]> git.ipfire.org Git - thirdparty/git.git/blame - t/helper/test-ref-store.c
Git 2.47-rc0
[thirdparty/git.git] / t / helper / test-ref-store.c
CommitLineData
e7da9385
PS
1#define USE_THE_REPOSITORY_VARIABLE
2
65370d81 3#include "test-tool.h"
41771fa4 4#include "hex.h"
80f2a609 5#include "refs.h"
e38da487 6#include "setup.h"
fa099d23 7#include "worktree.h"
a034e910 8#include "object-store-ll.h"
c339932b 9#include "path.h"
23a3f0cb 10#include "repository.h"
a034e910 11#include "strbuf.h"
4fe42f32 12#include "revision.h"
80f2a609 13
cd2d40fb
HWN
14struct flag_definition {
15 const char *name;
16 uint64_t mask;
17};
18
19#define FLAG_DEF(x) \
20 { \
21#x, (x) \
22 }
23
24static unsigned int parse_flags(const char *str, struct flag_definition *defs)
25{
26 struct string_list masks = STRING_LIST_INIT_DUP;
27 int i = 0;
28 unsigned int result = 0;
29
30 if (!strcmp(str, "0"))
31 return 0;
32
33 string_list_split(&masks, str, ',', 64);
34 for (; i < masks.nr; i++) {
35 const char *name = masks.items[i].string;
36 struct flag_definition *def = defs;
37 int found = 0;
38 while (def->name) {
39 if (!strcmp(def->name, name)) {
40 result |= def->mask;
41 found = 1;
42 break;
43 }
44 def++;
45 }
46 if (!found)
47 die("unknown flag \"%s\"", name);
48 }
49
50 string_list_clear(&masks, 0);
51 return result;
52}
53
54static struct flag_definition empty_flags[] = { { NULL, 0 } };
55
80f2a609
NTND
56static const char *notnull(const char *arg, const char *name)
57{
58 if (!arg)
59 die("%s required", name);
60 return arg;
61}
62
cd2d40fb
HWN
63static unsigned int arg_flags(const char *arg, const char *name,
64 struct flag_definition *defs)
80f2a609 65{
cd2d40fb 66 return parse_flags(notnull(arg, name), defs);
80f2a609
NTND
67}
68
69static const char **get_store(const char **argv, struct ref_store **refs)
70{
71 const char *gitdir;
72
73 if (!argv[0]) {
74 die("ref store required");
75 } else if (!strcmp(argv[0], "main")) {
23a3f0cb 76 *refs = get_main_ref_store(the_repository);
80f2a609
NTND
77 } else if (skip_prefix(argv[0], "submodule:", &gitdir)) {
78 struct strbuf sb = STRBUF_INIT;
79 int ret;
80
81 ret = strbuf_git_path_submodule(&sb, gitdir, "objects/");
82 if (ret)
83 die("strbuf_git_path_submodule failed: %d", ret);
84 add_to_alternates_memory(sb.buf);
85 strbuf_release(&sb);
86
965f8991 87 *refs = repo_get_submodule_ref_store(the_repository, gitdir);
fa099d23 88 } else if (skip_prefix(argv[0], "worktree:", &gitdir)) {
03f2465b 89 struct worktree **p, **worktrees = get_worktrees();
fa099d23
NTND
90
91 for (p = worktrees; *p; p++) {
92 struct worktree *wt = *p;
93
94 if (!wt->id) {
95 /* special case for main worktree */
96 if (!strcmp(gitdir, "main"))
97 break;
98 } else if (!strcmp(gitdir, wt->id))
99 break;
100 }
101 if (!*p)
102 die("no such worktree: %s", gitdir);
103
104 *refs = get_worktree_ref_store(*p);
34e69128 105 free_worktrees(worktrees);
80f2a609
NTND
106 } else
107 die("unknown backend %s", argv[0]);
108
109 if (!*refs)
110 die("no ref store");
111
112 /* consume store-specific optional arguments if needed */
113
114 return argv + 1;
115}
116
80f2a609
NTND
117static int cmd_create_symref(struct ref_store *refs, const char **argv)
118{
119 const char *refname = notnull(*argv++, "refname");
120 const char *target = notnull(*argv++, "target");
121 const char *logmsg = *argv++;
122
f151dfe3 123 return refs_update_symref(refs, refname, target, logmsg);
80f2a609
NTND
124}
125
cd2d40fb
HWN
126static struct flag_definition transaction_flags[] = {
127 FLAG_DEF(REF_NO_DEREF),
128 FLAG_DEF(REF_FORCE_CREATE_REFLOG),
e9706a18 129 FLAG_DEF(REF_SKIP_OID_VERIFICATION),
3c966c7b 130 FLAG_DEF(REF_SKIP_REFNAME_VERIFICATION),
fbd1a693 131 FLAG_DEF(REF_SKIP_CREATE_REFLOG),
cd2d40fb
HWN
132 { NULL, 0 }
133};
134
80f2a609
NTND
135static int cmd_delete_refs(struct ref_store *refs, const char **argv)
136{
cd2d40fb 137 unsigned int flags = arg_flags(*argv++, "flags", transaction_flags);
64da4199 138 const char *msg = *argv++;
80f2a609 139 struct string_list refnames = STRING_LIST_INIT_NODUP;
df25a19d 140 int result;
80f2a609
NTND
141
142 while (*argv)
143 string_list_append(&refnames, *argv++);
144
df25a19d
HWN
145 result = refs_delete_refs(refs, msg, &refnames, flags);
146 string_list_clear(&refnames, 0);
147 return result;
80f2a609
NTND
148}
149
150static int cmd_rename_ref(struct ref_store *refs, const char **argv)
151{
152 const char *oldref = notnull(*argv++, "oldref");
153 const char *newref = notnull(*argv++, "newref");
154 const char *logmsg = *argv++;
155
156 return refs_rename_ref(refs, oldref, newref, logmsg);
157}
158
e8207717 159static int each_ref(const char *refname, const char *referent UNUSED, const struct object_id *oid,
5cf88fd8 160 int flags, void *cb_data UNUSED)
80f2a609
NTND
161{
162 printf("%s %s 0x%x\n", oid_to_hex(oid), refname, flags);
163 return 0;
164}
165
166static int cmd_for_each_ref(struct ref_store *refs, const char **argv)
167{
168 const char *prefix = notnull(*argv++, "prefix");
169
170 return refs_for_each_ref_in(refs, prefix, each_ref, NULL);
171}
172
59c35fac
TB
173static int cmd_for_each_ref__exclude(struct ref_store *refs, const char **argv)
174{
175 const char *prefix = notnull(*argv++, "prefix");
176 const char **exclude_patterns = argv;
177
178 return refs_for_each_fullref_in(refs, prefix, exclude_patterns, each_ref,
179 NULL);
180}
181
80f2a609
NTND
182static int cmd_resolve_ref(struct ref_store *refs, const char **argv)
183{
0221eb86 184 struct object_id oid = *null_oid();
80f2a609 185 const char *refname = notnull(*argv++, "refname");
cd2d40fb 186 int resolve_flags = arg_flags(*argv++, "resolve-flags", empty_flags);
80f2a609
NTND
187 int flags;
188 const char *ref;
189
f1da24ca 190 ref = refs_resolve_ref_unsafe(refs, refname, resolve_flags,
ce14de03 191 &oid, &flags);
e7e456f5 192 printf("%s %s 0x%x\n", oid_to_hex(&oid), ref ? ref : "(null)", flags);
80f2a609
NTND
193 return ref ? 0 : 1;
194}
195
196static int cmd_verify_ref(struct ref_store *refs, const char **argv)
197{
198 const char *refname = notnull(*argv++, "refname");
199 struct strbuf err = STRBUF_INIT;
200 int ret;
201
202 ret = refs_verify_refname_available(refs, refname, NULL, NULL, &err);
203 if (err.len)
204 puts(err.buf);
205 return ret;
206}
207
31f89839
PS
208static int each_reflog(const char *refname, void *cb_data UNUSED)
209{
210 printf("%s\n", refname);
211 return 0;
212}
213
126e3b3d
JK
214static int cmd_for_each_reflog(struct ref_store *refs,
215 const char **argv UNUSED)
80f2a609 216{
31f89839 217 return refs_for_each_reflog(refs, each_reflog, NULL);
80f2a609
NTND
218}
219
31f89839
PS
220static int each_reflog_ent(struct object_id *old_oid, struct object_id *new_oid,
221 const char *committer, timestamp_t timestamp,
222 int tz, const char *msg, void *cb_data UNUSED)
80f2a609 223{
3474b602
HWN
224 printf("%s %s %s %" PRItime " %+05d%s%s", oid_to_hex(old_oid),
225 oid_to_hex(new_oid), committer, timestamp, tz,
226 *msg == '\n' ? "" : "\t", msg);
80f2a609
NTND
227 return 0;
228}
229
230static int cmd_for_each_reflog_ent(struct ref_store *refs, const char **argv)
231{
232 const char *refname = notnull(*argv++, "refname");
233
31f89839 234 return refs_for_each_reflog_ent(refs, refname, each_reflog_ent, refs);
80f2a609
NTND
235}
236
237static int cmd_for_each_reflog_ent_reverse(struct ref_store *refs, const char **argv)
238{
239 const char *refname = notnull(*argv++, "refname");
240
31f89839 241 return refs_for_each_reflog_ent_reverse(refs, refname, each_reflog_ent, refs);
80f2a609
NTND
242}
243
244static int cmd_reflog_exists(struct ref_store *refs, const char **argv)
245{
246 const char *refname = notnull(*argv++, "refname");
247
248 return !refs_reflog_exists(refs, refname);
249}
250
251static int cmd_create_reflog(struct ref_store *refs, const char **argv)
252{
253 const char *refname = notnull(*argv++, "refname");
80f2a609
NTND
254 struct strbuf err = STRBUF_INIT;
255 int ret;
256
7b089120 257 ret = refs_create_reflog(refs, refname, &err);
80f2a609
NTND
258 if (err.len)
259 puts(err.buf);
260 return ret;
261}
262
263static int cmd_delete_reflog(struct ref_store *refs, const char **argv)
264{
265 const char *refname = notnull(*argv++, "refname");
266
267 return refs_delete_reflog(refs, refname);
268}
269
80f2a609
NTND
270static int cmd_delete_ref(struct ref_store *refs, const char **argv)
271{
272 const char *msg = notnull(*argv++, "msg");
273 const char *refname = notnull(*argv++, "refname");
274 const char *sha1_buf = notnull(*argv++, "old-sha1");
cd2d40fb 275 unsigned int flags = arg_flags(*argv++, "flags", transaction_flags);
2616a5e5 276 struct object_id old_oid;
80f2a609 277
2616a5e5 278 if (get_oid_hex(sha1_buf, &old_oid))
57eb3681 279 die("cannot parse %s as %s", sha1_buf, the_hash_algo->name);
80f2a609 280
2616a5e5 281 return refs_delete_ref(refs, msg, refname, &old_oid, flags);
80f2a609
NTND
282}
283
284static int cmd_update_ref(struct ref_store *refs, const char **argv)
285{
286 const char *msg = notnull(*argv++, "msg");
287 const char *refname = notnull(*argv++, "refname");
3c27e2e0 288 const char *new_sha1_buf = notnull(*argv++, "new-sha1");
80f2a609 289 const char *old_sha1_buf = notnull(*argv++, "old-sha1");
cd2d40fb 290 unsigned int flags = arg_flags(*argv++, "flags", transaction_flags);
9ddd5b88 291 struct object_id old_oid, *old_oid_ptr = NULL;
ae077771 292 struct object_id new_oid;
80f2a609 293
9ddd5b88
PS
294 if (*old_sha1_buf) {
295 if (get_oid_hex(old_sha1_buf, &old_oid))
296 die("cannot parse %s as %s", old_sha1_buf, the_hash_algo->name);
297 old_oid_ptr = &old_oid;
298 }
57eb3681
HWN
299 if (get_oid_hex(new_sha1_buf, &new_oid))
300 die("cannot parse %s as %s", new_sha1_buf, the_hash_algo->name);
80f2a609
NTND
301
302 return refs_update_ref(refs, msg, refname,
9ddd5b88 303 &new_oid, old_oid_ptr,
80f2a609
NTND
304 flags, UPDATE_REFS_DIE_ON_ERR);
305}
306
307struct command {
308 const char *name;
309 int (*func)(struct ref_store *refs, const char **argv);
310};
311
312static struct command commands[] = {
80f2a609
NTND
313 { "create-symref", cmd_create_symref },
314 { "delete-refs", cmd_delete_refs },
315 { "rename-ref", cmd_rename_ref },
316 { "for-each-ref", cmd_for_each_ref },
59c35fac 317 { "for-each-ref--exclude", cmd_for_each_ref__exclude },
80f2a609
NTND
318 { "resolve-ref", cmd_resolve_ref },
319 { "verify-ref", cmd_verify_ref },
320 { "for-each-reflog", cmd_for_each_reflog },
321 { "for-each-reflog-ent", cmd_for_each_reflog_ent },
322 { "for-each-reflog-ent-reverse", cmd_for_each_reflog_ent_reverse },
323 { "reflog-exists", cmd_reflog_exists },
324 { "create-reflog", cmd_create_reflog },
325 { "delete-reflog", cmd_delete_reflog },
80f2a609
NTND
326 /*
327 * backend transaction functions can't be tested separately
328 */
329 { "delete-ref", cmd_delete_ref },
330 { "update-ref", cmd_update_ref },
331 { NULL, NULL }
332};
333
126e3b3d 334int cmd__ref_store(int argc UNUSED, const char **argv)
80f2a609
NTND
335{
336 struct ref_store *refs;
337 const char *func;
338 struct command *cmd;
339
340 setup_git_directory();
341
342 argv = get_store(argv + 1, &refs);
343
344 func = *argv++;
345 if (!func)
346 die("ref function required");
347 for (cmd = commands; cmd->name; cmd++) {
348 if (!strcmp(func, cmd->name))
349 return cmd->func(refs, argv);
350 }
351 die("unknown function %s", func);
352 return 0;
353}