1 #include "cache.h"
2 #include "notes-cache.h"
3 #include "object-store.h"
4 #include "repository.h"
5 #include "commit.h"
6 #include "refs.h"
7
notes_cache_match_validity(struct repository * r,const char * ref,const char * validity)8 static int notes_cache_match_validity(struct repository *r,
9 const char *ref,
10 const char *validity)
11 {
12 struct object_id oid;
13 struct commit *commit;
14 struct pretty_print_context pretty_ctx;
15 struct strbuf msg = STRBUF_INIT;
16 int ret;
17
18 if (read_ref(ref, &oid) < 0)
19 return 0;
20
21 commit = lookup_commit_reference_gently(r, &oid, 1);
22 if (!commit)
23 return 0;
24
25 memset(&pretty_ctx, 0, sizeof(pretty_ctx));
26 format_commit_message(commit, "%s", &msg, &pretty_ctx);
27 strbuf_trim(&msg);
28
29 ret = !strcmp(msg.buf, validity);
30 strbuf_release(&msg);
31
32 return ret;
33 }
34
notes_cache_init(struct repository * r,struct notes_cache * c,const char * name,const char * validity)35 void notes_cache_init(struct repository *r, struct notes_cache *c,
36 const char *name, const char *validity)
37 {
38 struct strbuf ref = STRBUF_INIT;
39 int flags = NOTES_INIT_WRITABLE;
40
41 memset(c, 0, sizeof(*c));
42 c->validity = xstrdup(validity);
43
44 strbuf_addf(&ref, "refs/notes/%s", name);
45 if (!notes_cache_match_validity(r, ref.buf, validity))
46 flags |= NOTES_INIT_EMPTY;
47 init_notes(&c->tree, ref.buf, combine_notes_overwrite, flags);
48 strbuf_release(&ref);
49 }
50
notes_cache_write(struct notes_cache * c)51 int notes_cache_write(struct notes_cache *c)
52 {
53 struct object_id tree_oid, commit_oid;
54
55 if (!c || !c->tree.initialized || !c->tree.update_ref ||
56 !*c->tree.update_ref)
57 return -1;
58 if (!c->tree.dirty)
59 return 0;
60
61 if (write_notes_tree(&c->tree, &tree_oid))
62 return -1;
63 if (commit_tree(c->validity, strlen(c->validity), &tree_oid, NULL,
64 &commit_oid, NULL, NULL) < 0)
65 return -1;
66 if (update_ref("update notes cache", c->tree.update_ref, &commit_oid,
67 NULL, 0, UPDATE_REFS_QUIET_ON_ERR) < 0)
68 return -1;
69
70 return 0;
71 }
72
notes_cache_get(struct notes_cache * c,struct object_id * key_oid,size_t * outsize)73 char *notes_cache_get(struct notes_cache *c, struct object_id *key_oid,
74 size_t *outsize)
75 {
76 const struct object_id *value_oid;
77 enum object_type type;
78 char *value;
79 unsigned long size;
80
81 value_oid = get_note(&c->tree, key_oid);
82 if (!value_oid)
83 return NULL;
84 value = read_object_file(value_oid, &type, &size);
85
86 *outsize = size;
87 return value;
88 }
89
notes_cache_put(struct notes_cache * c,struct object_id * key_oid,const char * data,size_t size)90 int notes_cache_put(struct notes_cache *c, struct object_id *key_oid,
91 const char *data, size_t size)
92 {
93 struct object_id value_oid;
94
95 if (write_object_file(data, size, "blob", &value_oid) < 0)
96 return -1;
97 return add_note(&c->tree, key_oid, &value_oid, NULL);
98 }
99