happy-sql 2.3.0 → 2.4.1

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

@@ -36,7 +36,9 @@ import {max} from './max/max.js';
 import {orderBy} from './order-by/order-by.js';
 import {desc} from './desc/desc.js';
 import {limit} from './limit/limit.js';
+import {namedWindow} from './named-window/named-window.js';
 import {distinct} from './distinct/distinct.js';
+import {distinctOn} from './distinct-on/distinct-on.js';
 import {notInList} from './not-in-list/not-in-list.js';
 import {between} from './between/between.js';
 import {notBetween} from './not-between/not-between.js';
@@ -52,7 +54,31 @@ import {lateralJoin} from './lateral-join/lateral-join.js';
 import {lateralSubquery} from './lateral-subquery/lateral-subquery.js';
 import {createIndex} from './create-index/create-index.js';
 import {createUniqueIndex} from './create-index/create-unique-index.js';
+import {concurrently} from './concurrently/concurrently.js';
+import {ifNotExists} from './if-not-exists/if-not-exists.js';
+import {explain} from './explain/explain.js';
+import {explainAnalyze} from './explain-analyze/explain-analyze.js';
+import {explainQueryPlan} from './explain-query-plan/explain-query-plan.js';
+import {setParam} from './set-param/set-param.js';
+import {showParam} from './show-param/show-param.js';
+import {createView} from './create-view/create-view.js';
+import {createMaterializedView} from './create-materialized-view/create-materialized-view.js';
+import {dropView} from './drop-view/drop-view.js';
+import {createExtension} from './create-extension/create-extension.js';
+import {createSchema} from './create-schema/create-schema.js';
+import {grant} from './grant/grant.js';
+import {privilege} from './privilege/privilege.js';
+import {grantOn} from './grant-on/grant-on.js';
+import {to} from './to/to.js';
+import {indexedBy} from './indexed-by/indexed-by.js';
+import {notIndexed} from './not-indexed/not-indexed.js';
+import {createVirtualTable} from './create-virtual-table/create-virtual-table.js';
+import {reindex} from './reindex/reindex.js';
+import {analyze} from './analyze/analyze.js';
+import {releaseSavepoint} from './release-savepoint/release-savepoint.js';
+import {doBlock} from './do-block/do-block.js';
 import {vacuum} from './vacuum/vacuum.js';
+import {truncate} from './truncate/truncate.js';
 import {row} from './row/row.js';
 import {notNull} from './not-null/not-null.js';
 import {default_} from './default/default.js';
@@ -65,10 +91,24 @@ import {leftJoin} from './left-join/left-join.js';
 import {rightJoin} from './right-join/right-join.js';
 import {innerJoin} from './inner-join/inner-join.js';
 import {leftOuterJoin} from './left-outer-join/left-outer-join.js';
+import {rightOuterJoin} from './right-outer-join/right-outer-join.js';
 import {crossJoin} from './cross-join/cross-join.js';
 import {createTable} from './create-table/create-table.js';
+import {
+    limitAll,
+    offset,
+    fetchFirst,
+    forUpdate,
+} from './limit-clauses.js';
+import {
+    createTempTable,
+    createTempTableIfNotExists,
+} from './create-temp-table/create-temp-table.js';
+import {withoutRowid} from './without-rowid/without-rowid.js';
+import {createTableAs} from './create-table-as/create-table-as.js';
 import {alterTable} from './alter-table/alter-table.js';
 import {addColumn} from './add-column/add-column.js';
+import {addConstraint} from './add-constraint/add-constraint.js';
 import {dropColumn} from './drop-column/drop-column.js';
 import {renameColumn} from './rename-column/rename-column.js';
 import {alterColumnType} from './alter-column-type/alter-column-type.js';
@@ -80,6 +120,7 @@ import {savepoint} from './savepoint/savepoint.js';
 import {createSequence} from './create-sequence/create-sequence.js';
 import {extract} from './extract/extract.js';
 import {concat} from './concat/concat.js';
+import {filter} from './filter/filter.js';
 import {array} from './array/array.js';
 import {over} from './over/over.js';
 import {partitionBy} from './partition-by/partition-by.js';
