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