Jeff King | 29c2bd5 | 2017-02-08 15:53:07 -0500 | [diff] [blame] | 1 | #include "cache.h" |
| 2 | #include "oidset.h" |
| 3 | |
René Scharfe | 8c84ae6 | 2018-10-04 17:14:37 +0200 | [diff] [blame] | 4 | void oidset_init(struct oidset *set, size_t initial_size) |
| 5 | { |
| 6 | memset(&set->set, 0, sizeof(set->set)); |
| 7 | if (initial_size) |
Jeff King | 8fbb558 | 2019-06-20 03:41:28 -0400 | [diff] [blame] | 8 | kh_resize_oid_set(&set->set, initial_size); |
René Scharfe | 8c84ae6 | 2018-10-04 17:14:37 +0200 | [diff] [blame] | 9 | } |
| 10 | |
Jeff King | 29c2bd5 | 2017-02-08 15:53:07 -0500 | [diff] [blame] | 11 | int oidset_contains(const struct oidset *set, const struct object_id *oid) |
| 12 | { |
Jeff King | 8fbb558 | 2019-06-20 03:41:28 -0400 | [diff] [blame] | 13 | khiter_t pos = kh_get_oid_set(&set->set, *oid); |
René Scharfe | 8b2f8cb | 2018-10-04 17:13:06 +0200 | [diff] [blame] | 14 | return pos != kh_end(&set->set); |
Jeff King | 29c2bd5 | 2017-02-08 15:53:07 -0500 | [diff] [blame] | 15 | } |
| 16 | |
| 17 | int oidset_insert(struct oidset *set, const struct object_id *oid) |
| 18 | { |
René Scharfe | 8b2f8cb | 2018-10-04 17:13:06 +0200 | [diff] [blame] | 19 | int added; |
Jeff King | 8fbb558 | 2019-06-20 03:41:28 -0400 | [diff] [blame] | 20 | kh_put_oid_set(&set->set, *oid, &added); |
René Scharfe | 8b2f8cb | 2018-10-04 17:13:06 +0200 | [diff] [blame] | 21 | return !added; |
Jeff King | 29c2bd5 | 2017-02-08 15:53:07 -0500 | [diff] [blame] | 22 | } |
| 23 | |
Jeff Hostetler | c3a9ad3 | 2017-11-21 20:58:49 +0000 | [diff] [blame] | 24 | int oidset_remove(struct oidset *set, const struct object_id *oid) |
| 25 | { |
Jeff King | 8fbb558 | 2019-06-20 03:41:28 -0400 | [diff] [blame] | 26 | khiter_t pos = kh_get_oid_set(&set->set, *oid); |
René Scharfe | 8b2f8cb | 2018-10-04 17:13:06 +0200 | [diff] [blame] | 27 | if (pos == kh_end(&set->set)) |
| 28 | return 0; |
Jeff King | 8fbb558 | 2019-06-20 03:41:28 -0400 | [diff] [blame] | 29 | kh_del_oid_set(&set->set, pos); |
René Scharfe | 8b2f8cb | 2018-10-04 17:13:06 +0200 | [diff] [blame] | 30 | return 1; |
Jeff Hostetler | c3a9ad3 | 2017-11-21 20:58:49 +0000 | [diff] [blame] | 31 | } |
| 32 | |
Jeff King | 29c2bd5 | 2017-02-08 15:53:07 -0500 | [diff] [blame] | 33 | void oidset_clear(struct oidset *set) |
| 34 | { |
Jeff King | 8fbb558 | 2019-06-20 03:41:28 -0400 | [diff] [blame] | 35 | kh_release_oid_set(&set->set); |
René Scharfe | 8b2f8cb | 2018-10-04 17:13:06 +0200 | [diff] [blame] | 36 | oidset_init(set, 0); |
Jeff King | 29c2bd5 | 2017-02-08 15:53:07 -0500 | [diff] [blame] | 37 | } |
Barret Rhoden | f93895f | 2019-05-15 17:44:57 -0400 | [diff] [blame] | 38 | |
Taylor Blau | f478106 | 2020-04-13 22:04:22 -0600 | [diff] [blame] | 39 | int oidset_size(struct oidset *set) |
| 40 | { |
| 41 | return kh_size(&set->set); |
| 42 | } |
| 43 | |
Barret Rhoden | f93895f | 2019-05-15 17:44:57 -0400 | [diff] [blame] | 44 | void oidset_parse_file(struct oidset *set, const char *path) |
| 45 | { |
Junio C Hamano | 610e2b9 | 2020-09-24 21:55:04 -0700 | [diff] [blame] | 46 | oidset_parse_file_carefully(set, path, NULL, NULL); |
| 47 | } |
| 48 | |
| 49 | void oidset_parse_file_carefully(struct oidset *set, const char *path, |
| 50 | oidset_parse_tweak_fn fn, void *cbdata) |
| 51 | { |
Barret Rhoden | f93895f | 2019-05-15 17:44:57 -0400 | [diff] [blame] | 52 | FILE *fp; |
| 53 | struct strbuf sb = STRBUF_INIT; |
| 54 | struct object_id oid; |
| 55 | |
| 56 | fp = fopen(path, "r"); |
| 57 | if (!fp) |
| 58 | die("could not open object name list: %s", path); |
| 59 | while (!strbuf_getline(&sb, fp)) { |
| 60 | const char *p; |
| 61 | const char *name; |
| 62 | |
| 63 | /* |
| 64 | * Allow trailing comments, leading whitespace |
| 65 | * (including before commits), and empty or whitespace |
| 66 | * only lines. |
| 67 | */ |
| 68 | name = strchr(sb.buf, '#'); |
| 69 | if (name) |
| 70 | strbuf_setlen(&sb, name - sb.buf); |
| 71 | strbuf_trim(&sb); |
| 72 | if (!sb.len) |
| 73 | continue; |
| 74 | |
Junio C Hamano | 610e2b9 | 2020-09-24 21:55:04 -0700 | [diff] [blame] | 75 | if (parse_oid_hex(sb.buf, &oid, &p) || *p != '\0' || |
| 76 | (fn && fn(&oid, cbdata))) |
Barret Rhoden | f93895f | 2019-05-15 17:44:57 -0400 | [diff] [blame] | 77 | die("invalid object name: %s", sb.buf); |
| 78 | oidset_insert(set, &oid); |
| 79 | } |
| 80 | if (ferror(fp)) |
| 81 | die_errno("Could not read '%s'", path); |
| 82 | fclose(fp); |
| 83 | strbuf_release(&sb); |
| 84 | } |