Elijah Newren | 36bf195 | 2023-02-24 00:09:24 +0000 | [diff] [blame] | 1 | #include "git-compat-util.h" |
Jeff King | dbbcd44 | 2020-07-28 16:23:39 -0400 | [diff] [blame] | 2 | #include "strvec.h" |
Jeff King | c1189ca | 2011-09-13 17:57:57 -0400 | [diff] [blame] | 3 | #include "strbuf.h" |
| 4 | |
Jeff King | 873cd28 | 2020-07-28 16:23:25 -0400 | [diff] [blame] | 5 | const char *empty_strvec[] = { NULL }; |
Jeff King | c1189ca | 2011-09-13 17:57:57 -0400 | [diff] [blame] | 6 | |
Jeff King | 873cd28 | 2020-07-28 16:23:25 -0400 | [diff] [blame] | 7 | void strvec_init(struct strvec *array) |
Jeff King | c1189ca | 2011-09-13 17:57:57 -0400 | [diff] [blame] | 8 | { |
Ævar Arnfjörð Bjarmason | 5726a6b | 2021-07-01 12:51:26 +0200 | [diff] [blame] | 9 | struct strvec blank = STRVEC_INIT; |
| 10 | memcpy(array, &blank, sizeof(*array)); |
Jeff King | c1189ca | 2011-09-13 17:57:57 -0400 | [diff] [blame] | 11 | } |
| 12 | |
Johannes Schindelin | ce68178 | 2024-07-13 21:08:19 +0000 | [diff] [blame] | 13 | void strvec_push_nodup(struct strvec *array, char *value) |
Jeff King | c1189ca | 2011-09-13 17:57:57 -0400 | [diff] [blame] | 14 | { |
Jeff King | d70a9eb | 2020-07-28 20:37:20 -0400 | [diff] [blame] | 15 | if (array->v == empty_strvec) |
| 16 | array->v = NULL; |
Jeff King | c1189ca | 2011-09-13 17:57:57 -0400 | [diff] [blame] | 17 | |
Jeff King | d70a9eb | 2020-07-28 20:37:20 -0400 | [diff] [blame] | 18 | ALLOC_GROW(array->v, array->nr + 2, array->alloc); |
| 19 | array->v[array->nr++] = value; |
| 20 | array->v[array->nr] = NULL; |
Jeff King | c1189ca | 2011-09-13 17:57:57 -0400 | [diff] [blame] | 21 | } |
| 22 | |
Jeff King | 873cd28 | 2020-07-28 16:23:25 -0400 | [diff] [blame] | 23 | const char *strvec_push(struct strvec *array, const char *value) |
Jeff King | c1189ca | 2011-09-13 17:57:57 -0400 | [diff] [blame] | 24 | { |
Jeff King | 873cd28 | 2020-07-28 16:23:25 -0400 | [diff] [blame] | 25 | strvec_push_nodup(array, xstrdup(value)); |
Jeff King | d70a9eb | 2020-07-28 20:37:20 -0400 | [diff] [blame] | 26 | return array->v[array->nr - 1]; |
Jeff King | c1189ca | 2011-09-13 17:57:57 -0400 | [diff] [blame] | 27 | } |
| 28 | |
Jeff King | 873cd28 | 2020-07-28 16:23:25 -0400 | [diff] [blame] | 29 | const char *strvec_pushf(struct strvec *array, const char *fmt, ...) |
Jeff King | c1189ca | 2011-09-13 17:57:57 -0400 | [diff] [blame] | 30 | { |
| 31 | va_list ap; |
| 32 | struct strbuf v = STRBUF_INIT; |
| 33 | |
| 34 | va_start(ap, fmt); |
| 35 | strbuf_vaddf(&v, fmt, ap); |
| 36 | va_end(ap); |
| 37 | |
Jeff King | 873cd28 | 2020-07-28 16:23:25 -0400 | [diff] [blame] | 38 | strvec_push_nodup(array, strbuf_detach(&v, NULL)); |
Jeff King | d70a9eb | 2020-07-28 20:37:20 -0400 | [diff] [blame] | 39 | return array->v[array->nr - 1]; |
Jeff King | c1189ca | 2011-09-13 17:57:57 -0400 | [diff] [blame] | 40 | } |
| 41 | |
Jeff King | 873cd28 | 2020-07-28 16:23:25 -0400 | [diff] [blame] | 42 | void strvec_pushl(struct strvec *array, ...) |
Jeff King | d15bbe1 | 2012-04-18 14:10:05 -0700 | [diff] [blame] | 43 | { |
| 44 | va_list ap; |
| 45 | const char *arg; |
| 46 | |
| 47 | va_start(ap, array); |
Denton Liu | 828e829 | 2019-11-19 15:55:29 -0800 | [diff] [blame] | 48 | while ((arg = va_arg(ap, const char *))) |
Jeff King | 873cd28 | 2020-07-28 16:23:25 -0400 | [diff] [blame] | 49 | strvec_push(array, arg); |
Jeff King | d15bbe1 | 2012-04-18 14:10:05 -0700 | [diff] [blame] | 50 | va_end(ap); |
| 51 | } |
| 52 | |
Jeff King | d70a9eb | 2020-07-28 20:37:20 -0400 | [diff] [blame] | 53 | void strvec_pushv(struct strvec *array, const char **items) |
Paul Tan | 85b3432 | 2015-06-14 16:41:50 +0800 | [diff] [blame] | 54 | { |
Jeff King | d70a9eb | 2020-07-28 20:37:20 -0400 | [diff] [blame] | 55 | for (; *items; items++) |
| 56 | strvec_push(array, *items); |
Paul Tan | 85b3432 | 2015-06-14 16:41:50 +0800 | [diff] [blame] | 57 | } |
| 58 | |
Patrick Steinhardt | 3f5fade | 2024-11-20 14:39:38 +0100 | [diff] [blame] | 59 | void strvec_splice(struct strvec *array, size_t idx, size_t len, |
| 60 | const char **replacement, size_t replacement_len) |
| 61 | { |
| 62 | if (idx + len > array->nr) |
| 63 | BUG("range outside of array boundary"); |
Rubén Justo | 14ef8c0 | 2024-12-04 23:44:25 +0100 | [diff] [blame] | 64 | if (replacement_len > len) { |
| 65 | if (array->v == empty_strvec) |
| 66 | array->v = NULL; |
Patrick Steinhardt | 3f5fade | 2024-11-20 14:39:38 +0100 | [diff] [blame] | 67 | ALLOC_GROW(array->v, array->nr + (replacement_len - len) + 1, |
| 68 | array->alloc); |
Rubén Justo | 14ef8c0 | 2024-12-04 23:44:25 +0100 | [diff] [blame] | 69 | array->v[array->nr + (replacement_len - len)] = NULL; |
| 70 | } |
Patrick Steinhardt | 3f5fade | 2024-11-20 14:39:38 +0100 | [diff] [blame] | 71 | for (size_t i = 0; i < len; i++) |
| 72 | free((char *)array->v[idx + i]); |
Rubén Justo | 14ef8c0 | 2024-12-04 23:44:25 +0100 | [diff] [blame] | 73 | if ((replacement_len != len) && array->nr) |
Patrick Steinhardt | 3f5fade | 2024-11-20 14:39:38 +0100 | [diff] [blame] | 74 | memmove(array->v + idx + replacement_len, array->v + idx + len, |
| 75 | (array->nr - idx - len + 1) * sizeof(char *)); |
Rubén Justo | 14ef8c0 | 2024-12-04 23:44:25 +0100 | [diff] [blame] | 76 | array->nr += replacement_len - len; |
Patrick Steinhardt | 3f5fade | 2024-11-20 14:39:38 +0100 | [diff] [blame] | 77 | for (size_t i = 0; i < replacement_len; i++) |
| 78 | array->v[idx + i] = xstrdup(replacement[i]); |
| 79 | } |
| 80 | |
Patrick Steinhardt | 11ce77b | 2024-05-27 13:47:04 +0200 | [diff] [blame] | 81 | const char *strvec_replace(struct strvec *array, size_t idx, const char *replacement) |
| 82 | { |
| 83 | char *to_free; |
| 84 | if (idx >= array->nr) |
| 85 | BUG("index outside of array boundary"); |
| 86 | to_free = (char *) array->v[idx]; |
| 87 | array->v[idx] = xstrdup(replacement); |
| 88 | free(to_free); |
| 89 | return array->v[idx]; |
| 90 | } |
| 91 | |
| 92 | void strvec_remove(struct strvec *array, size_t idx) |
| 93 | { |
| 94 | if (idx >= array->nr) |
| 95 | BUG("index outside of array boundary"); |
| 96 | free((char *)array->v[idx]); |
| 97 | memmove(array->v + idx, array->v + idx + 1, (array->nr - idx) * sizeof(char *)); |
| 98 | array->nr--; |
| 99 | } |
| 100 | |
Jeff King | 873cd28 | 2020-07-28 16:23:25 -0400 | [diff] [blame] | 101 | void strvec_pop(struct strvec *array) |
Jeff King | fe4a0a2 | 2012-09-01 07:25:27 -0400 | [diff] [blame] | 102 | { |
Jeff King | d70a9eb | 2020-07-28 20:37:20 -0400 | [diff] [blame] | 103 | if (!array->nr) |
Jeff King | fe4a0a2 | 2012-09-01 07:25:27 -0400 | [diff] [blame] | 104 | return; |
Jeff King | d70a9eb | 2020-07-28 20:37:20 -0400 | [diff] [blame] | 105 | free((char *)array->v[array->nr - 1]); |
| 106 | array->v[array->nr - 1] = NULL; |
| 107 | array->nr--; |
Jeff King | fe4a0a2 | 2012-09-01 07:25:27 -0400 | [diff] [blame] | 108 | } |
| 109 | |
Jeff King | 873cd28 | 2020-07-28 16:23:25 -0400 | [diff] [blame] | 110 | void strvec_split(struct strvec *array, const char *to_split) |
Johannes Schindelin | c5aa6db | 2018-04-25 11:53:57 +0200 | [diff] [blame] | 111 | { |
| 112 | while (isspace(*to_split)) |
| 113 | to_split++; |
| 114 | for (;;) { |
| 115 | const char *p = to_split; |
| 116 | |
| 117 | if (!*p) |
| 118 | break; |
| 119 | |
| 120 | while (*p && !isspace(*p)) |
| 121 | p++; |
Jeff King | 873cd28 | 2020-07-28 16:23:25 -0400 | [diff] [blame] | 122 | strvec_push_nodup(array, xstrndup(to_split, p - to_split)); |
Johannes Schindelin | c5aa6db | 2018-04-25 11:53:57 +0200 | [diff] [blame] | 123 | |
| 124 | while (isspace(*p)) |
| 125 | p++; |
| 126 | to_split = p; |
| 127 | } |
| 128 | } |
| 129 | |
Jeff King | 873cd28 | 2020-07-28 16:23:25 -0400 | [diff] [blame] | 130 | void strvec_clear(struct strvec *array) |
Jeff King | c1189ca | 2011-09-13 17:57:57 -0400 | [diff] [blame] | 131 | { |
Jeff King | d70a9eb | 2020-07-28 20:37:20 -0400 | [diff] [blame] | 132 | if (array->v != empty_strvec) { |
Patrick Steinhardt | 80c9e70 | 2024-12-06 11:27:24 +0100 | [diff] [blame] | 133 | for (size_t i = 0; i < array->nr; i++) |
Jeff King | d70a9eb | 2020-07-28 20:37:20 -0400 | [diff] [blame] | 134 | free((char *)array->v[i]); |
| 135 | free(array->v); |
Jeff King | c1189ca | 2011-09-13 17:57:57 -0400 | [diff] [blame] | 136 | } |
Jeff King | 873cd28 | 2020-07-28 16:23:25 -0400 | [diff] [blame] | 137 | strvec_init(array); |
Jeff King | c1189ca | 2011-09-13 17:57:57 -0400 | [diff] [blame] | 138 | } |
Jeff King | b992657 | 2016-02-22 17:44:15 -0500 | [diff] [blame] | 139 | |
Jeff King | 873cd28 | 2020-07-28 16:23:25 -0400 | [diff] [blame] | 140 | const char **strvec_detach(struct strvec *array) |
Jeff King | b992657 | 2016-02-22 17:44:15 -0500 | [diff] [blame] | 141 | { |
Jeff King | d70a9eb | 2020-07-28 20:37:20 -0400 | [diff] [blame] | 142 | if (array->v == empty_strvec) |
Jeff King | b992657 | 2016-02-22 17:44:15 -0500 | [diff] [blame] | 143 | return xcalloc(1, sizeof(const char *)); |
| 144 | else { |
Jeff King | d70a9eb | 2020-07-28 20:37:20 -0400 | [diff] [blame] | 145 | const char **ret = array->v; |
Jeff King | 873cd28 | 2020-07-28 16:23:25 -0400 | [diff] [blame] | 146 | strvec_init(array); |
Jeff King | b992657 | 2016-02-22 17:44:15 -0500 | [diff] [blame] | 147 | return ret; |
| 148 | } |
| 149 | } |