8ee1f9f03edb6ea91c7ecf442e874c819636ba8e
[buildfarm-client.git] / run_branches
1 #!/usr/bin/env perl
2
3 =comment
4
5 Copyright (c) 2003-2010, Andrew Dunstan
6
7 See accompanying License file for license details
8
9 =cut
10
11 use vars qw($VERSION); $VERSION = 'REL_0.1';
12
13 use strict;
14 use warnings;
15 use 5.010;
16
17 use Fcntl qw(:flock :seek);
18 use File::Basename;
19 use FindBin qw($RealBin);
20 use Cwd;
21
22 use lib $RealBin;
23 use EximBuild::Options;
24
25 sub branch_last_sort;
26
27 # Complain on old-style use (.pl extension), but only if a
28 # terminal is available.
29 if ($0 =~ /(.*)\.pl$/) {
30     die "$0: Please use `@{[join ' ' => $1, @ARGV]}' instead.\n"
31         if -t;
32     exec $1, @ARGV;
33 }
34
35 # Most of the client code assumes that our working directory
36 # is the client code directory.
37 my %CALLED = (
38     cwd => cwd(),
39     argv0 => $0,
40     argv => [@ARGV],    # get a copy!
41 );
42 chdir $RealBin or die "$0: Can't chdir to '$RealBin': $!\n";
43 say "Changed working directory to '$RealBin'" if -t;
44
45 my %branch_last;
46 my $run_build = './run_build';
47
48 my($run_all, $run_one);
49 my %extra_options =(
50     'run-all' => \$run_all,
51     'run-one' => \$run_one,
52 );
53
54 # process the command line
55 EximBuild::Options::fetch_options(%extra_options);
56
57 # no non-option args allowed here
58 die("$0: non-option arguments not permitted")
59   if @ARGV;
60
61 die "$0: only one of --run-all and --run-one permitted"
62   if ($run_all && $run_one);
63
64 die "$0: need one of --run-all and --run-one"
65   unless ($run_all || $run_one);
66
67 # common mistake
68 die "$0: need group searchable homedir"
69   unless (stat($ENV{HOME}) & 0550 == 0550);
70
71 # set up a "branch" variable for processing the config file
72 use vars qw($branch);
73 $branch = 'global';
74
75 #
76 # process config file
77 #
78 require $buildconf;
79
80 # Check if auto-update is wanted and possible
81 if (not exists $EximBuild::Conf{auto_update} or $EximBuild::Conf{auto_update})
82 {
83     my $remote = $EximBuild::Conf{auto_update} // 'origin';
84
85     die "$0: auto-update not possible: need write permissions in @{[cwd]}\n"
86         if not -w '.';
87
88     # Get information about our remote and calculate the chance for a
89     # successfull auto-updat. Based on:
90     # http://stackoverflow.com/questions/3258243/check-if-pull-needed-in-git
91     system("git fetch $remote") == 0 or die $? >> 8;
92     my ($upstream, $local, $base) = qx'git rev-parse ...@{upstream}';   die $? >> 8 if $?;
93     $base =~ s/^\^//;
94
95     if ($upstream ne $local) {
96
97         if ($base ne $local) {
98             warn "$0: the merge base is not local anymore. Refusing to `git pull`\n"
99         }
100         else {
101             # if we're the merge base, the ff-only should work
102             system('git pull --ff-only') == 0 or die $? >> 8;
103
104             say "$0: re-execute after update";
105             chdir $CALLED{cwd} or die "$0: Can't chdir to $CALLED{cwd}: $!\n";
106             exec $CALLED{argv0}, @{$CALLED{argv}};
107             die "Can't re-exec\n";
108         }
109     }
110 }
111
112 unless (
113     (
114         ref $EximBuild::conf{branches_to_build} eq 'ARRAY'
115         &&@{$EximBuild::conf{branches_to_build}}
116     )
117     ||$EximBuild::conf{branches_to_build} =~
118     /^(ALL|HEAD_PLUS_LATEST|HEAD_PLUS_LATEST2)$/
119   )
120 {
121     die "no branches_to_build specified in $buildconf";
122 }
123
124 my @branches;
125 if (ref $EximBuild::conf{branches_to_build})
126 {
127     @branches = @{$EximBuild::conf{branches_to_build}};
128 }
129 elsif ($EximBuild::conf{branches_to_build} =~
130     /^(ALL|HEAD_PLUS_LATEST|HEAD_PLUS_LATEST2)$/ )
131 {
132
133     # Need to set the path here so we make sure we pick up the right perl.
134     # It has to be the perl that the build script would choose
135     # i.e. specially *not* the MinGW SDK perl that is invoked for the
136     # build script, which means we need to put the path back the way it was
137     # when we're done
138     local $ENV{PATH} = $EximBuild::conf{build_env}->{PATH}
139       if ($EximBuild::conf{build_env}->{PATH});
140     (my $url = $EximBuild::conf{target}) =~s/cgi-bin.*/branches_of_interest.txt/;
141     my $branches_of_interest = `perl -MLWP::Simple -e "getprint(q{$url})"`;
142     die "getting branches of interest" unless $branches_of_interest;
143     push(@branches,$_)foreach (split(/\s+/,$branches_of_interest));
144     #splice(@branches,0,-2)
145     #  if $EximBuild::conf{branches_to_build} eq 'HEAD_PLUS_LATEST';
146     #splice(@branches,0,-3)
147     #  if $EximBuild::conf{branches_to_build} eq 'HEAD_PLUS_LATEST2';
148 }
149
150 @branches = apply_throttle(@branches);
151
152 my $global_lock_dir =
153     $EximBuild::conf{global_lock_dir}
154   ||$EximBuild::conf{build_root}
155   ||'';
156
157 unless ($global_lock_dir && -d $global_lock_dir)
158 {
159     die "no global lock directory: $global_lock_dir";
160 }
161
162 # acquire the lock
163
164 my $lockfile;
165
166 my $lockfilename = "$global_lock_dir/GLOBAL.lck";
167
168 open($lockfile, ">$lockfilename") || die "opening lockfile: $!";
169
170 if ( !flock($lockfile,LOCK_EX|LOCK_NB) )
171 {
172     print "Another process holds the lock on " ."$lockfilename. Exiting.\n"
173       if ($verbose);
174     exit(0);
175 }
176
177 if ($run_all)
178 {
179     foreach my $brnch(@branches)
180     {
181         run_branch($brnch);
182     }
183 }
184 elsif ($run_one)
185 {
186
187     # sort the branches by the order in which they last did actual work
188     # then try running them in that order until one does some work
189
190     %branch_last = map {$_ => find_last_status($_)} @branches;
191     foreach my $brnch(sort branch_last_sort @branches)
192     {
193         run_branch($brnch);
194         my $new_status = find_last_status($brnch);
195         last if $new_status != $branch_last{$brnch};
196     }
197 }
198
199 exit 0;
200
201 ##########################################################
202
203 sub run_branch
204 {
205     my $branch = shift;
206     my @args = ($run_build,EximBuild::Options::standard_option_list(), $branch);
207
208     # Explicitly use perl from the path (and not this perl, so don't use $^X)
209     # This script needs to run on Cygwin with non-cygwin perl if it's running
210     # in tandem with AS/MinGW perl, since Cygwin perl doesn't honor locks
211     # the samne way, and the global lock fails. But the build script needs
212     # to run with the native perl, even on Cygwin, which it picks up from
213     # the path. (Head exploding yet?).
214     system("perl",@args);
215 }
216
217 sub branch_last_sort
218 {
219     return $branch_last{$a} <=> $branch_last{$b};
220 }
221
222 sub find_last_status
223 {
224     my $brnch = shift;
225     my $status_file =
226       "$EximBuild::conf{build_root}/$brnch/$EximBuild::conf{animal}.last.status";
227     return 0 unless (-e  $status_file);
228     my $handle;
229     open($handle,$status_file) || dir $!;
230     my $ts = <$handle>;
231     chomp $ts;
232     close($handle);
233     return $ts + 0;
234 }
235
236 sub apply_throttle
237 {
238     my @branches = @_;
239     return @branches unless exists $EximBuild::conf{throttle};
240     my @result;
241     my %throttle = %{$EximBuild::conf{throttle}};
242
243     # implement throttle keywords ALL !HEAD and !RECENT
244     my @candidates;
245     my $replacement;
246     if (exists $throttle{ALL})
247     {
248         @candidates = @branches;
249         $replacement = $throttle{ALL};
250     }
251     elsif (exists  $throttle{'!HEAD'})
252     {
253         @candidates = grep { $_ ne 'HEAD' } @branches;
254         $replacement = $throttle{'!HEAD'};
255     }
256     elsif (exists  $throttle{'!RECENT'})
257     {
258
259         # sort branches, make sure we get numeric major version sorting right
260         my @stable = grep { $_ ne 'HEAD' } @branches;
261         s/^REL(\d)_/0$1/ foreach (@stable);
262         @stable = sort @stable;
263         s/^REL0/REL/ foreach (@stable);
264         pop @stable; # remove latest
265         @candidates = @stable;
266         $replacement = $throttle{'!RECENT'};
267     }
268     foreach my $cand (@candidates)
269     {
270
271         # only supply this for the branch if there isn't already
272         # a throttle
273         $throttle{$cand} ||= $replacement;
274     }
275
276     # apply throttle filters
277     foreach my $branch(@branches)
278     {
279         my $this_throttle =  $throttle{$branch};
280         unless (defined $this_throttle)
281         {
282             push(@result,$branch);
283             next;
284         }
285         my $minh = $this_throttle->{min_hours_since};
286         my $ts = find_last_status($branch);
287         next
288           if ( $ts
289             && (defined $minh)
290             &&($minh && $minh < ((time - $ts) / 3600.0)));
291         if (exists $this_throttle->{allowed_hours})
292         {
293             my @allowed_hours = split(/,/,$this_throttle->{allowed_hours});
294             my $hour = (localtime(time))[2];
295             next unless grep {$_ == $hour} @allowed_hours;
296         }
297         push(@result,$branch);
298     }
299
300     return @result;
301 }