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