builtin-remote: make rm() use properly named variable to hold return value
[git/git.git] / builtin-remote.c
1 #include "cache.h"
2 #include "parse-options.h"
3 #include "transport.h"
4 #include "remote.h"
5 #include "string-list.h"
6 #include "strbuf.h"
7 #include "run-command.h"
8 #include "refs.h"
9
10 static const char * const builtin_remote_usage[] = {
11 "git remote",
12 "git remote add <name> <url>",
13 "git remote rm <name>",
14 "git remote show <name>",
15 "git remote prune <name>",
16 "git remote update [group]",
17 NULL
18 };
19
20 static int verbose;
21
22 static int show_all(void);
23
24 static inline int postfixcmp(const char *string, const char *postfix)
25 {
26 int len1 = strlen(string), len2 = strlen(postfix);
27 if (len1 < len2)
28 return 1;
29 return strcmp(string + len1 - len2, postfix);
30 }
31
32 static int opt_parse_track(const struct option *opt, const char *arg, int not)
33 {
34 struct string_list *list = opt->value;
35 if (not)
36 string_list_clear(list, 0);
37 else
38 string_list_append(arg, list);
39 return 0;
40 }
41
42 static int fetch_remote(const char *name)
43 {
44 const char *argv[] = { "fetch", name, NULL };
45 printf("Updating %s\n", name);
46 if (run_command_v_opt(argv, RUN_GIT_CMD))
47 return error("Could not fetch %s", name);
48 return 0;
49 }
50
51 static int add(int argc, const char **argv)
52 {
53 int fetch = 0, mirror = 0;
54 struct string_list track = { NULL, 0, 0 };
55 const char *master = NULL;
56 struct remote *remote;
57 struct strbuf buf, buf2;
58 const char *name, *url;
59 int i;
60
61 struct option options[] = {
62 OPT_GROUP("add specific options"),
63 OPT_BOOLEAN('f', "fetch", &fetch, "fetch the remote branches"),
64 OPT_CALLBACK('t', "track", &track, "branch",
65 "branch(es) to track", opt_parse_track),
66 OPT_STRING('m', "master", &master, "branch", "master branch"),
67 OPT_BOOLEAN(0, "mirror", &mirror, "no separate remotes"),
68 OPT_END()
69 };
70
71 argc = parse_options(argc, argv, options, builtin_remote_usage, 0);
72
73 if (argc < 2)
74 usage_with_options(builtin_remote_usage, options);
75
76 name = argv[0];
77 url = argv[1];
78
79 remote = remote_get(name);
80 if (remote && (remote->url_nr > 1 || strcmp(name, remote->url[0]) ||
81 remote->fetch_refspec_nr))
82 die("remote %s already exists.", name);
83
84 strbuf_init(&buf, 0);
85 strbuf_init(&buf2, 0);
86
87 strbuf_addf(&buf2, "refs/heads/test:refs/remotes/%s/test", name);
88 if (!valid_fetch_refspec(buf2.buf))
89 die("'%s' is not a valid remote name", name);
90
91 strbuf_addf(&buf, "remote.%s.url", name);
92 if (git_config_set(buf.buf, url))
93 return 1;
94
95 strbuf_reset(&buf);
96 strbuf_addf(&buf, "remote.%s.fetch", name);
97
98 if (track.nr == 0)
99 string_list_append("*", &track);
100 for (i = 0; i < track.nr; i++) {
101 struct string_list_item *item = track.items + i;
102
103 strbuf_reset(&buf2);
104 strbuf_addch(&buf2, '+');
105 if (mirror)
106 strbuf_addf(&buf2, "refs/%s:refs/%s",
107 item->string, item->string);
108 else
109 strbuf_addf(&buf2, "refs/heads/%s:refs/remotes/%s/%s",
110 item->string, name, item->string);
111 if (git_config_set_multivar(buf.buf, buf2.buf, "^$", 0))
112 return 1;
113 }
114
115 if (mirror) {
116 strbuf_reset(&buf);
117 strbuf_addf(&buf, "remote.%s.mirror", name);
118 if (git_config_set(buf.buf, "true"))
119 return 1;
120 }
121
122 if (fetch && fetch_remote(name))
123 return 1;
124
125 if (master) {
126 strbuf_reset(&buf);
127 strbuf_addf(&buf, "refs/remotes/%s/HEAD", name);
128
129 strbuf_reset(&buf2);
130 strbuf_addf(&buf2, "refs/remotes/%s/%s", name, master);
131
132 if (create_symref(buf.buf, buf2.buf, "remote add"))
133 return error("Could not setup master '%s'", master);
134 }
135
136 strbuf_release(&buf);
137 strbuf_release(&buf2);
138 string_list_clear(&track, 0);
139
140 return 0;
141 }
142
143 struct branch_info {
144 char *remote;
145 struct string_list merge;
146 };
147
148 static struct string_list branch_list;
149
150 static const char *abbrev_ref(const char *name, const char *prefix)
151 {
152 const char *abbrev = skip_prefix(name, prefix);
153 if (abbrev)
154 return abbrev;
155 return name;
156 }
157 #define abbrev_branch(name) abbrev_ref((name), "refs/heads/")
158
159 static int config_read_branches(const char *key, const char *value, void *cb)
160 {
161 if (!prefixcmp(key, "branch.")) {
162 char *name;
163 struct string_list_item *item;
164 struct branch_info *info;
165 enum { REMOTE, MERGE } type;
166
167 key += 7;
168 if (!postfixcmp(key, ".remote")) {
169 name = xstrndup(key, strlen(key) - 7);
170 type = REMOTE;
171 } else if (!postfixcmp(key, ".merge")) {
172 name = xstrndup(key, strlen(key) - 6);
173 type = MERGE;
174 } else
175 return 0;
176
177 item = string_list_insert(name, &branch_list);
178
179 if (!item->util)
180 item->util = xcalloc(sizeof(struct branch_info), 1);
181 info = item->util;
182 if (type == REMOTE) {
183 if (info->remote)
184 warning("more than one branch.%s", key);
185 info->remote = xstrdup(value);
186 } else {
187 char *space = strchr(value, ' ');
188 value = abbrev_branch(value);
189 while (space) {
190 char *merge;
191 merge = xstrndup(value, space - value);
192 string_list_append(merge, &info->merge);
193 value = abbrev_branch(space + 1);
194 space = strchr(value, ' ');
195 }
196 string_list_append(xstrdup(value), &info->merge);
197 }
198 }
199 return 0;
200 }
201
202 static void read_branches(void)
203 {
204 if (branch_list.nr)
205 return;
206 git_config(config_read_branches, NULL);
207 sort_string_list(&branch_list);
208 }
209
210 struct ref_states {
211 struct remote *remote;
212 struct string_list new, stale, tracked;
213 };
214
215 static int handle_one_branch(const char *refname,
216 const unsigned char *sha1, int flags, void *cb_data)
217 {
218 struct ref_states *states = cb_data;
219 struct refspec refspec;
220
221 memset(&refspec, 0, sizeof(refspec));
222 refspec.dst = (char *)refname;
223 if (!remote_find_tracking(states->remote, &refspec)) {
224 struct string_list_item *item;
225 const char *name = abbrev_branch(refspec.src);
226 /* symbolic refs pointing nowhere were handled already */
227 if ((flags & REF_ISSYMREF) ||
228 unsorted_string_list_has_string(&states->tracked,
229 name) ||
230 unsorted_string_list_has_string(&states->new,
231 name))
232 return 0;
233 item = string_list_append(name, &states->stale);
234 item->util = xstrdup(refname);
235 }
236 return 0;
237 }
238
239 static int get_ref_states(const struct ref *ref, struct ref_states *states)
240 {
241 struct ref *fetch_map = NULL, **tail = &fetch_map;
242 int i;
243
244 for (i = 0; i < states->remote->fetch_refspec_nr; i++)
245 if (get_fetch_map(ref, states->remote->fetch + i, &tail, 1))
246 die("Could not get fetch map for refspec %s",
247 states->remote->fetch_refspec[i]);
248
249 states->new.strdup_strings = states->tracked.strdup_strings = 1;
250 for (ref = fetch_map; ref; ref = ref->next) {
251 struct string_list *target = &states->tracked;
252 unsigned char sha1[20];
253 void *util = NULL;
254
255 if (!ref->peer_ref || read_ref(ref->peer_ref->name, sha1))
256 target = &states->new;
257 else {
258 target = &states->tracked;
259 if (hashcmp(sha1, ref->new_sha1))
260 util = &states;
261 }
262 string_list_append(abbrev_branch(ref->name), target)->util = util;
263 }
264 free_refs(fetch_map);
265
266 for_each_ref(handle_one_branch, states);
267 sort_string_list(&states->stale);
268
269 return 0;
270 }
271
272 struct known_remote {
273 struct known_remote *next;
274 struct remote *remote;
275 };
276
277 struct known_remotes {
278 struct remote *to_delete;
279 struct known_remote *list;
280 };
281
282 static int add_known_remote(struct remote *remote, void *cb_data)
283 {
284 struct known_remotes *all = cb_data;
285 struct known_remote *r;
286
287 if (!strcmp(all->to_delete->name, remote->name))
288 return 0;
289
290 r = xmalloc(sizeof(*r));
291 r->remote = remote;
292 r->next = all->list;
293 all->list = r;
294 return 0;
295 }
296
297 struct branches_for_remote {
298 struct remote *remote;
299 struct string_list *branches;
300 struct known_remotes *keep;
301 };
302
303 static int add_branch_for_removal(const char *refname,
304 const unsigned char *sha1, int flags, void *cb_data)
305 {
306 struct branches_for_remote *branches = cb_data;
307 struct refspec refspec;
308 struct string_list_item *item;
309 struct known_remote *kr;
310
311 memset(&refspec, 0, sizeof(refspec));
312 refspec.dst = (char *)refname;
313 if (remote_find_tracking(branches->remote, &refspec))
314 return 0;
315
316 /* don't delete a branch if another remote also uses it */
317 for (kr = branches->keep->list; kr; kr = kr->next) {
318 memset(&refspec, 0, sizeof(refspec));
319 refspec.dst = (char *)refname;
320 if (!remote_find_tracking(kr->remote, &refspec))
321 return 0;
322 }
323
324 /* make sure that symrefs are deleted */
325 if (flags & REF_ISSYMREF)
326 return unlink(git_path("%s", refname));
327
328 item = string_list_append(refname, branches->branches);
329 item->util = xmalloc(20);
330 hashcpy(item->util, sha1);
331
332 return 0;
333 }
334
335 static int remove_branches(struct string_list *branches)
336 {
337 int i, result = 0;
338 for (i = 0; i < branches->nr; i++) {
339 struct string_list_item *item = branches->items + i;
340 const char *refname = item->string;
341 unsigned char *sha1 = item->util;
342
343 if (delete_ref(refname, sha1, 0))
344 result |= error("Could not remove branch %s", refname);
345 }
346 return result;
347 }
348
349 static int rm(int argc, const char **argv)
350 {
351 struct option options[] = {
352 OPT_END()
353 };
354 struct remote *remote;
355 struct strbuf buf;
356 struct known_remotes known_remotes = { NULL, NULL };
357 struct string_list branches = { NULL, 0, 0, 1 };
358 struct branches_for_remote cb_data = { NULL, &branches, &known_remotes };
359 int i, result;
360
361 if (argc != 2)
362 usage_with_options(builtin_remote_usage, options);
363
364 remote = remote_get(argv[1]);
365 if (!remote)
366 die("No such remote: %s", argv[1]);
367
368 known_remotes.to_delete = remote;
369 for_each_remote(add_known_remote, &known_remotes);
370
371 strbuf_init(&buf, 0);
372 strbuf_addf(&buf, "remote.%s", remote->name);
373 if (git_config_rename_section(buf.buf, NULL) < 1)
374 return error("Could not remove config section '%s'", buf.buf);
375
376 read_branches();
377 for (i = 0; i < branch_list.nr; i++) {
378 struct string_list_item *item = branch_list.items + i;
379 struct branch_info *info = item->util;
380 if (info->remote && !strcmp(info->remote, remote->name)) {
381 const char *keys[] = { "remote", "merge", NULL }, **k;
382 for (k = keys; *k; k++) {
383 strbuf_reset(&buf);
384 strbuf_addf(&buf, "branch.%s.%s",
385 item->string, *k);
386 if (git_config_set(buf.buf, NULL)) {
387 strbuf_release(&buf);
388 return -1;
389 }
390 }
391 }
392 }
393
394 /*
395 * We cannot just pass a function to for_each_ref() which deletes
396 * the branches one by one, since for_each_ref() relies on cached
397 * refs, which are invalidated when deleting a branch.
398 */
399 cb_data.remote = remote;
400 result = for_each_ref(add_branch_for_removal, &cb_data);
401 strbuf_release(&buf);
402
403 if (!result)
404 result = remove_branches(&branches);
405 string_list_clear(&branches, 1);
406
407 return result;
408 }
409
410 static void show_list(const char *title, struct string_list *list,
411 const char *extra_arg)
412 {
413 int i;
414
415 if (!list->nr)
416 return;
417
418 printf(title, list->nr > 1 ? "es" : "", extra_arg);
419 printf("\n ");
420 for (i = 0; i < list->nr; i++)
421 printf("%s%s", i ? " " : "", list->items[i].string);
422 printf("\n");
423 }
424
425 static int get_remote_ref_states(const char *name,
426 struct ref_states *states,
427 int query)
428 {
429 struct transport *transport;
430 const struct ref *ref;
431
432 states->remote = remote_get(name);
433 if (!states->remote)
434 return error("No such remote: %s", name);
435
436 read_branches();
437
438 if (query) {
439 transport = transport_get(NULL, states->remote->url_nr > 0 ?
440 states->remote->url[0] : NULL);
441 ref = transport_get_remote_refs(transport);
442 transport_disconnect(transport);
443
444 get_ref_states(ref, states);
445 }
446
447 return 0;
448 }
449
450 static int append_ref_to_tracked_list(const char *refname,
451 const unsigned char *sha1, int flags, void *cb_data)
452 {
453 struct ref_states *states = cb_data;
454 struct refspec refspec;
455
456 memset(&refspec, 0, sizeof(refspec));
457 refspec.dst = (char *)refname;
458 if (!remote_find_tracking(states->remote, &refspec))
459 string_list_append(abbrev_branch(refspec.src), &states->tracked);
460
461 return 0;
462 }
463
464 static int show(int argc, const char **argv)
465 {
466 int no_query = 0, result = 0;
467 struct option options[] = {
468 OPT_GROUP("show specific options"),
469 OPT_BOOLEAN('n', NULL, &no_query, "do not query remotes"),
470 OPT_END()
471 };
472 struct ref_states states;
473
474 argc = parse_options(argc, argv, options, builtin_remote_usage, 0);
475
476 if (argc < 1)
477 return show_all();
478
479 memset(&states, 0, sizeof(states));
480 for (; argc; argc--, argv++) {
481 int i;
482
483 get_remote_ref_states(*argv, &states, !no_query);
484
485 printf("* remote %s\n URL: %s\n", *argv,
486 states.remote->url_nr > 0 ?
487 states.remote->url[0] : "(no URL)");
488
489 for (i = 0; i < branch_list.nr; i++) {
490 struct string_list_item *branch = branch_list.items + i;
491 struct branch_info *info = branch->util;
492 int j;
493
494 if (!info->merge.nr || strcmp(*argv, info->remote))
495 continue;
496 printf(" Remote branch%s merged with 'git pull' "
497 "while on branch %s\n ",
498 info->merge.nr > 1 ? "es" : "",
499 branch->string);
500 for (j = 0; j < info->merge.nr; j++)
501 printf(" %s", info->merge.items[j].string);
502 printf("\n");
503 }
504
505 if (!no_query) {
506 show_list(" New remote branch%s (next fetch "
507 "will store in remotes/%s)",
508 &states.new, states.remote->name);
509 show_list(" Stale tracking branch%s (use 'git remote "
510 "prune')", &states.stale, "");
511 }
512
513 if (no_query)
514 for_each_ref(append_ref_to_tracked_list, &states);
515 show_list(" Tracked remote branch%s", &states.tracked, "");
516
517 if (states.remote->push_refspec_nr) {
518 printf(" Local branch%s pushed with 'git push'\n ",
519 states.remote->push_refspec_nr > 1 ?
520 "es" : "");
521 for (i = 0; i < states.remote->push_refspec_nr; i++) {
522 struct refspec *spec = states.remote->push + i;
523 printf(" %s%s%s%s", spec->force ? "+" : "",
524 abbrev_branch(spec->src),
525 spec->dst ? ":" : "",
526 spec->dst ? abbrev_branch(spec->dst) : "");
527 }
528 printf("\n");
529 }
530
531 /* NEEDSWORK: free remote */
532 string_list_clear(&states.new, 0);
533 string_list_clear(&states.stale, 0);
534 string_list_clear(&states.tracked, 0);
535 }
536
537 return result;
538 }
539
540 static int prune(int argc, const char **argv)
541 {
542 int dry_run = 0, result = 0;
543 struct option options[] = {
544 OPT_GROUP("prune specific options"),
545 OPT__DRY_RUN(&dry_run),
546 OPT_END()
547 };
548 struct ref_states states;
549
550 argc = parse_options(argc, argv, options, builtin_remote_usage, 0);
551
552 if (argc < 1)
553 usage_with_options(builtin_remote_usage, options);
554
555 memset(&states, 0, sizeof(states));
556 for (; argc; argc--, argv++) {
557 int i;
558
559 get_remote_ref_states(*argv, &states, 1);
560
561 if (states.stale.nr) {
562 printf("Pruning %s\n", *argv);
563 printf("URL: %s\n",
564 states.remote->url_nr
565 ? states.remote->url[0]
566 : "(no URL)");
567 }
568
569 for (i = 0; i < states.stale.nr; i++) {
570 const char *refname = states.stale.items[i].util;
571
572 if (!dry_run)
573 result |= delete_ref(refname, NULL, 0);
574
575 printf(" * [%s] %s\n", dry_run ? "would prune" : "pruned",
576 abbrev_ref(refname, "refs/remotes/"));
577 }
578
579 /* NEEDSWORK: free remote */
580 string_list_clear(&states.new, 0);
581 string_list_clear(&states.stale, 0);
582 string_list_clear(&states.tracked, 0);
583 }
584
585 return result;
586 }
587
588 static int get_one_remote_for_update(struct remote *remote, void *priv)
589 {
590 struct string_list *list = priv;
591 if (!remote->skip_default_update)
592 string_list_append(xstrdup(remote->name), list);
593 return 0;
594 }
595
596 struct remote_group {
597 const char *name;
598 struct string_list *list;
599 } remote_group;
600
601 static int get_remote_group(const char *key, const char *value, void *cb)
602 {
603 if (!prefixcmp(key, "remotes.") &&
604 !strcmp(key + 8, remote_group.name)) {
605 /* split list by white space */
606 int space = strcspn(value, " \t\n");
607 while (*value) {
608 if (space > 1)
609 string_list_append(xstrndup(value, space),
610 remote_group.list);
611 value += space + (value[space] != '\0');
612 space = strcspn(value, " \t\n");
613 }
614 }
615
616 return 0;
617 }
618
619 static int update(int argc, const char **argv)
620 {
621 int i, result = 0;
622 struct string_list list = { NULL, 0, 0, 0 };
623 static const char *default_argv[] = { NULL, "default", NULL };
624
625 if (argc < 2) {
626 argc = 2;
627 argv = default_argv;
628 }
629
630 remote_group.list = &list;
631 for (i = 1; i < argc; i++) {
632 remote_group.name = argv[i];
633 result = git_config(get_remote_group, NULL);
634 }
635
636 if (!result && !list.nr && argc == 2 && !strcmp(argv[1], "default"))
637 result = for_each_remote(get_one_remote_for_update, &list);
638
639 for (i = 0; i < list.nr; i++)
640 result |= fetch_remote(list.items[i].string);
641
642 /* all names were strdup()ed or strndup()ed */
643 list.strdup_strings = 1;
644 string_list_clear(&list, 0);
645
646 return result;
647 }
648
649 static int get_one_entry(struct remote *remote, void *priv)
650 {
651 struct string_list *list = priv;
652
653 if (remote->url_nr > 0) {
654 int i;
655
656 for (i = 0; i < remote->url_nr; i++)
657 string_list_append(remote->name, list)->util = (void *)remote->url[i];
658 } else
659 string_list_append(remote->name, list)->util = NULL;
660
661 return 0;
662 }
663
664 static int show_all(void)
665 {
666 struct string_list list = { NULL, 0, 0 };
667 int result = for_each_remote(get_one_entry, &list);
668
669 if (!result) {
670 int i;
671
672 sort_string_list(&list);
673 for (i = 0; i < list.nr; i++) {
674 struct string_list_item *item = list.items + i;
675 if (verbose)
676 printf("%s\t%s\n", item->string,
677 item->util ? (const char *)item->util : "");
678 else {
679 if (i && !strcmp((item - 1)->string, item->string))
680 continue;
681 printf("%s\n", item->string);
682 }
683 }
684 }
685 return result;
686 }
687
688 int cmd_remote(int argc, const char **argv, const char *prefix)
689 {
690 struct option options[] = {
691 OPT__VERBOSE(&verbose),
692 OPT_END()
693 };
694 int result;
695
696 argc = parse_options(argc, argv, options, builtin_remote_usage,
697 PARSE_OPT_STOP_AT_NON_OPTION);
698
699 if (argc < 1)
700 result = show_all();
701 else if (!strcmp(argv[0], "add"))
702 result = add(argc, argv);
703 else if (!strcmp(argv[0], "rm"))
704 result = rm(argc, argv);
705 else if (!strcmp(argv[0], "show"))
706 result = show(argc, argv);
707 else if (!strcmp(argv[0], "prune"))
708 result = prune(argc, argv);
709 else if (!strcmp(argv[0], "update"))
710 result = update(argc, argv);
711 else {
712 error("Unknown subcommand: %s", argv[0]);
713 usage_with_options(builtin_remote_usage, options);
714 }
715
716 return result ? 1 : 0;
717 }