<?php
require_once 'modules/admin/models/AddonGateway.php';
require_once 'modules/admin/models/ExportPlugin.php';
require_once 'modules/admin/models/PackageGateway.php';
require_once 'modules/admin/models/PackageTypeGateway.php';
require_once 'modules/billing/models/InvoiceListGateway.php';
require_once 'modules/billing/models/InvoiceEntriesGateway.php';
require_once 'modules/clients/models/ClientListGateway.php';
require_once 'modules/admin/models/ServerGateway.php';

/**
* @package Plugins
*/
class PluginCeformat extends ExportPlugin
{
    /**
     * Header columns names.
     */
    protected $_columns = array(
        'domains' => array('clientid', 'activateddate', 'domainname', 'recurring', 'registrationperiod', 'status', 'nextduedate'),
        'hosting' => array('id', 'clientid', 'username', 'plan', 'regdate', 'domain', 'status', 'nextinvoicedate', 'paymentterm', 'price', 'server'),
        'hosting_addons' => array('clientid', 'packageid', 'addonid', 'addonoptionid', 'nextinvoicedate', 'paymentterm', 'price', 'name'),
        'invoices' => array('id', 'clientid', 'amount', 'datedue', 'datepaid', 'description', 'detail', 'tax', 'subtotal', 'status'),
        'invoices_entries' => array('id', 'clientid', 'invoiceid', 'relid', 'amount', 'tax', 'datedue', 'detail', 'description'),
        'packages' => array('id', 'name', 'monthly', 'quarterly', 'semiannually', 'annually', 'biennial', 'setup', 'description', 'packagetype', 'tax'),
        'packages_addons' => array('id', 'name', 'description'),
        'packages_addons_options' => array('id', 'packageaddonid', 'detail', 'monthly', 'quarterly', 'semiannually', 'annually', 'biannually', 'setup'),
        'packages_groups' => array('id', 'name'),
        'servers' => array('id', 'name', 'ipaddress', 'assignedips', 'statusaddress', 'maxaccounts', 'type', 'username', 'password', 'accesshash', 'secure', 'nameserver1', 'nameserver1ip', 'nameserver2', 'nameserver2ip', 'nameserver3', 'nameserver3ip', 'nameserver4', 'nameserver4ip'),    
        'users' => array('id', 'firstname', 'lastname', 'address', 'email', 'city', 'state', 'zip', 'phone', 'country', 'company', 'status', 'language')
    );
    
    protected $_columnsBuffer = array();
    
    public $_description = 'This export plugin exports all your data to the default ClientExec format, which is a compressed CSV file.';
    
    protected $_filename = null;
    
    /**
     * Lines buffer. For performance propulse.
     */
    protected $_linesBuffer = array();
    
    /**
     * This controls the amount of lines needed to be processed before writting to the file.
     */
    protected $_linesBufferLimit = 100;
    
    public $_title = 'ClientExec Format';
    
    /**
     * Zlib file pointer
     */
    protected $_zp;
    
    protected function _addColumn ($value, $skipEscaping = false)
    {
        $value = str_replace("\r\n", "\n", $value);
        
        if (!$skipEscaping) {
            $value = json_encode($value);
            $value = str_replace(',', '\c', $value);
        }
        
        $this->_columnsBuffer[] = $value;
    }
    
    protected function _addHeader ($section)
    {
        if (!array_key_exists($section, $this->_columns)) {
            throw new Exception("Invalid section '{$section}'.");
        }
        
        $this->_addLine("; {$section}");
        
        foreach ($this->_columns[$section] as $column) {
            $this->_addColumn($column, true);
        }
        
        $this->_addLine();
    }
    
    protected function _addLine ($lineContents = null)
    {
        if ($lineContents === null) {
            if (count($this->_columnsBuffer) < 1) {
                throw new Exception('Cannot add a line without columns.');
            }
            
            $lineContents = implode(',', $this->_columnsBuffer);
            $this->_columnsBuffer = array();
        }
        
        $lineContents = trim($lineContents);
        
        if (empty($lineContents)) {
            throw new Exception('Cannot add an empty line to the file.');
        }
        
        $lineContents .= "\n";
        
        $this->_linesBuffer[] = $lineContents;
        
        if (count($this->_linesBuffer) >= $this->_linesBufferLimit) {
            $this->_clearLinesBuffer();
        }
    }
    
    protected function _clearLinesBuffer ()
    {
        foreach ($this->_linesBuffer as $line) {
            if (!gzwrite($this->_zp, $line)) {
                throw new Exception("Unable to write to the file '{$this->_filename}'.");
            }
        }
        
        $this->_linesBuffer = array();
    }
    
