1:   2:   3:   4:   5:   6:   7:   8:   9:  10:  11:  12:  13:  14:  15:  16:  17:  18:  19:  20:  21:  22:  23:  24:  25:  26:  27:  28:  29:  30:  31:  32:  33:  34:  35:  36:  37:  38:  39:  40:  41:  42:  43:  44:  45:  46:  47:  48:  49:  50:  51:  52:  53:  54:  55:  56:  57:  58:  59:  60:  61:  62:  63:  64:  65:  66:  67:  68:  69:  70:  71:  72:  73:  74:  75:  76:  77:  78:  79:  80:  81:  82:  83:  84:  85:  86:  87:  88:  89:  90:  91:  92:  93:  94:  95:  96:  97:  98:  99: 100: 101: 102: 103: 104: 105: 106: 107: 108: 109: 110: 111: 112: 113: 114: 115: 116: 117: 118: 119: 120: 121: 122: 123: 124: 125: 126: 127: 128: 129: 130: 131: 132: 133: 134: 135: 136: 137: 138: 139: 140: 141: 142: 143: 144: 145: 146: 147: 148: 149: 150: 151: 152: 153: 154: 155: 156: 157: 158: 159: 160: 161: 162: 163: 164: 165: 166: 167: 168: 169: 170: 

<?php

    namespace CzProject\SqlGenerator\Statements;

    use CzProject\SqlGenerator\DuplicateException;
    use CzProject\SqlGenerator\Helpers;
    use CzProject\SqlGenerator\IDriver;
    use CzProject\SqlGenerator\IStatement;


    class CreateTable implements IStatement
    {
        /** @var string */
        private $tableName;

        /** @var array  [name => ColumnDefinition] */
        private $columns = array();

        /** @var array  [name => IndexDefinition] */
        private $indexes = array();

        /** @var array  [name => ForeignKeyDefinition] */
        private $foreignKeys = array();

        /** @var string|NULL */
        private $comment;

        /** @var array  [name => value] */
        private $options = array();


        /**
         * @param  string
         */
        public function __construct($tableName)
        {
            $this->tableName = $tableName;
        }


        /**
         * @param  string
         * @param  string
         * @return ColumnDefinition
         */
        public function addColumn($name, $type, array $parameters = NULL, array $options = array())
        {
            if (isset($this->columns[$name])) {
                throw new DuplicateException("Column '$name' already exists.");
            }

            return $this->columns[$name] = new ColumnDefinition($name, $type, $parameters, $options);
        }


        /**
         * @param  string
         * @param  string
         * @return IndexDefinition
         */
        public function addIndex($name, $type)
        {
            if (isset($this->indexes[$name])) {
                throw new DuplicateException("Index '$name' already exists.");
            }

            return $this->indexes[$name] = new IndexDefinition($name, $type);
        }


        /**
         * @param  string
         * @param  string[]|string
         * @param  string
         * @param  string[]|string
         * @return ForeignKeyDefinition
         */
        public function addForeignKey($name, $columns = array(), $targetTable, $targetColumns = array())
        {
            if (isset($this->foreignKeys[$name])) {
                throw new DuplicateException("Foreign key '$name' already exists.");
            }

            return $this->foreignKeys[$name] = new ForeignKeyDefinition($name, $columns, $targetTable, $targetColumns);
        }


        /**
         * @param  string|NULL
         * @return static
         */
        public function setComment($comment)
        {
            $this->comment = $comment;
            return $this;
        }


        /**
         * @param  string
         * @param  string
         * @return static
         */
        public function setOption($name, $value)
        {
            $this->options[$name] = $value;
            return $this;
        }


        /**
         * @return string
         */
        public function toSql(IDriver $driver)
        {
            $output = 'CREATE TABLE ' . $driver->escapeIdentifier($this->tableName) . " (\n";

            // columns
            $isFirst = TRUE;

            foreach ($this->columns as $column) {
                if ($isFirst) {
                    $isFirst = FALSE;

                } else {
                    $output .= ",\n";
                }

                $output .= "\t" . $column->toSql($driver);
            }

            foreach ($this->indexes as $index) {
                if ($isFirst) {
                    $isFirst = FALSE;

                } else {
                    $output .= ",\n";
                }

                $output .= "\t" . $index->toSql($driver);
            }

            foreach ($this->foreignKeys as $foreignKey) {
                if ($isFirst) {
                    $isFirst = FALSE;

                } else {
                    $output .= ",\n";
                }

                $output .= "\t" . $foreignKey->toSql($driver);
            }

            $output .= "\n)";

            if (isset($this->comment)) {
                $output .= "\n";
                $output .= 'COMMENT ' . $driver->escapeText($this->comment);
            }

            foreach ($this->options as $optionName => $optionValue) {
                $output .= "\n";
                $output .= $optionName . '=' . $optionValue;
            }

            $output .= ';';
            return $output;
        }
    }