<?php
/**
 * The model file of budget module of ZenTaoPMS.
 *
 * @copyright   Copyright 2009-2015 青岛易软天创网络科技有限公司(QingDao Nature Easy Soft Network Technology Co,LTD, www.cnezsoft.com)
 * @license     ZPL (http://zpl.pub/page/zplv12.html)
 * @author      Chunsheng Wang <chunsheng@cnezsoft.com>
 * @package     budget
 * @version     $Id: model.php 5079 2013-07-10 00:44:34Z chencongzhi520@gmail.com $
 * @link        http://www.zentao.net
 */
?>
<?php
class budgetModel extends model
{
    /**
     * Get subject option.
     *
     * @access public
     * @return object
     */
    public function getSubjectOption()
    {
        $subjects    = $this->loadModel('tree')->getOptionMenu(0, $viewType = 'subject', $startModuleID = 0);
        $subjectList = $this->dao->select('*')->from(TABLE_MODULE)->where('id')->in(array_keys($subjects))->fetchAll();

        foreach($subjectList as $subject)
        {
            if(isset($subjects[$subject->parent])) unset($subjects[$subject->parent]);
            if($subject->grade > 2) unset($subjects[$subject->id]);
        }

        return $subjects;
    }

    /**
     * Get budget by id.
     *
     * @param  int  $budgetID
     * @access public
     * @return array
     */
    public function getByID($budgetID)
    {
        $budget = $this->dao->select('*')->from(TABLE_BUDGET)->where('id')->eq($budgetID)->fetch();
        $budget = $this->loadModel('file')->replaceImgURL($budget, 'desc');

        return $budget;
    }

    /**
     * Get budget list.
     *
     * @param  int    $projectID
     * @param  string $orderBy
     * @param  object $pager
     * @access public
     * @return object
     */
    public function getList($projectID, $orderBy = 'id_desc', $pager = null)
    {
        return $this->dao->select('*')->from(TABLE_BUDGET)->where('PRJ')->eq($projectID)->andWhere('deleted')->eq(0)->orderBy($orderBy)->page($pager)->fetchAll();
    }

    /**
     * Get stages.
     *
     * @param  int    $projectID
     * @access public
     * @return array
     */
    public function getStages($projectID)
    {
        return $this->dao->select('stage')->from(TABLE_BUDGET)->where('PRJ')->eq($projectID)->andWhere('deleted')->eq(0)->orderBy('stage_asc')->fetchPairs();
    }

    /**
     * Get subjects.
     *
     * @param  int    $projectID
     * @access public
     * @return array
     */
    public function getSubjectPairs($projectID)
    {
        return $this->dao->select('subject')->from(TABLE_BUDGET)->where('PRJ')->eq($projectID)->andWhere('deleted')->eq(0)->orderBy('subject_asc')->fetchPairs();
    }

    /**
     * Get subject structure.
     *
     * @access public
     * @return array
     */
    public function getSubjectStructure()
    {
        $this->loadModel('tree');

        $structure    = array();
        $subjectPairs = $this->getSubjectPairs($this->session->PRJ);
        $subjects     = $this->dao->select('*')->from(TABLE_MODULE)->where('id')->in($subjectPairs)->fetchAll('id');

        foreach($subjects as $subject)
        {
            if($subject->grade == 1)
            {
                $structure[$subject->id][] = $subject->id;
                continue;
            }

            $structure[$subject->parent][] = $subject->id;
        }

        return $structure;
    }

    /**
     * Check if has sub-subjects.
     *
     * @param  array  $subjects
     * @access public
     * @return bool
     */
    public function checkSubSubject($subjects)
    {
        foreach($subjects as $id => $subject)
        {
            if(count($subject) > 1 || $id != $subject[0]) return true;
        }

        return false;
    }

    /**
     * Get summary.
     *
     * @param  int    $projectID
     * @param  array  $subjects
     * @access public
     * @return array
     */
    public function getSummary($projectID, $subjects)
    {
        $budgets = $this->dao->select('*')->from(TABLE_BUDGET)->where('PRJ')->eq($projectID)->andWhere('deleted')->eq(0)->fetchAll();
        $stages  = $this->getStages($this->session->PRJ);

        /* Assign each subject to a stage. */
        $summary['stages']   = array();
        $summary['subjects'] = array();
        foreach($stages as $stageID)
        {
            foreach($subjects as $subject)
            {
                foreach($subject as $children)
                {
                    $summary['stages'][$stageID][$children] = 0;
                    $summary['subjects'][$children]         = 0;
                }
            }
        }

        /* Count the costs incurred by a stage corresponding to a subject. */
        foreach($budgets as $budget)
        {
            $summary['stages'][$budget->stage][$budget->subject] += $budget->amount;
            $summary['subjects'][$budget->subject] += $budget->amount;
        }

        return $summary;
    }

    /**
     * Create a budget.
     *
     * @access public
     * @return int
     */
    public function create()
    {
        $budget = fixer::input('post')
            ->setDefault('createdBy', $this->app->user->account)
            ->setDefault('createdDate', helper::today())
            ->setDefault('PRJ', $this->session->PRJ)
            ->cleanFloat('amount')
            ->stripTags($this->config->budget->editor->create['id'], $this->config->allowedTags)
            ->remove('uid')
            ->get();

        $this->dao->insert(TABLE_BUDGET)->data($budget)
            ->autoCheck()
            ->batchCheck($this->config->budget->create->requiredFields, 'notempty')
            ->exec();

        $budgetID = $this->dao->lastInsertID();
        return $budgetID;
    }

    /**
     * Batch create budgets.
     *
     * @access public
     * @return bool
     */
    public function batchCreate()
    {
        $today   = helper::today();
        $budgets = fixer::input('post')->get();

        foreach($budgets->name as $i => $name)
        {
            if(!$name) continue;
            $data = new stdclass();
            $data->name        = $name;
            $data->PRJ         = $this->session->PRJ;
            $data->stage       = $budgets->stage[$i];
            $data->subject     = $budgets->subject[$i];
            $data->amount      = (float)$budgets->amount[$i];
            $data->desc        = nl2br($budgets->desc[$i]);
            $data->createdBy   = $this->app->user->account;
            $data->createdDate = $today;

            foreach(explode(',', $this->config->budget->create->requiredFields) as $field)
            {
                $field = trim($field);
                if($field and empty($data->$field))
                {
                    dao::$errors['message'][] = sprintf($this->lang->error->notempty, $this->lang->budget->$field);
                    return false;
                }
            }


            $this->dao->insert(TABLE_BUDGET)->data($data)->autoCheck()->exec();

            $budgetID = $this->dao->lastInsertID();
            $this->loadModel('action')->create('budget', $budgetID, 'Opened');
        }

        return !dao::isError();
    }

    /**
     * Update a budget.
     *
     * @param  int  $budgetID 
     * @access public
     * @return array
     */
    public function update($budgetID)
    {
        $oldBudget = $this->getByID($budgetID);
        $budget = fixer::input('post')
            ->setDefault('lastEditedBy', $this->app->user->account)
            ->setDefault('lastEditedDate', helper::today())
            ->cleanFloat('amount')
            ->stripTags($this->config->budget->editor->edit['id'], $this->config->allowedTags)
            ->remove('uid')
            ->get();

        $this->dao->update(TABLE_BUDGET)->data($budget)
            ->where('id')->eq($budgetID)
            ->autoCheck()
            ->batchCheck($this->config->budget->create->requiredFields, 'notempty')
            ->exec();

        if(!dao::isError()) return common::createChanges($oldBudget, $budget);
    }
}