@crvouga/postgres-mem 1.0.0 → 1.1.0

package/dist/executor/plpgsql.d.ts

@@ -0,0 +1,69 @@
+import type { Expr, SelectStmt, TypeName } from "../ast/nodes.js";
+import type { FunctionData } from "../storage/database-state.js";
+import type { TypedValue } from "../types/value.js";
+import { type ExecEnv, type Relation } from "./relation.js";
+/**
+ * plpgsql-lite: DECLARE, nested BEGIN/EXCEPTION WHEN others, IF, CASE,
+ * assignment, RETURN / RETURN NEXT, PERFORM, FOR-IN-SELECT.
+ */
+export type PlStmt = {
+    kind: "assign";
+    target: string[];
+    expr: Expr;
+} | {
+    kind: "return_new";
+} | {
+    kind: "return_old";
+} | {
+    kind: "return_expr";
+    expr: Expr;
+} | {
+    kind: "return_empty";
+} | {
+    kind: "return_next";
+} | {
+    kind: "if";
+    branches: Array<{
+        cond: Expr;
+        body: PlStmt[];
+    }>;
+    elseBody: PlStmt[];
+} | {
+    kind: "case";
+    expr: Expr | null;
+    branches: Array<{
+        cond: Expr;
+        body: PlStmt[];
+    }>;
+    elseBody: PlStmt[];
+} | {
+    kind: "null";
+} | {
+    kind: "raise";
+    message: string;
+} | {
+    kind: "perform";
+    expr: Expr;
+} | {
+    kind: "block";
+    body: PlStmt[];
+    handler: PlStmt[] | null;
+} | {
+    kind: "for";
+    targets: string[];
+    query: SelectStmt;
+    body: PlStmt[];
+};
+export interface PlDecl {
+    name: string;
+    typeName: TypeName;
+    init: Expr | null;
+}
+export interface PlProgram {
+    decls: PlDecl[];
+    body: PlStmt[];
+}
+export declare function compilePlpgsql(raw: string): PlProgram;
+export declare function compileTriggerBody(raw: string): PlStmt[];
+export declare function callPlpgsqlScalar(env: ExecEnv, fn: FunctionData, args: TypedValue[]): TypedValue;
+export declare function callPlpgsqlSet(env: ExecEnv, fn: FunctionData, args: TypedValue[]): Relation;

package/dist/executor/select.d.ts

@@ -20,6 +20,8 @@ export declare function evalScalar(env: ExecEnv, scope: RowScope | null, e: Expr
 /** SQL three-valued predicate: null → false */
 export declare function evalPredicate(env: ExecEnv, scope: RowScope | null, e: Expr, extras?: ScopeExtras): boolean;
 export declare function resolveUserFunction(env: ExecEnv, name: string[], argCount: number): FunctionData | null;
+/** Prefer a UDF whose declared types accept the actual args via implicit cast. */
+export declare function resolveUserFunctionForArgs(env: ExecEnv, name: string[], args: TypedValue[]): FunctionData | null;
 export declare function callSqlFunctionScalar(env: ExecEnv, fn: FunctionData, args: TypedValue[]): TypedValue;
 export declare function callSqlFunctionSet(env: ExecEnv, fn: FunctionData, args: TypedValue[]): Relation;
 export declare function applyWith(env: ExecEnv, w: WithClause | null): ExecEnv;
@@ -27,7 +29,7 @@ interface Source {
     rel: Relation;
     rangeVars: Set<string>;
 }
-export declare function buildFrom(env: ExecEnv, items: FromItem[]): Source;
+export declare function buildFrom(env: ExecEnv, items: FromItem[], where?: Expr | null): Source;
 export interface SortSpec {
     colIdx: number;
     dir: "asc" | "desc";

package/dist/index.d.ts

@@ -23,6 +23,6 @@
  */
 import "./serialization/codec.js";
 export type { BindValue, JsValue, QueryRow } from "./api/bind.js";
-export { Database, type DatabaseOptions } from "./api/database.js";
+export { Database, type DatabaseOptions, type RegisterFunctionOptions, Snapshot } from "./api/database.js";
 export { type ResultSet, type RunResult, Statement } from "./api/statement.js";
 export { type ErrorCategory, PostgresError } from "./errors/error.js";