Andrew's git - gitweb.git/blob

   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
   8static 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
  33void 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
  49int 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
  71char *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
  88int 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}