happy-sql 2.1.1 → 2.2.0

package/lib/parser/visitors/clause/where/fixture/where-any-fix.js

@@ -0,0 +1,17 @@
+[
+    select(
+        '*',
+        from(
+            t,
+            where(
+                id === ANY(
+                    array(
+                        1,
+                        2,
+                        3,
+                    ),
+                ),
+            ),
+        ),
+    ),
+];

package/lib/parser/visitors/clause/where/fixture/where-any.sql

@@ -0,0 +1 @@
+SELECT * FROM t WHERE id = ANY(ARRAY[1, 2, 3])

package/lib/parser/visitors/clause/where/fixture/where-exists-fix.js

@@ -0,0 +1,21 @@
+[
+    select(
+        '*',
+        from(
+            t,
+            where(
+                exists(
+                    select(
+                        1,
+                        from(
+                            u,
+                            where(
+                                u.id === t.id,
+                            ),
+                        ),
+                    ),
+                ),
+            ),
+        ),
+    ),
+];

package/lib/parser/visitors/clause/where/fixture/where-exists.sql

@@ -0,0 +1 @@
+SELECT * FROM t WHERE EXISTS (SELECT 1 FROM u WHERE u.id = t.id)

package/lib/parser/visitors/clause/where/fixture/where-ilike-fix.js

@@ -0,0 +1,14 @@
+[
+    select(
+        '*',
+        from(
+            t,
+            where(
+                ilike(
+                    name,
+                    '%foo%',
+                ),
+            ),
+        ),
+    ),
+];

package/lib/parser/visitors/clause/where/fixture/where-ilike.sql

@@ -0,0 +1 @@
+SELECT * FROM t WHERE name ILIKE '%foo%'

package/lib/parser/visitors/clause/where/fixture/where-in-subquery-fix.js

@@ -0,0 +1,19 @@
+[
+    select(
+        id,
+        from(
+            t,
+            where(
+                inQuery(
+                    id,
+                    select(
+                        user_id,
+                        from(
+                            orders,
+                        ),
+                    ),
+                ),
+            ),
+        ),
+    ),
+];

package/lib/parser/visitors/clause/where/fixture/where-in-subquery.sql

@@ -0,0 +1 @@
+SELECT id FROM t WHERE id IN (SELECT user_id FROM orders)

package/lib/parser/visitors/clause/where/fixture/where-is-distinct-from-fix.js

@@ -0,0 +1,14 @@
+[
+    select(
+        '*',
+        from(
+            t,
+            where(
+                isDistinctFrom(
+                    a,
+                    b,
+                ),
+            ),
+        ),
+    ),
+];

package/lib/parser/visitors/clause/where/fixture/where-is-distinct-from.sql

@@ -0,0 +1 @@
+SELECT * FROM t WHERE a IS DISTINCT FROM b

package/lib/parser/visitors/clause/where/fixture/where-nested-parens-fix.js

@@ -0,0 +1,11 @@
+[
+    select(
+        '*',
+        from(
+            t,
+            where(
+                a === 1 || b === 2 && c === 3,
+            ),
+        ),
+    ),
+];

package/lib/parser/visitors/clause/where/fixture/where-nested-parens.sql

@@ -0,0 +1 @@
+SELECT * FROM t WHERE (a = 1 OR b = 2) AND c = 3

package/lib/parser/visitors/clause/where/fixture/where-not-between-fix.js

@@ -0,0 +1,15 @@
+[
+    select(
+        '*',
+        from(
+            t,
+            where(
+                notBetween(
+                    x,
+                    1,
+                    10,
+                ),
+            ),
+        ),
+    ),
+];

package/lib/parser/visitors/clause/where/fixture/where-not-between.sql

@@ -0,0 +1 @@
+SELECT * FROM t WHERE x NOT BETWEEN 1 AND 10

package/lib/parser/visitors/clause/where/fixture/where-not-exists-fix.js

