#
# Source step. Creates a source tarball equivalent to what was used to 
# build the binary release, in the Build step.
# 
package Bootstrap::Step::Source;
use Bootstrap::Step;
use Bootstrap::Config;
use File::Copy qw(move);
use File::Find qw(find);
use MozBuild::Util qw(MkdirWithPath);
use Bootstrap::Util qw(SyncToStaging);
@ISA = ("Bootstrap::Step");

sub Execute {
    my $this = shift;

    my $config = new Bootstrap::Config();
    my $product = $config->Get(var => 'product');
    my $appName = $config->Get(var => 'appName');
    my $productTag = $config->Get(var => 'productTag');
    my $version = $config->Get(var => 'version');
    my $rc = $config->Get(var => 'rc');
    my $logDir = $config->Get(sysvar => 'logDir');
    my $stageHome = $config->Get(var => 'stageHome');
    my $mozillaCvsroot = $config->Get(var => 'mozillaCvsroot');

    # create staging area
    my $stageDir = catfile($stageHome, $product . '-' . $version, 
                           'batch-source', 'rc' . $rc);

    if (not -d $stageDir) {
        MkdirWithPath(dir => $stageDir) 
          or die("Cannot create $stageDir: $!");
    }

    $this->CvsCo(cvsroot => $mozillaCvsroot,
                 tag => $productTag . '_RELEASE',
                 modules => ['mozilla/client.mk',
                             catfile('mozilla', $appName, 'config')],
                 workDir => $stageDir,
                 logFile => catfile($logDir, 'source.log')
    );
                 
    $this->Shell(
      cmd => 'make',
      cmdArgs => ['-f', 'client.mk', 'checkout',
                  'MOZ_CO_PROJECT=' . $appName],
      dir => catfile($stageDir, 'mozilla'),
      logFile => catfile($logDir, 'source.log'),
    );

    # change all CVS/Root files to anonymous CVSROOT
    File::Find::find(\&CvsChrootCallback, catfile($stageDir, 'mozilla'));

    # remove leftover mozconfig files
    unlink(glob(catfile($stageDir, 'mozilla', '.mozconfig*')));

    my $tarFile = $product . '-' . $version . '-' . 'source' . '.tar.bz2';

    $this->Shell(
      cmd => 'tar',
      cmdArgs => ['-cjf', $tarFile, 'mozilla'],
      dir => catfile($stageDir),
      logFile => catfile($logDir, 'source.log'),
    );
              
    chmod(0644, glob("$stageDir/$tarFile"));
}

sub Verify {
    my $this = shift;

    my $config = new Bootstrap::Config();
    my $logDir = $config->Get(sysvar => 'logDir');

    my $logFile = catfile($logDir, 'source.log');

    $this->CheckLog(
        log => $logFile,
        checkFor => '^checkout finish',
    );

    $this->CheckLog(
        log => $logFile,
        notAllowed => '^tar',
    );
}

sub Push {
    my $this = shift;

    my $config = new Bootstrap::Config();
    my $product = $config->Get(var => 'product');
    my $version = $config->Get(var => 'version');
    my $rc = $config->Get(var => 'rc');
    my $logDir = $config->Get(sysvar => 'logDir');
    my $stageHome = $config->Get(var => 'stageHome');

    my $stageDir =  catfile($stageHome, $product . '-' . $version);
    my $candidateDir = catfile('/home', 'ftp', 'pub', $product, 'nightly',
                            $version . '-candidates', 'rc' . $rc ) . '/';

    if (not -d $candidateDir) {
        MkdirWithPath(dir => $candidateDir) 
          or die("Could not mkdir $candidateDir: $!");
        $this->Log(msg => "Created directory $candidateDir");
    }

    $this->Shell(
      cmd => 'rsync',
      cmdArgs => ['-av', catfile('batch-source', 'rc' . $rc, 
                            $product . '-' . $version . '-source.tar.bz2'),
                  $candidateDir],
      logFile => catfile($logDir, 'source.log'),
      dir => catfile($stageDir),
    );

    SyncToStaging();
}

sub Announce {
    my $this = shift;

    my $config = new Bootstrap::Config();
    my $product = $config->Get(var => 'product');
    my $version = $config->Get(var => 'version');

    $this->SendAnnouncement(
      subject => "$product $version source step finished",
      message => "$product $version source archive was copied to the candidates dir.",
    );
}

# Change the CVS/Root file to be the anonymous CVS Root
sub CvsChrootCallback {
    my $config = new Bootstrap::Config();
    my $anonCvsroot = $config->Get(var => 'anonCvsroot');

    my $dirent = $File::Find::name;
    if ((-f $dirent) and ($dirent =~ /.*CVS\/Root$/)) {
        open(FILE, "> $dirent");
        print FILE "$anonCvsroot\n";
        close(FILE);
    }
}

1;