#!/usr/bin/perl

use strict;
use Getopt::Std;
use IO::File;
use Date::Calc;
use Data::Dumper;

use lib '/app/tools/common/lib';
use Common::RSApp;
use Common::Assert;

use lib '/app/tools/rps/lib';
use RPS::DB::Item::Expense;
use RPS::DB::Item::ExpenseType;
use RPS::DB::Item::ExpenseName;
use RPS::DB::Item::AlbumContract;
use RPS::DB::Item::TrackContract;
use RPS::DB::Item::AlbumContractAdvance;
use RPS::DB::Item::TrackContractAdvance;
use RPS::DB::Item::FinanceTransaction;

# Parse the command-line.
#
my ($clientID) = parseCommandLine();

$| = 1;

# Instantiate the application singleton object.
#
my $appSingleton = Common::RSApp->new( clientID => $clientID );

# create expense name
#
my $expenseNameObj = RPS::DB::Item::ExpenseName->Lookup( name => "Advance" );

if ( !$expenseNameObj ) {
    $expenseNameObj = RPS::DB::Item::ExpenseName->Create( name => "Advance" );
    $expenseNameObj->save();
}

# Commit any necessary updates.
#
convertAdvances();

sub convertAdvances {
    my %contractExpenseType = ();

    # do album advances first
    #
    my $albAdvColl = RPS::DB::Item::AlbumContractAdvance->GetAll();
    while ( $albAdvColl->hasNext() ) {
        my $albumContractAdvanceObj = $albAdvColl->next();

        my $albumContractObj = RPS::DB::Item::AlbumContract->Lookup( album_contract_id => $albumContractAdvanceObj->album_contract_id );
        next if ( !$albumContractObj );

        my $contractID = $albumContractObj->artist_contract_id;

        if ( !$contractExpenseType{$contractID} ) {

            # create an expense type for advance
            #
            my $newExpenseType = RPS::DB::Item::ExpenseType->Create(
                artist_contract_id => $contractID,
                expense_name_id    => $expenseNameObj->expense_name_id,
                percent            => 100,
            );
            $newExpenseType->save();
            $contractExpenseType{$contractID} = $newExpenseType->expense_type_id;
        }

        createExpenses(
            parentID      => $albumContractObj->album_contract_id,
            parentType    => RPS::DB::Item::Expense::kParentAlbumContract(),
            accountID     => $albumContractAdvanceObj->advance_account_id,
            expenseTypeID => $contractExpenseType{$contractID},
        );
    }

    # next do track advances
    #
    my $trkAdvColl = RPS::DB::Item::TrackContractAdvance->GetAll();
    while ( $trkAdvColl->hasNext() ) {
        my $trackContractAdvanceObj = $trkAdvColl->next();

        my $trackContractObj = RPS::DB::Item::TrackContract->Lookup( track_contract_id => $trackContractAdvanceObj->track_contract_id );
        next if ( !$trackContractObj );

        my $contractID = $trackContractObj->artist_contract_id;

        if ( !$contractExpenseType{$contractID} ) {

            # create an expense type for advance
            #
            my $newExpenseType = RPS::DB::Item::ExpenseType->Create(
                artist_contract_id => $contractID,
                expense_name_id    => $expenseNameObj->expense_name_id,
                percent            => 100,
            );
            $newExpenseType->save();
            $contractExpenseType{$contractID} = $newExpenseType->expense_type_id;
        }

        createExpenses(
            parentID      => $trackContractObj->track_contract_id,
            parentType    => RPS::DB::Item::Expense::kParentTrackContract(),
            accountID     => $trackContractAdvanceObj->advance_account_id,
            expenseTypeID => $contractExpenseType{$contractID},
        );
    }
}

sub createExpenses {
    my (%args) = @_;

    my $parentID      = $args{parentID};
    my $parentType    = $args{parentType};
    my $accountID     = $args{accountID};
    my $expenseTypeID = $args{expenseTypeID};

    my $transactionColl = RPS::DB::Item::FinanceTransaction->GetAccountTransactions( finance_account_id => $accountID );

    while ( $transactionColl->hasNext() ) {
        my $transactionObj = $transactionColl->next();

        my %params = (
            expense_type_id => $expenseTypeID,
            amount          => $transactionObj->amount,
            parent_type     => $parentType,
            parent_id       => $parentID,

            # usually this would be copied from the expense type, but we'll just hard code it here because we know it should always be 100
            percent => 100,
            memo    => $transactionObj->memo,
        );

        my $newExpense = RPS::DB::Item::Expense->Create(%params);
        $newExpense->save();
    }
}

sub parseCommandLine {
    my %opt;
    getopts( 'c:', \%opt );

    my $clientID = $opt{c};

    unless ( $clientID =~ /^\d+$/ && $clientID > 0 ) {
        die usage("You must specify a valid client ID");
    }

    return $clientID;
}

sub usage {
    my $errstr = shift;
    my $text = ($errstr) ? "ERROR: $errstr\n" : '';
    $text .= "Usage: $0 -c <clientID> -f <inputFile>\n";
    return $text;
}