    protected function _export ()
    {
        $this->_setupFile();
        
        if (!($this->_zp = gzopen($this->_filename, 'w9'))) {
            throw new Exception("Unable to open the file '{$this->_filename}'.");
        }
        
        $this->_exportUsers();
        $this->_exportServers();
        $this->_exportPackagesGroups();
        $this->_exportPackages();
        $this->_exportPackagesAddons();
        $this->_exportHosting();
        $this->_exportInvoices();
        $this->_exportInvoicesEntries();
        $this->_exportDomains();
        $this->_exportHostingAddons();
        $this->_clearLinesBuffer();
        
        if (!gzclose($this->_zp)) {
            throw new Exception("Unable to close the file '{$this->_filename}'.");
        }
        
        $contents = @file_get_contents($this->_filename);
        @unlink($this->_filename);
        
        return $contents;
    }
    
    function _exportServers()
    {
        require_once('modules/admin/models/server.php');
        $this->_addHeader('servers');
        $serverGateway = new ServerGateway();
        $servers = $serverGateway->getServersGridList();
        foreach ( $servers['data'] as $data ) {
            $nameServers = $serverGateway->getNameServers($data['id']);
            $this->_addColumn($data['id']);
            $this->_addColumn($data['name']);
            $this->_addColumn($data['sharedip']);
            $this->_addColumn('');
            $this->_addColumn($data['statsurl']);
            $this->_addColumn($data['rawAmount']);
            $this->_addColumn(strtolower($data['plugin']));
            $this->_addColumn('');
            $this->_addColumn('');
            $this->_addColumn('');
            $this->_addColumn('');
            $this->_addColumn(@$nameServers['data'][0]['hostname']);
            $this->_addColumn(@$nameServers['data'][0]['ip']);
            $this->_addColumn(@$nameServers['data'][1]['hostname']);
            $this->_addColumn(@$nameServers['data'][1]['ip']);
            $this->_addColumn(@$nameServers['data'][2]['hostname']);
            $this->_addColumn(@$nameServers['data'][2]['ip']);
            $this->_addColumn(@$nameServers['data'][3]['hostname']);
            $this->_addColumn(@$nameServers['data'][3]['ip']);
            $this->_addLine();
        }
    }
    
    protected function _exportDomains ()
    {
        $this->_addHeader('domains');
        // Domains are exported directly with Packages
    }
    
    protected function _exportHosting ()
    {
        $this->_addHeader('hosting');
        $gateway = new UserPackageGateway;
        $userPackages = $gateway->getUserPackagesIterator();
        while ($userPackage = $userPackages->fetch()) {
            $userPackage->loadCustomFields();
            $this->_addColumn($userPackage->getId());
            $this->_addColumn($userPackage->CustomerId);
            $this->_addColumn($userPackage->getCustomField('User Name'));
            $this->_addColumn($userPackage->Plan);
            $this->_addColumn($userPackage->dateActivated);
            $this->_addColumn($userPackage->getCustomField('Domain Name'));
            $this->_addColumn($userPackage->status);
            $this->_addColumn($userPackage->getRecurringFeeEntry()->getNextBillDate());
            $this->_addColumn($userPackage->getRecurringFeeEntry()->getPaymentTerm());
            $this->_addColumn($userPackage->getRecurringFeeEntry()->GetAmount());
            $this->_addColumn($userPackage->getCustomField('Server Id'));
            $this->_addLine();
        }
    }
    
    protected function _exportHostingAddons ()
    {
        $this->_addHeader('hosting_addons');
    }
    
    protected function _exportInvoices ()
    {
        $this->_addHeader('invoices');
        $gateway = new InvoiceListGateway;
        $invoices = $gateway->getAllInvoices();
        
        foreach ($invoices as $invoice) {
            $this->_addColumn($invoice->m_InvoiceID);
            $this->_addColumn($invoice->m_UserID);
            $this->_addColumn($invoice->m_Price);
            $this->_addColumn($invoice->m_Date);
            $this->_addColumn($invoice->m_DatePaid);
            $this->_addColumn($invoice->m_Description);
            $this->_addColumn(null);
            $this->_addColumn($invoice->m_Tax);
            $this->_addColumn($invoice->m_SubTotal);
            $this->_addColumn($invoice->m_Status);
            $this->_addLine();
        }
    }
    
    protected function _exportInvoicesEntries ()
    {
        $this->_addHeader('invoices_entries');
        $gateway = new InvoiceEntriesGateway;
        $entries = $gateway->getInvoiceEntries('id', 'ASC');
        $entries = $entries['invoiceentriesiterator'];
        while ($entry = $entries->fetch()) {
            $this->_addColumn($entry->getId());
            $this->_addColumn($entry->getCustomerId());
            $this->_addColumn($entry->getInvoiceId());
            $this->_addColumn('');
            $this->_addColumn($entry->getPrice());
            $this->_addColumn($entry->getTaxable());
            $this->_addColumn($entry->getDate());
            $this->_addColumn($entry->getDetail());
            $this->_addColumn($entry->getDescription());
            $this->_addLine();
        }
    }
    
