Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 1 | #include "cache.h" |
| 2 | #include "notes-cache.h" |
Stefan Beller | cbd53a2 | 2018-05-15 16:42:15 -0700 | [diff] [blame] | 3 | #include "object-store.h" |
Stefan Beller | 21e1ee8 | 2018-06-28 18:21:57 -0700 | [diff] [blame] | 4 | #include "repository.h" |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 5 | #include "commit.h" |
| 6 | #include "refs.h" |
| 7 | |
Nguyễn Thái Ngọc Duy | bd7ad45 | 2018-11-10 06:49:06 +0100 | [diff] [blame] | 8 | static int notes_cache_match_validity(struct repository *r, |
| 9 | const char *ref, |
| 10 | const char *validity) |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 11 | { |
brian m. carlson | 569aa37 | 2017-05-06 22:09:58 +0000 | [diff] [blame] | 12 | struct object_id oid; |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 13 | struct commit *commit; |
| 14 | struct pretty_print_context pretty_ctx; |
| 15 | struct strbuf msg = STRBUF_INIT; |
| 16 | int ret; |
| 17 | |
brian m. carlson | 34c290a | 2017-10-15 22:06:56 +0000 | [diff] [blame] | 18 | if (read_ref(ref, &oid) < 0) |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 19 | return 0; |
| 20 | |
Nguyễn Thái Ngọc Duy | bd7ad45 | 2018-11-10 06:49:06 +0100 | [diff] [blame] | 21 | commit = lookup_commit_reference_gently(r, &oid, 1); |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 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 | |
Nguyễn Thái Ngọc Duy | bd7ad45 | 2018-11-10 06:49:06 +0100 | [diff] [blame] | 35 | void notes_cache_init(struct repository *r, struct notes_cache *c, |
| 36 | const char *name, const char *validity) |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 37 | { |
| 38 | struct strbuf ref = STRBUF_INIT; |
Mike Hommey | ee76f92 | 2015-10-08 11:54:43 +0900 | [diff] [blame] | 39 | int flags = NOTES_INIT_WRITABLE; |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 40 | |
| 41 | memset(c, 0, sizeof(*c)); |
| 42 | c->validity = xstrdup(validity); |
| 43 | |
| 44 | strbuf_addf(&ref, "refs/notes/%s", name); |
Nguyễn Thái Ngọc Duy | bd7ad45 | 2018-11-10 06:49:06 +0100 | [diff] [blame] | 45 | if (!notes_cache_match_validity(r, ref.buf, validity)) |
Mike Hommey | ee76f92 | 2015-10-08 11:54:43 +0900 | [diff] [blame] | 46 | flags |= NOTES_INIT_EMPTY; |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 47 | init_notes(&c->tree, ref.buf, combine_notes_overwrite, flags); |
| 48 | strbuf_release(&ref); |
| 49 | } |
| 50 | |
| 51 | int notes_cache_write(struct notes_cache *c) |
| 52 | { |
brian m. carlson | 569aa37 | 2017-05-06 22:09:58 +0000 | [diff] [blame] | 53 | struct object_id tree_oid, commit_oid; |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 54 | |
Mike Hommey | ee76f92 | 2015-10-08 11:54:43 +0900 | [diff] [blame] | 55 | if (!c || !c->tree.initialized || !c->tree.update_ref || |
| 56 | !*c->tree.update_ref) |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 57 | return -1; |
| 58 | if (!c->tree.dirty) |
| 59 | return 0; |
| 60 | |
Patryk Obara | bbca96d | 2018-01-28 01:13:18 +0100 | [diff] [blame] | 61 | if (write_notes_tree(&c->tree, &tree_oid)) |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 62 | return -1; |
Patryk Obara | 5078f34 | 2018-01-28 01:13:16 +0100 | [diff] [blame] | 63 | if (commit_tree(c->validity, strlen(c->validity), &tree_oid, NULL, |
| 64 | &commit_oid, NULL, NULL) < 0) |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 65 | return -1; |
brian m. carlson | ae07777 | 2017-10-15 22:06:51 +0000 | [diff] [blame] | 66 | if (update_ref("update notes cache", c->tree.update_ref, &commit_oid, |
Mike Hommey | ee76f92 | 2015-10-08 11:54:43 +0900 | [diff] [blame] | 67 | NULL, 0, UPDATE_REFS_QUIET_ON_ERR) < 0) |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 68 | return -1; |
| 69 | |
| 70 | return 0; |
| 71 | } |
| 72 | |
brian m. carlson | 569aa37 | 2017-05-06 22:09:58 +0000 | [diff] [blame] | 73 | char *notes_cache_get(struct notes_cache *c, struct object_id *key_oid, |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 74 | size_t *outsize) |
| 75 | { |
brian m. carlson | 9ef7223 | 2017-05-30 10:30:40 -0700 | [diff] [blame] | 76 | const struct object_id *value_oid; |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 77 | enum object_type type; |
| 78 | char *value; |
| 79 | unsigned long size; |
| 80 | |
brian m. carlson | 5ee8a95 | 2017-05-30 10:30:43 -0700 | [diff] [blame] | 81 | value_oid = get_note(&c->tree, key_oid); |
brian m. carlson | 9ef7223 | 2017-05-30 10:30:40 -0700 | [diff] [blame] | 82 | if (!value_oid) |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 83 | return NULL; |
brian m. carlson | b4f5aca | 2018-03-12 02:27:53 +0000 | [diff] [blame] | 84 | value = read_object_file(value_oid, &type, &size); |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 85 | |
| 86 | *outsize = size; |
| 87 | return value; |
| 88 | } |
| 89 | |
brian m. carlson | 569aa37 | 2017-05-06 22:09:58 +0000 | [diff] [blame] | 90 | int notes_cache_put(struct notes_cache *c, struct object_id *key_oid, |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 91 | const char *data, size_t size) |
| 92 | { |
brian m. carlson | 569aa37 | 2017-05-06 22:09:58 +0000 | [diff] [blame] | 93 | struct object_id value_oid; |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 94 | |
Patryk Obara | a09c985 | 2018-01-28 01:13:19 +0100 | [diff] [blame] | 95 | if (write_object_file(data, size, "blob", &value_oid) < 0) |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 96 | return -1; |
brian m. carlson | 5ee8a95 | 2017-05-30 10:30:43 -0700 | [diff] [blame] | 97 | return add_note(&c->tree, key_oid, &value_oid, NULL); |
Jeff King | a941d5e | 2010-04-01 20:07:40 -0400 | [diff] [blame] | 98 | } |