Teach update-ref about a symbolic ref stored in a textfile.
[git.git] / git-relink.perl
1 #!/usr/bin/env perl
2 # Copyright 2005, Ryan Anderson <ryan@michonline.com>
3 # Distribution permitted under the GPL v2, as distributed
4 # by the Free Software Foundation.
5 # Later versions of the GPL at the discretion of Linus Torvalds
6 #
7 # Scan two git object-trees, and hardlink any common objects between them.
8
9 use 5.006;
10 use strict;
11 use warnings;
12 use Getopt::Long;
13
14 sub get_canonical_form($);
15 sub do_scan_directory($$$);
16 sub compare_two_files($$);
17 sub usage();
18 sub link_two_files($$);
19
20 # stats
21 my $total_linked = 0;
22 my $total_already = 0;
23 my ($linked,$already);
24
25 my $fail_on_different_sizes = 0;
26 my $help = 0;
27 GetOptions("safe" => \$fail_on_different_sizes,
28            "help" => \$help);
29
30 usage() if $help;
31
32 my (@dirs) = @ARGV;
33
34 usage() if (!defined $dirs[0] || !defined $dirs[1]);
35
36 $_ = get_canonical_form($_) foreach (@dirs);
37
38 my $master_dir = pop @dirs;
39
40 opendir(D,$master_dir . "objects/")
41         or die "Failed to open $master_dir/objects/ : $!";
42
43 my @hashdirs = grep !/^\.{1,2}$/, readdir(D);
44
45 foreach my $repo (@dirs) {
46         $linked = 0;
47         $already = 0;
48         printf("Searching '%s' and '%s' for common objects and hardlinking them...\n",
49                 $master_dir,$repo);
50
51         foreach my $hashdir (@hashdirs) {
52                 do_scan_directory($master_dir, $hashdir, $repo);
53         }
54
55         printf("Linked %d files, %d were already linked.\n",$linked, $already);
56
57         $total_linked += $linked;
58         $total_already += $already;
59 }
60
61 printf("Totals: Linked %d files, %d were already linked.\n",
62         $total_linked, $total_already);
63
64
65 sub do_scan_directory($$$) {
66         my ($srcdir, $subdir, $dstdir) = @_;
67
68         my $sfulldir = sprintf("%sobjects/%s/",$srcdir,$subdir);
69         my $dfulldir = sprintf("%sobjects/%s/",$dstdir,$subdir);
70
71         opendir(S,$sfulldir)
72                 or die "Failed to opendir $sfulldir: $!";
73
74         foreach my $file (grep(!/\.{1,2}$/, readdir(S))) {
75                 my $sfilename = $sfulldir . $file;
76                 my $dfilename = $dfulldir . $file;
77
78                 compare_two_files($sfilename,$dfilename);
79
80         }
81         closedir(S);
82 }
83
84 sub compare_two_files($$) {
85         my ($sfilename, $dfilename) = @_;
86
87         # Perl's stat returns relevant information as follows:
88         # 0 = dev number
89         # 1 = inode number
90         # 7 = size
91         my @sstatinfo = stat($sfilename);
92         my @dstatinfo = stat($dfilename);
93
94         if (@sstatinfo == 0 && @dstatinfo == 0) {
95                 die sprintf("Stat of both %s and %s failed: %s\n",$sfilename, $dfilename, $!);
96
97         } elsif (@dstatinfo == 0) {
98                 return;
99         }
100
101         if ( ($sstatinfo[0] == $dstatinfo[0]) &&
102              ($sstatinfo[1] != $dstatinfo[1])) {
103                 if ($sstatinfo[7] == $dstatinfo[7]) {
104                         link_two_files($sfilename, $dfilename);
105
106                 } else {
107                         my $err = sprintf("ERROR: File sizes are not the same, cannot relink %s to %s.\n",
108                                 $sfilename, $dfilename);
109                         if ($fail_on_different_sizes) {
110                                 die $err;
111                         } else {
112                                 warn $err;
113                         }
114                 }
115
116         } elsif ( ($sstatinfo[0] == $dstatinfo[0]) &&
117              ($sstatinfo[1] == $dstatinfo[1])) {
118                 $already++;
119         }
120 }
121
122 sub get_canonical_form($) {
123         my $dir = shift;
124         my $original = $dir;
125
126         die "$dir is not a directory." unless -d $dir;
127
128         $dir .= "/" unless $dir =~ m#/$#;
129         $dir .= ".git/" unless $dir =~ m#\.git/$#;
130
131         die "$original does not have a .git/ subdirectory.\n" unless -d $dir;
132
133         return $dir;
134 }
135
136 sub link_two_files($$) {
137         my ($sfilename, $dfilename) = @_;
138         my $tmpdname = sprintf("%s.old",$dfilename);
139         rename($dfilename,$tmpdname)
140                 or die sprintf("Failure renaming %s to %s: %s",
141                         $dfilename, $tmpdname, $!);
142
143         if (! link($sfilename,$dfilename)) {
144                 my $failtxt = "";
145                 unless (rename($tmpdname,$dfilename)) {
146                         $failtxt = sprintf(
147                                 "Git Repository containing %s is probably corrupted, " .
148                                 "please copy '%s' to '%s' to fix.\n",
149                                 $tmpdname, $dfilename);
150                 }
151
152                 die sprintf("Failed to link %s to %s: %s\n%s" .
153                         $sfilename, $dfilename,
154                         $!, $dfilename, $failtxt);
155         }
156
157         unlink($tmpdname)
158                 or die sprintf("Unlink of %s failed: %s\n",
159                         $dfilename, $!);
160
161         $linked++;
162 }
163
164
165 sub usage() {
166         print("Usage: $0 [--safe] <dir> [<dir> ...] <master_dir> \n");
167         print("All directories should contain a .git/objects/ subdirectory.\n");
168         print("Options\n");
169         print("\t--safe\t" .
170                 "Stops if two objects with the same hash exist but " .
171                 "have different sizes.  Default is to warn and continue.\n");
172         exit(1);
173 }