[thirdparty/git.git] / t / helper / test-ref-store.c

#include "cache.h"
#include "refs.h"
#include "worktree.h"

static const char *notnull(const char *arg, const char *name)
{
	if (!arg)
		die("%s required", name);
	return arg;
}

static unsigned int arg_flags(const char *arg, const char *name)
{
	return atoi(notnull(arg, name));
}

static const char **get_store(const char **argv, struct ref_store **refs)
{
	const char *gitdir;

	if (!argv[0]) {
		die("ref store required");
	} else if (!strcmp(argv[0], "main")) {
		*refs = get_main_ref_store();
	} else if (skip_prefix(argv[0], "submodule:", &gitdir)) {
		struct strbuf sb = STRBUF_INIT;
		int ret;

		ret = strbuf_git_path_submodule(&sb, gitdir, "objects/");
		if (ret)
			die("strbuf_git_path_submodule failed: %d", ret);
		add_to_alternates_memory(sb.buf);
		strbuf_release(&sb);

		*refs = get_submodule_ref_store(gitdir);
	} else if (skip_prefix(argv[0], "worktree:", &gitdir)) {
		struct worktree **p, **worktrees = get_worktrees(0);

		for (p = worktrees; *p; p++) {
			struct worktree *wt = *p;

			if (!wt->id) {
				/* special case for main worktree */
				if (!strcmp(gitdir, "main"))
					break;
			} else if (!strcmp(gitdir, wt->id))
				break;
		}
		if (!*p)
			die("no such worktree: %s", gitdir);

		*refs = get_worktree_ref_store(*p);
	} else
		die("unknown backend %s", argv[0]);

	if (!*refs)
		die("no ref store");

	/* consume store-specific optional arguments if needed */

	return argv + 1;
}


static int cmd_pack_refs(struct ref_store *refs, const char **argv)
{
	unsigned int flags = arg_flags(*argv++, "flags");

	return refs_pack_refs(refs, flags);
}

static int cmd_peel_ref(struct ref_store *refs, const char **argv)
{
	const char *refname = notnull(*argv++, "refname");
	struct object_id oid;
	int ret;

	ret = refs_peel_ref(refs, refname, &oid);
	if (!ret)
		puts(oid_to_hex(&oid));
	return ret;
}

static int cmd_create_symref(struct ref_store *refs, const char **argv)
{
	const char *refname = notnull(*argv++, "refname");
	const char *target = notnull(*argv++, "target");
	const char *logmsg = *argv++;

	return refs_create_symref(refs, refname, target, logmsg);
}

static int cmd_delete_refs(struct ref_store *refs, const char **argv)
{
	unsigned int flags = arg_flags(*argv++, "flags");
	const char *msg = *argv++;
	struct string_list refnames = STRING_LIST_INIT_NODUP;

	while (*argv)
		string_list_append(&refnames, *argv++);

	return refs_delete_refs(refs, msg, &refnames, flags);
}

static int cmd_rename_ref(struct ref_store *refs, const char **argv)
{
	const char *oldref = notnull(*argv++, "oldref");
	const char *newref = notnull(*argv++, "newref");
	const char *logmsg = *argv++;

	return refs_rename_ref(refs, oldref, newref, logmsg);
}

static int each_ref(const char *refname, const struct object_id *oid,
		    int flags, void *cb_data)
{
	printf("%s %s 0x%x\n", oid_to_hex(oid), refname, flags);
	return 0;
}

static int cmd_for_each_ref(struct ref_store *refs, const char **argv)
{
	const char *prefix = notnull(*argv++, "prefix");

	return refs_for_each_ref_in(refs, prefix, each_ref, NULL);
}

static int cmd_resolve_ref(struct ref_store *refs, const char **argv)
{
	struct object_id oid;
	const char *refname = notnull(*argv++, "refname");
	int resolve_flags = arg_flags(*argv++, "resolve-flags");
	int flags;
	const char *ref;

	ref = refs_resolve_ref_unsafe(refs, refname, resolve_flags,
				      &oid, &flags);
	printf("%s %s 0x%x\n", oid_to_hex(&oid), ref ? ref : "(null)", flags);
	return ref ? 0 : 1;
}

