check function argument isn't a select statement

nene · Aug 29, 2024 · 26258e4 · 26258e4
1 parent 626354c
commit 26258e4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/node_utils.ts b/src/node_utils.ts
@@ -20,6 +20,8 @@ export const isStringLiteral = is("string_literal");
 export const isArraySubscript = is("array_subscript");
 export const isAsClause = is("as_clause");
 export const isParenExpr = is("paren_expr");
+export const isSelectStmt = is("select_stmt");
+export const isCompoundSelectStmt = is("compound_select_stmt");
 export const isListExpr = is("list_expr");
 export const isCreateFunctionStmt = is("create_function_stmt");
 export const isLanguageClause = is("language_clause");

diff --git a/src/syntax/expr.ts b/src/syntax/expr.ts
@@ -19,6 +19,8 @@ import {
   isFuncArgs,
   isListExpr,
   isProgram,
+  isSelectStmt,
+  isCompoundSelectStmt,
 } from "../node_utils";
 import { isString, last } from "../utils";
 
@@ -48,7 +50,9 @@ export const exprMap: CstToDocMap<AllExprNodes> = {
     // Discard unnecessary parenthesis around function arguments
     if (
       isListExpr(path.getParentNode(0)) &&
-      isFuncArgs(path.getParentNode(1))
+      isFuncArgs(path.getParentNode(1)) &&
+      !isSelectStmt(node.expr) &&
+      !isCompoundSelectStmt(node.expr)
     ) {
       return print("expr");
     }