ATS worker target is made.
#!/usr/bin/perl
use strict;
#my $cache = "d:/HG_cache/";
#my $master = "//v800008/Builds01/";
my $cache = shift @ARGV;
my $csv = shift @ARGV;
my $generated = shift @ARGV;
my @recover;
my @nospace;
my $exitcode = 0;
if(defined $cache && defined $generated && defined $csv)
{
print "Cache:$cache\nIn:$csv\nOut:$generated\n";
# Format the cache directory path
if ( $cache !~ /(.*)[\\\/]$/ )
{
$cache .= "/";
}
open(IN, "<$csv") or die "Couldn't open $csv for reading";
open(OUT,">$generated") or die "Couldn't open $generated for writing";
my $header = <IN>;
print OUT $header;
while( my $line = <IN>)
{
my @args = split(',',$line);
my $repo = shift @args;
my $master = "";
if ( $repo =~ m/^(.*\/)(.*\/(oss|rnd|sfl)\/.*\/)$/i )
{
$master = $1;
$repo = $2;
}
if(-d $master.$repo.".hg")
{
# print "Found:\t".$master.$repo.".hg\n";
my $cmd;
if(-d $cache.$repo.".hg") # update
{
$cmd = "hg pull -R $cache$repo $master$repo";
}
else #clone
{
#taken from the normal clone script...
my @dirs = split ('\/', $cache.$repo);
my $destdir = pop @dirs;
my $path = "";
foreach my $dir (@dirs)
{
$path = ($path eq "") ? $dir : "$path/$dir";
if (!-d $path)
{
mkdir $path;
}
}
$cmd = "hg clone -U $master$repo $cache$repo";
}
if(cache($cmd))
{
print OUT $cache.$repo.",".join(',', @args);
}
else
{
print OUT $master.$repo.",".join(',', @args);
$exitcode = 1;
}
}
else
{
print "Error: cannot find ".$master.$repo.".hg\n";
$exitcode = 1;
}
}
close OUT;
close IN;
}
else
{
print "Usage: <cache_path> <source_csv> <generated_csv>";
$exitcode = 1;
}
foreach my $line (@recover)
{
print "WARNING: HG Recover: $line\n";
}
foreach my $line (@nospace)
{
print "WARNING: No Space: $line\n";
}
exit $exitcode;
sub cache($cmd)
{
my $cmd = shift;
print "$cmd\n";
open(CMD, "$cmd 2>&1 |") or die "Couldn't execute $cmd";
while(my $line = <CMD>)
{
# print $line;
# parse the output for failures. On fail return 0;
if($line =~ m/abort/i)
{
print $line;
if($line =~ m/hg\s+recover/i)
{
push(@recover, $cmd);
}
elsif($line =~ m/No\s+space/i)
{
push(@nospace, $cmd);
}
close CMD;
return 0;
}
}
close CMD;
return 1;
}