orez 0.2.30 → 0.2.32

package/dist/pg-proxy-do-backend.js

@@ -1033,6 +1033,18 @@ function equalityExpr(left, right) {
         },
     };
 }
+function numericLiteralValue(value) {
+    const literal = astLiteralValue(unwrapTypeCast(value));
+    if (typeof literal !== 'number')
+        return null;
+    return Number.isFinite(literal) ? literal : null;
+}
+function isDivisionByMillis(value) {
+    const expr = unwrapTypeCast(value)?.A_Expr;
+    if (expr?.kind !== 'AEXPR_OP' || operatorName(expr) !== '/')
+        return false;
+    return numericLiteralValue(expr.rexpr) === 1000;
+}
 function startsWithExpr(func, context) {
     const args = func.args ?? [];
     if (args.length < 2)
@@ -1130,6 +1142,14 @@ function postgresTimestampText(value) {
     const withSpace = raw.replace('T', ' ');
     return withSpace.endsWith('Z') ? `${withSpace.slice(0, -1)}+00` : withSpace;
 }
+function epochMillisParamValue(value) {
+    const millis = timestampMillisValue(value);
+    if (millis !== null)
+        return millis;
+    const raw = value instanceof Date ? value.toISOString() : String(value);
+    const date = new Date(raw);
+    return Number.isFinite(date.getTime()) ? date.getTime() : value;
+}
 function paramNumbersForOids(oids, matches) {
     const numbers = new Set();
     for (let index = 0; index < oids.length; index++) {
@@ -1750,8 +1770,12 @@ function rewriteNode(value, context) {
             return rewriteNode(value.FuncCall.args[0], context);
         }
         if (name === 'to_timestamp' && value.FuncCall.args?.[0]) {
+            const arg = value.FuncCall.args[0];
+            if (isDivisionByMillis(arg)) {
+                collectParamRefs(arg, (paramNumber) => context?.epochMillisParamNumbers?.add(paramNumber));
+            }
             return funcCallNode('datetime', [
-                rewriteNode(value.FuncCall.args[0], context),
+                rewriteNode(arg, context),
                 stringConst('unixepoch'),
             ]);
         }
@@ -2201,19 +2225,45 @@ function normalizeAlterTable(stmt) {
     const skipIfColumnMissingByCmd = new Map();
     const schemaColumnsByCmd = new Map();
     const metadataOnlySchemaColumns = [];
+    const syntheticStatements = [];
     for (const cmdNode of stmt.cmds ?? []) {
         const cmd = cmdNode.AlterTableCmd;
         if (!cmd)
             continue;
+        if (cmd.subtype === 'AT_AddConstraint') {
+            const constraint = cmd.def?.Constraint;
+            const contype = constraint?.contype;
+            if (table &&
+                (contype === 'CONSTR_UNIQUE' || contype === 'CONSTR_PRIMARY') &&
+                Array.isArray(constraint.keys)) {
+                const columns = constraint.keys.map(stringValue).filter(Boolean);
+                if (columns.length) {
+                    const name = constraint.conname ||
+                        `${table}_${columns.join('_')}_${contype === 'CONSTR_PRIMARY' ? 'pkey' : 'key'}`;
+                    syntheticStatements.push({
+                        sql: `CREATE UNIQUE INDEX IF NOT EXISTS ${quoteIdentifier(name)} ON ${quoteIdentifier(table)} (${columns.map(quoteIdentifier).join(', ')})`,
+                        isDDL: true,
+                    });
+                }
+            }
+            continue;
+        }
+        if (cmd.subtype === 'AT_DropConstraint') {
+            if (cmd.name) {
+                syntheticStatements.push({
+                    sql: `DROP INDEX IF EXISTS ${quoteIdentifier(cmd.name)}`,
+                    isDDL: true,
+                });
+            }
+            continue;
+        }
         if (cmd.subtype === 'AT_AlterColumnType') {
             const schemaColumn = schemaColumnForAlterColumnType(tableRef, cmd);
             if (schemaColumn)
                 metadataOnlySchemaColumns.push(schemaColumn);
             continue;
         }
-        if (cmd.subtype === 'AT_AddConstraint' ||
-            cmd.subtype === 'AT_DropConstraint' ||
-            cmd.subtype === 'AT_ReplicaIdentity' ||
+        if (cmd.subtype === 'AT_ReplicaIdentity' ||
             cmd.subtype?.startsWith('AT_AlterColumn') ||
             cmd.subtype === 'AT_ColumnDefault' ||
             cmd.subtype === 'AT_SetNotNull' ||
@@ -2244,6 +2294,7 @@ function normalizeAlterTable(stmt) {
         skipIfColumnMissingByCmd,
         schemaColumnsByCmd,
         metadataOnlySchemaColumns,
+        syntheticStatements,
     };
 }
 function normalizeRename(stmt) {
@@ -2927,11 +2978,47 @@ function sqliteTriggerEvents(events) {
 function sqliteTriggerName(base, event, eventCount) {
     return eventCount === 1 ? base : `${base}_${event.toLowerCase()}`;
 }
+function threadReplyCountTriggerStatements(node) {
+    const table = publicationTableRefForRangeVar(node.relation);
+    if (!table || table.table !== 'message')
+        return null;
+    const events = sqliteTriggerEvents(Number(node.events ?? 0));
+    if (events.length === 0)
+        return null;
+    return events.map((event) => {
+        const row = event === 'DELETE' ? 'OLD' : 'NEW';
+        const triggerName = sqliteTriggerName(node.trigname, event, events.length);
+        return {
+            sql: `CREATE TRIGGER IF NOT EXISTS ${quoteIdentifier(triggerName)}
+AFTER ${event} ON ${quoteIdentifier(table.table)}
+FOR EACH ROW
+WHEN ${row}."threadId" IS NOT NULL
+BEGIN
+  UPDATE "thread"
+  SET "replyCount" = min(11, (
+    SELECT COUNT(*)
+    FROM "message"
+    WHERE "threadId" = ${row}."threadId"
+      AND "deleted" = 0
+      AND "type" IS DISTINCT FROM 'draft'
+      AND "id" IS DISTINCT FROM (
+        SELECT "messageId" FROM "thread" WHERE "id" = ${row}."threadId"
+      )
+  ))
+  WHERE "id" = ${row}."threadId";
+END`,
+            isDDL: true,
+        };
+    });
+}
 function createTriggerStatements(node, context) {
     const fnName = functionName({ funcname: node.funcname });
     const fn = fnName ? context?.triggerFunctions?.get(fnName) : undefined;
     if (!fn)
         return null;
+    if (fnName === 'updatethreadreplycount') {
+        return threadReplyCountTriggerStatements(node);
+    }
     if (findKeywordOutsideQuotes(fn.body, 'TG_OP') >= 0 ||
         findKeywordOutsideQuotes(fn.body, 'TG_ARGV') >= 0 ||
         findKeywordOutsideQuotes(fn.body, 'TG_NAME') >= 0 ||
@@ -3055,13 +3142,16 @@ function rewriteParsedStatement(version, rawStmt, context) {
     if (nodeType === 'AlterTableStmt') {
         alterMetadata = normalizeAlterTable(node);
         if (!node.cmds?.length) {
-            return alterMetadata.metadataOnlySchemaColumns.length
-                ? {
+            const statements = [];
+            if (alterMetadata.metadataOnlySchemaColumns.length) {
+                statements.push({
                     sql: '',
                     isDDL: true,
                     schemaColumns: alterMetadata.metadataOnlySchemaColumns,
-                }
-                : null;
+                });
+            }
+            statements.push(...alterMetadata.syntheticStatements);
+            return statements.length ? statements : null;
         }
     }
     else if (nodeType === 'CreateStmt') {
@@ -3113,7 +3203,8 @@ function rewriteParsedStatement(version, rawStmt, context) {
     else {
         rewriteNode(node, context);
     }
-    if (nodeType === 'AlterTableStmt' && node.cmds.length > 1) {
+    if (nodeType === 'AlterTableStmt' &&
+        (node.cmds.length > 1 || alterMetadata?.syntheticStatements.length)) {
         const statements = [];
         if (alterMetadata?.metadataOnlySchemaColumns.length) {
             statements.push({
@@ -3122,6 +3213,7 @@ function rewriteParsedStatement(version, rawStmt, context) {
                 schemaColumns: alterMetadata.metadataOnlySchemaColumns,
             });
         }
+        statements.push(...(alterMetadata?.syntheticStatements ?? []));
         statements.push(...node.cmds.map((cmdNode) => {
             const singleStmt = {
                 [nodeType]: {
@@ -3141,6 +3233,11 @@ function rewriteParsedStatement(version, rawStmt, context) {
                 ...(context?.jsonParamNumbers?.size
                     ? { jsonParamNumbers: new Set(context.jsonParamNumbers) }
                     : null),
+                ...(context?.epochMillisParamNumbers?.size
+                    ? {
+                        epochMillisParamNumbers: new Set(context.epochMillisParamNumbers),
+                    }
+                    : null),
                 ...(cmdSchemaColumns?.length ? { schemaColumns: cmdSchemaColumns } : null),
                 ...(skipIfColumnExists ? { skipIfColumnExists } : null),
                 ...(skipIfColumnMissing ? { skipIfColumnMissing } : null),
@@ -3180,6 +3277,9 @@ function rewriteParsedStatement(version, rawStmt, context) {
         ...(context?.jsonParamNumbers?.size
             ? { jsonParamNumbers: new Set(context.jsonParamNumbers) }
             : null),
+        ...(context?.epochMillisParamNumbers?.size
+            ? { epochMillisParamNumbers: new Set(context.epochMillisParamNumbers) }
+            : null),
         ...(schemaColumns.length ? { schemaColumns } : null),
         ...(schemaMetadataChanges.length ? { schemaMetadataChanges } : null),
         ...(skipIfColumnExists ? { skipIfColumnExists } : null),
@@ -4803,6 +4903,7 @@ export class DoBackend {
             if (rewritten && !rewritten.startsWith('--')) {
                 const arrayParamNumbers = new Set();
                 const jsonParamNumbers = new Set(inferredJsonParamNumbers);
+                const epochMillisParamNumbers = new Set();
                 const schemaColumns = [];
                 const schemaMetadataChanges = [];
                 const publicationChanges = [];
@@ -4813,6 +4914,9 @@ export class DoBackend {
                     for (const number of statement.jsonParamNumbers ?? []) {
                         jsonParamNumbers.add(number);
                     }
+                    for (const number of statement.epochMillisParamNumbers ?? []) {
+                        epochMillisParamNumbers.add(number);
+                    }
                     schemaColumns.push(...(statement.schemaColumns ?? []));
                     schemaMetadataChanges.push(...(statement.schemaMetadataChanges ?? []));
                     publicationChanges.push(...(statement.publicationChanges ?? []));
@@ -4825,6 +4929,7 @@ export class DoBackend {
                     ...(arrayParamNumbers.size ? { arrayParamNumbers } : null),
                     ...(jsonParamNumbers.size ? { jsonParamNumbers } : null),
                     ...(timestampParamNumbers.size ? { timestampParamNumbers } : null),
+                    ...(epochMillisParamNumbers.size ? { epochMillisParamNumbers } : null),
                     ...(booleanParamNumbers.size ? { booleanParamNumbers } : null),
                     ...(schemaColumns.length ? { schemaColumns } : null),
                     ...(schemaMetadataChanges.length ? { schemaMetadataChanges } : null),
@@ -4841,6 +4946,13 @@ export class DoBackend {
                         ? { jsonParamNumbers: inferredJsonParamNumbers }
                         : null),
                     ...(timestampParamNumbers.size ? { timestampParamNumbers } : null),
+                    ...(rewrittenStatements.some((statement) => statement.epochMillisParamNumbers?.size)
+                        ? {
+                            epochMillisParamNumbers: new Set(rewrittenStatements.flatMap((statement) => [
+                                ...(statement.epochMillisParamNumbers ?? []),
+                            ])),
+                        }
+                        : null),
                     ...(booleanParamNumbers.size ? { booleanParamNumbers } : null),
                     schemaColumns: rewrittenStatements.flatMap((statement) => statement.schemaColumns ?? []),
                     schemaMetadataChanges: rewrittenStatements.flatMap((statement) => statement.schemaMetadataChanges ?? []),
@@ -4907,7 +5019,7 @@ export class DoBackend {
             return buildCommandComplete('SET');
         try {
             if (isCatalogQuery(sql)) {
-                const catalogSql = this.inlineParams(sql, portal.params, portal.arrayParamNumbers, portal.jsonParamNumbers, portal.timestampParamNumbers, portal.booleanParamNumbers);
+                const catalogSql = this.inlineParams(sql, portal.params, portal.arrayParamNumbers, portal.jsonParamNumbers, portal.timestampParamNumbers, portal.epochMillisParamNumbers, portal.booleanParamNumbers);
                 const result = await this.handleCatalogQuery(catalogSql);
                 return concat(buildRowDescription(result.fields), ...result.rows.map((r) => buildDataRow(r, result.fields)), buildCommandComplete(`SELECT ${result.rows.length}`));
             }
@@ -4918,7 +5030,7 @@ export class DoBackend {
                 await this.snapshotTransactionWrite(statement);
             const tracking = statement ? this.trackingForStatement(statement) : undefined;
             const execSql = tracking?.returningSQL ?? sql;
-            const bound = this.sqliteBoundSQL(execSql, portal.params, portal.arrayParamNumbers, portal.jsonParamNumbers, portal.timestampParamNumbers, portal.booleanParamNumbers);
+            const bound = this.sqliteBoundSQL(execSql, portal.params, portal.arrayParamNumbers, portal.jsonParamNumbers, portal.timestampParamNumbers, portal.epochMillisParamNumbers, portal.booleanParamNumbers);
             const result = await this.doExecResult(bound.sql, bound.params, tracking ? this.trackingRequest(tracking) : undefined);
             if (portal.schemaColumns?.length) {
                 await this.applyStatementMetadata([{ sql, schemaColumns: portal.schemaColumns }]);
@@ -5043,11 +5155,12 @@ export class DoBackend {
             return { rows: [] };
         }
         if (isCatalogQuery(rewritten)) {
-            const catalogSql = this.inlineStatementParams(rewritten, params, statements, inferredJsonParamNumbers, timestampParamNumbers, booleanParamNumbers);
+            const catalogSql = this.inlineStatementParams(rewritten, params, statements, inferredJsonParamNumbers, timestampParamNumbers, new Set(), booleanParamNumbers);
             return { rows: (await this.handleCatalogQuery(catalogSql)).rows };
         }
         const arrayParamNumbers = new Set();
         const jsonParamNumbers = new Set(inferredJsonParamNumbers);
+        const epochMillisParamNumbers = new Set();
         for (const statement of statements) {
             for (const number of statement.arrayParamNumbers ?? []) {
                 arrayParamNumbers.add(number);
@@ -5055,14 +5168,17 @@ export class DoBackend {
             for (const number of statement.jsonParamNumbers ?? []) {
                 jsonParamNumbers.add(number);
             }
+            for (const number of statement.epochMillisParamNumbers ?? []) {
+                epochMillisParamNumbers.add(number);
+            }
         }
-        const bound = this.sqliteBoundSQL(rewritten, params, arrayParamNumbers, jsonParamNumbers, timestampParamNumbers, booleanParamNumbers);
+        const bound = this.sqliteBoundSQL(rewritten, params, arrayParamNumbers, jsonParamNumbers, timestampParamNumbers, epochMillisParamNumbers, booleanParamNumbers);
         const statement = statements.length === 1 ? statements[0] : undefined;
         if (statement)
             await this.snapshotTransactionWrite(statement);
         const tracking = statement ? this.trackingForStatement(statement) : undefined;
         const execBound = tracking
-            ? this.sqliteBoundSQL(tracking.returningSQL, params, arrayParamNumbers, jsonParamNumbers, timestampParamNumbers, booleanParamNumbers)
+            ? this.sqliteBoundSQL(tracking.returningSQL, params, arrayParamNumbers, jsonParamNumbers, timestampParamNumbers, epochMillisParamNumbers, booleanParamNumbers)
             : bound;
         const result = await this.doExecResult(execBound.sql, execBound.params, tracking ? this.trackingRequest(tracking) : undefined);
         await this.applyStatementMetadata(statements);
@@ -5083,11 +5199,13 @@ export class DoBackend {
             return cached;
         const arrayParamNumbers = new Set();
         const jsonParamNumbers = new Set();
+        const epochMillisParamNumbers = new Set();
         const statements = rewriteSQLStatements(sql, {
             skippedFunctionNames: this.skippedFunctionNames,
             triggerFunctions: this.triggerFunctions,
             arrayParamNumbers,
             jsonParamNumbers,
+            epochMillisParamNumbers,
         });
         if (this.canCacheRewrite(statements)) {
             this.rememberRewrite(key, statements);
@@ -5118,14 +5236,16 @@ export class DoBackend {
     rewriteSQL(sql) {
         const arrayParamNumbers = new Set();
         const jsonParamNumbers = new Set();
+        const epochMillisParamNumbers = new Set();
         return rewriteSQL(sql, {
             skippedFunctionNames: this.skippedFunctionNames,
             triggerFunctions: this.triggerFunctions,
             arrayParamNumbers,
             jsonParamNumbers,
+            epochMillisParamNumbers,
         });
     }
-    inlineStatementParams(sql, params, statements, inferredJsonParamNumbers = new Set(), timestampParamNumbers = new Set(), booleanParamNumbers = new Set()) {
+    inlineStatementParams(sql, params, statements, inferredJsonParamNumbers = new Set(), timestampParamNumbers = new Set(), epochMillisParamNumbers = new Set(), booleanParamNumbers = new Set()) {
         if (!params?.length)
             return sql;
         const arrayParamNumbers = new Set();
@@ -5137,8 +5257,11 @@ export class DoBackend {
             for (const number of statement.jsonParamNumbers ?? []) {
                 jsonParamNumbers.add(number);
             }
+            for (const number of statement.epochMillisParamNumbers ?? []) {
+                epochMillisParamNumbers.add(number);
+            }
         }
-        return this.inlineParams(sql, params, arrayParamNumbers, jsonParamNumbers, timestampParamNumbers, booleanParamNumbers);
+        return this.inlineParams(sql, params, arrayParamNumbers, jsonParamNumbers, timestampParamNumbers, epochMillisParamNumbers, booleanParamNumbers);
     }
     async doExec(sql, params) {
         return (await this.doExecResult(sql, params)).rows;
@@ -5837,6 +5960,12 @@ export class DoBackend {
     sqlLiteral(val, options) {
         if (val === null || val === undefined)
             return 'NULL';
+        if (options?.pgEpochMillisAsNumber) {
+            const millis = epochMillisParamValue(val);
+            if (typeof millis === 'number')
+                return String(millis);
+            val = millis;
+        }
         if (options?.pgTimestampAsText)
             return `'${postgresTimestampText(val).replace(/'/g, "''")}'`;
         if (options?.pgJsonAsJson) {
@@ -5872,7 +6001,7 @@ export class DoBackend {
             return `'${val.replace(/'/g, "''")}'`;
         return String(val);
     }
-    inlineParams(sql, params, arrayParamNumbers = new Set(), jsonParamNumbers = new Set(), timestampParamNumbers = new Set(), booleanParamNumbers = new Set()) {
+    inlineParams(sql, params, arrayParamNumbers = new Set(), jsonParamNumbers = new Set(), timestampParamNumbers = new Set(), epochMillisParamNumbers = new Set(), booleanParamNumbers = new Set()) {
         let out = '';
         for (let i = 0; i < sql.length;) {
             const ch = sql[i];
@@ -5951,6 +6080,7 @@ export class DoBackend {
                             pgArrayAsJson: arrayParamNumbers.has(number),
                             pgJsonAsJson: jsonParamNumbers.has(number),
                             pgTimestampAsText: timestampParamNumbers.has(number),
+                            pgEpochMillisAsNumber: epochMillisParamNumbers.has(number),
                             pgBooleanAsInteger: booleanParamNumbers.has(number),
                         })
                         : sql.slice(i, end);
@@ -5965,6 +6095,8 @@ export class DoBackend {
     sqliteParamValue(value, options) {
         if (value === null || value === undefined)
             return null;
+        if (options?.pgEpochMillisAsNumber)
+            return epochMillisParamValue(value);
         if (options?.pgTimestampAsText)
             return postgresTimestampText(value);
         if (options?.pgJsonAsJson)
@@ -5991,7 +6123,7 @@ export class DoBackend {
             return JSON.stringify(value);
         return value;
     }
-    sqliteBoundSQL(sql, params, arrayParamNumbers = new Set(), jsonParamNumbers = new Set(), timestampParamNumbers = new Set(), booleanParamNumbers = new Set()) {
+    sqliteBoundSQL(sql, params, arrayParamNumbers = new Set(), jsonParamNumbers = new Set(), timestampParamNumbers = new Set(), epochMillisParamNumbers = new Set(), booleanParamNumbers = new Set()) {
         if (!params?.length)
             return { sql, params: [] };
         let out = '';
@@ -6073,6 +6205,7 @@ export class DoBackend {
                         pgArrayAsJson: arrayParamNumbers.has(number),
                         pgJsonAsJson: jsonParamNumbers.has(number),
                         pgTimestampAsText: timestampParamNumbers.has(number),
+                        pgEpochMillisAsNumber: epochMillisParamNumbers.has(number),
                         pgBooleanAsInteger: booleanParamNumbers.has(number),
                     }));
                 }