@dbml/core 3.5.1 → 3.5.2-alpha.0

package/lib/parse/ANTLR/ASTGeneration/snowflake/SnowflakeASTGen.js

@@ -0,0 +1,750 @@
+"use strict";
+
+function _typeof(obj) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (obj) { return typeof obj; } : function (obj) { return obj && "function" == typeof Symbol && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }, _typeof(obj); }
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports["default"] = void 0;
+var _lodash = require("lodash");
+var _SnowflakeParserVisitor = _interopRequireDefault(require("../../parsers/snowflake/SnowflakeParserVisitor"));
+var _AST = require("../AST");
+var _constants = require("../constants");
+var _helpers = require("../helpers");
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { "default": obj }; }
+function _toConsumableArray(arr) { return _arrayWithoutHoles(arr) || _iterableToArray(arr) || _unsupportedIterableToArray(arr) || _nonIterableSpread(); }
+function _nonIterableSpread() { throw new TypeError("Invalid attempt to spread non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
+function _iterableToArray(iter) { if (typeof Symbol !== "undefined" && iter[Symbol.iterator] != null || iter["@@iterator"] != null) return Array.from(iter); }
+function _arrayWithoutHoles(arr) { if (Array.isArray(arr)) return _arrayLikeToArray(arr); }
+function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
+function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
+function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
+function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }
+function _iterableToArrayLimit(arr, i) { var _i = null == arr ? null : "undefined" != typeof Symbol && arr[Symbol.iterator] || arr["@@iterator"]; if (null != _i) { var _s, _e, _x, _r, _arr = [], _n = !0, _d = !1; try { if (_x = (_i = _i.call(arr)).next, 0 === i) { if (Object(_i) !== _i) return; _n = !1; } else for (; !(_n = (_s = _x.call(_i)).done) && (_arr.push(_s.value), _arr.length !== i); _n = !0); } catch (err) { _d = !0, _e = err; } finally { try { if (!_n && null != _i["return"] && (_r = _i["return"](), Object(_r) !== _r)) return; } finally { if (_d) throw _e; } } return _arr; } }
+function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
+function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
+function _defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, _toPropertyKey(descriptor.key), descriptor); } }
+function _createClass(Constructor, protoProps, staticProps) { if (protoProps) _defineProperties(Constructor.prototype, protoProps); if (staticProps) _defineProperties(Constructor, staticProps); Object.defineProperty(Constructor, "prototype", { writable: false }); return Constructor; }
+function _toPropertyKey(arg) { var key = _toPrimitive(arg, "string"); return _typeof(key) === "symbol" ? key : String(key); }
+function _toPrimitive(input, hint) { if (_typeof(input) !== "object" || input === null) return input; var prim = input[Symbol.toPrimitive]; if (prim !== undefined) { var res = prim.call(input, hint || "default"); if (_typeof(res) !== "object") return res; throw new TypeError("@@toPrimitive must return a primitive value."); } return (hint === "string" ? String : Number)(input); }
+function _inherits(subClass, superClass) { if (typeof superClass !== "function" && superClass !== null) { throw new TypeError("Super expression must either be null or a function"); } subClass.prototype = Object.create(superClass && superClass.prototype, { constructor: { value: subClass, writable: true, configurable: true } }); Object.defineProperty(subClass, "prototype", { writable: false }); if (superClass) _setPrototypeOf(subClass, superClass); }
+function _setPrototypeOf(o, p) { _setPrototypeOf = Object.setPrototypeOf ? Object.setPrototypeOf.bind() : function _setPrototypeOf(o, p) { o.__proto__ = p; return o; }; return _setPrototypeOf(o, p); }
+function _createSuper(Derived) { var hasNativeReflectConstruct = _isNativeReflectConstruct(); return function _createSuperInternal() { var Super = _getPrototypeOf(Derived), result; if (hasNativeReflectConstruct) { var NewTarget = _getPrototypeOf(this).constructor; result = Reflect.construct(Super, arguments, NewTarget); } else { result = Super.apply(this, arguments); } return _possibleConstructorReturn(this, result); }; }
+function _possibleConstructorReturn(self, call) { if (call && (_typeof(call) === "object" || typeof call === "function")) { return call; } else if (call !== void 0) { throw new TypeError("Derived constructors may only return object or undefined"); } return _assertThisInitialized(self); }
+function _assertThisInitialized(self) { if (self === void 0) { throw new ReferenceError("this hasn't been initialised - super() hasn't been called"); } return self; }
+function _isNativeReflectConstruct() { if (typeof Reflect === "undefined" || !Reflect.construct) return false; if (Reflect.construct.sham) return false; if (typeof Proxy === "function") return true; try { Boolean.prototype.valueOf.call(Reflect.construct(Boolean, [], function () {})); return true; } catch (e) { return false; } }
+function _getPrototypeOf(o) { _getPrototypeOf = Object.setPrototypeOf ? Object.getPrototypeOf.bind() : function _getPrototypeOf(o) { return o.__proto__ || Object.getPrototypeOf(o); }; return _getPrototypeOf(o); } /* eslint-disable class-methods-use-this */
+var sanitizeComment = function sanitizeComment(stringContext) {
+  return (0, _helpers.getOriginalText)(stringContext).replace(/''/g, "'").slice(1, -1);
+};
+var SnowflakeASTGen = /*#__PURE__*/function (_SnowflakeParserVisit) {
+  _inherits(SnowflakeASTGen, _SnowflakeParserVisit);
+  var _super = _createSuper(SnowflakeASTGen);
+  function SnowflakeASTGen() {
+    var _this;
+    _classCallCheck(this, SnowflakeASTGen);
+    _this = _super.call(this);
+    _this.data = {
+      schemas: [],
+      tables: [],
+      refs: [],
+      enums: [],
+      tableGroups: [],
+      aliases: [],
+      project: {}
+    };
+    return _this;
+  }
+
+  // batch? EOF
+  _createClass(SnowflakeASTGen, [{
+    key: "visitSnowflake_file",
+    value: function visitSnowflake_file(ctx) {
+      var _ctx$batch;
+      (_ctx$batch = ctx.batch()) === null || _ctx$batch === void 0 ? void 0 : _ctx$batch.accept(this);
+      return this.data;
+    }
+
+    // sql_command (SEMI sql_command)* SEMI?
+  }, {
+    key: "visitBatch",
+    value: function visitBatch(ctx) {
+      var _this2 = this;
+      ctx.sql_command().forEach(function (c) {
+        c.accept(_this2);
+      });
+    }
+
+    // ddl_command | dml_command | show_command | use_command | describe_command | other_command
+  }, {
+    key: "visitSql_command",
+    value: function visitSql_command(ctx) {
+      if (ctx.ddl_command()) {
+        ctx.ddl_command().accept(this);
+      }
+    }
+
+    // alter_command | create_command | drop_command | undrop_command
+  }, {
+    key: "visitDdl_command",
+    value: function visitDdl_command(ctx) {
+      if (ctx.alter_command()) {
+        ctx.alter_command().accept(this);
+      } else if (ctx.create_command()) {
+        ctx.create_command().accept(this);
+      }
+    }
+
+    // check SnowflakeParser.g4 line 1442
+  }, {
+    key: "visitCreate_command",
+    value: function visitCreate_command(ctx) {
+      if (ctx.create_table()) {
+        var table = ctx.create_table().accept(this);
+        this.data.tables.push(table);
+      } else if (ctx.create_table_like()) {
+        var _ctx$create_table_lik = ctx.create_table_like().accept(this),
+          _ctx$create_table_lik2 = _slicedToArray(_ctx$create_table_lik, 4),
+          schemaNameLike = _ctx$create_table_lik2[0],
+          nameLike = _ctx$create_table_lik2[1],
+          schemaNameOrigin = _ctx$create_table_lik2[2],
+          nameOrigin = _ctx$create_table_lik2[3];
+        var originTable = this.data.tables.reduce(function (acc, ele) {
+          if (ele.name === nameOrigin && ele.schemaName === schemaNameOrigin) return ele;
+          return acc;
+        }, null);
+        if (originTable) {
+          var likeTable = new _AST.Table({
+            name: nameLike,
+            schemaName: schemaNameLike,
+            fields: originTable.fields
+          });
+          this.data.tables.push(likeTable);
+        }
+      }
+    }
+
+    // : CREATE or_replace? TRANSIENT? DATABASE if_not_exists? id_ clone_at_before? (
+    //     DATA_RETENTION_TIME_IN_DAYS EQ num
+    // )? (MAX_DATA_EXTENSION_TIME_IN_DAYS EQ num)? default_ddl_collation? with_tags? comment_clause?
+    // ;
+  }, {
+    key: "visitCreate_database",
+    value: function visitCreate_database(ctx) {
+      var schema = (0, _helpers.getOriginalText)(ctx.object_name());
+      this.data.schemas.push(schema);
+    }
+
+    // CREATE or_replace? table_type? TABLE (
+    //   if_not_exists? object_name
+    //   | object_name if_not_exists?
+    // ) ((comment_clause? create_table_clause) | (create_table_clause comment_clause?))
+  }, {
+    key: "visitCreate_table",
+    value: function visitCreate_table(ctx) {
+      var _ctx$comment_clause,
+        _this3 = this,
+        _this$data$refs,
+        _this$data$refs2;
+      var _ctx$object_name$acce = ctx.object_name().accept(this),
+        _ctx$object_name$acce2 = _slicedToArray(_ctx$object_name$acce, 3),
+        databaseName = _ctx$object_name$acce2[0],
+        schemaName = _ctx$object_name$acce2[1],
+        tableName = _ctx$object_name$acce2[2];
+      var _ctx$create_table_cla = ctx.create_table_clause().accept(this),
+        definitions = _ctx$create_table_cla.definitions,
+        tableNote = _ctx$create_table_cla.tableNote;
+      var note = ((_ctx$comment_clause = ctx.comment_clause()) === null || _ctx$comment_clause === void 0 ? void 0 : _ctx$comment_clause.accept(this)) || tableNote;
+      var _definitions$reduce = definitions.reduce(function (acc, ele) {
+          if (ele.kind === _constants.TABLE_CONSTRAINT_KIND.FIELD) acc[0].push(ele.value);else if (ele.kind === _constants.TABLE_CONSTRAINT_KIND.UNIQUE) acc[1].push(ele.value);else if (ele.kind === _constants.TABLE_CONSTRAINT_KIND.FK) acc[2].push(ele.value);else if (ele.kind === _constants.TABLE_CONSTRAINT_KIND.PK) {
+            /** @type {Index} */
+            var index = ele.value;
+            if (index.columns.length > 1) acc[1].push(ele.value);else acc[3] = index;
+          }
+          return acc;
+        }, [[], [], [], null]),
+        _definitions$reduce2 = _slicedToArray(_definitions$reduce, 4),
+        fieldsData = _definitions$reduce2[0],
+        indexes = _definitions$reduce2[1],
+        tableRefs = _definitions$reduce2[2],
+        singlePkIndex = _definitions$reduce2[3];
+      var inlineRefsOfFields = fieldsData.map(function (fieldData) {
+        var _field$type$type_name;
+        var field = fieldData.field,
+          inlineRefs = fieldData.inlineRefs;
+        if (((_field$type$type_name = field.type.type_name) === null || _field$type$type_name === void 0 ? void 0 : _field$type$type_name.toLowerCase()) === 'enum') {
+          var _values = field.type.args.map(function (arg) {
+            var newValue = arg.replace(/'|"|`/g, '').trim();
+            return {
+              name: newValue
+            };
+          });
+          var _enum = new _AST.Enum({
+            name: "".concat(tableName, "_").concat(field.name, "_enum"),
+            schemaName: schemaName,
+            values: _values
+          });
+          field.type.type_name = _enum.name;
+          field.type.schemaName = _enum.schemaName;
+          _this3.data.enums.push(_enum);
+        }
+        inlineRefs.forEach(function (inlineRef) {
+          inlineRef.endpoints[0].tableName = tableName;
+          inlineRef.endpoints[0].schemaName = schemaName;
+          inlineRef.endpoints[0].fieldNames = [field.name];
+        });
+        return inlineRefs;
+      });
+      (_this$data$refs = this.data.refs).push.apply(_this$data$refs, _toConsumableArray((0, _lodash.flatten)(inlineRefsOfFields)));
+      (_this$data$refs2 = this.data.refs).push.apply(_this$data$refs2, _toConsumableArray(tableRefs.map(function (tableRef) {
+        tableRef.endpoints[0].tableName = tableName;
+        tableRef.endpoints[0].schemaName = schemaName;
+        return tableRef;
+      })));
+      var table = new _AST.Table({
+        name: tableName,
+        schemaName: schemaName,
+        fields: fieldsData.map(function (fd) {
+          return fd.field;
+        }),
+        indexes: indexes,
+        note: note
+      });
+      if (singlePkIndex) {
+        var field = table.fields.find(function (f) {
+          return f.name === singlePkIndex.columns[0].value;
+        });
+        if (field) field.pk = true;
+      }
+      return table;
+    }
+
+    // : ALTER TABLE if_exists? object_name RENAME TO object_name
+    // | ALTER TABLE if_exists? object_name SWAP WITH object_name
+    // | ALTER TABLE if_exists? object_name (
+    //     clustering_action
+    //     | table_column_action
+    //     | constraint_action
+    // )
+    // | ALTER TABLE if_exists? object_name ext_table_column_action
+    // | ALTER TABLE if_exists? object_name search_optimization_action
+    // | ALTER TABLE if_exists? object_name SET stage_file_format? (
+    //     STAGE_COPY_OPTIONS EQ '(' copy_options ')'
+    // )? (DATA_RETENTION_TIME_IN_DAYS EQ num)? (MAX_DATA_EXTENSION_TIME_IN_DAYS EQ num)? (
+    //     CHANGE_TRACKING EQ true_false
+    // )? default_ddl_collation? comment_clause?
+    // | ALTER TABLE if_exists? object_name set_tags
+    // | ALTER TABLE if_exists? object_name unset_tags
+    // | ALTER TABLE if_exists? object_name UNSET (
+    //     DATA_RETENTION_TIME_IN_DAYS
+    //     | MAX_DATA_EXTENSION_TIME_IN_DAYS
+    //     | CHANGE_TRACKING
+    //     | DEFAULT_DDL_COLLATION_
+    //     | COMMENT
+    //     |
+    // )
+    // //[ , ... ]
+    // | ALTER TABLE if_exists? object_name ADD ROW ACCESS POLICY id_ ON column_list_in_parentheses
+    // | ALTER TABLE if_exists? object_name DROP ROW ACCESS POLICY id_
+    // | ALTER TABLE if_exists? object_name DROP ROW ACCESS POLICY id_ COMMA ADD ROW ACCESS POLICY id_ ON column_list_in_parentheses
+    // | ALTER TABLE if_exists? object_name DROP ALL ROW ACCESS POLICIES
+  }, {
+    key: "visitAlter_table",
+    value: function visitAlter_table(ctx) {
+      if (ctx.constraint_action()) {
+        var _ctx$object_name$0$ac = ctx.object_name()[0].accept(this),
+          _ctx$object_name$0$ac2 = _slicedToArray(_ctx$object_name$0$ac, 3),
+          databaseName = _ctx$object_name$0$ac2[0],
+          schemaName = _ctx$object_name$0$ac2[1],
+          tableName = _ctx$object_name$0$ac2[2];
+        var definition = ctx.constraint_action().accept(this);
+        if (definition) {
+          var _this$data$refs3;
+          var fieldsData = [];
+          var indexes = [];
+          var tableRefs = [];
+          if (definition.kind === _constants.TABLE_CONSTRAINT_KIND.FIELD) fieldsData.push(definition.value);else if (definition.kind === _constants.TABLE_CONSTRAINT_KIND.UNIQUE) indexes.push(definition.value);else if (definition.kind === _constants.TABLE_CONSTRAINT_KIND.FK) tableRefs.push(definition.value);
+          (_this$data$refs3 = this.data.refs).push.apply(_this$data$refs3, _toConsumableArray(tableRefs.map(function (tableRef) {
+            tableRef.endpoints[0].tableName = tableName;
+            tableRef.endpoints[0].schemaName = schemaName;
+            return tableRef;
+          })));
+          var table = this.data.tables.reduce(function (acc, ele) {
+            if (ele.name === tableName && ele.schemaName === schemaName) return ele;
+            return acc;
+          }, null);
+          if (table) {
+            var _table$fields, _table$indexes;
+            (_table$fields = table.fields).push.apply(_table$fields, fieldsData);
+            (_table$indexes = table.indexes).push.apply(_table$indexes, indexes);
+          }
+          return table;
+        }
+      }
+      return null;
+    }
+
+    // CREATE or_replace? TRANSIENT? TABLE if_not_exists? object_name LIKE object_name cluster_by? copy_grants?
+  }, {
+    key: "visitCreate_table_like",
+    value: function visitCreate_table_like(ctx) {
+      var _ctx$object_name$0$ac3 = ctx.object_name()[0].accept(this),
+        _ctx$object_name$0$ac4 = _slicedToArray(_ctx$object_name$0$ac3, 3),
+        databaseNameLike = _ctx$object_name$0$ac4[0],
+        schemaNameLike = _ctx$object_name$0$ac4[1],
+        nameLike = _ctx$object_name$0$ac4[2];
+      var _ctx$object_name$1$ac = ctx.object_name()[1].accept(this),
+        _ctx$object_name$1$ac2 = _slicedToArray(_ctx$object_name$1$ac, 3),
+        databaseNameOrigin = _ctx$object_name$1$ac2[0],
+        schemaNameOrigin = _ctx$object_name$1$ac2[1],
+        nameOrigin = _ctx$object_name$1$ac2[2];
+      return [schemaNameLike, nameLike, schemaNameOrigin, nameOrigin];
+    }
+
+    // : d = id_ DOT s = id_ DOT o = id_
+    // | s = id_ DOT o = id_
+    // | o = id_
+  }, {
+    key: "visitObject_name",
+    value: function visitObject_name(ctx) {
+      var _ctx$d, _ctx$s;
+      return [(_ctx$d = ctx.d) === null || _ctx$d === void 0 ? void 0 : _ctx$d.accept(this), (_ctx$s = ctx.s) === null || _ctx$s === void 0 ? void 0 : _ctx$s.accept(this), ctx.o.accept(this)];
+    }
+
+    // : ID | ID2 | DOUBLE_QUOTE_ID | DOUBLE_QUOTE_BLANK
+    // | keyword | non_reserved_words | object_type_plural | data_type
+    // | builtin_function | unary_or_binary_builtin_function | binary_builtin_function
+    // | binary_or_ternary_builtin_function | ternary_builtin_function
+  }, {
+    key: "visitId_",
+    value: function visitId_(ctx) {
+      if (ctx.DOUBLE_QUOTE_ID()) return (0, _helpers.getOriginalText)(ctx).slice(1, -1);
+      return (0, _helpers.getOriginalText)(ctx);
+    }
+
+    // : (
+    //     column_decl_item_list_paren cluster_by?
+    //     | cluster_by? comment_clause? column_decl_item_list_paren
+    // ) stage_file_format? (STAGE_COPY_OPTIONS EQ LR_BRACKET copy_options RR_BRACKET)? (
+    //     DATA_RETENTION_TIME_IN_DAYS EQ num
+    // )? (MAX_DATA_EXTENSION_TIME_IN_DAYS EQ num)? change_tracking? default_ddl_collation? copy_grants? comment_clause? with_row_access_policy?
+    //     with_tags?
+  }, {
+    key: "visitCreate_table_clause",
+    value: function visitCreate_table_clause(ctx) {
+      var _ctx$comment_clause$m,
+        _this4 = this;
+      return {
+        definitions: ctx.column_decl_item_list_paren().accept(this),
+        tableNote: (_ctx$comment_clause$m = ctx.comment_clause().map(function (c) {
+          return c.accept(_this4);
+        })) === null || _ctx$comment_clause$m === void 0 ? void 0 : _ctx$comment_clause$m[-1]
+      };
+    }
+
+    // : COMMENT EQ string
+  }, {
+    key: "visitComment_clause",
+    value: function visitComment_clause(ctx) {
+      return sanitizeComment(ctx.string());
+    }
+
+    // '(' column_decl_item_list ')'
+  }, {
+    key: "visitColumn_decl_item_list_paren",
+    value: function visitColumn_decl_item_list_paren(ctx) {
+      return ctx.column_decl_item_list().accept(this);
+    }
+
+    // column_decl_item (COMMA column_decl_item)*
+  }, {
+    key: "visitColumn_decl_item_list",
+    value: function visitColumn_decl_item_list(ctx) {
+      var _this5 = this;
+      return ctx.column_decl_item().map(function (c) {
+        return c.accept(_this5);
+      }).filter(function (r) {
+        return r;
+      });
+    }
+
+    // full_col_decl | out_of_line_constraint
+  }, {
+    key: "visitColumn_decl_item",
+    value: function visitColumn_decl_item(ctx) {
+      if (ctx.full_col_decl()) {
+        return {
+          kind: _constants.TABLE_CONSTRAINT_KIND.FIELD,
+          value: ctx.full_col_decl().accept(this)
+        };
+      }
+      if (ctx.out_of_line_constraint()) {
+        var test = ctx.out_of_line_constraint().accept(this);
+        return test;
+      }
+      return null;
+    }
+
+    // col_decl (collate | inline_constraint | null_not_null | (default_value | NULL_))* with_masking_policy? with_tags? ( COMMENT string )?
+  }, {
+    key: "visitFull_col_decl",
+    value: function visitFull_col_decl(ctx) {
+      var _this6 = this;
+      var field = ctx.col_decl().accept(this);
+      var inlineRefs = [];
+      if (ctx.inline_constraint()) {
+        var inlineConstraints = ctx.inline_constraint().map(function (c) {
+          return c.accept(_this6);
+        });
+        if (!(0, _lodash.isEmpty)(inlineConstraints)) {
+          inlineConstraints.forEach(function (inlineConstraint) {
+            if (inlineConstraint.kind === _constants.COLUMN_CONSTRAINT_KIND.UNIQUE) field.unique = true;else if (inlineConstraint.kind === _constants.COLUMN_CONSTRAINT_KIND.PK) field.pk = true;else if (inlineConstraint.kind === _constants.COLUMN_CONSTRAINT_KIND.FK) inlineRefs.push(inlineConstraint.value);
+          });
+        }
+      }
+      if (ctx.null_not_null()) {
+        var notNulls = ctx.null_not_null().map(function (c) {
+          return c.accept(_this6);
+        });
+        if (!(0, _lodash.isEmpty)(notNulls)) {
+          notNulls.forEach(function (notNull) {
+            if (notNull.kind === _constants.COLUMN_CONSTRAINT_KIND.NOT_NULL) field.not_null = notNull.value;
+          });
+        }
+      }
+      if (ctx.default_value()) {
+        var defaultOrIncrements = ctx.default_value().map(function (c) {
+          return c.accept(_this6);
+        });
+        defaultOrIncrements.forEach(function (defaultOrIncrement) {
+          if (defaultOrIncrement.kind === _constants.COLUMN_CONSTRAINT_KIND.DEFAULT) {
+            field.dbdefault = defaultOrIncrement.value;
+          } else if (defaultOrIncrement.kind === _constants.COLUMN_CONSTRAINT_KIND.INCREMENT) {
+            field.increment = defaultOrIncrement.value;
+          }
+        });
+      }
+      if (ctx.COMMENT()) {
+        field.note = sanitizeComment(ctx.string());
+      }
+      return {
+        field: field,
+        inlineRefs: inlineRefs
+      };
+    }
+
+    // column_name data_type virtual_column_decl?
+  }, {
+    key: "visitCol_decl",
+    value: function visitCol_decl(ctx) {
+      return new _AST.Field({
+        name: ctx.column_name().accept(this),
+        type: {
+          type_name: (0, _helpers.getOriginalText)(ctx.data_type()),
+          schemaName: null
+        }
+      });
+    }
+
+    // : (id_ '.')? id_
+  }, {
+    key: "visitColumn_name",
+    value: function visitColumn_name(ctx) {
+      var _this7 = this;
+      return ctx.id_().map(function (c) {
+        return c.accept(_this7);
+      }).join('.');
+    }
+
+    // : (CONSTRAINT id_)? (
+    //     (UNIQUE | primary_key) common_constraint_properties*
+    //     | foreign_key REFERENCES object_name (LR_BRACKET column_name RR_BRACKET)? constraint_properties
+    // )
+  }, {
+    key: "visitInline_constraint",
+    value: function visitInline_constraint(ctx) {
+      if (ctx.UNIQUE()) {
+        return {
+          kind: _constants.COLUMN_CONSTRAINT_KIND.UNIQUE,
+          value: true
+        };
+      }
+      if (ctx.primary_key()) {
+        return {
+          kind: _constants.COLUMN_CONSTRAINT_KIND.PK,
+          value: true
+        };
+      }
+      if (ctx.foreign_key() && ctx.column_name()) {
+        var _ctx$object_name$acce3 = ctx.object_name().accept(this),
+          _ctx$object_name$acce4 = _slicedToArray(_ctx$object_name$acce3, 3),
+          databaseName = _ctx$object_name$acce4[0],
+          schemaName = _ctx$object_name$acce4[1],
+          tableName = _ctx$object_name$acce4[2];
+        var destColumns = [ctx.column_name().accept(this)];
+        return {
+          kind: _constants.COLUMN_CONSTRAINT_KIND.FK,
+          value: {
+            endpoints: [{
+              tableName: null,
+              schemaName: null,
+              fieldNames: null,
+              relation: '*'
+            }, {
+              tableName: tableName,
+              schemaName: schemaName,
+              fieldNames: destColumns,
+              relation: '1'
+            }]
+          }
+        };
+      }
+      return {
+        kind: null,
+        value: null
+      };
+    }
+
+    // : (CONSTRAINT id_)? (
+    //     (UNIQUE | primary_key) column_list_in_parentheses common_constraint_properties*
+    //     | foreign_key column_list_in_parentheses REFERENCES object_name column_list_in_parentheses constraint_properties
+    // )
+  }, {
+    key: "visitOut_of_line_constraint",
+    value: function visitOut_of_line_constraint(ctx) {
+      var _this8 = this;
+      if (ctx.UNIQUE()) {
+        var _ctx$id_;
+        var name = (_ctx$id_ = ctx.id_()) === null || _ctx$id_ === void 0 ? void 0 : _ctx$id_.accept(this);
+        var colNames = (0, _lodash.flatten)(ctx.column_list_in_parentheses().map(function (c) {
+          return c.accept(_this8);
+        }));
+        var value = new _AST.Index({
+          unique: true,
+          columns: colNames.map(function (colName) {
+            return {
+              value: colName,
+              type: _constants.CONSTRAINT_TYPE.COLUMN
+            };
+          })
+        });
+        return {
+          name: name,
+          kind: _constants.TABLE_CONSTRAINT_KIND.UNIQUE,
+          value: value
+        };
+      }
+      if (ctx.primary_key()) {
+        var _ctx$id_2;
+        var _name = (_ctx$id_2 = ctx.id_()) === null || _ctx$id_2 === void 0 ? void 0 : _ctx$id_2.accept(this);
+        var _colNames = (0, _lodash.flatten)(ctx.column_list_in_parentheses().map(function (c) {
+          return c.accept(_this8);
+        }));
+        var _value = new _AST.Index({
+          name: _name,
+          pk: true,
+          columns: _colNames.map(function (colName) {
+            return {
+              value: colName,
+              type: _constants.CONSTRAINT_TYPE.COLUMN
+            };
+          })
+        });
+        return {
+          kind: _constants.TABLE_CONSTRAINT_KIND.PK,
+          value: _value
+        };
+      }
+      if (ctx.foreign_key() && ctx.column_list_in_parentheses().length === 2) {
+        var _ctx$object_name$acce5 = ctx.object_name().accept(this),
+          _ctx$object_name$acce6 = _slicedToArray(_ctx$object_name$acce5, 3),
+          databaseName = _ctx$object_name$acce6[0],
+          schemaName = _ctx$object_name$acce6[1],
+          tableName = _ctx$object_name$acce6[2];
+        var sourceColumns = ctx.column_list_in_parentheses()[0].accept(this);
+        var destColumns = ctx.column_list_in_parentheses()[1].accept(this);
+        return {
+          kind: _constants.TABLE_CONSTRAINT_KIND.FK,
+          value: {
+            endpoints: [{
+              tableName: null,
+              schemaName: null,
+              fieldNames: sourceColumns,
+              relation: '*'
+            }, {
+              tableName: tableName,
+              schemaName: schemaName,
+              fieldNames: destColumns,
+              relation: '1'
+            }]
+          }
+        };
+      }
+      return null;
+    }
+
+    // LR_BRACKET column_list RR_BRACKET
+  }, {
+    key: "visitColumn_list_in_parentheses",
+    value: function visitColumn_list_in_parentheses(ctx) {
+      return ctx.column_list().accept(this);
+    }
+
+    // column_name (COMMA column_name)*
+  }, {
+    key: "visitColumn_list",
+    value: function visitColumn_list(ctx) {
+      var _this9 = this;
+      return ctx.column_name().map(function (c) {
+        return c.accept(_this9);
+      });
+    }
+
+    // NOT? NULL_
+  }, {
+    key: "visitNull_not_null",
+    value: function visitNull_not_null(ctx) {
+      return {
+        kind: _constants.COLUMN_CONSTRAINT_KIND.NOT_NULL,
+        value: !!ctx.NOT()
+      };
+    }
+
+    // : DEFAULT expr
+    // | (AUTOINCREMENT | IDENTITY) (
+    //     LR_BRACKET num COMMA num RR_BRACKET
+    //     | start_with
+    //     | increment_by
+    //     | start_with increment_by
+    // )? order_noorder?
+  }, {
+    key: "visitDefault_value",
+    value: function visitDefault_value(ctx) {
+      // dbdefault: {value: string, type: 'string' | 'number' | 'boolean' | 'expression'},
+      if (ctx.DEFAULT()) {
+        return {
+          kind: _constants.COLUMN_CONSTRAINT_KIND.DEFAULT,
+          value: ctx.expr().accept(this)
+        };
+      }
+      if (ctx.AUTOINCREMENT() || ctx.IDENTITY()) {
+        return {
+          kind: _constants.COLUMN_CONSTRAINT_KIND.INCREMENT,
+          value: true
+        };
+      }
+      return null;
+    }
+
+    //  : object_name DOT NEXTVAL
+    //  | expr LSB expr RSB //array access
+    //  | expr COLON expr   //json access
+    //  | expr DOT (VALUE | expr)
+    //  | expr COLLATE string
+    //  | case_expression
+    //  | iff_expr
+    //  | bracket_expression
+    //  | op = ( PLUS | MINUS) expr
+    //  | expr op = (STAR | DIVIDE | MODULE) expr
+    //  | expr op = (PLUS | MINUS | PIPE_PIPE) expr
+    //  | expr comparison_operator expr
+    //  | op = NOT+ expr
+    //  | expr AND expr //bool operation
+    //  | expr OR expr  //bool operation
+    //  | arr_literal
+    //  //    | expr time_zone
+    //  | expr over_clause
+    //  | cast_expr
+    //  | expr COLON_COLON data_type // Cast also
+    //  | try_cast_expr
+    //  | json_literal
+    //  | trim_expression
+    //  | function_call
+    //  | subquery
+    //  | expr IS null_not_null
+    //  | expr NOT? IN LR_BRACKET (subquery | expr_list) RR_BRACKET
+    //  | expr NOT? ( LIKE | ILIKE) expr (ESCAPE expr)?
+    //  | expr NOT? RLIKE expr
+    //  | expr NOT? (LIKE | ILIKE) ANY LR_BRACKET expr (COMMA expr)* RR_BRACKET (ESCAPE expr)?
+    //  | primitive_expression //Should be latest rule as it's nearly a catch all
+  }, {
+    key: "visitExpr",
+    value: function visitExpr(ctx) {
+      if (ctx.primitive_expression()) return ctx.primitive_expression().accept(this);
+      return {
+        value: (0, _helpers.getOriginalText)(ctx),
+        type: 'expression'
+      };
+    }
+
+    // : DEFAULT //?
+    // | NULL_
+    // | id_ ('.' id_)* // json field access
+    // | full_column_name
+    // | literal
+    // | BOTH_Q
+    // | ARRAY_Q
+    // | OBJECT_Q
+    // //| json_literal
+    // //| arr_literal
+  }, {
+    key: "visitPrimitive_expression",
+    value: function visitPrimitive_expression(ctx) {
+      if (ctx.NULL_()) return {
+        value: null,
+        type: 'boolean'
+      };
+      if (ctx.literal()) return ctx.literal().accept(this);
+      return {
+        value: (0, _helpers.getOriginalText)(ctx),
+        type: 'expression'
+      };
+    }
+
+    //  : STRING // string, date, time, timestamp
+    //  | sign? DECIMAL
+    //  | sign? (REAL | FLOAT)
+    //  | true_false
+    //  | NULL_
+    //  | AT_Q
+    //  ;
+  }, {
+    key: "visitLiteral",
+    value: function visitLiteral(ctx) {
+      if (ctx.STRING()) return {
+        value: (0, _helpers.getOriginalText)(ctx),
+        type: 'string'
+      };
+      if (ctx.DECIMAL() || ctx.REAL || ctx.FLOAT) return {
+        value: (0, _helpers.getOriginalText)(ctx),
+        type: 'number'
+      };
+      if (ctx.true_false() || ctx.NULL_()) return {
+        value: (0, _helpers.getOriginalText)(ctx),
+        type: 'boolean'
+      };
+      return {
+        value: (0, _helpers.getOriginalText)(ctx),
+        type: 'expression'
+      };
+    }
+
+    // : ADD out_of_line_constraint
+    // | RENAME CONSTRAINT id_ TO id_
+    // | alter_modify (CONSTRAINT id_ | primary_key | UNIQUE | foreign_key) column_list_in_parentheses enforced_not_enforced? (
+    //     VALIDATE
+    //     | NOVALIDATE
+    // ) (RELY | NORELY)
+    // | DROP (CONSTRAINT id_ | primary_key | UNIQUE | foreign_key) column_list_in_parentheses? cascade_restrict?
+    // | DROP PRIMARY KEY
+  }, {
+    key: "visitConstraint_action",
+    value: function visitConstraint_action(ctx) {
+      if (ctx.ADD()) {
+        return ctx.out_of_line_constraint().accept(this);
+      }
+      return null;
+    }
+  }]);
+  return SnowflakeASTGen;
+}(_SnowflakeParserVisitor["default"]);
+exports["default"] = SnowflakeASTGen;