@prairielearn/postgres 4.0.1 → 4.1.1

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # @prairielearn/postgres
 
+## 4.1.1
+
+### Patch Changes
+
+- f571b40: Upgrade all JavaScript dependencies
+
+## 4.1.0
+
+### Minor Changes
+
+- 56a813a: Add `execute` and `executeRow` helpers that don't return the actual values from a query.
+
 ## 4.0.1
 
 ### Patch Changes

package/dist/default-pool.d.ts

@@ -53,6 +53,8 @@ export declare const runInTransactionAsync: <T>(fn: (client: import("pg").defaul
 /**
  * Executes a query with the specified parameters.
  *
+ * @deprecated Use {@link execute} instead.
+ *
  * Using the return value of this function directly is not recommended. Instead, use
  * {@link queryRows}, {@link queryRow}, or {@link queryOptionalRow}.
  */
@@ -61,7 +63,7 @@ export declare const queryAsync: (sql: string, params: QueryParams) => Promise<i
  * Executes a query with the specified parameters. Errors if the query does
  * not return exactly one row.
  *
- * @deprecated Use {@link queryRow} instead.
+ * @deprecated Use {@link executeRow} or {@link queryRow} instead.
  */
 export declare const queryOneRowAsync: (sql: string, params: QueryParams) => Promise<import("pg").default.QueryResult>;
 /**
@@ -158,6 +160,14 @@ export declare const callOptionalRow: {
     <Model extends import("zod").ZodTypeAny>(sql: string, model: Model): Promise<import("zod").TypeOf<Model> | null>;
     <Model extends import("zod").ZodTypeAny>(sql: string, params: any[], model: Model): Promise<import("zod").TypeOf<Model> | null>;
 };
+/**
+ * Executes a query with the specified parameters. Returns the number of rows affected.
+ */
+export declare const execute: (sql: string, params?: QueryParams) => Promise<number>;
+/**
+ * Executes a query with the specified parameter, and errors if the query doesn't return exactly one row.
+ */
+export declare const executeRow: (sql: string, params?: QueryParams) => Promise<void>;
 /**
  * Returns an {@link CursorIterator} that can be used to iterate over the
  * results of the query in batches, which is useful for large result sets.
@@ -169,7 +179,7 @@ export declare const queryCursor: <Model extends import("zod").ZodTypeAny>(sql:
  *
  * @param schema The schema name to use (can be "null" to unset the search path)
  */
-export declare const setSearchSchema: (schema: string) => Promise<void>;
+export declare const setSearchSchema: (schema: string | null) => Promise<void>;
 /**
  * Get the schema that is currently used for the search path.
  *

package/dist/default-pool.js

@@ -64,6 +64,8 @@ export const runInTransactionAsync = defaultPool.runInTransactionAsync.bind(defa
 /**
  * Executes a query with the specified parameters.
  *
+ * @deprecated Use {@link execute} instead.
+ *
  * Using the return value of this function directly is not recommended. Instead, use
  * {@link queryRows}, {@link queryRow}, or {@link queryOptionalRow}.
  */
@@ -72,7 +74,7 @@ export const queryAsync = defaultPool.queryAsync.bind(defaultPool);
  * Executes a query with the specified parameters. Errors if the query does
  * not return exactly one row.
  *
- * @deprecated Use {@link queryRow} instead.
+ * @deprecated Use {@link executeRow} or {@link queryRow} instead.
  */
 export const queryOneRowAsync = defaultPool.queryOneRowAsync.bind(defaultPool);
 /**
@@ -151,6 +153,14 @@ export const callRow = defaultPool.callRow.bind(defaultPool);
  * or a single row that conforms to the given Zod schema.
  */
 export const callOptionalRow = defaultPool.callOptionalRow.bind(defaultPool);
+/**
+ * Executes a query with the specified parameters. Returns the number of rows affected.
+ */
+export const execute = defaultPool.execute.bind(defaultPool);
+/**
+ * Executes a query with the specified parameter, and errors if the query doesn't return exactly one row.
+ */
+export const executeRow = defaultPool.executeRow.bind(defaultPool);
 /**
  * Returns an {@link CursorIterator} that can be used to iterate over the
  * results of the query in batches, which is useful for large result sets.

package/dist/default-pool.js.map

@@ -1 +1 @@
-{"version":3,"file":"default-pool.js","sourceRoot":"","sources":["../src/default-pool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAuB,YAAY,EAAoB,MAAM,WAAW,CAAC;AAEhF,MAAM,WAAW,GAAG,IAAI,YAAY,EAAE,CAAC;AACvC,OAAO,EAAE,WAAW,EAAyC,CAAC;AAE9D,2EAA2E;AAC3E,6EAA6E;AAC7E,sDAAsD;AACtD,EAAE;AACF,4EAA4E;AAC5E,6EAA6E;AAC7E,4EAA4E;AAC5E,sEAAsE;AAEtE,2EAA2E;AAC3E,2EAA2E;AAC3E,mDAAmD;AAEnD;;GAEG;AACH,MAAM,CAAC,MAAM,SAAS,GAAG,WAAW,CAAC,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACjE;;GAEG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG,WAAW,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACnE;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,cAAc,GAAG,WAAW,CAAC,cAAc,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAE3E;;GAEG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG,WAAW,CAAC,oBAAoB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAEvF;;;GAGG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAG,WAAW,CAAC,0BAA0B,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAEnG;;;GAGG;AACH,MAAM,CAAC,MAAM,gCAAgC,GAC3C,WAAW,CAAC,gCAAgC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACjE;;GAEG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG,WAAW,CAAC,uBAAuB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC7F,MAAM,CAAC,MAAM,qBAAqB,GAAG,WAAW,CAAC,qBAAqB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACzF;;;GAGG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,WAAW,CAAC,mBAAmB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACrF;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,WAAW,CAAC,qBAAqB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACzF;;;;;GAKG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG,WAAW,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACnE;;;;;GAKG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,WAAW,CAAC,gBAAgB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC/E;;;;;GAKG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAG,WAAW,CAAC,sBAAsB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC3F;;;;GAIG;AACH,MAAM,CAAC,MAAM,SAAS,GAAG,WAAW,CAAC,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACjE;;;;;GAKG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,WAAW,CAAC,eAAe,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC7E;;;;;GAKG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,WAAW,CAAC,qBAAqB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACzF;;GAEG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,WAAW,CAAC,mBAAmB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACrF;;;GAGG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,WAAW,CAAC,yBAAyB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACjG;;;GAGG;AACH,MAAM,CAAC,MAAM,+BAA+B,GAC1C,WAAW,CAAC,+BAA+B,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAChE;;;;;GAKG;AACH,MAAM,CAAC,MAAM,SAAS,GAAG,WAAW,CAAC,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACjE;;;;GAIG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC/D;;;;;GAKG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,WAAW,CAAC,gBAAgB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC/E;;;GAGG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC/D;;;GAGG;AACH,MAAM,CAAC,MAAM,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC7D;;;GAGG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,WAAW,CAAC,eAAe,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC7E;;;;GAIG;AACH,MAAM,CAAC,MAAM,WAAW,GAAG,WAAW,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACrE;;;;GAIG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,WAAW,CAAC,eAAe,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC7E;;;;GAIG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,WAAW,CAAC,eAAe,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC7E;;;;;GAKG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAG,WAAW,CAAC,0BAA0B,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC","sourcesContent":["import { type CursorIterator, PostgresPool, type QueryParams } from './pool.js';\n\nconst defaultPool = new PostgresPool();\nexport { defaultPool, type CursorIterator, type QueryParams };\n\n// We re-expose all functions from the default pool here to account for the\n// default case of a shared global pool of clients. If someone want to create\n// their own pool, we expose the `PostgresPool` class.\n//\n// Note that we explicitly bind all functions to `defaultPool`. This ensures\n// that they'll be invoked with the correct `this` context, specifically when\n// this module is imported as `import * as db from '...'` and that import is\n// subsequently transformed by Babel to `interopRequireWildcard(...)`.\n\n// VSCode currently doesn't allow for us to use `inheritDoc` to inherit the\n// documentation from the `PostgresPool` class. We mirror the documentation\n// here for *async methods* in VSCode intellisense.\n\n/**\n * Creates a new connection pool and attempts to connect to the database.\n */\nexport const initAsync = defaultPool.initAsync.bind(defaultPool);\n/**\n * Closes the connection pool.\n */\nexport const closeAsync = defaultPool.closeAsync.bind(defaultPool);\n/**\n * Gets a new client from the connection pool. The caller MUST call `release()` to\n * release the client, whether or not errors occurred while using\n * `client`. The client can call `done(truthy_value)` to force\n * destruction of the client, but this should not be used except in\n * unusual circumstances.\n */\nexport const getClientAsync = defaultPool.getClientAsync.bind(defaultPool);\n\n/**\n * Performs a query with the given client.\n */\nexport const queryWithClientAsync = defaultPool.queryWithClientAsync.bind(defaultPool);\n\n/**\n * Performs a query with the given client. Errors if the query returns more\n * than one row.\n */\nexport const queryWithClientOneRowAsync = defaultPool.queryWithClientOneRowAsync.bind(defaultPool);\n\n/**\n * Performs a query with the given client. Errors if the query returns more\n * than one row.\n */\nexport const queryWithClientZeroOrOneRowAsync =\n  defaultPool.queryWithClientZeroOrOneRowAsync.bind(defaultPool);\n/**\n * Rolls back the current transaction for the given client.\n */\nexport const rollbackWithClientAsync = defaultPool.rollbackWithClientAsync.bind(defaultPool);\nexport const beginTransactionAsync = defaultPool.beginTransactionAsync.bind(defaultPool);\n/**\n * Commits the transaction if err is null, otherwise rollbacks the transaction.\n * Also releases the client.\n */\nexport const endTransactionAsync = defaultPool.endTransactionAsync.bind(defaultPool);\n/**\n * Runs the specified function inside of a transaction. The function will\n * receive a database client as an argument, but it can also make queries\n * as usual, and the correct client will be used automatically.\n *\n * The transaction will be rolled back if the function throws an error, and\n * will be committed otherwise.\n */\nexport const runInTransactionAsync = defaultPool.runInTransactionAsync.bind(defaultPool);\n/**\n * Executes a query with the specified parameters.\n *\n * Using the return value of this function directly is not recommended. Instead, use\n * {@link queryRows}, {@link queryRow}, or {@link queryOptionalRow}.\n */\nexport const queryAsync = defaultPool.queryAsync.bind(defaultPool);\n/**\n * Executes a query with the specified parameters. Errors if the query does\n * not return exactly one row.\n *\n * @deprecated Use {@link queryRow} instead.\n */\nexport const queryOneRowAsync = defaultPool.queryOneRowAsync.bind(defaultPool);\n/**\n * Executes a query with the specified parameters. Errors if the query\n * returns more than one row.\n *\n * @deprecated Use {@link queryOptionalRow} instead.\n */\nexport const queryZeroOrOneRowAsync = defaultPool.queryZeroOrOneRowAsync.bind(defaultPool);\n/**\n * Calls the given sproc with the specified parameters.\n *\n * @deprecated Use {@link callRows} instead.\n */\nexport const callAsync = defaultPool.callAsync.bind(defaultPool);\n/**\n * Calls the given sproc with the specified parameters. Errors if the\n * sproc does not return exactly one row.\n *\n * @deprecated Use {@link callRow} instead.\n */\nexport const callOneRowAsync = defaultPool.callOneRowAsync.bind(defaultPool);\n/**\n * Calls the given sproc with the specified parameters. Errors if the\n * sproc returns more than one row.\n *\n * @deprecated Use {@link callOptionalRow} instead.\n */\nexport const callZeroOrOneRowAsync = defaultPool.callZeroOrOneRowAsync.bind(defaultPool);\n/**\n * Calls a sproc with the specified parameters using a specific client.\n */\nexport const callWithClientAsync = defaultPool.callWithClientAsync.bind(defaultPool);\n/**\n * Calls a sproc with the specified parameters using a specific client.\n * Errors if the sproc does not return exactly one row.\n */\nexport const callWithClientOneRowAsync = defaultPool.callWithClientOneRowAsync.bind(defaultPool);\n/**\n * Calls a sproc with the specified parameters using a specific client.\n * Errors if the sproc returns more than one row.\n */\nexport const callWithClientZeroOrOneRowAsync =\n  defaultPool.callWithClientZeroOrOneRowAsync.bind(defaultPool);\n/**\n * Executes a query with the specified parameters. Returns an array of rows\n * that conform to the given Zod schema.\n *\n * If the query returns a single column, the return value will be a list of column values.\n */\nexport const queryRows = defaultPool.queryRows.bind(defaultPool);\n/**\n * Executes a query with the specified parameters. Returns exactly one row that conforms to the given Zod schema.\n *\n * If the query returns a single column, the return value will be the column value itself.\n */\nexport const queryRow = defaultPool.queryRow.bind(defaultPool);\n/**\n * Executes a query with the specified parameters. Returns either null or a\n * single row that conforms to the given Zod schema, and errors otherwise.\n *\n * If the query returns a single column, the return value will be the column value itself.\n */\nexport const queryOptionalRow = defaultPool.queryOptionalRow.bind(defaultPool);\n/**\n * Calls the given sproc with the specified parameters.\n * Errors if the sproc does not return anything.\n */\nexport const callRows = defaultPool.callRows.bind(defaultPool);\n/**\n * Calls the given sproc with the specified parameters.\n * Returns exactly one row from the sproc that conforms to the given Zod schema.\n */\nexport const callRow = defaultPool.callRow.bind(defaultPool);\n/**\n * Calls the given sproc with the specified parameters. Returns either null\n * or a single row that conforms to the given Zod schema.\n */\nexport const callOptionalRow = defaultPool.callOptionalRow.bind(defaultPool);\n/**\n * Returns an {@link CursorIterator} that can be used to iterate over the\n * results of the query in batches, which is useful for large result sets.\n * Each row will be parsed by the given Zod schema.\n */\nexport const queryCursor = defaultPool.queryCursor.bind(defaultPool);\n/**\n * Set the schema to use for the search path.\n *\n * @param schema The schema name to use (can be \"null\" to unset the search path)\n */\nexport const setSearchSchema = defaultPool.setSearchSchema.bind(defaultPool);\n/**\n * Get the schema that is currently used for the search path.\n *\n * @returns schema in use (may be `null` to indicate no schema)\n */\nexport const getSearchSchema = defaultPool.getSearchSchema.bind(defaultPool);\n/**\n * Generate, set, and return a random schema name.\n *\n * @param prefix The prefix of the new schema, only the first 28 characters will be used (after lowercasing).\n * @returns The randomly-generated search schema.\n */\nexport const setRandomSearchSchemaAsync = defaultPool.setRandomSearchSchemaAsync.bind(defaultPool);\n"]}
+{"version":3,"file":"default-pool.js","sourceRoot":"","sources":["../src/default-pool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAuB,YAAY,EAAoB,MAAM,WAAW,CAAC;AAEhF,MAAM,WAAW,GAAG,IAAI,YAAY,EAAE,CAAC;AACvC,OAAO,EAAE,WAAW,EAAyC,CAAC;AAE9D,2EAA2E;AAC3E,6EAA6E;AAC7E,sDAAsD;AACtD,EAAE;AACF,4EAA4E;AAC5E,6EAA6E;AAC7E,4EAA4E;AAC5E,sEAAsE;AAEtE,2EAA2E;AAC3E,2EAA2E;AAC3E,mDAAmD;AAEnD;;GAEG;AACH,MAAM,CAAC,MAAM,SAAS,GAAG,WAAW,CAAC,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACjE;;GAEG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG,WAAW,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACnE;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,cAAc,GAAG,WAAW,CAAC,cAAc,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAE3E;;GAEG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG,WAAW,CAAC,oBAAoB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAEvF;;;GAGG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAG,WAAW,CAAC,0BAA0B,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAEnG;;;GAGG;AACH,MAAM,CAAC,MAAM,gCAAgC,GAC3C,WAAW,CAAC,gCAAgC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACjE;;GAEG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG,WAAW,CAAC,uBAAuB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC7F,MAAM,CAAC,MAAM,qBAAqB,GAAG,WAAW,CAAC,qBAAqB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACzF;;;GAGG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,WAAW,CAAC,mBAAmB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACrF;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,WAAW,CAAC,qBAAqB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACzF;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG,WAAW,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACnE;;;;;GAKG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,WAAW,CAAC,gBAAgB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC/E;;;;;GAKG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAG,WAAW,CAAC,sBAAsB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC3F;;;;GAIG;AACH,MAAM,CAAC,MAAM,SAAS,GAAG,WAAW,CAAC,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACjE;;;;;GAKG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,WAAW,CAAC,eAAe,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC7E;;;;;GAKG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,WAAW,CAAC,qBAAqB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACzF;;GAEG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,WAAW,CAAC,mBAAmB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACrF;;;GAGG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,WAAW,CAAC,yBAAyB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACjG;;;GAGG;AACH,MAAM,CAAC,MAAM,+BAA+B,GAC1C,WAAW,CAAC,+BAA+B,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAChE;;;;;GAKG;AACH,MAAM,CAAC,MAAM,SAAS,GAAG,WAAW,CAAC,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACjE;;;;GAIG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC/D;;;;;GAKG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,WAAW,CAAC,gBAAgB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC/E;;;GAGG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC/D;;;GAGG;AACH,MAAM,CAAC,MAAM,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC7D;;;GAGG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,WAAW,CAAC,eAAe,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAE7E;;GAEG;AACH,MAAM,CAAC,MAAM,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAE7D;;GAEG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG,WAAW,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAEnE;;;;GAIG;AACH,MAAM,CAAC,MAAM,WAAW,GAAG,WAAW,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACrE;;;;GAIG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,WAAW,CAAC,eAAe,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC7E;;;;GAIG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,WAAW,CAAC,eAAe,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AAC7E;;;;;GAKG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAG,WAAW,CAAC,0BAA0B,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC","sourcesContent":["import { type CursorIterator, PostgresPool, type QueryParams } from './pool.js';\n\nconst defaultPool = new PostgresPool();\nexport { defaultPool, type CursorIterator, type QueryParams };\n\n// We re-expose all functions from the default pool here to account for the\n// default case of a shared global pool of clients. If someone want to create\n// their own pool, we expose the `PostgresPool` class.\n//\n// Note that we explicitly bind all functions to `defaultPool`. This ensures\n// that they'll be invoked with the correct `this` context, specifically when\n// this module is imported as `import * as db from '...'` and that import is\n// subsequently transformed by Babel to `interopRequireWildcard(...)`.\n\n// VSCode currently doesn't allow for us to use `inheritDoc` to inherit the\n// documentation from the `PostgresPool` class. We mirror the documentation\n// here for *async methods* in VSCode intellisense.\n\n/**\n * Creates a new connection pool and attempts to connect to the database.\n */\nexport const initAsync = defaultPool.initAsync.bind(defaultPool);\n/**\n * Closes the connection pool.\n */\nexport const closeAsync = defaultPool.closeAsync.bind(defaultPool);\n/**\n * Gets a new client from the connection pool. The caller MUST call `release()` to\n * release the client, whether or not errors occurred while using\n * `client`. The client can call `done(truthy_value)` to force\n * destruction of the client, but this should not be used except in\n * unusual circumstances.\n */\nexport const getClientAsync = defaultPool.getClientAsync.bind(defaultPool);\n\n/**\n * Performs a query with the given client.\n */\nexport const queryWithClientAsync = defaultPool.queryWithClientAsync.bind(defaultPool);\n\n/**\n * Performs a query with the given client. Errors if the query returns more\n * than one row.\n */\nexport const queryWithClientOneRowAsync = defaultPool.queryWithClientOneRowAsync.bind(defaultPool);\n\n/**\n * Performs a query with the given client. Errors if the query returns more\n * than one row.\n */\nexport const queryWithClientZeroOrOneRowAsync =\n  defaultPool.queryWithClientZeroOrOneRowAsync.bind(defaultPool);\n/**\n * Rolls back the current transaction for the given client.\n */\nexport const rollbackWithClientAsync = defaultPool.rollbackWithClientAsync.bind(defaultPool);\nexport const beginTransactionAsync = defaultPool.beginTransactionAsync.bind(defaultPool);\n/**\n * Commits the transaction if err is null, otherwise rollbacks the transaction.\n * Also releases the client.\n */\nexport const endTransactionAsync = defaultPool.endTransactionAsync.bind(defaultPool);\n/**\n * Runs the specified function inside of a transaction. The function will\n * receive a database client as an argument, but it can also make queries\n * as usual, and the correct client will be used automatically.\n *\n * The transaction will be rolled back if the function throws an error, and\n * will be committed otherwise.\n */\nexport const runInTransactionAsync = defaultPool.runInTransactionAsync.bind(defaultPool);\n/**\n * Executes a query with the specified parameters.\n *\n * @deprecated Use {@link execute} instead.\n *\n * Using the return value of this function directly is not recommended. Instead, use\n * {@link queryRows}, {@link queryRow}, or {@link queryOptionalRow}.\n */\nexport const queryAsync = defaultPool.queryAsync.bind(defaultPool);\n/**\n * Executes a query with the specified parameters. Errors if the query does\n * not return exactly one row.\n *\n * @deprecated Use {@link executeRow} or {@link queryRow} instead.\n */\nexport const queryOneRowAsync = defaultPool.queryOneRowAsync.bind(defaultPool);\n/**\n * Executes a query with the specified parameters. Errors if the query\n * returns more than one row.\n *\n * @deprecated Use {@link queryOptionalRow} instead.\n */\nexport const queryZeroOrOneRowAsync = defaultPool.queryZeroOrOneRowAsync.bind(defaultPool);\n/**\n * Calls the given sproc with the specified parameters.\n *\n * @deprecated Use {@link callRows} instead.\n */\nexport const callAsync = defaultPool.callAsync.bind(defaultPool);\n/**\n * Calls the given sproc with the specified parameters. Errors if the\n * sproc does not return exactly one row.\n *\n * @deprecated Use {@link callRow} instead.\n */\nexport const callOneRowAsync = defaultPool.callOneRowAsync.bind(defaultPool);\n/**\n * Calls the given sproc with the specified parameters. Errors if the\n * sproc returns more than one row.\n *\n * @deprecated Use {@link callOptionalRow} instead.\n */\nexport const callZeroOrOneRowAsync = defaultPool.callZeroOrOneRowAsync.bind(defaultPool);\n/**\n * Calls a sproc with the specified parameters using a specific client.\n */\nexport const callWithClientAsync = defaultPool.callWithClientAsync.bind(defaultPool);\n/**\n * Calls a sproc with the specified parameters using a specific client.\n * Errors if the sproc does not return exactly one row.\n */\nexport const callWithClientOneRowAsync = defaultPool.callWithClientOneRowAsync.bind(defaultPool);\n/**\n * Calls a sproc with the specified parameters using a specific client.\n * Errors if the sproc returns more than one row.\n */\nexport const callWithClientZeroOrOneRowAsync =\n  defaultPool.callWithClientZeroOrOneRowAsync.bind(defaultPool);\n/**\n * Executes a query with the specified parameters. Returns an array of rows\n * that conform to the given Zod schema.\n *\n * If the query returns a single column, the return value will be a list of column values.\n */\nexport const queryRows = defaultPool.queryRows.bind(defaultPool);\n/**\n * Executes a query with the specified parameters. Returns exactly one row that conforms to the given Zod schema.\n *\n * If the query returns a single column, the return value will be the column value itself.\n */\nexport const queryRow = defaultPool.queryRow.bind(defaultPool);\n/**\n * Executes a query with the specified parameters. Returns either null or a\n * single row that conforms to the given Zod schema, and errors otherwise.\n *\n * If the query returns a single column, the return value will be the column value itself.\n */\nexport const queryOptionalRow = defaultPool.queryOptionalRow.bind(defaultPool);\n/**\n * Calls the given sproc with the specified parameters.\n * Errors if the sproc does not return anything.\n */\nexport const callRows = defaultPool.callRows.bind(defaultPool);\n/**\n * Calls the given sproc with the specified parameters.\n * Returns exactly one row from the sproc that conforms to the given Zod schema.\n */\nexport const callRow = defaultPool.callRow.bind(defaultPool);\n/**\n * Calls the given sproc with the specified parameters. Returns either null\n * or a single row that conforms to the given Zod schema.\n */\nexport const callOptionalRow = defaultPool.callOptionalRow.bind(defaultPool);\n\n/**\n * Executes a query with the specified parameters. Returns the number of rows affected.\n */\nexport const execute = defaultPool.execute.bind(defaultPool);\n\n/**\n * Executes a query with the specified parameter, and errors if the query doesn't return exactly one row.\n */\nexport const executeRow = defaultPool.executeRow.bind(defaultPool);\n\n/**\n * Returns an {@link CursorIterator} that can be used to iterate over the\n * results of the query in batches, which is useful for large result sets.\n * Each row will be parsed by the given Zod schema.\n */\nexport const queryCursor = defaultPool.queryCursor.bind(defaultPool);\n/**\n * Set the schema to use for the search path.\n *\n * @param schema The schema name to use (can be \"null\" to unset the search path)\n */\nexport const setSearchSchema = defaultPool.setSearchSchema.bind(defaultPool);\n/**\n * Get the schema that is currently used for the search path.\n *\n * @returns schema in use (may be `null` to indicate no schema)\n */\nexport const getSearchSchema = defaultPool.getSearchSchema.bind(defaultPool);\n/**\n * Generate, set, and return a random schema name.\n *\n * @param prefix The prefix of the new schema, only the first 28 characters will be used (after lowercasing).\n * @returns The randomly-generated search schema.\n */\nexport const setRandomSearchSchemaAsync = defaultPool.setRandomSearchSchemaAsync.bind(defaultPool);\n"]}

package/dist/error.js

@@ -6,7 +6,7 @@ export function formatQueryWithErrorPosition(query, position) {
     if (nextNewline < 0)
         nextNewline = query.length;
     const gap = ' '.repeat(Math.max(0, position - prevNewline - 1));
-    return (query.substring(0, nextNewline) +
+    return (query.slice(0, nextNewline) +
         '\n' +
         gap +
         '^\n' +
@@ -14,6 +14,6 @@ export function formatQueryWithErrorPosition(query, position) {
         '|\n' +
         gap +
         '+ ERROR POSITION SHOWN ABOVE\n' +
-        query.substring(nextNewline));
+        query.slice(nextNewline));
 }
 //# sourceMappingURL=error.js.map

package/dist/error.js.map

@@ -1 +1 @@
-{"version":3,"file":"error.js","sourceRoot":"","sources":["../src/error.ts"],"names":[],"mappings":"AAAA,MAAM,UAAU,4BAA4B,CAAC,KAAa,EAAE,QAAmC;IAC7F,IAAI,QAAQ,IAAI,IAAI;QAAE,OAAO,KAAK,CAAC;IAEnC,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,WAAW,CAAC,IAAI,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;IACvE,IAAI,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IAChD,IAAI,WAAW,GAAG,CAAC;QAAE,WAAW,GAAG,KAAK,CAAC,MAAM,CAAC;IAChD,MAAM,GAAG,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,GAAG,WAAW,GAAG,CAAC,CAAC,CAAC,CAAC;IAChE,OAAO,CACL,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,WAAW,CAAC;QAC/B,IAAI;QACJ,GAAG;QACH,KAAK;QACL,GAAG;QACH,KAAK;QACL,GAAG;QACH,gCAAgC;QAChC,KAAK,CAAC,SAAS,CAAC,WAAW,CAAC,CAC7B,CAAC;AACJ,CAAC","sourcesContent":["export function formatQueryWithErrorPosition(query: string, position: number | null | undefined) {\n  if (position == null) return query;\n\n  const prevNewline = Math.max(0, query.lastIndexOf('\\n', position) + 1);\n  let nextNewline = query.indexOf('\\n', position);\n  if (nextNewline < 0) nextNewline = query.length;\n  const gap = ' '.repeat(Math.max(0, position - prevNewline - 1));\n  return (\n    query.substring(0, nextNewline) +\n    '\\n' +\n    gap +\n    '^\\n' +\n    gap +\n    '|\\n' +\n    gap +\n    '+ ERROR POSITION SHOWN ABOVE\\n' +\n    query.substring(nextNewline)\n  );\n}\n"]}
+{"version":3,"file":"error.js","sourceRoot":"","sources":["../src/error.ts"],"names":[],"mappings":"AAAA,MAAM,UAAU,4BAA4B,CAAC,KAAa,EAAE,QAAmC;IAC7F,IAAI,QAAQ,IAAI,IAAI;QAAE,OAAO,KAAK,CAAC;IAEnC,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,WAAW,CAAC,IAAI,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;IACvE,IAAI,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IAChD,IAAI,WAAW,GAAG,CAAC;QAAE,WAAW,GAAG,KAAK,CAAC,MAAM,CAAC;IAChD,MAAM,GAAG,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,GAAG,WAAW,GAAG,CAAC,CAAC,CAAC,CAAC;IAChE,OAAO,CACL,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC;QAC3B,IAAI;QACJ,GAAG;QACH,KAAK;QACL,GAAG;QACH,KAAK;QACL,GAAG;QACH,gCAAgC;QAChC,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,CACzB,CAAC;AACJ,CAAC","sourcesContent":["export function formatQueryWithErrorPosition(query: string, position: number | null | undefined) {\n  if (position == null) return query;\n\n  const prevNewline = Math.max(0, query.lastIndexOf('\\n', position) + 1);\n  let nextNewline = query.indexOf('\\n', position);\n  if (nextNewline < 0) nextNewline = query.length;\n  const gap = ' '.repeat(Math.max(0, position - prevNewline - 1));\n  return (\n    query.slice(0, nextNewline) +\n    '\\n' +\n    gap +\n    '^\\n' +\n    gap +\n    '|\\n' +\n    gap +\n    '+ ERROR POSITION SHOWN ABOVE\\n' +\n    query.slice(nextNewline)\n  );\n}\n"]}

package/dist/pool.d.ts

@@ -80,11 +80,18 @@ export declare class PostgresPool {
     runInTransactionAsync<T>(fn: (client: pg.PoolClient) => Promise<T>): Promise<T>;
     /**
      * Executes a query with the specified parameters.
+     *
+     * @deprecated Use {@link execute} instead.
+     *
+     * Using the return value of this function directly is not recommended. Instead, use
+     * {@link queryRows}, {@link queryRow}, or {@link queryOptionalRow}.
      */
     queryAsync(sql: string, params: QueryParams): Promise<QueryResult>;
     /**
      * Executes a query with the specified parameters. Errors if the query does
      * not return exactly one row.
+     *
+     * @deprecated Use {@link executeRow} or {@link queryRow} instead.
      */
     queryOneRowAsync(sql: string, params: QueryParams): Promise<pg.QueryResult>;
     /**
@@ -132,6 +139,14 @@ export declare class PostgresPool {
     callRow<Model extends z.ZodTypeAny>(sql: string, params: any[], model: Model): Promise<z.infer<Model>>;
     callOptionalRow<Model extends z.ZodTypeAny>(sql: string, model: Model): Promise<z.infer<Model> | null>;
     callOptionalRow<Model extends z.ZodTypeAny>(sql: string, params: any[], model: Model): Promise<z.infer<Model> | null>;
+    /**
+     * Executes a query with the specified parameters. Returns the number of rows affected.
+     */
+    execute(sql: string, params?: QueryParams): Promise<number>;
+    /**
+     * Executes a query with the specified parameter, and errors if the query doesn't return exactly one row.
+     */
+    executeRow(sql: string, params?: QueryParams): Promise<void>;
     /**
      * Returns a {@link Cursor} for the given query. The cursor can be used to
      * read results in batches, which is useful for large result sets.
@@ -149,7 +164,7 @@ export declare class PostgresPool {
      *
      * @param schema The schema name to use (can be "null" to unset the search path)
      */
-    setSearchSchema(schema: string): Promise<void>;
+    setSearchSchema(schema: string | null): Promise<void>;
     /**
      * Get the schema that is currently used for the search path.
      *

package/dist/pool.js

@@ -3,16 +3,15 @@ import { Readable, Transform } from 'node:stream';
 import debugfn from 'debug';
 import _ from 'lodash';
 import multipipe from 'multipipe';
-import pg, {} from 'pg';
+import pg, { DatabaseError } from 'pg';
 import Cursor from 'pg-cursor';
-import { DatabaseError } from 'pg-protocol';
 import { z } from 'zod';
 const debug = debugfn('@prairielearn/postgres');
 const lastQueryMap = new WeakMap();
 const searchSchemaMap = new WeakMap();
 function addDataToError(err, data) {
     err.data = {
-        ...(err.data ?? {}),
+        ...err.data,
         ...data,
     };
     return err;
@@ -31,9 +30,9 @@ export class PostgresError extends Error {
 function debugString(s) {
     if (typeof s !== 'string')
         return 'NOT A STRING';
-    s = s.replace(/\n/g, '\\n');
+    s = s.replaceAll('\n', '\\n');
     if (s.length > 78)
-        s = s.substring(0, 75) + '...';
+        s = s.slice(0, 75) + '...';
     s = '"' + s + '"';
     return s;
 }
@@ -89,14 +88,14 @@ function paramsToArray(sql, params, errorOnUnusedParameters) {
                 paramsArray.push(params[v]);
             }
         }
-        processedSql += remainingSql.substring(0, result.index) + map[v];
-        remainingSql = remainingSql.substring(result.index + result[0].length);
+        processedSql += remainingSql.slice(0, result.index) + map[v];
+        remainingSql = remainingSql.slice(result.index + result[0].length);
     }
     processedSql += remainingSql;
     remainingSql = '';
     if (errorOnUnusedParameters) {
         const difference = _.difference(Object.keys(params), Object.keys(map));
-        if (difference.length) {
+        if (difference.length > 0) {
             throw new Error(`Unused parameters in SQL query: ${JSON.stringify(difference)}`);
         }
     }
@@ -402,6 +401,11 @@ export class PostgresPool {
     }
     /**
      * Executes a query with the specified parameters.
+     *
+     * @deprecated Use {@link execute} instead.
+     *
+     * Using the return value of this function directly is not recommended. Instead, use
+     * {@link queryRows}, {@link queryRow}, or {@link queryOptionalRow}.
      */
     async queryAsync(sql, params) {
         debug('query()', 'sql:', debugString(sql));
@@ -420,6 +424,8 @@ export class PostgresPool {
     /**
      * Executes a query with the specified parameters. Errors if the query does
      * not return exactly one row.
+     *
+     * @deprecated Use {@link executeRow} or {@link queryRow} instead.
      */
     async queryOneRowAsync(sql, params) {
         debug('queryOneRow()', 'sql:', debugString(sql));
@@ -457,7 +463,7 @@ export class PostgresPool {
     async callAsync(functionName, params) {
         debug('call()', 'function:', functionName);
         debug('call()', 'params:', debugParams(params));
-        const placeholders = params.map((_, v) => '$' + (v + 1)).join();
+        const placeholders = params.map((_, v) => '$' + (v + 1)).join(',');
         const sql = `SELECT * FROM ${escapeIdentifier(functionName)}(${placeholders});`;
         const result = await this.queryAsync(sql, params);
         debug('call() success', 'rowCount:', result.rowCount);
@@ -503,7 +509,7 @@ export class PostgresPool {
     async callWithClientAsync(client, functionName, params) {
         debug('callWithClient()', 'function:', functionName);
         debug('callWithClient()', 'params:', debugParams(params));
-        const placeholders = params.map((_, v) => '$' + (v + 1)).join();
+        const placeholders = params.map((_, v) => '$' + (v + 1)).join(',');
         const sql = `SELECT * FROM ${escapeIdentifier(functionName)}(${placeholders})`;
         const result = await this.queryWithClientAsync(client, sql, params);
         debug('callWithClient() success', 'rowCount:', result.rowCount);
@@ -652,6 +658,25 @@ export class PostgresPool {
             return model.parse(results.rows[0]);
         }
     }
+    /**
+     * Executes a query with the specified parameters. Returns the number of rows affected.
+     */
+    async execute(sql, params = {}) {
+        const result = await this.queryAsync(sql, params);
+        return result.rowCount ?? 0;
+    }
+    /**
+     * Executes a query with the specified parameter, and errors if the query doesn't return exactly one row.
+     */
+    async executeRow(sql, params = {}) {
+        const rowCount = await this.execute(sql, params);
+        if (rowCount !== 1) {
+            throw new PostgresError('Incorrect rowCount: ' + rowCount, {
+                sql,
+                sqlParams: params,
+            });
+        }
+    }
     /**
      * Returns a {@link Cursor} for the given query. The cursor can be used to
      * read results in batches, which is useful for large result sets.
@@ -768,7 +793,7 @@ export class PostgresPool {
      */
     async setRandomSearchSchemaAsync(prefix) {
         // truncated prefix (max 28 characters)
-        const truncPrefix = prefix.substring(0, 28);
+        const truncPrefix = prefix.slice(0, 28);
         // timestamp in format YYYY-MM-DDTHH:MM:SS.SSSZ (guaranteed to not exceed 27 characters in the spec)
         const timestamp = new Date().toISOString();
         // random 6-character suffix to avoid clashes (approx 2 billion possible values)

package/dist/pool.js.map

@@ -1 +1 @@
-{"version":3,"file":"pool.js","sourceRoot":"","sources":["../src/pool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAElD,OAAO,OAAO,MAAM,OAAO,CAAC;AAC5B,OAAO,CAAC,MAAM,QAAQ,CAAC;AACvB,OAAO,SAAS,MAAM,WAAW,CAAC;AAClC,OAAO,EAAE,EAAE,EAAoB,MAAM,IAAI,CAAC;AAC1C,OAAO,MAAM,MAAM,WAAW,CAAC;AAC/B,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAaxB,MAAM,KAAK,GAAG,OAAO,CAAC,wBAAwB,CAAC,CAAC;AAChD,MAAM,YAAY,GAAG,IAAI,OAAO,EAAyB,CAAC;AAC1D,MAAM,eAAe,GAAG,IAAI,OAAO,EAAyB,CAAC;AAE7D,SAAS,cAAc,CAAC,GAAU,EAAE,IAAyB;IAC1D,GAAW,CAAC,IAAI,GAAG;QAClB,GAAG,CAAE,GAAW,CAAC,IAAI,IAAI,EAAE,CAAC;QAC5B,GAAG,IAAI;KACR,CAAC;IACF,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,OAAO,aAAc,SAAQ,KAAK;IAC/B,IAAI,CAAsB;IAEjC,YAAY,OAAe,EAAE,IAAyB;QACpD,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,IAAI,GAAG,eAAe,CAAC;IAC9B,CAAC;CACF;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,CAAS;IAC5B,IAAI,OAAO,CAAC,KAAK,QAAQ;QAAE,OAAO,cAAc,CAAC;IACjD,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAC5B,IAAI,CAAC,CAAC,MAAM,GAAG,EAAE;QAAE,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,CAAC;IAClD,CAAC,GAAG,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC;IAClB,OAAO,CAAC,CAAC;AACX,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,MAAmB;IACtC,IAAI,CAAC,CAAC;IACN,IAAI,CAAC;QACH,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAC7B,CAAC;IAAC,MAAM,CAAC;QACP,CAAC,GAAG,uBAAuB,CAAC;IAC9B,CAAC;IACD,OAAO,WAAW,CAAC,CAAC,CAAC,CAAC;AACxB,CAAC;AAED;;;GAGG;AACH,SAAS,aAAa,CACpB,GAAW,EACX,MAAmB,EACnB,uBAAgC;IAEhC,IAAI,OAAO,GAAG,KAAK,QAAQ;QAAE,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;IACrE,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,OAAO;YACL,YAAY,EAAE,GAAG;YACjB,WAAW,EAAE,MAAM;SACpB,CAAC;IACJ,CAAC;IACD,IAAI,MAAM,IAAI,IAAI,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QACjD,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IAED,MAAM,EAAE,GAAG,oBAAoB,CAAC;IAChC,IAAI,MAAM,CAAC;IACX,IAAI,YAAY,GAAG,EAAE,CAAC;IACtB,IAAI,YAAY,GAAG,GAAG,CAAC;IACvB,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,MAAM,GAAG,GAA2B,EAAE,CAAC;IACvC,IAAI,WAAW,GAAU,EAAE,CAAC;IAC5B,OAAO,CAAC,MAAM,GAAG,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACjD,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QACpB,IAAI,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAChB,IAAI,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC;gBAAE,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,EAAE,CAAC,CAAC;YAC/D,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC7B,GAAG,CAAC,CAAC,CAAC,GAAG,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC;gBACrF,OAAO,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;gBAC5B,WAAW,GAAG,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YAC9C,CAAC;iBAAM,CAAC;gBACN,OAAO,EAAE,CAAC;gBACV,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,OAAO,CAAC;gBACvB,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;QACD,YAAY,IAAI,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,MAAM,CAAC,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;QACjE,YAAY,GAAG,YAAY,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;IACzE,CAAC;IACD,YAAY,IAAI,YAAY,CAAC;IAC7B,YAAY,GAAG,EAAE,CAAC;IAClB,IAAI,uBAAuB,EAAE,CAAC;QAC5B,MAAM,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;QACvE,IAAI,UAAU,CAAC,MAAM,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,mCAAmC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QACnF,CAAC;IACH,CAAC;IACD,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC;AACvC,CAAC;AAED;;;GAGG;AACH,SAAS,gBAAgB,CAAC,UAAkB;IAC1C,sEAAsE;IACtE,mDAAmD;IACnD,sDAAsD;IACtD,kEAAkE;IAClE,OAAO,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;AAC1D,CAAC;AAED,SAAS,YAAY,CAAC,GAAU,EAAE,GAAW,EAAE,MAAmB;IAChE,qEAAqE;IACrE,eAAe;IACf,MAAM,QAAQ,GAAG,EAAE,GAAG,GAAG,EAAE,CAAC;IAE5B,4EAA4E;IAC5E,oBAAoB;IACpB,QAAQ,CAAC,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;IAE/B,MAAM,gBAAgB,GAAG,GAAG,YAAY,aAAa,IAAI,GAAG,CAAC,QAAQ,IAAI,IAAI,CAAC;IAE9E,OAAO,cAAc,CAAC,GAAG,EAAE;QACzB,QAAQ;QACR,2EAA2E;QAC3E,0EAA0E;QAC1E,eAAe;QACf,GAAG,EAAE,gBAAgB,CAAC,CAAC,CAAC,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,GAAG;QAC5E,SAAS,EAAE,MAAM;KAClB,CAAC,CAAC;AACL,CAAC;AAED,MAAM,OAAO,YAAY;IACvB,oDAAoD;IAC5C,IAAI,GAAmB,IAAI,CAAC;IACpC;;;;;OAKG;IACK,SAAS,GAAG,IAAI,iBAAiB,EAAiB,CAAC;IACnD,YAAY,GAAkB,IAAI,CAAC;IAC3C,gEAAgE;IACxD,WAAW,GAAG,CAAC,CAAC;IAChB,uBAAuB,GAAG,KAAK,CAAC;IAExC;;OAEG;IACH,KAAK,CAAC,SAAS,CACb,QAA4B,EAC5B,gBAA+D;QAE/D,IAAI,CAAC,uBAAuB,GAAG,QAAQ,CAAC,uBAAuB,IAAI,KAAK,CAAC;QACzE,IAAI,CAAC,IAAI,GAAG,IAAI,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,UAAU,GAAG,EAAE,MAAM;YACzC,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC3C,gBAAgB,CAAC,cAAc,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,MAAM,EAAE,EAAE;YACjC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBACzB,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBAC3C,gBAAgB,CAAC,cAAc,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;YAC/D,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE;YAChC,sEAAsE;YACtE,oEAAoE;YACpE,oEAAoE;YACpE,yEAAyE;YACzE,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,oEAAoE;QACpE,wCAAwC;QACxC,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,MAAM,aAAa,GAAG,CAAC,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;QACrD,OAAO,UAAU,IAAI,aAAa,CAAC,MAAM,EAAE,CAAC;YAC1C,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACzC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACjB,OAAO;YACT,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAClB,IAAI,UAAU,KAAK,aAAa,CAAC,MAAM,EAAE,CAAC;oBACxC,MAAM,IAAI,KAAK,CACb,uCAAuC,aAAa,CAAC,MAAM,cAAc,GAAG,CAAC,OAAO,EAAE,CACvF,CAAC;gBACJ,CAAC;gBAED,MAAM,OAAO,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;gBAC1C,UAAU,EAAE,CAAC;gBACb,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,IAAI,CAAC,IAAI;YAAE,OAAO;QACvB,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;QACtB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,cAAc;QAClB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;QACjD,CAAC;QAED,wEAAwE;QACxE,sBAAsB;QACtB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QAExE,yEAAyE;QACzE,sEAAsE;QACtE,sEAAsE;QACtE,wEAAwE;QACxE,8BAA8B;QAC9B,EAAE;QACF,8DAA8D;QAC9D,kEAAkE;QAClE,wEAAwE;QACxE,mEAAmE;QACnE,qEAAqE;QACrE,gDAAgD;QAChD,EAAE;QACF,4EAA4E;QAC5E,4EAA4E;QAC5E,kEAAkE;QAClE,yEAAyE;QACzE,+CAA+C;QAC/C,MAAM,kBAAkB,GAAG,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACvD,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,IAAI,kBAAkB,KAAK,IAAI,CAAC,YAAY,EAAE,CAAC;YAC1E,MAAM,gBAAgB,GAAG,sBAAsB,gBAAgB,CAAC,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC;YAC5F,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,gBAAgB,EAAE,EAAE,CAAC,CAAC;YAChE,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,CAAC,OAAO,EAAE,CAAC;gBACjB,MAAM,GAAG,CAAC;YACZ,CAAC;YACD,eAAe,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QACjD,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,oBAAoB,CACxB,MAAqB,EACrB,GAAW,EACX,MAAmB;QAEnB,IAAI,CAAC,WAAW,IAAI,CAAC,CAAC;QACtB,KAAK,CAAC,mBAAmB,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;QACrD,KAAK,CAAC,mBAAmB,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QAC3D,MAAM,EAAE,YAAY,EAAE,WAAW,EAAE,GAAG,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAC/F,IAAI,CAAC;YACH,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YACvC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;YAC7D,KAAK,CAAC,2BAA2B,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;YACjE,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,MAAM,YAAY,CAAC,GAAG,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,0BAA0B,CAC9B,MAAqB,EACrB,GAAW,EACX,MAAmB;QAEnB,KAAK,CAAC,yBAAyB,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;QAC3D,KAAK,CAAC,yBAAyB,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QACjE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACpE,IAAI,MAAM,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,aAAa,CAAC,uBAAuB,MAAM,CAAC,QAAQ,EAAE,EAAE;gBAChE,GAAG;gBACH,SAAS,EAAE,MAAM;gBACjB,MAAM;aACP,CAAC,CAAC;QACL,CAAC;QACD,KAAK,CAAC,iCAAiC,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QACvE,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,gCAAgC,CACpC,MAAqB,EACrB,GAAW,EACX,MAAmB;QAEnB,KAAK,CAAC,+BAA+B,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;QACjE,KAAK,CAAC,+BAA+B,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QACvE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACpE,IAAI,MAAM,CAAC,QAAQ,IAAI,IAAI,IAAI,MAAM,CAAC,QAAQ,GAAG,CAAC,EAAE,CAAC;YACnD,MAAM,IAAI,aAAa,CAAC,uBAAuB,MAAM,CAAC,QAAQ,EAAE,EAAE;gBAChE,GAAG;gBACH,SAAS,EAAE,MAAM;gBACjB,MAAM;aACP,CAAC,CAAC;QACL,CAAC;QACD,KAAK,CAAC,uCAAuC,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC7E,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,uBAAuB,CAAC,MAAqB;QACjD,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC9B,uDAAuD;QACvD,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAC/B,sEAAsE;YACtE,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,KAAK,SAAS,EAAE,CAAC;gBAC5C,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,CAAC;QACH,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,8DAA8D;YAC9D,qEAAqE;YACrE,qEAAqE;YACrE,qEAAqE;YACrE,yBAAyB;YACzB,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,qBAAqB;QACzB,KAAK,CAAC,oBAAoB,CAAC,CAAC;QAC5B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC3C,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,oBAAoB,EAAE,EAAE,CAAC,CAAC;YAClE,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC,CAAC;YAC3C,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,mBAAmB,CAAC,MAAqB,EAAE,GAA6B;QAC5E,KAAK,CAAC,kBAAkB,CAAC,CAAC;QAC1B,IAAI,GAAG,EAAE,CAAC;YACR,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC,CAAC;YAC7C,CAAC;YAAC,OAAO,WAAgB,EAAE,CAAC;gBAC1B,MAAM,cAAc,CAAC,WAAW,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;YACxE,CAAC;YAED,qEAAqE;YACrE,2EAA2E;YAC3E,4EAA4E;YAC5E,MAAM,cAAc,CAAC,GAAG,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;QACrD,CAAC;aAAM,CAAC;YACN,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;YACxD,CAAC;oBAAS,CAAC;gBACT,0EAA0E;gBAC1E,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,KAAK,SAAS,EAAE,CAAC;oBAC5C,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,qBAAqB,CAAI,EAAyC;QACtE,kFAAkF;QAClF,mDAAmD;QACnD,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC;QACzC,MAAM,mBAAmB,GAAG,MAAM,KAAK,SAAS,CAAC;QACjD,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,MAAM,IAAI,CAAC,qBAAqB,EAAE,CAAC,CAAC;QAEzE,IAAI,MAAS,CAAC;QACd,IAAI,CAAC;YACH,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,iBAAiB,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,iBAAiB,CAAC,CAAC,CAAC;QACpF,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBACzB,2EAA2E;gBAC3E,uDAAuD;gBACvD,MAAM,IAAI,CAAC,mBAAmB,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC;YACzD,CAAC;YACD,MAAM,GAAG,CAAC;QACZ,CAAC;QAED,IAAI,CAAC,mBAAmB,EAAE,CAAC;YACzB,0EAA0E;YAC1E,6CAA6C;YAC7C,EAAE;YACF,yEAAyE;YACzE,wEAAwE;YACxE,iDAAiD;YACjD,MAAM,IAAI,CAAC,mBAAmB,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QAC1D,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,GAAW,EAAE,MAAmB;QAC/C,KAAK,CAAC,SAAS,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;QAC3C,KAAK,CAAC,SAAS,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QACjD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC3C,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QAC9D,CAAC;gBAAS,CAAC;YACT,qDAAqD;YACrD,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,KAAK,SAAS,EAAE,CAAC;gBAC5C,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,gBAAgB,CAAC,GAAW,EAAE,MAAmB;QACrD,KAAK,CAAC,eAAe,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;QACjD,KAAK,CAAC,eAAe,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QACvD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAClD,IAAI,MAAM,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,aAAa,CAAC,uBAAuB,MAAM,CAAC,QAAQ,EAAE,EAAE;gBAChE,GAAG;gBACH,SAAS,EAAE,MAAM;aAClB,CAAC,CAAC;QACL,CAAC;QACD,KAAK,CAAC,uBAAuB,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC7D,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,sBAAsB,CAAC,GAAW,EAAE,MAAmB;QAC3D,KAAK,CAAC,qBAAqB,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;QACvD,KAAK,CAAC,qBAAqB,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QAC7D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAClD,IAAI,MAAM,CAAC,QAAQ,IAAI,IAAI,IAAI,MAAM,CAAC,QAAQ,GAAG,CAAC,EAAE,CAAC;YACnD,MAAM,IAAI,aAAa,CAAC,uBAAuB,MAAM,CAAC,QAAQ,EAAE,EAAE;gBAChE,GAAG;gBACH,SAAS,EAAE,MAAM;aAClB,CAAC,CAAC;QACL,CAAC;QACD,KAAK,CAAC,6BAA6B,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QACnE,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,YAAoB,EAAE,MAAa;QACjD,KAAK,CAAC,QAAQ,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QAC3C,KAAK,CAAC,QAAQ,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QAChD,MAAM,YAAY,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAChE,MAAM,GAAG,GAAG,iBAAiB,gBAAgB,CAAC,YAAY,CAAC,IAAI,YAAY,IAAI,CAAC;QAChF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAClD,KAAK,CAAC,gBAAgB,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QACtD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,eAAe,CAAC,YAAoB,EAAE,MAAa;QACvD,KAAK,CAAC,cAAc,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QACjD,KAAK,CAAC,cAAc,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QACtD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QAC1D,IAAI,MAAM,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,aAAa,CAAC,sBAAsB,GAAG,MAAM,CAAC,QAAQ,EAAE;gBAChE,YAAY;gBACZ,SAAS,EAAE,MAAM;aAClB,CAAC,CAAC;QACL,CAAC;QACD,KAAK,CAAC,sBAAsB,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5D,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,qBAAqB,CAAC,YAAoB,EAAE,MAAa;QAC7D,KAAK,CAAC,oBAAoB,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QACvD,KAAK,CAAC,oBAAoB,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QAC5D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QAC1D,IAAI,MAAM,CAAC,QAAQ,IAAI,IAAI,IAAI,MAAM,CAAC,QAAQ,GAAG,CAAC,EAAE,CAAC;YACnD,MAAM,IAAI,aAAa,CAAC,sBAAsB,GAAG,MAAM,CAAC,QAAQ,EAAE;gBAChE,YAAY;gBACZ,SAAS,EAAE,MAAM;aAClB,CAAC,CAAC;QACL,CAAC;QACD,KAAK,CAAC,4BAA4B,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAClE,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,mBAAmB,CACvB,MAAqB,EACrB,YAAoB,EACpB,MAAa;QAEb,KAAK,CAAC,kBAAkB,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QACrD,KAAK,CAAC,kBAAkB,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QAC1D,MAAM,YAAY,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAChE,MAAM,GAAG,GAAG,iBAAiB,gBAAgB,CAAC,YAAY,CAAC,IAAI,YAAY,GAAG,CAAC;QAC/E,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACpE,KAAK,CAAC,0BAA0B,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAChE,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,yBAAyB,CAC7B,MAAqB,EACrB,YAAoB,EACpB,MAAa;QAEb,KAAK,CAAC,wBAAwB,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QAC3D,KAAK,CAAC,wBAAwB,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QAChE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;QAC5E,IAAI,MAAM,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,aAAa,CAAC,sBAAsB,GAAG,MAAM,CAAC,QAAQ,EAAE;gBAChE,YAAY;gBACZ,SAAS,EAAE,MAAM;aAClB,CAAC,CAAC;QACL,CAAC;QACD,KAAK,CAAC,gCAAgC,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QACtE,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,+BAA+B,CACnC,MAAqB,EACrB,YAAoB,EACpB,MAAa;QAEb,KAAK,CAAC,8BAA8B,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QACjE,KAAK,CAAC,8BAA8B,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QACtE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;QAC5E,IAAI,MAAM,CAAC,QAAQ,IAAI,IAAI,IAAI,MAAM,CAAC,QAAQ,GAAG,CAAC,EAAE,CAAC;YACnD,MAAM,IAAI,aAAa,CAAC,sBAAsB,GAAG,MAAM,CAAC,QAAQ,EAAE;gBAChE,YAAY;gBACZ,SAAS,EAAE,MAAM;aAClB,CAAC,CAAC;QACL,CAAC;QACD,KAAK,CAAC,sCAAsC,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5E,OAAO,MAAM,CAAC;IAChB,CAAC;IAQD;;;;;OAKG;IACH,KAAK,CAAC,SAAS,CACb,GAAW,EACX,cAAmC,EACnC,UAAkB;QAElB,MAAM,MAAM,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAE,cAA8B,CAAC;QAC/E,MAAM,KAAK,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAE,cAAwB,CAAC,CAAC,CAAC,UAAU,CAAC;QAChF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACnD,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC1C,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;YAC3D,OAAO,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;IAQD;;;;OAIG;IACH,KAAK,CAAC,QAAQ,CACZ,GAAW,EACX,cAAmC,EACnC,UAAkB;QAElB,MAAM,MAAM,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAE,cAA8B,CAAC;QAC/E,MAAM,KAAK,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAE,cAAwB,CAAC,CAAC,CAAC,UAAU,CAAC;QAChF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACzD,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC1C,OAAO,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAClD,CAAC;aAAM,CAAC;YACN,OAAO,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAWD;;;;;OAKG;IACH,KAAK,CAAC,gBAAgB,CACpB,GAAW,EACX,cAAmC,EACnC,UAAkB;QAElB,MAAM,MAAM,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAE,cAA8B,CAAC;QAC/E,MAAM,KAAK,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAE,cAAwB,CAAC,CAAC,CAAC,UAAU,CAAC;QAChF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAC/D,IAAI,OAAO,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9B,OAAO,IAAI,CAAC;QACd,CAAC;aAAM,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvC,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC1C,OAAO,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAClD,CAAC;aAAM,CAAC;YACN,OAAO,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAQD;;;OAGG;IACH,KAAK,CAAC,QAAQ,CACZ,GAAW,EACX,cAA6B,EAC7B,UAAkB;QAElB,MAAM,MAAM,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAE,cAAwB,CAAC;QACzE,MAAM,KAAK,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAE,cAAwB,CAAC,CAAC,CAAC,UAAU,CAAC;QAChF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAClD,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC1C,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;YAC3D,OAAO,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;IAQD;;;OAGG;IACH,KAAK,CAAC,OAAO,CACX,GAAW,EACX,cAA6B,EAC7B,UAAkB;QAElB,MAAM,MAAM,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAE,cAAwB,CAAC;QACzE,MAAM,KAAK,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAE,cAAwB,CAAC,CAAC,CAAC,UAAU,CAAC;QAChF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACxD,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC1C,OAAO,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAClD,CAAC;aAAM,CAAC;YACN,OAAO,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAWD;;;OAGG;IACH,KAAK,CAAC,eAAe,CACnB,GAAW,EACX,cAA6B,EAC7B,UAAkB;QAElB,MAAM,MAAM,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAE,cAAwB,CAAC;QACzE,MAAM,KAAK,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAE,cAAwB,CAAC,CAAC,CAAC,UAAU,CAAC;QAChF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAC9D,IAAI,OAAO,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9B,OAAO,IAAI,CAAC;QACd,CAAC;aAAM,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvC,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC1C,OAAO,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAClD,CAAC;aAAM,CAAC;YACN,OAAO,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,qBAAqB,CACjC,MAAqB,EACrB,GAAW,EACX,MAAmB;QAEnB,IAAI,CAAC,WAAW,IAAI,CAAC,CAAC;QACtB,KAAK,CAAC,yBAAyB,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;QAC3D,KAAK,CAAC,yBAAyB,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QACjE,MAAM,EAAE,YAAY,EAAE,WAAW,EAAE,GAAG,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAC/F,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QACvC,OAAO,MAAM,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC,CAAC;IAC7D,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,WAAW,CACf,GAAW,EACX,MAAmB,EACnB,KAAY;QAEZ,OAAO,IAAI,CAAC,mBAAmB,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;IACtD,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAC/B,GAAW,EACX,MAAmB,EACnB,KAAa;QAEb,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC3C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QAErE,IAAI,aAAa,GAAG,KAAK,CAAC;QAC1B,MAAM,QAAQ,GAAmC;YAC/C,KAAK,CAAC,CAAC,OAAO,CAAC,SAAiB;gBAC9B,iEAAiE;gBACjE,oCAAoC;gBACpC,IAAI,aAAa,EAAE,CAAC;oBAClB,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;gBACrD,CAAC;gBACD,aAAa,GAAG,IAAI,CAAC;gBAErB,IAAI,CAAC;oBACH,OAAO,IAAI,EAAE,CAAC;wBACZ,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;wBAC1C,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;4BACtB,MAAM;wBACR,CAAC;wBAED,IAAI,KAAK,EAAE,CAAC;4BACV,MAAM,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBACnC,CAAC;6BAAM,CAAC;4BACN,MAAM,IAAI,CAAC;wBACb,CAAC;oBACH,CAAC;gBACH,CAAC;gBAAC,OAAO,GAAQ,EAAE,CAAC;oBAClB,MAAM,YAAY,CAAC,GAAG,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;gBACvC,CAAC;wBAAS,CAAC;oBACT,IAAI,CAAC;wBACH,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;oBACvB,CAAC;4BAAS,CAAC;wBACT,MAAM,CAAC,OAAO,EAAE,CAAC;oBACnB,CAAC;gBACH,CAAC;YACH,CAAC;YACD,MAAM,CAAC,SAAiB;gBACtB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC;oBAC9B,kBAAkB,EAAE,IAAI;oBACxB,kBAAkB,EAAE,IAAI;oBACxB,SAAS,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ;wBAClC,KAAK,MAAM,GAAG,IAAI,KAAK,EAAE,CAAC;4BACxB,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;wBACjB,CAAC;wBACD,QAAQ,EAAE,CAAC;oBACb,CAAC;iBACF,CAAC,CAAC;gBAEH,2DAA2D;gBAC3D,MAAM,SAAS,GAAG,QAAQ,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBAC9C,MAAM,IAAI,GAAG,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,CAAC,CAAC;gBAE5D,sEAAsE;gBACtE,4EAA4E;gBAC5E,qEAAqE;gBACrE,uEAAuE;gBACvE,uEAAuE;gBACvE,uDAAuD;gBACvD,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE;oBACtB,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBACzB,CAAC,CAAC,CAAC;gBAEH,OAAO,IAAI,CAAC;YACd,CAAC;SACF,CAAC;QACF,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,eAAe,CAAC,MAAc;QAClC,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;YACnB,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC;YAC3B,OAAO;QACT,CAAC;QAED,MAAM,IAAI,CAAC,UAAU,CAAC,+BAA+B,gBAAgB,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QACrF,4FAA4F;QAC5F,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC;IAC7B,CAAC;IAED;;;;OAIG;IACH,eAAe;QACb,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,0BAA0B,CAAC,MAAc;QAC7C,uCAAuC;QACvC,MAAM,WAAW,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAC5C,oGAAoG;QACpG,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAC3C,gFAAgF;QAChF,MAAM,KAAK,GAAG,sCAAsC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;aACrC,GAAG,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;aAC1D,IAAI,CAAC,EAAE,CAAC,CAAC;QAEZ,0EAA0E;QAC1E,oDAAoD;QACpD,wFAAwF;QACxF,MAAM,MAAM,GAAG,GAAG,WAAW,IAAI,SAAS,IAAI,MAAM,EAAE,CAAC;QACvD,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QACnC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,6CAA6C;IAC7C,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,IAAI,EAAE,UAAU,IAAI,CAAC,CAAC;IACpC,CAAC;IAED,sCAAsC;IACtC,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,IAAI,EAAE,SAAS,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,0EAA0E;IAC1E,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,IAAI,EAAE,YAAY,IAAI,CAAC,CAAC;IACtC,CAAC;IAED,wEAAwE;IACxE,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;CACF","sourcesContent":["import { AsyncLocalStorage } from 'node:async_hooks';\nimport { Readable, Transform } from 'node:stream';\n\nimport debugfn from 'debug';\nimport _ from 'lodash';\nimport multipipe from 'multipipe';\nimport pg, { type QueryResult } from 'pg';\nimport Cursor from 'pg-cursor';\nimport { DatabaseError } from 'pg-protocol';\nimport { z } from 'zod';\n\nexport type QueryParams = Record<string, any> | any[];\n\nexport interface CursorIterator<T> {\n  iterate: (batchSize: number) => AsyncGenerator<T[]>;\n  stream: (batchSize: number) => NodeJS.ReadWriteStream;\n}\n\nexport interface PostgresPoolConfig extends pg.PoolConfig {\n  errorOnUnusedParameters?: boolean;\n}\n\nconst debug = debugfn('@prairielearn/postgres');\nconst lastQueryMap = new WeakMap<pg.PoolClient, string>();\nconst searchSchemaMap = new WeakMap<pg.PoolClient, string>();\n\nfunction addDataToError(err: Error, data: Record<string, any>): Error {\n  (err as any).data = {\n    ...((err as any).data ?? {}),\n    ...data,\n  };\n  return err;\n}\n\nexport class PostgresError extends Error {\n  public data: Record<string, any>;\n\n  constructor(message: string, data: Record<string, any>) {\n    super(message);\n    this.data = data;\n    this.name = 'PostgresError';\n  }\n}\n\n/**\n * Formats a string for debugging.\n */\nfunction debugString(s: string): string {\n  if (typeof s !== 'string') return 'NOT A STRING';\n  s = s.replace(/\\n/g, '\\\\n');\n  if (s.length > 78) s = s.substring(0, 75) + '...';\n  s = '\"' + s + '\"';\n  return s;\n}\n\n/**\n * Formats a set of params for debugging.\n */\nfunction debugParams(params: QueryParams): string {\n  let s;\n  try {\n    s = JSON.stringify(params);\n  } catch {\n    s = 'CANNOT JSON STRINGIFY';\n  }\n  return debugString(s);\n}\n\n/**\n * Given an SQL string and params, creates an array of params and an SQL string\n * with any named dollar-sign placeholders replaced with parameters.\n */\nfunction paramsToArray(\n  sql: string,\n  params: QueryParams,\n  errorOnUnusedParameters: boolean,\n): { processedSql: string; paramsArray: any } {\n  if (typeof sql !== 'string') throw new Error('SQL must be a string');\n  if (Array.isArray(params)) {\n    return {\n      processedSql: sql,\n      paramsArray: params,\n    };\n  }\n  if (params == null || typeof params !== 'object') {\n    throw new Error('params must be array or object');\n  }\n\n  const re = /\\$([-_a-zA-Z0-9]+)/;\n  let result;\n  let processedSql = '';\n  let remainingSql = sql;\n  let nParams = 0;\n  const map: Record<string, string> = {};\n  let paramsArray: any[] = [];\n  while ((result = re.exec(remainingSql)) !== null) {\n    const v = result[1];\n    if (!(v in map)) {\n      if (!(v in params)) throw new Error(`Missing parameter: ${v}`);\n      if (Array.isArray(params[v])) {\n        map[v] = 'ARRAY[' + params[v].map((_, n) => '$' + (n + nParams + 1)).join(',') + ']';\n        nParams += params[v].length;\n        paramsArray = paramsArray.concat(params[v]);\n      } else {\n        nParams++;\n        map[v] = '$' + nParams;\n        paramsArray.push(params[v]);\n      }\n    }\n    processedSql += remainingSql.substring(0, result.index) + map[v];\n    remainingSql = remainingSql.substring(result.index + result[0].length);\n  }\n  processedSql += remainingSql;\n  remainingSql = '';\n  if (errorOnUnusedParameters) {\n    const difference = _.difference(Object.keys(params), Object.keys(map));\n    if (difference.length) {\n      throw new Error(`Unused parameters in SQL query: ${JSON.stringify(difference)}`);\n    }\n  }\n  return { processedSql, paramsArray };\n}\n\n/**\n * Escapes the given identifier for use in an SQL query. Useful for preventing\n * SQL injection.\n */\nfunction escapeIdentifier(identifier: string): string {\n  // Note that as of 2021-06-29 escapeIdentifier() is undocumented. See:\n  // https://github.com/brianc/node-postgres/pull/396\n  // https://github.com/brianc/node-postgres/issues/1978\n  // https://www.postgresql.org/docs/current/sql-syntax-lexical.html\n  return pg.Client.prototype.escapeIdentifier(identifier);\n}\n\nfunction enhanceError(err: Error, sql: string, params: QueryParams): Error {\n  // Copy the error so we don't end up with a circular reference in the\n  // final error.\n  const sqlError = { ...err };\n\n  // `message` is a non-enumerable property, so we need to copy it manually to\n  // the error object.\n  sqlError.message = err.message;\n\n  const errorHasPosition = err instanceof DatabaseError && err.position != null;\n\n  return addDataToError(err, {\n    sqlError,\n    // If the error has a `position` field, we need to use the processed source\n    // (where e.g. `$foobar` has been replaced with `$1`) so that the position\n    // is accurate.\n    sql: errorHasPosition ? paramsToArray(sql, params, false).processedSql : sql,\n    sqlParams: params,\n  });\n}\n\nexport class PostgresPool {\n  /** The pool from which clients will be acquired. */\n  private pool: pg.Pool | null = null;\n  /**\n   * We use this to propagate the client associated with the current transaction\n   * to any nested queries. In the past, we had some nasty bugs associated with\n   * the fact that we tried to acquire new clients inside of transactions, which\n   * ultimately lead to a deadlock.\n   */\n  private alsClient = new AsyncLocalStorage<pg.PoolClient>();\n  private searchSchema: string | null = null;\n  /** Tracks the total number of queries executed by this pool. */\n  private _queryCount = 0;\n  private errorOnUnusedParameters = false;\n\n  /**\n   * Creates a new connection pool and attempts to connect to the database.\n   */\n  async initAsync(\n    pgConfig: PostgresPoolConfig,\n    idleErrorHandler: (error: Error, client: pg.PoolClient) => void,\n  ): Promise<void> {\n    this.errorOnUnusedParameters = pgConfig.errorOnUnusedParameters ?? false;\n    this.pool = new pg.Pool(pgConfig);\n    this.pool.on('error', function (err, client) {\n      const lastQuery = lastQueryMap.get(client);\n      idleErrorHandler(addDataToError(err, { lastQuery }), client);\n    });\n    this.pool.on('connect', (client) => {\n      client.on('error', (err) => {\n        const lastQuery = lastQueryMap.get(client);\n        idleErrorHandler(addDataToError(err, { lastQuery }), client);\n      });\n    });\n    this.pool.on('remove', (client) => {\n      // This shouldn't be necessary, as `pg` currently allows clients to be\n      // garbage collected after they're removed. However, if `pg` someday\n      // starts reusing client objects across difference connections, this\n      // will ensure that we re-set the search path when the client reconnects.\n      searchSchemaMap.delete(client);\n    });\n\n    // Attempt to connect to the database so that we can fail quickly if\n    // something isn't configured correctly.\n    let retryCount = 0;\n    const retryTimeouts = [500, 1000, 2000, 5000, 10000];\n    while (retryCount <= retryTimeouts.length) {\n      try {\n        const client = await this.pool.connect();\n        client.release();\n        return;\n      } catch (err: any) {\n        if (retryCount === retryTimeouts.length) {\n          throw new Error(\n            `Could not connect to Postgres after ${retryTimeouts.length} attempts: ${err.message}`,\n          );\n        }\n\n        const timeout = retryTimeouts[retryCount];\n        retryCount++;\n        await new Promise((resolve) => setTimeout(resolve, timeout));\n      }\n    }\n  }\n\n  /**\n   * Closes the connection pool.\n   */\n  async closeAsync(): Promise<void> {\n    if (!this.pool) return;\n    await this.pool.end();\n    this.pool = null;\n  }\n\n  /**\n   * Gets a new client from the connection pool. The caller MUST call `release()` to\n   * release the client, whether or not errors occurred while using\n   * `client`. The client can call `done(truthy_value)` to force\n   * destruction of the client, but this should not be used except in\n   * unusual circumstances.\n   */\n  async getClientAsync(): Promise<pg.PoolClient> {\n    if (!this.pool) {\n      throw new Error('Connection pool is not open');\n    }\n\n    // If we're inside a transaction, we'll reuse the same client to avoid a\n    // potential deadlock.\n    const client = this.alsClient.getStore() ?? (await this.pool.connect());\n\n    // If we're configured to use a particular schema, we'll store whether or\n    // not the search path has already been configured for this particular\n    // client. If we acquire a client and it's already had its search path\n    // set, we can avoid setting it again since the search path will persist\n    // for the life of the client.\n    //\n    // We do this check for each call to `getClient` instead of on\n    // `pool.connect` so that we don't have to be really careful about\n    // destroying old clients that were created before `setSearchSchema` was\n    // called. Instead, we'll just check if the search path matches the\n    // currently-desired schema, and if it's a mismatch (or doesn't exist\n    // at all), we re-set it for the current client.\n    //\n    // Note that this accidentally supports changing the search_path on the fly,\n    // although that's not something we currently do (or would be likely to do).\n    // It does NOT support clearing the existing search schema - e.g.,\n    // `setSearchSchema(null)` would not work as you expect. This is fine, as\n    // that's not something we ever do in practice.\n    const clientSearchSchema = searchSchemaMap.get(client);\n    if (this.searchSchema != null && clientSearchSchema !== this.searchSchema) {\n      const setSearchPathSql = `SET search_path TO ${escapeIdentifier(this.searchSchema)},public`;\n      try {\n        await this.queryWithClientAsync(client, setSearchPathSql, {});\n      } catch (err) {\n        client.release();\n        throw err;\n      }\n      searchSchemaMap.set(client, this.searchSchema);\n    }\n\n    return client;\n  }\n\n  /**\n   * Performs a query with the given client.\n   */\n  async queryWithClientAsync(\n    client: pg.PoolClient,\n    sql: string,\n    params: QueryParams,\n  ): Promise<pg.QueryResult> {\n    this._queryCount += 1;\n    debug('queryWithClient()', 'sql:', debugString(sql));\n    debug('queryWithClient()', 'params:', debugParams(params));\n    const { processedSql, paramsArray } = paramsToArray(sql, params, this.errorOnUnusedParameters);\n    try {\n      lastQueryMap.set(client, processedSql);\n      const result = await client.query(processedSql, paramsArray);\n      debug('queryWithClient() success', 'rowCount:', result.rowCount);\n      return result;\n    } catch (err: any) {\n      throw enhanceError(err, sql, params);\n    }\n  }\n\n  /**\n   * Performs a query with the given client. Errors if the query returns more\n   * than one row.\n   */\n  async queryWithClientOneRowAsync(\n    client: pg.PoolClient,\n    sql: string,\n    params: QueryParams,\n  ): Promise<pg.QueryResult> {\n    debug('queryWithClientOneRow()', 'sql:', debugString(sql));\n    debug('queryWithClientOneRow()', 'params:', debugParams(params));\n    const result = await this.queryWithClientAsync(client, sql, params);\n    if (result.rowCount !== 1) {\n      throw new PostgresError(`Incorrect rowCount: ${result.rowCount}`, {\n        sql,\n        sqlParams: params,\n        result,\n      });\n    }\n    debug('queryWithClientOneRow() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  /**\n   * Performs a query with the given client. Errors if the query returns more\n   * than one row.\n   */\n  async queryWithClientZeroOrOneRowAsync(\n    client: pg.PoolClient,\n    sql: string,\n    params: QueryParams,\n  ): Promise<QueryResult> {\n    debug('queryWithClientZeroOrOneRow()', 'sql:', debugString(sql));\n    debug('queryWithClientZeroOrOneRow()', 'params:', debugParams(params));\n    const result = await this.queryWithClientAsync(client, sql, params);\n    if (result.rowCount == null || result.rowCount > 1) {\n      throw new PostgresError(`Incorrect rowCount: ${result.rowCount}`, {\n        sql,\n        sqlParams: params,\n        result,\n      });\n    }\n    debug('queryWithClientZeroOrOneRow() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  /**\n   * Rolls back the current transaction for the given client.\n   */\n  async rollbackWithClientAsync(client: pg.PoolClient) {\n    debug('rollbackWithClient()');\n    // From https://node-postgres.com/features/transactions\n    try {\n      await client.query('ROLLBACK');\n      // Only release the client if we weren't already inside a transaction.\n      if (this.alsClient.getStore() === undefined) {\n        client.release();\n      }\n    } catch (err: any) {\n      // If there was a problem rolling back the query, something is\n      // seriously messed up. Return the error to the release() function to\n      // close & remove this client from the pool. If you leave a client in\n      // the pool with an unaborted transaction, weird and hard to diagnose\n      // problems might happen.\n      client.release(err);\n    }\n  }\n\n  /**\n   * Begins a new transaction.\n   */\n  async beginTransactionAsync(): Promise<pg.PoolClient> {\n    debug('beginTransaction()');\n    const client = await this.getClientAsync();\n    try {\n      await this.queryWithClientAsync(client, 'START TRANSACTION;', {});\n      return client;\n    } catch (err) {\n      await this.rollbackWithClientAsync(client);\n      throw err;\n    }\n  }\n\n  /**\n   * Commits the transaction if err is null, otherwise rollbacks the transaction.\n   * Also releases the client.\n   */\n  async endTransactionAsync(client: pg.PoolClient, err: Error | null | undefined) {\n    debug('endTransaction()');\n    if (err) {\n      try {\n        await this.rollbackWithClientAsync(client);\n      } catch (rollbackErr: any) {\n        throw addDataToError(rollbackErr, { prevErr: err, rollback: 'fail' });\n      }\n\n      // Even though we successfully rolled back the transaction, there was\n      // still an error in the first place that necessitated a rollback. Re-throw\n      // that error here so that everything downstream of here will know about it.\n      throw addDataToError(err, { rollback: 'success' });\n    } else {\n      try {\n        await this.queryWithClientAsync(client, 'COMMIT', {});\n      } finally {\n        // Only release the client if we aren't nested inside another transaction.\n        if (this.alsClient.getStore() === undefined) {\n          client.release();\n        }\n      }\n    }\n  }\n\n  /**\n   * Runs the specified function inside of a transaction. The function will\n   * receive a database client as an argument, but it can also make queries\n   * as usual, and the correct client will be used automatically.\n   *\n   * The transaction will be rolled back if the function throws an error, and\n   * will be committed otherwise.\n   */\n  async runInTransactionAsync<T>(fn: (client: pg.PoolClient) => Promise<T>): Promise<T> {\n    // Check if we're already inside a transaction. If so, we won't start another one,\n    // as Postgres doesn't support nested transactions.\n    const client = this.alsClient.getStore();\n    const isNestedTransaction = client !== undefined;\n    const transactionClient = client ?? (await this.beginTransactionAsync());\n\n    let result: T;\n    try {\n      result = await this.alsClient.run(transactionClient, () => fn(transactionClient));\n    } catch (err: any) {\n      if (!isNestedTransaction) {\n        // If we're inside another transaction, we assume that the root transaction\n        // will catch this error and roll back the transaction.\n        await this.endTransactionAsync(transactionClient, err);\n      }\n      throw err;\n    }\n\n    if (!isNestedTransaction) {\n      // If we're inside another transaction; don't commit it prematurely. Allow\n      // the root transaction to commit it instead.\n      //\n      // Note that we don't invoke `endTransactionAsync` inside the `try` block\n      // because we don't want an error thrown by it to trigger *another* call\n      // to `endTransactionAsync` in the `catch` block.\n      await this.endTransactionAsync(transactionClient, null);\n    }\n\n    return result;\n  }\n\n  /**\n   * Executes a query with the specified parameters.\n   */\n  async queryAsync(sql: string, params: QueryParams): Promise<QueryResult> {\n    debug('query()', 'sql:', debugString(sql));\n    debug('query()', 'params:', debugParams(params));\n    const client = await this.getClientAsync();\n    try {\n      return await this.queryWithClientAsync(client, sql, params);\n    } finally {\n      // Only release if we aren't nested in a transaction.\n      if (this.alsClient.getStore() === undefined) {\n        client.release();\n      }\n    }\n  }\n\n  /**\n   * Executes a query with the specified parameters. Errors if the query does\n   * not return exactly one row.\n   */\n  async queryOneRowAsync(sql: string, params: QueryParams): Promise<pg.QueryResult> {\n    debug('queryOneRow()', 'sql:', debugString(sql));\n    debug('queryOneRow()', 'params:', debugParams(params));\n    const result = await this.queryAsync(sql, params);\n    if (result.rowCount !== 1) {\n      throw new PostgresError(`Incorrect rowCount: ${result.rowCount}`, {\n        sql,\n        sqlParams: params,\n      });\n    }\n    debug('queryOneRow() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  /**\n   * Executes a query with the specified parameters. Errors if the query\n   * returns more than one row.\n   */\n  async queryZeroOrOneRowAsync(sql: string, params: QueryParams): Promise<pg.QueryResult> {\n    debug('queryZeroOrOneRow()', 'sql:', debugString(sql));\n    debug('queryZeroOrOneRow()', 'params:', debugParams(params));\n    const result = await this.queryAsync(sql, params);\n    if (result.rowCount == null || result.rowCount > 1) {\n      throw new PostgresError(`Incorrect rowCount: ${result.rowCount}`, {\n        sql,\n        sqlParams: params,\n      });\n    }\n    debug('queryZeroOrOneRow() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  /**\n   * Calls the given sproc with the specified parameters.\n   */\n  async callAsync(functionName: string, params: any[]): Promise<pg.QueryResult> {\n    debug('call()', 'function:', functionName);\n    debug('call()', 'params:', debugParams(params));\n    const placeholders = params.map((_, v) => '$' + (v + 1)).join();\n    const sql = `SELECT * FROM ${escapeIdentifier(functionName)}(${placeholders});`;\n    const result = await this.queryAsync(sql, params);\n    debug('call() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  /**\n   * Calls the given sproc with the specified parameters. Errors if the\n   * sproc does not return exactly one row.\n   */\n  async callOneRowAsync(functionName: string, params: any[]): Promise<pg.QueryResult> {\n    debug('callOneRow()', 'function:', functionName);\n    debug('callOneRow()', 'params:', debugParams(params));\n    const result = await this.callAsync(functionName, params);\n    if (result.rowCount !== 1) {\n      throw new PostgresError('Incorrect rowCount: ' + result.rowCount, {\n        functionName,\n        sqlParams: params,\n      });\n    }\n    debug('callOneRow() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  /**\n   * Calls the given sproc with the specified parameters. Errors if the\n   * sproc returns more than one row.\n   */\n  async callZeroOrOneRowAsync(functionName: string, params: any[]): Promise<pg.QueryResult> {\n    debug('callZeroOrOneRow()', 'function:', functionName);\n    debug('callZeroOrOneRow()', 'params:', debugParams(params));\n    const result = await this.callAsync(functionName, params);\n    if (result.rowCount == null || result.rowCount > 1) {\n      throw new PostgresError('Incorrect rowCount: ' + result.rowCount, {\n        functionName,\n        sqlParams: params,\n      });\n    }\n    debug('callZeroOrOneRow() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  /**\n   * Calls a sproc with the specified parameters using a specific client.\n   */\n  async callWithClientAsync(\n    client: pg.PoolClient,\n    functionName: string,\n    params: any[],\n  ): Promise<pg.QueryResult> {\n    debug('callWithClient()', 'function:', functionName);\n    debug('callWithClient()', 'params:', debugParams(params));\n    const placeholders = params.map((_, v) => '$' + (v + 1)).join();\n    const sql = `SELECT * FROM ${escapeIdentifier(functionName)}(${placeholders})`;\n    const result = await this.queryWithClientAsync(client, sql, params);\n    debug('callWithClient() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  /**\n   * Calls a sproc with the specified parameters using a specific client.\n   * Errors if the sproc does not return exactly one row.\n   */\n  async callWithClientOneRowAsync(\n    client: pg.PoolClient,\n    functionName: string,\n    params: any[],\n  ): Promise<pg.QueryResult> {\n    debug('callWithClientOneRow()', 'function:', functionName);\n    debug('callWithClientOneRow()', 'params:', debugParams(params));\n    const result = await this.callWithClientAsync(client, functionName, params);\n    if (result.rowCount !== 1) {\n      throw new PostgresError('Incorrect rowCount: ' + result.rowCount, {\n        functionName,\n        sqlParams: params,\n      });\n    }\n    debug('callWithClientOneRow() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  /**\n   * Calls a function with the specified parameters using a specific client.\n   * Errors if the sproc returns more than one row.\n   */\n  async callWithClientZeroOrOneRowAsync(\n    client: pg.PoolClient,\n    functionName: string,\n    params: any[],\n  ): Promise<pg.QueryResult> {\n    debug('callWithClientZeroOrOneRow()', 'function:', functionName);\n    debug('callWithClientZeroOrOneRow()', 'params:', debugParams(params));\n    const result = await this.callWithClientAsync(client, functionName, params);\n    if (result.rowCount == null || result.rowCount > 1) {\n      throw new PostgresError('Incorrect rowCount: ' + result.rowCount, {\n        functionName,\n        sqlParams: params,\n      });\n    }\n    debug('callWithClientZeroOrOneRow() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  async queryRows<Model extends z.ZodTypeAny>(sql: string, model: Model): Promise<z.infer<Model>[]>;\n  async queryRows<Model extends z.ZodTypeAny>(\n    sql: string,\n    params: QueryParams,\n    model: Model,\n  ): Promise<z.infer<Model>[]>;\n  /**\n   * Executes a query with the specified parameters. Returns an array of rows\n   * that conform to the given Zod schema.\n   *\n   * If the query returns a single column, the return value will be a list of column values.\n   */\n  async queryRows<Model extends z.ZodTypeAny>(\n    sql: string,\n    paramsOrSchema: QueryParams | Model,\n    maybeModel?: Model,\n  ) {\n    const params = maybeModel === undefined ? {} : (paramsOrSchema as QueryParams);\n    const model = maybeModel === undefined ? (paramsOrSchema as Model) : maybeModel;\n    const results = await this.queryAsync(sql, params);\n    if (results.fields.length === 1) {\n      const columnName = results.fields[0].name;\n      const rawData = results.rows.map((row) => row[columnName]);\n      return z.array(model).parse(rawData);\n    } else {\n      return z.array(model).parse(results.rows);\n    }\n  }\n\n  async queryRow<Model extends z.ZodTypeAny>(sql: string, model: Model): Promise<z.infer<Model>>;\n  async queryRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    params: QueryParams,\n    model: Model,\n  ): Promise<z.infer<Model>>;\n  /**\n   * Executes a query with the specified parameters. Returns exactly one row that conforms to the given Zod schema.\n   *\n   * If the query returns a single column, the return value will be the column value itself.\n   */\n  async queryRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    paramsOrSchema: QueryParams | Model,\n    maybeModel?: Model,\n  ) {\n    const params = maybeModel === undefined ? {} : (paramsOrSchema as QueryParams);\n    const model = maybeModel === undefined ? (paramsOrSchema as Model) : maybeModel;\n    const results = await this.queryOneRowAsync(sql, params);\n    if (results.fields.length === 1) {\n      const columnName = results.fields[0].name;\n      return model.parse(results.rows[0][columnName]);\n    } else {\n      return model.parse(results.rows[0]);\n    }\n  }\n\n  async queryOptionalRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    model: Model,\n  ): Promise<z.infer<Model> | null>;\n  async queryOptionalRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    params: QueryParams,\n    model: Model,\n  ): Promise<z.infer<Model> | null>;\n  /**\n   * Executes a query with the specified parameters. Returns either null or a\n   * single row that conforms to the given Zod schema, and errors otherwise.\n   *\n   * If the query returns a single column, the return value will be the column value itself.\n   */\n  async queryOptionalRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    paramsOrSchema: QueryParams | Model,\n    maybeModel?: Model,\n  ) {\n    const params = maybeModel === undefined ? {} : (paramsOrSchema as QueryParams);\n    const model = maybeModel === undefined ? (paramsOrSchema as Model) : maybeModel;\n    const results = await this.queryZeroOrOneRowAsync(sql, params);\n    if (results.rows.length === 0) {\n      return null;\n    } else if (results.fields.length === 1) {\n      const columnName = results.fields[0].name;\n      return model.parse(results.rows[0][columnName]);\n    } else {\n      return model.parse(results.rows[0]);\n    }\n  }\n\n  async callRows<Model extends z.ZodTypeAny>(sql: string, model: Model): Promise<z.infer<Model>[]>;\n  async callRows<Model extends z.ZodTypeAny>(\n    sql: string,\n    params: any[],\n    model: Model,\n  ): Promise<z.infer<Model>[]>;\n  /**\n   * Calls the given sproc with the specified parameters.\n   * Errors if the sproc does not return anything.\n   */\n  async callRows<Model extends z.ZodTypeAny>(\n    sql: string,\n    paramsOrSchema: any[] | Model,\n    maybeModel?: Model,\n  ) {\n    const params = maybeModel === undefined ? [] : (paramsOrSchema as any[]);\n    const model = maybeModel === undefined ? (paramsOrSchema as Model) : maybeModel;\n    const results = await this.callAsync(sql, params);\n    if (results.fields.length === 1) {\n      const columnName = results.fields[0].name;\n      const rawData = results.rows.map((row) => row[columnName]);\n      return z.array(model).parse(rawData);\n    } else {\n      return z.array(model).parse(results.rows);\n    }\n  }\n\n  async callRow<Model extends z.ZodTypeAny>(sql: string, model: Model): Promise<z.infer<Model>>;\n  async callRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    params: any[],\n    model: Model,\n  ): Promise<z.infer<Model>>;\n  /**\n   * Calls the given sproc with the specified parameters.\n   * Returns exactly one row from the sproc that conforms to the given Zod schema.\n   */\n  async callRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    paramsOrSchema: any[] | Model,\n    maybeModel?: Model,\n  ) {\n    const params = maybeModel === undefined ? [] : (paramsOrSchema as any[]);\n    const model = maybeModel === undefined ? (paramsOrSchema as Model) : maybeModel;\n    const results = await this.callOneRowAsync(sql, params);\n    if (results.fields.length === 1) {\n      const columnName = results.fields[0].name;\n      return model.parse(results.rows[0][columnName]);\n    } else {\n      return model.parse(results.rows[0]);\n    }\n  }\n\n  async callOptionalRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    model: Model,\n  ): Promise<z.infer<Model> | null>;\n  async callOptionalRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    params: any[],\n    model: Model,\n  ): Promise<z.infer<Model> | null>;\n  /**\n   * Calls the given sproc with the specified parameters. Returns either null\n   * or a single row that conforms to the given Zod schema.\n   */\n  async callOptionalRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    paramsOrSchema: any[] | Model,\n    maybeModel?: Model,\n  ) {\n    const params = maybeModel === undefined ? [] : (paramsOrSchema as any[]);\n    const model = maybeModel === undefined ? (paramsOrSchema as Model) : maybeModel;\n    const results = await this.callZeroOrOneRowAsync(sql, params);\n    if (results.rows.length === 0) {\n      return null;\n    } else if (results.fields.length === 1) {\n      const columnName = results.fields[0].name;\n      return model.parse(results.rows[0][columnName]);\n    } else {\n      return model.parse(results.rows[0]);\n    }\n  }\n\n  /**\n   * Returns a {@link Cursor} for the given query. The cursor can be used to\n   * read results in batches, which is useful for large result sets.\n   */\n  private async queryCursorWithClient(\n    client: pg.PoolClient,\n    sql: string,\n    params: QueryParams,\n  ): Promise<Cursor> {\n    this._queryCount += 1;\n    debug('queryCursorWithClient()', 'sql:', debugString(sql));\n    debug('queryCursorWithClient()', 'params:', debugParams(params));\n    const { processedSql, paramsArray } = paramsToArray(sql, params, this.errorOnUnusedParameters);\n    lastQueryMap.set(client, processedSql);\n    return client.query(new Cursor(processedSql, paramsArray));\n  }\n\n  /**\n   * Returns an {@link CursorIterator} that can be used to iterate over the\n   * results of the query in batches, which is useful for large result sets.\n   * Each row will be parsed by the given Zod schema.\n   */\n  async queryCursor<Model extends z.ZodTypeAny>(\n    sql: string,\n    params: QueryParams,\n    model: Model,\n  ): Promise<CursorIterator<z.infer<Model>>> {\n    return this.queryCursorInternal(sql, params, model);\n  }\n\n  private async queryCursorInternal<Model extends z.ZodTypeAny>(\n    sql: string,\n    params: QueryParams,\n    model?: Model,\n  ): Promise<CursorIterator<z.infer<Model>>> {\n    const client = await this.getClientAsync();\n    const cursor = await this.queryCursorWithClient(client, sql, params);\n\n    let iterateCalled = false;\n    const iterator: CursorIterator<z.infer<Model>> = {\n      async *iterate(batchSize: number) {\n        // Safety check: if someone calls iterate multiple times, they're\n        // definitely doing something wrong.\n        if (iterateCalled) {\n          throw new Error('iterate() called multiple times');\n        }\n        iterateCalled = true;\n\n        try {\n          while (true) {\n            const rows = await cursor.read(batchSize);\n            if (rows.length === 0) {\n              break;\n            }\n\n            if (model) {\n              yield z.array(model).parse(rows);\n            } else {\n              yield rows;\n            }\n          }\n        } catch (err: any) {\n          throw enhanceError(err, sql, params);\n        } finally {\n          try {\n            await cursor.close();\n          } finally {\n            client.release();\n          }\n        }\n      },\n      stream(batchSize: number) {\n        const transform = new Transform({\n          readableObjectMode: true,\n          writableObjectMode: true,\n          transform(chunk, _encoding, callback) {\n            for (const row of chunk) {\n              this.push(row);\n            }\n            callback();\n          },\n        });\n\n        // TODO: use native `node:stream#compose` once it's stable.\n        const generator = iterator.iterate(batchSize);\n        const pipe = multipipe(Readable.from(generator), transform);\n\n        // When the underlying stream is closed, we need to make sure that the\n        // cursor is also closed. We do this by calling `return()` on the generator,\n        // which will trigger its `finally` block, which will in turn release\n        // the client and close the cursor. The fact that the stream is already\n        // closed by this point means that someone reading from the stream will\n        // never actually see the `null` value that's returned.\n        pipe.once('close', () => {\n          generator.return(null);\n        });\n\n        return pipe;\n      },\n    };\n    return iterator;\n  }\n\n  /**\n   * Set the schema to use for the search path.\n   *\n   * @param schema The schema name to use (can be \"null\" to unset the search path)\n   */\n  async setSearchSchema(schema: string) {\n    if (schema == null) {\n      this.searchSchema = schema;\n      return;\n    }\n\n    await this.queryAsync(`CREATE SCHEMA IF NOT EXISTS ${escapeIdentifier(schema)}`, {});\n    // We only set searchSchema after CREATE to avoid the above query() call using searchSchema.\n    this.searchSchema = schema;\n  }\n\n  /**\n   * Get the schema that is currently used for the search path.\n   *\n   * @returns schema in use (may be `null` to indicate no schema)\n   */\n  getSearchSchema(): string | null {\n    return this.searchSchema;\n  }\n\n  /**\n   * Generate, set, and return a random schema name.\n   *\n   * @param prefix The prefix of the new schema, only the first 28 characters will be used (after lowercasing).\n   * @returns The randomly-generated search schema.\n   */\n  async setRandomSearchSchemaAsync(prefix: string): Promise<string> {\n    // truncated prefix (max 28 characters)\n    const truncPrefix = prefix.substring(0, 28);\n    // timestamp in format YYYY-MM-DDTHH:MM:SS.SSSZ (guaranteed to not exceed 27 characters in the spec)\n    const timestamp = new Date().toISOString();\n    // random 6-character suffix to avoid clashes (approx 2 billion possible values)\n    const chars = '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ'.split('');\n    const suffix = Array.from({ length: 6 })\n      .map(() => chars[Math.floor(Math.random() * chars.length)])\n      .join('');\n\n    // Schema is guaranteed to have length at most 63 (= 28 + 1 + 27 + 1 + 6),\n    // which is the default PostgreSQL identifier limit.\n    // Note that this schema name will need quoting because of characters like ':', '-', etc\n    const schema = `${truncPrefix}_${timestamp}_${suffix}`;\n    await this.setSearchSchema(schema);\n    return schema;\n  }\n\n  /** The number of established connections. */\n  get totalCount() {\n    return this.pool?.totalCount ?? 0;\n  }\n\n  /** The number of idle connections. */\n  get idleCount() {\n    return this.pool?.idleCount ?? 0;\n  }\n\n  /** The number of queries waiting for a connection to become available. */\n  get waitingCount() {\n    return this.pool?.waitingCount ?? 0;\n  }\n\n  /** The total number of queries that have been executed by this pool. */\n  get queryCount() {\n    return this._queryCount;\n  }\n}\n"]}
+{"version":3,"file":"pool.js","sourceRoot":"","sources":["../src/pool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAElD,OAAO,OAAO,MAAM,OAAO,CAAC;AAC5B,OAAO,CAAC,MAAM,QAAQ,CAAC;AACvB,OAAO,SAAS,MAAM,WAAW,CAAC;AAClC,OAAO,EAAE,EAAE,EAAE,aAAa,EAAoB,MAAM,IAAI,CAAC;AACzD,OAAO,MAAM,MAAM,WAAW,CAAC;AAC/B,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAaxB,MAAM,KAAK,GAAG,OAAO,CAAC,wBAAwB,CAAC,CAAC;AAChD,MAAM,YAAY,GAAG,IAAI,OAAO,EAAyB,CAAC;AAC1D,MAAM,eAAe,GAAG,IAAI,OAAO,EAAyB,CAAC;AAE7D,SAAS,cAAc,CAAC,GAAU,EAAE,IAAyB;IAC1D,GAAW,CAAC,IAAI,GAAG;QAClB,GAAI,GAAW,CAAC,IAAI;QACpB,GAAG,IAAI;KACR,CAAC;IACF,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,OAAO,aAAc,SAAQ,KAAK;IAC/B,IAAI,CAAsB;IAEjC,YAAY,OAAe,EAAE,IAAyB;QACpD,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,IAAI,GAAG,eAAe,CAAC;IAC9B,CAAC;CACF;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,CAAS;IAC5B,IAAI,OAAO,CAAC,KAAK,QAAQ;QAAE,OAAO,cAAc,CAAC;IACjD,CAAC,GAAG,CAAC,CAAC,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAC9B,IAAI,CAAC,CAAC,MAAM,GAAG,EAAE;QAAE,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,CAAC;IAC9C,CAAC,GAAG,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC;IAClB,OAAO,CAAC,CAAC;AACX,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,MAAmB;IACtC,IAAI,CAAC,CAAC;IACN,IAAI,CAAC;QACH,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAC7B,CAAC;IAAC,MAAM,CAAC;QACP,CAAC,GAAG,uBAAuB,CAAC;IAC9B,CAAC;IACD,OAAO,WAAW,CAAC,CAAC,CAAC,CAAC;AACxB,CAAC;AAED;;;GAGG;AACH,SAAS,aAAa,CACpB,GAAW,EACX,MAAmB,EACnB,uBAAgC;IAEhC,IAAI,OAAO,GAAG,KAAK,QAAQ;QAAE,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;IACrE,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,OAAO;YACL,YAAY,EAAE,GAAG;YACjB,WAAW,EAAE,MAAM;SACpB,CAAC;IACJ,CAAC;IACD,IAAI,MAAM,IAAI,IAAI,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QACjD,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IAED,MAAM,EAAE,GAAG,oBAAoB,CAAC;IAChC,IAAI,MAAM,CAAC;IACX,IAAI,YAAY,GAAG,EAAE,CAAC;IACtB,IAAI,YAAY,GAAG,GAAG,CAAC;IACvB,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,MAAM,GAAG,GAA2B,EAAE,CAAC;IACvC,IAAI,WAAW,GAAU,EAAE,CAAC;IAC5B,OAAO,CAAC,MAAM,GAAG,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACjD,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QACpB,IAAI,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAChB,IAAI,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC;gBAAE,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,EAAE,CAAC,CAAC;YAC/D,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC7B,GAAG,CAAC,CAAC,CAAC,GAAG,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC;gBACrF,OAAO,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;gBAC5B,WAAW,GAAG,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YAC9C,CAAC;iBAAM,CAAC;gBACN,OAAO,EAAE,CAAC;gBACV,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,OAAO,CAAC;gBACvB,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;QACD,YAAY,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;QAC7D,YAAY,GAAG,YAAY,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;IACrE,CAAC;IACD,YAAY,IAAI,YAAY,CAAC;IAC7B,YAAY,GAAG,EAAE,CAAC;IAClB,IAAI,uBAAuB,EAAE,CAAC;QAC5B,MAAM,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;QACvE,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,mCAAmC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QACnF,CAAC;IACH,CAAC;IACD,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC;AACvC,CAAC;AAED;;;GAGG;AACH,SAAS,gBAAgB,CAAC,UAAkB;IAC1C,sEAAsE;IACtE,mDAAmD;IACnD,sDAAsD;IACtD,kEAAkE;IAClE,OAAO,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;AAC1D,CAAC;AAED,SAAS,YAAY,CAAC,GAAU,EAAE,GAAW,EAAE,MAAmB;IAChE,qEAAqE;IACrE,eAAe;IACf,MAAM,QAAQ,GAAG,EAAE,GAAG,GAAG,EAAE,CAAC;IAE5B,4EAA4E;IAC5E,oBAAoB;IACpB,QAAQ,CAAC,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;IAE/B,MAAM,gBAAgB,GAAG,GAAG,YAAY,aAAa,IAAI,GAAG,CAAC,QAAQ,IAAI,IAAI,CAAC;IAE9E,OAAO,cAAc,CAAC,GAAG,EAAE;QACzB,QAAQ;QACR,2EAA2E;QAC3E,0EAA0E;QAC1E,eAAe;QACf,GAAG,EAAE,gBAAgB,CAAC,CAAC,CAAC,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,GAAG;QAC5E,SAAS,EAAE,MAAM;KAClB,CAAC,CAAC;AACL,CAAC;AAED,MAAM,OAAO,YAAY;IACvB,oDAAoD;IAC5C,IAAI,GAAmB,IAAI,CAAC;IACpC;;;;;OAKG;IACK,SAAS,GAAG,IAAI,iBAAiB,EAAiB,CAAC;IACnD,YAAY,GAAkB,IAAI,CAAC;IAC3C,gEAAgE;IACxD,WAAW,GAAG,CAAC,CAAC;IAChB,uBAAuB,GAAG,KAAK,CAAC;IAExC;;OAEG;IACH,KAAK,CAAC,SAAS,CACb,QAA4B,EAC5B,gBAA+D;QAE/D,IAAI,CAAC,uBAAuB,GAAG,QAAQ,CAAC,uBAAuB,IAAI,KAAK,CAAC;QACzE,IAAI,CAAC,IAAI,GAAG,IAAI,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,UAAU,GAAG,EAAE,MAAM;YACzC,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC3C,gBAAgB,CAAC,cAAc,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,MAAM,EAAE,EAAE;YACjC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBACzB,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBAC3C,gBAAgB,CAAC,cAAc,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;YAC/D,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE;YAChC,sEAAsE;YACtE,oEAAoE;YACpE,oEAAoE;YACpE,yEAAyE;YACzE,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,oEAAoE;QACpE,wCAAwC;QACxC,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,MAAM,aAAa,GAAG,CAAC,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;QACrD,OAAO,UAAU,IAAI,aAAa,CAAC,MAAM,EAAE,CAAC;YAC1C,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACzC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACjB,OAAO;YACT,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAClB,IAAI,UAAU,KAAK,aAAa,CAAC,MAAM,EAAE,CAAC;oBACxC,MAAM,IAAI,KAAK,CACb,uCAAuC,aAAa,CAAC,MAAM,cAAc,GAAG,CAAC,OAAO,EAAE,CACvF,CAAC;gBACJ,CAAC;gBAED,MAAM,OAAO,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;gBAC1C,UAAU,EAAE,CAAC;gBACb,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,IAAI,CAAC,IAAI;YAAE,OAAO;QACvB,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;QACtB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,cAAc;QAClB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;QACjD,CAAC;QAED,wEAAwE;QACxE,sBAAsB;QACtB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QAExE,yEAAyE;QACzE,sEAAsE;QACtE,sEAAsE;QACtE,wEAAwE;QACxE,8BAA8B;QAC9B,EAAE;QACF,8DAA8D;QAC9D,kEAAkE;QAClE,wEAAwE;QACxE,mEAAmE;QACnE,qEAAqE;QACrE,gDAAgD;QAChD,EAAE;QACF,4EAA4E;QAC5E,4EAA4E;QAC5E,kEAAkE;QAClE,yEAAyE;QACzE,+CAA+C;QAC/C,MAAM,kBAAkB,GAAG,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACvD,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,IAAI,kBAAkB,KAAK,IAAI,CAAC,YAAY,EAAE,CAAC;YAC1E,MAAM,gBAAgB,GAAG,sBAAsB,gBAAgB,CAAC,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC;YAC5F,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,gBAAgB,EAAE,EAAE,CAAC,CAAC;YAChE,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,CAAC,OAAO,EAAE,CAAC;gBACjB,MAAM,GAAG,CAAC;YACZ,CAAC;YACD,eAAe,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QACjD,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,oBAAoB,CACxB,MAAqB,EACrB,GAAW,EACX,MAAmB;QAEnB,IAAI,CAAC,WAAW,IAAI,CAAC,CAAC;QACtB,KAAK,CAAC,mBAAmB,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;QACrD,KAAK,CAAC,mBAAmB,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QAC3D,MAAM,EAAE,YAAY,EAAE,WAAW,EAAE,GAAG,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAC/F,IAAI,CAAC;YACH,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YACvC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;YAC7D,KAAK,CAAC,2BAA2B,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;YACjE,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,MAAM,YAAY,CAAC,GAAG,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,0BAA0B,CAC9B,MAAqB,EACrB,GAAW,EACX,MAAmB;QAEnB,KAAK,CAAC,yBAAyB,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;QAC3D,KAAK,CAAC,yBAAyB,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QACjE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACpE,IAAI,MAAM,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,aAAa,CAAC,uBAAuB,MAAM,CAAC,QAAQ,EAAE,EAAE;gBAChE,GAAG;gBACH,SAAS,EAAE,MAAM;gBACjB,MAAM;aACP,CAAC,CAAC;QACL,CAAC;QACD,KAAK,CAAC,iCAAiC,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QACvE,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,gCAAgC,CACpC,MAAqB,EACrB,GAAW,EACX,MAAmB;QAEnB,KAAK,CAAC,+BAA+B,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;QACjE,KAAK,CAAC,+BAA+B,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QACvE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACpE,IAAI,MAAM,CAAC,QAAQ,IAAI,IAAI,IAAI,MAAM,CAAC,QAAQ,GAAG,CAAC,EAAE,CAAC;YACnD,MAAM,IAAI,aAAa,CAAC,uBAAuB,MAAM,CAAC,QAAQ,EAAE,EAAE;gBAChE,GAAG;gBACH,SAAS,EAAE,MAAM;gBACjB,MAAM;aACP,CAAC,CAAC;QACL,CAAC;QACD,KAAK,CAAC,uCAAuC,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC7E,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,uBAAuB,CAAC,MAAqB;QACjD,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC9B,uDAAuD;QACvD,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAC/B,sEAAsE;YACtE,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,KAAK,SAAS,EAAE,CAAC;gBAC5C,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,CAAC;QACH,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,8DAA8D;YAC9D,qEAAqE;YACrE,qEAAqE;YACrE,qEAAqE;YACrE,yBAAyB;YACzB,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,qBAAqB;QACzB,KAAK,CAAC,oBAAoB,CAAC,CAAC;QAC5B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC3C,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,oBAAoB,EAAE,EAAE,CAAC,CAAC;YAClE,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC,CAAC;YAC3C,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,mBAAmB,CAAC,MAAqB,EAAE,GAA6B;QAC5E,KAAK,CAAC,kBAAkB,CAAC,CAAC;QAC1B,IAAI,GAAG,EAAE,CAAC;YACR,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC,CAAC;YAC7C,CAAC;YAAC,OAAO,WAAgB,EAAE,CAAC;gBAC1B,MAAM,cAAc,CAAC,WAAW,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;YACxE,CAAC;YAED,qEAAqE;YACrE,2EAA2E;YAC3E,4EAA4E;YAC5E,MAAM,cAAc,CAAC,GAAG,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;QACrD,CAAC;aAAM,CAAC;YACN,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;YACxD,CAAC;oBAAS,CAAC;gBACT,0EAA0E;gBAC1E,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,KAAK,SAAS,EAAE,CAAC;oBAC5C,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,qBAAqB,CAAI,EAAyC;QACtE,kFAAkF;QAClF,mDAAmD;QACnD,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC;QACzC,MAAM,mBAAmB,GAAG,MAAM,KAAK,SAAS,CAAC;QACjD,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,MAAM,IAAI,CAAC,qBAAqB,EAAE,CAAC,CAAC;QAEzE,IAAI,MAAS,CAAC;QACd,IAAI,CAAC;YACH,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,iBAAiB,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,iBAAiB,CAAC,CAAC,CAAC;QACpF,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBACzB,2EAA2E;gBAC3E,uDAAuD;gBACvD,MAAM,IAAI,CAAC,mBAAmB,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC;YACzD,CAAC;YACD,MAAM,GAAG,CAAC;QACZ,CAAC;QAED,IAAI,CAAC,mBAAmB,EAAE,CAAC;YACzB,0EAA0E;YAC1E,6CAA6C;YAC7C,EAAE;YACF,yEAAyE;YACzE,wEAAwE;YACxE,iDAAiD;YACjD,MAAM,IAAI,CAAC,mBAAmB,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QAC1D,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,UAAU,CAAC,GAAW,EAAE,MAAmB;QAC/C,KAAK,CAAC,SAAS,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;QAC3C,KAAK,CAAC,SAAS,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QACjD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC3C,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QAC9D,CAAC;gBAAS,CAAC;YACT,qDAAqD;YACrD,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,KAAK,SAAS,EAAE,CAAC;gBAC5C,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,gBAAgB,CAAC,GAAW,EAAE,MAAmB;QACrD,KAAK,CAAC,eAAe,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;QACjD,KAAK,CAAC,eAAe,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QACvD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAClD,IAAI,MAAM,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,aAAa,CAAC,uBAAuB,MAAM,CAAC,QAAQ,EAAE,EAAE;gBAChE,GAAG;gBACH,SAAS,EAAE,MAAM;aAClB,CAAC,CAAC;QACL,CAAC;QACD,KAAK,CAAC,uBAAuB,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC7D,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,sBAAsB,CAAC,GAAW,EAAE,MAAmB;QAC3D,KAAK,CAAC,qBAAqB,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;QACvD,KAAK,CAAC,qBAAqB,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QAC7D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAClD,IAAI,MAAM,CAAC,QAAQ,IAAI,IAAI,IAAI,MAAM,CAAC,QAAQ,GAAG,CAAC,EAAE,CAAC;YACnD,MAAM,IAAI,aAAa,CAAC,uBAAuB,MAAM,CAAC,QAAQ,EAAE,EAAE;gBAChE,GAAG;gBACH,SAAS,EAAE,MAAM;aAClB,CAAC,CAAC;QACL,CAAC;QACD,KAAK,CAAC,6BAA6B,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QACnE,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,YAAoB,EAAE,MAAa;QACjD,KAAK,CAAC,QAAQ,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QAC3C,KAAK,CAAC,QAAQ,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QAChD,MAAM,YAAY,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnE,MAAM,GAAG,GAAG,iBAAiB,gBAAgB,CAAC,YAAY,CAAC,IAAI,YAAY,IAAI,CAAC;QAChF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAClD,KAAK,CAAC,gBAAgB,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QACtD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,eAAe,CAAC,YAAoB,EAAE,MAAa;QACvD,KAAK,CAAC,cAAc,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QACjD,KAAK,CAAC,cAAc,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QACtD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QAC1D,IAAI,MAAM,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,aAAa,CAAC,sBAAsB,GAAG,MAAM,CAAC,QAAQ,EAAE;gBAChE,YAAY;gBACZ,SAAS,EAAE,MAAM;aAClB,CAAC,CAAC;QACL,CAAC;QACD,KAAK,CAAC,sBAAsB,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5D,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,qBAAqB,CAAC,YAAoB,EAAE,MAAa;QAC7D,KAAK,CAAC,oBAAoB,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QACvD,KAAK,CAAC,oBAAoB,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QAC5D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QAC1D,IAAI,MAAM,CAAC,QAAQ,IAAI,IAAI,IAAI,MAAM,CAAC,QAAQ,GAAG,CAAC,EAAE,CAAC;YACnD,MAAM,IAAI,aAAa,CAAC,sBAAsB,GAAG,MAAM,CAAC,QAAQ,EAAE;gBAChE,YAAY;gBACZ,SAAS,EAAE,MAAM;aAClB,CAAC,CAAC;QACL,CAAC;QACD,KAAK,CAAC,4BAA4B,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAClE,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,mBAAmB,CACvB,MAAqB,EACrB,YAAoB,EACpB,MAAa;QAEb,KAAK,CAAC,kBAAkB,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QACrD,KAAK,CAAC,kBAAkB,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QAC1D,MAAM,YAAY,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnE,MAAM,GAAG,GAAG,iBAAiB,gBAAgB,CAAC,YAAY,CAAC,IAAI,YAAY,GAAG,CAAC;QAC/E,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACpE,KAAK,CAAC,0BAA0B,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAChE,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,yBAAyB,CAC7B,MAAqB,EACrB,YAAoB,EACpB,MAAa;QAEb,KAAK,CAAC,wBAAwB,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QAC3D,KAAK,CAAC,wBAAwB,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QAChE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;QAC5E,IAAI,MAAM,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,aAAa,CAAC,sBAAsB,GAAG,MAAM,CAAC,QAAQ,EAAE;gBAChE,YAAY;gBACZ,SAAS,EAAE,MAAM;aAClB,CAAC,CAAC;QACL,CAAC;QACD,KAAK,CAAC,gCAAgC,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QACtE,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,+BAA+B,CACnC,MAAqB,EACrB,YAAoB,EACpB,MAAa;QAEb,KAAK,CAAC,8BAA8B,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QACjE,KAAK,CAAC,8BAA8B,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QACtE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;QAC5E,IAAI,MAAM,CAAC,QAAQ,IAAI,IAAI,IAAI,MAAM,CAAC,QAAQ,GAAG,CAAC,EAAE,CAAC;YACnD,MAAM,IAAI,aAAa,CAAC,sBAAsB,GAAG,MAAM,CAAC,QAAQ,EAAE;gBAChE,YAAY;gBACZ,SAAS,EAAE,MAAM;aAClB,CAAC,CAAC;QACL,CAAC;QACD,KAAK,CAAC,sCAAsC,EAAE,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5E,OAAO,MAAM,CAAC;IAChB,CAAC;IAQD;;;;;OAKG;IACH,KAAK,CAAC,SAAS,CACb,GAAW,EACX,cAAmC,EACnC,UAAkB;QAElB,MAAM,MAAM,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAE,cAA8B,CAAC;QAC/E,MAAM,KAAK,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAE,cAAwB,CAAC,CAAC,CAAC,UAAU,CAAC;QAChF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACnD,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC1C,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;YAC3D,OAAO,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;IAQD;;;;OAIG;IACH,KAAK,CAAC,QAAQ,CACZ,GAAW,EACX,cAAmC,EACnC,UAAkB;QAElB,MAAM,MAAM,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAE,cAA8B,CAAC;QAC/E,MAAM,KAAK,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAE,cAAwB,CAAC,CAAC,CAAC,UAAU,CAAC;QAChF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACzD,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC1C,OAAO,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAClD,CAAC;aAAM,CAAC;YACN,OAAO,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAWD;;;;;OAKG;IACH,KAAK,CAAC,gBAAgB,CACpB,GAAW,EACX,cAAmC,EACnC,UAAkB;QAElB,MAAM,MAAM,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAE,cAA8B,CAAC;QAC/E,MAAM,KAAK,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAE,cAAwB,CAAC,CAAC,CAAC,UAAU,CAAC;QAChF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAC/D,IAAI,OAAO,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9B,OAAO,IAAI,CAAC;QACd,CAAC;aAAM,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvC,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC1C,OAAO,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAClD,CAAC;aAAM,CAAC;YACN,OAAO,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAQD;;;OAGG;IACH,KAAK,CAAC,QAAQ,CACZ,GAAW,EACX,cAA6B,EAC7B,UAAkB;QAElB,MAAM,MAAM,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAE,cAAwB,CAAC;QACzE,MAAM,KAAK,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAE,cAAwB,CAAC,CAAC,CAAC,UAAU,CAAC;QAChF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAClD,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC1C,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;YAC3D,OAAO,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;IAQD;;;OAGG;IACH,KAAK,CAAC,OAAO,CACX,GAAW,EACX,cAA6B,EAC7B,UAAkB;QAElB,MAAM,MAAM,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAE,cAAwB,CAAC;QACzE,MAAM,KAAK,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAE,cAAwB,CAAC,CAAC,CAAC,UAAU,CAAC;QAChF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACxD,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC1C,OAAO,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAClD,CAAC;aAAM,CAAC;YACN,OAAO,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAWD;;;OAGG;IACH,KAAK,CAAC,eAAe,CACnB,GAAW,EACX,cAA6B,EAC7B,UAAkB;QAElB,MAAM,MAAM,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAE,cAAwB,CAAC;QACzE,MAAM,KAAK,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAE,cAAwB,CAAC,CAAC,CAAC,UAAU,CAAC;QAChF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAC9D,IAAI,OAAO,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9B,OAAO,IAAI,CAAC;QACd,CAAC;aAAM,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvC,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC1C,OAAO,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAClD,CAAC;aAAM,CAAC;YACN,OAAO,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CAAC,GAAW,EAAE,SAAsB,EAAE;QACjD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAClD,OAAO,MAAM,CAAC,QAAQ,IAAI,CAAC,CAAC;IAC9B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,GAAW,EAAE,SAAsB,EAAE;QACpD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACjD,IAAI,QAAQ,KAAK,CAAC,EAAE,CAAC;YACnB,MAAM,IAAI,aAAa,CAAC,sBAAsB,GAAG,QAAQ,EAAE;gBACzD,GAAG;gBACH,SAAS,EAAE,MAAM;aAClB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,qBAAqB,CACjC,MAAqB,EACrB,GAAW,EACX,MAAmB;QAEnB,IAAI,CAAC,WAAW,IAAI,CAAC,CAAC;QACtB,KAAK,CAAC,yBAAyB,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;QAC3D,KAAK,CAAC,yBAAyB,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QACjE,MAAM,EAAE,YAAY,EAAE,WAAW,EAAE,GAAG,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAC/F,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QACvC,OAAO,MAAM,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC,CAAC;IAC7D,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,WAAW,CACf,GAAW,EACX,MAAmB,EACnB,KAAY;QAEZ,OAAO,IAAI,CAAC,mBAAmB,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;IACtD,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAC/B,GAAW,EACX,MAAmB,EACnB,KAAa;QAEb,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC3C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QAErE,IAAI,aAAa,GAAG,KAAK,CAAC;QAC1B,MAAM,QAAQ,GAAmC;YAC/C,KAAK,CAAC,CAAC,OAAO,CAAC,SAAiB;gBAC9B,iEAAiE;gBACjE,oCAAoC;gBACpC,IAAI,aAAa,EAAE,CAAC;oBAClB,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;gBACrD,CAAC;gBACD,aAAa,GAAG,IAAI,CAAC;gBAErB,IAAI,CAAC;oBACH,OAAO,IAAI,EAAE,CAAC;wBACZ,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;wBAC1C,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;4BACtB,MAAM;wBACR,CAAC;wBAED,IAAI,KAAK,EAAE,CAAC;4BACV,MAAM,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBACnC,CAAC;6BAAM,CAAC;4BACN,MAAM,IAAI,CAAC;wBACb,CAAC;oBACH,CAAC;gBACH,CAAC;gBAAC,OAAO,GAAQ,EAAE,CAAC;oBAClB,MAAM,YAAY,CAAC,GAAG,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;gBACvC,CAAC;wBAAS,CAAC;oBACT,IAAI,CAAC;wBACH,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;oBACvB,CAAC;4BAAS,CAAC;wBACT,MAAM,CAAC,OAAO,EAAE,CAAC;oBACnB,CAAC;gBACH,CAAC;YACH,CAAC;YACD,MAAM,CAAC,SAAiB;gBACtB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC;oBAC9B,kBAAkB,EAAE,IAAI;oBACxB,kBAAkB,EAAE,IAAI;oBACxB,SAAS,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ;wBAClC,KAAK,MAAM,GAAG,IAAI,KAAK,EAAE,CAAC;4BACxB,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;wBACjB,CAAC;wBACD,QAAQ,EAAE,CAAC;oBACb,CAAC;iBACF,CAAC,CAAC;gBAEH,2DAA2D;gBAC3D,MAAM,SAAS,GAAG,QAAQ,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBAC9C,MAAM,IAAI,GAAG,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,CAAC,CAAC;gBAE5D,sEAAsE;gBACtE,4EAA4E;gBAC5E,qEAAqE;gBACrE,uEAAuE;gBACvE,uEAAuE;gBACvE,uDAAuD;gBACvD,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE;oBACtB,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBACzB,CAAC,CAAC,CAAC;gBAEH,OAAO,IAAI,CAAC;YACd,CAAC;SACF,CAAC;QACF,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,eAAe,CAAC,MAAqB;QACzC,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;YACnB,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC;YAC3B,OAAO;QACT,CAAC;QAED,MAAM,IAAI,CAAC,UAAU,CAAC,+BAA+B,gBAAgB,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QACrF,4FAA4F;QAC5F,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC;IAC7B,CAAC;IAED;;;;OAIG;IACH,eAAe;QACb,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,0BAA0B,CAAC,MAAc;QAC7C,uCAAuC;QACvC,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACxC,oGAAoG;QACpG,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAC3C,gFAAgF;QAChF,MAAM,KAAK,GAAG,sCAAsC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;aACrC,GAAG,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;aAC1D,IAAI,CAAC,EAAE,CAAC,CAAC;QAEZ,0EAA0E;QAC1E,oDAAoD;QACpD,wFAAwF;QACxF,MAAM,MAAM,GAAG,GAAG,WAAW,IAAI,SAAS,IAAI,MAAM,EAAE,CAAC;QACvD,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QACnC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,6CAA6C;IAC7C,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,IAAI,EAAE,UAAU,IAAI,CAAC,CAAC;IACpC,CAAC;IAED,sCAAsC;IACtC,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,IAAI,EAAE,SAAS,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,0EAA0E;IAC1E,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,IAAI,EAAE,YAAY,IAAI,CAAC,CAAC;IACtC,CAAC;IAED,wEAAwE;IACxE,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;CACF","sourcesContent":["import { AsyncLocalStorage } from 'node:async_hooks';\nimport { Readable, Transform } from 'node:stream';\n\nimport debugfn from 'debug';\nimport _ from 'lodash';\nimport multipipe from 'multipipe';\nimport pg, { DatabaseError, type QueryResult } from 'pg';\nimport Cursor from 'pg-cursor';\nimport { z } from 'zod';\n\nexport type QueryParams = Record<string, any> | any[];\n\nexport interface CursorIterator<T> {\n  iterate: (batchSize: number) => AsyncGenerator<T[]>;\n  stream: (batchSize: number) => NodeJS.ReadWriteStream;\n}\n\nexport interface PostgresPoolConfig extends pg.PoolConfig {\n  errorOnUnusedParameters?: boolean;\n}\n\nconst debug = debugfn('@prairielearn/postgres');\nconst lastQueryMap = new WeakMap<pg.PoolClient, string>();\nconst searchSchemaMap = new WeakMap<pg.PoolClient, string>();\n\nfunction addDataToError(err: Error, data: Record<string, any>): Error {\n  (err as any).data = {\n    ...(err as any).data,\n    ...data,\n  };\n  return err;\n}\n\nexport class PostgresError extends Error {\n  public data: Record<string, any>;\n\n  constructor(message: string, data: Record<string, any>) {\n    super(message);\n    this.data = data;\n    this.name = 'PostgresError';\n  }\n}\n\n/**\n * Formats a string for debugging.\n */\nfunction debugString(s: string): string {\n  if (typeof s !== 'string') return 'NOT A STRING';\n  s = s.replaceAll('\\n', '\\\\n');\n  if (s.length > 78) s = s.slice(0, 75) + '...';\n  s = '\"' + s + '\"';\n  return s;\n}\n\n/**\n * Formats a set of params for debugging.\n */\nfunction debugParams(params: QueryParams): string {\n  let s;\n  try {\n    s = JSON.stringify(params);\n  } catch {\n    s = 'CANNOT JSON STRINGIFY';\n  }\n  return debugString(s);\n}\n\n/**\n * Given an SQL string and params, creates an array of params and an SQL string\n * with any named dollar-sign placeholders replaced with parameters.\n */\nfunction paramsToArray(\n  sql: string,\n  params: QueryParams,\n  errorOnUnusedParameters: boolean,\n): { processedSql: string; paramsArray: any } {\n  if (typeof sql !== 'string') throw new Error('SQL must be a string');\n  if (Array.isArray(params)) {\n    return {\n      processedSql: sql,\n      paramsArray: params,\n    };\n  }\n  if (params == null || typeof params !== 'object') {\n    throw new Error('params must be array or object');\n  }\n\n  const re = /\\$([-_a-zA-Z0-9]+)/;\n  let result;\n  let processedSql = '';\n  let remainingSql = sql;\n  let nParams = 0;\n  const map: Record<string, string> = {};\n  let paramsArray: any[] = [];\n  while ((result = re.exec(remainingSql)) !== null) {\n    const v = result[1];\n    if (!(v in map)) {\n      if (!(v in params)) throw new Error(`Missing parameter: ${v}`);\n      if (Array.isArray(params[v])) {\n        map[v] = 'ARRAY[' + params[v].map((_, n) => '$' + (n + nParams + 1)).join(',') + ']';\n        nParams += params[v].length;\n        paramsArray = paramsArray.concat(params[v]);\n      } else {\n        nParams++;\n        map[v] = '$' + nParams;\n        paramsArray.push(params[v]);\n      }\n    }\n    processedSql += remainingSql.slice(0, result.index) + map[v];\n    remainingSql = remainingSql.slice(result.index + result[0].length);\n  }\n  processedSql += remainingSql;\n  remainingSql = '';\n  if (errorOnUnusedParameters) {\n    const difference = _.difference(Object.keys(params), Object.keys(map));\n    if (difference.length > 0) {\n      throw new Error(`Unused parameters in SQL query: ${JSON.stringify(difference)}`);\n    }\n  }\n  return { processedSql, paramsArray };\n}\n\n/**\n * Escapes the given identifier for use in an SQL query. Useful for preventing\n * SQL injection.\n */\nfunction escapeIdentifier(identifier: string): string {\n  // Note that as of 2021-06-29 escapeIdentifier() is undocumented. See:\n  // https://github.com/brianc/node-postgres/pull/396\n  // https://github.com/brianc/node-postgres/issues/1978\n  // https://www.postgresql.org/docs/current/sql-syntax-lexical.html\n  return pg.Client.prototype.escapeIdentifier(identifier);\n}\n\nfunction enhanceError(err: Error, sql: string, params: QueryParams): Error {\n  // Copy the error so we don't end up with a circular reference in the\n  // final error.\n  const sqlError = { ...err };\n\n  // `message` is a non-enumerable property, so we need to copy it manually to\n  // the error object.\n  sqlError.message = err.message;\n\n  const errorHasPosition = err instanceof DatabaseError && err.position != null;\n\n  return addDataToError(err, {\n    sqlError,\n    // If the error has a `position` field, we need to use the processed source\n    // (where e.g. `$foobar` has been replaced with `$1`) so that the position\n    // is accurate.\n    sql: errorHasPosition ? paramsToArray(sql, params, false).processedSql : sql,\n    sqlParams: params,\n  });\n}\n\nexport class PostgresPool {\n  /** The pool from which clients will be acquired. */\n  private pool: pg.Pool | null = null;\n  /**\n   * We use this to propagate the client associated with the current transaction\n   * to any nested queries. In the past, we had some nasty bugs associated with\n   * the fact that we tried to acquire new clients inside of transactions, which\n   * ultimately lead to a deadlock.\n   */\n  private alsClient = new AsyncLocalStorage<pg.PoolClient>();\n  private searchSchema: string | null = null;\n  /** Tracks the total number of queries executed by this pool. */\n  private _queryCount = 0;\n  private errorOnUnusedParameters = false;\n\n  /**\n   * Creates a new connection pool and attempts to connect to the database.\n   */\n  async initAsync(\n    pgConfig: PostgresPoolConfig,\n    idleErrorHandler: (error: Error, client: pg.PoolClient) => void,\n  ): Promise<void> {\n    this.errorOnUnusedParameters = pgConfig.errorOnUnusedParameters ?? false;\n    this.pool = new pg.Pool(pgConfig);\n    this.pool.on('error', function (err, client) {\n      const lastQuery = lastQueryMap.get(client);\n      idleErrorHandler(addDataToError(err, { lastQuery }), client);\n    });\n    this.pool.on('connect', (client) => {\n      client.on('error', (err) => {\n        const lastQuery = lastQueryMap.get(client);\n        idleErrorHandler(addDataToError(err, { lastQuery }), client);\n      });\n    });\n    this.pool.on('remove', (client) => {\n      // This shouldn't be necessary, as `pg` currently allows clients to be\n      // garbage collected after they're removed. However, if `pg` someday\n      // starts reusing client objects across difference connections, this\n      // will ensure that we re-set the search path when the client reconnects.\n      searchSchemaMap.delete(client);\n    });\n\n    // Attempt to connect to the database so that we can fail quickly if\n    // something isn't configured correctly.\n    let retryCount = 0;\n    const retryTimeouts = [500, 1000, 2000, 5000, 10000];\n    while (retryCount <= retryTimeouts.length) {\n      try {\n        const client = await this.pool.connect();\n        client.release();\n        return;\n      } catch (err: any) {\n        if (retryCount === retryTimeouts.length) {\n          throw new Error(\n            `Could not connect to Postgres after ${retryTimeouts.length} attempts: ${err.message}`,\n          );\n        }\n\n        const timeout = retryTimeouts[retryCount];\n        retryCount++;\n        await new Promise((resolve) => setTimeout(resolve, timeout));\n      }\n    }\n  }\n\n  /**\n   * Closes the connection pool.\n   */\n  async closeAsync(): Promise<void> {\n    if (!this.pool) return;\n    await this.pool.end();\n    this.pool = null;\n  }\n\n  /**\n   * Gets a new client from the connection pool. The caller MUST call `release()` to\n   * release the client, whether or not errors occurred while using\n   * `client`. The client can call `done(truthy_value)` to force\n   * destruction of the client, but this should not be used except in\n   * unusual circumstances.\n   */\n  async getClientAsync(): Promise<pg.PoolClient> {\n    if (!this.pool) {\n      throw new Error('Connection pool is not open');\n    }\n\n    // If we're inside a transaction, we'll reuse the same client to avoid a\n    // potential deadlock.\n    const client = this.alsClient.getStore() ?? (await this.pool.connect());\n\n    // If we're configured to use a particular schema, we'll store whether or\n    // not the search path has already been configured for this particular\n    // client. If we acquire a client and it's already had its search path\n    // set, we can avoid setting it again since the search path will persist\n    // for the life of the client.\n    //\n    // We do this check for each call to `getClient` instead of on\n    // `pool.connect` so that we don't have to be really careful about\n    // destroying old clients that were created before `setSearchSchema` was\n    // called. Instead, we'll just check if the search path matches the\n    // currently-desired schema, and if it's a mismatch (or doesn't exist\n    // at all), we re-set it for the current client.\n    //\n    // Note that this accidentally supports changing the search_path on the fly,\n    // although that's not something we currently do (or would be likely to do).\n    // It does NOT support clearing the existing search schema - e.g.,\n    // `setSearchSchema(null)` would not work as you expect. This is fine, as\n    // that's not something we ever do in practice.\n    const clientSearchSchema = searchSchemaMap.get(client);\n    if (this.searchSchema != null && clientSearchSchema !== this.searchSchema) {\n      const setSearchPathSql = `SET search_path TO ${escapeIdentifier(this.searchSchema)},public`;\n      try {\n        await this.queryWithClientAsync(client, setSearchPathSql, {});\n      } catch (err) {\n        client.release();\n        throw err;\n      }\n      searchSchemaMap.set(client, this.searchSchema);\n    }\n\n    return client;\n  }\n\n  /**\n   * Performs a query with the given client.\n   */\n  async queryWithClientAsync(\n    client: pg.PoolClient,\n    sql: string,\n    params: QueryParams,\n  ): Promise<pg.QueryResult> {\n    this._queryCount += 1;\n    debug('queryWithClient()', 'sql:', debugString(sql));\n    debug('queryWithClient()', 'params:', debugParams(params));\n    const { processedSql, paramsArray } = paramsToArray(sql, params, this.errorOnUnusedParameters);\n    try {\n      lastQueryMap.set(client, processedSql);\n      const result = await client.query(processedSql, paramsArray);\n      debug('queryWithClient() success', 'rowCount:', result.rowCount);\n      return result;\n    } catch (err: any) {\n      throw enhanceError(err, sql, params);\n    }\n  }\n\n  /**\n   * Performs a query with the given client. Errors if the query returns more\n   * than one row.\n   */\n  async queryWithClientOneRowAsync(\n    client: pg.PoolClient,\n    sql: string,\n    params: QueryParams,\n  ): Promise<pg.QueryResult> {\n    debug('queryWithClientOneRow()', 'sql:', debugString(sql));\n    debug('queryWithClientOneRow()', 'params:', debugParams(params));\n    const result = await this.queryWithClientAsync(client, sql, params);\n    if (result.rowCount !== 1) {\n      throw new PostgresError(`Incorrect rowCount: ${result.rowCount}`, {\n        sql,\n        sqlParams: params,\n        result,\n      });\n    }\n    debug('queryWithClientOneRow() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  /**\n   * Performs a query with the given client. Errors if the query returns more\n   * than one row.\n   */\n  async queryWithClientZeroOrOneRowAsync(\n    client: pg.PoolClient,\n    sql: string,\n    params: QueryParams,\n  ): Promise<QueryResult> {\n    debug('queryWithClientZeroOrOneRow()', 'sql:', debugString(sql));\n    debug('queryWithClientZeroOrOneRow()', 'params:', debugParams(params));\n    const result = await this.queryWithClientAsync(client, sql, params);\n    if (result.rowCount == null || result.rowCount > 1) {\n      throw new PostgresError(`Incorrect rowCount: ${result.rowCount}`, {\n        sql,\n        sqlParams: params,\n        result,\n      });\n    }\n    debug('queryWithClientZeroOrOneRow() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  /**\n   * Rolls back the current transaction for the given client.\n   */\n  async rollbackWithClientAsync(client: pg.PoolClient) {\n    debug('rollbackWithClient()');\n    // From https://node-postgres.com/features/transactions\n    try {\n      await client.query('ROLLBACK');\n      // Only release the client if we weren't already inside a transaction.\n      if (this.alsClient.getStore() === undefined) {\n        client.release();\n      }\n    } catch (err: any) {\n      // If there was a problem rolling back the query, something is\n      // seriously messed up. Return the error to the release() function to\n      // close & remove this client from the pool. If you leave a client in\n      // the pool with an unaborted transaction, weird and hard to diagnose\n      // problems might happen.\n      client.release(err);\n    }\n  }\n\n  /**\n   * Begins a new transaction.\n   */\n  async beginTransactionAsync(): Promise<pg.PoolClient> {\n    debug('beginTransaction()');\n    const client = await this.getClientAsync();\n    try {\n      await this.queryWithClientAsync(client, 'START TRANSACTION;', {});\n      return client;\n    } catch (err) {\n      await this.rollbackWithClientAsync(client);\n      throw err;\n    }\n  }\n\n  /**\n   * Commits the transaction if err is null, otherwise rollbacks the transaction.\n   * Also releases the client.\n   */\n  async endTransactionAsync(client: pg.PoolClient, err: Error | null | undefined) {\n    debug('endTransaction()');\n    if (err) {\n      try {\n        await this.rollbackWithClientAsync(client);\n      } catch (rollbackErr: any) {\n        throw addDataToError(rollbackErr, { prevErr: err, rollback: 'fail' });\n      }\n\n      // Even though we successfully rolled back the transaction, there was\n      // still an error in the first place that necessitated a rollback. Re-throw\n      // that error here so that everything downstream of here will know about it.\n      throw addDataToError(err, { rollback: 'success' });\n    } else {\n      try {\n        await this.queryWithClientAsync(client, 'COMMIT', {});\n      } finally {\n        // Only release the client if we aren't nested inside another transaction.\n        if (this.alsClient.getStore() === undefined) {\n          client.release();\n        }\n      }\n    }\n  }\n\n  /**\n   * Runs the specified function inside of a transaction. The function will\n   * receive a database client as an argument, but it can also make queries\n   * as usual, and the correct client will be used automatically.\n   *\n   * The transaction will be rolled back if the function throws an error, and\n   * will be committed otherwise.\n   */\n  async runInTransactionAsync<T>(fn: (client: pg.PoolClient) => Promise<T>): Promise<T> {\n    // Check if we're already inside a transaction. If so, we won't start another one,\n    // as Postgres doesn't support nested transactions.\n    const client = this.alsClient.getStore();\n    const isNestedTransaction = client !== undefined;\n    const transactionClient = client ?? (await this.beginTransactionAsync());\n\n    let result: T;\n    try {\n      result = await this.alsClient.run(transactionClient, () => fn(transactionClient));\n    } catch (err: any) {\n      if (!isNestedTransaction) {\n        // If we're inside another transaction, we assume that the root transaction\n        // will catch this error and roll back the transaction.\n        await this.endTransactionAsync(transactionClient, err);\n      }\n      throw err;\n    }\n\n    if (!isNestedTransaction) {\n      // If we're inside another transaction; don't commit it prematurely. Allow\n      // the root transaction to commit it instead.\n      //\n      // Note that we don't invoke `endTransactionAsync` inside the `try` block\n      // because we don't want an error thrown by it to trigger *another* call\n      // to `endTransactionAsync` in the `catch` block.\n      await this.endTransactionAsync(transactionClient, null);\n    }\n\n    return result;\n  }\n\n  /**\n   * Executes a query with the specified parameters.\n   *\n   * @deprecated Use {@link execute} instead.\n   *\n   * Using the return value of this function directly is not recommended. Instead, use\n   * {@link queryRows}, {@link queryRow}, or {@link queryOptionalRow}.\n   */\n  async queryAsync(sql: string, params: QueryParams): Promise<QueryResult> {\n    debug('query()', 'sql:', debugString(sql));\n    debug('query()', 'params:', debugParams(params));\n    const client = await this.getClientAsync();\n    try {\n      return await this.queryWithClientAsync(client, sql, params);\n    } finally {\n      // Only release if we aren't nested in a transaction.\n      if (this.alsClient.getStore() === undefined) {\n        client.release();\n      }\n    }\n  }\n\n  /**\n   * Executes a query with the specified parameters. Errors if the query does\n   * not return exactly one row.\n   *\n   * @deprecated Use {@link executeRow} or {@link queryRow} instead.\n   */\n  async queryOneRowAsync(sql: string, params: QueryParams): Promise<pg.QueryResult> {\n    debug('queryOneRow()', 'sql:', debugString(sql));\n    debug('queryOneRow()', 'params:', debugParams(params));\n    const result = await this.queryAsync(sql, params);\n    if (result.rowCount !== 1) {\n      throw new PostgresError(`Incorrect rowCount: ${result.rowCount}`, {\n        sql,\n        sqlParams: params,\n      });\n    }\n    debug('queryOneRow() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  /**\n   * Executes a query with the specified parameters. Errors if the query\n   * returns more than one row.\n   */\n  async queryZeroOrOneRowAsync(sql: string, params: QueryParams): Promise<pg.QueryResult> {\n    debug('queryZeroOrOneRow()', 'sql:', debugString(sql));\n    debug('queryZeroOrOneRow()', 'params:', debugParams(params));\n    const result = await this.queryAsync(sql, params);\n    if (result.rowCount == null || result.rowCount > 1) {\n      throw new PostgresError(`Incorrect rowCount: ${result.rowCount}`, {\n        sql,\n        sqlParams: params,\n      });\n    }\n    debug('queryZeroOrOneRow() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  /**\n   * Calls the given sproc with the specified parameters.\n   */\n  async callAsync(functionName: string, params: any[]): Promise<pg.QueryResult> {\n    debug('call()', 'function:', functionName);\n    debug('call()', 'params:', debugParams(params));\n    const placeholders = params.map((_, v) => '$' + (v + 1)).join(',');\n    const sql = `SELECT * FROM ${escapeIdentifier(functionName)}(${placeholders});`;\n    const result = await this.queryAsync(sql, params);\n    debug('call() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  /**\n   * Calls the given sproc with the specified parameters. Errors if the\n   * sproc does not return exactly one row.\n   */\n  async callOneRowAsync(functionName: string, params: any[]): Promise<pg.QueryResult> {\n    debug('callOneRow()', 'function:', functionName);\n    debug('callOneRow()', 'params:', debugParams(params));\n    const result = await this.callAsync(functionName, params);\n    if (result.rowCount !== 1) {\n      throw new PostgresError('Incorrect rowCount: ' + result.rowCount, {\n        functionName,\n        sqlParams: params,\n      });\n    }\n    debug('callOneRow() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  /**\n   * Calls the given sproc with the specified parameters. Errors if the\n   * sproc returns more than one row.\n   */\n  async callZeroOrOneRowAsync(functionName: string, params: any[]): Promise<pg.QueryResult> {\n    debug('callZeroOrOneRow()', 'function:', functionName);\n    debug('callZeroOrOneRow()', 'params:', debugParams(params));\n    const result = await this.callAsync(functionName, params);\n    if (result.rowCount == null || result.rowCount > 1) {\n      throw new PostgresError('Incorrect rowCount: ' + result.rowCount, {\n        functionName,\n        sqlParams: params,\n      });\n    }\n    debug('callZeroOrOneRow() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  /**\n   * Calls a sproc with the specified parameters using a specific client.\n   */\n  async callWithClientAsync(\n    client: pg.PoolClient,\n    functionName: string,\n    params: any[],\n  ): Promise<pg.QueryResult> {\n    debug('callWithClient()', 'function:', functionName);\n    debug('callWithClient()', 'params:', debugParams(params));\n    const placeholders = params.map((_, v) => '$' + (v + 1)).join(',');\n    const sql = `SELECT * FROM ${escapeIdentifier(functionName)}(${placeholders})`;\n    const result = await this.queryWithClientAsync(client, sql, params);\n    debug('callWithClient() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  /**\n   * Calls a sproc with the specified parameters using a specific client.\n   * Errors if the sproc does not return exactly one row.\n   */\n  async callWithClientOneRowAsync(\n    client: pg.PoolClient,\n    functionName: string,\n    params: any[],\n  ): Promise<pg.QueryResult> {\n    debug('callWithClientOneRow()', 'function:', functionName);\n    debug('callWithClientOneRow()', 'params:', debugParams(params));\n    const result = await this.callWithClientAsync(client, functionName, params);\n    if (result.rowCount !== 1) {\n      throw new PostgresError('Incorrect rowCount: ' + result.rowCount, {\n        functionName,\n        sqlParams: params,\n      });\n    }\n    debug('callWithClientOneRow() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  /**\n   * Calls a function with the specified parameters using a specific client.\n   * Errors if the sproc returns more than one row.\n   */\n  async callWithClientZeroOrOneRowAsync(\n    client: pg.PoolClient,\n    functionName: string,\n    params: any[],\n  ): Promise<pg.QueryResult> {\n    debug('callWithClientZeroOrOneRow()', 'function:', functionName);\n    debug('callWithClientZeroOrOneRow()', 'params:', debugParams(params));\n    const result = await this.callWithClientAsync(client, functionName, params);\n    if (result.rowCount == null || result.rowCount > 1) {\n      throw new PostgresError('Incorrect rowCount: ' + result.rowCount, {\n        functionName,\n        sqlParams: params,\n      });\n    }\n    debug('callWithClientZeroOrOneRow() success', 'rowCount:', result.rowCount);\n    return result;\n  }\n\n  async queryRows<Model extends z.ZodTypeAny>(sql: string, model: Model): Promise<z.infer<Model>[]>;\n  async queryRows<Model extends z.ZodTypeAny>(\n    sql: string,\n    params: QueryParams,\n    model: Model,\n  ): Promise<z.infer<Model>[]>;\n  /**\n   * Executes a query with the specified parameters. Returns an array of rows\n   * that conform to the given Zod schema.\n   *\n   * If the query returns a single column, the return value will be a list of column values.\n   */\n  async queryRows<Model extends z.ZodTypeAny>(\n    sql: string,\n    paramsOrSchema: QueryParams | Model,\n    maybeModel?: Model,\n  ) {\n    const params = maybeModel === undefined ? {} : (paramsOrSchema as QueryParams);\n    const model = maybeModel === undefined ? (paramsOrSchema as Model) : maybeModel;\n    const results = await this.queryAsync(sql, params);\n    if (results.fields.length === 1) {\n      const columnName = results.fields[0].name;\n      const rawData = results.rows.map((row) => row[columnName]);\n      return z.array(model).parse(rawData);\n    } else {\n      return z.array(model).parse(results.rows);\n    }\n  }\n\n  async queryRow<Model extends z.ZodTypeAny>(sql: string, model: Model): Promise<z.infer<Model>>;\n  async queryRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    params: QueryParams,\n    model: Model,\n  ): Promise<z.infer<Model>>;\n  /**\n   * Executes a query with the specified parameters. Returns exactly one row that conforms to the given Zod schema.\n   *\n   * If the query returns a single column, the return value will be the column value itself.\n   */\n  async queryRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    paramsOrSchema: QueryParams | Model,\n    maybeModel?: Model,\n  ) {\n    const params = maybeModel === undefined ? {} : (paramsOrSchema as QueryParams);\n    const model = maybeModel === undefined ? (paramsOrSchema as Model) : maybeModel;\n    const results = await this.queryOneRowAsync(sql, params);\n    if (results.fields.length === 1) {\n      const columnName = results.fields[0].name;\n      return model.parse(results.rows[0][columnName]);\n    } else {\n      return model.parse(results.rows[0]);\n    }\n  }\n\n  async queryOptionalRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    model: Model,\n  ): Promise<z.infer<Model> | null>;\n  async queryOptionalRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    params: QueryParams,\n    model: Model,\n  ): Promise<z.infer<Model> | null>;\n  /**\n   * Executes a query with the specified parameters. Returns either null or a\n   * single row that conforms to the given Zod schema, and errors otherwise.\n   *\n   * If the query returns a single column, the return value will be the column value itself.\n   */\n  async queryOptionalRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    paramsOrSchema: QueryParams | Model,\n    maybeModel?: Model,\n  ) {\n    const params = maybeModel === undefined ? {} : (paramsOrSchema as QueryParams);\n    const model = maybeModel === undefined ? (paramsOrSchema as Model) : maybeModel;\n    const results = await this.queryZeroOrOneRowAsync(sql, params);\n    if (results.rows.length === 0) {\n      return null;\n    } else if (results.fields.length === 1) {\n      const columnName = results.fields[0].name;\n      return model.parse(results.rows[0][columnName]);\n    } else {\n      return model.parse(results.rows[0]);\n    }\n  }\n\n  async callRows<Model extends z.ZodTypeAny>(sql: string, model: Model): Promise<z.infer<Model>[]>;\n  async callRows<Model extends z.ZodTypeAny>(\n    sql: string,\n    params: any[],\n    model: Model,\n  ): Promise<z.infer<Model>[]>;\n  /**\n   * Calls the given sproc with the specified parameters.\n   * Errors if the sproc does not return anything.\n   */\n  async callRows<Model extends z.ZodTypeAny>(\n    sql: string,\n    paramsOrSchema: any[] | Model,\n    maybeModel?: Model,\n  ) {\n    const params = maybeModel === undefined ? [] : (paramsOrSchema as any[]);\n    const model = maybeModel === undefined ? (paramsOrSchema as Model) : maybeModel;\n    const results = await this.callAsync(sql, params);\n    if (results.fields.length === 1) {\n      const columnName = results.fields[0].name;\n      const rawData = results.rows.map((row) => row[columnName]);\n      return z.array(model).parse(rawData);\n    } else {\n      return z.array(model).parse(results.rows);\n    }\n  }\n\n  async callRow<Model extends z.ZodTypeAny>(sql: string, model: Model): Promise<z.infer<Model>>;\n  async callRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    params: any[],\n    model: Model,\n  ): Promise<z.infer<Model>>;\n  /**\n   * Calls the given sproc with the specified parameters.\n   * Returns exactly one row from the sproc that conforms to the given Zod schema.\n   */\n  async callRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    paramsOrSchema: any[] | Model,\n    maybeModel?: Model,\n  ) {\n    const params = maybeModel === undefined ? [] : (paramsOrSchema as any[]);\n    const model = maybeModel === undefined ? (paramsOrSchema as Model) : maybeModel;\n    const results = await this.callOneRowAsync(sql, params);\n    if (results.fields.length === 1) {\n      const columnName = results.fields[0].name;\n      return model.parse(results.rows[0][columnName]);\n    } else {\n      return model.parse(results.rows[0]);\n    }\n  }\n\n  async callOptionalRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    model: Model,\n  ): Promise<z.infer<Model> | null>;\n  async callOptionalRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    params: any[],\n    model: Model,\n  ): Promise<z.infer<Model> | null>;\n  /**\n   * Calls the given sproc with the specified parameters. Returns either null\n   * or a single row that conforms to the given Zod schema.\n   */\n  async callOptionalRow<Model extends z.ZodTypeAny>(\n    sql: string,\n    paramsOrSchema: any[] | Model,\n    maybeModel?: Model,\n  ) {\n    const params = maybeModel === undefined ? [] : (paramsOrSchema as any[]);\n    const model = maybeModel === undefined ? (paramsOrSchema as Model) : maybeModel;\n    const results = await this.callZeroOrOneRowAsync(sql, params);\n    if (results.rows.length === 0) {\n      return null;\n    } else if (results.fields.length === 1) {\n      const columnName = results.fields[0].name;\n      return model.parse(results.rows[0][columnName]);\n    } else {\n      return model.parse(results.rows[0]);\n    }\n  }\n\n  /**\n   * Executes a query with the specified parameters. Returns the number of rows affected.\n   */\n  async execute(sql: string, params: QueryParams = {}): Promise<number> {\n    const result = await this.queryAsync(sql, params);\n    return result.rowCount ?? 0;\n  }\n\n  /**\n   * Executes a query with the specified parameter, and errors if the query doesn't return exactly one row.\n   */\n  async executeRow(sql: string, params: QueryParams = {}) {\n    const rowCount = await this.execute(sql, params);\n    if (rowCount !== 1) {\n      throw new PostgresError('Incorrect rowCount: ' + rowCount, {\n        sql,\n        sqlParams: params,\n      });\n    }\n  }\n\n  /**\n   * Returns a {@link Cursor} for the given query. The cursor can be used to\n   * read results in batches, which is useful for large result sets.\n   */\n  private async queryCursorWithClient(\n    client: pg.PoolClient,\n    sql: string,\n    params: QueryParams,\n  ): Promise<Cursor> {\n    this._queryCount += 1;\n    debug('queryCursorWithClient()', 'sql:', debugString(sql));\n    debug('queryCursorWithClient()', 'params:', debugParams(params));\n    const { processedSql, paramsArray } = paramsToArray(sql, params, this.errorOnUnusedParameters);\n    lastQueryMap.set(client, processedSql);\n    return client.query(new Cursor(processedSql, paramsArray));\n  }\n\n  /**\n   * Returns an {@link CursorIterator} that can be used to iterate over the\n   * results of the query in batches, which is useful for large result sets.\n   * Each row will be parsed by the given Zod schema.\n   */\n  async queryCursor<Model extends z.ZodTypeAny>(\n    sql: string,\n    params: QueryParams,\n    model: Model,\n  ): Promise<CursorIterator<z.infer<Model>>> {\n    return this.queryCursorInternal(sql, params, model);\n  }\n\n  private async queryCursorInternal<Model extends z.ZodTypeAny>(\n    sql: string,\n    params: QueryParams,\n    model?: Model,\n  ): Promise<CursorIterator<z.infer<Model>>> {\n    const client = await this.getClientAsync();\n    const cursor = await this.queryCursorWithClient(client, sql, params);\n\n    let iterateCalled = false;\n    const iterator: CursorIterator<z.infer<Model>> = {\n      async *iterate(batchSize: number) {\n        // Safety check: if someone calls iterate multiple times, they're\n        // definitely doing something wrong.\n        if (iterateCalled) {\n          throw new Error('iterate() called multiple times');\n        }\n        iterateCalled = true;\n\n        try {\n          while (true) {\n            const rows = await cursor.read(batchSize);\n            if (rows.length === 0) {\n              break;\n            }\n\n            if (model) {\n              yield z.array(model).parse(rows);\n            } else {\n              yield rows;\n            }\n          }\n        } catch (err: any) {\n          throw enhanceError(err, sql, params);\n        } finally {\n          try {\n            await cursor.close();\n          } finally {\n            client.release();\n          }\n        }\n      },\n      stream(batchSize: number) {\n        const transform = new Transform({\n          readableObjectMode: true,\n          writableObjectMode: true,\n          transform(chunk, _encoding, callback) {\n            for (const row of chunk) {\n              this.push(row);\n            }\n            callback();\n          },\n        });\n\n        // TODO: use native `node:stream#compose` once it's stable.\n        const generator = iterator.iterate(batchSize);\n        const pipe = multipipe(Readable.from(generator), transform);\n\n        // When the underlying stream is closed, we need to make sure that the\n        // cursor is also closed. We do this by calling `return()` on the generator,\n        // which will trigger its `finally` block, which will in turn release\n        // the client and close the cursor. The fact that the stream is already\n        // closed by this point means that someone reading from the stream will\n        // never actually see the `null` value that's returned.\n        pipe.once('close', () => {\n          generator.return(null);\n        });\n\n        return pipe;\n      },\n    };\n    return iterator;\n  }\n\n  /**\n   * Set the schema to use for the search path.\n   *\n   * @param schema The schema name to use (can be \"null\" to unset the search path)\n   */\n  async setSearchSchema(schema: string | null) {\n    if (schema == null) {\n      this.searchSchema = schema;\n      return;\n    }\n\n    await this.queryAsync(`CREATE SCHEMA IF NOT EXISTS ${escapeIdentifier(schema)}`, {});\n    // We only set searchSchema after CREATE to avoid the above query() call using searchSchema.\n    this.searchSchema = schema;\n  }\n\n  /**\n   * Get the schema that is currently used for the search path.\n   *\n   * @returns schema in use (may be `null` to indicate no schema)\n   */\n  getSearchSchema(): string | null {\n    return this.searchSchema;\n  }\n\n  /**\n   * Generate, set, and return a random schema name.\n   *\n   * @param prefix The prefix of the new schema, only the first 28 characters will be used (after lowercasing).\n   * @returns The randomly-generated search schema.\n   */\n  async setRandomSearchSchemaAsync(prefix: string): Promise<string> {\n    // truncated prefix (max 28 characters)\n    const truncPrefix = prefix.slice(0, 28);\n    // timestamp in format YYYY-MM-DDTHH:MM:SS.SSSZ (guaranteed to not exceed 27 characters in the spec)\n    const timestamp = new Date().toISOString();\n    // random 6-character suffix to avoid clashes (approx 2 billion possible values)\n    const chars = '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ'.split('');\n    const suffix = Array.from({ length: 6 })\n      .map(() => chars[Math.floor(Math.random() * chars.length)])\n      .join('');\n\n    // Schema is guaranteed to have length at most 63 (= 28 + 1 + 27 + 1 + 6),\n    // which is the default PostgreSQL identifier limit.\n    // Note that this schema name will need quoting because of characters like ':', '-', etc\n    const schema = `${truncPrefix}_${timestamp}_${suffix}`;\n    await this.setSearchSchema(schema);\n    return schema;\n  }\n\n  /** The number of established connections. */\n  get totalCount() {\n    return this.pool?.totalCount ?? 0;\n  }\n\n  /** The number of idle connections. */\n  get idleCount() {\n    return this.pool?.idleCount ?? 0;\n  }\n\n  /** The number of queries waiting for a connection to become available. */\n  get waitingCount() {\n    return this.pool?.waitingCount ?? 0;\n  }\n\n  /** The total number of queries that have been executed by this pool. */\n  get queryCount() {\n    return this._queryCount;\n  }\n}\n"]}

package/dist/pool.test.js

@@ -2,7 +2,7 @@ import { Writable } from 'node:stream';
 import { pipeline } from 'node:stream/promises';
 import { afterAll, assert, beforeAll, describe, expect, it } from 'vitest';
 import { ZodError, z } from 'zod';
-import { callOptionalRow, callRow, callRows, queryAsync, queryCursor, queryOptionalRow, queryRow, queryRows, } from './default-pool.js';
+import { callOptionalRow, callRow, callRows, execute, queryAsync, queryCursor, queryOptionalRow, queryRow, queryRows, } from './default-pool.js';
 import { makePostgresTestUtils } from './test-utils.js';
 const postgresTestUtils = makePostgresTestUtils({
     database: 'prairielearn_postgres',
@@ -18,12 +18,12 @@ const SprocTwoColumnsSchema = z.object({
 describe('@prairielearn/postgres', function () {
     beforeAll(async () => {
         await postgresTestUtils.createDatabase();
-        await queryAsync('CREATE TABLE workspaces (id BIGSERIAL PRIMARY KEY, created_at TIMESTAMPTZ DEFAULT CURRENT_TIMESTAMP);', {});
-        await queryAsync('INSERT INTO workspaces (id) SELECT s FROM generate_series(1, 100) AS s', {});
-        await queryAsync('CREATE FUNCTION test_sproc_one_column(num_entries INT) RETURNS TABLE (id BIGINT) AS $$ BEGIN RETURN QUERY SELECT s::BIGINT AS id FROM generate_series(1, num_entries) AS s; END; $$ LANGUAGE plpgsql;', {});
-        await queryAsync('CREATE FUNCTION test_sproc_two_columns(num_entries INT) RETURNS TABLE (id BIGINT, negative INT) AS $$ BEGIN RETURN QUERY SELECT s::BIGINT AS id, -s AS negative FROM generate_series(1, num_entries) AS s; END; $$ LANGUAGE plpgsql;', {});
-        await queryAsync('CREATE FUNCTION test_sproc_one_column_ten_rows() RETURNS TABLE (id BIGINT) AS $$ BEGIN RETURN QUERY SELECT s::BIGINT AS id FROM generate_series(1, 10) AS s; END; $$ LANGUAGE plpgsql;', {});
-        await queryAsync('CREATE FUNCTION test_sproc_one_column_one_row(OUT id BIGINT) AS $$ BEGIN id = 1; END; $$ LANGUAGE plpgsql;', {});
+        await execute('CREATE TABLE workspaces (id BIGSERIAL PRIMARY KEY, created_at TIMESTAMPTZ DEFAULT CURRENT_TIMESTAMP);');
+        await execute('INSERT INTO workspaces (id) SELECT s FROM generate_series(1, 100) AS s');
+        await execute('CREATE FUNCTION test_sproc_one_column(num_entries INT) RETURNS TABLE (id BIGINT) AS $$ BEGIN RETURN QUERY SELECT s::BIGINT AS id FROM generate_series(1, num_entries) AS s; END; $$ LANGUAGE plpgsql;');
+        await execute('CREATE FUNCTION test_sproc_two_columns(num_entries INT) RETURNS TABLE (id BIGINT, negative INT) AS $$ BEGIN RETURN QUERY SELECT s::BIGINT AS id, -s AS negative FROM generate_series(1, num_entries) AS s; END; $$ LANGUAGE plpgsql;');
+        await execute('CREATE FUNCTION test_sproc_one_column_ten_rows() RETURNS TABLE (id BIGINT) AS $$ BEGIN RETURN QUERY SELECT s::BIGINT AS id FROM generate_series(1, 10) AS s; END; $$ LANGUAGE plpgsql;');
+        await execute('CREATE FUNCTION test_sproc_one_column_one_row(OUT id BIGINT) AS $$ BEGIN id = 1; END; $$ LANGUAGE plpgsql;');
     });
     afterAll(async () => {
         await postgresTestUtils.dropDatabase();

package/dist/pool.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"pool.test.js","sourceRoot":"","sources":["../src/pool.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAEhD,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC3E,OAAO,EAAE,QAAQ,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAElC,OAAO,EACL,eAAe,EACf,OAAO,EACP,QAAQ,EACR,UAAU,EACV,WAAW,EACX,gBAAgB,EAChB,QAAQ,EACR,SAAS,GACV,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,qBAAqB,EAAE,MAAM,iBAAiB,CAAC;AAExD,MAAM,iBAAiB,GAAG,qBAAqB,CAAC;IAC9C,QAAQ,EAAE,uBAAuB;CAClC,CAAC,CAAC;AAEH,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC;IAC/B,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;IACd,UAAU,EAAE,CAAC,CAAC,IAAI,EAAE;CACrB,CAAC,CAAC;AAEH,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC;IACrC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;IACd,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;CACrB,CAAC,CAAC;AAEH,QAAQ,CAAC,wBAAwB,EAAE;IACjC,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,iBAAiB,CAAC,cAAc,EAAE,CAAC;QACzC,MAAM,UAAU,CACd,uGAAuG,EACvG,EAAE,CACH,CAAC;QACF,MAAM,UAAU,CAAC,wEAAwE,EAAE,EAAE,CAAC,CAAC;QAC/F,MAAM,UAAU,CACd,uMAAuM,EACvM,EAAE,CACH,CAAC;QACF,MAAM,UAAU,CACd,sOAAsO,EACtO,EAAE,CACH,CAAC;QACF,MAAM,UAAU,CACd,wLAAwL,EACxL,EAAE,CACH,CAAC;QACF,MAAM,UAAU,CACd,4GAA4G,EAC5G,EAAE,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,KAAK,IAAI,EAAE;QAClB,MAAM,iBAAiB,CAAC,YAAY,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,wCAAwC;YACxC,MAAM,IAAI,GAAG,UAAU,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;YAC/C,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,qDAAqD;YACrD,MAAM,IAAI,GAAG,UAAU,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;YAC1C,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,gCAAgC,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,IAAI,GAAG,UAAU,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC;YAChD,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,IAAI,GAAG,UAAU,CAAC,YAAY,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC;YACjE,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,2CAA2C,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YACtF,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,0CAA0C,EAAE,eAAe,CAAC,CAAC;YAC1F,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAC9B,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;YAClC,MAAM,IAAI,GAAG,MAAM,SAAS,CAC1B,0CAA0C,EAC1C,CAAC,EAAE,CAAC,EACJ,eAAe,CAChB,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,yCAAyC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YAClF,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,wCAAwC,EAAE,eAAe,CAAC,CAAC;YACtF,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAC1B,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;YAClC,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,yCAAyC,EAAE,CAAC,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC;YAC5F,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,IAAI,GAAG,QAAQ,CAAC,yCAAyC,EAAE,eAAe,CAAC,CAAC;YAClF,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,IAAI,GAAG,QAAQ,CAAC,0BAA0B,EAAE,eAAe,CAAC,CAAC;YACnE,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,yCAAyC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YAClF,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,wCAAwC,EAAE,eAAe,CAAC,CAAC;YAC9F,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YACtB,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;YAC3B,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;YAClC,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAChC,yCAAyC,EACzC,CAAC,CAAC,CAAC,EACH,eAAe,CAChB,CAAC;YACF,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YACtB,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;YACtC,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAChC,yCAAyC,EACzC,eAAe,CAChB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACrB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,IAAI,GAAG,gBAAgB,CAAC,0BAA0B,EAAE,eAAe,CAAC,CAAC;YAC3E,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,gCAAgC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YAC1E,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;YAClC,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,uBAAuB,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YACvE,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,wBAAwB,EAAE,CAAC,EAAE,CAAC,EAAE,qBAAqB,CAAC,CAAC;YACnF,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAC9B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC;YACnC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YAChC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,+BAA+B,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YACvE,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;YAClC,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YACpE,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,wBAAwB,EAAE,CAAC,CAAC,CAAC,EAAE,qBAAqB,CAAC,CAAC;YAChF,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAC1B,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,GAAG,GAAG,OAAO,CAAC,wBAAwB,EAAE,CAAC,CAAC,CAAC,EAAE,qBAAqB,CAAC,CAAC;YAC1E,MAAM,MAAM,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,IAAI,GAAG,OAAO,CAAC,wBAAwB,EAAE,CAAC,GAAG,CAAC,EAAE,qBAAqB,CAAC,CAAC;YAC7E,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,+BAA+B,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YAC/E,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;YAClC,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YAC5E,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,wBAAwB,EAAE,CAAC,CAAC,CAAC,EAAE,qBAAqB,CAAC,CAAC;YACxF,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YACtB,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;YAC3B,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,wBAAwB,EAAE,CAAC,CAAC,CAAC,EAAE,qBAAqB,CAAC,CAAC;YACxF,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACrB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,IAAI,GAAG,eAAe,CAAC,wBAAwB,EAAE,CAAC,GAAG,CAAC,EAAE,qBAAqB,CAAC,CAAC;YACrF,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,mBAAmB,EAAE,KAAK,IAAI,EAAE;YACjC,MAAM,MAAM,GAAG,MAAM,WAAW,CAC9B,4CAA4C,EAC5C,EAAE,EACF,CAAC,CAAC,OAAO,EAAE,CACZ,CAAC;YACF,MAAM,UAAU,GAAG,EAAE,CAAC;YACtB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;gBAC5C,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC;YACD,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,yCAAyC,EAAE,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;YAC7F,MAAM,UAAU,GAAG,EAAE,CAAC;YACtB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3C,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC;YACD,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAClC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,MAAM,GAAG,WAAW,CAAC,0CAA0C,EAAE,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;YACxF,MAAM,UAAU,GAAG,EAAE,CAAC;YACtB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,CAAC,MAAM,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;gBACpD,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC;YACD,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gBAAgB,EAAE,KAAK,IAAI,EAAE;YAC9B,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,eAAe,EAAE,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;YAEnE,KAAK,UAAU,WAAW;gBACxB,MAAM,OAAO,GAAG,EAAE,CAAC;gBACnB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;oBAC5C,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;gBACxB,CAAC;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC;YAED,MAAM,UAAU,GAAG,MAAM,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;YAC3D,MAAM,CAAC,UAAU,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YACrC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;YACjD,MAAM,CAAC,SAAS,CAAE,UAAkB,CAAC,IAAI,CAAC,CAAC;YAC3C,MAAM,CAAC,KAAK,CAAE,UAAkB,CAAC,IAAI,CAAC,GAAG,EAAE,eAAe,CAAC,CAAC;YAC5D,MAAM,CAAC,SAAS,CAAE,UAAkB,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;YACzD,MAAM,CAAC,SAAS,CAAE,UAAkB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACpD,MAAM,CAAC,KAAK,CAAE,UAAkB,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC;YAC/B,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;SACf,CAAC,CAAC;QAEH,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC;YAClC,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;SACxB,CAAC,CAAC;QAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;YACxB,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;gBAC9C,MAAM,MAAM,GAAG,MAAM,WAAW,CAC9B,0DAA0D,EAC1D,EAAE,EACF,eAAe,CAChB,CAAC;gBACF,MAAM,OAAO,GAAG,EAAE,CAAC;gBACnB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;oBAC5C,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;gBACxB,CAAC;gBACD,MAAM,CAAC,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gBAC7B,MAAM,SAAS,GAAG,OAAO,CAAC,CAAC,CAAQ,CAAC;gBACpC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;gBAChC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YACtC,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;gBAClD,MAAM,MAAM,GAAG,MAAM,WAAW,CAC9B,0DAA0D,EAC1D,EAAE,EACF,kBAAkB,CACnB,CAAC;gBAEF,KAAK,UAAU,WAAW;oBACxB,MAAM,OAAO,GAAG,EAAE,CAAC;oBACnB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;wBAC5C,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;oBACxB,CAAC;oBACD,OAAO,OAAO,CAAC;gBACjB,CAAC;gBAED,MAAM,UAAU,GAAG,MAAM,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;gBAC3D,MAAM,CAAC,UAAU,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;gBACxC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YACzC,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;YACtB,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;gBAC9C,MAAM,MAAM,GAAG,MAAM,WAAW,CAC9B,0DAA0D,EAC1D,EAAE,EACF,eAAe,CAChB,CAAC;gBACF,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAChC,MAAM,OAAO,GAAG,EAAE,CAAC;gBACnB,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;oBAC/B,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACpB,CAAC;gBAED,MAAM,CAAC,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YAC/B,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;gBACpD,MAAM,MAAM,GAAG,MAAM,WAAW,CAC9B,2CAA2C,EAC3C,EAAE,EACF,kBAAkB,CACnB,CAAC;gBACF,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAEhC,KAAK,UAAU,WAAW;oBACxB,MAAM,OAAO,GAAG,EAAE,CAAC;oBACnB,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;wBAC/B,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBACpB,CAAC;oBACD,OAAO,OAAO,CAAC;gBACjB,CAAC;gBAED,MAAM,UAAU,GAAG,MAAM,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;gBAC3D,MAAM,CAAC,UAAU,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;gBACxC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACxC,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;gBAC3D,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,2BAA2B,EAAE,EAAE,EAAE,eAAe,CAAC,CAAC;gBACnF,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAEhC,MAAM,IAAI,GAAU,EAAE,CAAC;gBACvB,MAAM,EAAE,GAAG,IAAI,eAAe,EAAE,CAAC;gBACjC,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC;oBAC5B,UAAU,EAAE,IAAI;oBAChB,KAAK,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ;wBAC9B,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;wBAEjB,qEAAqE;wBACrE,oEAAoE;wBACpE,yCAAyC;wBACzC,EAAE,CAAC,KAAK,EAAE,CAAC;wBACX,QAAQ,EAAE,CAAC;oBACb,CAAC;iBACF,CAAC,CAAC;gBAEH,MAAM,MAAM,CAAC,QAAQ,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;gBAClF,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;YAC3B,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { Writable } from 'node:stream';\nimport { pipeline } from 'node:stream/promises';\n\nimport { afterAll, assert, beforeAll, describe, expect, it } from 'vitest';\nimport { ZodError, z } from 'zod';\n\nimport {\n  callOptionalRow,\n  callRow,\n  callRows,\n  queryAsync,\n  queryCursor,\n  queryOptionalRow,\n  queryRow,\n  queryRows,\n} from './default-pool.js';\nimport { makePostgresTestUtils } from './test-utils.js';\n\nconst postgresTestUtils = makePostgresTestUtils({\n  database: 'prairielearn_postgres',\n});\n\nconst WorkspaceSchema = z.object({\n  id: z.string(),\n  created_at: z.date(),\n});\n\nconst SprocTwoColumnsSchema = z.object({\n  id: z.string(),\n  negative: z.number(),\n});\n\ndescribe('@prairielearn/postgres', function () {\n  beforeAll(async () => {\n    await postgresTestUtils.createDatabase();\n    await queryAsync(\n      'CREATE TABLE workspaces (id BIGSERIAL PRIMARY KEY, created_at TIMESTAMPTZ DEFAULT CURRENT_TIMESTAMP);',\n      {},\n    );\n    await queryAsync('INSERT INTO workspaces (id) SELECT s FROM generate_series(1, 100) AS s', {});\n    await queryAsync(\n      'CREATE FUNCTION test_sproc_one_column(num_entries INT) RETURNS TABLE (id BIGINT) AS $$ BEGIN RETURN QUERY SELECT s::BIGINT AS id FROM generate_series(1, num_entries) AS s; END; $$ LANGUAGE plpgsql;',\n      {},\n    );\n    await queryAsync(\n      'CREATE FUNCTION test_sproc_two_columns(num_entries INT) RETURNS TABLE (id BIGINT, negative INT) AS $$ BEGIN RETURN QUERY SELECT s::BIGINT AS id, -s AS negative FROM generate_series(1, num_entries) AS s; END; $$ LANGUAGE plpgsql;',\n      {},\n    );\n    await queryAsync(\n      'CREATE FUNCTION test_sproc_one_column_ten_rows() RETURNS TABLE (id BIGINT) AS $$ BEGIN RETURN QUERY SELECT s::BIGINT AS id FROM generate_series(1, 10) AS s; END; $$ LANGUAGE plpgsql;',\n      {},\n    );\n    await queryAsync(\n      'CREATE FUNCTION test_sproc_one_column_one_row(OUT id BIGINT) AS $$ BEGIN id = 1; END; $$ LANGUAGE plpgsql;',\n      {},\n    );\n  });\n\n  afterAll(async () => {\n    await postgresTestUtils.dropDatabase();\n  });\n\n  describe('paramsToArray', () => {\n    it('enforces SQL must be a string', async () => {\n      // @ts-expect-error SQL must be a string\n      const rows = queryAsync({ invalid: true }, {});\n      await expect(rows).rejects.toThrow('SQL must be a string');\n    });\n\n    it('enforces params must be array or object', async () => {\n      // @ts-expect-error params must be an array or object\n      const rows = queryAsync('SELECT 33;', 33);\n      await expect(rows).rejects.toThrow('params must be array or object');\n    });\n\n    it('rejects missing parameters', async () => {\n      const rows = queryAsync('SELECT $missing;', {});\n      await expect(rows).rejects.toThrow('Missing parameter');\n    });\n\n    it('rejects unused parameters in testing', async () => {\n      const rows = queryAsync('SELECT 33;', { unsed_parameter: true });\n      await expect(rows).rejects.toThrow('Unused parameter');\n    });\n  });\n\n  describe('queryRows', () => {\n    it('handles single column', async () => {\n      const rows = await queryRows('SELECT id FROM workspaces WHERE id <= 10;', z.string());\n      assert.lengthOf(rows, 10);\n      assert.equal(rows[0], '1');\n    });\n\n    it('handles multiple columns', async () => {\n      const rows = await queryRows('SELECT * FROM workspaces WHERE id <= 10;', WorkspaceSchema);\n      assert.lengthOf(rows, 10);\n      assert.equal(rows[0].id, '1');\n      assert.isNotNull(rows[0].created_at);\n    });\n\n    it('handles parameters', async () => {\n      const rows = await queryRows(\n        'SELECT * FROM workspaces WHERE id <= $1;',\n        [10],\n        WorkspaceSchema,\n      );\n      assert.lengthOf(rows, 10);\n    });\n  });\n\n  describe('queryRow', () => {\n    it('handles single column', async () => {\n      const row = await queryRow('SELECT id FROM workspaces WHERE id = 1;', z.string());\n      assert.equal(row, '1');\n    });\n\n    it('handles multiple columns', async () => {\n      const row = await queryRow('SELECT * FROM workspaces WHERE id = 1;', WorkspaceSchema);\n      assert.equal(row.id, '1');\n      assert.isNotNull(row.created_at);\n    });\n\n    it('handles parameters', async () => {\n      const row = await queryRow('SELECT * FROM workspaces WHERE id = $1;', [1], WorkspaceSchema);\n      assert.equal(row.id, '1');\n    });\n\n    it('rejects results with zero rows', async () => {\n      const rows = queryRow('SELECT * FROM workspaces WHERE id = -1;', WorkspaceSchema);\n      await expect(rows).rejects.toThrow('Incorrect rowCount: 0');\n    });\n\n    it('rejects results with multiple rows', async () => {\n      const rows = queryRow('SELECT * FROM workspaces', WorkspaceSchema);\n      await expect(rows).rejects.toThrow('Incorrect rowCount: 100');\n    });\n  });\n\n  describe('queryOptionalRow', () => {\n    it('handles single column', async () => {\n      const row = await queryRow('SELECT id FROM workspaces WHERE id = 1;', z.string());\n      assert.equal(row, '1');\n    });\n\n    it('handles multiple columns', async () => {\n      const row = await queryOptionalRow('SELECT * FROM workspaces WHERE id = 1;', WorkspaceSchema);\n      assert.isNotNull(row);\n      assert.equal(row?.id, '1');\n      assert.isNotNull(row?.created_at);\n    });\n\n    it('handles parameters', async () => {\n      const row = await queryOptionalRow(\n        'SELECT * FROM workspaces WHERE id = $1;',\n        [1],\n        WorkspaceSchema,\n      );\n      assert.isNotNull(row);\n      assert.equal(row?.id, '1');\n    });\n\n    it('handles missing result', async () => {\n      const row = await queryOptionalRow(\n        'SELECT * FROM workspaces WHERE id = -1;',\n        WorkspaceSchema,\n      );\n      assert.isNull(row);\n    });\n\n    it('rejects with multiple rows', async () => {\n      const rows = queryOptionalRow('SELECT * FROM workspaces', WorkspaceSchema);\n      await expect(rows).rejects.toThrow('Incorrect rowCount: 100');\n    });\n  });\n\n  describe('callRows', () => {\n    it('handles single column', async () => {\n      const rows = await callRows('test_sproc_one_column_ten_rows', z.string());\n      assert.lengthOf(rows, 10);\n      assert.equal(rows[0], '1');\n    });\n\n    it('handles parameters', async () => {\n      const rows = await callRows('test_sproc_one_column', [10], z.string());\n      assert.lengthOf(rows, 10);\n      assert.equal(rows[0], '1');\n    });\n\n    it('handles multiple columns', async () => {\n      const rows = await callRows('test_sproc_two_columns', [20], SprocTwoColumnsSchema);\n      assert.lengthOf(rows, 20);\n      assert.equal(rows[0].id, '1');\n      assert.equal(rows[0].negative, -1);\n      assert.equal(rows[19].id, '20');\n      assert.equal(rows[19].negative, -20);\n    });\n  });\n\n  describe('callRow', () => {\n    it('handles single column', async () => {\n      const row = await callRow('test_sproc_one_column_one_row', z.string());\n      assert.equal(row, '1');\n    });\n\n    it('handles parameters', async () => {\n      const row = await callRow('test_sproc_one_column', [1], z.string());\n      assert.equal(row, '1');\n    });\n\n    it('handles multiple columns', async () => {\n      const row = await callRow('test_sproc_two_columns', [1], SprocTwoColumnsSchema);\n      assert.equal(row.id, '1');\n      assert.equal(row.negative, -1);\n    });\n\n    it('rejects results with zero rows', async () => {\n      const row = callRow('test_sproc_two_columns', [0], SprocTwoColumnsSchema);\n      await expect(row).rejects.toThrow('Incorrect rowCount: 0');\n    });\n\n    it('rejects results with multiple rows', async () => {\n      const rows = callRow('test_sproc_two_columns', [100], SprocTwoColumnsSchema);\n      await expect(rows).rejects.toThrow('Incorrect rowCount: 100');\n    });\n  });\n\n  describe('callOptionalRow', () => {\n    it('handles single column', async () => {\n      const row = await callOptionalRow('test_sproc_one_column_one_row', z.string());\n      assert.equal(row, '1');\n    });\n\n    it('handles parameters', async () => {\n      const row = await callOptionalRow('test_sproc_one_column', [1], z.string());\n      assert.equal(row, '1');\n    });\n\n    it('handles multiple columns', async () => {\n      const row = await callOptionalRow('test_sproc_two_columns', [1], SprocTwoColumnsSchema);\n      assert.isNotNull(row);\n      assert.equal(row?.id, '1');\n      assert.equal(row?.negative, -1);\n    });\n\n    it('handles results with zero rows', async () => {\n      const row = await callOptionalRow('test_sproc_two_columns', [0], SprocTwoColumnsSchema);\n      assert.isNull(row);\n    });\n\n    it('rejects results with multiple rows', async () => {\n      const rows = callOptionalRow('test_sproc_two_columns', [100], SprocTwoColumnsSchema);\n      await expect(rows).rejects.toThrow('Incorrect rowCount: 100');\n    });\n  });\n\n  describe('queryCursor', () => {\n    it('returns zero rows', async () => {\n      const cursor = await queryCursor(\n        'SELECT * FROM workspaces WHERE id = 10000;',\n        {},\n        z.unknown(),\n      );\n      const rowBatches = [];\n      for await (const rows of cursor.iterate(10)) {\n        rowBatches.push(rows);\n      }\n      assert.lengthOf(rowBatches, 0);\n    });\n\n    it('returns one row at a time', async () => {\n      const cursor = await queryCursor('SELECT * FROM workspaces WHERE id <= 2;', {}, z.unknown());\n      const rowBatches = [];\n      for await (const rows of cursor.iterate(1)) {\n        rowBatches.push(rows);\n      }\n      assert.lengthOf(rowBatches, 2);\n      assert.lengthOf(rowBatches[0], 1);\n      assert.lengthOf(rowBatches[1], 1);\n    });\n\n    it('returns all rows at once', async () => {\n      const cursor = queryCursor('SELECT * FROM workspaces WHERE id <= 10;', {}, z.unknown());\n      const rowBatches = [];\n      for await (const rows of (await cursor).iterate(10)) {\n        rowBatches.push(rows);\n      }\n      assert.lengthOf(rowBatches, 1);\n      assert.lengthOf(rowBatches[0], 10);\n    });\n\n    it('handles errors', async () => {\n      const cursor = await queryCursor('NOT VALID SQL', {}, z.unknown());\n\n      async function readAllRows() {\n        const allRows = [];\n        for await (const rows of cursor.iterate(10)) {\n          allRows.push(...rows);\n        }\n        return allRows;\n      }\n\n      const maybeError = await readAllRows().catch((err) => err);\n      assert.instanceOf(maybeError, Error);\n      assert.match(maybeError.message, /syntax error/);\n      assert.isDefined((maybeError as any).data);\n      assert.equal((maybeError as any).data.sql, 'NOT VALID SQL');\n      assert.deepEqual((maybeError as any).data.sqlParams, {});\n      assert.isDefined((maybeError as any).data.sqlError);\n      assert.equal((maybeError as any).data.sqlError.severity, 'ERROR');\n    });\n  });\n\n  describe('queryCursor', () => {\n    const WorkspaceSchema = z.object({\n      id: z.string(),\n    });\n\n    const BadWorkspaceSchema = z.object({\n      badProperty: z.string(),\n    });\n\n    describe('iterator', () => {\n      it('validates with provided schema', async () => {\n        const cursor = await queryCursor(\n          'SELECT * FROM workspaces WHERE id <= 10 ORDER BY id ASC;',\n          {},\n          WorkspaceSchema,\n        );\n        const allRows = [];\n        for await (const rows of cursor.iterate(10)) {\n          allRows.push(...rows);\n        }\n        assert.lengthOf(allRows, 10);\n        const workspace = allRows[0] as any;\n        assert.equal(workspace.id, '1');\n        assert.isUndefined(workspace.state);\n      });\n\n      it('throws error when validation fails', async () => {\n        const cursor = await queryCursor(\n          'SELECT * FROM workspaces WHERE id <= 10 ORDER BY id ASC;',\n          {},\n          BadWorkspaceSchema,\n        );\n\n        async function readAllRows() {\n          const allRows = [];\n          for await (const rows of cursor.iterate(10)) {\n            allRows.push(...rows);\n          }\n          return allRows;\n        }\n\n        const maybeError = await readAllRows().catch((err) => err);\n        assert.instanceOf(maybeError, ZodError);\n        assert.lengthOf(maybeError.errors, 10);\n      });\n    });\n\n    describe('stream', () => {\n      it('validates with provided schema', async () => {\n        const cursor = await queryCursor(\n          'SELECT * FROM workspaces WHERE id <= 10 ORDER BY id ASC;',\n          {},\n          WorkspaceSchema,\n        );\n        const stream = cursor.stream(1);\n        const allRows = [];\n        for await (const row of stream) {\n          allRows.push(row);\n        }\n\n        assert.lengthOf(allRows, 10);\n      });\n\n      it('emits an error when validation fails', async () => {\n        const cursor = await queryCursor(\n          'SELECT * FROM workspaces ORDER BY id ASC;',\n          {},\n          BadWorkspaceSchema,\n        );\n        const stream = cursor.stream(1);\n\n        async function readAllRows() {\n          const allRows = [];\n          for await (const row of stream) {\n            allRows.push(row);\n          }\n          return allRows;\n        }\n\n        const maybeError = await readAllRows().catch((err) => err);\n        assert.instanceOf(maybeError, ZodError);\n        assert.lengthOf(maybeError.errors, 1);\n      });\n\n      it('closes the cursor when the stream is closed', async () => {\n        const cursor = await queryCursor('SELECT * FROM workspaces;', {}, WorkspaceSchema);\n        const stream = cursor.stream(1);\n\n        const rows: any[] = [];\n        const ac = new AbortController();\n        const writable = new Writable({\n          objectMode: true,\n          write(chunk, _encoding, callback) {\n            rows.push(chunk);\n\n            // After receiving the first row, abort the stream. This lets us test\n            // that the underlying cursor is closed. If it is *not* closed, this\n            // `after` hook will fail with a timeout.\n            ac.abort();\n            callback();\n          },\n        });\n\n        await expect(pipeline(stream, writable, { signal: ac.signal })).rejects.toThrow();\n        assert.lengthOf(rows, 1);\n      });\n    });\n  });\n});\n"]}
+{"version":3,"file":"pool.test.js","sourceRoot":"","sources":["../src/pool.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAEhD,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC3E,OAAO,EAAE,QAAQ,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAElC,OAAO,EACL,eAAe,EACf,OAAO,EACP,QAAQ,EACR,OAAO,EACP,UAAU,EACV,WAAW,EACX,gBAAgB,EAChB,QAAQ,EACR,SAAS,GACV,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,qBAAqB,EAAE,MAAM,iBAAiB,CAAC;AAExD,MAAM,iBAAiB,GAAG,qBAAqB,CAAC;IAC9C,QAAQ,EAAE,uBAAuB;CAClC,CAAC,CAAC;AAEH,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC;IAC/B,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;IACd,UAAU,EAAE,CAAC,CAAC,IAAI,EAAE;CACrB,CAAC,CAAC;AAEH,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC;IACrC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;IACd,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;CACrB,CAAC,CAAC;AAEH,QAAQ,CAAC,wBAAwB,EAAE;IACjC,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,iBAAiB,CAAC,cAAc,EAAE,CAAC;QACzC,MAAM,OAAO,CACX,uGAAuG,CACxG,CAAC;QACF,MAAM,OAAO,CAAC,wEAAwE,CAAC,CAAC;QACxF,MAAM,OAAO,CACX,uMAAuM,CACxM,CAAC;QACF,MAAM,OAAO,CACX,sOAAsO,CACvO,CAAC;QACF,MAAM,OAAO,CACX,wLAAwL,CACzL,CAAC;QACF,MAAM,OAAO,CACX,4GAA4G,CAC7G,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,KAAK,IAAI,EAAE;QAClB,MAAM,iBAAiB,CAAC,YAAY,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,wCAAwC;YACxC,MAAM,IAAI,GAAG,UAAU,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;YAC/C,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,qDAAqD;YACrD,MAAM,IAAI,GAAG,UAAU,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;YAC1C,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,gCAAgC,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,IAAI,GAAG,UAAU,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC;YAChD,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,IAAI,GAAG,UAAU,CAAC,YAAY,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC;YACjE,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,2CAA2C,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YACtF,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,0CAA0C,EAAE,eAAe,CAAC,CAAC;YAC1F,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAC9B,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;YAClC,MAAM,IAAI,GAAG,MAAM,SAAS,CAC1B,0CAA0C,EAC1C,CAAC,EAAE,CAAC,EACJ,eAAe,CAChB,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,yCAAyC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YAClF,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,wCAAwC,EAAE,eAAe,CAAC,CAAC;YACtF,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAC1B,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;YAClC,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,yCAAyC,EAAE,CAAC,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC;YAC5F,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,IAAI,GAAG,QAAQ,CAAC,yCAAyC,EAAE,eAAe,CAAC,CAAC;YAClF,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,IAAI,GAAG,QAAQ,CAAC,0BAA0B,EAAE,eAAe,CAAC,CAAC;YACnE,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,yCAAyC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YAClF,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,wCAAwC,EAAE,eAAe,CAAC,CAAC;YAC9F,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YACtB,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;YAC3B,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;YAClC,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAChC,yCAAyC,EACzC,CAAC,CAAC,CAAC,EACH,eAAe,CAChB,CAAC;YACF,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YACtB,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;YACtC,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAChC,yCAAyC,EACzC,eAAe,CAChB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACrB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,IAAI,GAAG,gBAAgB,CAAC,0BAA0B,EAAE,eAAe,CAAC,CAAC;YAC3E,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,gCAAgC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YAC1E,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;YAClC,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,uBAAuB,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YACvE,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,wBAAwB,EAAE,CAAC,EAAE,CAAC,EAAE,qBAAqB,CAAC,CAAC;YACnF,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAC9B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC;YACnC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YAChC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,+BAA+B,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YACvE,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;YAClC,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YACpE,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,wBAAwB,EAAE,CAAC,CAAC,CAAC,EAAE,qBAAqB,CAAC,CAAC;YAChF,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAC1B,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,GAAG,GAAG,OAAO,CAAC,wBAAwB,EAAE,CAAC,CAAC,CAAC,EAAE,qBAAqB,CAAC,CAAC;YAC1E,MAAM,MAAM,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,IAAI,GAAG,OAAO,CAAC,wBAAwB,EAAE,CAAC,GAAG,CAAC,EAAE,qBAAqB,CAAC,CAAC;YAC7E,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,+BAA+B,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YAC/E,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;YAClC,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YAC5E,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,wBAAwB,EAAE,CAAC,CAAC,CAAC,EAAE,qBAAqB,CAAC,CAAC;YACxF,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YACtB,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;YAC3B,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,wBAAwB,EAAE,CAAC,CAAC,CAAC,EAAE,qBAAqB,CAAC,CAAC;YACxF,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACrB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,IAAI,GAAG,eAAe,CAAC,wBAAwB,EAAE,CAAC,GAAG,CAAC,EAAE,qBAAqB,CAAC,CAAC;YACrF,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,mBAAmB,EAAE,KAAK,IAAI,EAAE;YACjC,MAAM,MAAM,GAAG,MAAM,WAAW,CAC9B,4CAA4C,EAC5C,EAAE,EACF,CAAC,CAAC,OAAO,EAAE,CACZ,CAAC;YACF,MAAM,UAAU,GAAG,EAAE,CAAC;YACtB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;gBAC5C,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC;YACD,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,yCAAyC,EAAE,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;YAC7F,MAAM,UAAU,GAAG,EAAE,CAAC;YACtB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3C,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC;YACD,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAClC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,MAAM,GAAG,WAAW,CAAC,0CAA0C,EAAE,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;YACxF,MAAM,UAAU,GAAG,EAAE,CAAC;YACtB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,CAAC,MAAM,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;gBACpD,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC;YACD,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gBAAgB,EAAE,KAAK,IAAI,EAAE;YAC9B,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,eAAe,EAAE,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;YAEnE,KAAK,UAAU,WAAW;gBACxB,MAAM,OAAO,GAAG,EAAE,CAAC;gBACnB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;oBAC5C,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;gBACxB,CAAC;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC;YAED,MAAM,UAAU,GAAG,MAAM,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;YAC3D,MAAM,CAAC,UAAU,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YACrC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;YACjD,MAAM,CAAC,SAAS,CAAE,UAAkB,CAAC,IAAI,CAAC,CAAC;YAC3C,MAAM,CAAC,KAAK,CAAE,UAAkB,CAAC,IAAI,CAAC,GAAG,EAAE,eAAe,CAAC,CAAC;YAC5D,MAAM,CAAC,SAAS,CAAE,UAAkB,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;YACzD,MAAM,CAAC,SAAS,CAAE,UAAkB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACpD,MAAM,CAAC,KAAK,CAAE,UAAkB,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC;YAC/B,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;SACf,CAAC,CAAC;QAEH,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC;YAClC,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;SACxB,CAAC,CAAC;QAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;YACxB,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;gBAC9C,MAAM,MAAM,GAAG,MAAM,WAAW,CAC9B,0DAA0D,EAC1D,EAAE,EACF,eAAe,CAChB,CAAC;gBACF,MAAM,OAAO,GAAG,EAAE,CAAC;gBACnB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;oBAC5C,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;gBACxB,CAAC;gBACD,MAAM,CAAC,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gBAC7B,MAAM,SAAS,GAAG,OAAO,CAAC,CAAC,CAAQ,CAAC;gBACpC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;gBAChC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YACtC,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;gBAClD,MAAM,MAAM,GAAG,MAAM,WAAW,CAC9B,0DAA0D,EAC1D,EAAE,EACF,kBAAkB,CACnB,CAAC;gBAEF,KAAK,UAAU,WAAW;oBACxB,MAAM,OAAO,GAAG,EAAE,CAAC;oBACnB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;wBAC5C,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;oBACxB,CAAC;oBACD,OAAO,OAAO,CAAC;gBACjB,CAAC;gBAED,MAAM,UAAU,GAAG,MAAM,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;gBAC3D,MAAM,CAAC,UAAU,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;gBACxC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YACzC,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;YACtB,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;gBAC9C,MAAM,MAAM,GAAG,MAAM,WAAW,CAC9B,0DAA0D,EAC1D,EAAE,EACF,eAAe,CAChB,CAAC;gBACF,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAChC,MAAM,OAAO,GAAG,EAAE,CAAC;gBACnB,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;oBAC/B,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACpB,CAAC;gBAED,MAAM,CAAC,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YAC/B,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;gBACpD,MAAM,MAAM,GAAG,MAAM,WAAW,CAC9B,2CAA2C,EAC3C,EAAE,EACF,kBAAkB,CACnB,CAAC;gBACF,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAEhC,KAAK,UAAU,WAAW;oBACxB,MAAM,OAAO,GAAG,EAAE,CAAC;oBACnB,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;wBAC/B,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBACpB,CAAC;oBACD,OAAO,OAAO,CAAC;gBACjB,CAAC;gBAED,MAAM,UAAU,GAAG,MAAM,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;gBAC3D,MAAM,CAAC,UAAU,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;gBACxC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACxC,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;gBAC3D,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,2BAA2B,EAAE,EAAE,EAAE,eAAe,CAAC,CAAC;gBACnF,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAEhC,MAAM,IAAI,GAAU,EAAE,CAAC;gBACvB,MAAM,EAAE,GAAG,IAAI,eAAe,EAAE,CAAC;gBACjC,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC;oBAC5B,UAAU,EAAE,IAAI;oBAChB,KAAK,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ;wBAC9B,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;wBAEjB,qEAAqE;wBACrE,oEAAoE;wBACpE,yCAAyC;wBACzC,EAAE,CAAC,KAAK,EAAE,CAAC;wBACX,QAAQ,EAAE,CAAC;oBACb,CAAC;iBACF,CAAC,CAAC;gBAEH,MAAM,MAAM,CAAC,QAAQ,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;gBAClF,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;YAC3B,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { Writable } from 'node:stream';\nimport { pipeline } from 'node:stream/promises';\n\nimport { afterAll, assert, beforeAll, describe, expect, it } from 'vitest';\nimport { ZodError, z } from 'zod';\n\nimport {\n  callOptionalRow,\n  callRow,\n  callRows,\n  execute,\n  queryAsync,\n  queryCursor,\n  queryOptionalRow,\n  queryRow,\n  queryRows,\n} from './default-pool.js';\nimport { makePostgresTestUtils } from './test-utils.js';\n\nconst postgresTestUtils = makePostgresTestUtils({\n  database: 'prairielearn_postgres',\n});\n\nconst WorkspaceSchema = z.object({\n  id: z.string(),\n  created_at: z.date(),\n});\n\nconst SprocTwoColumnsSchema = z.object({\n  id: z.string(),\n  negative: z.number(),\n});\n\ndescribe('@prairielearn/postgres', function () {\n  beforeAll(async () => {\n    await postgresTestUtils.createDatabase();\n    await execute(\n      'CREATE TABLE workspaces (id BIGSERIAL PRIMARY KEY, created_at TIMESTAMPTZ DEFAULT CURRENT_TIMESTAMP);',\n    );\n    await execute('INSERT INTO workspaces (id) SELECT s FROM generate_series(1, 100) AS s');\n    await execute(\n      'CREATE FUNCTION test_sproc_one_column(num_entries INT) RETURNS TABLE (id BIGINT) AS $$ BEGIN RETURN QUERY SELECT s::BIGINT AS id FROM generate_series(1, num_entries) AS s; END; $$ LANGUAGE plpgsql;',\n    );\n    await execute(\n      'CREATE FUNCTION test_sproc_two_columns(num_entries INT) RETURNS TABLE (id BIGINT, negative INT) AS $$ BEGIN RETURN QUERY SELECT s::BIGINT AS id, -s AS negative FROM generate_series(1, num_entries) AS s; END; $$ LANGUAGE plpgsql;',\n    );\n    await execute(\n      'CREATE FUNCTION test_sproc_one_column_ten_rows() RETURNS TABLE (id BIGINT) AS $$ BEGIN RETURN QUERY SELECT s::BIGINT AS id FROM generate_series(1, 10) AS s; END; $$ LANGUAGE plpgsql;',\n    );\n    await execute(\n      'CREATE FUNCTION test_sproc_one_column_one_row(OUT id BIGINT) AS $$ BEGIN id = 1; END; $$ LANGUAGE plpgsql;',\n    );\n  });\n\n  afterAll(async () => {\n    await postgresTestUtils.dropDatabase();\n  });\n\n  describe('paramsToArray', () => {\n    it('enforces SQL must be a string', async () => {\n      // @ts-expect-error SQL must be a string\n      const rows = queryAsync({ invalid: true }, {});\n      await expect(rows).rejects.toThrow('SQL must be a string');\n    });\n\n    it('enforces params must be array or object', async () => {\n      // @ts-expect-error params must be an array or object\n      const rows = queryAsync('SELECT 33;', 33);\n      await expect(rows).rejects.toThrow('params must be array or object');\n    });\n\n    it('rejects missing parameters', async () => {\n      const rows = queryAsync('SELECT $missing;', {});\n      await expect(rows).rejects.toThrow('Missing parameter');\n    });\n\n    it('rejects unused parameters in testing', async () => {\n      const rows = queryAsync('SELECT 33;', { unsed_parameter: true });\n      await expect(rows).rejects.toThrow('Unused parameter');\n    });\n  });\n\n  describe('queryRows', () => {\n    it('handles single column', async () => {\n      const rows = await queryRows('SELECT id FROM workspaces WHERE id <= 10;', z.string());\n      assert.lengthOf(rows, 10);\n      assert.equal(rows[0], '1');\n    });\n\n    it('handles multiple columns', async () => {\n      const rows = await queryRows('SELECT * FROM workspaces WHERE id <= 10;', WorkspaceSchema);\n      assert.lengthOf(rows, 10);\n      assert.equal(rows[0].id, '1');\n      assert.isNotNull(rows[0].created_at);\n    });\n\n    it('handles parameters', async () => {\n      const rows = await queryRows(\n        'SELECT * FROM workspaces WHERE id <= $1;',\n        [10],\n        WorkspaceSchema,\n      );\n      assert.lengthOf(rows, 10);\n    });\n  });\n\n  describe('queryRow', () => {\n    it('handles single column', async () => {\n      const row = await queryRow('SELECT id FROM workspaces WHERE id = 1;', z.string());\n      assert.equal(row, '1');\n    });\n\n    it('handles multiple columns', async () => {\n      const row = await queryRow('SELECT * FROM workspaces WHERE id = 1;', WorkspaceSchema);\n      assert.equal(row.id, '1');\n      assert.isNotNull(row.created_at);\n    });\n\n    it('handles parameters', async () => {\n      const row = await queryRow('SELECT * FROM workspaces WHERE id = $1;', [1], WorkspaceSchema);\n      assert.equal(row.id, '1');\n    });\n\n    it('rejects results with zero rows', async () => {\n      const rows = queryRow('SELECT * FROM workspaces WHERE id = -1;', WorkspaceSchema);\n      await expect(rows).rejects.toThrow('Incorrect rowCount: 0');\n    });\n\n    it('rejects results with multiple rows', async () => {\n      const rows = queryRow('SELECT * FROM workspaces', WorkspaceSchema);\n      await expect(rows).rejects.toThrow('Incorrect rowCount: 100');\n    });\n  });\n\n  describe('queryOptionalRow', () => {\n    it('handles single column', async () => {\n      const row = await queryRow('SELECT id FROM workspaces WHERE id = 1;', z.string());\n      assert.equal(row, '1');\n    });\n\n    it('handles multiple columns', async () => {\n      const row = await queryOptionalRow('SELECT * FROM workspaces WHERE id = 1;', WorkspaceSchema);\n      assert.isNotNull(row);\n      assert.equal(row?.id, '1');\n      assert.isNotNull(row?.created_at);\n    });\n\n    it('handles parameters', async () => {\n      const row = await queryOptionalRow(\n        'SELECT * FROM workspaces WHERE id = $1;',\n        [1],\n        WorkspaceSchema,\n      );\n      assert.isNotNull(row);\n      assert.equal(row?.id, '1');\n    });\n\n    it('handles missing result', async () => {\n      const row = await queryOptionalRow(\n        'SELECT * FROM workspaces WHERE id = -1;',\n        WorkspaceSchema,\n      );\n      assert.isNull(row);\n    });\n\n    it('rejects with multiple rows', async () => {\n      const rows = queryOptionalRow('SELECT * FROM workspaces', WorkspaceSchema);\n      await expect(rows).rejects.toThrow('Incorrect rowCount: 100');\n    });\n  });\n\n  describe('callRows', () => {\n    it('handles single column', async () => {\n      const rows = await callRows('test_sproc_one_column_ten_rows', z.string());\n      assert.lengthOf(rows, 10);\n      assert.equal(rows[0], '1');\n    });\n\n    it('handles parameters', async () => {\n      const rows = await callRows('test_sproc_one_column', [10], z.string());\n      assert.lengthOf(rows, 10);\n      assert.equal(rows[0], '1');\n    });\n\n    it('handles multiple columns', async () => {\n      const rows = await callRows('test_sproc_two_columns', [20], SprocTwoColumnsSchema);\n      assert.lengthOf(rows, 20);\n      assert.equal(rows[0].id, '1');\n      assert.equal(rows[0].negative, -1);\n      assert.equal(rows[19].id, '20');\n      assert.equal(rows[19].negative, -20);\n    });\n  });\n\n  describe('callRow', () => {\n    it('handles single column', async () => {\n      const row = await callRow('test_sproc_one_column_one_row', z.string());\n      assert.equal(row, '1');\n    });\n\n    it('handles parameters', async () => {\n      const row = await callRow('test_sproc_one_column', [1], z.string());\n      assert.equal(row, '1');\n    });\n\n    it('handles multiple columns', async () => {\n      const row = await callRow('test_sproc_two_columns', [1], SprocTwoColumnsSchema);\n      assert.equal(row.id, '1');\n      assert.equal(row.negative, -1);\n    });\n\n    it('rejects results with zero rows', async () => {\n      const row = callRow('test_sproc_two_columns', [0], SprocTwoColumnsSchema);\n      await expect(row).rejects.toThrow('Incorrect rowCount: 0');\n    });\n\n    it('rejects results with multiple rows', async () => {\n      const rows = callRow('test_sproc_two_columns', [100], SprocTwoColumnsSchema);\n      await expect(rows).rejects.toThrow('Incorrect rowCount: 100');\n    });\n  });\n\n  describe('callOptionalRow', () => {\n    it('handles single column', async () => {\n      const row = await callOptionalRow('test_sproc_one_column_one_row', z.string());\n      assert.equal(row, '1');\n    });\n\n    it('handles parameters', async () => {\n      const row = await callOptionalRow('test_sproc_one_column', [1], z.string());\n      assert.equal(row, '1');\n    });\n\n    it('handles multiple columns', async () => {\n      const row = await callOptionalRow('test_sproc_two_columns', [1], SprocTwoColumnsSchema);\n      assert.isNotNull(row);\n      assert.equal(row?.id, '1');\n      assert.equal(row?.negative, -1);\n    });\n\n    it('handles results with zero rows', async () => {\n      const row = await callOptionalRow('test_sproc_two_columns', [0], SprocTwoColumnsSchema);\n      assert.isNull(row);\n    });\n\n    it('rejects results with multiple rows', async () => {\n      const rows = callOptionalRow('test_sproc_two_columns', [100], SprocTwoColumnsSchema);\n      await expect(rows).rejects.toThrow('Incorrect rowCount: 100');\n    });\n  });\n\n  describe('queryCursor', () => {\n    it('returns zero rows', async () => {\n      const cursor = await queryCursor(\n        'SELECT * FROM workspaces WHERE id = 10000;',\n        {},\n        z.unknown(),\n      );\n      const rowBatches = [];\n      for await (const rows of cursor.iterate(10)) {\n        rowBatches.push(rows);\n      }\n      assert.lengthOf(rowBatches, 0);\n    });\n\n    it('returns one row at a time', async () => {\n      const cursor = await queryCursor('SELECT * FROM workspaces WHERE id <= 2;', {}, z.unknown());\n      const rowBatches = [];\n      for await (const rows of cursor.iterate(1)) {\n        rowBatches.push(rows);\n      }\n      assert.lengthOf(rowBatches, 2);\n      assert.lengthOf(rowBatches[0], 1);\n      assert.lengthOf(rowBatches[1], 1);\n    });\n\n    it('returns all rows at once', async () => {\n      const cursor = queryCursor('SELECT * FROM workspaces WHERE id <= 10;', {}, z.unknown());\n      const rowBatches = [];\n      for await (const rows of (await cursor).iterate(10)) {\n        rowBatches.push(rows);\n      }\n      assert.lengthOf(rowBatches, 1);\n      assert.lengthOf(rowBatches[0], 10);\n    });\n\n    it('handles errors', async () => {\n      const cursor = await queryCursor('NOT VALID SQL', {}, z.unknown());\n\n      async function readAllRows() {\n        const allRows = [];\n        for await (const rows of cursor.iterate(10)) {\n          allRows.push(...rows);\n        }\n        return allRows;\n      }\n\n      const maybeError = await readAllRows().catch((err) => err);\n      assert.instanceOf(maybeError, Error);\n      assert.match(maybeError.message, /syntax error/);\n      assert.isDefined((maybeError as any).data);\n      assert.equal((maybeError as any).data.sql, 'NOT VALID SQL');\n      assert.deepEqual((maybeError as any).data.sqlParams, {});\n      assert.isDefined((maybeError as any).data.sqlError);\n      assert.equal((maybeError as any).data.sqlError.severity, 'ERROR');\n    });\n  });\n\n  describe('queryCursor', () => {\n    const WorkspaceSchema = z.object({\n      id: z.string(),\n    });\n\n    const BadWorkspaceSchema = z.object({\n      badProperty: z.string(),\n    });\n\n    describe('iterator', () => {\n      it('validates with provided schema', async () => {\n        const cursor = await queryCursor(\n          'SELECT * FROM workspaces WHERE id <= 10 ORDER BY id ASC;',\n          {},\n          WorkspaceSchema,\n        );\n        const allRows = [];\n        for await (const rows of cursor.iterate(10)) {\n          allRows.push(...rows);\n        }\n        assert.lengthOf(allRows, 10);\n        const workspace = allRows[0] as any;\n        assert.equal(workspace.id, '1');\n        assert.isUndefined(workspace.state);\n      });\n\n      it('throws error when validation fails', async () => {\n        const cursor = await queryCursor(\n          'SELECT * FROM workspaces WHERE id <= 10 ORDER BY id ASC;',\n          {},\n          BadWorkspaceSchema,\n        );\n\n        async function readAllRows() {\n          const allRows = [];\n          for await (const rows of cursor.iterate(10)) {\n            allRows.push(...rows);\n          }\n          return allRows;\n        }\n\n        const maybeError = await readAllRows().catch((err) => err);\n        assert.instanceOf(maybeError, ZodError);\n        assert.lengthOf(maybeError.errors, 10);\n      });\n    });\n\n    describe('stream', () => {\n      it('validates with provided schema', async () => {\n        const cursor = await queryCursor(\n          'SELECT * FROM workspaces WHERE id <= 10 ORDER BY id ASC;',\n          {},\n          WorkspaceSchema,\n        );\n        const stream = cursor.stream(1);\n        const allRows = [];\n        for await (const row of stream) {\n          allRows.push(row);\n        }\n\n        assert.lengthOf(allRows, 10);\n      });\n\n      it('emits an error when validation fails', async () => {\n        const cursor = await queryCursor(\n          'SELECT * FROM workspaces ORDER BY id ASC;',\n          {},\n          BadWorkspaceSchema,\n        );\n        const stream = cursor.stream(1);\n\n        async function readAllRows() {\n          const allRows = [];\n          for await (const row of stream) {\n            allRows.push(row);\n          }\n          return allRows;\n        }\n\n        const maybeError = await readAllRows().catch((err) => err);\n        assert.instanceOf(maybeError, ZodError);\n        assert.lengthOf(maybeError.errors, 1);\n      });\n\n      it('closes the cursor when the stream is closed', async () => {\n        const cursor = await queryCursor('SELECT * FROM workspaces;', {}, WorkspaceSchema);\n        const stream = cursor.stream(1);\n\n        const rows: any[] = [];\n        const ac = new AbortController();\n        const writable = new Writable({\n          objectMode: true,\n          write(chunk, _encoding, callback) {\n            rows.push(chunk);\n\n            // After receiving the first row, abort the stream. This lets us test\n            // that the underlying cursor is closed. If it is *not* closed, this\n            // `after` hook will fail with a timeout.\n            ac.abort();\n            callback();\n          },\n        });\n\n        await expect(pipeline(stream, writable, { signal: ac.signal })).rejects.toThrow();\n        assert.lengthOf(rows, 1);\n      });\n    });\n  });\n});\n"]}

package/dist/test-utils.js

@@ -31,7 +31,7 @@ async function createDatabase(options, { dropExistingDatabase = true, configureP
             max: 10,
             idleTimeoutMillis: 30000,
             errorOnUnusedParameters: true,
-            ...(options.poolConfig ?? {}),
+            ...options.poolConfig,
         }, (err) => {
             throw err;
         });

package/dist/test-utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"test-utils.js","sourceRoot":"","sources":["../src/test-utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AAEpB,OAAO,KAAK,WAAW,MAAM,mBAAmB,CAAC;AAEjD,MAAM,aAAa,GAAG,UAAU,CAAC;AACjC,MAAM,aAAa,GAAG,WAAW,CAAC;AAClC,MAAM,iBAAiB,GAAG,UAAU,CAAC;AAyBrC,KAAK,UAAU,cAAc,CAC3B,OAAiC,EACjC,EACE,oBAAoB,GAAG,IAAI,EAC3B,aAAa,GAAG,IAAI,EACpB,QAAQ,EACR,gBAAgB,EAChB,OAAO,MACkB,EAAE;IAE7B,MAAM,MAAM,GAAG,IAAI,EAAE,CAAC,MAAM,CAAC;QAC3B,GAAG,aAAa,CAAC,OAAO,CAAC;QACzB,QAAQ,EAAE,OAAO,CAAC,eAAe,IAAI,iBAAiB;KACvD,CAAC,CAAC;IACH,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;IAEvB,MAAM,eAAe,GAAG,MAAM,CAAC,gBAAgB,CAC7C,QAAQ,IAAI,oCAAoC,CAAC,OAAO,CAAC,QAAQ,CAAC,CACnE,CAAC;IACF,IAAI,oBAAoB,IAAI,IAAI,EAAE,CAAC;QACjC,MAAM,MAAM,CAAC,KAAK,CAAC,2BAA2B,eAAe,EAAE,CAAC,CAAC;IACnE,CAAC;IAED,IAAI,gBAAgB,EAAE,CAAC;QACrB,MAAM,uBAAuB,GAAG,MAAM,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,CAAC;QAC1E,MAAM,MAAM,CAAC,KAAK,CAAC,mBAAmB,eAAe,aAAa,uBAAuB,EAAE,CAAC,CAAC;IAC/F,CAAC;SAAM,CAAC;QACN,MAAM,MAAM,CAAC,KAAK,CAAC,mBAAmB,eAAe,EAAE,CAAC,CAAC;IAC3D,CAAC;IAED,MAAM,MAAM,CAAC,GAAG,EAAE,CAAC;IAEnB,MAAM,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC;IAExB,IAAI,aAAa,EAAE,CAAC;QAClB,MAAM,WAAW,CAAC,SAAS,CACzB;YACE,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,aAAa;YACnC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,aAAa;YACnC,QAAQ,EAAE,oCAAoC,CAAC,OAAO,CAAC,QAAQ,CAAC;YAChE,+EAA+E;YAC/E,GAAG,EAAE,EAAE;YACP,iBAAiB,EAAE,KAAK;YACxB,uBAAuB,EAAE,IAAI;YAC7B,GAAG,CAAC,OAAO,CAAC,UAAU,IAAI,EAAE,CAAC;SAC9B,EACD,CAAC,GAAG,EAAE,EAAE;YACN,MAAM,GAAG,CAAC;QACZ,CAAC,CACF,CAAC;IACJ,CAAC;AACH,CAAC;AAED,KAAK,UAAU,aAAa,CAAC,OAAiC;IAC5D,MAAM,MAAM,GAAG,IAAI,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC;IACrD,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;IACvB,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;;;;;;;;GAYlB,CAAC,CAAC;IACH,MAAM,OAAO,CAAC,iBAAiB,EAAE,CAAC,MAAM,CAAC,CAAC;IAC1C,MAAM,MAAM,CAAC,GAAG,EAAE,CAAC;AACrB,CAAC;AAED,KAAK,UAAU,YAAY,CACzB,OAAiC,EACjC,EAAE,SAAS,GAAG,IAAI,EAAE,KAAK,GAAG,KAAK,EAAE,QAAQ,KAA0B,EAAE;IAEvE,IAAI,SAAS,EAAE,CAAC;QACd,MAAM,WAAW,CAAC,UAAU,EAAE,CAAC;IACjC,CAAC;IAED,MAAM,YAAY,GAAG,QAAQ,IAAI,oCAAoC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACxF,IAAI,iBAAiB,IAAI,OAAO,CAAC,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;QAC/C,sCAAsC;QACtC,OAAO,CAAC,GAAG,CAAC,mEAAmE,YAAY,EAAE,CAAC,CAAC;QAC/F,OAAO;IACT,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,EAAE,CAAC,MAAM,CAAC;QAC3B,GAAG,aAAa,CAAC,OAAO,CAAC;QACzB,QAAQ,EAAE,OAAO,CAAC,eAAe,IAAI,iBAAiB;KACvD,CAAC,CAAC;IACH,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;IACvB,MAAM,MAAM,CAAC,KAAK,CAAC,2BAA2B,MAAM,CAAC,gBAAgB,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;IACvF,MAAM,MAAM,CAAC,GAAG,EAAE,CAAC;AACrB,CAAC;AAED,SAAS,oCAAoC,CAAC,SAAiB;IAC7D,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,GAAG,CAAC;IAClF,OAAO,GAAG,SAAS,IAAI,QAAQ,EAAE,CAAC;AACpC,CAAC;AAED,SAAS,aAAa,CAAC,OAAiC;IACtD,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,aAAa;QACnC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,aAAa;QACnC,QAAQ,EAAE,oCAAoC,CAAC,OAAO,CAAC,QAAQ,CAAC;KACjE,CAAC;AACJ,CAAC;AAUD,MAAM,UAAU,qBAAqB,CAAC,OAAiC;IACrE,OAAO;QACL,cAAc,EAAE,CAAC,aAAqC,EAAE,EAAE,CACxD,cAAc,CAAC,OAAO,EAAE,aAAa,CAAC;QACxC,aAAa,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,OAAO,CAAC;QAC3C,YAAY,EAAE,CAAC,WAAiC,EAAE,EAAE,CAAC,YAAY,CAAC,OAAO,EAAE,WAAW,CAAC;QACvF,oCAAoC,EAAE,GAAG,EAAE,CACzC,oCAAoC,CAAC,OAAO,CAAC,QAAQ,CAAC;QACxD,aAAa,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,OAAO,CAAC;KAC5C,CAAC;AACJ,CAAC","sourcesContent":["import pg from 'pg';\n\nimport * as defaultPool from './default-pool.js';\n\nconst POSTGRES_USER = 'postgres';\nconst POSTGRES_HOST = 'localhost';\nconst POSTGRES_DATABASE = 'postgres';\n\nexport interface PostgresTestUtilsOptions {\n  database: string;\n  user?: string;\n  host?: string;\n  poolConfig?: Pick<pg.PoolConfig, 'max' | 'idleTimeoutMillis'>;\n  defaultDatabase?: string;\n  prepareAfterReset?: (client: pg.Client) => Promise<void>;\n}\n\ninterface CreateDatabaseOptions {\n  dropExistingDatabase?: boolean;\n  database?: string;\n  templateDatabase?: string;\n  configurePool?: boolean;\n  prepare?: (client: pg.Client) => Promise<void>;\n}\n\ninterface DropDatabaseOptions {\n  database?: string;\n  force?: boolean;\n  closePool?: boolean;\n}\n\nasync function createDatabase(\n  options: PostgresTestUtilsOptions,\n  {\n    dropExistingDatabase = true,\n    configurePool = true,\n    database,\n    templateDatabase,\n    prepare,\n  }: CreateDatabaseOptions = {},\n): Promise<void> {\n  const client = new pg.Client({\n    ...getPoolConfig(options),\n    database: options.defaultDatabase ?? POSTGRES_DATABASE,\n  });\n  await client.connect();\n\n  const escapedDatabase = client.escapeIdentifier(\n    database ?? getDatabaseNameForCurrentMochaWorker(options.database),\n  );\n  if (dropExistingDatabase ?? true) {\n    await client.query(`DROP DATABASE IF EXISTS ${escapedDatabase}`);\n  }\n\n  if (templateDatabase) {\n    const escapedTemplateDatabase = client.escapeIdentifier(templateDatabase);\n    await client.query(`CREATE DATABASE ${escapedDatabase} TEMPLATE ${escapedTemplateDatabase}`);\n  } else {\n    await client.query(`CREATE DATABASE ${escapedDatabase}`);\n  }\n\n  await client.end();\n\n  await prepare?.(client);\n\n  if (configurePool) {\n    await defaultPool.initAsync(\n      {\n        user: options.user ?? POSTGRES_USER,\n        host: options.host ?? POSTGRES_HOST,\n        database: getDatabaseNameForCurrentMochaWorker(options.database),\n        // Offer sensible default, but these can be overridden by `options.poolConfig`.\n        max: 10,\n        idleTimeoutMillis: 30000,\n        errorOnUnusedParameters: true,\n        ...(options.poolConfig ?? {}),\n      },\n      (err) => {\n        throw err;\n      },\n    );\n  }\n}\n\nasync function resetDatabase(options: PostgresTestUtilsOptions): Promise<void> {\n  const client = new pg.Client(getPoolConfig(options));\n  await client.connect();\n  await client.query(`\n    DO\n    $func$\n    BEGIN\n      EXECUTE (\n        SELECT 'TRUNCATE TABLE ' || string_agg(oid::regclass::text, ', ') || ' RESTART IDENTITY CASCADE'\n          FROM pg_class\n          WHERE relkind = 'r'\n          AND relnamespace = 'public'::regnamespace\n      );\n    END\n    $func$;\n  `);\n  await options.prepareAfterReset?.(client);\n  await client.end();\n}\n\nasync function dropDatabase(\n  options: PostgresTestUtilsOptions,\n  { closePool = true, force = false, database }: DropDatabaseOptions = {},\n): Promise<void> {\n  if (closePool) {\n    await defaultPool.closeAsync();\n  }\n\n  const databaseName = database ?? getDatabaseNameForCurrentMochaWorker(options.database);\n  if ('PL_KEEP_TEST_DB' in process.env && !force) {\n    // eslint-disable-next-line no-console\n    console.log(`PL_KEEP_TEST_DB environment variable set, not dropping database ${databaseName}`);\n    return;\n  }\n\n  const client = new pg.Client({\n    ...getPoolConfig(options),\n    database: options.defaultDatabase ?? POSTGRES_DATABASE,\n  });\n  await client.connect();\n  await client.query(`DROP DATABASE IF EXISTS ${client.escapeIdentifier(databaseName)}`);\n  await client.end();\n}\n\nfunction getDatabaseNameForCurrentMochaWorker(namespace: string): string {\n  const workerId = process.env.MOCHA_WORKER_ID ?? process.env.VITEST_POOL_ID ?? '1';\n  return `${namespace}_${workerId}`;\n}\n\nfunction getPoolConfig(options: PostgresTestUtilsOptions): pg.PoolConfig {\n  return {\n    user: options.user ?? POSTGRES_USER,\n    host: options.host ?? POSTGRES_HOST,\n    database: getDatabaseNameForCurrentMochaWorker(options.database),\n  };\n}\n\nexport interface PostgresTestUtils {\n  createDatabase: (options?: CreateDatabaseOptions) => Promise<void>;\n  resetDatabase: () => Promise<void>;\n  dropDatabase: (options?: DropDatabaseOptions) => Promise<void>;\n  getDatabaseNameForCurrentMochaWorker: () => string;\n  getPoolConfig: () => pg.PoolConfig;\n}\n\nexport function makePostgresTestUtils(options: PostgresTestUtilsOptions): PostgresTestUtils {\n  return {\n    createDatabase: (createOptions?: CreateDatabaseOptions) =>\n      createDatabase(options, createOptions),\n    resetDatabase: () => resetDatabase(options),\n    dropDatabase: (dropOptions?: DropDatabaseOptions) => dropDatabase(options, dropOptions),\n    getDatabaseNameForCurrentMochaWorker: () =>\n      getDatabaseNameForCurrentMochaWorker(options.database),\n    getPoolConfig: () => getPoolConfig(options),\n  };\n}\n"]}
+{"version":3,"file":"test-utils.js","sourceRoot":"","sources":["../src/test-utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AAEpB,OAAO,KAAK,WAAW,MAAM,mBAAmB,CAAC;AAEjD,MAAM,aAAa,GAAG,UAAU,CAAC;AACjC,MAAM,aAAa,GAAG,WAAW,CAAC;AAClC,MAAM,iBAAiB,GAAG,UAAU,CAAC;AAyBrC,KAAK,UAAU,cAAc,CAC3B,OAAiC,EACjC,EACE,oBAAoB,GAAG,IAAI,EAC3B,aAAa,GAAG,IAAI,EACpB,QAAQ,EACR,gBAAgB,EAChB,OAAO,MACkB,EAAE;IAE7B,MAAM,MAAM,GAAG,IAAI,EAAE,CAAC,MAAM,CAAC;QAC3B,GAAG,aAAa,CAAC,OAAO,CAAC;QACzB,QAAQ,EAAE,OAAO,CAAC,eAAe,IAAI,iBAAiB;KACvD,CAAC,CAAC;IACH,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;IAEvB,MAAM,eAAe,GAAG,MAAM,CAAC,gBAAgB,CAC7C,QAAQ,IAAI,oCAAoC,CAAC,OAAO,CAAC,QAAQ,CAAC,CACnE,CAAC;IACF,IAAI,oBAAoB,IAAI,IAAI,EAAE,CAAC;QACjC,MAAM,MAAM,CAAC,KAAK,CAAC,2BAA2B,eAAe,EAAE,CAAC,CAAC;IACnE,CAAC;IAED,IAAI,gBAAgB,EAAE,CAAC;QACrB,MAAM,uBAAuB,GAAG,MAAM,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,CAAC;QAC1E,MAAM,MAAM,CAAC,KAAK,CAAC,mBAAmB,eAAe,aAAa,uBAAuB,EAAE,CAAC,CAAC;IAC/F,CAAC;SAAM,CAAC;QACN,MAAM,MAAM,CAAC,KAAK,CAAC,mBAAmB,eAAe,EAAE,CAAC,CAAC;IAC3D,CAAC;IAED,MAAM,MAAM,CAAC,GAAG,EAAE,CAAC;IAEnB,MAAM,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC;IAExB,IAAI,aAAa,EAAE,CAAC;QAClB,MAAM,WAAW,CAAC,SAAS,CACzB;YACE,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,aAAa;YACnC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,aAAa;YACnC,QAAQ,EAAE,oCAAoC,CAAC,OAAO,CAAC,QAAQ,CAAC;YAChE,+EAA+E;YAC/E,GAAG,EAAE,EAAE;YACP,iBAAiB,EAAE,KAAK;YACxB,uBAAuB,EAAE,IAAI;YAC7B,GAAG,OAAO,CAAC,UAAU;SACtB,EACD,CAAC,GAAG,EAAE,EAAE;YACN,MAAM,GAAG,CAAC;QACZ,CAAC,CACF,CAAC;IACJ,CAAC;AACH,CAAC;AAED,KAAK,UAAU,aAAa,CAAC,OAAiC;IAC5D,MAAM,MAAM,GAAG,IAAI,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC;IACrD,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;IACvB,MAAM,MAAM,CAAC,KAAK,CAAC;;;;;;;;;;;;GAYlB,CAAC,CAAC;IACH,MAAM,OAAO,CAAC,iBAAiB,EAAE,CAAC,MAAM,CAAC,CAAC;IAC1C,MAAM,MAAM,CAAC,GAAG,EAAE,CAAC;AACrB,CAAC;AAED,KAAK,UAAU,YAAY,CACzB,OAAiC,EACjC,EAAE,SAAS,GAAG,IAAI,EAAE,KAAK,GAAG,KAAK,EAAE,QAAQ,KAA0B,EAAE;IAEvE,IAAI,SAAS,EAAE,CAAC;QACd,MAAM,WAAW,CAAC,UAAU,EAAE,CAAC;IACjC,CAAC;IAED,MAAM,YAAY,GAAG,QAAQ,IAAI,oCAAoC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACxF,IAAI,iBAAiB,IAAI,OAAO,CAAC,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;QAC/C,sCAAsC;QACtC,OAAO,CAAC,GAAG,CAAC,mEAAmE,YAAY,EAAE,CAAC,CAAC;QAC/F,OAAO;IACT,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,EAAE,CAAC,MAAM,CAAC;QAC3B,GAAG,aAAa,CAAC,OAAO,CAAC;QACzB,QAAQ,EAAE,OAAO,CAAC,eAAe,IAAI,iBAAiB;KACvD,CAAC,CAAC;IACH,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;IACvB,MAAM,MAAM,CAAC,KAAK,CAAC,2BAA2B,MAAM,CAAC,gBAAgB,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;IACvF,MAAM,MAAM,CAAC,GAAG,EAAE,CAAC;AACrB,CAAC;AAED,SAAS,oCAAoC,CAAC,SAAiB;IAC7D,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,GAAG,CAAC;IAClF,OAAO,GAAG,SAAS,IAAI,QAAQ,EAAE,CAAC;AACpC,CAAC;AAED,SAAS,aAAa,CAAC,OAAiC;IACtD,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,aAAa;QACnC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,aAAa;QACnC,QAAQ,EAAE,oCAAoC,CAAC,OAAO,CAAC,QAAQ,CAAC;KACjE,CAAC;AACJ,CAAC;AAUD,MAAM,UAAU,qBAAqB,CAAC,OAAiC;IACrE,OAAO;QACL,cAAc,EAAE,CAAC,aAAqC,EAAE,EAAE,CACxD,cAAc,CAAC,OAAO,EAAE,aAAa,CAAC;QACxC,aAAa,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,OAAO,CAAC;QAC3C,YAAY,EAAE,CAAC,WAAiC,EAAE,EAAE,CAAC,YAAY,CAAC,OAAO,EAAE,WAAW,CAAC;QACvF,oCAAoC,EAAE,GAAG,EAAE,CACzC,oCAAoC,CAAC,OAAO,CAAC,QAAQ,CAAC;QACxD,aAAa,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,OAAO,CAAC;KAC5C,CAAC;AACJ,CAAC","sourcesContent":["import pg from 'pg';\n\nimport * as defaultPool from './default-pool.js';\n\nconst POSTGRES_USER = 'postgres';\nconst POSTGRES_HOST = 'localhost';\nconst POSTGRES_DATABASE = 'postgres';\n\nexport interface PostgresTestUtilsOptions {\n  database: string;\n  user?: string;\n  host?: string;\n  poolConfig?: Pick<pg.PoolConfig, 'max' | 'idleTimeoutMillis'>;\n  defaultDatabase?: string;\n  prepareAfterReset?: (client: pg.Client) => Promise<void>;\n}\n\ninterface CreateDatabaseOptions {\n  dropExistingDatabase?: boolean;\n  database?: string;\n  templateDatabase?: string;\n  configurePool?: boolean;\n  prepare?: (client: pg.Client) => Promise<void>;\n}\n\ninterface DropDatabaseOptions {\n  database?: string;\n  force?: boolean;\n  closePool?: boolean;\n}\n\nasync function createDatabase(\n  options: PostgresTestUtilsOptions,\n  {\n    dropExistingDatabase = true,\n    configurePool = true,\n    database,\n    templateDatabase,\n    prepare,\n  }: CreateDatabaseOptions = {},\n): Promise<void> {\n  const client = new pg.Client({\n    ...getPoolConfig(options),\n    database: options.defaultDatabase ?? POSTGRES_DATABASE,\n  });\n  await client.connect();\n\n  const escapedDatabase = client.escapeIdentifier(\n    database ?? getDatabaseNameForCurrentMochaWorker(options.database),\n  );\n  if (dropExistingDatabase ?? true) {\n    await client.query(`DROP DATABASE IF EXISTS ${escapedDatabase}`);\n  }\n\n  if (templateDatabase) {\n    const escapedTemplateDatabase = client.escapeIdentifier(templateDatabase);\n    await client.query(`CREATE DATABASE ${escapedDatabase} TEMPLATE ${escapedTemplateDatabase}`);\n  } else {\n    await client.query(`CREATE DATABASE ${escapedDatabase}`);\n  }\n\n  await client.end();\n\n  await prepare?.(client);\n\n  if (configurePool) {\n    await defaultPool.initAsync(\n      {\n        user: options.user ?? POSTGRES_USER,\n        host: options.host ?? POSTGRES_HOST,\n        database: getDatabaseNameForCurrentMochaWorker(options.database),\n        // Offer sensible default, but these can be overridden by `options.poolConfig`.\n        max: 10,\n        idleTimeoutMillis: 30000,\n        errorOnUnusedParameters: true,\n        ...options.poolConfig,\n      },\n      (err) => {\n        throw err;\n      },\n    );\n  }\n}\n\nasync function resetDatabase(options: PostgresTestUtilsOptions): Promise<void> {\n  const client = new pg.Client(getPoolConfig(options));\n  await client.connect();\n  await client.query(`\n    DO\n    $func$\n    BEGIN\n      EXECUTE (\n        SELECT 'TRUNCATE TABLE ' || string_agg(oid::regclass::text, ', ') || ' RESTART IDENTITY CASCADE'\n          FROM pg_class\n          WHERE relkind = 'r'\n          AND relnamespace = 'public'::regnamespace\n      );\n    END\n    $func$;\n  `);\n  await options.prepareAfterReset?.(client);\n  await client.end();\n}\n\nasync function dropDatabase(\n  options: PostgresTestUtilsOptions,\n  { closePool = true, force = false, database }: DropDatabaseOptions = {},\n): Promise<void> {\n  if (closePool) {\n    await defaultPool.closeAsync();\n  }\n\n  const databaseName = database ?? getDatabaseNameForCurrentMochaWorker(options.database);\n  if ('PL_KEEP_TEST_DB' in process.env && !force) {\n    // eslint-disable-next-line no-console\n    console.log(`PL_KEEP_TEST_DB environment variable set, not dropping database ${databaseName}`);\n    return;\n  }\n\n  const client = new pg.Client({\n    ...getPoolConfig(options),\n    database: options.defaultDatabase ?? POSTGRES_DATABASE,\n  });\n  await client.connect();\n  await client.query(`DROP DATABASE IF EXISTS ${client.escapeIdentifier(databaseName)}`);\n  await client.end();\n}\n\nfunction getDatabaseNameForCurrentMochaWorker(namespace: string): string {\n  const workerId = process.env.MOCHA_WORKER_ID ?? process.env.VITEST_POOL_ID ?? '1';\n  return `${namespace}_${workerId}`;\n}\n\nfunction getPoolConfig(options: PostgresTestUtilsOptions): pg.PoolConfig {\n  return {\n    user: options.user ?? POSTGRES_USER,\n    host: options.host ?? POSTGRES_HOST,\n    database: getDatabaseNameForCurrentMochaWorker(options.database),\n  };\n}\n\nexport interface PostgresTestUtils {\n  createDatabase: (options?: CreateDatabaseOptions) => Promise<void>;\n  resetDatabase: () => Promise<void>;\n  dropDatabase: (options?: DropDatabaseOptions) => Promise<void>;\n  getDatabaseNameForCurrentMochaWorker: () => string;\n  getPoolConfig: () => pg.PoolConfig;\n}\n\nexport function makePostgresTestUtils(options: PostgresTestUtilsOptions): PostgresTestUtils {\n  return {\n    createDatabase: (createOptions?: CreateDatabaseOptions) =>\n      createDatabase(options, createOptions),\n    resetDatabase: () => resetDatabase(options),\n    dropDatabase: (dropOptions?: DropDatabaseOptions) => dropDatabase(options, dropOptions),\n    getDatabaseNameForCurrentMochaWorker: () =>\n      getDatabaseNameForCurrentMochaWorker(options.database),\n    getPoolConfig: () => getPoolConfig(options),\n  };\n}\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@prairielearn/postgres",
-  "version": "4.0.1",
+  "version": "4.1.1",
   "type": "module",
   "repository": {
     "type": "git",
@@ -27,9 +27,9 @@
   "devDependencies": {
     "@prairielearn/tsconfig": "^0.0.0",
     "@types/multipipe": "^3.0.5",
-    "@types/node": "^22.17.0",
+    "@types/node": "^22.18.0",
     "@vitest/coverage-v8": "^3.2.4",
-    "tsx": "^4.20.3",
+    "tsx": "^4.20.5",
     "typescript": "^5.9.2",
     "vitest": "^3.2.4"
   }

package/src/default-pool.ts

@@ -72,6 +72,8 @@ export const runInTransactionAsync = defaultPool.runInTransactionAsync.bind(defa
 /**
  * Executes a query with the specified parameters.
  *
+ * @deprecated Use {@link execute} instead.
+ *
  * Using the return value of this function directly is not recommended. Instead, use
  * {@link queryRows}, {@link queryRow}, or {@link queryOptionalRow}.
  */
@@ -80,7 +82,7 @@ export const queryAsync = defaultPool.queryAsync.bind(defaultPool);
  * Executes a query with the specified parameters. Errors if the query does
  * not return exactly one row.
  *
- * @deprecated Use {@link queryRow} instead.
+ * @deprecated Use {@link executeRow} or {@link queryRow} instead.
  */
 export const queryOneRowAsync = defaultPool.queryOneRowAsync.bind(defaultPool);
 /**
@@ -160,6 +162,17 @@ export const callRow = defaultPool.callRow.bind(defaultPool);
  * or a single row that conforms to the given Zod schema.
  */
 export const callOptionalRow = defaultPool.callOptionalRow.bind(defaultPool);
+
+/**
+ * Executes a query with the specified parameters. Returns the number of rows affected.
+ */
+export const execute = defaultPool.execute.bind(defaultPool);
+
+/**
+ * Executes a query with the specified parameter, and errors if the query doesn't return exactly one row.
+ */
+export const executeRow = defaultPool.executeRow.bind(defaultPool);
+
 /**
  * Returns an {@link CursorIterator} that can be used to iterate over the
  * results of the query in batches, which is useful for large result sets.

package/src/error.ts

@@ -6,7 +6,7 @@ export function formatQueryWithErrorPosition(query: string, position: number | n
   if (nextNewline < 0) nextNewline = query.length;
   const gap = ' '.repeat(Math.max(0, position - prevNewline - 1));
   return (
-    query.substring(0, nextNewline) +
+    query.slice(0, nextNewline) +
     '\n' +
     gap +
     '^\n' +
@@ -14,6 +14,6 @@ export function formatQueryWithErrorPosition(query: string, position: number | n
     '|\n' +
     gap +
     '+ ERROR POSITION SHOWN ABOVE\n' +
-    query.substring(nextNewline)
+    query.slice(nextNewline)
   );
 }

package/src/pool.test.ts

@@ -8,6 +8,7 @@ import {
   callOptionalRow,
   callRow,
   callRows,
+  execute,
   queryAsync,
   queryCursor,
   queryOptionalRow,
@@ -33,26 +34,21 @@ const SprocTwoColumnsSchema = z.object({
 describe('@prairielearn/postgres', function () {
   beforeAll(async () => {
     await postgresTestUtils.createDatabase();
-    await queryAsync(
+    await execute(
       'CREATE TABLE workspaces (id BIGSERIAL PRIMARY KEY, created_at TIMESTAMPTZ DEFAULT CURRENT_TIMESTAMP);',
-      {},
     );
-    await queryAsync('INSERT INTO workspaces (id) SELECT s FROM generate_series(1, 100) AS s', {});
-    await queryAsync(
+    await execute('INSERT INTO workspaces (id) SELECT s FROM generate_series(1, 100) AS s');
+    await execute(
       'CREATE FUNCTION test_sproc_one_column(num_entries INT) RETURNS TABLE (id BIGINT) AS $$ BEGIN RETURN QUERY SELECT s::BIGINT AS id FROM generate_series(1, num_entries) AS s; END; $$ LANGUAGE plpgsql;',
-      {},
     );
-    await queryAsync(
+    await execute(
       'CREATE FUNCTION test_sproc_two_columns(num_entries INT) RETURNS TABLE (id BIGINT, negative INT) AS $$ BEGIN RETURN QUERY SELECT s::BIGINT AS id, -s AS negative FROM generate_series(1, num_entries) AS s; END; $$ LANGUAGE plpgsql;',
-      {},
     );
-    await queryAsync(
+    await execute(
       'CREATE FUNCTION test_sproc_one_column_ten_rows() RETURNS TABLE (id BIGINT) AS $$ BEGIN RETURN QUERY SELECT s::BIGINT AS id FROM generate_series(1, 10) AS s; END; $$ LANGUAGE plpgsql;',
-      {},
     );
-    await queryAsync(
+    await execute(
       'CREATE FUNCTION test_sproc_one_column_one_row(OUT id BIGINT) AS $$ BEGIN id = 1; END; $$ LANGUAGE plpgsql;',
-      {},
     );
   });
 

package/src/pool.ts

@@ -4,9 +4,8 @@ import { Readable, Transform } from 'node:stream';
 import debugfn from 'debug';
 import _ from 'lodash';
 import multipipe from 'multipipe';
-import pg, { type QueryResult } from 'pg';
+import pg, { DatabaseError, type QueryResult } from 'pg';
 import Cursor from 'pg-cursor';
-import { DatabaseError } from 'pg-protocol';
 import { z } from 'zod';
 
 export type QueryParams = Record<string, any> | any[];
@@ -26,7 +25,7 @@ const searchSchemaMap = new WeakMap<pg.PoolClient, string>();
 
 function addDataToError(err: Error, data: Record<string, any>): Error {
   (err as any).data = {
-    ...((err as any).data ?? {}),
+    ...(err as any).data,
     ...data,
   };
   return err;
@@ -47,8 +46,8 @@ export class PostgresError extends Error {
  */
 function debugString(s: string): string {
   if (typeof s !== 'string') return 'NOT A STRING';
-  s = s.replace(/\n/g, '\\n');
-  if (s.length > 78) s = s.substring(0, 75) + '...';
+  s = s.replaceAll('\n', '\\n');
+  if (s.length > 78) s = s.slice(0, 75) + '...';
   s = '"' + s + '"';
   return s;
 }
@@ -107,14 +106,14 @@ function paramsToArray(
         paramsArray.push(params[v]);
       }
     }
-    processedSql += remainingSql.substring(0, result.index) + map[v];
-    remainingSql = remainingSql.substring(result.index + result[0].length);
+    processedSql += remainingSql.slice(0, result.index) + map[v];
+    remainingSql = remainingSql.slice(result.index + result[0].length);
   }
   processedSql += remainingSql;
   remainingSql = '';
   if (errorOnUnusedParameters) {
     const difference = _.difference(Object.keys(params), Object.keys(map));
-    if (difference.length) {
+    if (difference.length > 0) {
       throw new Error(`Unused parameters in SQL query: ${JSON.stringify(difference)}`);
     }
   }
@@ -453,6 +452,11 @@ export class PostgresPool {
 
   /**
    * Executes a query with the specified parameters.
+   *
+   * @deprecated Use {@link execute} instead.
+   *
+   * Using the return value of this function directly is not recommended. Instead, use
+   * {@link queryRows}, {@link queryRow}, or {@link queryOptionalRow}.
    */
   async queryAsync(sql: string, params: QueryParams): Promise<QueryResult> {
     debug('query()', 'sql:', debugString(sql));
@@ -471,6 +475,8 @@ export class PostgresPool {
   /**
    * Executes a query with the specified parameters. Errors if the query does
    * not return exactly one row.
+   *
+   * @deprecated Use {@link executeRow} or {@link queryRow} instead.
    */
   async queryOneRowAsync(sql: string, params: QueryParams): Promise<pg.QueryResult> {
     debug('queryOneRow()', 'sql:', debugString(sql));
@@ -510,7 +516,7 @@ export class PostgresPool {
   async callAsync(functionName: string, params: any[]): Promise<pg.QueryResult> {
     debug('call()', 'function:', functionName);
     debug('call()', 'params:', debugParams(params));
-    const placeholders = params.map((_, v) => '$' + (v + 1)).join();
+    const placeholders = params.map((_, v) => '$' + (v + 1)).join(',');
     const sql = `SELECT * FROM ${escapeIdentifier(functionName)}(${placeholders});`;
     const result = await this.queryAsync(sql, params);
     debug('call() success', 'rowCount:', result.rowCount);
@@ -563,7 +569,7 @@ export class PostgresPool {
   ): Promise<pg.QueryResult> {
     debug('callWithClient()', 'function:', functionName);
     debug('callWithClient()', 'params:', debugParams(params));
-    const placeholders = params.map((_, v) => '$' + (v + 1)).join();
+    const placeholders = params.map((_, v) => '$' + (v + 1)).join(',');
     const sql = `SELECT * FROM ${escapeIdentifier(functionName)}(${placeholders})`;
     const result = await this.queryWithClientAsync(client, sql, params);
     debug('callWithClient() success', 'rowCount:', result.rowCount);
@@ -787,6 +793,27 @@ export class PostgresPool {
     }
   }
 
+  /**
+   * Executes a query with the specified parameters. Returns the number of rows affected.
+   */
+  async execute(sql: string, params: QueryParams = {}): Promise<number> {
+    const result = await this.queryAsync(sql, params);
+    return result.rowCount ?? 0;
+  }
+
+  /**
+   * Executes a query with the specified parameter, and errors if the query doesn't return exactly one row.
+   */
+  async executeRow(sql: string, params: QueryParams = {}) {
+    const rowCount = await this.execute(sql, params);
+    if (rowCount !== 1) {
+      throw new PostgresError('Incorrect rowCount: ' + rowCount, {
+        sql,
+        sqlParams: params,
+      });
+    }
+  }
+
   /**
    * Returns a {@link Cursor} for the given query. The cursor can be used to
    * read results in batches, which is useful for large result sets.
@@ -895,7 +922,7 @@ export class PostgresPool {
    *
    * @param schema The schema name to use (can be "null" to unset the search path)
    */
-  async setSearchSchema(schema: string) {
+  async setSearchSchema(schema: string | null) {
     if (schema == null) {
       this.searchSchema = schema;
       return;
@@ -923,7 +950,7 @@ export class PostgresPool {
    */
   async setRandomSearchSchemaAsync(prefix: string): Promise<string> {
     // truncated prefix (max 28 characters)
-    const truncPrefix = prefix.substring(0, 28);
+    const truncPrefix = prefix.slice(0, 28);
     // timestamp in format YYYY-MM-DDTHH:MM:SS.SSSZ (guaranteed to not exceed 27 characters in the spec)
     const timestamp = new Date().toISOString();
     // random 6-character suffix to avoid clashes (approx 2 billion possible values)

package/src/test-utils.ts

@@ -73,7 +73,7 @@ async function createDatabase(
         max: 10,
         idleTimeoutMillis: 30000,
         errorOnUnusedParameters: true,
-        ...(options.poolConfig ?? {}),
+        ...options.poolConfig,
       },
       (err) => {
         throw err;