typesql-cli 0.18.3 → 0.18.5

package/postgres-query-analyzer/traverse.js

@@ -2,21 +2,23 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.defaultOptions = defaultOptions;
 exports.traverseSmt = traverseSmt;
+exports.parseReturnType = parseReturnType;
 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");
+const postgres_1 = require("@wsporto/typesql-parser/postgres");
+const case_nullability_checker_1 = require("./case-nullability-checker");
 function defaultOptions() {
     return {
         collectNestedInfo: false,
         collectDynamicQueryInfo: false
     };
 }
-function traverseSmt(stmt, dbSchema, checkConstraints, options) {
+function traverseSmt(stmt, dbSchema, checkConstraints, userFunctions, options) {
     const { collectNestedInfo = false, collectDynamicQueryInfo = false } = options;
     const traverseResult = {
         columnsNullability: [],
-        parameters: [],
-        singleRow: false
+        parameters: []
     };
     if (collectNestedInfo) {
         traverseResult.relations = [];
@@ -35,7 +37,10 @@ function traverseSmt(stmt, dbSchema, checkConstraints, options) {
         collectNestedInfo,
         collectDynamicQueryInfo,
         fromColumns: [],
-        checkConstraints
+        parentColumns: [],
+        withColumns: [],
+        checkConstraints,
+        userFunctions
     };
     const selectstmt = stmt.selectstmt();
     if (selectstmt) {
@@ -82,15 +87,15 @@ function getInParameterList(ctx) {
 }
 function traverseSelectstmt(selectstmt, context, traverseResult) {
     const paramIsListResult = getInParameterList(selectstmt);
-    const columns = traverse_selectstmt(selectstmt, context, traverseResult);
+    const selectResult = traverse_selectstmt(selectstmt, context, traverseResult);
     //select parameters are collected after from paramters
     traverseResult.parameters.sort((param1, param2) => param1.paramIndex - param2.paramIndex);
-    const multipleRowsResult = !isSingleRowResult(selectstmt, context.dbSchema);
+    const multipleRowsResult = !(selectResult.singleRow || isSingleRowResult(selectstmt, selectResult.columns));
     const limit = checkLimit(selectstmt);
     const postgresTraverseResult = {
         queryType: 'Select',
         multipleRowsResult,
-        columns,
+        columns: selectResult.columns,
         parametersNullability: traverseResult.parameters.map(param => (Object.assign({ isNotNull: param.isNotNull }, addConstraintIfNotNull(param.checkConstraint)))),
         parameterList: paramIsListResult,
         limit
@@ -108,22 +113,23 @@ function traverse_selectstmt(selectstmt, context, traverseResult) {
     if (select_no_parens) {
         return traverse_select_no_parens(select_no_parens, context, traverseResult);
     }
-    return [];
+    //select_with_parens
+    return {
+        columns: [],
+        singleRow: true
+    };
 }
 function traverse_select_no_parens(select_no_parens, context, traverseResult) {
-    let withColumns = [];
     const with_clause = select_no_parens.with_clause();
     if (with_clause) {
         with_clause.cte_list().common_table_expr_list()
             .forEach(common_table_expr => {
-            const newContext = Object.assign(Object.assign({}, context), { fromColumns: withColumns.concat(context.fromColumns) });
-            const withResult = traverse_common_table_expr(common_table_expr, newContext, traverseResult);
-            withColumns.push(...withResult);
+            const withResult = traverse_common_table_expr(common_table_expr, context, traverseResult);
+            context.withColumns.push(...withResult);
         });
     }
     const select_clause = select_no_parens.select_clause();
-    const newContext = Object.assign(Object.assign({}, context), { fromColumns: withColumns.concat(context.fromColumns) });
-    const selectResult = traverse_select_clause(select_clause, newContext, traverseResult);
+    const selectResult = traverse_select_clause(select_clause, context, traverseResult);
     const select_limit = select_no_parens.select_limit();
     if (select_limit) {
         const numParamsBefore = traverseResult.parameters.length;
@@ -159,8 +165,8 @@ function traverse_common_table_expr(common_table_expr, context, traverseResult)
     const tableName = common_table_expr.name().getText();
     const select_stmt = common_table_expr.preparablestmt().selectstmt();
     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 })));
+    const selectResult = traverse_selectstmt(select_stmt, Object.assign(Object.assign({}, context), { collectDynamicQueryInfo: false }), traverseResult);
+    const columnsWithTalbeName = selectResult.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({
@@ -173,45 +179,55 @@ function traverse_common_table_expr(common_table_expr, context, traverseResult)
 }
 function traverse_select_clause(select_clause, context, traverseResult) {
     const simple_select_intersect_list = select_clause.simple_select_intersect_list();
-    let selectColumns = [];
-    if (simple_select_intersect_list) {
-        selectColumns = traverse_simple_select_intersect(simple_select_intersect_list[0], context, traverseResult);
-    }
+    const mainSelectResult = traverse_simple_select_intersect(simple_select_intersect_list[0], context, traverseResult);
+    let columns = mainSelectResult.columns;
     //union
     for (let index = 1; index < simple_select_intersect_list.length; index++) {
-        const unionNotNull = traverse_simple_select_intersect(simple_select_intersect_list[index], context, traverseResult);
-        selectColumns = selectColumns.map((value, columnIndex) => {
-            const col = Object.assign({ column_name: value.column_name, is_nullable: value.is_nullable || unionNotNull[columnIndex].is_nullable, table_name: '', table_schema: '', type: value.type }, (value.column_default && { column_default: value.column_default }));
+        const unionResult = traverse_simple_select_intersect(simple_select_intersect_list[index], context, traverseResult);
+        columns = columns.map((value, columnIndex) => {
+            const col = Object.assign({ column_name: value.column_name, is_nullable: value.is_nullable || unionResult.columns[columnIndex].is_nullable, table_name: '', table_schema: '', type: value.type }, (value.column_default && { column_default: value.column_default }));
             return col;
         });
     }
-    return selectColumns;
+    return {
+        columns,
+        singleRow: simple_select_intersect_list.length == 1 ? mainSelectResult.singleRow : false
+    };
 }
 function traverse_simple_select_intersect(simple_select_intersect, context, traverseResult) {
     const simple_select_pramary = simple_select_intersect.simple_select_pramary_list()[0];
     if (simple_select_pramary) {
         return traverse_simple_select_pramary(simple_select_pramary, context, traverseResult);
     }
-    return [];
+    return {
+        columns: [],
+        singleRow: true
+    };
 }
 function traverse_simple_select_pramary(simple_select_pramary, context, traverseResult) {
     var _a, _b, _c, _d, _e;
-    const fromColumns = [];
+    let fromResult = {
+        columns: [],
+        singleRow: false
+    };
     const from_clause = simple_select_pramary.from_clause();
     if (from_clause) {
         const where_clause = simple_select_pramary.where_clause();
-        const fields = traverse_from_clause(from_clause, context, traverseResult);
-        const fieldsNotNull = where_clause != null ? fields.map(field => checkIsNullable(where_clause, field)) : fields;
-        fromColumns.push(...fieldsNotNull);
+        fromResult = traverse_from_clause(from_clause, context, traverseResult);
+        fromResult.columns = where_clause != null ? fromResult.columns.map(field => checkIsNullable(where_clause, field)) : fromResult.columns;
     }
     const values_clause = simple_select_pramary.values_clause();
     if (values_clause) {
         const valuesColumns = traverse_values_clause(values_clause, context, traverseResult);
-        return valuesColumns;
+        return {
+            columns: valuesColumns,
+            singleRow: false
+        };
     }
     const where_a_expr = (_a = simple_select_pramary.where_clause()) === null || _a === void 0 ? void 0 : _a.a_expr();
     //fromColumns has precedence
-    const newContext = Object.assign(Object.assign({}, context), { fromColumns: fromColumns.concat(context.fromColumns) });
+    //context.fromColumns only becase of insert. update
+    const newContext = Object.assign(Object.assign({}, context), { fromColumns: [...context.fromColumns, ...fromResult.columns, ...context.parentColumns] });
     if (where_a_expr) {
         const numParamsBefore = traverseResult.parameters.length;
         traverse_a_expr(where_a_expr, newContext, traverseResult);
@@ -236,7 +252,10 @@ function traverse_simple_select_pramary(simple_select_pramary, context, traverse
         traverse_a_expr(having_expr, newContext, traverseResult);
     }
     const filteredColumns = filterColumns_simple_select_pramary(simple_select_pramary, newContext, traverseResult);
-    return filteredColumns;
+    return {
+        columns: filteredColumns,
+        singleRow: fromResult.singleRow
+    };
 }
 function extractRelations(a_expr) {
     const columnsRef = collectContextsOfType(a_expr, PostgreSQLParser_1.ColumnrefContext);
@@ -340,7 +359,7 @@ function traverse_target_el(target_el, context, traverseResult) {
                 parameters
             });
         }
-        return Object.assign({ column_name: alias || exprResult.column_name, is_nullable: exprResult.is_nullable && exprResult.column_default !== true, table_name: exprResult.table_name, table_schema: exprResult.table_schema, type: exprResult.type }, (exprResult.jsonType != null && { jsonType: exprResult.jsonType }));
+        return Object.assign(Object.assign(Object.assign({ column_name: alias || exprResult.column_name, is_nullable: exprResult.is_nullable, table_name: exprResult.table_name, table_schema: exprResult.table_schema, type: exprResult.type }, (exprResult.column_default != null) && { column_default: exprResult.column_default }), (exprResult.column_key != null && { column_key: exprResult.column_key })), (exprResult.jsonType != null && { jsonType: exprResult.jsonType }));
     }
     throw Error('Column not found');
 }
@@ -470,6 +489,7 @@ function traverse_expr_is_not(a_expr_is_not, context, traverseResult) {
         const result = traverse_expr_compare(a_expr_compare, context, traverseResult);
         if (a_expr_is_not.IS() && a_expr_is_not.NULL_P()) {
             checkParamterNullability(result, traverseResult);
+            return Object.assign(Object.assign({}, result), { is_nullable: false });
         }
         return result;
     }
@@ -518,10 +538,10 @@ function traverse_expr_compare(a_expr_compare, context, traverseResult) {
         const result = traverse_select_with_parens(select_with_parens, context, traverseResult);
         return {
             column_name: '?column?',
-            is_nullable: result.some(col => col.is_nullable),
+            is_nullable: result.columns.some(col => col.is_nullable),
             table_name: '',
             table_schema: '',
-            type: result[0].type
+            type: result.columns[0].type
         };
     }
     const a_expr = a_expr_compare.a_expr();
@@ -710,7 +730,7 @@ function getNameAndTypeIdFromAExprConst(a_expr_const) {
     if (a_expr_const.NULL_P()) {
         return {
             name: a_expr_const.getText(),
-            type: 'unknown'
+            type: 'null'
         };
     }
     return {
@@ -742,12 +762,15 @@ function traversec_expr(c_expr, context, traverseResult) {
         const columnref = c_expr.columnref();
         if (columnref) {
             if (context.columnRefIsRecord) {
+                const table = (0, select_columns_1.splitTableName)(columnref.getText());
+                const columns = filterColumns(context.fromColumns, table);
                 return {
                     column_name: columnref.getText(),
                     is_nullable: false,
                     table_name: '',
                     table_schema: '',
-                    type: 'unknown',
+                    type: 'record',
+                    recordTypes: columns
                 };
             }
             else {
@@ -782,6 +805,7 @@ function traversec_expr(c_expr, context, traverseResult) {
         }
         const fun_expr = c_expr.func_expr();
         if (fun_expr) {
+            context.columnRefIsRecord = undefined;
             const func_application = fun_expr.func_application();
             if (func_application) {
                 if (is_json_build_object_func(func_application)) {
@@ -807,14 +831,16 @@ function traversec_expr(c_expr, context, traverseResult) {
         }
         const select_with_parens = c_expr.select_with_parens();
         if (select_with_parens) {
-            const result = traverse_select_with_parens(select_with_parens, context, traverseResult);
+            const result = traverse_select_with_parens(select_with_parens, Object.assign(Object.assign({}, context), { parentColumns: context.fromColumns, fromColumns: [] }), traverseResult);
+            const jsonType = result.columns[0].jsonType;
+            const is_nullable = jsonType == null ? true : jsonType.name !== 'json[]' && jsonType.name !== 'json_map'; //json[], json_map are not nullable
             return {
                 column_name: '?column?',
-                is_nullable: true,
+                is_nullable,
                 table_name: '',
                 table_schema: '',
-                type: result[0].type,
-                jsonType: result[0].jsonType
+                type: result.columns[0].type,
+                jsonType: jsonType != null && jsonType.name === 'json' ? Object.assign(Object.assign({}, jsonType), { notNull: !is_nullable }) : jsonType
             };
         }
         const a_expr_in_parens = c_expr._a_expr_in_parens;
@@ -830,8 +856,8 @@ function traversec_expr(c_expr, context, traverseResult) {
                 is_nullable: expr_list.some(col => col.is_nullable),
                 table_name: '',
                 table_schema: '',
-                type: 'unknown',
-                recordTypes: expr_list.map(rec => ({ type: rec.type, notNull: !rec.is_nullable }))
+                type: 'record',
+                recordTypes: expr_list.map(expr => (Object.assign(Object.assign({}, expr), { column_name: '' })))
             };
         }
         const implicit_row = c_expr.implicit_row();
@@ -867,7 +893,7 @@ function traversec_expr(c_expr, context, traverseResult) {
 function filterColumns(fromColumns, fieldName) {
     return fromColumns.filter(col => (fieldName.prefix === '' || col.table_name === fieldName.prefix)
         && (fieldName.name === '*' || col.column_name === fieldName.name)).map(col => {
-        const result = Object.assign({ column_name: col.column_name, is_nullable: col.is_nullable, table_name: col.table_name, table_schema: col.table_schema, type: col.type }, (col.original_is_nullable !== undefined && { original_is_nullable: col.original_is_nullable }));
+        const result = Object.assign(Object.assign(Object.assign({ column_name: col.column_name, is_nullable: col.is_nullable, table_name: col.table_name, table_schema: col.table_schema, type: col.type }, (col.column_key !== undefined) && { column_key: col.column_key }), (col.jsonType !== undefined) && { jsonType: col.jsonType }), (col.original_is_nullable !== undefined && { original_is_nullable: col.original_is_nullable }));
         return result;
     });
 }
@@ -879,36 +905,62 @@ function excludeColumns(fromColumns, excludeList) {
     });
 }
 function traversec_expr_case(c_expr_case, context, traverseResult) {
-    var _a, _b;
+    var _a;
     const case_expr = c_expr_case.case_expr();
-    const whenResult = case_expr.when_clause_list().when_clause_list().map(when_clause => traversewhen_clause(when_clause, context, traverseResult));
-    const whenIsNotNull = whenResult.every(when => when);
-    const elseExpr = (_a = case_expr.case_default()) === null || _a === void 0 ? void 0 : _a.a_expr();
-    const elseIsNotNull = elseExpr ? !traverse_a_expr(elseExpr, context, traverseResult).is_nullable : false;
+    const whenClauseList = case_expr.when_clause_list().when_clause_list();
+    const whenResult = whenClauseList.map(when_clause => traversewhen_clause(when_clause, context, traverseResult));
+    const whenIsNotNull = whenResult.every(when => !when.is_nullable);
+    const elseExpr = case_expr.case_default();
+    const elseResult = elseExpr ? traverse_a_expr(elseExpr.a_expr(), Object.assign({}, context), traverseResult) : null;
+    const branchNotNull = elseResult ? evaluateBranches(whenClauseList, elseExpr, whenResult.map(col => !col.is_nullable)) : false;
+    const elseIsNotNull = (elseResult === null || elseResult === void 0 ? void 0 : elseResult.is_nullable) === false || branchNotNull;
     const notNull = elseIsNotNull && whenIsNotNull;
     return {
         column_name: '?column?',
         is_nullable: !notNull,
         table_name: '',
         table_schema: '',
-        type: (_b = whenResult[0].type) !== null && _b !== void 0 ? _b : 'unknown'
+        type: (_a = whenResult[0].type) !== null && _a !== void 0 ? _a : 'unknown',
+        jsonType: allJsonTypesMatch(whenResult, elseResult) ? whenResult[0].jsonType : undefined
     };
 }
+function evaluateBranches(whenClauseList, elseExpr, whenIsNotNull) {
+    const exprIndex = whenClauseList.findIndex(when => {
+        const whenExpr = when.a_expr(0);
+        const evaluatesIfNull = (0, case_nullability_checker_1.evaluatesTrueIfNull)(elseExpr, whenExpr);
+        if (evaluatesIfNull) {
+            return true;
+        }
+        return false;
+    });
+    const result = exprIndex !== -1 ? whenIsNotNull[exprIndex] : false;
+    return result;
+}
+function allJsonTypesMatch(whenResultList, elseResult) {
+    var _a;
+    const firstType = (_a = whenResultList[0]) === null || _a === void 0 ? void 0 : _a.jsonType;
+    const allMatch = whenResultList.every(res => {
+        const match = res.jsonType == firstType;
+        return match;
+    }) && ((elseResult === null || elseResult === void 0 ? void 0 : elseResult.type) === 'null' || (elseResult === null || elseResult === void 0 ? void 0 : elseResult.jsonType) == firstType);
+    return allMatch;
+}
 function traversewhen_clause(when_clause, context, traverseResult) {
     const a_expr_list = when_clause.a_expr_list();
     const [whenExprList, thenExprList] = partition(a_expr_list, (index) => index % 2 === 0);
     const whenExprResult = thenExprList.map((thenExpr, index) => {
         traverse_a_expr(whenExprList[index], context, traverseResult);
-        const thenExprResult = traverse_a_expr(thenExpr, context, traverseResult);
+        const thenExprResult = traverse_a_expr(thenExpr, Object.assign(Object.assign({}, context), { filter_expr: whenExprList[index] }), traverseResult);
         return thenExprResult;
     });
-    const notNull = whenExprResult.every(res => res);
+    const notNull = whenExprResult.every(res => !res.is_nullable);
     return {
         column_name: '?column?',
         is_nullable: !notNull,
         table_name: '',
         table_schema: '',
-        type: whenExprResult[0].type
+        type: whenExprResult[0].type,
+        jsonType: whenExprResult[0].jsonType
     };
 }
 function partition(array, predicate) {
@@ -952,33 +1004,31 @@ function mapJsonBuildArgsToJsonProperty(args, filterExpr) {
     return properties;
 }
 function inferJsonNullability(columns, filterExpr) {
-    const someIsNotNull = columns.some(col => filterExpr && col.original_is_nullable === false && isNotNull_a_expr(col, filterExpr));
+    const tables = columns.filter(col => filterExpr && col.original_is_nullable === false
+        && isNotNull_a_expr({ name: col.column_name, prefix: col.table_name }, filterExpr))
+        .map(col => col.table_name);
     const fields = columns.map(col => {
-        if (someIsNotNull) {
-            return col.original_is_nullable != null ? !col.original_is_nullable : Boolean(filterExpr && isNotNull_a_expr(col, filterExpr));
-        }
-        else {
-            return !col.is_nullable || Boolean(filterExpr && isNotNull_a_expr(col, filterExpr));
-        }
+        return col.original_is_nullable != null && tables.includes(col.table_name) ? !col.original_is_nullable : !col.is_nullable;
     });
     return fields;
 }
 function transformFieldsToJsonObjType(fields) {
     const jsonObject = {
         name: 'json',
+        notNull: true,
         properties: fields.map(col => mapFieldToPropertyDef(col))
     };
     return jsonObject;
 }
 function mapFieldToPropertyDef(field) {
     const prop = {
-        key: field.name,
+        key: field.column_name,
         type: transformFieldToJsonField(field)
     };
     return prop;
 }
 function transformFieldToJsonField(field) {
-    const jsonField = { name: 'json_field', type: field.type, notNull: field.notNull };
+    const jsonField = field.jsonType ? field.jsonType : { name: 'json_field', type: field.type, notNull: !field.is_nullable };
     return jsonField;
 }
 function traverse_json_build_obj_func(func_application, context, traverseResult) {
@@ -986,36 +1036,54 @@ function traverse_json_build_obj_func(func_application, context, traverseResult)
     const columnName = ((_a = func_application.func_name()) === null || _a === void 0 ? void 0 : _a.getText()) || func_application.getText();
     const func_arg_expr_list = ((_b = func_application.func_arg_list()) === null || _b === void 0 ? void 0 : _b.func_arg_expr_list()) || [];
     const argsResult = func_arg_expr_list.map(func_arg_expr => traversefunc_arg_expr(func_arg_expr, context, traverseResult));
-    return {
+    const result = {
         column_name: columnName,
-        is_nullable: true,
+        is_nullable: false,
         table_name: '',
         table_schema: '',
         type: 'json',
         jsonType: {
             name: 'json',
+            notNull: true,
             properties: mapJsonBuildArgsToJsonProperty(argsResult, context.filter_expr),
         }
     };
+    return result;
 }
 function traverse_json_agg(func_application, context, traverseResult) {
     var _a, _b;
     const columnName = ((_a = func_application.func_name()) === null || _a === void 0 ? void 0 : _a.getText()) || func_application.getText();
     const func_arg_expr_list = ((_b = func_application.func_arg_list()) === null || _b === void 0 ? void 0 : _b.func_arg_expr_list()) || [];
-    const argsResult = func_arg_expr_list.map(func_arg_expr => traversefunc_arg_expr(func_arg_expr, context, traverseResult));
+    const argsResult = func_arg_expr_list.map(func_arg_expr => traversefunc_arg_expr(func_arg_expr, Object.assign(Object.assign({}, context), { columnRefIsRecord: true }), traverseResult));
     const result = {
         column_name: columnName,
         is_nullable: context.filter_expr != null,
         table_name: '',
         table_schema: '',
         type: 'json[]',
-        jsonType: {
-            name: 'json[]',
-            properties: argsResult.map(arg => arg.jsonType || { name: 'json_field', type: arg.type, notNull: !arg.is_nullable })
-        }
+        jsonType: createJsonTypeForJsonAgg(argsResult[0], context.filter_expr)
     };
     return result;
 }
+function createJsonTypeForJsonAgg(arg, filter_expr) {
+    if (arg.recordTypes) {
+        const jsonType = mapRecordsToJsonType(arg.recordTypes, filter_expr);
+        return {
+            name: 'json[]',
+            properties: [jsonType]
+        };
+    }
+    const jsonType = arg.jsonType ? { name: 'json[]', properties: [arg.jsonType] } : undefined;
+    const fieldType = { name: 'json[]', properties: [{ name: 'json_field', type: arg.type, notNull: !arg.is_nullable }] };
+    const result = jsonType || fieldType;
+    return result;
+}
+function mapRecordsToJsonType(recordTypes, filterExpr) {
+    const jsonNullability = inferJsonNullability(recordTypes, filterExpr);
+    const fields = recordTypes.map((col, index) => (Object.assign(Object.assign({}, col), { column_name: col.column_name ? col.column_name : `f${index + 1}`, is_nullable: !jsonNullability[index] })));
+    const jsonType = transformFieldsToJsonObjType(fields);
+    return jsonType;
+}
 function traversefunc_application(func_application, context, traverseResult) {
     var _a;
     const functionName = getFunctionName(func_application);
@@ -1023,22 +1091,7 @@ function traversefunc_application(func_application, context, traverseResult) {
     if (functionName === 'row_to_json') {
         const argResult = traversefunc_arg_expr(func_arg_expr_list[0], Object.assign(Object.assign({}, context), { columnRefIsRecord: true }), traverseResult);
         if (argResult.recordTypes) {
-            const fields = argResult.recordTypes.map((record, index) => ({ name: `f${index + 1}`, type: record.type, notNull: record.notNull }));
-            const jsonType = transformFieldsToJsonObjType(fields);
-            return {
-                column_name: functionName,
-                is_nullable: false,
-                table_name: '',
-                table_schema: '',
-                type: 'json',
-                jsonType
-            };
-        }
-        else {
-            const columns = filterColumns(context.fromColumns, { name: '*', prefix: argResult.column_name });
-            const jsonNullability = inferJsonNullability(columns, context.filter_expr);
-            const fields = columns.map((col, index) => ({ name: col.column_name, type: col.type, notNull: jsonNullability[index] }));
-            const jsonType = transformFieldsToJsonObjType(fields);
+            const jsonType = mapRecordsToJsonType(argResult.recordTypes, context.filter_expr);
             return {
                 column_name: functionName,
                 is_nullable: false,
@@ -1170,6 +1223,28 @@ function traversefunc_application(func_application, context, traverseResult) {
     if (functionName === 'sum') {
         return Object.assign(Object.assign({}, argsResult[0]), { column_name: functionName, is_nullable: true, type: argsResult[0].type ? mapSumType(argsResult[0].type) : 'unknown' });
     }
+    if (functionName === 'json_object_agg') {
+        return {
+            column_name: functionName,
+            is_nullable: false,
+            table_name: '',
+            table_schema: '',
+            type: 'json',
+            jsonType: {
+                name: 'json_map',
+                type: argsResult[1].jsonType || { name: 'json_field', type: argsResult[1].type, notNull: !argsResult[1].is_nullable }
+            }
+        };
+    }
+    if (functionName === 'unnest') {
+        return {
+            column_name: functionName,
+            is_nullable: argsResult[0].is_nullable,
+            type: argsResult[0].type,
+            table_name: '',
+            table_schema: ''
+        };
+    }
     return {
         column_name: functionName,
         is_nullable: true,
@@ -1229,7 +1304,7 @@ function traverse_array_expr(array_expr, context, traverseResult) {
     const expr_list = array_expr.expr_list();
     if (expr_list) {
         const traverse_expr_list_result = traverse_expr_list(expr_list, context, traverseResult);
-        return Object.assign(Object.assign({}, traverse_expr_list_result[0]), { column_name: '?column?', table_name: '', table_schema: '' });
+        return Object.assign(Object.assign({}, traverse_expr_list_result[0]), { is_nullable: traverse_expr_list_result.some(expr => expr.is_nullable), column_name: '?column?', table_name: '', table_schema: '' });
     }
     const array_expr_list = array_expr.array_expr_list();
     if (array_expr_list) {
@@ -1251,44 +1326,92 @@ function traverse_array_expr_list(array_expr_list, context, traverseResult) {
     return result;
 }
 function findColumn(fieldName, fromColumns) {
-    const col = fromColumns.find(col => (fieldName.prefix === '' || col.table_name.toLowerCase() === fieldName.prefix.toLowerCase()) && col.column_name.toLowerCase() === fieldName.name.toLowerCase());
+    const col = findColumnOrNull(fieldName, fromColumns);
     if (col == null) {
         throw Error('Column not found: ' + fieldNameToString(fieldName));
     }
     return col;
 }
+function findColumnOrNull(fieldName, fromColumns) {
+    const col = fromColumns.find(col => (fieldName.prefix === '' || col.table_name.toLowerCase() === fieldName.prefix.toLowerCase()) && col.column_name.toLowerCase() === fieldName.name.toLowerCase());
+    return col;
+}
 function fieldNameToString(fieldName) {
     return fieldName.prefix !== '' ? `${fieldName.prefix}.${fieldName.name}` : fieldName.name;
 }
 function checkIsNullable(where_clause, field) {
-    const isNotNullResult = !field.is_nullable || isNotNull(field, where_clause);
+    const isNotNullResult = !field.is_nullable || isNotNull({ name: field.column_name, prefix: field.table_name }, where_clause);
     const col = Object.assign(Object.assign({}, field), { is_nullable: !isNotNullResult });
     return col;
 }
 function traverse_from_clause(from_clause, context, traverseResult) {
     const from_list = from_clause.from_list();
     if (from_list) {
-        return traverse_from_list(from_list, context, traverseResult);
+        const fromListResult = traverse_from_list(from_list, context, traverseResult);
+        return fromListResult;
     }
-    return [];
+    return {
+        columns: [],
+        singleRow: false
+    };
 }
 function traverse_from_list(from_list, context, traverseResult) {
-    const newColumns = from_list.table_ref_list().flatMap(table_ref => traverse_table_ref(table_ref, context, traverseResult));
-    return newColumns;
+    const fromListResult = from_list.table_ref_list().map(table_ref => traverse_table_ref(table_ref, context, traverseResult));
+    const columns = fromListResult.flatMap(tableRes => tableRes.columns);
+    return {
+        columns: columns,
+        singleRow: fromListResult.length === 1 ? fromListResult[0].singleRow : false
+    };
+}
+function getFromColumns(tableName, withColumns, dbSchema) {
+    const filteredWithColumns = withColumns.filter(col => col.table_name.toLowerCase() === tableName.name.toLowerCase());
+    const filteredSchema = filteredWithColumns.length > 0 ? filteredWithColumns : dbSchema.filter(col => col.table_name.toLowerCase() === tableName.name.toLowerCase());
+    return filteredSchema;
+}
+function checkLeftJoinIsNullable(leftJoin, subsequentJoins, fromColumns) {
+    if (!leftJoin.joinQual) {
+        return true; // No condition = always nullable
+    }
+    const leftTable = getTableName(leftJoin.tableRef);
+    const leftJoinColumns = getJoinColumns(leftJoin.joinQual)
+        .filter(col => {
+        if (col.prefix === leftTable.name && col.prefix === leftTable.alias) {
+            return true;
+        }
+        if (findColumnOrNull(col, fromColumns)) { //column without table reference (ex: fk_user)
+            return true;
+        }
+        return false;
+    });
+    for (const join of subsequentJoins) {
+        // Only INNER JOINs can cancel nullability
+        if (!(join.joinType === null || join.joinType.INNER_P())) {
+            continue;
+        }
+        const joinConditionColumns = getJoinColumns(join.joinQual);
+        const filteredJoinColumnsFromLeft = joinConditionColumns.filter(col => leftJoinColumns.some(lc => lc.prefix === col.prefix));
+        // Are *all* columns from the left join used in not-null filtering?
+        const referencesLeftJoin = filteredJoinColumnsFromLeft.length > 0 &&
+            filteredJoinColumnsFromLeft.every(col => { var _a; return isNotNull_a_expr(col, (_a = join.joinQual) === null || _a === void 0 ? void 0 : _a.a_expr()); });
+        if (referencesLeftJoin) {
+            return false; // LEFT JOIN is effectively filtered by INNER JOIN — not nullable
+        }
+    }
+    return true; // No INNER JOIN filtered it — remains nullable
 }
 function traverse_table_ref(table_ref, context, traverseResult) {
-    var _a, _b, _c, _d;
-    const { fromColumns, dbSchema } = context;
+    var _a, _b, _c, _d, _e, _f, _g;
     const allColumns = [];
     const relation_expr = table_ref.relation_expr();
     const aliasClause = table_ref.alias_clause();
     const aliasNameList = (_b = (_a = aliasClause === null || aliasClause === void 0 ? void 0 : aliasClause.name_list()) === null || _a === void 0 ? void 0 : _a.name_list()) === null || _b === void 0 ? void 0 : _b.map(name => name.getText());
     const alias = aliasClause ? aliasClause.colid().getText() : '';
     if (relation_expr) {
-        const tableName = traverse_relation_expr(relation_expr, dbSchema);
+        const tableName = traverse_relation_expr(relation_expr);
+        const fromColumns = getFromColumns(tableName, context.withColumns, context.dbSchema);
         const tableNameWithAlias = alias ? alias : tableName.name;
-        const fromColumnsResult = fromColumns.concat(dbSchema).filter(col => col.table_name.toLowerCase() === tableName.name.toLowerCase())
-            .map(col => (Object.assign(Object.assign({}, col), { table_name: tableNameWithAlias.toLowerCase() })));
+        const columnsWithAlias = fromColumns.map(col => (Object.assign(Object.assign({}, col), { table_name: tableNameWithAlias.toLowerCase() })));
+        const fromColumnsResult = columnsWithAlias.concat(context.parentColumns);
         allColumns.push(...fromColumnsResult);
         if (context.collectNestedInfo) {
             const key = fromColumnsResult.filter(col => col.column_key === 'PRI');
@@ -1304,47 +1427,79 @@ function traverse_table_ref(table_ref, context, traverseResult) {
             };
             (_d = traverseResult.relations) === null || _d === void 0 ? void 0 : _d.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 (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, original_is_nullable: col.is_nullable }))) : 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 joinList = extractJoins(table_ref);
+        const joinColumns = joinList.flatMap((join, index) => {
+            const joinType = join.joinType; //INNER, LEFT
+            const joinQual = join.joinQual;
+            const numParamsBefore = traverseResult.parameters.length;
+            const parentColumns = join.tableRef.LATERAL_P() ? fromColumnsResult : [];
+            const joinTableRefResult = traverse_table_ref(join.tableRef, Object.assign(Object.assign({}, context), { parentColumns }), traverseResult);
+            const isLeftJoin = joinType === null || joinType === void 0 ? void 0 : joinType.LEFT();
+            const filteredColumns = joinQual && (joinQual === null || joinQual === void 0 ? void 0 : joinQual.USING()) ? filterUsingColumns(joinTableRefResult.columns, joinQual) : joinTableRefResult.columns;
+            const subsequentJoints = joinList.slice(index + 1);
+            const checkIsNullable = isLeftJoin ? checkLeftJoinIsNullable(join, subsequentJoints, filteredColumns) : false;
+            ;
+            const resultColumns = isLeftJoin ? filteredColumns.map(col => {
+                const colResult = Object.assign(Object.assign({}, col), { is_nullable: checkIsNullable ? true : col.is_nullable, original_is_nullable: col.is_nullable });
+                return colResult;
+            }) : filteredColumns;
+            if (context.collectNestedInfo && joinQual) {
+                collectNestedInfo(joinQual, resultColumns, traverseResult);
+            }
+            if (context.collectDynamicQueryInfo) {
+                const parameters = traverseResult.parameters.slice(numParamsBefore).map((_, index) => index + numParamsBefore);
+                collectDynamicQueryInfoTableRef(join.tableRef, joinType, joinQual, resultColumns, parameters, traverseResult);
+            }
+            return resultColumns;
+        });
+        allColumns.push(...joinColumns);
+    }
+    const func_table = table_ref.func_table();
+    if (func_table) {
+        const funcAlias = ((_g = (_f = (_e = table_ref.func_alias_clause()) === null || _e === void 0 ? void 0 : _e.alias_clause()) === null || _f === void 0 ? void 0 : _f.colid()) === null || _g === void 0 ? void 0 : _g.getText()) || '';
+        const result = traverse_func_table(func_table, context, traverseResult);
+        const resultWithAlias = result.columns.map(col => (Object.assign(Object.assign({}, col), { table_name: funcAlias || col.table_name })));
+        return {
+            columns: resultWithAlias,
+            singleRow: result.singleRow
+        };
     }
     const select_with_parens = table_ref.select_with_parens();
     if (select_with_parens) {
         const columns = traverse_select_with_parens(select_with_parens, Object.assign(Object.assign({}, context), { collectDynamicQueryInfo: false }), traverseResult);
-        const withAlias = columns.map((col, i) => ({
-            column_name: (aliasNameList === null || aliasNameList === void 0 ? void 0 : aliasNameList[i]) || col.column_name,
-            is_nullable: col.is_nullable,
-            table_name: alias || col.table_name,
-            table_schema: col.table_schema,
-            type: col.type
-        }));
-        return withAlias;
+        const withAlias = columns.columns.map((col, i) => (Object.assign(Object.assign({}, col), { column_name: (aliasNameList === null || aliasNameList === void 0 ? void 0 : aliasNameList[i]) || col.column_name, table_name: alias || col.table_name })));
+        return {
+            columns: withAlias,
+            singleRow: false
+        };
+    }
+    return {
+        columns: allColumns,
+        singleRow: false
+    };
+}
+function extractJoins(table_ref) {
+    const joinList = [];
+    let currentJoinType = null;
+    for (const child of table_ref.children || []) {
+        if (child instanceof PostgreSQLParser_1.Join_typeContext) {
+            currentJoinType = child;
+        }
+        if (child instanceof PostgreSQLParser_1.Table_refContext) {
+            joinList.push({ joinQual: null, joinType: currentJoinType, tableRef: child });
+            currentJoinType = null;
+        }
+        if (child instanceof PostgreSQLParser_1.Join_qualContext) {
+            const lastJoin = joinList.at(-1);
+            if (lastJoin) {
+                lastJoin.joinQual = child;
+            }
+        }
     }
-    return allColumns;
+    return joinList;
 }
 function collectDynamicQueryInfoTableRef(table_ref, joinType, joinQual, columns, parameters, traverseResult) {
     var _a, _b;
@@ -1414,16 +1569,88 @@ function traverse_select_with_parens(select_with_parens, context, traverseResult
     if (select_no_parens) {
         return traverse_select_no_parens(select_no_parens, context, traverseResult);
     }
-    return [];
+    return {
+        columns: [],
+        singleRow: false
+    };
+}
+function traverse_func_table(func_table, context, traverseResult) {
+    const func_expr_windowless = func_table.func_expr_windowless();
+    if (func_expr_windowless) {
+        const result = traverse_func_expr_windowless(func_expr_windowless, context, traverseResult);
+        return result;
+    }
+    throw Error('Stmt not supported: ' + func_table.getText());
+}
+function parseReturnType(returnType) {
+    const trimmed = returnType.trim();
+    // Handle TABLE(...)
+    if (trimmed.toLowerCase().startsWith('table(') && trimmed.endsWith(')')) {
+        const inside = trimmed.slice(6, -1); // remove "TABLE(" and final ")"
+        const columns = [];
+        const parts = inside.split(',').map(part => part.trim());
+        for (const part of parts) {
+            const match = part.match(/^(\w+)\s+(.+)$/);
+            if (!match) {
+                throw new Error(`Invalid column definition: ${part}`);
+            }
+            const [, name, type] = match;
+            columns.push({ name, type });
+        }
+        return { kind: 'table', columns };
+    }
+    // Handle SETOF typename
+    const setofMatch = trimmed.match(/^SETOF\s+(\w+)$/i);
+    if (setofMatch) {
+        return { kind: 'setof', table: setofMatch[1] };
+    }
+    throw new Error(`Unsupported return type format: ${returnType}`);
+}
+function traverse_func_expr_windowless(func_expr_windowless, context, traverseResult) {
+    const func_application = func_expr_windowless.func_application();
+    if (func_application) {
+        const func_name = func_application.func_name().getText().toLowerCase();
+        const funcSchema = context.userFunctions.find(func => func.function_name.toLowerCase() === func_name);
+        if (funcSchema) {
+            const definition = funcSchema.definition;
+            const returnType = parseReturnType(funcSchema.return_type);
+            const functionColumns = returnType.kind === 'table' ? returnType.columns.map(col => {
+                const columnInfo = {
+                    column_name: col.name,
+                    type: col.type,
+                    is_nullable: true,
+                    table_name: '',
+                    table_schema: ''
+                };
+                return columnInfo;
+            }) : context.dbSchema.filter(col => col.table_name.toLowerCase() === returnType.table);
+            if (funcSchema.language.toLowerCase() === 'sql') {
+                const parser = (0, postgres_1.parseSql)(definition);
+                const selectstmt = parser.stmt().selectstmt();
+                const { columns, multipleRowsResult } = traverseSelectstmt(selectstmt, context, traverseResult);
+                return {
+                    columns: columns.map((c) => (Object.assign(Object.assign({}, c), { table_name: funcSchema.function_name }))),
+                    singleRow: !multipleRowsResult
+                };
+            }
+            else {
+                return {
+                    columns: functionColumns,
+                    singleRow: false
+                };
+            }
+        }
+    }
+    throw Error('Stmt not supported: ' + func_expr_windowless.getText());
 }
-function traverse_relation_expr(relation_expr, dbSchema) {
+function traverse_relation_expr(relation_expr) {
     const qualified_name = relation_expr.qualified_name();
-    const name = traverse_qualified_name(qualified_name, dbSchema);
+    const name = traverse_qualified_name(qualified_name);
     return name;
 }
-function traverse_qualified_name(qualified_name, dbSchema) {
+function traverse_qualified_name(qualified_name) {
     var _a, _b;
-    const colid_name = qualified_name.colid() ? get_colid_text(qualified_name.colid(), dbSchema) : '';
+    const colid_name = qualified_name.colid() ? get_colid_text(qualified_name.colid()) : '';
     const indirection_el_list = (_a = qualified_name.indirection()) === null || _a === void 0 ? void 0 : _a.indirection_el_list();
     if (indirection_el_list && indirection_el_list.length === 1) {
         return {
@@ -1436,22 +1663,22 @@ function traverse_qualified_name(qualified_name, dbSchema) {
         alias: ''
     };
 }
-function get_colid_text(colid, dbSchema) {
+function get_colid_text(colid) {
     const identifier = colid.identifier();
     if (identifier) {
-        return traverse_identifier(identifier, dbSchema);
+        return traverse_identifier(identifier);
     }
     const unreserved_keyword = colid.unreserved_keyword();
     if (unreserved_keyword) {
-        return traverse_unreserved_keyword(unreserved_keyword, dbSchema);
+        return traverse_unreserved_keyword(unreserved_keyword);
     }
     return '';
 }
-function traverse_identifier(identifier, dbSchema) {
+function traverse_identifier(identifier) {
     const tableName = identifier.Identifier().getText();
     return tableName;
 }
-function traverse_unreserved_keyword(unreserved_keyword, dbSchema) {
+function traverse_unreserved_keyword(unreserved_keyword) {
     return unreserved_keyword.getText();
 }
 function paramIsList(c_expr) {
@@ -1486,8 +1713,7 @@ function paramIsList(c_expr) {
 function traverseInsertstmt(insertstmt, dbSchema) {
     const traverseResult = {
         columnsNullability: [],
-        parameters: [],
-        singleRow: false
+        parameters: []
     };
     const insert_target = insertstmt.insert_target();
     const tableName = insert_target.getText();
@@ -1499,7 +1725,10 @@ function traverseInsertstmt(insertstmt, dbSchema) {
     const context = {
         dbSchema,
         fromColumns: insertColumns,
+        parentColumns: [],
+        withColumns: [],
         checkConstraints: {},
+        userFunctions: [],
         collectNestedInfo: false,
         collectDynamicQueryInfo: false
     };
@@ -1535,7 +1764,10 @@ function traverseDeletestmt(deleteStmt, dbSchema, traverseResult) {
     const context = {
         dbSchema,
         fromColumns: deleteColumns,
+        parentColumns: [],
+        withColumns: [],
         checkConstraints: {},
+        userFunctions: [],
         collectNestedInfo: false,
         collectDynamicQueryInfo: false
     };
@@ -1557,7 +1789,7 @@ function traverseDeletestmt(deleteStmt, dbSchema, traverseResult) {
     return result;
 }
 function addConstraintIfNotNull(checkConstraint) {
-    return checkConstraint !== undefined ? { checkConstraint } : {};
+    return checkConstraint !== undefined ? { checkConstraint } : undefined;
 }
 function traverseUpdatestmt(updatestmt, traverseContext, traverseResult) {
     var _a;
@@ -1570,12 +1802,12 @@ function traverseUpdatestmt(updatestmt, traverseContext, traverseResult) {
     updatestmt.set_clause_list().set_clause_list()
         .forEach(set_clause => traverse_set_clause(set_clause, context, traverseResult));
     const from_clause = updatestmt.from_clause();
-    const fromColumns = from_clause ? traverse_from_clause(from_clause, context, traverseResult) : [];
+    const fromResult = from_clause ? traverse_from_clause(from_clause, context, traverseResult) : { columns: [], singleRow: true };
     const parametersBefore = traverseResult.parameters.length;
     const where_clause = updatestmt.where_or_current_clause();
     if (where_clause) {
         const a_expr = where_clause.a_expr();
-        traverse_a_expr(a_expr, Object.assign(Object.assign({}, context), { fromColumns: updateColumns.concat(fromColumns) }), traverseResult);
+        traverse_a_expr(a_expr, Object.assign(Object.assign({}, context), { fromColumns: updateColumns.concat(fromResult.columns) }), traverseResult);
     }
     const whereParameters = traverseResult.parameters.slice(parametersBefore);
     const returning_clause = updatestmt.returning_clause();
@@ -1690,7 +1922,14 @@ function isNotNull_a_expr_unary_not(a_expr_unary_not, field) {
 function isNotNull_a_expr_isnull(a_expr_isnull, field) {
     const a_expr_is_not = a_expr_isnull.a_expr_is_not();
     if (a_expr_is_not) {
-        return isNotNull_a_expr_is_not(a_expr_is_not, field);
+        const isNotNull = isNotNull_a_expr_is_not(a_expr_is_not, field);
+        if (isNotNull && a_expr_is_not.IS() && a_expr_is_not.NOT() && a_expr_is_not.NULL_P()) {
+            return true;
+        }
+        if (a_expr_is_not.IS() && a_expr_is_not.NULL_P()) {
+            return false;
+        }
+        return isNotNull;
     }
     return false;
 }
@@ -1814,7 +2053,7 @@ function isNotNull_c_expr(c_expr, field) {
         const columnref = c_expr.columnref();
         if (columnref) {
             const fieldName = (0, select_columns_1.splitName)(columnref.getText());
-            return (fieldName.name === field.column_name && (fieldName.prefix === '' || field.table_name === fieldName.prefix));
+            return (fieldName.name === field.name && (fieldName.prefix === '' || field.prefix === fieldName.prefix));
         }
         const aexprconst = c_expr.aexprconst();
         if (aexprconst) {
@@ -1840,8 +2079,8 @@ function isParameter(str) {
     const paramPattern = /^\$[0-9]+(::[a-zA-Z_][a-zA-Z0-9_]*)?$/;
     return paramPattern.test(str);
 }
-function isSingleRowResult(selectstmt, dbSchema) {
-    var _a, _b;
+function isSingleRowResult(selectstmt, fromColumns) {
+    var _a;
     const limit = checkLimit(selectstmt);
     if (limit === 1) {
         return true;
@@ -1855,11 +2094,11 @@ function isSingleRowResult(selectstmt, dbSchema) {
     const simple_select_pramary = simple_select_pramary_list[0];
     const from_clause = simple_select_pramary.from_clause();
     if (!from_clause) {
-        const hasSetReturningFunction = (_b = simple_select_pramary.target_list_()) === null || _b === void 0 ? void 0 : _b.target_list().target_el_list().some(target_el => isSetReturningFunction_target_el(target_el));
+        const hasSetReturningFunction = hasFunction(simple_select_pramary, fun => isSetReturningFunction(fun));
         return !hasSetReturningFunction;
     }
     if (!simple_select_pramary.group_clause()) {
-        const agreegateFunction = hasAggregateFunction(simple_select_pramary);
+        const agreegateFunction = hasFunction(simple_select_pramary, fun => isAggregateFunction(fun));
         if (agreegateFunction) {
             return true;
         }
@@ -1873,36 +2112,38 @@ function isSingleRowResult(selectstmt, dbSchema) {
     }
     const where_clause = simple_select_pramary.where_clause();
     if (where_clause) {
-        const tableName = getTableName(table_ref_list[0]);
-        const uniqueKeys = dbSchema.filter(col => col.table_name.toLowerCase() === tableName.name.toLowerCase()
-            && (col.column_key === 'PRI' || col.column_key === 'UNI'))
+        const uniqueKeys = fromColumns.filter(col => col.column_key === 'PRI' || col.column_key === 'UNI')
             .map(col => col.column_name);
         return isSingleRowResult_where(where_clause.a_expr(), uniqueKeys);
     }
     return false;
 }
-function hasAggregateFunction(simple_select_pramary) {
+function hasFunction(simple_select_pramary, checkFunction) {
     const target_list_ = simple_select_pramary.target_list_();
     if (target_list_) {
-        return target_list_.target_list().target_el_list().some(target_el => isAggregateFunction_target_el(target_el));
+        return target_list_.target_list().target_el_list().some(target_el => checkFunction_target_el(target_el, checkFunction));
     }
     const target_list = simple_select_pramary.target_list();
-    return target_list.target_el_list().some(target_el => isAggregateFunction_target_el(target_el));
+    if (target_list) {
+        return target_list.target_el_list().some(target_el => checkFunction_target_el(target_el, checkFunction));
+    }
+    return false;
 }
 function getTableName(table_ref) {
     const relation_expr = table_ref.relation_expr();
-    const tableName = relation_expr.qualified_name().getText();
+    if (relation_expr) {
+        return traverse_relation_expr(relation_expr);
+    }
     const aliasClause = table_ref.alias_clause();
-    const tableAlias = aliasClause ? aliasClause.colid().getText() : '';
     return {
-        name: tableName,
-        alias: tableAlias
+        name: aliasClause.colid().getText(),
+        alias: ''
     };
 }
-function isAggregateFunction_target_el(target_el) {
+function checkFunction_target_el(target_el, checkFunction) {
     if (target_el instanceof PostgreSQLParser_1.Target_labelContext) {
         const c_expr_list = collectContextsOfType(target_el, PostgreSQLParser_1.Func_exprContext, false);
-        const aggrFunction = c_expr_list.some(func_expr => isAggregateFunction_c_expr(func_expr));
+        const aggrFunction = c_expr_list.some(func_expr => checkFunction(func_expr));
         return aggrFunction;
     }
     return false;
@@ -1955,25 +2196,98 @@ const aggregateFunctions = new Set([
     'variance',
     'xmlagg'
 ]);
-function isAggregateFunction_c_expr(func_expr) {
+function isAggregateFunction(func_expr) {
     var _a, _b, _c;
     const funcName = (_c = (_b = (_a = func_expr === null || func_expr === void 0 ? void 0 : func_expr.func_application()) === null || _a === void 0 ? void 0 : _a.func_name()) === null || _b === void 0 ? void 0 : _b.getText()) === null || _c === void 0 ? void 0 : _c.toLowerCase();
     // RANK(), DENSE_RANK(), PERCENT_RANK() - they are window functions, not aggregate functions, 
     // even though your query is returning them from a join involving pg_aggregate.
     return !func_expr.over_clause() && aggregateFunctions.has(funcName);
 }
-function isSetReturningFunction_target_el(target_el) {
-    if (target_el instanceof PostgreSQLParser_1.Target_labelContext) {
-        const c_expr_list = collectContextsOfType(target_el, PostgreSQLParser_1.Func_exprContext);
-        const setReturningFunction = c_expr_list.some(func_expr => isSetReturningFunction_c_expr(func_expr));
-        return setReturningFunction;
-    }
-    return false;
-}
-function isSetReturningFunction_c_expr(func_expr) {
+// SELECT distinct '''' || p.proname || ''',' AS function_name
+// FROM pg_proc p
+// JOIN pg_namespace n ON p.pronamespace = n.oid
+// WHERE p.proretset = true
+// ORDER BY function_name;
+function isSetReturningFunction(func_expr) {
     var _a, _b, _c;
     const funcName = (_c = (_b = (_a = func_expr === null || func_expr === void 0 ? void 0 : func_expr.func_application()) === null || _a === void 0 ? void 0 : _a.func_name()) === null || _b === void 0 ? void 0 : _b.getText()) === null || _c === void 0 ? void 0 : _c.toLowerCase();
-    return funcName === 'generate_series';
+    const setReturning = new Set([
+        '_pg_expandarray',
+        'aclexplode',
+        'generate_series',
+        'generate_subscripts',
+        'get_clients_with_addresses',
+        'get_mytable1',
+        'get_mytable1_by_id',
+        'get_mytable1_with_nested_function',
+        'get_mytable_plpgsql',
+        'get_users_with_posts',
+        'get_users_with_posts_plpgsql',
+        'json_array_elements',
+        'json_array_elements_text',
+        'json_each',
+        'json_each_text',
+        'json_object_keys',
+        'json_populate_recordset',
+        'json_to_recordset',
+        'jsonb_array_elements',
+        'jsonb_array_elements_text',
+        'jsonb_each',
+        'jsonb_each_text',
+        'jsonb_object_keys',
+        'jsonb_path_query',
+        'jsonb_populate_recordset',
+        'jsonb_to_recordset',
+        'pg_available_extension_versions',
+        'pg_available_extensions',
+        'pg_config',
+        'pg_cursor',
+        'pg_event_trigger_ddl_commands',
+        'pg_event_trigger_dropped_objects',
+        'pg_extension_update_paths',
+        'pg_get_keywords',
+        'pg_get_multixact_members',
+        'pg_get_publication_tables',
+        'pg_get_replication_slots',
+        'pg_hba_file_rules',
+        'pg_listening_channels',
+        'pg_lock_status',
+        'pg_logical_slot_get_binary_changes',
+        'pg_logical_slot_get_changes',
+        'pg_logical_slot_peek_binary_changes',
+        'pg_logical_slot_peek_changes',
+        'pg_ls_archive_statusdir',
+        'pg_ls_dir',
+        'pg_ls_logdir',
+        'pg_ls_tmpdir',
+        'pg_ls_waldir',
+        'pg_mcv_list_items',
+        'pg_options_to_table',
+        'pg_partition_ancestors',
+        'pg_partition_tree',
+        'pg_prepared_statement',
+        'pg_prepared_xact',
+        'pg_show_all_file_settings',
+        'pg_show_all_settings',
+        'pg_show_replication_origin_status',
+        'pg_stat_get_activity',
+        'pg_stat_get_backend_idset',
+        'pg_stat_get_progress_info',
+        'pg_stat_get_wal_senders',
+        'pg_stop_backup',
+        'pg_tablespace_databases',
+        'pg_timezone_abbrevs',
+        'pg_timezone_names',
+        'regexp_matches',
+        'regexp_split_to_table',
+        'ts_debug',
+        'ts_parse',
+        'ts_stat',
+        'ts_token_type',
+        'txid_snapshot_xip',
+        'unnest'
+    ]);
+    return setReturning.has(funcName);
 }
 function isSingleRowResult_where(a_expr, uniqueKeys) {
     var _a, _b;