@@ -0,0 +1,18 @@
+[
+    select(
+        '*',
+        from(
+            t,
+            where(
+                notExists(
+                    select(
+                        1,
+                        from(
+                            u,
+                        ),
+                    ),
+                ),
+            ),
+        ),
+    ),
+];

package/lib/parser/visitors/clause/where/fixture/where-not-exists.sql

@@ -0,0 +1 @@
+SELECT * FROM t WHERE NOT EXISTS (SELECT 1 FROM u)

package/lib/parser/visitors/clause/where/fixture/where-not-like-fix.js

@@ -0,0 +1,14 @@
+[
+    select(
+        '*',
+        from(
+            t,
+            where(
+                notLike(
+                    name,
+                    '%x%',
+                ),
+            ),
+        ),
+    ),
+];

package/lib/parser/visitors/clause/where/fixture/where-not-like.sql

@@ -0,0 +1 @@
+SELECT * FROM t WHERE name NOT LIKE '%x%'

package/lib/parser/visitors/clause/where/fixture/where-similar-to-fix.js

@@ -0,0 +1,14 @@
+[
+    select(
+        '*',
+        from(
+            t,
+            where(
+                similarTo(
+                    name,
+                    '%foo%',
+                ),
+            ),
+        ),
+    ),
+];

package/lib/parser/visitors/clause/where/fixture/where-similar-to.sql

@@ -0,0 +1 @@
+SELECT * FROM t WHERE name SIMILAR TO '%foo%'

package/lib/parser/visitors/clause/where/where.js

@@ -1,101 +1,59 @@
 import {types} from '@putout/babel';
 import {convertOperand} from '#parser/clause/operand';
-
-const isString = (a) => typeof a === 'string';
+import {convertClause} from '#parser/clause';
+import {convertBinary} from '#parser/clause/binary';
 
 const {
     logicalExpression,
     identifier,
     callExpression,
-    binaryExpression,
 } = types;
 
-const opName = (op) => {
-    if (Array.isArray(op))
-        return op
-            .map((k) => k.name)
-            .join(' ');
-    
-    if (isString(op))
-        return op;
-    
-    return op?.name ?? '';
-};
-
-const SQL_TO_JS_OP = {
-    '=': '===',
-    '!=': '!==',
-    '>=': '>=',
-    '<=': '<=',
-    '>': '>',
-    '<': '<',
-};
-
 const convertAnd = (node) => logicalExpression('&&', convertCondition(node.left), convertCondition(node.right));
 const convertOr = (node) => logicalExpression('||', convertCondition(node.left), convertCondition(node.right));
 
-export const convertBetween = (node) => callExpression(identifier('between'), [
-    convertOperand(node.left),
-    convertOperand(node.begin),
-    convertOperand(node.end),
-]);
+const isNotBetween = (node) => Array.isArray(node.betweenKw) && node.betweenKw[0].name === 'NOT';
 
