6 #include "object-store-ll.h"
8 #include "repository.h"
12 struct flag_definition
{
22 static unsigned int parse_flags(const char *str
, struct flag_definition
*defs
)
24 struct string_list masks
= STRING_LIST_INIT_DUP
;
26 unsigned int result
= 0;
28 if (!strcmp(str
, "0"))
31 string_list_split(&masks
, str
, ',', 64);
32 for (; i
< masks
.nr
; i
++) {
33 const char *name
= masks
.items
[i
].string
;
34 struct flag_definition
*def
= defs
;
37 if (!strcmp(def
->name
, name
)) {
45 die("unknown flag \"%s\"", name
);
48 string_list_clear(&masks
, 0);
52 static struct flag_definition empty_flags
[] = { { NULL
, 0 } };
54 static const char *notnull(const char *arg
, const char *name
)
57 die("%s required", name
);
61 static unsigned int arg_flags(const char *arg
, const char *name
,
62 struct flag_definition
*defs
)
64 return parse_flags(notnull(arg
, name
), defs
);
67 static const char **get_store(const char **argv
, struct ref_store
**refs
)
72 die("ref store required");
73 } else if (!strcmp(argv
[0], "main")) {
74 *refs
= get_main_ref_store(the_repository
);
75 } else if (skip_prefix(argv
[0], "submodule:", &gitdir
)) {
76 struct strbuf sb
= STRBUF_INIT
;
79 ret
= strbuf_git_path_submodule(&sb
, gitdir
, "objects/");
81 die("strbuf_git_path_submodule failed: %d", ret
);
82 add_to_alternates_memory(sb
.buf
);
85 *refs
= get_submodule_ref_store(gitdir
);
86 } else if (skip_prefix(argv
[0], "worktree:", &gitdir
)) {
87 struct worktree
**p
, **worktrees
= get_worktrees();
89 for (p
= worktrees
; *p
; p
++) {
90 struct worktree
*wt
= *p
;
93 /* special case for main worktree */
94 if (!strcmp(gitdir
, "main"))
96 } else if (!strcmp(gitdir
, wt
->id
))
100 die("no such worktree: %s", gitdir
);
102 *refs
= get_worktree_ref_store(*p
);
103 free_worktrees(worktrees
);
105 die("unknown backend %s", argv
[0]);
110 /* consume store-specific optional arguments if needed */
115 static int cmd_create_symref(struct ref_store
*refs
, const char **argv
)
117 const char *refname
= notnull(*argv
++, "refname");
118 const char *target
= notnull(*argv
++, "target");
119 const char *logmsg
= *argv
++;
121 return refs_create_symref(refs
, refname
, target
, logmsg
);
124 static struct flag_definition transaction_flags
[] = {
125 FLAG_DEF(REF_NO_DEREF
),
126 FLAG_DEF(REF_FORCE_CREATE_REFLOG
),
127 FLAG_DEF(REF_SKIP_OID_VERIFICATION
),
128 FLAG_DEF(REF_SKIP_REFNAME_VERIFICATION
),
132 static int cmd_delete_refs(struct ref_store
*refs
, const char **argv
)
134 unsigned int flags
= arg_flags(*argv
++, "flags", transaction_flags
);
135 const char *msg
= *argv
++;
136 struct string_list refnames
= STRING_LIST_INIT_NODUP
;
140 string_list_append(&refnames
, *argv
++);
142 result
= refs_delete_refs(refs
, msg
, &refnames
, flags
);
143 string_list_clear(&refnames
, 0);
147 static int cmd_rename_ref(struct ref_store
*refs
, const char **argv
)
149 const char *oldref
= notnull(*argv
++, "oldref");
150 const char *newref
= notnull(*argv
++, "newref");
151 const char *logmsg
= *argv
++;
153 return refs_rename_ref(refs
, oldref
, newref
, logmsg
);
156 static int each_ref(const char *refname
, const struct object_id
*oid
,
157 int flags
, void *cb_data UNUSED
)
159 printf("%s %s 0x%x\n", oid_to_hex(oid
), refname
, flags
);
163 static int cmd_for_each_ref(struct ref_store
*refs
, const char **argv
)
165 const char *prefix
= notnull(*argv
++, "prefix");
167 return refs_for_each_ref_in(refs
, prefix
, each_ref
, NULL
);
170 static int cmd_for_each_ref__exclude(struct ref_store
*refs
, const char **argv
)
172 const char *prefix
= notnull(*argv
++, "prefix");
173 const char **exclude_patterns
= argv
;
175 return refs_for_each_fullref_in(refs
, prefix
, exclude_patterns
, each_ref
,
179 static int cmd_resolve_ref(struct ref_store
*refs
, const char **argv
)
181 struct object_id oid
= *null_oid();
182 const char *refname
= notnull(*argv
++, "refname");
183 int resolve_flags
= arg_flags(*argv
++, "resolve-flags", empty_flags
);
187 ref
= refs_resolve_ref_unsafe(refs
, refname
, resolve_flags
,
189 printf("%s %s 0x%x\n", oid_to_hex(&oid
), ref
? ref
: "(null)", flags
);
193 static int cmd_verify_ref(struct ref_store
*refs
, const char **argv
)
195 const char *refname
= notnull(*argv
++, "refname");
196 struct strbuf err
= STRBUF_INIT
;
199 ret
= refs_verify_refname_available(refs
, refname
, NULL
, NULL
, &err
);
205 static int each_reflog(const char *refname
, void *cb_data UNUSED
)
207 printf("%s\n", refname
);
211 static int cmd_for_each_reflog(struct ref_store
*refs
,
212 const char **argv UNUSED
)
214 return refs_for_each_reflog(refs
, each_reflog
, NULL
);
217 static int each_reflog_ent(struct object_id
*old_oid
, struct object_id
*new_oid
,
218 const char *committer
, timestamp_t timestamp
,
219 int tz
, const char *msg
, void *cb_data UNUSED
)
221 printf("%s %s %s %" PRItime
" %+05d%s%s", oid_to_hex(old_oid
),
222 oid_to_hex(new_oid
), committer
, timestamp
, tz
,
223 *msg
== '\n' ? "" : "\t", msg
);
227 static int cmd_for_each_reflog_ent(struct ref_store
*refs
, const char **argv
)
229 const char *refname
= notnull(*argv
++, "refname");
231 return refs_for_each_reflog_ent(refs
, refname
, each_reflog_ent
, refs
);
234 static int cmd_for_each_reflog_ent_reverse(struct ref_store
*refs
, const char **argv
)
236 const char *refname
= notnull(*argv
++, "refname");
238 return refs_for_each_reflog_ent_reverse(refs
, refname
, each_reflog_ent
, refs
);
241 static int cmd_reflog_exists(struct ref_store
*refs
, const char **argv
)
243 const char *refname
= notnull(*argv
++, "refname");
245 return !refs_reflog_exists(refs
, refname
);
248 static int cmd_create_reflog(struct ref_store
*refs
, const char **argv
)
250 const char *refname
= notnull(*argv
++, "refname");
251 struct strbuf err
= STRBUF_INIT
;
254 ret
= refs_create_reflog(refs
, refname
, &err
);
260 static int cmd_delete_reflog(struct ref_store
*refs
, const char **argv
)
262 const char *refname
= notnull(*argv
++, "refname");
264 return refs_delete_reflog(refs
, refname
);
267 static int cmd_delete_ref(struct ref_store
*refs
, const char **argv
)
269 const char *msg
= notnull(*argv
++, "msg");
270 const char *refname
= notnull(*argv
++, "refname");
271 const char *sha1_buf
= notnull(*argv
++, "old-sha1");
272 unsigned int flags
= arg_flags(*argv
++, "flags", transaction_flags
);
273 struct object_id old_oid
;
275 if (get_oid_hex(sha1_buf
, &old_oid
))
276 die("cannot parse %s as %s", sha1_buf
, the_hash_algo
->name
);
278 return refs_delete_ref(refs
, msg
, refname
, &old_oid
, flags
);
281 static int cmd_update_ref(struct ref_store
*refs
, const char **argv
)
283 const char *msg
= notnull(*argv
++, "msg");
284 const char *refname
= notnull(*argv
++, "refname");
285 const char *new_sha1_buf
= notnull(*argv
++, "new-sha1");
286 const char *old_sha1_buf
= notnull(*argv
++, "old-sha1");
287 unsigned int flags
= arg_flags(*argv
++, "flags", transaction_flags
);
288 struct object_id old_oid
, *old_oid_ptr
= NULL
;
289 struct object_id new_oid
;
292 if (get_oid_hex(old_sha1_buf
, &old_oid
))
293 die("cannot parse %s as %s", old_sha1_buf
, the_hash_algo
->name
);
294 old_oid_ptr
= &old_oid
;
296 if (get_oid_hex(new_sha1_buf
, &new_oid
))
297 die("cannot parse %s as %s", new_sha1_buf
, the_hash_algo
->name
);
299 return refs_update_ref(refs
, msg
, refname
,
300 &new_oid
, old_oid_ptr
,
301 flags
, UPDATE_REFS_DIE_ON_ERR
);
306 int (*func
)(struct ref_store
*refs
, const char **argv
);
309 static struct command commands
[] = {
310 { "create-symref", cmd_create_symref
},
311 { "delete-refs", cmd_delete_refs
},
312 { "rename-ref", cmd_rename_ref
},
313 { "for-each-ref", cmd_for_each_ref
},
314 { "for-each-ref--exclude", cmd_for_each_ref__exclude
},
315 { "resolve-ref", cmd_resolve_ref
},
316 { "verify-ref", cmd_verify_ref
},
317 { "for-each-reflog", cmd_for_each_reflog
},
318 { "for-each-reflog-ent", cmd_for_each_reflog_ent
},
319 { "for-each-reflog-ent-reverse", cmd_for_each_reflog_ent_reverse
},
320 { "reflog-exists", cmd_reflog_exists
},
321 { "create-reflog", cmd_create_reflog
},
322 { "delete-reflog", cmd_delete_reflog
},
324 * backend transaction functions can't be tested separately
326 { "delete-ref", cmd_delete_ref
},
327 { "update-ref", cmd_update_ref
},
331 int cmd__ref_store(int argc UNUSED
, const char **argv
)
333 struct ref_store
*refs
;
337 setup_git_directory();
339 argv
= get_store(argv
+ 1, &refs
);
343 die("ref function required");
344 for (cmd
= commands
; cmd
->name
; cmd
++) {
345 if (!strcmp(func
, cmd
->name
))
346 return cmd
->func(refs
, argv
);
348 die("unknown function %s", func
);