Move 'unpack_trees()' over to 'traverse_trees()' interface
[git] / git-remote.perl
1 #!/usr/bin/perl -w
2
3 use strict;
4 use Git;
5 my $git = Git->repository();
6
7 sub add_remote_config {
8         my ($hash, $name, $what, $value) = @_;
9         if ($what eq 'url') {
10                 # Having more than one is Ok -- it is used for push.
11                 if (! exists $hash->{'URL'}) {
12                         $hash->{$name}{'URL'} = $value;
13                 }
14         }
15         elsif ($what eq 'fetch') {
16                 $hash->{$name}{'FETCH'} ||= [];
17                 push @{$hash->{$name}{'FETCH'}}, $value;
18         }
19         elsif ($what eq 'push') {
20                 $hash->{$name}{'PUSH'} ||= [];
21                 push @{$hash->{$name}{'PUSH'}}, $value;
22         }
23         if (!exists $hash->{$name}{'SOURCE'}) {
24                 $hash->{$name}{'SOURCE'} = 'config';
25         }
26 }
27
28 sub add_remote_remotes {
29         my ($hash, $file, $name) = @_;
30
31         if (exists $hash->{$name}) {
32                 $hash->{$name}{'WARNING'} = 'ignored due to config';
33                 return;
34         }
35
36         my $fh;
37         if (!open($fh, '<', $file)) {
38                 print STDERR "Warning: cannot open $file\n";
39                 return;
40         }
41         my $it = { 'SOURCE' => 'remotes' };
42         $hash->{$name} = $it;
43         while (<$fh>) {
44                 chomp;
45                 if (/^URL:\s*(.*)$/) {
46                         # Having more than one is Ok -- it is used for push.
47                         if (! exists $it->{'URL'}) {
48                                 $it->{'URL'} = $1;
49                         }
50                 }
51                 elsif (/^Push:\s*(.*)$/) {
52                         $it->{'PUSH'} ||= [];
53                         push @{$it->{'PUSH'}}, $1;
54                 }
55                 elsif (/^Pull:\s*(.*)$/) {
56                         $it->{'FETCH'} ||= [];
57                         push @{$it->{'FETCH'}}, $1;
58                 }
59                 elsif (/^\#/) {
60                         ; # ignore
61                 }
62                 else {
63                         print STDERR "Warning: funny line in $file: $_\n";
64                 }
65         }
66         close($fh);
67 }
68
69 sub list_remote {
70         my ($git) = @_;
71         my %seen = ();
72         my @remotes = eval {
73                 $git->command(qw(config --get-regexp), '^remote\.');
74         };
75         for (@remotes) {
76                 if (/^remote\.(\S+?)\.([^.\s]+)\s+(.*)$/) {
77                         add_remote_config(\%seen, $1, $2, $3);
78                 }
79         }
80
81         my $dir = $git->repo_path() . "/remotes";
82         if (opendir(my $dh, $dir)) {
83                 local $_;
84                 while ($_ = readdir($dh)) {
85                         chomp;
86                         next if (! -f "$dir/$_" || ! -r _);
87                         add_remote_remotes(\%seen, "$dir/$_", $_);
88                 }
89         }
90
91         return \%seen;
92 }
93
94 sub add_branch_config {
95         my ($hash, $name, $what, $value) = @_;
96         if ($what eq 'remote') {
97                 if (exists $hash->{$name}{'REMOTE'}) {
98                         print STDERR "Warning: more than one branch.$name.remote\n";
99                 }
100                 $hash->{$name}{'REMOTE'} = $value;
101         }
102         elsif ($what eq 'merge') {
103                 $hash->{$name}{'MERGE'} ||= [];
104                 push @{$hash->{$name}{'MERGE'}}, $value;
105         }
106 }
107
108 sub list_branch {
109         my ($git) = @_;
110         my %seen = ();
111         my @branches = eval {
112                 $git->command(qw(config --get-regexp), '^branch\.');
113         };
114         for (@branches) {
115                 if (/^branch\.([^.]*)\.(\S*)\s+(.*)$/) {
116                         add_branch_config(\%seen, $1, $2, $3);
117                 }
118         }
119
120         return \%seen;
121 }
122
123 my $remote = list_remote($git);
124 my $branch = list_branch($git);
125
126 sub update_ls_remote {
127         my ($harder, $info) = @_;
128
129         return if (($harder == 0) ||
130                    (($harder == 1) && exists $info->{'LS_REMOTE'}));
131
132         my @ref = map {
133                 s|^[0-9a-f]{40}\s+refs/heads/||;
134                 $_;
135         } $git->command(qw(ls-remote --heads), $info->{'URL'});
136         $info->{'LS_REMOTE'} = \@ref;
137 }
138
139 sub list_wildcard_mapping {
140         my ($forced, $ours, $ls) = @_;
141         my %refs;
142         for (@$ls) {
143                 $refs{$_} = 01; # bit #0 to say "they have"
144         }
145         for ($git->command('for-each-ref', "refs/remotes/$ours")) {
146                 chomp;
147                 next unless (s|^[0-9a-f]{40}\s[a-z]+\srefs/remotes/$ours/||);
148                 next if ($_ eq 'HEAD');
149                 $refs{$_} ||= 0;
150                 $refs{$_} |= 02; # bit #1 to say "we have"
151         }
152         my (@new, @stale, @tracked);
153         for (sort keys %refs) {
154                 my $have = $refs{$_};
155                 if ($have == 1) {
156                         push @new, $_;
157                 }
158                 elsif ($have == 2) {
159                         push @stale, $_;
160                 }
161                 elsif ($have == 3) {
162                         push @tracked, $_;
163                 }
164         }
165         return \@new, \@stale, \@tracked;
166 }
167
168 sub list_mapping {
169         my ($name, $info) = @_;
170         my $fetch = $info->{'FETCH'};
171         my $ls = $info->{'LS_REMOTE'};
172         my (@new, @stale, @tracked);
173
174         for (@$fetch) {
175                 next unless (/(\+)?([^:]+):(.*)/);
176                 my ($forced, $theirs, $ours) = ($1, $2, $3);
177                 if ($theirs eq 'refs/heads/*' &&
178                     $ours =~ /^refs\/remotes\/(.*)\/\*$/) {
179                         # wildcard mapping
180                         my ($w_new, $w_stale, $w_tracked)
181                                 = list_wildcard_mapping($forced, $1, $ls);
182                         push @new, @$w_new;
183                         push @stale, @$w_stale;
184                         push @tracked, @$w_tracked;
185                 }
186                 elsif ($theirs =~ /\*/ || $ours =~ /\*/) {
187                         print STDERR "Warning: unrecognized mapping in remotes.$name.fetch: $_\n";
188                 }
189                 elsif ($theirs =~ s|^refs/heads/||) {
190                         if (!grep { $_ eq $theirs } @$ls) {
191                                 push @stale, $theirs;
192                         }
193                         elsif ($ours ne '') {
194                                 push @tracked, $theirs;
195                         }
196                 }
197         }
198         return \@new, \@stale, \@tracked;
199 }
200
201 sub show_mapping {
202         my ($name, $info) = @_;
203         my ($new, $stale, $tracked) = list_mapping($name, $info);
204         if (@$new) {
205                 print "  New remote branches (next fetch will store in remotes/$name)\n";
206                 print "    @$new\n";
207         }
208         if (@$stale) {
209                 print "  Stale tracking branches in remotes/$name (use 'git remote prune')\n";
210                 print "    @$stale\n";
211         }
212         if (@$tracked) {
213                 print "  Tracked remote branches\n";
214                 print "    @$tracked\n";
215         }
216 }
217
218 sub prune_remote {
219         my ($name, $ls_remote) = @_;
220         if (!exists $remote->{$name}) {
221                 print STDERR "No such remote $name\n";
222                 return 1;
223         }
224         my $info = $remote->{$name};
225         update_ls_remote($ls_remote, $info);
226
227         my ($new, $stale, $tracked) = list_mapping($name, $info);
228         my $prefix = "refs/remotes/$name";
229         foreach my $to_prune (@$stale) {
230                 my @v = $git->command(qw(rev-parse --verify), "$prefix/$to_prune");
231                 $git->command(qw(update-ref -d), "$prefix/$to_prune", $v[0]);
232         }
233         return 0;
234 }
235
236 sub show_remote {
237         my ($name, $ls_remote) = @_;
238         if (!exists $remote->{$name}) {
239                 print STDERR "No such remote $name\n";
240                 return 1;
241         }
242         my $info = $remote->{$name};
243         update_ls_remote($ls_remote, $info);
244
245         print "* remote $name\n";
246         print "  URL: $info->{'URL'}\n";
247         for my $branchname (sort keys %$branch) {
248                 next unless (defined $branch->{$branchname}{'REMOTE'} &&
249                              $branch->{$branchname}{'REMOTE'} eq $name);
250                 my @merged = map {
251                         s|^refs/heads/||;
252                         $_;
253                 } split(' ',"@{$branch->{$branchname}{'MERGE'}}");
254                 next unless (@merged);
255                 print "  Remote branch(es) merged with 'git pull' while on branch $branchname\n";
256                 print "    @merged\n";
257         }
258         if ($info->{'LS_REMOTE'}) {
259                 show_mapping($name, $info);
260         }
261         if ($info->{'PUSH'}) {
262                 my @pushed = map {
263                         s|^refs/heads/||;
264                         s|^\+refs/heads/|+|;
265                         s|:refs/heads/|:|;
266                         $_;
267                 } @{$info->{'PUSH'}};
268                 print "  Local branch(es) pushed with 'git push'\n";
269                 print "    @pushed\n";
270         }
271         return 0;
272 }
273
274 sub add_remote {
275         my ($name, $url, $opts) = @_;
276         if (exists $remote->{$name}) {
277                 print STDERR "remote $name already exists.\n";
278                 exit(1);
279         }
280         $git->command('config', "remote.$name.url", $url);
281         my $track = $opts->{'track'} || ["*"];
282
283         for (@$track) {
284                 $git->command('config', '--add', "remote.$name.fetch",
285                                 $opts->{'mirror'} ?
286                                 "+refs/$_:refs/$_" :
287                                 "+refs/heads/$_:refs/remotes/$name/$_");
288         }
289         if ($opts->{'fetch'}) {
290                 $git->command('fetch', $name);
291         }
292         if (exists $opts->{'master'}) {
293                 $git->command('symbolic-ref', "refs/remotes/$name/HEAD",
294                               "refs/remotes/$name/$opts->{'master'}");
295         }
296 }
297
298 sub update_remote {
299         my ($name) = @_;
300         my @remotes;
301
302         my $conf = $git->config("remotes." . $name);
303         if (defined($conf)) {
304                 @remotes = split(' ', $conf);
305         } elsif ($name eq 'default') {
306                 @remotes = ();
307                 for (sort keys %$remote) {
308                         my $do_fetch = $git->config_bool("remote." . $_ .
309                                                     ".skipDefaultUpdate");
310                         unless ($do_fetch) {
311                                 push @remotes, $_;
312                         }
313                 }
314         } else {
315                 print STDERR "Remote group $name does not exists.\n";
316                 exit(1);
317         }
318         for (@remotes) {
319                 print "Updating $_\n";
320                 $git->command('fetch', "$_");
321         }
322 }
323
324 sub rm_remote {
325         my ($name) = @_;
326         if (!exists $remote->{$name}) {
327                 print STDERR "No such remote $name\n";
328                 return 1;
329         }
330
331         $git->command('config', '--remove-section', "remote.$name");
332
333         eval {
334             my @trackers = $git->command('config', '--get-regexp',
335                         'branch.*.remote', $name);
336                 for (@trackers) {
337                         /^branch\.(.*)?\.remote/;
338                         $git->config('--unset', "branch.$1.remote");
339                         $git->config('--unset', "branch.$1.merge");
340                 }
341         };
342
343         my @refs = $git->command('for-each-ref',
344                 '--format=%(refname) %(objectname)', "refs/remotes/$name");
345         for (@refs) {
346                 my ($ref, $object) = split;
347                 $git->command(qw(update-ref -d), $ref, $object);
348         }
349         return 0;
350 }
351
352 sub add_usage {
353         print STDERR "Usage: git remote add [-f] [-t track]* [-m master] <name> <url>\n";
354         exit(1);
355 }
356
357 my $VERBOSE = 0;
358 @ARGV = grep {
359         if ($_ eq '-v' or $_ eq '--verbose') {
360                 $VERBOSE=1;
361                 0
362         } else {
363                 1
364         }
365 } @ARGV;
366
367 if (!@ARGV) {
368         for (sort keys %$remote) {
369                 print "$_";
370                 print "\t$remote->{$_}->{URL}" if $VERBOSE;
371                 print "\n";
372         }
373 }
374 elsif ($ARGV[0] eq 'show') {
375         my $ls_remote = 1;
376         my $i;
377         for ($i = 1; $i < @ARGV; $i++) {
378                 if ($ARGV[$i] eq '-n') {
379                         $ls_remote = 0;
380                 }
381                 else {
382                         last;
383                 }
384         }
385         if ($i >= @ARGV) {
386                 print STDERR "Usage: git remote show <remote>\n";
387                 exit(1);
388         }
389         my $status = 0;
390         for (; $i < @ARGV; $i++) {
391                 $status |= show_remote($ARGV[$i], $ls_remote);
392         }
393         exit($status);
394 }
395 elsif ($ARGV[0] eq 'update') {
396         if (@ARGV <= 1) {
397                 update_remote("default");
398                 exit(1);
399         }
400         for (my $i = 1; $i < @ARGV; $i++) {
401                 update_remote($ARGV[$i]);
402         }
403 }
404 elsif ($ARGV[0] eq 'prune') {
405         my $ls_remote = 1;
406         my $i;
407         for ($i = 1; $i < @ARGV; $i++) {
408                 if ($ARGV[$i] eq '-n') {
409                         $ls_remote = 0;
410                 }
411                 else {
412                         last;
413                 }
414         }
415         if ($i >= @ARGV) {
416                 print STDERR "Usage: git remote prune <remote>\n";
417                 exit(1);
418         }
419         my $status = 0;
420         for (; $i < @ARGV; $i++) {
421                 $status |= prune_remote($ARGV[$i], $ls_remote);
422         }
423         exit($status);
424 }
425 elsif ($ARGV[0] eq 'add') {
426         my %opts = ();
427         while (1 < @ARGV && $ARGV[1] =~ /^-/) {
428                 my $opt = $ARGV[1];
429                 shift @ARGV;
430                 if ($opt eq '-f' || $opt eq '--fetch') {
431                         $opts{'fetch'} = 1;
432                         next;
433                 }
434                 if ($opt eq '-t' || $opt eq '--track') {
435                         if (@ARGV < 1) {
436                                 add_usage();
437                         }
438                         $opts{'track'} ||= [];
439                         push @{$opts{'track'}}, $ARGV[1];
440                         shift @ARGV;
441                         next;
442                 }
443                 if ($opt eq '-m' || $opt eq '--master') {
444                         if ((@ARGV < 1) || exists $opts{'master'}) {
445                                 add_usage();
446                         }
447                         $opts{'master'} = $ARGV[1];
448                         shift @ARGV;
449                         next;
450                 }
451                 if ($opt eq '--mirror') {
452                         $opts{'mirror'} = 1;
453                         next;
454                 }
455                 add_usage();
456         }
457         if (@ARGV != 3) {
458                 add_usage();
459         }
460         add_remote($ARGV[1], $ARGV[2], \%opts);
461 }
462 elsif ($ARGV[0] eq 'rm') {
463         if (@ARGV <= 1) {
464                 print STDERR "Usage: git remote rm <remote>\n";
465                 exit(1);
466         }
467         exit(rm_remote($ARGV[1]));
468 }
469 else {
470         print STDERR "Usage: git remote\n";
471         print STDERR "       git remote add <name> <url>\n";
472         print STDERR "       git remote rm <name>\n";
473         print STDERR "       git remote show <name>\n";
474         print STDERR "       git remote prune <name>\n";
475         print STDERR "       git remote update [group]\n";
476         exit(1);
477 }