-const convertBinary = (node) => {
-    const op = opName(node.operator);
-    
-    if (op === 'LIKE')
-        return callExpression(identifier('like'), [
-            convertOperand(node.left),
-            convertOperand(node.right),
-        ]);
-    
-    if (op === 'IS')
-        return callExpression(identifier('isNull'), [
-            convertOperand(node.left),
-        ]);
-    
-    if (op === 'IS NOT')
-        return callExpression(identifier('isNotNull'), [
-            convertOperand(node.left),
-        ]);
-    
-    if (op === 'IN') {
-        const items = node.right.expr.items.map(convertOperand);
-        
-        return callExpression(identifier('inList'), [
-            convertOperand(node.left),
-            ...items,
-        ]);
-    }
-    
-    if (op === 'NOT IN') {
-        const items = node.right.expr.items.map(convertOperand);
-        
-        return callExpression(identifier('notInList'), [
-            convertOperand(node.left),
-            ...items,
-        ]);
-    }
+export const convertBetween = (node) => {
+    const args = [
+        convertOperand(node.left),
+        convertOperand(node.begin),
+        convertOperand(node.end),
+    ];
     
-    const jsOp = SQL_TO_JS_OP[op] ?? '===';
+    if (isNotBetween(node))
+        return callExpression(identifier('notBetween'), args);
     
-    return binaryExpression(
-        jsOp,
-        convertOperand(node.left),
-        convertOperand(node.right),
-    );
+    return callExpression(identifier('between'), args);
 };
 
+const isExists = (node) => node.type === 'prefix_op_expr' && node.operator.name === 'EXISTS';
+const isNotExists = (node) => node.type === 'prefix_op_expr' && node.operator.name === 'NOT' && isExists(node.expr);
+const isParenExpr = (node) => node.type === 'paren_expr';
+
 export const convertCondition = (node) => {
-    if (node.operator?.name === 'AND')
+    if (node.operator && node.operator.name === 'AND')
         return convertAnd(node);
     
-    if (node.operator?.name === 'OR')
+    if (node.operator && node.operator.name === 'OR')
         return convertOr(node);
     
     if (node.type === 'between_expr')
         return convertBetween(node);
     
+    if (isNotExists(node))
+        return callExpression(identifier('notExists'), [
+            convertClause(node.expr.expr.expr),
+        ]);
+    
+    if (isExists(node))
+        return callExpression(identifier('exists'), [
+            convertClause(node.expr.expr),
+        ]);
+    
+    if (isParenExpr(node))
+        return convertCondition(node.expr);
+    
     return convertBinary(node);
 };
 

package/lib/printer/visitors/array-expression.js

@@ -1,4 +1,11 @@
-export const ArrayExpression = (path, {write, traverse}) => {
+import {isJsonTuple, printJsonTuple} from './call-expression/clauses/json-op.js';
+
+export const ArrayExpression = (path, printer) => {
+    const {write, traverse} = printer;
+    
+    if (isJsonTuple(path.node))
+        return printJsonTuple(path, printer);
+    
     const elements = path.get('elements');
     const n = elements.length - 1;
     

package/lib/printer/visitors/boolean-literal.js

@@ -0,0 +1,8 @@
+export const BooleanLiteral = (path, {write}) => {
+    const {value} = path.node;
+    
+    if (value)
+        return write('TRUE');
+    
+    write('FALSE');
+};

package/lib/printer/visitors/call-expression/call-expression.js

@@ -1,11 +1,28 @@
 import {clauses} from './clauses/clauses.js';
 
+const printGenericCall = (path, {write, traverse}) => {
+    const {name} = path.node.callee;
+    const args = path.get('arguments');
+    
+    write(name);
+    write('(');
+    
+    for (const [i, arg] of args.entries()) {
+        if (i > 0)
+            write(', ');
+        
+        traverse(arg);
+    }
+    
+    write(')');
+};
+
 export const CallExpression = (path, printer) => {
     const {name} = path.node.callee;
     const visit = clauses[name];
     
     if (!visit)
-        throw Error(`${name} not supported yet`);
+        return printGenericCall(path, printer);
     
     visit(path, printer);
 };

package/lib/printer/visitors/call-expression/clauses/add-column/add-column.js

@@ -0,0 +1,4 @@
+export const addColumn = (path, {write, traverse}) => {
+    write('ADD COLUMN ');
+    traverse(path.get('arguments')[0]);
+};

package/lib/printer/visitors/call-expression/clauses/all/all.js

@@ -0,0 +1,10 @@
+export const ALL = (path, {write, traverse}) => {
+    write('ALL(');
+    
+    const args = path.get('arguments');
+    
+    for (const arg of args)
+        traverse(arg);
+    
+    write(')');
+};

package/lib/printer/visitors/call-expression/clauses/alter-column-type/alter-column-type.js

@@ -0,0 +1,8 @@
+export const alterColumnType = (path, {write, traverse}) => {
+    const [col, type] = path.get('arguments');
+    
+    write('ALTER COLUMN ');
+    traverse(col);
+    write(' TYPE ');
+    traverse(type);
+};

package/lib/printer/visitors/call-expression/clauses/alter-table/alter-table.js

@@ -0,0 +1,12 @@
+export const alterTable = (path, {write, traverse}) => {
+    const args = path.get('arguments');
+    const [table] = args;
+    
+    write('ALTER TABLE ');
+    traverse(table);
+    
+    for (const arg of args.slice(1)) {
+        write.newline();
+        traverse(arg);
+    }
+};

package/lib/printer/visitors/call-expression/clauses/alter-table/fixture/alter-table-add-column-fix.sql

@@ -0,0 +1,2 @@
+ALTER TABLE t
+ADD COLUMN y INT

package/lib/printer/visitors/call-expression/clauses/alter-table/fixture/alter-table-add-column.js

@@ -0,0 +1,11 @@
+[
+    alterTable(
+        t,
+        addColumn(
+            column(
+                y,
+                INT,
+            ),
+        ),
+    ),
+];

package/lib/printer/visitors/call-expression/clauses/alter-table/fixture/alter-table-alter-column-type-fix.sql

@@ -0,0 +1,2 @@
+ALTER TABLE t
+ALTER COLUMN x TYPE TEXT

package/lib/printer/visitors/call-expression/clauses/alter-table/fixture/alter-table-alter-column-type.js

@@ -0,0 +1 @@
+[alterTable(t, alterColumnType(x, TEXT))];

package/lib/printer/visitors/call-expression/clauses/alter-table/fixture/alter-table-drop-column-fix.sql

@@ -0,0 +1,2 @@
+ALTER TABLE t
+DROP COLUMN name

package/lib/printer/visitors/call-expression/clauses/alter-table/fixture/alter-table-drop-column.js

@@ -0,0 +1 @@
+[alterTable(t, dropColumn(name))];

package/lib/printer/visitors/call-expression/clauses/alter-table/fixture/alter-table-rename-column-fix.sql

@@ -0,0 +1,2 @@
+ALTER TABLE t
+RENAME COLUMN old_name TO new_name

package/lib/printer/visitors/call-expression/clauses/alter-table/fixture/alter-table-rename-column.js

@@ -0,0 +1 @@
+[alterTable(t, renameColumn(old_name, new_name))];

package/lib/printer/visitors/call-expression/clauses/any/any.js

@@ -0,0 +1,10 @@
+export const ANY = (path, {write, traverse}) => {
+    write('ANY(');
+    
+    const args = path.get('arguments');
+    
+    for (const arg of args)
+        traverse(arg);
+    
+    write(')');
+};

package/lib/printer/visitors/call-expression/clauses/any/fixture/any-fix.sql

@@ -0,0 +1,3 @@
+SELECT *
+FROM t
+WHERE id = ANY(ARRAY[1, 2, 3])

package/lib/printer/visitors/call-expression/clauses/any/fixture/any.js

@@ -0,0 +1,3 @@
+[
+    select('*', from(t, where(id === ANY(array(1, 2, 3))))),
+];

package/lib/printer/visitors/call-expression/clauses/array/array.js

@@ -0,0 +1,14 @@
+export const array = (path, {write, traverse}) => {
+    const args = path.get('arguments');
+    
+    write('ARRAY[');
+    
+    for (const [i, arg] of args.entries()) {
+        if (i > 0)
+            write(', ');
+        
+        traverse(arg);
+    }
+    
+    write(']');
+};

package/lib/printer/visitors/call-expression/clauses/attach/attach.js

@@ -0,0 +1,8 @@
+export const attach = (path, {write, traverse}) => {
+    const [file, schema] = path.get('arguments');
+    
+    write('ATTACH DATABASE ');
+    traverse(file);
+    write(' AS ');
+    traverse(schema);
+};

package/lib/printer/visitors/call-expression/clauses/attach/fixture/attach-fix.sql

@@ -0,0 +1 @@
+ATTACH DATABASE 'file.db' AS other

package/lib/printer/visitors/call-expression/clauses/attach/fixture/attach.js

@@ -0,0 +1 @@
+[attach('file.db', other)];

package/lib/printer/visitors/call-expression/clauses/begin/begin.js

@@ -0,0 +1,3 @@
+export const begin = (path, {write}) => {
+    write('BEGIN');
+};

package/lib/printer/visitors/call-expression/clauses/begin/fixture/begin-fix.sql

@@ -0,0 +1 @@
+BEGIN

package/lib/printer/visitors/call-expression/clauses/begin/fixture/begin.js

@@ -0,0 +1 @@
+[begin()];

package/lib/printer/visitors/call-expression/clauses/case-when/case-when.js

@@ -0,0 +1,33 @@
+const isElseClause = (path) => {
+    const {callee} = path.node;
+    
+    return callee && callee.name === 'else_';
+};
+
+export const caseWhen = (path, printer) => {
+    const {write} = printer;
+    const args = path.get('arguments');
+    const lastArg = args.at(-1);
+    const hasElse = isElseClause(lastArg);
+    
+    write('CASE ');
+    
+    for (let i = 0; i < args.length - 1; i += 2) {
+        write('WHEN ');
+        printer.traverse(args[i]);
+        write(' THEN ');
+        printer.traverse(args[i + 1]);
+        
+        // space between clauses / before ELSE — but not trailing
+        if (i < args.length - 3)
+            write(' ');
+    }
+    
+    if (hasElse) {
+        write(' ');
+        write('ELSE ');
+        printer.traverse(lastArg.get('arguments')[0]);
+    }
+    
+    write(' END');
+};

package/lib/printer/visitors/call-expression/clauses/case-when/fixture/case-when-fix.sql

@@ -0,0 +1,2 @@
+SELECT CASE WHEN kind = 'const' THEN 1 WHEN kind = 'let' THEN 2 ELSE 0 END
+FROM t

package/lib/printer/visitors/call-expression/clauses/case-when/fixture/case-when.js

@@ -0,0 +1,16 @@
+[
+    select(
+        caseWhen(
+            kind === 'const',
+            1,
+            kind === 'let',
+            2,
+            else_(
+                0,
+            ),
+        ),
+        from(
+            t,
+        ),
+    ),
+];

package/lib/printer/visitors/call-expression/clauses/cast/cast.js

@@ -0,0 +1,9 @@
+export const cast = (path, {write, traverse}) => {
+    const [expr, typeArg] = path.get('arguments');
+    
+    write('CAST(');
+    traverse(expr);
+    write(' AS ');
+    traverse(typeArg);
+    write(')');
+};

package/lib/printer/visitors/call-expression/clauses/cast/fixture/cast-fix.sql

@@ -0,0 +1,2 @@
+SELECT CAST(id AS TEXT)
+FROM t

package/lib/printer/visitors/call-expression/clauses/cast/fixture/cast.js

@@ -0,0 +1,11 @@
+[
+    select(
+        cast(
+            id,
+            TEXT,
+        ),
+        from(
+            t,
+        ),
+    ),
+];

package/lib/printer/visitors/call-expression/clauses/check/check.js

@@ -0,0 +1,8 @@
+export const check = (path, {write, traverse}) => {
+    write('CHECK (');
+    
+    const [arg] = path.get('arguments');
+    traverse(arg);
+    
+    write(')');
+};

package/lib/printer/visitors/call-expression/clauses/check/fixture/check-fix.sql

@@ -0,0 +1 @@
+x INT CHECK (x > 0)

package/lib/printer/visitors/call-expression/clauses/check/fixture/check.js

@@ -0,0 +1,3 @@
+[
+    column(x, INT, check(x > 0)),
+];