refspec: consolidate ref-prefix generation logic
[git/git.git] / refspec.c
1 #include "cache.h"
2 #include "argv-array.h"
3 #include "refs.h"
4 #include "refspec.h"
5
6 static struct refspec_item s_tag_refspec = {
7 0,
8 1,
9 0,
10 0,
11 "refs/tags/*",
12 "refs/tags/*"
13 };
14
15 /* See TAG_REFSPEC for the string version */
16 const struct refspec_item *tag_refspec = &s_tag_refspec;
17
18 /*
19 * Parses the provided refspec 'refspec' and populates the refspec_item 'item'.
20 * Returns 1 if successful and 0 if the refspec is invalid.
21 */
22 static int parse_refspec(struct refspec_item *item, const char *refspec, int fetch)
23 {
24 size_t llen;
25 int is_glob;
26 const char *lhs, *rhs;
27 int flags;
28
29 is_glob = 0;
30
31 lhs = refspec;
32 if (*lhs == '+') {
33 item->force = 1;
34 lhs++;
35 }
36
37 rhs = strrchr(lhs, ':');
38
39 /*
40 * Before going on, special case ":" (or "+:") as a refspec
41 * for pushing matching refs.
42 */
43 if (!fetch && rhs == lhs && rhs[1] == '\0') {
44 item->matching = 1;
45 return 1;
46 }
47
48 if (rhs) {
49 size_t rlen = strlen(++rhs);
50 is_glob = (1 <= rlen && strchr(rhs, '*'));
51 item->dst = xstrndup(rhs, rlen);
52 }
53
54 llen = (rhs ? (rhs - lhs - 1) : strlen(lhs));
55 if (1 <= llen && memchr(lhs, '*', llen)) {
56 if ((rhs && !is_glob) || (!rhs && fetch))
57 return 0;
58 is_glob = 1;
59 } else if (rhs && is_glob) {
60 return 0;
61 }
62
63 item->pattern = is_glob;
64 item->src = xstrndup(lhs, llen);
65 flags = REFNAME_ALLOW_ONELEVEL | (is_glob ? REFNAME_REFSPEC_PATTERN : 0);
66
67 if (fetch) {
68 struct object_id unused;
69
70 /* LHS */
71 if (!*item->src)
72 ; /* empty is ok; it means "HEAD" */
73 else if (llen == GIT_SHA1_HEXSZ && !get_oid_hex(item->src, &unused))
74 item->exact_sha1 = 1; /* ok */
75 else if (!check_refname_format(item->src, flags))
76 ; /* valid looking ref is ok */
77 else
78 return 0;
79 /* RHS */
80 if (!item->dst)
81 ; /* missing is ok; it is the same as empty */
82 else if (!*item->dst)
83 ; /* empty is ok; it means "do not store" */
84 else if (!check_refname_format(item->dst, flags))
85 ; /* valid looking ref is ok */
86 else
87 return 0;
88 } else {
89 /*
90 * LHS
91 * - empty is allowed; it means delete.
92 * - when wildcarded, it must be a valid looking ref.
93 * - otherwise, it must be an extended SHA-1, but
94 * there is no existing way to validate this.
95 */
96 if (!*item->src)
97 ; /* empty is ok */
98 else if (is_glob) {
99 if (check_refname_format(item->src, flags))
100 return 0;
101 }
102 else
103 ; /* anything goes, for now */
104 /*
105 * RHS
106 * - missing is allowed, but LHS then must be a
107 * valid looking ref.
108 * - empty is not allowed.
109 * - otherwise it must be a valid looking ref.
110 */
111 if (!item->dst) {
112 if (check_refname_format(item->src, flags))
113 return 0;
114 } else if (!*item->dst) {
115 return 0;
116 } else {
117 if (check_refname_format(item->dst, flags))
118 return 0;
119 }
120 }
121
122 return 1;
123 }
124
125 void refspec_item_init(struct refspec_item *item, const char *refspec, int fetch)
126 {
127 memset(item, 0, sizeof(*item));
128
129 if (!parse_refspec(item, refspec, fetch))
130 die("Invalid refspec '%s'", refspec);
131 }
132
133 void refspec_item_clear(struct refspec_item *item)
134 {
135 FREE_AND_NULL(item->src);
136 FREE_AND_NULL(item->dst);
137 item->force = 0;
138 item->pattern = 0;
139 item->matching = 0;
140 item->exact_sha1 = 0;
141 }
142
143 void refspec_init(struct refspec *rs, int fetch)
144 {
145 memset(rs, 0, sizeof(*rs));
146 rs->fetch = fetch;
147 }
148
149 void refspec_append(struct refspec *rs, const char *refspec)
150 {
151 struct refspec_item item;
152
153 refspec_item_init(&item, refspec, rs->fetch);
154
155 ALLOC_GROW(rs->items, rs->nr + 1, rs->alloc);
156 rs->items[rs->nr++] = item;
157
158 ALLOC_GROW(rs->raw, rs->raw_nr + 1, rs->raw_alloc);
159 rs->raw[rs->raw_nr++] = xstrdup(refspec);
160 }
161
162 void refspec_appendn(struct refspec *rs, const char **refspecs, int nr)
163 {
164 int i;
165 for (i = 0; i < nr; i++)
166 refspec_append(rs, refspecs[i]);
167 }
168
169 void refspec_clear(struct refspec *rs)
170 {
171 int i;
172
173 for (i = 0; i < rs->nr; i++)
174 refspec_item_clear(&rs->items[i]);
175
176 FREE_AND_NULL(rs->items);
177 rs->alloc = 0;
178 rs->nr = 0;
179
180 for (i = 0; i < rs->raw_nr; i++)
181 free((char *)rs->raw[i]);
182 FREE_AND_NULL(rs->raw);
183 rs->raw_alloc = 0;
184 rs->raw_nr = 0;
185
186 rs->fetch = 0;
187 }
188
189 int valid_fetch_refspec(const char *fetch_refspec_str)
190 {
191 struct refspec_item refspec;
192 int ret = parse_refspec(&refspec, fetch_refspec_str, REFSPEC_FETCH);
193 refspec_item_clear(&refspec);
194 return ret;
195 }
196
197 void refspec_ref_prefixes(const struct refspec *rs,
198 struct argv_array *ref_prefixes)
199 {
200 int i;
201 for (i = 0; i < rs->nr; i++) {
202 const struct refspec_item *item = &rs->items[i];
203 const char *prefix = NULL;
204
205 if (rs->fetch == REFSPEC_FETCH)
206 prefix = item->src;
207 else if (item->dst)
208 prefix = item->dst;
209 else if (item->src && !item->exact_sha1)
210 prefix = item->src;
211
212 if (prefix) {
213 if (item->pattern) {
214 const char *glob = strchr(prefix, '*');
215 argv_array_pushf(ref_prefixes, "%.*s",
216 (int)(glob - prefix),
217 prefix);
218 } else {
219 expand_ref_prefix(ref_prefixes, prefix);
220 }
221 }
222 }
223 }