#
# Updates step. Generates binary update (MAR) files as well as AUS config
# snippets.
# 
package Bootstrap::Step::Updates;
use Bootstrap::Step;
use Bootstrap::Config;
use File::Find qw(find);
use MozBuild::Util qw(MkdirWithPath);
@ISA = ("Bootstrap::Step");

my $config = new Bootstrap::Config;

sub Execute {
    my $this = shift;

    my $product = $config->Get('var' => 'product');
    my $logDir = $config->Get('var' => 'logDir');
    my $version = $config->Get('var' => 'version');
    my $mozillaCvsroot = $config->Get('var' => 'mozillaCvsroot');
    my $mofoCvsroot = $config->Get('var' => 'mofoCvsroot');
    my $updateDir = $config->Get('var' => 'updateDir');
    my $patcherConfig = $config->Get('var' => 'patcherConfig');

    # Create updates area.
    if (not -d $updateDir) {
        MkdirWithPath('dir' => $updateDir) 
          or die "Cannot mkdir $updateDir: $!";
    }

    $this->Shell(
      'cmd' => 'cvs -d ' . $mozillaCvsroot . ' co -d patcher mozilla/tools/patcher',
      'logFile' => $logDir . '/patcher_checkout.log',
      'dir' => $updateDir,
      'timeout' => 3600,
    );

    # config lives in private repo
    $this->Shell(
      'cmd' => 'cvs -d ' . $mofoCvsroot . ' co -d config release/patcher/' . $patcherConfig,
      'logFile' => $logDir . '/patcher_config-checkout.log',
      'dir' => $updateDir,
    );

    # build tools
    my $originalCvsrootEnv = $ENV{'CVSROOT'};
    $ENV{'CVSROOT'} = $mozillaCvsroot;
    $this->Shell(
      'cmd' => './patcher2.pl --build-tools --app=' . $product . ' --config=../config/moz180-branch-patcher2.cfg',
      'logFile' => $logDir . '/patcher_build-tools.log',
      'dir' => $updateDir . '/patcher',
      'timeout' => 3600,
    );
    if ($originalCvsrootEnv) {
        $ENV{'CVSROOT'} = $originalCvsrootEnv;
    }
    
    # download complete MARs
    $this->Shell(
      'cmd' => './patcher2.pl --download --app=' . $product . ' --config=../config/moz180-branch-patcher2.cfg',
      'logFile' => $logDir . '/patcher_download.log',
      'dir' => $updateDir . '/patcher',
      'timeout' => 3600,
    );

    # Create partial patches and snippets
    $this->Shell(
      'cmd' => './patcher2.pl --create-patches -app=' . $product . ' --config=../config/moz180-branch-patcher2.cfg',
      'logFile' => $logDir . '/patcher_create-patches.log',
      'dir' => $updateDir . '/patcher',
      'timeout' => 18000,
    );
    
    # prepare aus2-staging
    # ssh aus2-staging.mozilla.org
    # cd /opt/aus2/incoming/3-staging
    # tmpdir="`date +%Y%m%d`-${SHORT_PRODUCT}-${VERSION}"
    # sudo mkdir ${tmpdir}-test
    # sudo chown cltbld ${tmpdir}-test
    # sudo mkdir ${tmpdir}
    # sudo chown cltbld ${tmpdir}
    
    # # copy updates from prometheus-vm.mozilla.org
    # ssh prometheus-vm.mozilla.org
    # cd /builds/${VERSION}-updates/release/patcher/temp/firefox/${PREVIOUS_VERSION}-${VERSION}/
    # rsync -nav -e "ssh -i $HOME/.ssh/aus" aus2.test/ aus2-staging.mozilla.org:/opt/aus2/incoming/3-staging/${tmpdir}-test/
    # rsync -nav -e "ssh -i $HOME/.ssh/aus" aus2/ aus2-staging.mozilla.org:/opt/aus2/incoming/3-staging/${tmpdir}/
}

sub Verify {
    my $this = shift;

    my $logDir = $config->Get('var' => 'logDir');
    my $version = $config->Get('var' => 'version');
    my $oldVersion = $config->Get('var' => 'oldVersion');
    my $mozillaCvsroot = $config->Get('var' => 'mozillaCvsroot');
    my $updateDir = $config->Get('var' => 'updateDir');
    my $verifyDir = $config->Get('var' => 'verifyDir');
    my $product = $config->Get('var' => 'product');

    ### quick verification
    # ensure that there are only test channels
    my $testDir = $updateDir . '/patcher/temp/' . $product . '/' . $oldVersion . '-' . $version . '/aus2.test';

    File::Find::find(\&TestAusCallback, $testDir);
}

sub TestAusCallback { 
    my $dir = $File::Find::name;
    if (($dir =~ /beta/) or ($dir =~ /release/)) {
        if (not $dir =~ /test/) { 
            die("Non-test directory found in $testDir/aus2.test: $dir");
        }
    }
}

1;