<?php
/**
 * AddInvoicePage.class.php
 *
 * This file contains the definition for the AddInvoicePage class
 *
 * @package Pages
 * @author John Diamond <hide@address.com>
 * @copyright John Diamond <hide@address.com>
 * @license http://www.opensource.org/licenses/gpl-license.php GNU Public License
 */

// Include the parent class
require BASE_PATH . "include/SolidStatePage.class.php";

/**
 * AddInvoicePage
 *
 * Creates a new invoice.
 *
 * @package Pages
 * @author John Diamond <hide@address.com>
 */
class AddInvoicePage extends SolidStatepage {
    /**
     * Initialize Add Invoice Page
     */
    function init() {
        parent::init();

        if ( !isset( $this->post['periodend'] ) ) {
            // Set the end of the invoice period to be 1 month ahead of today
            $today = getdate( time() );
            $newDate = DBConnection::format_datetime( mktime( null, null, null, $today['mon']+1 ) );
            $this->smarty->assign( "nextMonth", $newDate );
        }

        if ( isset( $this->get['account'] ) ) {
            $this->setURLField( "account", $this->get['account']->getID() );
            $this->session['account_dbo'] =& $this->get['account'];
            $this->smarty->assign( "account",
                    $this->get['account']->getID() );
            $this->smarty->assign( "account_name",
                    $this->get['account']->getAccountName() );
        }
    }

    /**
     * Action
     *
     * Actions handled by this page:
     *   new_invoice
     *
     * @param string $action_name Action
     */
    function action( $action_name ) {
        switch ( $action_name ) {
            case "new_invoice":
                if ( isset( $this->post['cancel'] ) ) {
                    // Cancel
                    $this->cancel();
                }
                elseif  ( isset( $this->post['continue'] ) ) {
                    // Generate Invoice
                    $this->generate_invoice();
                }
                break;

            default:
                // No matching action, refer to base class
                parent::action( $action_name );
        }
    }

    /**
     * Form Canceled
     */
    function cancel() {
        $this->goback();
    }

    /**
     * Generate Invoice
     *
     * Creates a new Invoice and adds it to the database.
     */
    function generate_invoice() {
        // Determine the correct source of the account ID
        $account_id = isset( $this->get['account'] ) ?
                $this->get['account']->getID() : $this->post['account']->getID();

        // Create a new invoice DBO
        $invoice = new InvoiceDBO();
        $invoice->setAccountID( $account_id );
        $invoice->setDate( DBConnection::format_datetime( $this->post['date'] ) );
        $invoice->setPeriodBegin( DBConnection::format_datetime( $this->post['periodbegin'] ) );
        $invoice->setPeriodEnd( DBConnection::format_datetime( $this->post['periodend'] ) );
        $invoice->setNote( $this->post['note'] );
        $invoice->setTerms( $this->post['terms'] );

        // Generate lineitems
        $invoice->generate();

        // Insert invoice into database
        add_InvoiceDBO( $invoice );

        // Success
        $this->setMessage( array( "type" => "[INVOICE_CREATED]" ) );
        $this->gotoPage( "billing_view_invoice",
                null,
                "invoice=" . $invoice->getID() );
    }
}

?>