happy-sql 2.1.0 → 2.2.0

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

@@ -0,0 +1,2 @@
+CREATE TABLE t (
+id INT NOT NULL)

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

@@ -0,0 +1,11 @@
+[
+    createTable(
+        t, [
+            column(
+                id,
+                INT,
+                notNull(),
+            ),
+        ],
+    ),
+];

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

@@ -0,0 +1,2 @@
+CREATE TABLE orders (
+user_id INT REFERENCES users(id))

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

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

package/lib/printer/visitors/call-expression/clauses/create-table/fixture/create-table-unique-fix.sql

@@ -0,0 +1,2 @@
+CREATE TABLE t (
+x INT UNIQUE)

package/lib/printer/visitors/call-expression/clauses/create-table/fixture/create-table-unique.js

@@ -0,0 +1,11 @@
+[
+    createTable(
+        t, [
+            column(
+                x,
+                INT,
+                unique(),
+            ),
+        ],
+    ),
+];

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

@@ -0,0 +1,6 @@
+import {printTableBody} from '../create-table/create-table.js';
+
+export const createTableIfNotExists = (path, printer) => {
+    printer.write('CREATE TABLE IF NOT EXISTS ');
+    printTableBody(path, printer);
+};

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

@@ -0,0 +1,2 @@
+CREATE TABLE IF NOT EXISTS t (
+id INT)

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

@@ -0,0 +1,10 @@
+[
+    createTableIfNotExists(
+        t, [
+            column(
+                id,
+                INT,
+            ),
+        ],
+    ),
+];

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

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

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

@@ -0,0 +1,3 @@
+SELECT *
+FROM t
+CROSS JOIN u

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

@@ -0,0 +1 @@
+[select("*", from(t, crossJoin(u)))];

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

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

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

@@ -0,0 +1 @@
+id INT DEFAULT 0

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

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

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

