notes-cache.c

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