happy-sql 2.1.1 → 2.2.0

package/lib/printer/visitors/call-expression/clauses/full-outer-join/full-outer-join.js

@@ -0,0 +1,9 @@
+import {printJoinBody} from '../join/join.js';
+
+export const fullOuterJoin = (path, printer) => {
+    const {write} = printer;
+    
+    write.breakline();
+    write('FULL OUTER JOIN ');
+    printJoinBody(path, printer);
+};

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

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

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

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

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

@@ -0,0 +1,7 @@
+export const ilike = (path, {write, traverse}) => {
+    const args = path.get('arguments');
+    
+    traverse(args[0]);
+    write(' ILIKE ');
+    traverse(args[1]);
+};

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

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

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

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

package/lib/printer/visitors/call-expression/clauses/in-query/in-query.js

@@ -0,0 +1,8 @@
+export const inQuery = (path, {write, traverse}) => {
+    const [col, stmt] = path.get('arguments');
+    
+    traverse(col);
+    write(' IN (');
+    traverse(stmt);
+    write(')');
+};

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

@@ -0,0 +1,3 @@
+SELECT *
+FROM users
+INNER JOIN orders ON users.id = orders.user_id

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

@@ -0,0 +1 @@
+[select('*', from(users, innerJoin(orders, on(users.id === orders.user_id))))];

package/lib/printer/visitors/call-expression/clauses/inner-join/inner-join.js

@@ -0,0 +1,8 @@
+import {printJoinBody} from '../join/join.js';
+
+export const innerJoin = (path, printer) => {
+    const {write} = printer;
+    write.breakline();
+    write('INNER JOIN ');
+    printJoinBody(path, printer);
+};

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

