happy-sql 2.1.0 → 2.2.0

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

@@ -7,22 +7,27 @@ const printAsExpression = (path, {write, traverse}) => {
     write(` AS ${path.node.typeAnnotation.literal.value}`);
 };
 
-const printTable = (path, ctx) => {
+export const printTable = (path, printer) => {
+    const {traverse} = printer;
+    
     if (isTSAsExpression(path.node))
-        return printAsExpression(path, ctx);
+        return printAsExpression(path, printer);
     
-    ctx.traverse(path);
+    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/limit/fixture/limit-fix.sql

@@ -0,0 +1,3 @@
+SELECT id
+FROM users
+LIMIT 10

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

@@ -0,0 +1,4 @@
+SELECT id
+FROM users
+LIMIT 10
+OFFSET 5

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

@@ -0,0 +1,12 @@
+[
+    select(
+        id,
+        from(
+            users,
+            limit(
+                10,
+                5,
+            ),
+        ),
+    ),
+];

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

@@ -0,0 +1,11 @@
+[
+    select(
+        id,
+        from(
+            users,
+            limit(
+                10,
+            ),
+        ),
+    ),
+];

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

@@ -0,0 +1,12 @@
+export const limit = (path, {write, traverse}) => {
+    const args = path.get('arguments');
+    
+    write('LIMIT ');
+    traverse(args[0]);
+    
+    if (args.length > 1) {
+        write.breakline();
+        write('OFFSET ');
+        traverse(args[1]);
+    }
+};

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

@@ -0,0 +1,2 @@
+SELECT MAX(price)
+FROM products

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

@@ -0,0 +1 @@
+select(max(price), from(products));

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

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

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

@@ -0,0 +1,2 @@
+SELECT MIN(price)
+FROM products

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

@@ -0,0 +1 @@
+select(min(price), from(products));

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

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

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-in-list/fixture/not-in-list-fix.sql

@@ -0,0 +1,3 @@
+SELECT *
+FROM users
+WHERE id NOT IN (1, 2, 3)

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

@@ -0,0 +1,3 @@
+[
+    select('*', from(users, where(notInList(id, 1, 2, 3)))),
+];

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

@@ -0,0 +1,15 @@
+export const notInList = (path, {write, traverse}) => {
+    const [head, ...items] = path.get('arguments');
+    
+    traverse(head);
+    write(' NOT IN (');
+    
+    for (let i = 0; i < items.length; i++) {
+        if (i > 0)
+            write(', ');
+        
+        traverse(items[i]);
+    }
+    
+    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/order-by/fixture/order-by-fix.sql

@@ -0,0 +1,3 @@
+SELECT id
+FROM users
+ORDER BY name

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

@@ -0,0 +1,3 @@
+SELECT id
+FROM users
+ORDER BY name, id DESC

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

@@ -0,0 +1,14 @@
+[
+    select(
+        id,
+        from(
+            users,
+            orderBy(
+                name,
+                desc(
+                    id,
+                ),
+            ),
+        ),
+    ),
+];

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

@@ -0,0 +1,11 @@
+[
+    select(
+        id,
+        from(
+            users,
+            orderBy(
+                name,
+            ),
+        ),
+    ),
+];

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

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

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/returning/returning.js

@@ -7,11 +7,13 @@ const printAsExpression = (path, {write, traverse}) => {
     write(` AS ${path.node.typeAnnotation.literal.value}`);
 };
 
-const printItem = (path, ctx) => {
+const printItem = (path, printer) => {
+    const {traverse} = printer;
+    
     if (isTSAsExpression(path.node))
-        return printAsExpression(path, ctx);
+        return printAsExpression(path, printer);
     
-    ctx.traverse(path);
+    traverse(path);
 };
 
 export const returning = (path, {write, traverse}) => {

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