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