<?php
/**
 *本文件主要进行生成每个脚本文件对应的测试数据yaml文件。
 *
 * All request of entries should be routed by this router.
 *
 * @copyright   Copyright 2009-2017 青岛易软天创网络科技有限公司(QingDao Nature Easy Soft Network Technology Co,LTD, www.cnezsoft.com)
 * @license     ZPL (http://zpl.pub/page/zplv12.html)
 * @author      liyang <liyang@easycorp.ltd>
 * @package     ZenTaoPMS
 * @version     $Id: $
 * @link        http://www.zentao.net/
 */
class yaml
{
    /**
     * Global config.
     *
     * @var object
     * @access public
     */
    public $config;

    /**
     * Filed Arr.
     *
     * @var array
     * @access public
     */
    public  $fieldArr = array();

    /**
     * Field.
     *
     * @var int
     * @access private
     */
    private $field;

    /**
     * Yaml Dir root.
     *
     * @var int
     * @access public
     */
    public $yamlDir;

    /**
     * __construct function load config.
     *
     * @access public
     * @return void
     */
    public function __construct()
    {
        global $config;
        $this->config  = $config;
        $this->yamlDir = dirname(__FILE__, 2) . '/model';
    }

    /**
     * Magic method, return fild.
     *
     * @param  string    $property_name
     * @access protected
     * @return object
     */
    public function __get($property_name)
    {
        $this->setField($property_name);
        return $this;
    }

    /**
     * Set yaml filed.
     *
     * @param  string    $value
     * @access public
     * @return object
     */
    public function setField($value)
    {
        $this->fieldArr[$value] = array();
        $this->field = $value;
        return $this;
    }

    /**
     * Set filed rang.
     *
     * @param  string    $range
     * @access public
     * @return object
     */
    public function range($range)
    {
        $this->fieldArr[$this->field]['range'] = $range;
        return $this;
    }

    /**
     * Set field prefix.
     *
     * @param  string    $prefix
     * @access public
     * @return object
     */
    public function prefix($prefix)
    {
        $this->fieldArr[$this->field]['prefix'] = $prefix;
        return $this;
    }

    /**
     * Set field postfix.
     *
     * @param  string    $postfix
     * @access public
     * @return object
     */
    function postfix($postfix)
    {
        $this->fieldArr[$this->field]['postfix'] = $postfix;
        return $this;
    }

    /**
     * Set field type.
     *
     * @param  string    $type
     * @access public
     * @return object
     */
    function type($type)
    {
        $this->fieldArr[$this->field]['type'] =  $type;
        return $this;
    }

    /**
     * Set field format.
     *
     * @param  string    $format
     * @access public
     * @return object
     */
    function format($format)
    {
        $this->fieldArr[$this->field]['format'] =  $format;
        return $this;
    }

    /**
     * set field fields.
     *
     * @param  array    $fields
     * @access public
     * @return object
     */
    function fields($fields)
    {
        if(!is_array($fields))
        {
            echo "fileds must be an array";
            return;
        }

        $this->fieldArr[$this->field]['fields'] =  $fields;
        return $this;
    }

    /**
     * get field array value.
     *
     * @access public
     * @return array
     */
    function getField()
    {
        return $this->fieldArr;
    }

    /**
     * Assembly field generation rules.
     *
     * @param  array     $filedArr
     * @access public
     * @return array
     */
    function setFieldRule($filedArr)
    {
        $ruleArr = array();
        $index   = 0;

        foreach($filedArr as $filed => $rule)
        {
            $ruleArr[$index]['field'] = $filed;

            if(array_key_exists('fields', $rule))
            {
                $ruleArr[$index]['fields'] = $this->setFieldRule($rule['fields']);
            }
            else
            {
                if(!empty($rule['range'])) $ruleArr[$index]['range']   = $rule['range'];
            }

            if(!empty($rule['prefix']))  $ruleArr[$index]['prefix']  = $rule['prefix'];
            if(!empty($rule['postfix'])) $ruleArr[$index]['postfix'] = $rule['postfix'];
            if(!empty($rule['type']))    $ruleArr[$index]['type']    = $rule['type'];
            if(!empty($rule['format']))  $ruleArr[$index]['format']  = $rule['format'];
            $index++;
        }

        return $ruleArr;
    }

    /**
     * Build yaml file.
     *
     * @param  string    $model
     * @param  string    $name
     * @param  string    $version
     * @access public
     * @return void
     */
    function build($model, $name, $version = '')
    {
        if(!is_dir($this->yamlDir . "/{$model}/data")) mkdir($this->yamlDir . "/{$model}/data", 0700);
        $yamlFile = $this->yamlDir . "/{$model}/data/{$name}.yaml";

        $yamlDataArr = array();

        $yamlDataArr['title']   = "zt_{$name}";
        $yamlDataArr['author']  = "auto_{$name}";
        $version ? $yamlDataArr['version'] = $version : $yamlDataArr['version'] = '1.0';

        if(empty($this->fieldArr)) return;
        $yamlDataArr['fields'] = $this->setFieldRule($this->fieldArr);

        yaml_emit_file($yamlFile, $yamlDataArr);
    }

    /**
     * Insert the data into database.
     *
     * @param  string    $model
     * @param  string    $tableName
     * @param  int       $rows
     * @access public
     * @return string
     */
    function insertDB($model, $file, $tableName, $rows, $isClear = false)
    {
        $yamlFile     = $this->yamlDir . "/{$model}/data/{$file}.yaml";
        $tableSqlDir  = $this->yamlDir . "/{$model}/data/sql";

        if(!is_dir($tableSqlDir)) mkdir($tableSqlDir, 0700);
        $dumpCommand = "mysqldump -u%s -p%s -h%s -P%s %s %s > {$tableSqlDir}/{$tableName}.sql";

        $runtimeRoot = dirname(dirname(__FILE__)) . '/runtime/';
        $zdPath      = $runtimeRoot . 'zd';
        $configYaml  = $runtimeRoot . 'tmp/config.yaml';

        $tableName = $this->config->db->prefix . $tableName;
        $dbName    = $this->config->db->name;
        $dbHost    = $this->config->db->host;
        $dbPort    = $this->config->db->port;
        $dbUser    = $this->config->db->user;
        $dbPWD     = $this->config->db->password;

        $command  = "$zdPath -c %s -d %s -n %d -t %s --trim -dns mysql://%s:%s@%s:%s/%s#utf8";
        if($isClear === true) $command .= ' --clear';
        $execYaml = sprintf($command, $configYaml, $yamlFile, $rows, $tableName, $dbUser, $dbPWD, $dbHost, $dbPort, $dbName);
        $execDump = sprintf($dumpCommand, $dbUser, $dbPWD, $dbHost, $dbPort, $dbName, $tableName);
        system($execDump);
        system($execYaml);
    }

    /**
     * Restore table data.
     *
     * @param  string    $model
     * @param  string    $tableName
     * @access public
     * @return mixed
     */
    public function restoreTable($model, $tableName)
    {
        $tableSql = $this->yamlDir . "/{$model}/data/sql/$tableName.sql";
        if(!is_file($tableSql)) return false;

        $dbName    = $this->config->db->name;
        $dbHost    = $this->config->db->host;
        $dbPort    = $this->config->db->port;
        $dbUser    = $this->config->db->user;
        $dbPWD     = $this->config->db->password;

        $command     = "mysql -u%s -p%s -h%s -P%s %s < %s";
        $execRestore = sprintf($command, $dbUser, $dbPWD, $dbHost, $dbPort, $dbName, $tableSql);
        system($execRestore);
    }
}