1 files changed, 69 insertions, 0 deletions
diff --git a/src/Utils/Table.php b/src/Utils/Table.php
new file mode 100644
index 0000000..7560f9d
--- /dev/null
+++ b/src/Utils/Table.php
@@ -0,0 +1,69 @@
+<?php
+
+namespace SqlParser\Utils;
+
+use SqlParser\Lexer;
+use SqlParser\Parser;
+use SqlParser\Fragments\DataTypeFragment;
+use SqlParser\Fragments\ParamDefFragment;
+use SqlParser\Statements\CreateStatement;
+
+/**
+ * Table utilities.
+ *
+ * @category   Tables
+ * @package    SqlParser
+ * @subpackage Utils
+ * @author     Dan Ungureanu <udan1107@gmail.com>
+ * @license    http://opensource.org/licenses/GPL-2.0 GNU Public License
+ */
+class Table
+{
+
+    public static function getFields(CreateStatement $tree)
+    {
+        if (($tree->fields === null) || (!$tree->options->has('TABLE'))) {
+            return array();
+        }
+
+        $ret = array();
+
+        foreach ($tree->fields as $field) {
+
+            // Skipping keys.
+            if (is_string($field->type)) {
+                continue;
+            }
+
+            $ret[$field->name] = array(
+                'type' => $field->type->name,
+                'timestamp_not_null' => false,
+            );
+
+            if ($field->options) {
+
+                if ($field->type->name === 'TIMESTAMP') {
+                    if ($field->options->has('NOT NULL')) {
+                        $ret[$field->name]['timestamp_not_null'] = true;
+                    }
+                }
+
+                if (($option = $field->options->has('DEFAULT'))) {
+                    $ret[$field->name]['default_value'] = $option;
+                    if ($option === 'CURRENT_TIMESTAMP') {
+                        $ret[$field->name]['default_current_timestamp'] = true;
+                    }
+                }
+
+                if (($option = $field->options->has('ON UPDATE'))) {
+                    if ($option === 'CURRENT_TIMESTAMP') {
+                        $ret[$field->name]['on_update_current_timestamp'] = true;
+                    }
+                }
+            }
+
+        }
+
+        return $ret;
+    }
+}