Fix #49: Add Support for 'CREATE TABLE `table_copy` LIKE `table`;'

Signed-off-by: Deven Bansod <devenbansod.bits@gmail.com>

1 files changed, 38 insertions, 6 deletions

diff --git a/src/Statements/CreateStatement.php b/src/Statements/CreateStatement.php
index 3ccf78f..e9ffe85 100644
--- a/src/Statements/CreateStatement.php
+++ b/src/Statements/CreateStatement.php
@@ -11,6 +11,7 @@ namespace SqlParser\Statements;
 use SqlParser\Parser;
 use SqlParser\Statement;
 use SqlParser\Token;
+use SqlParser\Context;
 use SqlParser\TokensList;
 use SqlParser\Components\ArrayObj;
 use SqlParser\Components\DataType;
@@ -188,6 +189,15 @@ class CreateStatement extends Statement
     public $select;
 
     /**
+     * If `CREATE TABLE ... LIKE`
+     *
+     * Used by `CREATE TABLE`
+     *
+     * @var Expression
+     */
+    public $like;
+
+    /**
      * Expression used for partitioning.
      *
      * @var string
@@ -282,6 +292,11 @@ class CreateStatement extends Statement
                 . OptionsArray::build($this->options) . ' '
                 . Expression::build($this->name) . ' '
                 . $this->select->build();
+        } elseif ($this->options->has('TABLE') && !is_null($this->like)) {
+            return 'CREATE '
+                . OptionsArray::build($this->options) . ' '
+                . Expression::build($this->name) . ' LIKE '
+                . Expression::build($this->like);
         } elseif ($this->options->has('TABLE')) {
             $partition = '';
 
@@ -389,17 +404,34 @@ class CreateStatement extends Statement
                 $list,
                 static::$DB_OPTIONS
             );
-        } elseif ($this->options->has('TABLE') && ($token->type == Token::TYPE_KEYWORD) && ($token->value == 'SELECT')) {
+        } elseif ($this->options->has('TABLE')
+            && ($token->type == Token::TYPE_KEYWORD)
+            && ($token->value == 'SELECT')
+        ) {
             /* CREATE TABLE ... SELECT */
             $this->select = new SelectStatement($parser, $list);
-        } elseif (
-                $this->options->has('TABLE') &&
-                ($token->type == Token::TYPE_KEYWORD) && ($token->value == 'AS') &&
-                ($list->tokens[$nextidx]->type == Token::TYPE_KEYWORD) && ($list->tokens[$nextidx]->value == 'SELECT')
-            ) {
+        } elseif ($this->options->has('TABLE')
+            && ($token->type == Token::TYPE_KEYWORD) && ($token->value == 'AS')
+            && ($list->tokens[$nextidx]->type == Token::TYPE_KEYWORD)
+            && ($list->tokens[$nextidx]->value == 'SELECT')
+        ) {
             /* CREATE TABLE ... AS SELECT */
             $list->idx = $nextidx;
             $this->select = new SelectStatement($parser, $list);
+        } elseif ($this->options->has('TABLE')
+            && $token->type == Token::TYPE_KEYWORD
+            && $token->value == 'LIKE'
+        ) {
+            /* CREATE TABLE `new_tbl` LIKE 'orig_tbl' */
+            $list->idx = $nextidx;
+            $this->like = $this->table = Expression::parse(
+                $parser,
+                $list,
+                array(
+                    'parseField' => 'table',
+                    'breakOnAlias' => true,
+                )
+            );
         } elseif ($this->options->has('TABLE')) {
             $this->fields = CreateDefinition::parse($parser, $list);
             if (empty($this->fields)) {