typesql-cli 0.9.6 → 0.10.1

package/sqlite-query-analyzer/traverse.js

@@ -1,6 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.isMultipleRowResult = exports.isNotNull = exports.tryTraverse_Sql_stmtContext = void 0;
+exports.tryTraverse_Sql_stmtContext = tryTraverse_Sql_stmtContext;
+exports.isNotNull = isNotNull;
+exports.isMultipleRowResult = isMultipleRowResult;
 const sqlite_1 = require("@wsporto/ts-mysql-parser/dist/sqlite");
 const select_columns_1 = require("../mysql-query-analyzer/select-columns");
 const collect_constraints_1 = require("../mysql-query-analyzer/collect-constraints");
@@ -27,7 +29,7 @@ function traverse_Sql_stmtContext(sql_stmt, traverseContext) {
         const deleteResult = traverse_delete_stmt(delete_stmt, traverseContext);
         return deleteResult;
     }
-    throw Error("traverse_Sql_stmtContext");
+    throw Error('traverse_Sql_stmtContext');
 }
 function tryTraverse_Sql_stmtContext(sql_stmt, traverseContext) {
     try {
@@ -42,22 +44,21 @@ function tryTraverse_Sql_stmtContext(sql_stmt, traverseContext) {
         });
     }
 }
-exports.tryTraverse_Sql_stmtContext = tryTraverse_Sql_stmtContext;
 function traverse_select_stmt(select_stmt, traverseContext, subQuery = false, recursive = false, recursiveNames = []) {
     const common_table_stmt = select_stmt.common_table_stmt();
     if (common_table_stmt) {
         const recursive = common_table_stmt.RECURSIVE_() != null;
         const common_table_expression = common_table_stmt.common_table_expression_list();
-        common_table_expression.forEach(common_table_expression => {
+        common_table_expression.forEach((common_table_expression) => {
             const table_name = common_table_expression.table_name();
-            const recursiveNames = common_table_expression.column_name_list().map(column_name => column_name.getText());
+            const recursiveNames = common_table_expression.column_name_list().map((column_name) => column_name.getText());
             const select_stmt = common_table_expression.select_stmt();
             const select_stmt_result = traverse_select_stmt(select_stmt, Object.assign(Object.assign({}, traverseContext), { subQuery: true }), subQuery, recursive, recursiveNames);
             select_stmt_result.columns.forEach((col, index) => {
                 var _a;
                 traverseContext.withSchema.push({
                     table: table_name.getText(),
-                    columnName: recursive ? (_a = recursiveNames[index]) !== null && _a !== void 0 ? _a : col.name : col.name,
+                    columnName: recursive ? ((_a = recursiveNames[index]) !== null && _a !== void 0 ? _a : col.name) : col.name,
                     columnType: col.type,
                     columnKey: '',
                     notNull: col.notNull
@@ -72,8 +73,10 @@ function traverse_select_stmt(select_stmt, traverseContext, subQuery = false, re
     }
     const [mainSelect, ...unionSelect] = select_stmt.select_core_list();
     const mainQueryResult = traverse_select_core(mainSelect, traverseContext, subQuery, recursive, recursiveNames);
-    const fromColumns = recursive ? mainQueryResult.columns.map((col, index) => mapTypeAndNullInferToColumnDef(col, recursiveNames[index])) : traverseContext.fromColumns;
-    unionSelect.forEach(select_core => {
+    const fromColumns = recursive
+        ? mainQueryResult.columns.map((col, index) => mapTypeAndNullInferToColumnDef(col, recursiveNames[index]))
+        : traverseContext.fromColumns;
+    unionSelect.forEach((select_core) => {
         const unionResult = traverse_select_core(select_core, Object.assign(Object.assign({}, traverseContext), { fromColumns }), subQuery, recursive);
         unionResult.columns.forEach((col, colIndex) => {
             mainQueryResult.columns[colIndex].table = '';
@@ -99,9 +102,9 @@ function traverse_select_stmt(select_stmt, traverseContext, subQuery = false, re
     if (order_by_stmt) {
         const selectColumns = mainQueryResult.columns.map((col, index) => mapTypeAndNullInferToColumnDef(col, recursiveNames[index]));
         const ordering_term_list = order_by_stmt.ordering_term_list();
-        ordering_term_list.forEach(ordering_term => {
+        ordering_term_list.forEach((ordering_term) => {
             const expr = ordering_term.expr();
-            if (expr.getText() == '?') {
+            if (expr.getText() === '?') {
                 hasOrderByParameter = true;
             }
             else {
@@ -129,7 +132,7 @@ function traverse_select_stmt(select_stmt, traverseContext, subQuery = false, re
             type1: exrp1Type.type,
             type2: (0, collect_constraints_1.freshVar)('INTEGER', 'INTEGER')
         });
-        if (expr_list.length == 2) {
+        if (expr_list.length === 2) {
             const expr2 = expr_list[1];
             const exrp2Type = traverse_expr(expr2, traverseContext);
             exrp2Type.notNull = true;
@@ -174,11 +177,11 @@ function traverse_select_core(select_core, traverseContext, subQuery = false, re
     }
     const result_column = select_core.result_column_list();
     const fromColumns = subQuery || recursive ? traverseContext.fromColumns.concat(columnsResult) : columnsResult;
-    result_column.forEach(result_column => {
+    result_column.forEach((result_column) => {
         var _a, _b;
         if (result_column.STAR()) {
             const tableName = (_a = result_column.table_name()) === null || _a === void 0 ? void 0 : _a.getText();
-            columnsResult.forEach(col => {
+            columnsResult.forEach((col) => {
                 const table = col.tableAlias || col.table;
                 if (!tableName || (0, select_columns_1.includeColumn)(col, tableName)) {
                     listType.push({
@@ -203,11 +206,13 @@ function traverse_select_core(select_core, traverseContext, subQuery = false, re
         if (expr) {
             const exprType = traverse_expr(expr, Object.assign(Object.assign({}, traverseContext), { fromColumns: fromColumns }));
             if (alias) {
-                traverseContext.relations.filter(relation => relation.joinColumn == exprType.name && (relation.name == exprType.table || relation.alias == exprType.table)).forEach(relation => {
+                traverseContext.relations
+                    .filter((relation) => relation.joinColumn === exprType.name && (relation.name === exprType.table || relation.alias === exprType.table))
+                    .forEach((relation) => {
                     relation.joinColumn = alias;
                 });
             }
-            if (exprType.type.kind == 'TypeVar') {
+            if (exprType.type.kind === 'TypeVar') {
                 if (alias) {
                     exprType.name = alias;
                 }
@@ -236,40 +241,44 @@ function traverse_select_core(select_core, traverseContext, subQuery = false, re
             });
         }
     }
-    const newColumns = listType.map(selectField => {
+    const newColumns = listType.map((selectField) => {
         const col = {
             columnName: selectField.name,
             table: selectField.table,
             columnType: selectField.type,
             notNull: selectField.notNull,
-            columnKey: ""
+            columnKey: ''
         };
         return col;
     });
     const groupByExprList = select_core._groupByExpr || [];
-    groupByExprList.forEach(groupByExpr => {
+    groupByExprList.forEach((groupByExpr) => {
         traverse_expr(groupByExpr, Object.assign(Object.assign({}, traverseContext), { fromColumns: newColumns.concat(fromColumns) }));
     });
     const havingExpr = select_core._havingExpr;
     if (havingExpr) {
-        //select have precedence: newColumns.concat(fromColumns) 
+        //select have precedence: newColumns.concat(fromColumns)
         traverse_expr(havingExpr, Object.assign(Object.assign({}, traverseContext), { fromColumns: newColumns.concat(fromColumns) }));
     }
     const querySpecification = {
-        columns: listType.map(col => (Object.assign(Object.assign({}, col), { notNull: col.notNull || isNotNull(col.name, whereExpr) || isNotNull(col.name, havingExpr) }))),
+        columns: listType.map((col) => (Object.assign(Object.assign({}, col), { notNull: col.notNull || isNotNull(col.name, whereExpr) || isNotNull(col.name, havingExpr) }))),
         fromColumns: columnsResult //TODO - return isMultipleRowResult instead
     };
     return querySpecification;
 }
 function extractRelationsAndParams(expr, fromColumns, parameters) {
     const columnsRef = (0, select_columns_1.getExpressions)(expr, sqlite_1.Column_nameContext);
-    const relations = columnsRef.filter(expr => !expr.isSubQuery).map(colRef => {
+    const relations = columnsRef
+        .filter((expr) => !expr.isSubQuery)
+        .map((colRef) => {
         const fieldName = (0, select_columns_1.splitName)(colRef.expr.parentCtx.getText());
         const column = (0, select_columns_1.findColumn)(fieldName, fromColumns);
         return column.tableAlias || column.table;
     });
     const expressionList = (0, select_columns_1.getExpressions)(expr, sqlite_1.ExprContext);
-    const paramsIds = expressionList.filter(expr => expr.expr.BIND_PARAMETER() != null).map(expr => expr.expr.BIND_PARAMETER().symbol.start);
+    const paramsIds = expressionList
+        .filter((expr) => expr.expr.BIND_PARAMETER() != null)
+        .map((expr) => expr.expr.BIND_PARAMETER().symbol.start);
     const params = getParamsIndexes(parameters, paramsIds);
     return {
         relations,
@@ -286,12 +295,14 @@ function traverse_table_or_subquery(table_or_subquery_list, join_constraint_list
         const table_alias_temp = ((_b = table_or_subquery.table_alias()) === null || _b === void 0 ? void 0 : _b.getText()) || '';
         let tableOrSubqueryFields = [];
         //grammar error: select * from table1 inner join table2....; inner is parsed as table_alias
-        let table_alias = table_alias_temp.toLowerCase() == 'left'
-            || table_alias_temp.toLowerCase() == 'right'
-            || table_alias_temp.toLowerCase() == 'full'
-            || table_alias_temp.toLowerCase() == 'outer'
-            || table_alias_temp.toLowerCase() == 'inner'
-            || table_alias_temp.toLowerCase() == 'cross' ? '' : table_alias_temp;
+        const table_alias = table_alias_temp.toLowerCase() === 'left' ||
+            table_alias_temp.toLowerCase() === 'right' ||
+            table_alias_temp.toLowerCase() === 'full' ||
+            table_alias_temp.toLowerCase() === 'outer' ||
+            table_alias_temp.toLowerCase() === 'inner' ||
+            table_alias_temp.toLowerCase() === 'cross'
+            ? ''
+            : table_alias_temp;
         const join_constraint = join_constraint_list && index > 0 ? join_constraint_list[index - 1] : undefined;
         const asAlias = table_or_subquery.AS_() || false;
         const tableAlias = (_c = table_or_subquery.table_alias()) === null || _c === void 0 ? void 0 : _c.getText();
@@ -303,10 +314,10 @@ function traverse_table_or_subquery(table_or_subquery_list, join_constraint_list
                 prefix: schema
             };
             tableOrSubqueryFields = (0, select_columns_1.filterColumns)(traverseContext.dbSchema, traverseContext.withSchema, table_alias, tableName);
-            const usingFields = (join_constraint === null || join_constraint === void 0 ? void 0 : join_constraint.USING_()) ? join_constraint === null || join_constraint === void 0 ? void 0 : join_constraint.column_name_list().map(column_name => column_name.getText()) : [];
+            const usingFields = (join_constraint === null || join_constraint === void 0 ? void 0 : join_constraint.USING_()) ? join_constraint === null || join_constraint === void 0 ? void 0 : join_constraint.column_name_list().map((column_name) => column_name.getText()) : [];
             const filteredFields = usingFields.length > 0 ? filterUsingFields(tableOrSubqueryFields, usingFields) : tableOrSubqueryFields;
             if (isLeftJoin) {
-                allFields.push(...filteredFields.map(field => (Object.assign(Object.assign({}, field), { notNull: false }))));
+                allFields.push(...filteredFields.map((field) => (Object.assign(Object.assign({}, field), { notNull: false }))));
             }
             else {
                 allFields.push(...filteredFields);
@@ -315,12 +326,12 @@ function traverse_table_or_subquery(table_or_subquery_list, join_constraint_list
         const select_stmt = table_or_subquery.select_stmt();
         if (select_stmt) {
             const subQueryResult = traverse_select_stmt(select_stmt, Object.assign(Object.assign({}, traverseContext), { subQuery: true }));
-            tableOrSubqueryFields = subQueryResult.columns.map(t => {
+            tableOrSubqueryFields = subQueryResult.columns.map((t) => {
                 const colDef = {
                     table: t.table ? tableAlias || '' : '',
                     columnName: t.name,
                     columnType: t.type,
-                    columnKey: "",
+                    columnKey: '',
                     notNull: t.notNull,
                     tableAlias: tableAlias
                 };
@@ -333,7 +344,7 @@ function traverse_table_or_subquery(table_or_subquery_list, join_constraint_list
             tableOrSubqueryFields = traverse_table_or_subquery(table_or_subquery_list2, null, null, traverseContext);
             allFields.push(...tableOrSubqueryFields);
         }
-        const idColumn = (_e = tableOrSubqueryFields.find(field => field.columnKey == 'PRI')) === null || _e === void 0 ? void 0 : _e.columnName;
+        const idColumn = (_e = tableOrSubqueryFields.find((field) => field.columnKey === 'PRI')) === null || _e === void 0 ? void 0 : _e.columnName;
         const relation = {
             name: asAlias ? table_alias : tableOrSubqueryName,
             alias: table_alias,
@@ -342,23 +353,24 @@ function traverse_table_or_subquery(table_or_subquery_list, join_constraint_list
             parentCardinality: 'one',
             joinColumn: idColumn
         };
-        if (join_constraint) { //index 0 is the FROM (root relation)
+        if (join_constraint) {
+            //index 0 is the FROM (root relation)
             const expr = join_constraint.expr(); //ON expr
             if (expr) {
                 traverse_expr(expr, Object.assign(Object.assign({}, traverseContext), { fromColumns: allFields }));
                 const allJoinColumsn = getAllColumns(expr);
-                allJoinColumsn.forEach(joinColumn => {
-                    const column = allFields.find(col => col.columnName == joinColumn.name && (col.tableAlias == joinColumn.prefix || col.table == joinColumn.prefix));
-                    const filterUniqueKeys = allFields.filter(col => (joinColumn.prefix == col.table || joinColumn.prefix == col.tableAlias) && (col.columnKey == 'PRI'));
-                    const compositeKey = filterUniqueKeys.find(uni => uni.columnName == column.columnName);
-                    const notUnique = (filterUniqueKeys.length > 1 && compositeKey) || ((column === null || column === void 0 ? void 0 : column.columnKey) != 'UNI' && (column === null || column === void 0 ? void 0 : column.columnKey) != 'PRI');
-                    if (joinColumn.prefix != relation.name && joinColumn.prefix != relation.alias) {
+                allJoinColumsn.forEach((joinColumn) => {
+                    const column = allFields.find((col) => col.columnName === joinColumn.name && (col.tableAlias === joinColumn.prefix || col.table === joinColumn.prefix));
+                    const filterUniqueKeys = allFields.filter((col) => (joinColumn.prefix === col.table || joinColumn.prefix === col.tableAlias) && col.columnKey === 'PRI');
+                    const compositeKey = filterUniqueKeys.find((uni) => uni.columnName === column.columnName);
+                    const notUnique = (filterUniqueKeys.length > 1 && compositeKey) || ((column === null || column === void 0 ? void 0 : column.columnKey) !== 'UNI' && (column === null || column === void 0 ? void 0 : column.columnKey) !== 'PRI');
+                    if (joinColumn.prefix !== relation.name && joinColumn.prefix !== relation.alias) {
                         relation.parentRelation = joinColumn.prefix;
                         if (notUnique) {
                             relation.parentCardinality = 'many';
                         }
                     }
-                    if (joinColumn.prefix == relation.name || joinColumn.prefix == relation.alias) {
+                    if (joinColumn.prefix === relation.name || joinColumn.prefix === relation.alias) {
                         if (notUnique) {
                             relation.cardinality = 'many';
                         }
@@ -369,16 +381,14 @@ function traverse_table_or_subquery(table_or_subquery_list, join_constraint_list
         if (!traverseContext.subQuery) {
             traverseContext.relations.push(relation);
             //dynamic query
-            const fragment = (join_operator_list != null && index > 0 ? extractOriginalSql(join_operator_list[index - 1]) : 'FROM')
-                + ' ' + extractOriginalSql(table_or_subquery_list[index])
-                + (join_constraint != null ? ' ' + extractOriginalSql(join_constraint) : '');
+            const fragment = `${join_operator_list != null && index > 0 ? extractOriginalSql(join_operator_list[index - 1]) : 'FROM'} ${extractOriginalSql(table_or_subquery_list[index])}${join_constraint != null ? ` ${extractOriginalSql(join_constraint)}` : ''}`;
             const params = traverseContext.parameters.slice(numParamsBefore).map((_, index) => index + numParamsBefore);
             traverseContext.dynamicSqlInfo2.from.push({
                 fragment: fragment,
                 relationName: relation.name,
                 relationAlias: relation.alias,
                 parentRelation: relation.parentRelation,
-                fields: tableOrSubqueryFields.map(field => field.columnName),
+                fields: tableOrSubqueryFields.map((field) => field.columnName),
                 parameters: params
             });
         }
@@ -388,11 +398,11 @@ function traverse_table_or_subquery(table_or_subquery_list, join_constraint_list
 function traverse_expr(expr, traverseContext) {
     var _a, _b;
     const function_name = (_a = expr.function_name()) === null || _a === void 0 ? void 0 : _a.getText().toLowerCase();
-    if (function_name == 'avg') {
+    if (function_name === 'avg') {
         const functionType = (0, collect_constraints_1.freshVar)(expr.getText(), 'REAL');
         const sumParamExpr = expr.expr(0);
         const paramType = traverse_expr(sumParamExpr, traverseContext);
-        if (paramType.type.kind == 'TypeVar') {
+        if (paramType.type.kind === 'TypeVar') {
             functionType.table = paramType.table;
         }
         return {
@@ -402,7 +412,7 @@ function traverse_expr(expr, traverseContext) {
             table: functionType.table || ''
         };
     }
-    if (function_name == 'sum') {
+    if (function_name === 'sum') {
         const sumParamExpr = expr.expr(0);
         const paramType = traverse_expr(sumParamExpr, traverseContext);
         return {
@@ -412,7 +422,7 @@ function traverse_expr(expr, traverseContext) {
             table: paramType.table || ''
         };
     }
-    if (function_name == 'min' || function_name == 'max') {
+    if (function_name === 'min' || function_name === 'max') {
         const functionType = (0, collect_constraints_1.freshVar)(expr.getText(), '?');
         const sumParamExpr = expr.expr(0);
         const paramType = traverse_expr(sumParamExpr, traverseContext);
@@ -428,12 +438,12 @@ function traverse_expr(expr, traverseContext) {
             table: functionType.table || ''
         };
     }
-    if (function_name == 'count') {
+    if (function_name === 'count') {
         const functionType = (0, collect_constraints_1.freshVar)(expr.getText(), 'INTEGER');
-        if (expr.expr_list().length == 1) {
+        if (expr.expr_list().length === 1) {
             const sumParamExpr = expr.expr(0);
             const paramType = traverse_expr(sumParamExpr, traverseContext);
-            if (paramType.type.kind == 'TypeVar') {
+            if (paramType.type.kind === 'TypeVar') {
                 functionType.table = paramType.table;
             }
         }
@@ -444,16 +454,16 @@ function traverse_expr(expr, traverseContext) {
             table: functionType.table || ''
         };
     }
-    if (function_name == 'concat') {
+    if (function_name === 'concat') {
         const functionType = (0, collect_constraints_1.freshVar)(expr.getText(), 'TEXT');
-        expr.expr_list().forEach(paramExpr => {
+        expr.expr_list().forEach((paramExpr) => {
             const paramType = traverse_expr(paramExpr, traverseContext);
             traverseContext.constraints.push({
                 expression: expr.getText(),
                 type1: functionType,
                 type2: paramType.type
             });
-            if (paramType.type.kind == 'TypeVar') {
+            if (paramType.type.kind === 'TypeVar') {
                 functionType.table = paramType.table;
             }
         });
@@ -464,7 +474,7 @@ function traverse_expr(expr, traverseContext) {
             table: functionType.table || ''
         };
     }
-    if (function_name == 'length') {
+    if (function_name === 'length') {
         const functionType = (0, collect_constraints_1.freshVar)(expr.getText(), 'INTEGER');
         const paramExpr = expr.expr(0);
         const paramType = traverse_expr(paramExpr, traverseContext);
@@ -473,7 +483,7 @@ function traverse_expr(expr, traverseContext) {
             type1: (0, collect_constraints_1.freshVar)(expr.getText(), '?'), //str or blob
             type2: paramType.type
         });
-        if (paramType.type.kind == 'TypeVar') {
+        if (paramType.type.kind === 'TypeVar') {
             functionType.table = paramType.table;
         }
         return {
@@ -483,8 +493,8 @@ function traverse_expr(expr, traverseContext) {
             table: functionType.table || ''
         };
     }
-    if (function_name == 'group_concat') {
-        expr.expr_list().forEach(paramExpr => {
+    if (function_name === 'group_concat') {
+        expr.expr_list().forEach((paramExpr) => {
             const param1Type = traverse_expr(paramExpr, traverseContext);
             traverseContext.constraints.push({
                 expression: expr.getText(),
@@ -499,7 +509,16 @@ function traverse_expr(expr, traverseContext) {
             table: ''
         };
     }
-    if (function_name == 'round') {
+    if (function_name === 'random') {
+        const functionType = (0, collect_constraints_1.freshVar)(expr.getText(), 'INTEGER');
+        return {
+            name: functionType.name,
+            type: functionType,
+            notNull: true,
+            table: ''
+        };
+    }
+    if (function_name === 'round') {
         const functionType = (0, collect_constraints_1.freshVar)(expr.getText(), 'REAL');
         const param1Expr = expr.expr(0);
         const param1Type = traverse_expr(param1Expr, traverseContext);
@@ -524,9 +543,9 @@ function traverse_expr(expr, traverseContext) {
             table: param1Type.table || ''
         };
     }
-    if (function_name == 'coalesce') {
+    if (function_name === 'coalesce') {
         const functionType = (0, collect_constraints_1.freshVar)(expr.getText(), '?');
-        const paramTypes = expr.expr_list().map(paramExpr => {
+        const paramTypes = expr.expr_list().map((paramExpr) => {
             const paramType = traverse_expr(paramExpr, traverseContext);
             traverseContext.constraints.push({
                 expression: expr.getText(),
@@ -538,11 +557,11 @@ function traverse_expr(expr, traverseContext) {
         return {
             name: functionType.name,
             type: functionType,
-            notNull: paramTypes.some(param => param.notNull),
+            notNull: paramTypes.some((param) => param.notNull),
             table: functionType.table || ''
         };
     }
-    if (function_name == 'strftime') {
+    if (function_name === 'strftime') {
         const functionType = (0, collect_constraints_1.freshVar)(expr.getText(), 'TEXT');
         const paramExpr = expr.expr(1);
         const paramType = traverse_expr(paramExpr, traverseContext);
@@ -559,7 +578,7 @@ function traverse_expr(expr, traverseContext) {
             table: functionType.table || ''
         };
     }
-    if (function_name == 'date' || function_name == 'time' || function_name == 'datetime') {
+    if (function_name === 'date' || function_name === 'time' || function_name === 'datetime') {
         const functionType = (0, collect_constraints_1.freshVar)(expr.getText(), 'DATE');
         const paramExpr = expr.expr(0);
         const paramType = traverse_expr(paramExpr, traverseContext);
@@ -576,10 +595,10 @@ function traverse_expr(expr, traverseContext) {
             table: functionType.table || ''
         };
     }
-    if (function_name == 'julianday') {
+    if (function_name === 'julianday') {
         const functionType = (0, collect_constraints_1.freshVar)(expr.getText(), 'REAL');
         const paramExpr = expr.expr(0);
-        const notNull = paramExpr.getText().toLowerCase() == `'now'` ? true : false;
+        const notNull = paramExpr.getText().toLowerCase() === `'now'`;
         const paramType = traverse_expr(paramExpr, traverseContext);
         traverseContext.constraints.push({
             expression: paramExpr.getText(),
@@ -593,10 +612,10 @@ function traverse_expr(expr, traverseContext) {
             table: functionType.table || ''
         };
     }
-    if (function_name == 'unixepoch') {
+    if (function_name === 'unixepoch') {
         const functionType = (0, collect_constraints_1.freshVar)(expr.getText(), 'INTEGER');
         const paramExpr = expr.expr(0);
-        const notNull = paramExpr.getText().toLowerCase() == `'now'` ? true : false;
+        const notNull = paramExpr.getText().toLowerCase() === `'now'`;
         const paramType = traverse_expr(paramExpr, traverseContext);
         traverseContext.constraints.push({
             expression: paramExpr.getText(),
@@ -610,11 +629,11 @@ function traverse_expr(expr, traverseContext) {
             table: functionType.table || ''
         };
     }
-    if (function_name == 'ifnull') {
+    if (function_name === 'ifnull') {
         const functionType = (0, collect_constraints_1.freshVar)(expr.getText(), '?');
-        const paramTypes = expr.expr_list().map(paramExpr => {
+        const paramTypes = expr.expr_list().map((paramExpr) => {
             const paramType = traverse_expr(paramExpr, traverseContext);
-            if (paramType.name == '?') {
+            if (paramType.name === '?') {
                 paramType.notNull = false;
             }
             traverseContext.constraints.push({
@@ -627,13 +646,11 @@ function traverse_expr(expr, traverseContext) {
         return {
             name: functionType.name,
             type: functionType,
-            notNull: paramTypes.every(param => param.notNull),
+            notNull: paramTypes.every((param) => param.notNull),
             table: functionType.table || ''
         };
     }
-    if (function_name == 'row_number'
-        || function_name == 'rank'
-        || function_name == 'dense_rank') {
+    if (function_name === 'row_number' || function_name === 'rank' || function_name === 'dense_rank') {
         const functionType = (0, collect_constraints_1.freshVar)(expr.getText(), 'INTEGER');
         return {
             name: functionType.name,
@@ -642,19 +659,17 @@ function traverse_expr(expr, traverseContext) {
             table: functionType.table || ''
         };
     }
-    if (function_name == 'first_value'
-        || function_name == 'last_value') {
+    if (function_name === 'first_value' || function_name === 'last_value') {
         const paramExpr = expr.expr(0);
         const paramType = traverse_expr(paramExpr, traverseContext);
         return paramType;
     }
-    if (function_name == 'lead'
-        || function_name == 'lag') {
+    if (function_name === 'lead' || function_name === 'lag') {
         const paramExpr = expr.expr(0);
         const paramType = traverse_expr(paramExpr, traverseContext);
         return Object.assign(Object.assign({}, paramType), { notNull: false });
     }
-    if (function_name == 'iif') {
+    if (function_name === 'iif') {
         const expr1 = expr.expr(0);
         traverse_expr(expr1, traverseContext);
         const expr2 = expr.expr(1);
@@ -668,7 +683,7 @@ function traverse_expr(expr, traverseContext) {
         });
         return Object.assign(Object.assign({}, expr2Type), { notNull: expr2Type.notNull && expr3Type.notNull });
     }
-    if (function_name == 'vector') {
+    if (function_name === 'vector') {
         const functionType = (0, collect_constraints_1.freshVar)(expr.getText(), 'BLOB');
         const param1Expr = expr.expr(0);
         const param1Type = traverse_expr(param1Expr, traverseContext);
@@ -685,7 +700,7 @@ function traverse_expr(expr, traverseContext) {
             table: ''
         };
     }
-    if (function_name == 'vector_extract') {
+    if (function_name === 'vector_extract') {
         const functionType = (0, collect_constraints_1.freshVar)(expr.getText(), 'TEXT');
         const param1Expr = expr.expr(0);
         const param1Type = traverse_expr(param1Expr, traverseContext);
@@ -701,7 +716,7 @@ function traverse_expr(expr, traverseContext) {
             table: ''
         };
     }
-    if (function_name == 'vector_distance_cos') {
+    if (function_name === 'vector_distance_cos') {
         const functionType = (0, collect_constraints_1.freshVar)(expr.getText(), 'REAL');
         const param1Expr = expr.expr(0);
         if (param1Expr) {
@@ -731,7 +746,7 @@ function traverse_expr(expr, traverseContext) {
         };
     }
     if (function_name) {
-        throw Error('traverse_expr: function not supported:' + function_name);
+        throw Error(`traverse_expr: function not supported:${function_name}`);
     }
     const column_name = expr.column_name();
     const table_name = expr.table_name();
@@ -768,7 +783,7 @@ function traverse_expr(expr, traverseContext) {
         return {
             name: type.name,
             type: type,
-            notNull: literal.NULL_() != null ? false : true,
+            notNull: literal.NULL_() == null,
             table: type.table || ''
         };
     }
@@ -821,8 +836,7 @@ function traverse_expr(expr, traverseContext) {
             type1: returnType,
             type2: typeLeft.type
         });
-        const isDateFunctionContext = (expr.parentCtx instanceof sqlite_1.ExprContext)
-            && ((_b = expr.parentCtx.function_name()) === null || _b === void 0 ? void 0 : _b.getText().toLowerCase()) == 'date';
+        const isDateFunctionContext = expr.parentCtx instanceof sqlite_1.ExprContext && ((_b = expr.parentCtx.function_name()) === null || _b === void 0 ? void 0 : _b.getText().toLowerCase()) === 'date';
         if (!isDateFunctionContext) {
             traverseContext.constraints.push({
                 expression: exprRight.getText(),
@@ -832,15 +846,24 @@ function traverse_expr(expr, traverseContext) {
         }
         return Object.assign(Object.assign({}, typeLeft), { notNull: typeLeft.notNull && typeRight.notNull });
     }
-    if (expr.LT2() || expr.GT2() || expr.AMP() || expr.PIPE() || expr.LT() || expr.LT_EQ() || expr.GT() || expr.GT_EQ() || expr.NOT_EQ1() || expr.NOT_EQ2()) {
+    if (expr.LT2() ||
+        expr.GT2() ||
+        expr.AMP() ||
+        expr.PIPE() ||
+        expr.LT() ||
+        expr.LT_EQ() ||
+        expr.GT() ||
+        expr.GT_EQ() ||
+        expr.NOT_EQ1() ||
+        expr.NOT_EQ2()) {
         const exprLeft = expr.expr(0);
         const exprRight = expr.expr(1);
         const typeLeft = traverse_expr(exprLeft, traverseContext);
         const typeRight = traverse_expr(exprRight, traverseContext);
-        if (typeLeft.name == '?') {
+        if (typeLeft.name === '?') {
             typeLeft.notNull = true;
         }
-        if (typeRight.name == '?') {
+        if (typeRight.name === '?') {
             typeRight.notNull = true;
         }
         traverseContext.constraints.push({
@@ -856,7 +879,8 @@ function traverse_expr(expr, traverseContext) {
             table: resultType.table || ''
         };
     }
-    if (expr.IS_()) { //is null/is not null
+    if (expr.IS_()) {
+        //is null/is not null
         const expr_ = expr.expr(0);
         traverse_expr(expr_, traverseContext);
         const type = (0, collect_constraints_1.freshVar)(expr.getText(), 'INTEGER');
@@ -867,15 +891,16 @@ function traverse_expr(expr, traverseContext) {
             table: type.table || ''
         };
     }
-    if (expr.ASSIGN()) { //=
+    if (expr.ASSIGN()) {
+        //=
         const exprLeft = expr.expr(0);
         const exprRight = expr.expr(1);
         const typeLeft = traverse_expr(exprLeft, traverseContext);
         const typeRight = traverse_expr(exprRight, traverseContext);
-        if (typeLeft.name == '?') {
+        if (typeLeft.name === '?') {
             typeLeft.notNull = true;
         }
-        if (typeRight.name == '?') {
+        if (typeRight.name === '?') {
             typeRight.notNull = true;
         }
         traverseContext.constraints.push({
@@ -895,10 +920,10 @@ function traverse_expr(expr, traverseContext) {
         const exprType = traverse_expr(expr.expr(0), traverseContext);
         const between1 = traverse_expr(expr.expr(1), traverseContext);
         const between2 = traverse_expr(expr.expr(2), traverseContext);
-        if (between1.name == '?') {
+        if (between1.name === '?') {
             between1.notNull = true;
         }
-        if (between2.name == '?') {
+        if (between2.name === '?') {
             between2.notNull = true;
         }
         traverseContext.constraints.push({
@@ -928,7 +953,7 @@ function traverse_expr(expr, traverseContext) {
         const exprList = expr.expr_list();
         const inExprLeft = exprList[0];
         const typeLeft = traverse_expr(inExprLeft, traverseContext);
-        if (typeLeft.name == '?') {
+        if (typeLeft.name === '?') {
             typeLeft.notNull = true;
         }
         if (expr.NOT_()) {
@@ -944,9 +969,9 @@ function traverse_expr(expr, traverseContext) {
                 });
             }
             //NOT IN (1, 2, 3)
-            exprList.slice(1).forEach(inExpr => {
+            exprList.slice(1).forEach((inExpr) => {
                 const typeRight = traverse_expr(inExpr, traverseContext);
-                if (typeRight.name == '?') {
+                if (typeRight.name === '?') {
                     typeRight.notNull = true;
                 }
                 traverseContext.constraints.push({
@@ -970,9 +995,9 @@ function traverse_expr(expr, traverseContext) {
                 });
             }
             //IN (1, 2, 3)
-            rightExpr.expr_list().forEach(inExpr2 => {
+            rightExpr.expr_list().forEach((inExpr2) => {
                 const typeRight = traverse_expr(inExpr2, traverseContext);
-                if (typeRight.name == '?') {
+                if (typeRight.name === '?') {
                     typeRight.notNull = true;
                 }
                 traverseContext.constraints.push({
@@ -995,10 +1020,10 @@ function traverse_expr(expr, traverseContext) {
         const exprRight = expr.expr(1);
         const typeLeft = traverse_expr(exprLeft, traverseContext);
         const typeRight = traverse_expr(exprRight, traverseContext);
-        if (typeLeft.name == '?') {
+        if (typeLeft.name === '?') {
             typeLeft.notNull = true;
         }
-        if (typeRight.name == '?') {
+        if (typeRight.name === '?') {
             typeRight.notNull = true;
         }
         traverseContext.constraints.push({
@@ -1032,7 +1057,7 @@ function traverse_expr(expr, traverseContext) {
     }
     if (expr.OPEN_PAR() && expr.CLOSE_PAR()) {
         const type = (0, collect_constraints_1.freshVar)(expr.getText(), '?');
-        const exprTypes = expr.expr_list().map(innerExpr => {
+        const exprTypes = expr.expr_list().map((innerExpr) => {
             const exprType = traverse_expr(innerExpr, traverseContext);
             traverseContext.constraints.push({
                 expression: innerExpr.getText(),
@@ -1044,7 +1069,7 @@ function traverse_expr(expr, traverseContext) {
         return {
             name: type.name,
             type: type,
-            notNull: exprTypes.every(type => type.notNull),
+            notNull: exprTypes.every((type) => type.notNull),
             table: type.table || ''
         };
     }
@@ -1053,7 +1078,7 @@ function traverse_expr(expr, traverseContext) {
         const whenTypes = [];
         expr.expr_list().forEach((expr_, index) => {
             const type = traverse_expr(expr_, traverseContext);
-            if (index % 2 == 0 && (!expr.ELSE_() || index < expr.expr_list().length - 1)) {
+            if (index % 2 === 0 && (!expr.ELSE_() || index < expr.expr_list().length - 1)) {
                 whenTypes.push(type);
             }
             else {
@@ -1080,11 +1105,11 @@ function traverse_expr(expr, traverseContext) {
         return {
             name: extractOriginalSql(expr),
             type: type.type,
-            notNull: expr.ELSE_() ? resultTypes.every(type => type.notNull) : false,
+            notNull: expr.ELSE_() ? resultTypes.every((type) => type.notNull) : false,
             table: type.table || ''
         };
     }
-    throw Error('traverse_expr not supported:' + expr.getText());
+    throw Error(`traverse_expr not supported:${expr.getText()}`);
 }
 function extractOriginalSql(rule) {
     var _a, _b;
@@ -1094,7 +1119,10 @@ function extractOriginalSql(rule) {
     return result;
 }
 function traverse_column_name(column_name, table_name, traverseContext) {
-    const fieldName = { name: column_name.getText(), prefix: (table_name === null || table_name === void 0 ? void 0 : table_name.getText()) || '' };
+    const fieldName = {
+        name: column_name.getText(),
+        prefix: (table_name === null || table_name === void 0 ? void 0 : table_name.getText()) || ''
+    };
     const column = (0, select_columns_1.findColumn)(fieldName, traverseContext.fromColumns);
     // const typeVar = freshVar(column.columnName, column.columnType.type, column.tableAlias || column.table);
     return column;
@@ -1104,29 +1132,28 @@ function isNotNull(columnName, where) {
         return false;
     }
     if (where.AND_()) {
-        const ifNullList = where.expr_list().map(expr => isNotNull(columnName, expr));
-        const result = ifNullList.some(v => v);
+        const ifNullList = where.expr_list().map((expr) => isNotNull(columnName, expr));
+        const result = ifNullList.some((v) => v);
         return result;
     }
-    else if (where.OR_()) {
-        const possibleNullList = where.expr_list().map(expr => isNotNull(columnName, expr));
-        const result = possibleNullList.every(v => v);
+    if (where.OR_()) {
+        const possibleNullList = where.expr_list().map((expr) => isNotNull(columnName, expr));
+        const result = possibleNullList.every((v) => v);
         return result;
     }
-    else {
-        return isNotNullExpr(columnName, where);
-    }
+    return isNotNullExpr(columnName, where);
 }
-exports.isNotNull = isNotNull;
 function isNotNullExpr(columnName, expr) {
     if (expr.OPEN_PAR() && expr.CLOSE_PAR()) {
         const innerExpr = expr.expr(0);
         return isNotNull(columnName, innerExpr);
     }
-    if (expr.ASSIGN()
-        || expr.GT() || expr.GT_EQ()
-        || expr.LT() || expr.LT_EQ()
-        || (expr.IS_() && expr.expr_list().length == 2 && expr.expr(1).getText() == 'notnull')) {
+    if (expr.ASSIGN() ||
+        expr.GT() ||
+        expr.GT_EQ() ||
+        expr.LT() ||
+        expr.LT_EQ() ||
+        (expr.IS_() && expr.expr_list().length === 2 && expr.expr(1).getText() === 'notnull')) {
         const exprLeft = expr.expr(0);
         const exprRight = expr.expr(1);
         const column_name_left = exprLeft.column_name();
@@ -1134,7 +1161,7 @@ function isNotNullExpr(columnName, expr) {
         if (column_name_left || column_name_right) {
             const columnLeft = column_name_left === null || column_name_left === void 0 ? void 0 : column_name_left.getText();
             const columnRight = column_name_right === null || column_name_right === void 0 ? void 0 : column_name_right.getText();
-            if (columnLeft == columnName || columnRight == columnName) {
+            if (columnLeft === columnName || columnRight === columnName) {
                 return true;
             }
         }
@@ -1142,23 +1169,24 @@ function isNotNullExpr(columnName, expr) {
     return false;
 }
 function isMultipleRowResult(select_stmt, fromColumns) {
-    if (select_stmt.select_core_list().length == 1) { //UNION queries are multipleRowsResult = true
+    if (select_stmt.select_core_list().length === 1) {
+        //UNION queries are multipleRowsResult = true
         const select_core = select_stmt.select_core(0);
         const from = select_core.FROM_();
         if (!from) {
             return false;
         }
-        const groupBy = select_stmt.select_core_list().some(select_core => select_core.GROUP_() != null);
+        const groupBy = select_stmt.select_core_list().some((select_core) => select_core.GROUP_() != null);
         if (groupBy) {
             return true;
         }
-        const agreegateFunction = select_core.result_column_list().some(result_column => isAgregateFunction(result_column));
+        const agreegateFunction = select_core.result_column_list().some((result_column) => isAgregateFunction(result_column));
         if (agreegateFunction) {
             return false;
         }
         const _whereExpr = select_core._whereExpr;
         const isSingleResult = select_core.join_clause() == null && _whereExpr && where_is_single_result(_whereExpr, fromColumns);
-        if (isSingleResult == true) {
+        if (isSingleResult === true) {
             return false;
         }
     }
@@ -1167,10 +1195,10 @@ function isMultipleRowResult(select_stmt, fromColumns) {
     }
     return true;
 }
-exports.isMultipleRowResult = isMultipleRowResult;
 function isAgregateFunction(result_column) {
     var _a;
-    if (((_a = result_column.expr()) === null || _a === void 0 ? void 0 : _a.over_clause()) != null) { //window function isMultipleRow = true
+    if (((_a = result_column.expr()) === null || _a === void 0 ? void 0 : _a.over_clause()) != null) {
+        //window function isMultipleRow = true
         return false;
     }
     const expr = result_column.expr();
@@ -1180,21 +1208,21 @@ function isAgregateFunction(result_column) {
 function isAgregateFunctionExpr(expr) {
     var _a;
     //ex. min(value)/100, 100/min(value)
-    const isAgrr = expr.expr_list().some(expr => isAgregateFunctionExpr(expr));
+    const isAgrr = expr.expr_list().some((expr) => isAgregateFunctionExpr(expr));
     if (isAgrr) {
         return isAgrr;
     }
     const function_name = (_a = expr.function_name()) === null || _a === void 0 ? void 0 : _a.getText().toLowerCase();
-    return function_name == 'count'
-        || function_name == 'sum'
-        || function_name == 'avg'
-        || function_name == 'min'
-        || function_name == 'max'
-        || function_name == 'group_concat';
+    return (function_name === 'count' ||
+        function_name === 'sum' ||
+        function_name === 'avg' ||
+        function_name === 'min' ||
+        function_name === 'max' ||
+        function_name === 'group_concat');
 }
 function isLimitOne(select_stmt) {
     const limit_stmt = select_stmt.limit_stmt();
-    if (limit_stmt && limit_stmt.expr(0).getText() == '1') {
+    if (limit_stmt && limit_stmt.expr(0).getText() === '1') {
         return true;
     }
     return false;
@@ -1206,7 +1234,7 @@ function where_is_single_result(whereExpr, fromColumns) {
     }
     const expr_list = whereExpr.expr_list();
     const onlyAnd = !whereExpr.OR_();
-    const oneSingle = expr_list.some(expr => is_single_result(expr, fromColumns));
+    const oneSingle = expr_list.some((expr) => is_single_result(expr, fromColumns));
     if (onlyAnd && oneSingle) {
         return true;
     }
@@ -1219,7 +1247,7 @@ function is_single_result(expr, fromColumns) {
     if (column_name && expr.ASSIGN()) {
         const fieldName = (0, select_columns_1.splitName)(column_name.getText());
         const column = (0, select_columns_1.findColumn)(fieldName, fromColumns);
-        if (column.columnKey == 'PRI') {
+        if (column.columnKey === 'PRI') {
             return true;
         }
     }
@@ -1229,7 +1257,7 @@ function traverse_insert_stmt(insert_stmt, traverseContext) {
     var _a;
     const table_name = insert_stmt.table_name();
     const fromColumns = (0, select_columns_1.filterColumns)(traverseContext.dbSchema, [], '', (0, select_columns_1.splitName)(table_name.getText()));
-    const columns = insert_stmt.column_name_list().map(column_name => {
+    const columns = insert_stmt.column_name_list().map((column_name) => {
         return traverse_column_name(column_name, null, Object.assign(Object.assign({}, traverseContext), { fromColumns }));
     });
     const insertColumns = [];
@@ -1245,8 +1273,8 @@ function traverse_insert_stmt(insert_stmt, traverseContext) {
                     type1: col.columnType,
                     type2: exprType.type
                 });
-                const notNullColumn = (col.columnKey == 'PRI' && col.columnType.type == 'INTEGER') ? false : col.notNull;
-                insertColumns.push(Object.assign(Object.assign({}, param), { notNull: exprType.name == '?' ? notNullColumn : param.notNull }));
+                const notNullColumn = col.columnKey === 'PRI' && col.columnType.type === 'INTEGER' ? false : col.notNull;
+                insertColumns.push(Object.assign(Object.assign({}, param), { notNull: exprType.name === '?' ? notNullColumn : param.notNull }));
             });
         });
     });
@@ -1261,10 +1289,10 @@ function traverse_insert_stmt(insert_stmt, traverseContext) {
                 type1: col.columnType,
                 type2: selectColumn.type
             });
-            const notNullColumn = (col.columnKey == 'PRI' && col.columnType.type == 'INTEGER') ? false : col.notNull;
+            const notNullColumn = col.columnKey === 'PRI' && col.columnType.type === 'INTEGER' ? false : col.notNull;
             columnNullability.set(selectColumn.type.id, notNullColumn);
         });
-        traverseContext.parameters.forEach(param => {
+        traverseContext.parameters.forEach((param) => {
             insertColumns.push(Object.assign(Object.assign({}, param), { notNull: columnNullability.get(param.type.id) != null ? columnNullability.get(param.type.id) : param.notNull }));
         });
     }
@@ -1276,14 +1304,16 @@ function traverse_insert_stmt(insert_stmt, traverseContext) {
             const column_name = upsert_clause.column_name(index);
             const col = traverse_column_name(column_name, null, Object.assign(Object.assign({}, traverseContext), { fromColumns }));
             const expr = upsert_clause.expr(index);
-            const exprType = traverse_expr(expr, Object.assign(Object.assign({}, traverseContext), { fromColumns }));
+            const table_name = expr.table_name();
+            const excludedColumns = table_name && table_name.getText() === 'excluded' ? fromColumns.map((col) => (Object.assign(Object.assign({}, col), { table: 'excluded' }))) : [];
+            const exprType = traverse_expr(expr, Object.assign(Object.assign({}, traverseContext), { fromColumns: fromColumns.concat(excludedColumns) }));
             traverseContext.constraints.push({
                 expression: column_name.getText(),
                 type1: col.columnType,
                 type2: exprType.type
             });
         });
-        traverseContext.parameters.slice(paramsBefore).forEach(param => {
+        traverseContext.parameters.slice(paramsBefore).forEach((param) => {
             insertColumns.push(Object.assign(Object.assign({}, param), { notNull: param.notNull }));
         });
     }
@@ -1300,9 +1330,9 @@ function traverse_insert_stmt(insert_stmt, traverseContext) {
 }
 function traverse_returning_clause(returning_clause, fromColumns) {
     const result_column_list = returning_clause.result_column_list();
-    const result = result_column_list.flatMap(result_column => {
+    const result = result_column_list.flatMap((result_column) => {
         if (result_column.STAR()) {
-            return fromColumns.map(col => {
+            return fromColumns.map((col) => {
                 const newCol = {
                     name: col.columnName,
                     type: col.columnType,
@@ -1319,9 +1349,10 @@ function traverse_returning_clause(returning_clause, fromColumns) {
 function traverse_update_stmt(update_stmt, traverseContext) {
     const table_name = update_stmt.qualified_table_name().getText();
     const fromColumns = (0, select_columns_1.filterColumns)(traverseContext.dbSchema, [], '', (0, select_columns_1.splitName)(table_name));
-    const column_name_list = Array.from({ length: update_stmt.ASSIGN_list().length })
-        .map((_, i) => update_stmt.column_name(i));
-    const columns = column_name_list.map(column_name => {
+    const column_name_list = Array.from({
+        length: update_stmt.ASSIGN_list().length
+    }).map((_, i) => update_stmt.column_name(i));
+    const columns = column_name_list.map((column_name) => {
         return traverse_column_name(column_name, null, Object.assign(Object.assign({}, traverseContext), { fromColumns }));
     });
     const updateColumns = [];
@@ -1377,11 +1408,10 @@ function getAllColumns(expr) {
         columns.push((0, select_columns_1.splitName)(expr1.getText()));
         columns.push((0, select_columns_1.splitName)(expr2.getText()));
     }
-    ;
     return columns;
 }
 function filterUsingFields(fields, usingFields) {
-    const result = fields.filter(field => !usingFields.includes(field.columnName));
+    const result = fields.filter((field) => !usingFields.includes(field.columnName));
     return result;
 }
 function getParamsIndexes(parameters, paramsIds) {
@@ -1389,7 +1419,7 @@ function getParamsIndexes(parameters, paramsIds) {
     parameters.forEach((param, index) => {
         map.set(param.paramIndex, index);
     });
-    return paramsIds.map(id => map.get(id));
+    return paramsIds.map((id) => map.get(id));
 }
 function getWhereFragmentExpressions(whereExpr) {
     const exprList = [];