branch/checkout --track: Ensure that upstream branch is indeed a branch
[git] / notes-cache.c
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 }