DidbsExtractor.pm

package DidbsExtractor;

use DidbsUtils;
use DidbsPackageState;

use File::Path qw/rmtree/;

# Contents set up from values from the dbibsPackage
#
# downloadSuccessFilePath
# 

sub new
{
    my $self = bless {}, shift;
    my $scriptLocation = shift;
    my $packageId = shift;
    my $packageDir = shift;
    my $buildDir = shift;
    my $didbsPackage = shift;
    my $packageState = shift;

    $self->{scriptLocation} = $scriptLocation;
    $self->{packageId} = $packageId;
    $self->{packageDir} = $packageDir;
    $self->{buildDir} = $buildDir;
    $self->{didbsPackage} = $didbsPackage;
    $self->{packageState} = $packageState;

    return $self;
}

sub getState
{
    my $self = shift;
    return $self->{packageState}->getState();
}

sub setState
{
    my $self = shift;
    my $newState = shift;
    $self->{packageState}->setState($newState);
}

sub extractionSuccess
{
    my $self = shift;

    return $self->getState() eq EXTRACTED;
}

sub extractit
{
    my $self = shift;

    my $destdir = $self->{packageDir}."/srcballs";
    mkdirp($destdir);
    my $destfile = $self->{didbsPackage}->{packageFile};
    my $fulldestfile = $destdir."/".$destfile;

    if( $self->getState() ne FETCHED )
    {
        mkdirp($destdir);
        my $destfile = $self->{didbsPackage}->{packageFile};
        my $fulldestfile = $destdir."/".$destfile;

        my $sourceurl = $self->{didbsPackage}->{packageSource};
        my $checksum = $self->{didbsPackage}->{packageChecksum};

        didbsprint "Checking for already downloaded archive from $sourceurl\n";
        my $fetchcmd = $self->{scriptLocation}."/scripts/wgethelper.sh ".$self->{scriptLocation}." $destdir \"$sourceurl\"";

        # If the file exists check the signature
        my $fileexistsgood = 0;
        if( -e $fulldestfile )
        {
            if(!verifyChecksum($self,$fulldestfile,$checksum))
            {
                didbsprint "Stale download at $fulldestfile.\n";
                didbsprint "Removing is commented out.\n";
                exit -1;
                unlink $fulldestfile;
            }
            else
            {
		didbsprint "File exists ($fulldestfile)\n";
                $fileexistsgood = 1;
            }
        }

        if(!$fileexistsgood)
        {
            didbsprint "Fetching...\n";
            didbsprint "\n\n";

            system($fetchcmd) == 0 || die $!;
            $self->setState( FETCHED );
            if(!verifyChecksum($self,$fulldestfile,$checksum))
            {
                didbsprint "Second download attempt failed.\n";
                exit -1;
            }
        }

        # File is good
        didbsprint "Checksum match.\n";
        $self->setState(SIGCHECKED);
    }

    if( $self->getState() == SIGCHECKED )
    {
        my $extractdir = $self->{buildDir}."/".$self->{packageId};
        didbsprint "Removing any existing content at $extractdir\n";
        rmtree $extractdir || die $!;
        mkdirp( $extractdir );
        my $extractor = $self->{didbsPackage}->{packageExtraction};
        my $fullpathext = $self->{scriptLocation}."/scripts/".$extractor;
        didbsprint "Extracting to $extractdir using $fullpathext\n";
        my $cmd = "$fullpathext $extractdir $fulldestfile";
        didbsprint "Command is $cmd\n";
        system($cmd) == 0 || die $!;

        $self->setState(EXTRACTED);
    }

    return $self->getState() == EXTRACTED;
}

sub verifyChecksum
{
    my $self=shift;
    my $filename=shift;
    my $checksum=shift;
    #    didbsprint "Verifying checksum of $filename\n";
    
    my $calcchecksum = sumfile($self->{scriptLocation}, $filename);
    didbsprint "Expected($checksum) - received($calcchecksum)\n";

    return $calcchecksum eq $checksum;
}

sub debug
{
    my $self = shift;
    didbsprint "DibsExtractor constructed for $self->{packageId}\n";
    didbsprint "Status is " . $self->getState() ."\n";
}

1;