typesql-cli 0.15.2 → 0.16.0-experimental.0

package/postgres-query-analyzer/traverse.js

@@ -1,21 +1,39 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.defaultOptions = defaultOptions;
 exports.traverseSmt = traverseSmt;
 const PostgreSQLParser_1 = require("@wsporto/typesql-parser/postgres/PostgreSQLParser");
 const typesql_parser_1 = require("@wsporto/typesql-parser");
 const select_columns_1 = require("../mysql-query-analyzer/select-columns");
-function traverseSmt(stmt, dbSchema, generateNestedInfo) {
+function defaultOptions() {
+    return {
+        collectNestedInfo: false,
+        collectDynamicQueryInfo: false
+    };
+}
+function traverseSmt(stmt, dbSchema, options) {
+    const { collectNestedInfo = false, collectDynamicQueryInfo = false } = options;
     const traverseResult = {
         columnsNullability: [],
         parameters: [],
         singleRow: false
     };
-    if (generateNestedInfo) {
+    if (collectNestedInfo) {
         traverseResult.relations = [];
     }
+    if (collectDynamicQueryInfo) {
+        const dynamicQueryInfo = {
+            with: [],
+            select: [],
+            from: [],
+            where: []
+        };
+        traverseResult.dynamicQueryInfo = dynamicQueryInfo;
+    }
     const traverseContext = {
         dbSchema,
-        generateNestedInfo,
+        collectNestedInfo,
+        collectDynamicQueryInfo,
         fromColumns: []
     };
     const selectstmt = stmt.selectstmt();
@@ -69,6 +87,9 @@ function traverseSelectstmt(selectstmt, context, traverseResult) {
     if (traverseResult.relations) {
         postgresTraverseResult.relations = traverseResult.relations;
     }
+    if (traverseResult.dynamicQueryInfo) {
+        postgresTraverseResult.dynamicQueryInfo = traverseResult.dynamicQueryInfo;
+    }
     return postgresTraverseResult;
 }
 function traverse_selectstmt(selectstmt, context, traverseResult) {
@@ -90,21 +111,54 @@ function traverse_select_no_parens(select_no_parens, context, traverseResult) {
         });
     }
     const select_clause = select_no_parens.select_clause();
-    if (select_clause) {
-        const newContext = Object.assign(Object.assign({}, context), { fromColumns: withColumns.concat(context.fromColumns) });
-        return traverse_select_clause(select_clause, newContext, traverseResult);
+    const newContext = Object.assign(Object.assign({}, context), { fromColumns: withColumns.concat(context.fromColumns) });
+    const selectResult = traverse_select_clause(select_clause, newContext, traverseResult);
+    const select_limit = select_no_parens.select_limit();
+    if (select_limit) {
+        const numParamsBefore = traverseResult.parameters.length;
+        const limit_clause = select_limit.limit_clause();
+        const limit_a_expr = limit_clause.select_limit_value().a_expr();
+        traverse_a_expr(limit_a_expr, context, traverseResult);
+        let fragment = '';
+        if (limit_clause) {
+            if (context.collectDynamicQueryInfo) {
+                fragment += (0, select_columns_1.extractOriginalSql)(limit_clause);
+            }
+        }
+        const offset_clause = select_limit.offset_clause();
+        if (offset_clause) {
+            const offset_a_expr = offset_clause.select_offset_value().a_expr();
+            traverse_a_expr(offset_a_expr, context, traverseResult);
+            if (context.collectDynamicQueryInfo) {
+                fragment += ' ' + (0, select_columns_1.extractOriginalSql)(offset_clause);
+            }
+        }
+        if (fragment) {
+            const parameters = traverseResult.parameters.slice(numParamsBefore).map((_, index) => index + numParamsBefore);
+            traverseResult.dynamicQueryInfo.limitOffset = {
+                fragment,
+                parameters
+            };
+        }
     }
-    return [];
+    return selectResult;
 }
 function traverse_common_table_expr(common_table_expr, context, traverseResult) {
+    var _a;
     const tableName = common_table_expr.name().getText();
     const select_stmt = common_table_expr.preparablestmt().selectstmt();
-    if (select_stmt) {
-        const columns = traverse_selectstmt(select_stmt, context, traverseResult);
-        const columnsWithTalbeName = columns.map(col => (Object.assign(Object.assign({}, col), { table_name: tableName })));
-        return columnsWithTalbeName;
+    const numParamsBefore = traverseResult.parameters.length;
+    const columns = traverse_selectstmt(select_stmt, Object.assign(Object.assign({}, context), { collectDynamicQueryInfo: false }), traverseResult);
+    const columnsWithTalbeName = columns.map(col => (Object.assign(Object.assign({}, col), { table_name: tableName })));
+    if (context.collectDynamicQueryInfo) {
+        const parameters = traverseResult.parameters.slice(numParamsBefore).map((_, index) => index + numParamsBefore);
+        (_a = traverseResult.dynamicQueryInfo) === null || _a === void 0 ? void 0 : _a.with.push({
+            fragment: (0, select_columns_1.extractOriginalSql)(common_table_expr),
+            relationName: tableName,
+            parameters
+        });
     }
-    return [];
+    return columnsWithTalbeName;
 }
 function traverse_select_clause(select_clause, context, traverseResult) {
     const simple_select_intersect_list = select_clause.simple_select_intersect_list();
@@ -135,7 +189,7 @@ function traverse_simple_select_intersect(simple_select_intersect, context, trav
     return [];
 }
 function traverse_simple_select_pramary(simple_select_pramary, context, traverseResult) {
-    var _a;
+    var _a, _b;
     const fromColumns = [];
     const from_clause = simple_select_pramary.from_clause();
     if (from_clause) {
@@ -153,11 +207,32 @@ function traverse_simple_select_pramary(simple_select_pramary, context, traverse
     //fromColumns has precedence
     const newContext = Object.assign(Object.assign({}, context), { fromColumns: fromColumns.concat(context.fromColumns) });
     if (where_a_expr) {
+        const numParamsBefore = traverseResult.parameters.length;
         traverse_a_expr(where_a_expr, newContext, traverseResult);
+        if (context.collectDynamicQueryInfo) {
+            const parameters = traverseResult.parameters.slice(numParamsBefore).map((_, index) => index + numParamsBefore);
+            const relations = extractRelations(where_a_expr);
+            (_b = traverseResult.dynamicQueryInfo) === null || _b === void 0 ? void 0 : _b.where.push({
+                fragment: `AND ${(0, select_columns_1.extractOriginalSql)(where_a_expr)}`,
+                parameters,
+                dependOnRelations: relations
+            });
+        }
     }
     const filteredColumns = filterColumns_simple_select_pramary(simple_select_pramary, newContext, traverseResult);
     return filteredColumns;
 }
+function extractRelations(a_expr) {
+    const columnsRef = collectContextsOfType(a_expr, PostgreSQLParser_1.ColumnrefContext);
+    const relations = columnsRef
+        .map((colRefExpr) => {
+        const colRef = colRefExpr;
+        const tableName = (0, select_columns_1.splitName)(colRef.getText());
+        return tableName;
+    });
+    const uniqueRelations = [...new Set(relations.map(relation => relation.prefix))];
+    return uniqueRelations;
+}
 function traverse_values_clause(values_clause, context, traverseResult) {
     const expr_list_list = values_clause.expr_list_list();
     if (expr_list_list) {
@@ -197,17 +272,29 @@ function traverse_target_list(target_list, context, traverseResult) {
         const fieldName = (0, select_columns_1.splitName)(target_el.getText());
         if (fieldName.name == '*') {
             const columns = filterColumns(context.fromColumns, fieldName);
+            if (context.collectDynamicQueryInfo) {
+                columns.forEach(col => {
+                    var _a;
+                    (_a = traverseResult.dynamicQueryInfo) === null || _a === void 0 ? void 0 : _a.select.push({
+                        fragment: `${col.table_name}.${col.column_name}`,
+                        fragmentWitoutAlias: `${col.table_name}.${col.column_name}`,
+                        dependOnRelations: [col.table_name],
+                        parameters: []
+                    });
+                });
+            }
             return columns;
         }
-        const column = isNotNull_target_el(target_el, context, traverseResult);
+        const column = traverse_target_el(target_el, context, traverseResult);
         return [column];
     });
     return columns;
 }
-function isNotNull_target_el(target_el, context, traverseResult) {
-    var _a;
+function traverse_target_el(target_el, context, traverseResult) {
+    var _a, _b;
     if (target_el instanceof PostgreSQLParser_1.Target_labelContext) {
         const a_expr = target_el.a_expr();
+        const numParamsBefore = traverseResult.parameters.length;
         const exprResult = traverse_a_expr(a_expr, context, traverseResult);
         const colLabel = target_el.colLabel();
         const alias = colLabel != null ? colLabel.getText() : '';
@@ -219,6 +306,16 @@ function isNotNull_target_el(target_el, context, traverseResult) {
                 }
             });
         }
+        if (context.collectDynamicQueryInfo) {
+            const parameters = traverseResult.parameters.slice(numParamsBefore).map((_, index) => index + numParamsBefore);
+            const relations = extractRelations(target_el.a_expr());
+            (_b = traverseResult.dynamicQueryInfo) === null || _b === void 0 ? void 0 : _b.select.push({
+                fragment: (0, select_columns_1.extractOriginalSql)(target_el),
+                fragmentWitoutAlias: (0, select_columns_1.extractOriginalSql)(target_el.a_expr()),
+                dependOnRelations: relations,
+                parameters
+            });
+        }
         return {
             column_name: alias || exprResult.column_name,
             is_nullable: exprResult.is_nullable,
@@ -256,11 +353,17 @@ function traverse_a_expr_lessless(a_expr_lessless, context, traverseResult) {
     throw Error('traverse_a_expr_lessless -  Not expected:' + a_expr_lessless.getText());
 }
 function traverse_expr_or(a_expr_or, context, traverseResult) {
-    const a_expr_and = a_expr_or.a_expr_and_list()[0];
-    if (a_expr_and) {
-        return traverse_expr_and(a_expr_and, context, traverseResult);
+    // expr1 OR expr2
+    const result = a_expr_or.a_expr_and_list().map(a_expr_and => traverse_expr_and(a_expr_and, context, traverseResult));
+    if (result.length === 1) {
+        return result[0];
     }
-    throw Error('traverse_expr_or -  Not expected:' + a_expr_or.getText());
+    return {
+        column_name: '?column?',
+        is_nullable: result.some(col => col.is_nullable),
+        table_name: '',
+        table_schema: ''
+    };
 }
 function traverse_expr_and(a_expr_and, context, traverseResult) {
     const result = a_expr_and.a_expr_between_list().map(a_expr_between => traverse_expr_between(a_expr_between, context, traverseResult));
@@ -480,11 +583,11 @@ function traversec_expr(c_expr, context, traverseResult) {
         if (c_expr.PARAM()) {
             traverseResult.parameters.push({
                 paramIndex: c_expr.start.start,
-                isNotNull: true
+                isNotNull: !context.propagatesNull
             });
             return {
                 column_name: c_expr.PARAM().getText(),
-                is_nullable: false,
+                is_nullable: !!context.propagatesNull,
                 table_name: '',
                 table_schema: ''
             };
@@ -523,6 +626,28 @@ function traversec_expr(c_expr, context, traverseResult) {
         if (a_expr_in_parens) {
             return traverse_a_expr(a_expr_in_parens, context, traverseResult);
         }
+        const explicit_row = c_expr.explicit_row();
+        if (explicit_row) {
+            const expr_list = explicit_row.expr_list().a_expr_list()
+                .map(a_expr => traverse_a_expr(a_expr, context, traverseResult));
+            return {
+                column_name: '?column?',
+                is_nullable: expr_list.some(col => col.is_nullable),
+                table_name: '',
+                table_schema: ''
+            };
+        }
+        const implicit_row = c_expr.implicit_row();
+        if (implicit_row) {
+            const expr_list = implicit_row.expr_list().a_expr_list().concat(implicit_row.a_expr())
+                .map(a_expr => traverse_a_expr(a_expr, context, traverseResult));
+            return {
+                column_name: '?column?',
+                is_nullable: expr_list.some(col => col.is_nullable),
+                table_name: '',
+                table_schema: ''
+            };
+        }
     }
     if (c_expr instanceof PostgreSQLParser_1.C_expr_caseContext) {
         const isNotNull = traversec_expr_case(c_expr, context, traverseResult);
@@ -614,20 +739,13 @@ function traversefunc_application(func_application, context, traverseResult) {
     if (functionName === 'generate_series') {
         return true;
     }
-    if (func_arg_expr_list) {
-        func_arg_expr_list.forEach(func_arg_expr => traversefunc_arg_expr(func_arg_expr, context, traverseResult));
-    }
     return false;
 }
 function traversefunc_expr_common_subexpr(func_expr_common_subexpr, context, traverseResult) {
-    if (func_expr_common_subexpr.COALESCE()) {
+    if (func_expr_common_subexpr.COALESCE() || func_expr_common_subexpr.GREATEST() || func_expr_common_subexpr.LEAST()) {
         const func_arg_list = func_expr_common_subexpr.expr_list().a_expr_list();
         const result = func_arg_list.map(func_arg_expr => {
-            const paramResult = traverse_a_expr(func_arg_expr, context, traverseResult);
-            if (isParameter(paramResult.column_name)) {
-                traverseResult.parameters[traverseResult.parameters.length - 1].isNotNull = false;
-                paramResult.is_nullable = true;
-            }
+            const paramResult = traverse_a_expr(func_arg_expr, Object.assign(Object.assign({}, context), { propagatesNull: true }), traverseResult);
             return paramResult;
         });
         return result.some(col => !col.is_nullable);
@@ -675,19 +793,19 @@ function traverse_table_ref(table_ref, context, traverseResult) {
     const allColumns = [];
     const relation_expr = table_ref.relation_expr();
     const aliasClause = table_ref.alias_clause();
-    const alias = aliasClause ? aliasClause.colid().getText() : undefined;
+    const alias = aliasClause ? aliasClause.colid().getText() : '';
     if (relation_expr) {
         const tableName = traverse_relation_expr(relation_expr, dbSchema);
         const tableNameWithAlias = alias ? alias : tableName.name;
         const fromColumnsResult = fromColumns.concat(dbSchema).filter(col => col.table_name === tableName.name)
             .map(col => (Object.assign(Object.assign({}, col), { table_name: tableNameWithAlias })));
         allColumns.push(...fromColumnsResult);
-        if (context.generateNestedInfo) {
+        if (context.collectNestedInfo) {
             const key = fromColumnsResult.filter(col => col.column_key === 'PRI');
             const renameAs = (aliasClause === null || aliasClause === void 0 ? void 0 : aliasClause.AS()) != null;
             const relation = {
                 name: tableName.name,
-                alias: alias || '',
+                alias: alias,
                 renameAs,
                 parentRelation: '',
                 joinColumn: ((_a = key[0]) === null || _a === void 0 ? void 0 : _a.column_name) || '',
@@ -696,61 +814,95 @@ function traverse_table_ref(table_ref, context, traverseResult) {
             };
             (_b = traverseResult.relations) === null || _b === void 0 ? void 0 : _b.push(relation);
         }
-    }
-    const table_ref_list = table_ref.table_ref_list();
-    const join_type_list = table_ref.join_type_list();
-    const join_qual_list = table_ref.join_qual_list();
-    if (table_ref_list) {
-        const joinColumns = table_ref_list.flatMap((table_ref, joinIndex) => {
-            const joinType = join_type_list[joinIndex]; //INNER, LEFT
-            const joinQual = join_qual_list[joinIndex];
-            const joinColumns = traverse_table_ref(table_ref, context, traverseResult);
-            const isUsing = (joinQual === null || joinQual === void 0 ? void 0 : joinQual.USING()) ? true : false;
-            const isLeftJoin = joinType === null || joinType === void 0 ? void 0 : joinType.LEFT();
-            const filteredColumns = isUsing ? filterUsingColumns(joinColumns, joinQual) : joinColumns;
-            const resultColumns = isLeftJoin ? filteredColumns.map(col => (Object.assign(Object.assign({}, col), { is_nullable: true }))) : filteredColumns;
-            if (context.generateNestedInfo) {
-                collectNestedInfo(joinQual, resultColumns, traverseResult);
-            }
-            return resultColumns;
-        });
-        allColumns.push(...joinColumns);
+        const table_ref_list = table_ref.table_ref_list();
+        const join_type_list = table_ref.join_type_list();
+        const join_qual_list = table_ref.join_qual_list();
+        if (context.collectDynamicQueryInfo && traverseResult.dynamicQueryInfo.from.length == 0) {
+            collectDynamicQueryInfoTableRef(table_ref, null, null, fromColumnsResult, [], traverseResult);
+        }
+        if (table_ref_list) {
+            const joinColumns = table_ref_list.flatMap((table_ref, joinIndex) => {
+                const joinType = join_type_list[joinIndex]; //INNER, LEFT
+                const joinQual = join_qual_list[joinIndex];
+                const numParamsBefore = traverseResult.parameters.length;
+                const joinColumns = traverse_table_ref(table_ref, context, traverseResult);
+                const isUsing = (joinQual === null || joinQual === void 0 ? void 0 : joinQual.USING()) ? true : false;
+                const isLeftJoin = joinType === null || joinType === void 0 ? void 0 : joinType.LEFT();
+                const filteredColumns = isUsing ? filterUsingColumns(joinColumns, joinQual) : joinColumns;
+                const resultColumns = isLeftJoin ? filteredColumns.map(col => (Object.assign(Object.assign({}, col), { is_nullable: true }))) : filteredColumns;
+                if (context.collectNestedInfo) {
+                    collectNestedInfo(joinQual, resultColumns, traverseResult);
+                }
+                if (context.collectDynamicQueryInfo) {
+                    const parameters = traverseResult.parameters.slice(numParamsBefore).map((_, index) => index + numParamsBefore);
+                    collectDynamicQueryInfoTableRef(table_ref, joinType, joinQual, resultColumns, parameters, traverseResult);
+                }
+                return resultColumns;
+            });
+            allColumns.push(...joinColumns);
+        }
     }
     const select_with_parens = table_ref.select_with_parens();
     if (select_with_parens) {
-        const columns = traverse_select_with_parens(select_with_parens, context, traverseResult);
+        const columns = traverse_select_with_parens(select_with_parens, Object.assign(Object.assign({}, context), { collectDynamicQueryInfo: false }), traverseResult);
         const withAlias = columns.map(col => (Object.assign(Object.assign({}, col), { table_name: alias || col.table_name })));
         return withAlias;
     }
     return allColumns;
 }
-function collectNestedInfo(joinQual, resultColumns, traverseResult) {
-    var _a;
+function collectDynamicQueryInfoTableRef(table_ref, joinType, joinQual, columns, parameters, traverseResult) {
+    var _a, _b;
+    const alias = table_ref.alias_clause() ? (0, select_columns_1.extractOriginalSql)(table_ref.alias_clause()) : '';
+    const fromExpr = (0, select_columns_1.extractOriginalSql)(table_ref.relation_expr() || table_ref.select_with_parens());
+    const tableName = ((_a = table_ref.relation_expr()) === null || _a === void 0 ? void 0 : _a.getText()) || alias;
+    const fromOrJoin = joinType ? `${(0, select_columns_1.extractOriginalSql)(joinType)} JOIN` : 'FROM';
+    const join = joinQual ? ` ${(0, select_columns_1.extractOriginalSql)(joinQual)}` : '';
+    const fromFragment = `${fromOrJoin} ${fromExpr} ${alias}${join}`;
+    const fields = columns.map(col => col.column_name);
+    const joinColumns = joinQual ? getJoinColumns(joinQual) : [];
+    const parentList = joinColumns.filter(joinRef => joinRef.prefix !== tableName && joinRef.prefix !== alias);
+    const parentRelation = parentList.length === 1 ? parentList[0].prefix : '';
+    (_b = traverseResult.dynamicQueryInfo) === null || _b === void 0 ? void 0 : _b.from.push({
+        fragment: fromFragment,
+        fields,
+        parameters,
+        relationName: tableName,
+        relationAlias: alias,
+        parentRelation
+    });
+}
+function getJoinColumns(joinQual) {
     const a_expr_or_list = joinQual ? collectContextsOfType(joinQual, PostgreSQLParser_1.A_expr_orContext) : [];
     if (a_expr_or_list.length == 1) {
         const a_expr_or = a_expr_or_list[0];
         const a_expr_and = a_expr_or.a_expr_and_list()[0];
         const columnref = collectContextsOfType(a_expr_and, PostgreSQLParser_1.ColumnrefContext);
         const joinColumns = columnref.map(colRef => (0, select_columns_1.splitName)(colRef.getText()));
-        const currentRelation = (_a = traverseResult.relations) === null || _a === void 0 ? void 0 : _a.at(-1);
-        joinColumns.forEach(joinRef => {
-            if (currentRelation) {
-                const joinColumn = resultColumns.filter(col => col.column_name === joinRef.name)[0];
-                const unique = joinColumn && (joinColumn.column_key === 'PRI' || joinColumn.column_key === 'UNI');
-                if (joinRef.prefix === currentRelation.name || joinRef.prefix === currentRelation.alias) {
-                    if (!unique) {
-                        currentRelation.cardinality = 'many';
-                    }
+        return joinColumns;
+    }
+    return [];
+}
+function collectNestedInfo(joinQual, resultColumns, traverseResult) {
+    var _a;
+    const joinColumns = getJoinColumns(joinQual);
+    const currentRelation = (_a = traverseResult.relations) === null || _a === void 0 ? void 0 : _a.at(-1);
+    joinColumns.forEach(joinRef => {
+        if (currentRelation) {
+            const joinColumn = resultColumns.filter(col => col.column_name === joinRef.name)[0];
+            const unique = joinColumn && (joinColumn.column_key === 'PRI' || joinColumn.column_key === 'UNI');
+            if (joinRef.prefix === currentRelation.name || joinRef.prefix === currentRelation.alias) {
+                if (!unique) {
+                    currentRelation.cardinality = 'many';
                 }
-                else {
-                    currentRelation.parentRelation = joinRef.prefix;
-                    if (!unique) {
-                        currentRelation.parentCardinality = 'many';
-                    }
+            }
+            else {
+                currentRelation.parentRelation = joinRef.prefix;
+                if (!unique) {
+                    currentRelation.parentCardinality = 'many';
                 }
             }
-        });
-    }
+        }
+    });
 }
 function filterUsingColumns(fromColumns, joinQual) {
     const excludeList = joinQual.name_list().name_list().map(name => (0, select_columns_1.splitName)(name.getText()));
@@ -851,7 +1003,8 @@ function traverseInsertstmt(insertstmt, dbSchema) {
     const context = {
         dbSchema,
         fromColumns: insertColumns,
-        generateNestedInfo: false
+        collectNestedInfo: false,
+        collectDynamicQueryInfo: false
     };
     const selectstmt = insert_rest.selectstmt();
     traverse_insert_select_stmt(selectstmt, Object.assign(Object.assign({}, context), { fromColumns: insertColumnsList }), traverseResult);
@@ -890,7 +1043,7 @@ function traverse_insert_select_stmt(selectstmt, context, traverseResult) {
                 const from_clause = simple_select_pramary.from_clause();
                 const fromColumns = from_clause ? traverse_from_clause(from_clause, Object.assign(Object.assign({}, context), { fromColumns: [] }), traverseResult) : [];
                 target_list.target_el_list().forEach((target_el, index) => {
-                    const targetResult = isNotNull_target_el(target_el, Object.assign(Object.assign({}, context), { fromColumns }), traverseResult);
+                    const targetResult = traverse_target_el(target_el, Object.assign(Object.assign({}, context), { fromColumns }), traverseResult);
                     if (isParameter(targetResult.column_name)) {
                         traverseResult.parameters.at(-1).isNotNull = !context.fromColumns[index].is_nullable;
                     }
@@ -915,7 +1068,8 @@ function traverseDeletestmt(deleteStmt, dbSchema, traverseResult) {
     const context = {
         dbSchema,
         fromColumns: deleteColumns,
-        generateNestedInfo: false
+        collectNestedInfo: false,
+        collectDynamicQueryInfo: false
     };
     const returninColumns = returning_clause ? traverse_target_list(returning_clause.target_list(), context, traverseResult) : [];
     const result = {
@@ -937,7 +1091,8 @@ function traverseUpdatestmt(updatestmt, dbSchema, traverseResult) {
     const context = {
         dbSchema,
         fromColumns: updateColumns,
-        generateNestedInfo: false
+        collectNestedInfo: false,
+        collectDynamicQueryInfo: false
     };
     updatestmt.set_clause_list().set_clause_list()
         .forEach(set_clause => traverse_set_clause(set_clause, context, traverseResult));