#------------------------------------------------------------
# Copyright (C) 2006 RoyaltyShare, Inc.   All Rights Reserved
#------------------------------------------------------------
package RPS::Statement::Command::DownloadMechanicalPDF;
use strict;
use warnings;

use IO::File;

use lib '/app/tools/rps/lib';
use RPS::DB::Item::MechanicalStatement;
use RPS::Statement::Mechanical::US::PDF;

use lib '/app/tools/common/lib';
use Common::WriteXML;
use RSApache::Response::Download;
use Common::Timer;

use RPS::Command;
use RPS::CommandLog;
use base 'RPS::Command';

sub cmd  { return 'pdf_mechanical'; }
sub area { return 'statement'; }

sub execute {
    my ($self) = @_;

    my $timer = Common::Timer->new();

    my $response = $self->SUPER::execute();
    return $response if $response;

    my $id = $self->getParam("MechanicalStatementID");

    my $statementData = RPS::DB::Item::MechanicalStatement->Lookup( mechanical_statement_id => $id );
    my $runID = $statementData->mechanical_run_id;

    # Need to wrap a bit of abstraction around these file paths. Hard coding isn't that great...
    #
    my $prettyFileName;
    my $fileName;

    my $fullStatement = $self->getParam("_full");
    if ($fullStatement) {
        $prettyFileName = RPS::Statement::Mechanical::US::PDF->PrettyLabelStatementFileNameFromDBItem($statementData);
        $fileName       = RPS::Statement::Mechanical::US::PDF->LabelStatementFileNameFromDBItem($statementData);
    } else {
        $prettyFileName = RPS::Statement::Mechanical::US::PDF->PrettyStatementFileNameFromDBItem($statementData);
        $fileName       = RPS::Statement::Mechanical::US::PDF->StatementFileNameFromDBItem($statementData);
    }

    my $fullPath = RPS::Statement::Mechanical::US::PDF->StatementPathFromRunID($runID) . $fileName;

    my $fh = new IO::File;
    open( $fh, "$fullPath" ) or die "ERROR: Unable to open $fullPath for reading: $!";

    my $contentType = 'application/pdf; name=' . $prettyFileName;

    my @stat     = $fh->stat;
    my $fileSize = $stat[7];
    $response = RSApache::Response::Download->new( $fh, $contentType, $prettyFileName, $fileSize );

    return $response;
}

1;