notes-cache.con commit completion: unite --reuse-message and --reedit-message handling (f8e49e1)
   1#include "cache.h"
   2#include "notes-cache.h"
   3#include "commit.h"
   4#include "refs.h"
   5
   6static 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
  31void 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
  47int 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
  68char *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
  85int 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        return add_note(&c->tree, key_sha1, value_sha1, NULL);
  93}