@@ -88,13 +129,14 @@ import {dropTable} from './drop-table/drop-table.js';
 import {dropIndex} from './drop-index/drop-index.js';
 import {pragma} from './pragma/pragma.js';
 import {attach} from './attach/attach.js';
+import {detach} from './detach/detach.js';
 import {column} from './column/column.js';
 import {autoIncrement} from './auto-increment/auto-increment.js';
 import {serial} from './serial/serial.js';
 import {primaryKey} from './primary-key/primary-key.js';
 import {lastInsertRowid} from './last-insert-rowid/last-insert-rowid.js';
 import {lastval} from './lastval/lastval.js';
-import {nextval} from './nextval/nextval.js';
+import {defaultNextval} from './default-nextval/default-nextval.js';
 import {like} from './like/like.js';
 import {isNull, isNotNull} from './is-null/is-null.js';
 import {orReplace} from './or-replace/or-replace.js';
@@ -104,6 +146,33 @@ import {
     identity,
     identityByDefault,
 } from './identity/identity.js';
+import {not} from './not/not.js';
+import {unaryMinus} from './unary-minus/unary-minus.js';
+import {arrayType} from './array-type/array-type.js';
+import {using} from './using/using.js';
+import {defaultValues} from './default-values/default-values.js';
+import {tablePrimaryKey} from './table-primary-key/table-primary-key.js';
+import {tableUnique} from './table-unique/table-unique.js';
+import {constraint} from './constraint/constraint.js';
+import {renameTable} from './rename-table/rename-table.js';
+import {
+    enableRowLevelSecurity,
+    disableRowLevelSecurity,
+} from './row-level-security/row-level-security.js';
+import {nullsFirst, nullsLast} from './nulls-order/nulls-order.js';
+import {orAbort} from './or-abort/or-abort.js';
+import {orFail} from './or-fail/or-fail.js';
+import {countDistinct} from './count-distinct/count-distinct.js';
+import {generatedAlwaysAs} from './generated-always-as/generated-always-as.js';
+import {accessMethod} from './access-method/access-method.js';
+import {onDelete, onUpdate} from './on-delete/on-delete.js';
+import {collate} from './collate/collate.js';
+import {
+    rowsBetween,
+    preceding,
+    following,
+    currentRow,
+} from './window-frame/window-frame.js';
 
 export const clauses = {
     join,
@@ -124,7 +193,31 @@ export const clauses = {
     lateralSubquery,
     createIndex,
     createUniqueIndex,
+    concurrently,
+    ifNotExists,
+    explain,
+    explainAnalyze,
+    explainQueryPlan,
+    setParam,
+    showParam,
+    createView,
+    createMaterializedView,
+    dropView,
+    createExtension,
+    createSchema,
+    grant,
+    privilege,
+    grantOn,
+    to,
+    indexedBy,
+    notIndexed,
+    createVirtualTable,
+    reindex,
+    analyze,
+    releaseSavepoint,
+    doBlock,
     vacuum,
+    truncate,
     row,
     notNull,
     default_,
@@ -155,7 +248,13 @@ export const clauses = {
     orderBy,
     desc,
     limit,
+    limitAll,
+    offset,
+    fetchFirst,
+    forUpdate,
+    namedWindow,
     distinct,
+    distinctOn,
     notInList,
     notBetween,
     between,
@@ -169,12 +268,18 @@ export const clauses = {
     rightJoin,
     innerJoin,
     leftOuterJoin,
+    rightOuterJoin,
     fullOuterJoin,
     naturalJoin,
     crossJoin,
     createTable,
+    createTempTable,
+    createTempTableIfNotExists,
+    withoutRowid,
+    createTableAs,
     alterTable,
     addColumn,
+    addConstraint,
     dropColumn,
     renameColumn,
     alterColumnType,
@@ -186,6 +291,7 @@ export const clauses = {
     createSequence,
     extract,
     concat,
+    filter,
     array,
     over,
     partitionBy,
@@ -194,21 +300,47 @@ export const clauses = {
     dropIndex,
     pragma,
     attach,
+    detach,
     column,
     autoIncrement,
     serial,
     primaryKey,
     lastInsertRowid,
     lastval,
-    nextval,
+    defaultNextval,
     like,
     notLike,
     ilike,
     similarTo,
     isDistinctFrom,
+    collate,
     isNull,
     isNotNull,
     inList,
     identity,
     identityByDefault,
+    not,
+    unaryMinus,
+    arrayType,
+    using,
+    defaultValues,
+    tablePrimaryKey,
+    tableUnique,
+    constraint,
+    renameTable,
+    enableRowLevelSecurity,
+    disableRowLevelSecurity,
+    nullsFirst,
+    nullsLast,
+    orAbort,
+    orFail,
+    countDistinct,
+    generatedAlwaysAs,
+    accessMethod,
+    onDelete,
+    onUpdate,
+    rowsBetween,
+    preceding,
+    following,
+    currentRow,
 };

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

@@ -0,0 +1,7 @@
+export const collate = (path, {write, traverse}) => {
+    const [left, right] = path.get('arguments');
+    
+    traverse(left);
+    write(' COLLATE ');
+    traverse(right);
+};

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

@@ -0,0 +1,2 @@
+// sentinel — consumed by createIndex/createUniqueIndex printer, never called
+export const concurrently = null;

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

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

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

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

package/lib/printer/visitors/call-expression/clauses/create-extension/create-extension.js

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

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

@@ -1,20 +1,54 @@
-export const createIndex = (path, {write, traverse}) => {
-    const args = path.get('arguments');
+import {types} from '@putout/babel';
+
+const {isCallExpression} = types;
+
+const isConcurrently = (arg) => isCallExpression(arg.node) && arg.node.callee.name === 'concurrently';
+const isIfNotExists = (arg) => isCallExpression(arg.node) && arg.node.callee.name === 'ifNotExists';
+const isWhere = (arg) => isCallExpression(arg.node) && arg.node.callee.name === 'where';
+const isAccessMethod = (arg) => isCallExpression(arg.node) && arg.node.callee.name === 'accessMethod';
+const isCol = (arg) => !isConcurrently(arg) && !isIfNotExists(arg) && !isWhere(arg) && !isAccessMethod(arg);
+
+export const printIndexBody = (path, {write, traverse}) => {
+    const [name, table, ...rest] = path.get('arguments');
+    const cols = rest.filter(isCol);
+    const hasIfNotExists = rest.some(isIfNotExists);
+    const hasConcurrently = rest.some(isConcurrently);
+    const whereArg = rest.find(isWhere);
+    const accessMethod = rest.find(isAccessMethod);
     
-    write('CREATE INDEX ');
-    traverse(args[0]);
+    if (hasIfNotExists)
+        write('IF NOT EXISTS ');
+    
+    if (hasConcurrently)
+        write('CONCURRENTLY ');
+    
+    traverse(name);
     write(' ON ');
-    traverse(args[1]);
+    traverse(table);
+    
+    if (accessMethod) {
+        write(' USING ');
+        traverse(accessMethod.get('arguments.0'));
+    }
+    
     write(' (');
     
-    for (const [i, arg] of args
-        .slice(2)
-        .entries()) {
+    for (const [i, col] of cols.entries()) {
         if (i > 0)
             write(', ');
         
-        traverse(arg);
+        traverse(col);
     }
     
     write(')');
+    
+    if (whereArg) {
+        write.newline();
+        traverse(whereArg);
+    }
+};
+
+export const createIndex = (path, printer) => {
+    printer.write('CREATE INDEX ');
+    printIndexBody(path, printer);
 };

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

@@ -1,20 +1,6 @@
-export const createUniqueIndex = (path, {write, traverse}) => {
-    const args = path.get('arguments');
-    
-    write('CREATE UNIQUE INDEX ');
-    traverse(args[0]);
-    write(' ON ');
-    traverse(args[1]);
-    write(' (');
-    
-    for (const [i, arg] of args
-        .slice(2)
-        .entries()) {
-        if (i > 0)
-            write(', ');
-        
-        traverse(arg);
-    }
-    
-    write(')');
+import {printIndexBody} from './create-index.js';
+
+export const createUniqueIndex = (path, printer) => {
+    printer.write('CREATE UNIQUE INDEX ');
+    printIndexBody(path, printer);
 };

package/lib/printer/visitors/call-expression/clauses/create-materialized-view/create-materialized-view.js

@@ -0,0 +1,8 @@
+export const createMaterializedView = (path, {write, traverse}) => {
+    const [name, stmt] = path.get('arguments');
+    
+    write('CREATE MATERIALIZED VIEW ');
+    traverse(name);
+    write(' AS ');
+    traverse(stmt);
+};

package/lib/printer/visitors/call-expression/clauses/create-schema/create-schema.js

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

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

@@ -1,3 +1,7 @@
+import {types} from '@putout/babel';
+
+const {isCallExpression} = types;
+
 export const printTableBody = (path, {write, traverse}) => {
     const [name, columns] = path.get('arguments');
     
@@ -7,9 +11,17 @@ export const printTableBody = (path, {write, traverse}) => {
     write.newline();
     
     const elements = columns.get('elements');
+    let wroteRowid = false;
     
     for (const [i, element] of elements.entries()) {
-        if (i > 0) {
+        const isRowid = isCallExpression(element.node) && element.node.callee.name === 'withoutRowid';
+        
+        if (isRowid) {
+            wroteRowid = true;
+            continue;
+        }
+        
+        if (i > 0 && !wroteRowid) {
             write(',');
             write.newline();
         }
@@ -18,6 +30,9 @@ export const printTableBody = (path, {write, traverse}) => {
     }
     
     write(')');
+    
+    if (wroteRowid)
+        write(' WITHOUT ROWID');
 };
 
 export const createTable = (path, printer) => {

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

@@ -0,0 +1,8 @@
+export const createTableAs = (path, {write, traverse}) => {
+    const [name, stmt] = path.get('arguments');
+    
+    write('CREATE TABLE ');
+    traverse(name);
+    write(' AS ');
+    traverse(stmt);
+};

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

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

package/lib/printer/visitors/call-expression/clauses/create-view/create-view.js

@@ -0,0 +1,8 @@
+export const createView = (path, {write, traverse}) => {
+    const [name, stmt] = path.get('arguments');
+    
+    write('CREATE VIEW ');
+    traverse(name);
+    write(' AS ');
+    traverse(stmt);
+};

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

@@ -0,0 +1,8 @@
+export const createVirtualTable = (path, {write, traverse}) => {
+    const [name, module] = path.get('arguments');
+    
+    write('CREATE VIRTUAL TABLE ');
+    traverse(name);
+    write(' USING ');
+    traverse(module);
+};

package/lib/printer/visitors/call-expression/clauses/{nextval/nextval.js → default-nextval/default-nextval.js}

@@ -1,4 +1,5 @@
-export const nextval = (path, {write}) => {
+export const defaultNextval = (path, {write}) => {
     const [seq] = path.node.arguments;
+    
     write(`DEFAULT nextval('${seq.name}')`);
 };

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

@@ -0,0 +1,3 @@
+export const defaultValues = (path, {write}) => {
+    write('DEFAULT VALUES');
+};

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

@@ -0,0 +1,6 @@
+export const detach = (path, {write, traverse}) => {
+    const [schema] = path.get('arguments');
+    
+    write('DETACH DATABASE ');
+    traverse(schema);
+};

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

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

package/lib/printer/visitors/call-expression/clauses/do-block/do-block.js

@@ -0,0 +1,5 @@
+export const doBlock = (path, {write}) => {
+    const [arg] = path.get('arguments');
+    
+    write(`DO $$${arg.node.value}$$`);
+};

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

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

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

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

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

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

package/lib/printer/visitors/call-expression/clauses/explain-analyze/explain-analyze.js

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

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

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

package/lib/printer/visitors/call-expression/clauses/fetch-first/fetch-first.js

@@ -0,0 +1,5 @@
+export const fetchFirst = (path, {write, traverse}) => {
+    write('FETCH FIRST ');
+    traverse(path.get('arguments')[0]);
+    write(' ROWS ONLY');
+};

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

@@ -0,0 +1,8 @@
+export const filter = (path, {write, traverse}) => {
+    const [fn, condition] = path.get('arguments');
+    
+    traverse(fn);
+    write(' FILTER (WHERE ');
+    traverse(condition);
+    write(')');
+};

package/lib/printer/visitors/call-expression/clauses/for-update/for-update.js

@@ -0,0 +1,3 @@
+export const forUpdate = (path, {write}) => {
+    write('FOR UPDATE');
+};