replace-object: move replace_map to object store
[git/git.git] / replace_object.c
1 #include "cache.h"
2 #include "oidmap.h"
3 #include "object-store.h"
4 #include "replace-object.h"
5 #include "refs.h"
6 #include "repository.h"
7 #include "commit.h"
8
9 static int register_replace_ref(const char *refname,
10 const struct object_id *oid,
11 int flag, void *cb_data)
12 {
13 /* Get sha1 from refname */
14 const char *slash = strrchr(refname, '/');
15 const char *hash = slash ? slash + 1 : refname;
16 struct replace_object *repl_obj = xmalloc(sizeof(*repl_obj));
17
18 if (get_oid_hex(hash, &repl_obj->original.oid)) {
19 free(repl_obj);
20 warning("bad replace ref name: %s", refname);
21 return 0;
22 }
23
24 /* Copy sha1 from the read ref */
25 oidcpy(&repl_obj->replacement, oid);
26
27 /* Register new object */
28 if (oidmap_put(&the_repository->objects->replace_map, repl_obj))
29 die("duplicate replace ref: %s", refname);
30
31 return 0;
32 }
33
34 static void prepare_replace_object(void)
35 {
36 static int replace_object_prepared;
37
38 if (replace_object_prepared)
39 return;
40
41 for_each_replace_ref(register_replace_ref, NULL);
42 replace_object_prepared = 1;
43 if (!the_repository->objects->replace_map.map.tablesize)
44 check_replace_refs = 0;
45 }
46
47 /* We allow "recursive" replacement. Only within reason, though */
48 #define MAXREPLACEDEPTH 5
49
50 /*
51 * If a replacement for object oid has been set up, return the
52 * replacement object's name (replaced recursively, if necessary).
53 * The return value is either oid or a pointer to a
54 * permanently-allocated value. This function always respects replace
55 * references, regardless of the value of check_replace_refs.
56 */
57 const struct object_id *do_lookup_replace_object(const struct object_id *oid)
58 {
59 int depth = MAXREPLACEDEPTH;
60 const struct object_id *cur = oid;
61
62 prepare_replace_object();
63
64 /* Try to recursively replace the object */
65 while (depth-- > 0) {
66 struct replace_object *repl_obj =
67 oidmap_get(&the_repository->objects->replace_map, cur);
68 if (!repl_obj)
69 return cur;
70 cur = &repl_obj->replacement;
71 }
72 die("replace depth too high for object %s", oid_to_hex(oid));
73 }