oidset.c

   1 #include "git-compat-util.h"
   2 #include "oidset.h"
   3 #include "hex.h"
   4 #include "strbuf.h"
   5
   6 void oidset_init(struct oidset *set, size_t initial_size)
   7 {
   8         memset(&set->set, 0, sizeof(set->set));
   9         if (initial_size)
  10                 kh_resize_oid_set(&set->set, initial_size);
  11 }
  12
  13 int oidset_contains(const struct oidset *set, const struct object_id *oid)
  14 {
  15         khiter_t pos = kh_get_oid_set(&set->set, *oid);
  16         return pos != kh_end(&set->set);
  17 }
  18
  19 int oidset_insert(struct oidset *set, const struct object_id *oid)
  20 {
  21         int added;
  22         kh_put_oid_set(&set->set, *oid, &added);
  23         return !added;
  24 }
  25
  26 void oidset_insert_from_set(struct oidset *dest, struct oidset *src)
  27 {
  28         struct oidset_iter iter;
  29         struct object_id *src_oid;
  30
  31         oidset_iter_init(src, &iter);
  32         while ((src_oid = oidset_iter_next(&iter)))
  33                 oidset_insert(dest, src_oid);
  34 }
  35
  36 int oidset_remove(struct oidset *set, const struct object_id *oid)
  37 {
  38         khiter_t pos = kh_get_oid_set(&set->set, *oid);
  39         if (pos == kh_end(&set->set))
  40                 return 0;
  41         kh_del_oid_set(&set->set, pos);
  42         return 1;
  43 }
  44
  45 void oidset_clear(struct oidset *set)
  46 {
  47         kh_release_oid_set(&set->set);
  48         oidset_init(set, 0);
  49 }
  50
  51 void oidset_parse_file(struct oidset *set, const char *path,
  52                        const struct git_hash_algo *algop)
  53 {
  54         oidset_parse_file_carefully(set, path, algop, NULL, NULL);
  55 }
  56
  57 void oidset_parse_file_carefully(struct oidset *set, const char *path,
  58                                  const struct git_hash_algo *algop,
  59                                  oidset_parse_tweak_fn fn, void *cbdata)
  60 {
  61         FILE *fp;
  62         struct strbuf sb = STRBUF_INIT;
  63         struct object_id oid;
  64
  65         fp = fopen(path, "r");
  66         if (!fp)
  67                 die("could not open object name list: %s", path);
  68         while (!strbuf_getline(&sb, fp)) {
  69                 const char *p;
  70                 const char *name;
  71
  72                 /*
  73                  * Allow trailing comments, leading whitespace
  74                  * (including before commits), and empty or whitespace
  75                  * only lines.
  76                  */
  77                 name = strchr(sb.buf, '#');
  78                 if (name)
  79                         strbuf_setlen(&sb, name - sb.buf);
  80                 strbuf_trim(&sb);
  81                 if (!sb.len)
  82                         continue;
  83
  84                 if (parse_oid_hex_algop(sb.buf, &oid, &p, algop) || *p != '\0')
  85                         die("invalid object name: %s", sb.buf);
  86                 if (fn && fn(&oid, cbdata))
  87                         continue;
  88                 oidset_insert(set, &oid);
  89         }
  90         if (ferror(fp))
  91                 die_errno("Could not read '%s'", path);
  92         fclose(fp);
  93         strbuf_release(&sb);
  94 }