@dbml/core 3.5.1 → 3.6.0

package/lib/parse/ANTLR/ASTGeneration/helpers.js

@@ -0,0 +1,14 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.getOriginalText = getOriginalText;
+var _lodash = require("lodash");
+function getOriginalText(ctx) {
+  var a = ctx.start.start;
+  var b = ctx.stop.stop;
+  if ((0, _lodash.isNil)(a) || (0, _lodash.isNil)(b)) return undefined;
+  var input = ctx.start.getInputStream();
+  return input.getText(a, b);
+}

package/lib/parse/ANTLR/ASTGeneration/index.js

@@ -11,6 +11,9 @@ var _PostgresASTGen = _interopRequireDefault(require("./postgres/PostgresASTGen"
 var _MySqlLexer = _interopRequireDefault(require("../parsers/mysql/MySqlLexer"));
 var _MySqlParser = _interopRequireDefault(require("../parsers/mysql/MySqlParser"));
 var _MySQLASTGen = _interopRequireDefault(require("./mysql/MySQLASTGen"));
+var _SnowflakeLexer = _interopRequireDefault(require("../parsers/snowflake/SnowflakeLexer"));
+var _SnowflakeParser = _interopRequireDefault(require("../parsers/snowflake/SnowflakeParser"));
+var _SnowflakeASTGen = _interopRequireDefault(require("./snowflake/SnowflakeASTGen"));
 var _ParserErrorListener = _interopRequireDefault(require("./ParserErrorListener"));
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { "default": obj }; }
 /* eslint-disable import/no-named-as-default-member */
@@ -47,6 +50,19 @@ function parse(input, format) {
         if (errorListener.errors.length) throw errorListener.errors;
         break;
       }
+    case 'snowflake':
+      {
+        var _lexer2 = new _SnowflakeLexer["default"](chars);
+        var _tokens2 = new _antlr["default"].CommonTokenStream(_lexer2);
+        var _parser2 = new _SnowflakeParser["default"](_tokens2);
+        _parser2.buildParseTrees = true;
+        _parser2.removeErrorListeners();
+        _parser2.addErrorListener(errorListener);
+        var _parseTree2 = _parser2.snowflake_file();
+        database = _parseTree2.accept(new _SnowflakeASTGen["default"]());
+        if (errorListener.errors.length) throw errorListener.errors;
+        break;
+      }
     default:
       throw new Error("Format not supported: ".concat(format));
   }