@@ -1,11 +1,24 @@
+const isOrAction = (arg) => {
+    const {name} = arg.node.callee;
+    
+    return name === 'orReplace' || name === 'orIgnore';
+};
+
 export const insert = (path, {write, traverse}) => {
     const args = path.get('arguments');
-    const [intoArg] = args;
-    const rest = args.slice(1);
+    const [first] = args;
+    const rest = isOrAction(first) ? args.slice(1) : args;
+    
+    write('INSERT ');
+    
+    if (isOrAction(first)) {
+        traverse(first);
+        write.breakline();
+    }
     
-    traverse(intoArg);
+    traverse(rest[0]);
     
-    for (const arg of rest) {
+    for (const arg of rest.slice(1)) {
         write.breakline();
         traverse(arg);
     }

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

@@ -1 +1 @@
-INSERT INTO t (x) VALUES (:x)
+INTO t (x) VALUES (:x)

package/lib/printer/visitors/call-expression/clauses/into/fixture/single-column-fix.sql

@@ -1 +1 @@
-INSERT INTO t (x) VALUES (:x)
+INTO t (x) VALUES (:x)

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

@@ -4,7 +4,7 @@ export const into = (path, {write, traverse}) => {
     const [, columns] = args;
     const values = args[2];
     
-    write('INSERT INTO ');
+    write('INTO ');
     traverse(table);
     write(' (');
     

package/lib/printer/visitors/call-expression/clauses/is-distinct-from/fixture/is-distinct-from-fix.sql

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

package/lib/printer/visitors/call-expression/clauses/is-distinct-from/fixture/is-distinct-from.js

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

package/lib/printer/visitors/call-expression/clauses/is-distinct-from/is-distinct-from.js

@@ -0,0 +1,7 @@
+export const isDistinctFrom = (path, {write, traverse}) => {
+    const args = path.get('arguments');
+    
+    traverse(args[0]);
+    write(' IS DISTINCT FROM ');
+    traverse(args[1]);
+};

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

@@ -7,7 +7,7 @@ const printAsExpression = (path, {write, traverse}) => {
     write(` AS ${path.node.typeAnnotation.literal.value}`);
 };
 
-const printTable = (path, printer) => {
+export const printTable = (path, printer) => {
     const {traverse} = printer;
     
     if (isTSAsExpression(path.node))
@@ -16,15 +16,18 @@ const printTable = (path, printer) => {
     traverse(path);
 };
 
-export const join = (path, {write, traverse}) => {
+export const printJoinBody = (path, printer) => {
+    const {write, traverse} = printer;
     const [table, onClause] = path.get('arguments');
     
-    write.breakline();
-    write('JOIN ');
-    printTable(table, {
-        write,
-        traverse,
-    });
+    printTable(table, printer);
     write.space();
     traverse(onClause);
 };
+
+export const join = (path, printer) => {
+    const {write} = printer;
+    write.breakline();
+    write('JOIN ');
+    printJoinBody(path, printer);
+};

package/lib/printer/visitors/call-expression/clauses/json-op.js

@@ -0,0 +1,21 @@
+const JSON_OPS = new Set([
+    '->',
+    '->>',
+    '#>',
+    '#>>',
+    '?',
+    '?|',
+    '?&',
+    '@>',
+    '<@',
+]);
+
+export const isJsonTuple = (node) => node.type === 'ArrayExpression' && node.elements.length === 3 && node.elements[1].type === 'StringLiteral' && JSON_OPS.has(node.elements[1].value);
+
+export const printJsonTuple = (path, {write, traverse}) => {
+    const elems = path.get('elements');
+    
+    traverse(elems[0]);
+    write(` ${elems[1].node.value} `);
+    traverse(elems[2]);
+};

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

@@ -0,0 +1,2 @@
+SELECT *
+FROM t, LATERAL (SELECT 1) AS sub

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

@@ -0,0 +1,15 @@
+[
+    select(
+        '*',
+        from(
+            t,
+            lateralJoin(
+                lateralSubquery(
+                    select(
+                        1,
+                    ),
+                ) as 'sub',
+            ),
+        ),
+    ),
+];

package/lib/printer/visitors/call-expression/clauses/lateral-join/fixture/lateral-no-as-fix.sql

@@ -0,0 +1,2 @@
+SELECT *
+FROM t, LATERAL (SELECT 1)

package/lib/printer/visitors/call-expression/clauses/lateral-join/fixture/lateral-no-as.js

@@ -0,0 +1,15 @@
+[
+    select(
+        '*',
+        from(
+            t,
+            lateralJoin(
+                lateralSubquery(
+                    select(
+                        1,
+                    ),
+                ),
+            ),
+        ),
+    ),
+];

package/lib/printer/visitors/call-expression/clauses/lateral-join/lateral-join.js

@@ -0,0 +1,19 @@
+import {types} from '@putout/babel';
+
+const {isTSAsExpression} = types;
+
+export const lateralJoin = (path, {write, traverse}) => {
+    write(', ');
+    
+    const [arg] = path.get('arguments');
+    const {node} = arg;
+    
+    if (isTSAsExpression(node)) {
+        traverse(arg.get('expression'));
+        write(` AS ${node.typeAnnotation.literal.value}`);
+        
+        return;
+    }
+    
+    traverse(arg);
+};

package/lib/printer/visitors/call-expression/clauses/lateral-subquery/lateral-subquery.js

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

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

@@ -0,0 +1,3 @@
+SELECT *
+FROM users
+LEFT JOIN orders ON users.id = orders.user_id

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

@@ -0,0 +1 @@
+[select('*', from(users, leftJoin(orders, on(users.id === orders.user_id))))];

package/lib/printer/visitors/call-expression/clauses/left-join/left-join.js

@@ -0,0 +1,8 @@
+import {printJoinBody} from '../join/join.js';
+
+export const leftJoin = (path, printer) => {
+    const {write} = printer;
+    write.breakline();
+    write('LEFT JOIN ');
+    printJoinBody(path, printer);
+};

package/lib/printer/visitors/call-expression/clauses/left-outer-join/fixture/left-outer-join-fix.sql

@@ -0,0 +1,3 @@
+SELECT *
+FROM users
+LEFT OUTER JOIN orders ON users.id = orders.user_id

package/lib/printer/visitors/call-expression/clauses/left-outer-join/fixture/left-outer-join.js

@@ -0,0 +1 @@
+[select('*', from(users, leftOuterJoin(orders, on(users.id === orders.user_id))))];

package/lib/printer/visitors/call-expression/clauses/left-outer-join/left-outer-join.js

@@ -0,0 +1,8 @@
+import {printJoinBody} from '../join/join.js';
+
+export const leftOuterJoin = (path, printer) => {
+    const {write} = printer;
+    write.breakline();
+    write('LEFT OUTER JOIN ');
+    printJoinBody(path, printer);
+};

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

@@ -0,0 +1,3 @@
+SELECT *
+FROM a
+NATURAL JOIN b

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

@@ -0,0 +1 @@
+[select('*', from(a, naturalJoin(b)))];

package/lib/printer/visitors/call-expression/clauses/natural-join/natural-join.js

@@ -0,0 +1,9 @@
+import {printTable} from '../join/join.js';
+
+export const naturalJoin = (path, printer) => {
+    const {write} = printer;
+    
+    write.breakline();
+    write('NATURAL JOIN ');
+    printTable(path.get('arguments')[0], printer);
+};

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

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

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

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

package/lib/printer/visitors/call-expression/clauses/not-between/not-between.js

@@ -0,0 +1,9 @@
+export const notBetween = (path, {write, traverse}) => {
+    const [col, lo, hi] = path.get('arguments');
+    
+    traverse(col);
+    write(' NOT BETWEEN ');
+    traverse(lo);
+    write(' AND ');
+    traverse(hi);
+};

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

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

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

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

package/lib/printer/visitors/call-expression/clauses/not-exists/not-exists.js

@@ -0,0 +1,5 @@
+export const notExists = (path, {write, traverse}) => {
+    write('NOT EXISTS (');
+    traverse(path.get('arguments')[0]);
+    write(')');
+};

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

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

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

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

package/lib/printer/visitors/call-expression/clauses/not-like/not-like.js

@@ -0,0 +1,7 @@
+export const notLike = (path, {write, traverse}) => {
+    const args = path.get('arguments');
+    
+    traverse(args[0]);
+    write(' NOT LIKE ');
+    traverse(args[1]);
+};

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

@@ -0,0 +1 @@
+id INT NOT NULL

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

@@ -0,0 +1,3 @@
+[
+    column(id, INT, notNull()),
+];

package/lib/printer/visitors/call-expression/clauses/not-null/not-null.js

@@ -0,0 +1,3 @@
+export const notNull = (path, {write}) => {
+    write('NOT NULL');
+};

package/lib/printer/visitors/call-expression/clauses/on/fixture/on-json-tuple-fix.sql

@@ -0,0 +1,3 @@
+SELECT *
+FROM a
+JOIN b ON a.data ? 'x'

package/lib/printer/visitors/call-expression/clauses/on/fixture/on-json-tuple.js

@@ -0,0 +1 @@
+[select('*', from(a, join(b, on([a.data, '?', 'x']))))];

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

@@ -0,0 +1,2 @@
+INSERT OR IGNORE
+INTO t (id) VALUES (1)

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

@@ -0,0 +1 @@
+[insert(orIgnore(), into(t, id, values(1)))];

package/lib/printer/visitors/call-expression/clauses/or-ignore/or-ignore.js

@@ -0,0 +1,3 @@
+export const orIgnore = (path, {write}) => {
+    write('OR IGNORE');
+};

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

@@ -0,0 +1,2 @@
+INSERT OR REPLACE
+INTO t (id) VALUES (1)

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

@@ -0,0 +1 @@
+[insert(orReplace(), into(t, id, values(1)))];

package/lib/printer/visitors/call-expression/clauses/or-replace/or-replace.js

@@ -0,0 +1,3 @@
+export const orReplace = (path, {write}) => {
+    write('OR REPLACE');
+};

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

@@ -0,0 +1,2 @@
+SELECT ROW_NUMBER() OVER (PARTITION BY kind ORDER BY id)
+FROM users

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

@@ -0,0 +1 @@
+[select(over(rowNumber(), partitionBy(kind), orderBy(id)), from(users))];

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

@@ -0,0 +1,17 @@
+export const over = (path, {write, traverse}) => {
+    const args = path.get('arguments');
+    const [func] = args;
+    const clauses = args.slice(1);
+    
+    traverse(func);
+    write(' OVER (');
+    
+    for (const [i, clause] of clauses.entries()) {
+        if (i > 0)
+            write(' ');
+        
+        traverse(clause);
+    }
+    
+    write(')');
+};

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

@@ -0,0 +1 @@
+PARTITION BY kind, name

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

@@ -0,0 +1 @@
+[partitionBy(kind, name)];

package/lib/printer/visitors/call-expression/clauses/partition-by/partition-by.js

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

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

@@ -0,0 +1,3 @@
+SELECT *
+FROM t
+WHERE data @> '{}'::jsonb

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

@@ -0,0 +1,3 @@
+[
+    select('*', from(t, where([data, '@>', pgCast('{}', jsonb)]))),
+];

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

@@ -0,0 +1,7 @@
+export const pgCast = (path, {write, traverse}) => {
+    const [value, type] = path.get('arguments');
+    
+    traverse(value);
+    write('::');
+    traverse(type);
+};

package/lib/printer/visitors/call-expression/clauses/pragma/fixture/pragma-table-info-fix.sql

@@ -0,0 +1 @@
+PRAGMA table_info(users)

package/lib/printer/visitors/call-expression/clauses/pragma/fixture/pragma-table-info.js

@@ -0,0 +1 @@
+[pragma(table_info, users)];

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

@@ -0,0 +1,9 @@
+export const pragma = (path, {write, traverse}) => {
+    const [name, arg] = path.get('arguments');
+    
+    write('PRAGMA ');
+    traverse(name);
+    write('(');
+    traverse(arg);
+    write(')');
+};

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

@@ -0,0 +1 @@
+user_id INT REFERENCES users(id, countery)

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

@@ -0,0 +1,3 @@
+[
+    column(user_id, INT, references(users, id, countery)),
+];

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

@@ -0,0 +1 @@
+user_id INT REFERENCES users(id)

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

@@ -0,0 +1,3 @@
+[
+    column(user_id, INT, references(users, id)),
+];

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

@@ -0,0 +1,17 @@
+export const references = (path, {write, traverse}) => {
+    write('REFERENCES ');
+    const [first, ...other] = path.get('arguments');
+    
+    traverse(first);
+    
+    write('(');
+    
+    for (const [i, arg] of other.entries()) {
+        if (i > 0)
+            write(', ');
+        
+        traverse(arg);
+    }
+    
+    write(')');
+};

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

@@ -0,0 +1,8 @@
+export const renameColumn = (path, {write, traverse}) => {
+    const [from, to] = path.get('arguments');
+    
+    write('RENAME COLUMN ');
+    traverse(from);
+    write(' TO ');
+    traverse(to);
+};

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

@@ -0,0 +1,2 @@
+INSERT INTO t (name) VALUES ('test')
+RETURNING *

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

@@ -0,0 +1,14 @@
+[
+    insert(
+        into(
+            t,
+            name,
+            values(
+                'test',
+            ),
+        ),
+        returning(
+            '*',
+        ),
+    ),
+];

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

@@ -0,0 +1,3 @@
+SELECT *
+FROM users
+RIGHT JOIN orders ON users.id = orders.user_id

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

@@ -0,0 +1 @@
+[select('*', from(users, rightJoin(orders, on(users.id === orders.user_id))))];

package/lib/printer/visitors/call-expression/clauses/right-join/right-join.js

@@ -0,0 +1,8 @@
+import {printJoinBody} from '../join/join.js';
+
+export const rightJoin = (path, printer) => {
+    const {write} = printer;
+    write.breakline();
+    write('RIGHT JOIN ');
+    printJoinBody(path, printer);
+};

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

@@ -0,0 +1 @@
+ROLLBACK

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

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

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

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

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

@@ -0,0 +1 @@
+ROLLBACK TO SAVEPOINT sp1