static int cmd_verify_ref(struct ref_store *refs, const char **argv)
{
	const char *refname = notnull(*argv++, "refname");
	struct strbuf err = STRBUF_INIT;
	int ret;

	ret = refs_verify_refname_available(refs, refname, NULL, NULL, &err);
	if (err.len)
		puts(err.buf);
	return ret;
}

static int cmd_for_each_reflog(struct ref_store *refs, const char **argv)
{
	return refs_for_each_reflog(refs, each_ref, NULL);
}

static int each_reflog(struct object_id *old_oid, struct object_id *new_oid,
		       const char *committer, timestamp_t timestamp,
		       int tz, const char *msg, void *cb_data)
{
	printf("%s %s %s %"PRItime" %d %s\n",
	       oid_to_hex(old_oid), oid_to_hex(new_oid),
	       committer, timestamp, tz, msg);
	return 0;
}

static int cmd_for_each_reflog_ent(struct ref_store *refs, const char **argv)
{
	const char *refname = notnull(*argv++, "refname");

	return refs_for_each_reflog_ent(refs, refname, each_reflog, refs);
}

static int cmd_for_each_reflog_ent_reverse(struct ref_store *refs, const char **argv)
{
	const char *refname = notnull(*argv++, "refname");

	return refs_for_each_reflog_ent_reverse(refs, refname, each_reflog, refs);
}

static int cmd_reflog_exists(struct ref_store *refs, const char **argv)
{
	const char *refname = notnull(*argv++, "refname");

	return !refs_reflog_exists(refs, refname);
}

static int cmd_create_reflog(struct ref_store *refs, const char **argv)
{
	const char *refname = notnull(*argv++, "refname");
	int force_create = arg_flags(*argv++, "force-create");
	struct strbuf err = STRBUF_INIT;
	int ret;

	ret = refs_create_reflog(refs, refname, force_create, &err);
	if (err.len)
		puts(err.buf);
	return ret;
}

static int cmd_delete_reflog(struct ref_store *refs, const char **argv)
{
	const char *refname = notnull(*argv++, "refname");

	return refs_delete_reflog(refs, refname);
}

static int cmd_reflog_expire(struct ref_store *refs, const char **argv)
{
	die("not supported yet");
}

static int cmd_delete_ref(struct ref_store *refs, const char **argv)
{
	const char *msg = notnull(*argv++, "msg");
	const char *refname = notnull(*argv++, "refname");
	const char *sha1_buf = notnull(*argv++, "old-sha1");
	unsigned int flags = arg_flags(*argv++, "flags");
	struct object_id old_oid;

	if (get_oid_hex(sha1_buf, &old_oid))
		die("not sha-1");

	return refs_delete_ref(refs, msg, refname, &old_oid, flags);
}

static int cmd_update_ref(struct ref_store *refs, const char **argv)
{
	const char *msg = notnull(*argv++, "msg");
	const char *refname = notnull(*argv++, "refname");
	const char *new_sha1_buf = notnull(*argv++, "old-sha1");
	const char *old_sha1_buf = notnull(*argv++, "old-sha1");
	unsigned int flags = arg_flags(*argv++, "flags");
	struct object_id old_oid;
	struct object_id new_oid;

	if (get_oid_hex(old_sha1_buf, &old_oid) ||
	    get_oid_hex(new_sha1_buf, &new_oid))
		die("not sha-1");

	return refs_update_ref(refs, msg, refname,
			       &new_oid, &old_oid,
			       flags, UPDATE_REFS_DIE_ON_ERR);
}

struct command {
	const char *name;
	int (*func)(struct ref_store *refs, const char **argv);
};

static struct command commands[] = {
	{ "pack-refs", cmd_pack_refs },
	{ "peel-ref", cmd_peel_ref },
	{ "create-symref", cmd_create_symref },
	{ "delete-refs", cmd_delete_refs },
	{ "rename-ref", cmd_rename_ref },
	{ "for-each-ref", cmd_for_each_ref },
	{ "resolve-ref", cmd_resolve_ref },
	{ "verify-ref", cmd_verify_ref },
	{ "for-each-reflog", cmd_for_each_reflog },
	{ "for-each-reflog-ent", cmd_for_each_reflog_ent },
	{ "for-each-reflog-ent-reverse", cmd_for_each_reflog_ent_reverse },
	{ "reflog-exists", cmd_reflog_exists },
	{ "create-reflog", cmd_create_reflog },
	{ "delete-reflog", cmd_delete_reflog },
	{ "reflog-expire", cmd_reflog_expire },
	/*
	 * backend transaction functions can't be tested separately
	 */
	{ "delete-ref", cmd_delete_ref },
	{ "update-ref", cmd_update_ref },
	{ NULL, NULL }
};

