prettier-plugin-bq 1.0.0-beta.2 → 1.0.0-beta.4

package/dist/printer.js

@@ -1,7 +1,10 @@
-import { reservedKeywords, globalFunctions, keysFunctions, aeadFunctions, hllCountFunctions, kllQuantilesFunctions, netFunctions, objFunctions, mlFunctions, aiFunctions, vectorIndexFunctions, } from "./keywords.js";
-import { doc } from "prettier";
-const { builders: { breakParent, group, hardline, ifBreak, indent, join, line, lineSuffix, softline, }, } = doc;
-export const isNodeChild = (child) => {
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.printSQL = exports.isNodeVecChild = exports.isNodeChild = void 0;
+const keywords_1 = require("./keywords");
+const prettier_1 = require("prettier");
+const { builders: { breakParent, group, hardline, ifBreak, indent, join, line, lineSuffix, softline, }, } = prettier_1.doc;
+const isNodeChild = (child) => {
     if (child &&
         typeof child === "object" &&
         Object.keys(child).length === 1 &&
@@ -10,7 +13,8 @@ export const isNodeChild = (child) => {
     }
     return false;
 };
-export const isNodeVecChild = (child) => {
+exports.isNodeChild = isNodeChild;
+const isNodeVecChild = (child) => {
     if (child &&
         typeof child === "object" &&
         Object.keys(child).length === 1 &&
@@ -19,6 +23,7 @@ export const isNodeVecChild = (child) => {
     }
     return false;
 };
+exports.isNodeVecChild = isNodeVecChild;
 class Printer {
     constructor(path, options, print, node) {
         this.path = path;
@@ -30,7 +35,7 @@ class Printer {
     child(key, transform, consumeLeadingComments, sep) {
         const child = this.children[key];
         let f = (x) => x;
-        if (isNodeChild(child)) {
+        if ((0, exports.isNodeChild)(child)) {
             if (typeof transform === "function") {
                 f = transform;
             }
@@ -43,7 +48,7 @@ class Printer {
                 this.path.call((p) => p.call((p) => f(p.call(this.print, "Node")), key), "children"),
             ];
         }
-        else if (isNodeVecChild(child)) {
+        else if ((0, exports.isNodeVecChild)(child)) {
             if (typeof transform === "function") {
                 f = transform;
             }
@@ -62,7 +67,7 @@ class Printer {
     consumeAllCommentsOfX(key) {
         let res = "";
         const child = this.children[key];
-        if (isNodeChild(child)) {
+        if ((0, exports.isNodeChild)(child)) {
             const leading_comments = child.Node.children.leading_comments;
             if (leading_comments) {
                 res = leading_comments.NodeVec.map((x) => lineSuffix([" ", x.token.literal]));
@@ -107,10 +112,10 @@ class Printer {
         else if (target === "all") {
             const child = this.children[key];
             let firstNodes = [];
-            if (isNodeChild(child)) {
+            if ((0, exports.isNodeChild)(child)) {
                 firstNodes = [getFirstNode(child.Node)];
             }
-            else if (isNodeVecChild(child)) {
+            else if ((0, exports.isNodeVecChild)(child)) {
                 firstNodes = child.NodeVec.map((x) => getFirstNode(x));
             }
             const res = [];
@@ -137,10 +142,10 @@ class Printer {
     getFirstNode(key) {
         const child = this.children[key];
         let firstNode;
-        if (isNodeChild(child)) {
+        if ((0, exports.isNodeChild)(child)) {
             firstNode = getFirstNode(child.Node);
         }
-        else if (isNodeVecChild(child) && child.NodeVec.length > 0) {
+        else if ((0, exports.isNodeVecChild)(child) && child.NodeVec.length > 0) {
             firstNode = getFirstNode(child.NodeVec[0]);
         }
         else {
@@ -173,7 +178,7 @@ class Printer {
     }
     len(key) {
         const nodeVec = this.children[key];
-        if (isNodeVecChild(nodeVec)) {
+        if ((0, exports.isNodeVecChild)(nodeVec)) {
             return nodeVec.NodeVec.length;
         }
         return 0;
@@ -199,7 +204,7 @@ class Printer {
             if (upperOrLower === "upper") {
                 literal = literal.toUpperCase();
             }
-            else if (!this.node.notGlobal && this.includedIn(reservedKeywords)) {
+            else if (!this.node.notGlobal && this.includedIn(keywords_1.reservedKeywords)) {
                 literal = literal.toUpperCase();
             }
             else if (this.options.printKeywordsInUpperCase &&
@@ -226,7 +231,7 @@ class Printer {
     }
     setBreakRecommended(key) {
         const child = this.children[key];
-        if (isNodeChild(child)) {
+        if ((0, exports.isNodeChild)(child)) {
             child.Node.breakRecommended = true;
         }
     }
@@ -237,10 +242,10 @@ class Printer {
          * e.g. when grouping a part of `this.child(key)`.
          */
         const child = this.children[key];
-        if (isNodeChild(child)) {
+        if ((0, exports.isNodeChild)(child)) {
             child.Node.groupRecommended = true;
         }
-        else if (isNodeVecChild(child)) {
+        else if ((0, exports.isNodeVecChild)(child)) {
             child.NodeVec.forEach((x) => {
                 x.groupRecommended = true;
             });
@@ -248,7 +253,7 @@ class Printer {
     }
     setLiteral(key, literal) {
         const child = this.children[key];
-        if (isNodeChild(child)) {
+        if ((0, exports.isNodeChild)(child)) {
             const token = child.Node.token;
             if (token) {
                 token.literal = literal;
@@ -257,10 +262,10 @@ class Printer {
     }
     setNotRoot(key) {
         const child = this.children[key];
-        if (isNodeChild(child)) {
+        if ((0, exports.isNodeChild)(child)) {
             child.Node.notRoot = true;
         }
-        else if (isNodeVecChild(child)) {
+        else if ((0, exports.isNodeVecChild)(child)) {
             child.NodeVec.forEach((x) => {
                 x.notRoot = true;
             });
@@ -268,7 +273,7 @@ class Printer {
     }
     setNotGlobal(key) {
         const child = this.children[key];
-        if (isNodeChild(child)) {
+        if ((0, exports.isNodeChild)(child)) {
             child.Node.notGlobal = true;
         }
     }
@@ -276,7 +281,7 @@ class Printer {
         if (this.hasLeadingComments(key))
             return true;
         const child = this.children[key];
-        if (!isNodeChild(child))
+        if (!(0, exports.isNodeChild)(child))
             return false; // unnexpected pattern
         const token = child.Node.token;
         if (!token)
@@ -303,13 +308,13 @@ class Printer {
         if (!this.options.printKeywordsInUpperCase) {
             return;
         }
-        if (isNodeChild(child)) {
+        if ((0, exports.isNodeChild)(child)) {
             const token = child.Node.token;
             if (token) {
                 token.literal = token.literal.toUpperCase();
             }
         }
-        else if (isNodeVecChild(child)) {
+        else if ((0, exports.isNodeVecChild)(child)) {
             child.NodeVec.forEach((x) => {
                 const token = x.token;
                 if (token) {
@@ -326,10 +331,10 @@ const getFirstNode = (node, includeComment = false) => {
             !includeComment) {
             continue;
         }
-        if (isNodeChild(v)) {
+        if ((0, exports.isNodeChild)(v)) {
             candidates.push(getFirstNode(v.Node, includeComment));
         }
-        else if (isNodeVecChild(v)) {
+        else if ((0, exports.isNodeVecChild)(v)) {
             // NOTE maybe you don't have to check 2nd, 3rd, or latter node
             v.NodeVec.forEach((x) => candidates.push(getFirstNode(x, includeComment)));
         }
@@ -357,10 +362,10 @@ const getLastNode = (node, includeComment = false) => {
             !includeComment) {
             continue;
         }
-        if (isNodeChild(v)) {
+        if ((0, exports.isNodeChild)(v)) {
             candidates.push(getLastNode(v.Node, includeComment));
         }
-        else if (isNodeVecChild(v)) {
+        else if ((0, exports.isNodeVecChild)(v)) {
             // NOTE maybe you don't have to check 2nd, 3rd, or latter node
             v.NodeVec.forEach((x) => candidates.push(getLastNode(x, includeComment)));
         }
@@ -381,7 +386,7 @@ const getLastNode = (node, includeComment = false) => {
     }
     return res;
 };
-export const printSQL = (path, options, print) => {
+const printSQL = (path, options, print) => {
     const node = path.node;
     if (Array.isArray(node)) {
         for (let i = 0; i < node.length - 1; i++) {
@@ -718,6 +723,7 @@ export const printSQL = (path, options, print) => {
             throw new Error(`Not implemented node type: ${JSON.stringify(node)}`);
     }
 };
+exports.printSQL = printSQL;
 const printAccessOperator = (path, options, print, node) => {
     const p = new Printer(path, options, print, node);
     const docs = {
@@ -1536,6 +1542,9 @@ const printCallingFunctionGeneral = (path, options, print, node) => {
     const p = new Printer(path, options, print, node);
     p.setNotRoot("args");
     let func = node.children.func.Node;
+    if (func.node_type === "GroupedExpr" && node.isChained) {
+        func = func.children.expr.Node;
+    }
     let parent;
     let grandParent;
     if (node.isDatePart) {
@@ -1550,7 +1559,7 @@ const printCallingFunctionGeneral = (path, options, print, node) => {
     };
     if (func.node_type === "Identifier") {
         // SUBSTR("foo", 0, 2)
-        if (globalFunctions.includes(func.token.literal.toUpperCase())) {
+        if (keywords_1.globalFunctions.includes(func.token.literal.toUpperCase())) {
             func.isPreDefinedFunction = true;
         }
     }
@@ -1562,61 +1571,61 @@ const printCallingFunctionGeneral = (path, options, print, node) => {
             // KEYS.NEW_KEYSET('AEAD_AES_GCM_256')
             switch (parent.token.literal.toUpperCase()) {
                 case "SAFE":
-                    if (globalFunctions.includes(func.token.literal.toUpperCase())) {
+                    if (keywords_1.globalFunctions.includes(func.token.literal.toUpperCase())) {
                         func.isPreDefinedFunction = true;
                         toUpper(parent.token);
                     }
                     break;
                 case "KEYS":
-                    if (keysFunctions.includes(func.token.literal.toUpperCase())) {
+                    if (keywords_1.keysFunctions.includes(func.token.literal.toUpperCase())) {
                         func.isPreDefinedFunction = true;
                         toUpper(parent.token);
                     }
                     break;
                 case "AEAD":
-                    if (aeadFunctions.includes(func.token.literal.toUpperCase())) {
+                    if (keywords_1.aeadFunctions.includes(func.token.literal.toUpperCase())) {
                         func.isPreDefinedFunction = true;
                         toUpper(parent.token);
                     }
                     break;
                 case "NET":
-                    if (netFunctions.includes(func.token.literal.toUpperCase())) {
+                    if (keywords_1.netFunctions.includes(func.token.literal.toUpperCase())) {
                         func.isPreDefinedFunction = true;
                         toUpper(parent.token);
                     }
                     break;
                 case "HLL_COUNT":
-                    if (hllCountFunctions.includes(func.token.literal.toUpperCase())) {
+                    if (keywords_1.hllCountFunctions.includes(func.token.literal.toUpperCase())) {
                         func.isPreDefinedFunction = true;
                         toUpper(parent.token);
                     }
                     break;
                 case "KLL_QUANTILES":
-                    if (kllQuantilesFunctions.includes(func.token.literal.toUpperCase())) {
+                    if (keywords_1.kllQuantilesFunctions.includes(func.token.literal.toUpperCase())) {
                         func.isPreDefinedFunction = true;
                         toUpper(parent.token);
                     }
                     break;
                 case "OBJ":
-                    if (objFunctions.includes(func.token.literal.toUpperCase())) {
+                    if (keywords_1.objFunctions.includes(func.token.literal.toUpperCase())) {
                         func.isPreDefinedFunction = true;
                         toUpper(parent.token);
                     }
                     break;
                 case "ML":
-                    if (mlFunctions.includes(func.token.literal.toUpperCase())) {
+                    if (keywords_1.mlFunctions.includes(func.token.literal.toUpperCase())) {
                         func.isPreDefinedFunction = true;
                         toUpper(parent.token);
                     }
                     break;
                 case "AI":
-                    if (aiFunctions.includes(func.token.literal.toUpperCase())) {
+                    if (keywords_1.aiFunctions.includes(func.token.literal.toUpperCase())) {
                         func.isPreDefinedFunction = true;
                         toUpper(parent.token);
                     }
                     break;
                 case "VECTOR_INDEX":
-                    if (vectorIndexFunctions.includes(func.token.literal.toUpperCase())) {
+                    if (keywords_1.vectorIndexFunctions.includes(func.token.literal.toUpperCase())) {
                         func.isPreDefinedFunction = true;
                         toUpper(parent.token);
                     }
@@ -1630,63 +1639,63 @@ const printCallingFunctionGeneral = (path, options, print, node) => {
             if (grandParent.token.literal.toUpperCase() === "SAFE") {
                 switch (parent.token.literal.toUpperCase()) {
                     case "KEYS":
-                        if (keysFunctions.includes(func.token.literal.toUpperCase())) {
+                        if (keywords_1.keysFunctions.includes(func.token.literal.toUpperCase())) {
                             func.isPreDefinedFunction = true;
                             toUpper(parent.token);
                             toUpper(grandParent.token);
                         }
                         break;
                     case "AEAD":
-                        if (aeadFunctions.includes(func.token.literal.toUpperCase())) {
+                        if (keywords_1.aeadFunctions.includes(func.token.literal.toUpperCase())) {
                             func.isPreDefinedFunction = true;
                             toUpper(parent.token);
                             toUpper(grandParent.token);
                         }
                         break;
                     case "NET":
-                        if (netFunctions.includes(func.token.literal.toUpperCase())) {
+                        if (keywords_1.netFunctions.includes(func.token.literal.toUpperCase())) {
                             func.isPreDefinedFunction = true;
                             toUpper(parent.token);
                             toUpper(grandParent.token);
                         }
                         break;
                     case "HLL_COUNT":
-                        if (hllCountFunctions.includes(func.token.literal.toUpperCase())) {
+                        if (keywords_1.hllCountFunctions.includes(func.token.literal.toUpperCase())) {
                             func.isPreDefinedFunction = true;
                             toUpper(parent.token);
                             toUpper(grandParent.token);
                         }
                         break;
                     case "KLL_QUANTILES":
-                        if (kllQuantilesFunctions.includes(func.token.literal.toUpperCase())) {
+                        if (keywords_1.kllQuantilesFunctions.includes(func.token.literal.toUpperCase())) {
                             func.isPreDefinedFunction = true;
                             toUpper(parent.token);
                             toUpper(grandParent.token);
                         }
                         break;
                     case "OBJ":
-                        if (objFunctions.includes(func.token.literal.toUpperCase())) {
+                        if (keywords_1.objFunctions.includes(func.token.literal.toUpperCase())) {
                             func.isPreDefinedFunction = true;
                             toUpper(parent.token);
                             toUpper(grandParent.token);
                         }
                         break;
                     case "ML":
-                        if (mlFunctions.includes(func.token.literal.toUpperCase())) {
+                        if (keywords_1.mlFunctions.includes(func.token.literal.toUpperCase())) {
                             func.isPreDefinedFunction = true;
                             toUpper(parent.token);
                             toUpper(grandParent.token);
                         }
                         break;
                     case "AI":
-                        if (aiFunctions.includes(func.token.literal.toUpperCase())) {
+                        if (keywords_1.aiFunctions.includes(func.token.literal.toUpperCase())) {
                             func.isPreDefinedFunction = true;
                             toUpper(parent.token);
                             toUpper(grandParent.token);
                         }
                         break;
                     case "VECTOR_INDEX":
-                        if (vectorIndexFunctions.includes(func.token.literal.toUpperCase())) {
+                        if (keywords_1.vectorIndexFunctions.includes(func.token.literal.toUpperCase())) {
                             func.isPreDefinedFunction = true;
                             toUpper(parent.token);
                             toUpper(grandParent.token);
@@ -2091,7 +2100,7 @@ const printCreateFunctionStatement = (path, options, print, node) => {
         leading_comments: printLeadingComments(path, options, print, node),
         self: p.self("upper"),
         trailing_comments: printTrailingComments(path, options, print, node),
-        or_replace: p.child("or_replace", (x) => group([line, x])),
+        or_replace: p.child("or_replace", (x) => [" ", x], "all"),
         temp: p.child("temp", undefined, "all"),
         table: p.child("table", undefined, "all"),
         aggregate: p.child("aggregate", undefined, "all"),
@@ -2153,7 +2162,7 @@ const printCreateModelStatement = (path, options, print, node) => {
         leading_comments: printLeadingComments(path, options, print, node),
         self: p.self("upper"),
         trailing_comments: printTrailingComments(path, options, print, node),
-        or_replace: p.child("or_replace", (x) => group([line, x])),
+        or_replace: p.child("or_replace", (x) => [" ", x], "all"),
         what: p.child("what", undefined, "all"),
         if_not_exists: p.child("if_not_exists", (x) => group([line, x])),
         ident: p.child("ident", undefined, "all"),
@@ -2204,7 +2213,7 @@ const printCreateProcedureStatement = (path, options, print, node) => {
         leading_comments: printLeadingComments(path, options, print, node),
         self: p.self("upper"),
         trailing_comments: printTrailingComments(path, options, print, node),
-        or_replace: p.child("or_replace", (x) => group([line, x])),
+        or_replace: p.child("or_replace", (x) => [" ", x], "all"),
         what: p.child("what", undefined, "all"),
         if_not_exists: p.child("if_not_exists", (x) => group([line, x])),
         ident: p.child("ident", undefined, "all"),
@@ -2291,7 +2300,7 @@ const printCreateRowAccessPolicyStatement = (path, options, print, node) => {
         leading_comments: printLeadingComments(path, options, print, node),
         self: p.self("upper"),
         trailing_comments: printTrailingComments(path, options, print, node),
-        or_replace: p.child("or_replace", (x) => group([line, x])),
+        or_replace: p.child("or_replace", (x) => [" ", x], "all"),
         what: p.child("what", undefined, "all", " "),
         if_not_exists: p.child("if_not_exists", (x) => group([line, x])),
         ident: p.child("ident", undefined, "all"),
@@ -2374,7 +2383,7 @@ const printCreateIndexStatement = (path, options, print, node) => {
         leading_comments: printLeadingComments(path, options, print, node),
         self: p.self("upper"),
         trailing_comments: printTrailingComments(path, options, print, node),
-        or_replace: p.child("or_replace", (x) => group([line, x])),
+        or_replace: p.child("or_replace", (x) => [" ", x], "all"),
         what: p.child("what", undefined, "all"),
         if_not_exists: p.child("if_not_exists", (x) => group([line, x])),
         ident: p.child("ident", undefined, "all"),
@@ -2422,7 +2431,7 @@ const printCreateTableStatement = (path, options, print, node) => {
         leading_comments: printLeadingComments(path, options, print, node),
         self: p.self("upper"),
         trailing_comments: printTrailingComments(path, options, print, node),
-        or_replace: p.child("or_replace", (x) => group([line, x])),
+        or_replace: p.child("or_replace", (x) => [" ", x], "all"),
         temp: p.child("temp", undefined, "all"),
         external: p.child("external", undefined, "all"),
         snapshot: p.child("snapshot", undefined, "all"),
@@ -2493,7 +2502,7 @@ const printCreateViewStatement = (path, options, print, node) => {
         leading_comments: printLeadingComments(path, options, print, node),
         self: p.self("upper"),
         trailing_comments: printTrailingComments(path, options, print, node),
-        or_replace: p.child("or_replace", (x) => group([line, x])),
+        or_replace: p.child("or_replace", (x) => [" ", x], "all"),
         materialized: p.child("materialized", undefined, "all"),
         what: p.child("what", undefined, "all"),
         if_not_exists: p.child("if_not_exists", (x) => group([line, x])),
@@ -3349,7 +3358,7 @@ const printIfStatement = (path, options, print, node) => {
             docs.self,
             docs.trailing_comments,
             " ",
-            docs.condition,
+            group(docs.condition),
             " ",
             docs.then,
             docs.elseifs,
@@ -4361,16 +4370,18 @@ const printSelectPipeOperator = (path, options, print, node) => {
         self: p.self(),
         trailing_comments: printTrailingComments(path, options, print, node),
         keywords: p.child("keywords", undefined, "all"),
-        exprs: p.child("exprs", (x) => group([line, x])),
+        exprs: p.child("exprs", (x) => [line, x]),
         window: p.child("window"),
     };
     return [
         docs.leading_comments,
-        docs.self,
-        docs.trailing_comments,
-        p.has("keywords") ? " " : "",
-        docs.keywords,
-        indent(docs.exprs),
+        group([
+            docs.self,
+            docs.trailing_comments,
+            p.has("keywords") ? " " : "",
+            docs.keywords,
+            indent(docs.exprs),
+        ]),
         p.has("window") ? line : "",
         docs.window,
     ];