@@ -1,13 +1,11 @@
 export const deleteFrom = (path, {write, traverse}) => {
-    const args = path.get('arguments');
-    const [fromArg] = args;
-    const [, whereArg] = args;
+    const [fromArg, ...rest] = path.get('arguments');
     
     write('DELETE ');
     traverse(fromArg);
     
-    if (whereArg) {
+    for (const arg of rest) {
         write.newline();
-        traverse(whereArg);
+        traverse(arg);
     }
 };

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

@@ -0,0 +1,3 @@
+DELETE FROM t
+WHERE id = 1
+RETURNING id

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

@@ -0,0 +1,13 @@
+[
+    deleteFrom(
+        from(
+            t,
+        ),
+        where(
+            id === 1,
+        ),
+        returning(
+            id,
+        ),
+    ),
+];

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

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

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

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

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

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

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

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

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

@@ -0,0 +1,2 @@
+SELECT DISTINCT name
+FROM users

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

@@ -0,0 +1,2 @@
+SELECT DISTINCT name, age
+FROM users

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

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

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

@@ -0,0 +1,10 @@
+[
+    select(
+        distinct(
+            name,
+        ),
+        from(
+            users,
+        ),
+    ),
+];

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

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

package/lib/printer/visitors/call-expression/clauses/drop-index/drop-index.js

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

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

@@ -0,0 +1 @@
+DROP INDEX idx

package/lib/printer/visitors/call-expression/clauses/drop-index/fixture/drop-index-multi-fix.sql

@@ -0,0 +1 @@
+DROP INDEX idx1, idx2

package/lib/printer/visitors/call-expression/clauses/drop-index/fixture/drop-index-multi.js

@@ -0,0 +1 @@
+[dropIndex(idx1, idx2)];

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

@@ -0,0 +1 @@
+[dropIndex(idx)];

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

@@ -0,0 +1,25 @@
+import {types} from '@putout/babel';
+
+const {isCallExpression} = types;
+
+const isIfExists = (arg) => isCallExpression(arg.node) && arg.node.callee.name === 'ifExists';
+
+export const dropTable = (path, {write, traverse}) => {
+    const args = path.get('arguments');
+    
+    write('DROP TABLE ');
+    
+    let tableArgs = args;
+    
+    if (args.length > 0 && isIfExists(args[0])) {
+        write('IF EXISTS ');
+        tableArgs = args.slice(1);
+    }
+    
+    for (const [i, arg] of tableArgs.entries()) {
+        if (i > 0)
+            write(', ');
+        
+        traverse(arg);
+    }
+};

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

@@ -0,0 +1 @@
+DROP TABLE users

package/lib/printer/visitors/call-expression/clauses/drop-table/fixture/drop-table-if-exists-fix.sql

@@ -0,0 +1 @@
+DROP TABLE IF EXISTS users

package/lib/printer/visitors/call-expression/clauses/drop-table/fixture/drop-table-if-exists.js

@@ -0,0 +1 @@
+[dropTable(ifExists(), users)];

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

@@ -0,0 +1 @@
+DROP TABLE a, b

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

@@ -0,0 +1 @@
+[dropTable(a, b)];

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

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

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

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

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

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

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

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

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

@@ -0,0 +1,9 @@
+export const extract = (path, {write, traverse}) => {
+    const [unit, col] = path.get('arguments');
+    
+    write('EXTRACT(');
+    traverse(unit);
+    write(' FROM ');
+    traverse(col);
+    write(')');
+};

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

@@ -0,0 +1,2 @@
+SELECT EXTRACT(YEAR FROM created_at)
+FROM t

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

@@ -0,0 +1 @@
+[select(extract(YEAR, created_at), from(t))];

package/lib/printer/visitors/call-expression/clauses/from/fixture/generate-series-fix.sql

@@ -0,0 +1,2 @@
+SELECT *
+FROM generate_series(1, 10, 2)

package/lib/printer/visitors/call-expression/clauses/from/fixture/generate-series.js

@@ -0,0 +1,3 @@
+[
+    select('*', from(generateSeries(1, 10, 2))),
+];

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

@@ -1,6 +1,9 @@
 import {types} from '@putout/babel';
 
-const {isTSAsExpression} = types;
+const {
+    isTSAsExpression,
+    isCallExpression,
+} = types;
 
 const printAsExpression = (path, {write, traverse}) => {
     traverse(path.get('expression'));
@@ -11,15 +14,61 @@ const CLAUSE_NAMES = [
     'where',
     'groupBy',
     'having',
+    'orderBy',
+    'limit',
 ];
 
-const isClause = ({node}) => node.type === 'CallExpression' && CLAUSE_NAMES.includes(node.callee?.name);
+const TABLE_ARG_NAMES = [
+    ...CLAUSE_NAMES,
+    'join',
+    'leftJoin',
+    'rightJoin',
+    'innerJoin',
+    'leftOuterJoin',
+    'fullOuterJoin',
+    'naturalJoin',
+    'crossJoin',
+    'on',
+    'subquery',
+    'lateralJoin',
+    'lateralSubquery',
+];
+
+const isClause = ({node}) => isCallExpression(node) && CLAUSE_NAMES.includes(node.callee.name);
+
+const CAMEL_RE = /([A-Z])/g;
+const toSnake = (a) => a.replace(CAMEL_RE, (_, c) => `_${c.toLowerCase()}`);
+
+const isTableFunc = (node) => isCallExpression(node) && !TABLE_ARG_NAMES.includes(node.callee.name);
 
-const printArg = (path, ctx) => {
-    if (isTSAsExpression(path.node))
-        return printAsExpression(path, ctx);
+const printTableFunc = (path, {write, traverse}) => {
+    const {name} = path.node.callee;
+    const args = path.get('arguments');
+    
+    write(toSnake(name));
+    write('(');
+    
+    for (const [i, arg] of args.entries()) {
+        if (i > 0)
+            write(', ');
+        
+        traverse(arg);
+    }
+    
+    write(')');
+};
+
+const printArg = (path, printer) => {
+    const {node} = path;
+    const {traverse} = printer;
+    
+    if (isTSAsExpression(node))
+        return printAsExpression(path, printer);
+    
+    if (isTableFunc(node))
+        return printTableFunc(path, printer);
     
-    ctx.traverse(path);
+    traverse(path);
 };
 
 export const from = (path, {write, traverse}) => {

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

@@ -0,0 +1,3 @@
+SELECT *
+FROM a
+FULL OUTER JOIN b ON a.id = b.id

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

@@ -0,0 +1 @@
+[select('*', from(a, fullOuterJoin(b, on(a.id === b.id))))];

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/group-by/fixture/group-by-sum-fix.sql

@@ -0,0 +1,4 @@
+SELECT dept, SUM(salary)
+FROM emp
+GROUP BY dept
+HAVING SUM(salary) > 1000

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

@@ -0,0 +1,19 @@
+[
+    select(
+        dept,
+        sum(
+            salary,
+        ),
+        from(
+            emp,
+            groupBy(
+                dept,
+            ),
+            having(
+                sum(
+                    salary,
+                ) > 1000,
+            ),
+        ),
+    ),
+];

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]);
+};