int cmd_main(int argc, const char **argv)
{
	struct ref_store *refs;
	const char *func;
	struct command *cmd;

	setup_git_directory();

	argv = get_store(argv + 1, &refs);

	func = *argv++;
	if (!func)
		die("ref function required");
	for (cmd = commands; cmd->name; cmd++) {
		if (!strcmp(func, cmd->name))
			return cmd->func(refs, argv);
	}
	die("unknown function %s", func);
	return 0;
}
Commit	Line	Data
80f2a609 NTND	1	#include "cache.h"
80f2a609 NTND	2	#include "refs.h"
fa099d23	3	#include "worktree.h"
80f2a609 NTND	4
	5	static const char notnull(const char arg, const char *name)
	6	{
	7	if (!arg)
	8	die("%s required", name);
	9	return arg;
	10	}
	11
	12	static unsigned int arg_flags(const char arg, const char name)
	13	{
	14	return atoi(notnull(arg, name));
	15	}
	16
	17	static const char get_store(const char argv, struct ref_store **refs)
	18	{
	19	const char *gitdir;
	20
	21	if (!argv[0]) {
	22	die("ref store required");
	23	} else if (!strcmp(argv[0], "main")) {
	24	*refs = get_main_ref_store();
	25	} else if (skip_prefix(argv[0], "submodule:", &gitdir)) {
	26	struct strbuf sb = STRBUF_INIT;
	27	int ret;
	28
	29	ret = strbuf_git_path_submodule(&sb, gitdir, "objects/");
	30	if (ret)
	31	die("strbuf_git_path_submodule failed: %d", ret);
	32	add_to_alternates_memory(sb.buf);
	33	strbuf_release(&sb);
	34
	35	*refs = get_submodule_ref_store(gitdir);
fa099d23 NTND	36	} else if (skip_prefix(argv[0], "worktree:", &gitdir)) {
	37	struct worktree p, worktrees = get_worktrees(0);
	38
	39	for (p = worktrees; *p; p++) {
	40	struct worktree wt = p;
	41
	42	if (!wt->id) {
	43	/* special case for main worktree */
	44	if (!strcmp(gitdir, "main"))
	45	break;
	46	} else if (!strcmp(gitdir, wt->id))
	47	break;
	48	}
	49	if (!*p)
	50	die("no such worktree: %s", gitdir);
	51
	52	refs = get_worktree_ref_store(p);
80f2a609 NTND	53	} else
	54	die("unknown backend %s", argv[0]);
	55
	56	if (!*refs)
	57	die("no ref store");
	58
	59	/* consume store-specific optional arguments if needed */
	60
	61	return argv + 1;
	62	}
	63
	64
	65	static int cmd_pack_refs(struct ref_store refs, const char *argv)
	66	{
	67	unsigned int flags = arg_flags(*argv++, "flags");
	68
	69	return refs_pack_refs(refs, flags);
	70	}
	71
	72	static int cmd_peel_ref(struct ref_store refs, const char *argv)
	73	{
	74	const char refname = notnull(argv++, "refname");
b420d909	75	struct object_id oid;
80f2a609 NTND	76	int ret;
80f2a609 NTND	77
b420d909	78	ret = refs_peel_ref(refs, refname, &oid);
80f2a609	79	if (!ret)
b420d909	80	puts(oid_to_hex(&oid));
80f2a609 NTND	81	return ret;
	82	}
	83
	84	static int cmd_create_symref(struct ref_store refs, const char *argv)
	85	{
	86	const char refname = notnull(argv++, "refname");
	87	const char target = notnull(argv++, "target");
	88	const char logmsg = argv++;
	89
	90	return refs_create_symref(refs, refname, target, logmsg);
	91	}
	92
	93	static int cmd_delete_refs(struct ref_store refs, const char *argv)
	94	{
	95	unsigned int flags = arg_flags(*argv++, "flags");
64da4199	96	const char msg = argv++;
80f2a609 NTND	97	struct string_list refnames = STRING_LIST_INIT_NODUP;
	98
	99	while (*argv)
	100	string_list_append(&refnames, *argv++);
	101
64da4199	102	return refs_delete_refs(refs, msg, &refnames, flags);
80f2a609 NTND	103	}
	104
	105	static int cmd_rename_ref(struct ref_store refs, const char *argv)
	106	{
	107	const char oldref = notnull(argv++, "oldref");
	108	const char newref = notnull(argv++, "newref");
	109	const char logmsg = argv++;
	110
	111	return refs_rename_ref(refs, oldref, newref, logmsg);
	112	}
	113
	114	static int each_ref(const char refname, const struct object_id oid,
	115	int flags, void *cb_data)
	116	{
	117	printf("%s %s 0x%x\n", oid_to_hex(oid), refname, flags);
	118	return 0;
	119	}
	120
	121	static int cmd_for_each_ref(struct ref_store refs, const char *argv)
	122	{
	123	const char prefix = notnull(argv++, "prefix");
	124
	125	return refs_for_each_ref_in(refs, prefix, each_ref, NULL);
	126	}
	127
	128	static int cmd_resolve_ref(struct ref_store refs, const char *argv)
	129	{
49e61479	130	struct object_id oid;
80f2a609 NTND	131	const char refname = notnull(argv++, "refname");
	132	int resolve_flags = arg_flags(*argv++, "resolve-flags");
	133	int flags;
	134	const char *ref;
	135
	136	ref = refs_resolve_ref_unsafe(refs, refname, resolve_flags,
49e61479	137	&oid, &flags);
e7e456f5	138	printf("%s %s 0x%x\n", oid_to_hex(&oid), ref ? ref : "(null)", flags);
80f2a609 NTND	139	return ref ? 0 : 1;
	140	}
	141
	142	static int cmd_verify_ref(struct ref_store refs, const char *argv)
	143	{
	144	const char refname = notnull(argv++, "refname");
	145	struct strbuf err = STRBUF_INIT;
	146	int ret;
	147
	148	ret = refs_verify_refname_available(refs, refname, NULL, NULL, &err);
	149	if (err.len)
	150	puts(err.buf);
	151	return ret;
	152	}
	153
	154	static int cmd_for_each_reflog(struct ref_store refs, const char *argv)
	155	{
	156	return refs_for_each_reflog(refs, each_ref, NULL);
	157	}
	158
	159	static int each_reflog(struct object_id old_oid, struct object_id new_oid,
dddbad72	160	const char *committer, timestamp_t timestamp,
80f2a609 NTND	161	int tz, const char msg, void cb_data)
80f2a609 NTND	162	{
cb71f8bd	163	printf("%s %s %s %"PRItime" %d %s\n",
80f2a609 NTND	164	oid_to_hex(old_oid), oid_to_hex(new_oid),
	165	committer, timestamp, tz, msg);
	166	return 0;
	167	}
	168
	169	static int cmd_for_each_reflog_ent(struct ref_store refs, const char *argv)
	170	{
	171	const char refname = notnull(argv++, "refname");
	172
	173	return refs_for_each_reflog_ent(refs, refname, each_reflog, refs);
	174	}
	175
	176	static int cmd_for_each_reflog_ent_reverse(struct ref_store refs, const char *argv)
	177	{
	178	const char refname = notnull(argv++, "refname");
	179
	180	return refs_for_each_reflog_ent_reverse(refs, refname, each_reflog, refs);
	181	}
	182
	183	static int cmd_reflog_exists(struct ref_store refs, const char *argv)
	184	{
	185	const char refname = notnull(argv++, "refname");
	186
	187	return !refs_reflog_exists(refs, refname);
	188	}
	189
	190	static int cmd_create_reflog(struct ref_store refs, const char *argv)
	191	{
	192	const char refname = notnull(argv++, "refname");
	193	int force_create = arg_flags(*argv++, "force-create");
	194	struct strbuf err = STRBUF_INIT;
	195	int ret;
	196
	197	ret = refs_create_reflog(refs, refname, force_create, &err);
	198	if (err.len)
	199	puts(err.buf);
	200	return ret;
	201	}
	202
	203	static int cmd_delete_reflog(struct ref_store refs, const char *argv)
	204	{
	205	const char refname = notnull(argv++, "refname");
	206
	207	return refs_delete_reflog(refs, refname);
	208	}
	209
	210	static int cmd_reflog_expire(struct ref_store refs, const char *argv)
	211	{
	212	die("not supported yet");
	213	}
	214
	215	static int cmd_delete_ref(struct ref_store refs, const char *argv)
	216	{
	217	const char msg = notnull(argv++, "msg");
	218	const char refname = notnull(argv++, "refname");
	219	const char sha1_buf = notnull(argv++, "old-sha1");
	220	unsigned int flags = arg_flags(*argv++, "flags");
2616a5e5	221	struct object_id old_oid;
80f2a609	222
2616a5e5	223	if (get_oid_hex(sha1_buf, &old_oid))
80f2a609 NTND	224	die("not sha-1");
80f2a609 NTND	225
2616a5e5	226	return refs_delete_ref(refs, msg, refname, &old_oid, flags);
80f2a609 NTND	227	}
	228
	229	static int cmd_update_ref(struct ref_store refs, const char *argv)
	230	{
	231	const char msg = notnull(argv++, "msg");
	232	const char refname = notnull(argv++, "refname");
	233	const char new_sha1_buf = notnull(argv++, "old-sha1");
	234	const char old_sha1_buf = notnull(argv++, "old-sha1");
	235	unsigned int flags = arg_flags(*argv++, "flags");
ae077771	236	struct object_id old_oid;
ae077771	237	struct object_id new_oid;
80f2a609	238
ae077771	239	if (get_oid_hex(old_sha1_buf, &old_oid) \|\|
ae077771	240	get_oid_hex(new_sha1_buf, &new_oid))
80f2a609 NTND	241	die("not sha-1");
	242
	243	return refs_update_ref(refs, msg, refname,
ae077771	244	&new_oid, &old_oid,
80f2a609 NTND	245	flags, UPDATE_REFS_DIE_ON_ERR);
	246	}
	247
	248	struct command {
	249	const char *name;
	250	int (func)(struct ref_store refs, const char **argv);
	251	};
	252
	253	static struct command commands[] = {
	254	{ "pack-refs", cmd_pack_refs },
	255	{ "peel-ref", cmd_peel_ref },
	256	{ "create-symref", cmd_create_symref },
	257	{ "delete-refs", cmd_delete_refs },
	258	{ "rename-ref", cmd_rename_ref },
	259	{ "for-each-ref", cmd_for_each_ref },
	260	{ "resolve-ref", cmd_resolve_ref },
	261	{ "verify-ref", cmd_verify_ref },
	262	{ "for-each-reflog", cmd_for_each_reflog },
	263	{ "for-each-reflog-ent", cmd_for_each_reflog_ent },
	264	{ "for-each-reflog-ent-reverse", cmd_for_each_reflog_ent_reverse },
	265	{ "reflog-exists", cmd_reflog_exists },
	266	{ "create-reflog", cmd_create_reflog },
	267	{ "delete-reflog", cmd_delete_reflog },
	268	{ "reflog-expire", cmd_reflog_expire },
	269	/*
	270	* backend transaction functions can't be tested separately
	271	*/
	272	{ "delete-ref", cmd_delete_ref },
	273	{ "update-ref", cmd_update_ref },
	274	{ NULL, NULL }
	275	};
	276
	277	int cmd_main(int argc, const char **argv)
	278	{
	279	struct ref_store *refs;
	280	const char *func;
	281	struct command *cmd;
	282
	283	setup_git_directory();
	284
	285	argv = get_store(argv + 1, &refs);
	286
	287	func = *argv++;
	288	if (!func)
	289	die("ref function required");
	290	for (cmd = commands; cmd->name; cmd++) {
	291	if (!strcmp(func, cmd->name))
	292	return cmd->func(refs, argv);
	293	}
	294	die("unknown function %s", func);
	295	return 0;
	296	}