    protected function _exportPackages ()
    {
        $this->_addHeader('packages');
        $gateway = new PackageGateway;
        $packages = $gateway->getProductsGrid();
        
        foreach ($packages as $package) {
            $model = new Package($package['id']);
            $pricing = $model->getProductPricing();
            $this->_addColumn($model->id);
            $this->_addColumn($model->planname);
            
            if (is_array($pricing)) {
                $this->_addColumn($model->getPrice(1));
                $this->_addColumn($model->getPrice(3));
                $this->_addColumn($model->getPrice(6));
                $this->_addColumn($model->getPrice(12));
                $this->_addColumn($model->getPrice(24));
                $this->_addColumn($model->getSetupFee());
            } else {
                $this->_addColumn('');
                $this->_addColumn('');
                $this->_addColumn('');
                $this->_addColumn('');
                $this->_addColumn('');
                $this->_addColumn('');
            }
            
            $this->_addColumn($model->description);
            $this->_addColumn($model->planid);
            
            if (is_array($pricing)) {
                $this->_addColumn($model->isTaxable()); // #fix - correct?
            } else {
                $this->_addColumn('');
            }
            
            $this->_addLine();
        }
    }
    
    protected function _exportPackagesAddons ()
    {
        $gateway = new AddonGateway;
        $addons = $gateway->getAllAddons();
        $aAddon = array();
        
        while ($addon = $addons->fetch()) {
            $aAddon[$addon->getId()] = $addon;
        }
        
        $this->_addHeader('packages_addons');
        
        foreach ($aAddon as $addon) {
            $this->_addColumn($addon->getId());
            $this->_addColumn($addon->getName());
            $this->_addColumn($addon->getDescription());
            $this->_addLine();
        }
        
        $this->_addHeader('packages_addons_options');
        $ids = array();
        
        foreach ($aAddon as $addon) {
            $addon->getPrices();
            
            foreach ($addon->prices as $price) {
                $this->_addColumn($price['id']);
                $this->_addColumn($addon->getId());
                $this->_addColumn($price['detail']);
                $this->_addColumn($price['price1']);
                $this->_addColumn($price['price3']);
                $this->_addColumn($price['price6']);
                $this->_addColumn($price['price12']);
                $this->_addColumn($price['price24']);
                $this->_addColumn(0); // #fix - setup price?
                $this->_addLine();
                
                $ids[] = $price['id'];
            }
        }
    }
    
    protected function _exportPackagesGroups ()
    {
        $this->_addHeader('packages_groups');
        $gateway = new PackageTypeGateway;
        $groups = $gateway->getPackageTypes();
        
        while ($group = $groups->fetch()) {
            $this->_addColumn($group->getId());
            $this->_addColumn($group->getName());
            $this->_addLine();
        }
    }
    
    protected function _exportUsers ()
    {
        $this->_addHeader('users');
        $gateway = new ClientListGateway;
        $clients = $gateway->getClients(false, 0, 'id', 'asc', null, '', 'allclients', false);
        
        while ($client = $clients->fetch()) {
            $model = new User($client->getId());
            $this->_addColumn($model->getId());
            $this->_addColumn($model->getFirstName());
            $this->_addColumn($model->getLastName());
            $this->_addColumn($model->getAddress());
            $this->_addColumn($model->getEmail());
            $this->_addColumn($model->getCity());
            $this->_addColumn($model->getState());
            $this->_addColumn($model->getZipCode());
            $this->_addColumn($model->getPhone());
            $this->_addColumn($model->getCountry());
            $this->_addColumn($model->getOrganization());
            $this->_addColumn($model->getStatus());
            $this->_addColumn($model->getRealLanguage());
            $this->_addLine();
        }
    }
    
    protected function _setupFile ()
    {
        $this->_filename = tempnam(sys_get_temp_dir(), 'PHP');
        
        if (!file_exists($this->_filename)) {
            // Attempt to create the file
            if (!touch($this->_filename)) {
                throw new Exception('Unable to create the temporary file.');
            }
        }
        
        if (!is_writable($this->_filename)) {
            // Attempt to give write permissions
            if (!chmod($this->_filename, 0666)) {
                throw new Exception('Unable to set temporary file permissions.');
            }
        }
    }
    
    public function getFileContentsUncompressed ()
    {
        $fileContents = '';
        
        $zd = gzopen($this->_filename, 'r');
        
        while (!gzeof($zd)) {
            $buffer = gzgets($zd);
            $fileContents .= $buffer;
        }
        
        gzclose($zd);
        
        return $fileContents;
    }
    
    public function getForm()
    {
        $this->tpl->define(array('PluginClientdata' => 'views/admin/PluginCeformat.tpl'));

        return $this->tpl->subst('PluginClientdata_out', 'PluginClientdata');
    }

    public function process($post)
    {
        $contents = $this->_export();
        $filename = 'clientexec.csv.gz';
        NE_Lib::download($contents, $filename);
    }
}
