@rocicorp/zero 1.12.0-canary.2 → 1.12.0-canary.4

package/out/zero-server/src/process-mutations.js.map

@@ -1 +1 @@
-{"version":3,"file":"process-mutations.js","names":["#dbProvider","#req","#params","#lc","#transactImpl","#shouldRetry","#checkAndIncrementLastMutationID","#getTransactionInput"],"sources":["../../../../zero-server/src/process-mutations.ts"],"sourcesContent":["import {\n  consoleLogSink,\n  type LogContext,\n  type LogLevel,\n  type LogSink,\n} from '@rocicorp/logger';\nimport {assert} from '../../shared/src/asserts.ts';\nimport {getErrorDetails, getErrorMessage} from '../../shared/src/error.ts';\nimport type {ReadonlyJSONValue} from '../../shared/src/json.ts';\nimport {promiseVoid} from '../../shared/src/resolved-promises.ts';\nimport type {MaybePromise} from '../../shared/src/types.ts';\nimport * as v from '../../shared/src/valita.ts';\nimport {MutationAlreadyProcessedError} from '../../zero-cache/src/services/mutagen/error.ts';\nimport type {ApplicationError} from '../../zero-protocol/src/application-error.ts';\nimport {\n  isApplicationError,\n  wrapWithApplicationError,\n} from '../../zero-protocol/src/application-error.ts';\nimport {ErrorKind} from '../../zero-protocol/src/error-kind.ts';\nimport {ErrorOrigin} from '../../zero-protocol/src/error-origin.ts';\nimport {ErrorReason} from '../../zero-protocol/src/error-reason.ts';\nimport type {PushFailedBody} from '../../zero-protocol/src/error.ts';\nimport {\n  mutateParamsSchema,\n  type MutateResponse,\n} from '../../zero-protocol/src/mutate-server.ts';\nimport type {MutationID} from '../../zero-protocol/src/mutation-id.ts';\nimport {\n  CLEANUP_RESULTS_MUTATION_NAME,\n  cleanupResultsArgSchema,\n  type CleanupResultsArg,\n  type CustomMutation,\n  type Mutation,\n} from '../../zero-protocol/src/mutation.ts';\nimport {\n  pushBodySchema,\n  type MutationResponse,\n  type PushBody,\n} from '../../zero-protocol/src/push.ts';\nimport type {RetryPredicate} from '../../zql/src/mutate/custom.ts';\nimport type {AnyMutatorRegistry} from '../../zql/src/mutate/mutator-registry.ts';\nimport {isMutator} from '../../zql/src/mutate/mutator.ts';\nimport type {CustomMutatorDefs, CustomMutatorImpl} from './custom.ts';\nimport {createLogContext} from './logging.ts';\nimport {separatorRe} from './push-processor.ts';\n\nexport interface TransactionProviderHooks {\n  updateClientMutationID: () => Promise<{lastMutationID: number | bigint}>;\n  writeMutationResult: (result: MutationResponse) => Promise<void>;\n  deleteMutationResults: (args: CleanupResultsArg) => Promise<void>;\n}\n\nexport interface TransactionProviderInput {\n  upstreamSchema: string;\n  clientGroupID: string;\n  clientID: string;\n  mutationID: number;\n  /**\n   * Receives each predicate the mutator registers with `tx.retryOn()`. A\n   * `Database` passes it on to the server transaction it creates.\n   */\n  addRetryPredicate?: ((shouldRetry: RetryPredicate) => void) | undefined;\n}\n\n/**\n * Defines the abstract interface for a database that PushProcessor can execute\n * transactions against.\n */\nexport interface Database<T> {\n  transaction: <R>(\n    callback: (\n      tx: T,\n      transactionHooks: TransactionProviderHooks,\n    ) => MaybePromise<R>,\n    transactionInput?: TransactionProviderInput,\n  ) => Promise<R>;\n}\n\nexport type ExtractTransactionType<D> = D extends Database<infer T> ? T : never;\nexport type Params = v.Infer<typeof mutateParamsSchema>;\n\nexport type TransactFn<D extends Database<ExtractTransactionType<D>>> = (\n  cb: TransactFnCallback<D>,\n) => Promise<MutationResponse>;\n\nexport type TransactFnCallback<D extends Database<ExtractTransactionType<D>>> =\n  (\n    tx: ExtractTransactionType<D>,\n    mutatorName: string,\n    mutatorArgs: ReadonlyJSONValue | undefined,\n  ) => Promise<ReadonlyJSONValue | void>;\n\nexport type Parsed<D extends Database<ExtractTransactionType<D>>> = {\n  transact: TransactFn<D>;\n  mutations: CustomMutation[];\n};\n\ntype MutationPhase = 'preTransaction' | 'transactionPending' | 'postCommit';\n\nconst applicationErrorWrapper = async <T>(fn: () => Promise<T>): Promise<T> => {\n  try {\n    return await fn();\n  } catch (error) {\n    if (\n      error instanceof DatabaseTransactionError ||\n      error instanceof OutOfOrderMutation ||\n      error instanceof MutationAlreadyProcessedError ||\n      isApplicationError(error)\n    ) {\n      throw error;\n    }\n\n    throw wrapWithApplicationError(error);\n  }\n};\n\n/**\n * @deprecated Use {@link handleMutateRequest} instead.\n */\nexport const handleMutationRequest = handleMutateRequest;\n\n/**\n * Parsed query params accepted by {@link handleMutateRequest} when the\n * incoming request URL has already been handled by your framework.\n */\nexport type MutateSearchParams = URLSearchParams | Record<string, string>;\n\n/**\n * Runs once per custom mutation in a `/mutate` request.\n *\n * Call `transact` to execute database work for the mutation. Throwing before\n * `transact` persists an application error; throwing after `transact` resolves\n * only affects server-side logging because the mutation is already committed.\n */\nexport type MutateRequestHandler<\n  D extends Database<ExtractTransactionType<D>>,\n> = (\n  transact: TransactFn<D>,\n  mutation: CustomMutation,\n) => Promise<MutationResponse>;\n\nexport type HandleMutateRequestArgs<\n  D extends Database<ExtractTransactionType<D>>,\n> = {\n  /** Database used to run transactions and store mutation results. */\n  dbProvider: D;\n  handler: MutateRequestHandler<D>;\n  /**\n   * Authenticated user ID. Null or undefined means the user is logged out.\n   */\n  userID: string | null | undefined;\n  /** Optional log level for request parsing and execution. */\n  logLevel?: LogLevel | undefined;\n  /**\n   * Destination for this request's log output. When omitted, logs are written\n   * with `console.log`/`info`/`warn`/`error` etc.\n   *\n   * Provide a custom {@link LogSink} to redirect logs — for example into an\n   * application's own structured logging pipeline, so that records emitted while\n   * processing a push carry the same fields as the rest of the server's logs.\n   */\n  logSink?: LogSink | undefined;\n} & (\n  | {\n      /** Fetch request containing both query params and the JSON body. */\n      request: Request;\n    }\n  | {\n      /** Parsed query params from the `/mutate` request URL. */\n      query: MutateSearchParams;\n      /** Parsed JSON body from the `/mutate` request. */\n      body: ReadonlyJSONValue;\n    }\n);\n\ntype NormalizedMutateRequestArgs<\n  D extends Database<ExtractTransactionType<D>>,\n> = {\n  readonly dbProvider: D;\n  readonly handler: MutateRequestHandler<D>;\n  // Note: semantics of undefined differ from HandleMutateRequestArgs.userID.\n  // Here, undefined means the app didn't provide a user ID - we do not know if\n  // the user is logged in or not. This is legacy behavior needed to support\n  // deprecated signatures of handleMutateRequest which did not receive userID\n  // from app.\n  readonly userID: string | null | undefined;\n  readonly logLevel: LogLevel;\n  readonly logSink: LogSink;\n} & (\n  | {\n      readonly type: 'request';\n      readonly request: Request;\n    }\n  | {\n      readonly type: 'body';\n      readonly queryParams: Record<string, string>;\n      readonly jsonBody: ReadonlyJSONValue;\n    }\n);\n\n/**\n * Process a `/mutate` request from a Fetch `Request`.\n */\nexport function handleMutateRequest<\n  D extends Database<ExtractTransactionType<D>>,\n>(input: HandleMutateRequestArgs<D>): Promise<MutateResponse>;\n\n/**\n * @deprecated Pass a single object instead:\n * `handleMutateRequest({dbProvider, handler, query, body, userID, logLevel})`.\n */\nexport function handleMutateRequest<\n  D extends Database<ExtractTransactionType<D>>,\n>(\n  dbProvider: D,\n  handler: MutateRequestHandler<D>,\n  query: MutateSearchParams,\n  body: ReadonlyJSONValue,\n  logLevel?: LogLevel,\n  logSink?: LogSink,\n): Promise<MutateResponse>;\n\n/**\n * @deprecated Pass a single object instead:\n * `handleMutateRequest({dbProvider, handler, request, userID, logLevel})`.\n */\nexport function handleMutateRequest<\n  D extends Database<ExtractTransactionType<D>>,\n>(\n  dbProvider: D,\n  handler: MutateRequestHandler<D>,\n  request: Request,\n  logLevel?: LogLevel,\n  logSink?: LogSink,\n): Promise<MutateResponse>;\n\nexport async function handleMutateRequest<\n  D extends Database<ExtractTransactionType<D>>,\n>(\n  inputOrDbProvider: HandleMutateRequestArgs<D> | D,\n  maybeHandler?: MutateRequestHandler<D> | undefined,\n  requestOrQuery?: Request | MutateSearchParams | undefined,\n  bodyOrLogLevel?: ReadonlyJSONValue | LogLevel | undefined,\n  logLevelOrLogSink?: LogLevel | LogSink | undefined,\n  maybeLogSink?: LogSink | undefined,\n): Promise<MutateResponse> {\n  const normalized =\n    typeof inputOrDbProvider === 'object' && 'handler' in inputOrDbProvider\n      ? normalizeMutateRequestInput(inputOrDbProvider)\n      : normalizeLegacyMutateRequestArgs(\n          inputOrDbProvider,\n          maybeHandler,\n          requestOrQuery,\n          bodyOrLogLevel,\n          logLevelOrLogSink,\n          maybeLogSink,\n        );\n\n  const lc = createLogContext(\n    normalized.logLevel,\n    normalized.logSink,\n  ).withContext('PushProcessor');\n  let jsonBody: unknown;\n\n  if (normalized.type === 'request') {\n    try {\n      jsonBody = await normalized.request.json();\n    } catch (error) {\n      lc.error?.('Failed to parse push body', error);\n      const message = `Failed to parse push body: ${getErrorMessage(error)}`;\n      const details = getErrorDetails(error);\n      return {\n        kind: ErrorKind.PushFailed,\n        origin: ErrorOrigin.Server,\n        reason: ErrorReason.Parse,\n        message,\n        mutationIDs: [],\n        ...(details ? {details} : {}),\n      } as const satisfies PushFailedBody;\n    }\n  } else {\n    jsonBody = normalized.jsonBody;\n  }\n\n  let mutationIDs: MutationID[] = [];\n\n  let pushBody: PushBody;\n  try {\n    pushBody = v.parse(jsonBody, pushBodySchema, 'passthrough');\n    mutationIDs = pushBody.mutations.map(m => ({\n      id: m.id,\n      clientID: m.clientID,\n    }));\n  } catch (error) {\n    lc.error?.('Failed to parse push body', error);\n    const message = `Failed to parse push body: ${getErrorMessage(error)}`;\n    const details = getErrorDetails(error);\n    return {\n      kind: ErrorKind.PushFailed,\n      origin: ErrorOrigin.Server,\n      reason: ErrorReason.Parse,\n      message,\n      mutationIDs,\n      ...(details ? {details} : {}),\n    } as const satisfies PushFailedBody;\n  }\n\n  let parsedQueryParams: Params;\n  try {\n    parsedQueryParams = v.parse(\n      normalized.type === 'request'\n        ? Object.fromEntries(new URL(normalized.request.url).searchParams)\n        : normalized.queryParams,\n      mutateParamsSchema,\n      'passthrough',\n    );\n  } catch (error) {\n    lc.error?.('Failed to parse push query parameters', error);\n    const message = `Failed to parse push query parameters: ${getErrorMessage(error)}`;\n    const details = getErrorDetails(error);\n    return {\n      kind: ErrorKind.PushFailed,\n      origin: ErrorOrigin.Server,\n      reason: ErrorReason.Parse,\n      message,\n      mutationIDs,\n      ...(details ? {details} : {}),\n    } as const satisfies PushFailedBody;\n  }\n\n  if (pushBody.pushVersion !== 1) {\n    const response = {\n      kind: ErrorKind.PushFailed,\n      origin: ErrorOrigin.Server,\n      reason: ErrorReason.UnsupportedPushVersion,\n      mutationIDs,\n      message: `Unsupported push version: ${pushBody.pushVersion}`,\n    } as const satisfies PushFailedBody;\n    return response;\n  }\n\n  const responses: MutationResponse[] = [];\n  let processedCount = 0;\n\n  try {\n    const transactor = new Transactor(\n      normalized.dbProvider,\n      pushBody,\n      parsedQueryParams,\n      lc,\n    );\n\n    // Each mutation goes through three phases:\n    //   1. Pre-transaction: user logic that runs before `transact` is called. If\n    //      this throws we still advance LMID and persist the failure result.\n    //   2. Transaction: the callback passed to `transact`, which can be retried\n    //      if it fails with an ApplicationError.\n    //   3. Post-commit: any logic that runs after `transact` resolves. Failures\n    //      here are logged but the mutation remains committed.\n    for (const m of pushBody.mutations) {\n      // Handle internal mutations (like cleanup) directly without user dispatch\n      if (m.type === 'custom' && m.name === CLEANUP_RESULTS_MUTATION_NAME) {\n        lc.debug?.(\n          `Processing internal mutation '${m.name}' (clientID=${m.clientID})`,\n        );\n        try {\n          await processCleanupResultsMutation(\n            normalized.dbProvider,\n            m,\n            parsedQueryParams,\n            lc,\n          );\n          // No response added - this is fire-and-forget\n          processedCount++;\n        } catch (error) {\n          lc.warn?.(\n            `Failed to process cleanup mutation for client ${m.clientID}`,\n            error,\n          );\n          // Don't fail the whole push for cleanup errors\n          processedCount++;\n        }\n        continue;\n      }\n\n      assert(m.type === 'custom', 'Expected custom mutation');\n      lc.debug?.(\n        `Processing mutation '${m.name}' (id=${m.id}, clientID=${m.clientID})`,\n      );\n\n      let mutationPhase: MutationPhase = 'preTransaction';\n\n      const transactProxy: TransactFn<D> = async innerCb => {\n        mutationPhase = 'transactionPending';\n        const result = await transactor.transact(m, innerCb);\n        mutationPhase = 'postCommit';\n        return result;\n      };\n\n      try {\n        const res = await applicationErrorWrapper(() =>\n          normalized.handler(transactProxy, m),\n        );\n        responses.push(res);\n        lc.debug?.(`Mutation '${m.name}' (id=${m.id}) completed successfully`);\n\n        processedCount++;\n      } catch (error) {\n        if (!isApplicationError(error)) {\n          throw error;\n        }\n\n        if (mutationPhase === 'preTransaction') {\n          // Pre-transaction\n          await transactor.persistPreTransactionFailure(m, error);\n        } else if (mutationPhase === 'postCommit') {\n          // Post-commit\n          lc.error?.(\n            `Post-commit mutation handler failed for mutation ${m.id} for client ${m.clientID}`,\n            error,\n          );\n        }\n\n        lc.warn?.(\n          `Application error processing mutation ${m.id} for client ${m.clientID}`,\n          error,\n        );\n        responses.push(makeAppErrorResponse(m, error));\n\n        processedCount++;\n      }\n    }\n\n    return {\n      kind: 'MutateResponse',\n      mutations: responses,\n      ...(typeof normalized.userID !== 'undefined'\n        ? {userID: normalized.userID}\n        : {}),\n    } as const satisfies MutateResponse;\n  } catch (error) {\n    lc.error?.('Failed to process push request', error);\n    // only include mutationIDs for mutations that were not processed\n    const unprocessedMutationIDs = mutationIDs.slice(processedCount);\n\n    const message = getErrorMessage(error);\n    const details = getErrorDetails(error);\n\n    return {\n      kind: ErrorKind.PushFailed,\n      origin: ErrorOrigin.Server,\n      reason:\n        error instanceof OutOfOrderMutation\n          ? ErrorReason.OutOfOrderMutation\n          : error instanceof DatabaseTransactionError\n            ? ErrorReason.Database\n            : ErrorReason.Internal,\n      message,\n      mutationIDs: unprocessedMutationIDs,\n      ...(details ? {details} : {}),\n    };\n  }\n}\n\nfunction normalizeMutateRequestInput<\n  D extends Database<ExtractTransactionType<D>>,\n>(input: HandleMutateRequestArgs<D>): NormalizedMutateRequestArgs<D> {\n  if ('request' in input) {\n    return {\n      type: 'request',\n      dbProvider: input.dbProvider,\n      handler: input.handler,\n      request: input.request,\n      userID: input.userID ?? null,\n      logLevel: input.logLevel ?? 'info',\n      logSink: input.logSink ?? consoleLogSink,\n    };\n  }\n\n  return {\n    type: 'body',\n    dbProvider: input.dbProvider,\n    handler: input.handler,\n    jsonBody: input.body,\n    userID: input.userID ?? null,\n    queryParams:\n      input.query instanceof URLSearchParams\n        ? Object.fromEntries(input.query)\n        : input.query,\n    logLevel: input.logLevel ?? 'info',\n    logSink: input.logSink ?? consoleLogSink,\n  };\n}\n\nfunction normalizeLegacyMutateRequestArgs<\n  D extends Database<ExtractTransactionType<D>>,\n>(\n  dbProvider: D,\n  handler: MutateRequestHandler<D> | undefined,\n  requestOrQuery: Request | MutateSearchParams | undefined,\n  bodyOrLogLevel: ReadonlyJSONValue | LogLevel | undefined,\n  logLevelOrLogSink: LogLevel | LogSink | undefined,\n  maybeLogSink: LogSink | undefined,\n): NormalizedMutateRequestArgs<D> {\n  assert(typeof handler === 'function', 'Handler function is required');\n  assert(\n    typeof requestOrQuery !== 'undefined',\n    'Request or query parameters are required',\n  );\n\n  if (requestOrQuery instanceof Request) {\n    return {\n      type: 'request',\n      dbProvider,\n      handler,\n      request: requestOrQuery,\n      userID: undefined,\n      logLevel: (bodyOrLogLevel as LogLevel | undefined) ?? 'info',\n      // The `request` overload has no `body` slot, so the sink arrives one\n      // position earlier than it does in the `query`/`body` overload.\n      logSink: (logLevelOrLogSink as LogSink | undefined) ?? consoleLogSink,\n    };\n  }\n\n  assert(\n    typeof bodyOrLogLevel !== 'undefined',\n    'JSON body cannot be undefined',\n  );\n\n  return {\n    type: 'body',\n    dbProvider,\n    handler,\n    jsonBody: bodyOrLogLevel,\n    userID: undefined,\n    queryParams:\n      requestOrQuery instanceof URLSearchParams\n        ? Object.fromEntries(requestOrQuery)\n        : requestOrQuery,\n    logLevel: (logLevelOrLogSink as LogLevel | undefined) ?? 'info',\n    logSink: maybeLogSink ?? consoleLogSink,\n  };\n}\n\n/**\n * The most times a mutator runs for one mutation within one push when its\n * `tx.retryOn()` predicate keeps answering yes: the first run and four\n * re-runs.\n */\nexport const MAX_MUTATOR_RUNS = 5;\n\nclass Transactor<D extends Database<ExtractTransactionType<D>>> {\n  readonly #dbProvider: D;\n  readonly #req: PushBody;\n  readonly #params: Params;\n  readonly #lc: LogContext;\n\n  constructor(dbProvider: D, req: PushBody, params: Params, lc: LogContext) {\n    this.#dbProvider = dbProvider;\n    this.#req = req;\n    this.#params = params;\n    this.#lc = lc;\n  }\n\n  /**\n   * Asks all of the run's `tx.retryOn()` predicates at once whether to run\n   * the mutator again. The first to answer yes decides it and the other\n   * answers are ignored; it is no once every predicate has answered no. A\n   * predicate that throws or rejects answers no: escaping here would skip\n   * recording the mutation's result, and the client would send it again.\n   */\n  #shouldRetry(\n    predicates: readonly RetryPredicate[],\n    error: unknown,\n    attempt: number,\n  ): Promise<boolean> {\n    if (predicates.length === 0) {\n      return Promise.resolve(false);\n    }\n    const ask = async (shouldRetry: RetryPredicate): Promise<boolean> => {\n      try {\n        return (await shouldRetry(error, attempt)) === true;\n      } catch (retryError) {\n        this.#lc.warn?.(\n          'retryOn predicate threw; treating it as no',\n          retryError,\n        );\n        return false;\n      }\n    };\n    return new Promise(resolve => {\n      let pending = predicates.length;\n      for (const shouldRetry of predicates) {\n        void ask(shouldRetry).then(yes => {\n          if (yes) {\n            resolve(true);\n          } else if (--pending === 0) {\n            resolve(false);\n          }\n        });\n      }\n    });\n  }\n\n  transact = async (\n    mutation: CustomMutation,\n    cb: TransactFnCallback<D>,\n  ): Promise<MutationResponse> => {\n    let appError: ApplicationError | undefined = undefined;\n    // Which run of the mutator this is: 1, then 2 for the first re-run, and so\n    // on. When a run fails, it is also the number of the re-run being decided.\n    let attempt = 1;\n    for (;;) {\n      // What this run registers with `tx.retryOn()`. Each run starts empty.\n      const predicates: RetryPredicate[] = [];\n      try {\n        const ret = await this.#transactImpl(\n          mutation,\n          cb,\n          appError,\n          shouldRetry => {\n            predicates.push(shouldRetry);\n          },\n        );\n        if (appError !== undefined) {\n          this.#lc.warn?.(\n            // log-leak-ignore -- mutation and client ids, not row data\n            `Mutation ${mutation.id} for client ${mutation.clientID} was retried after an error`,\n            appError,\n          );\n          return makeAppErrorResponse(mutation, appError);\n        }\n\n        return ret;\n      } catch (error) {\n        if (error instanceof OutOfOrderMutation) {\n          this.#lc.error?.(error);\n          throw error;\n        }\n\n        if (error instanceof MutationAlreadyProcessedError) {\n          this.#lc.warn?.(error);\n          return {\n            id: {\n              clientID: mutation.clientID,\n              id: mutation.id,\n            },\n            result: {\n              error: 'alreadyProcessed',\n              details: error.message,\n            },\n          };\n        }\n\n        if (appError !== undefined) {\n          // Retry also failed → internal error, cannot skip mutation\n          this.#lc.error?.(\n            // log-leak-ignore -- mutation and client ids, not row data\n            `Retry also failed for mutation ${mutation.id} for client ${mutation.clientID}`,\n            error,\n          );\n          throw error;\n        }\n\n        // What the mutator or the database threw, without the wrapper. This\n        // includes errors raised at COMMIT, which the mutator cannot catch.\n        const originalError =\n          error instanceof DatabaseTransactionError\n            ? (error.cause ?? error)\n            : error;\n\n        // One of the run's `tx.retryOn()` predicates says to run it again → do\n        // so, in a fresh transaction.\n        //\n        // `continue` WITHOUT setting `appError` is the whole mechanism: it is\n        // `appError` being set that makes the next `#transactImpl` skip\n        // `cb(...)`. `#transactImpl` opens a new transaction each pass, so the\n        // re-run gets a new snapshot. `#checkAndIncrementLastMutationID` re-runs\n        // safely because the failed run rolled its increment back.\n        if (\n          attempt < MAX_MUTATOR_RUNS &&\n          (await this.#shouldRetry(predicates, originalError, attempt))\n        ) {\n          this.#lc.warn?.(\n            // log-leak-ignore -- mutation and client ids, not row data\n            `Mutation ${mutation.id} for client ${mutation.clientID} failed, re-running mutator (retry ${attempt})`,\n            originalError,\n          );\n          attempt++;\n          continue;\n        }\n\n        // Not re-run → store error and retry without mutator\n        appError = wrapWithApplicationError(originalError);\n        this.#lc.warn?.(\n          // log-leak-ignore -- mutation and client ids, not row data\n          `Error processing mutation ${mutation.id} for client ${mutation.clientID}, retrying without mutator`,\n          appError,\n        );\n        continue;\n      }\n    }\n  };\n\n  async persistPreTransactionFailure(\n    mutation: CustomMutation,\n    appError: ApplicationError<ReadonlyJSONValue | undefined>,\n  ): Promise<MutationResponse> {\n    // User-land code threw before calling `transact`. We still need to bump the\n    // LMID for this mutation and persist the error so that the client knows it failed.\n    const ret = await this.#transactImpl(\n      mutation,\n      // noop callback since there's no transaction to execute\n      () => promiseVoid,\n      appError,\n      undefined,\n    );\n    return ret;\n  }\n\n  async #transactImpl(\n    mutation: CustomMutation,\n    cb: TransactFnCallback<D>,\n    appError: ApplicationError | undefined,\n    addRetryPredicate: ((shouldRetry: RetryPredicate) => void) | undefined,\n  ): Promise<MutationResponse> {\n    let transactionPhase: DatabaseTransactionPhase = 'open';\n\n    try {\n      const ret = await this.#dbProvider.transaction(\n        async (dbTx, transactionHooks) => {\n          // update the transaction phase to 'execute' after the transaction is opened\n          transactionPhase = 'execute';\n\n          await this.#checkAndIncrementLastMutationID(\n            transactionHooks,\n            mutation.clientID,\n            mutation.id,\n          );\n\n          if (appError === undefined) {\n            this.#lc.debug?.(\n              // log-leak-ignore -- mutator name and mutation id, not row data\n              `Executing mutator '${mutation.name}' (id=${mutation.id})`,\n            );\n            const data = (await cb(dbTx, mutation.name, mutation.args[0])) as\n              | ReadonlyJSONValue\n              | undefined;\n            const id = {clientID: mutation.clientID, id: mutation.id};\n            if (data === undefined) {\n              return {id, result: {}};\n            }\n            // Persisted so the data still reaches the client if the push\n            // response is lost. A mutator that returns nothing writes no row.\n            const successResult: MutationResponse = {id, result: {data}};\n            await transactionHooks.writeMutationResult(successResult);\n            return successResult;\n          } else {\n            const mutationResult = makeAppErrorResponse(mutation, appError);\n            await transactionHooks.writeMutationResult(mutationResult);\n            return mutationResult;\n          }\n        },\n        this.#getTransactionInput(mutation, addRetryPredicate),\n      );\n\n      return ret;\n    } catch (error) {\n      if (\n        isApplicationError(error) ||\n        error instanceof OutOfOrderMutation ||\n        error instanceof MutationAlreadyProcessedError\n      ) {\n        throw error;\n      }\n\n      throw new DatabaseTransactionError(transactionPhase, {cause: error});\n    }\n  }\n\n  #getTransactionInput(\n    mutation: CustomMutation,\n    addRetryPredicate: ((shouldRetry: RetryPredicate) => void) | undefined,\n  ): TransactionProviderInput {\n    return {\n      upstreamSchema: this.#params.schema,\n      clientGroupID: this.#req.clientGroupID,\n      clientID: mutation.clientID,\n      mutationID: mutation.id,\n      addRetryPredicate,\n    };\n  }\n\n  async #checkAndIncrementLastMutationID(\n    transactionHooks: TransactionProviderHooks,\n    clientID: string,\n    receivedMutationID: number,\n  ) {\n    const {lastMutationID} = await transactionHooks.updateClientMutationID();\n\n    if (receivedMutationID < lastMutationID) {\n      throw new MutationAlreadyProcessedError(\n        clientID,\n        receivedMutationID,\n        lastMutationID,\n      );\n    } else if (receivedMutationID > lastMutationID) {\n      throw new OutOfOrderMutation(\n        clientID,\n        receivedMutationID,\n        lastMutationID,\n      );\n    }\n  }\n}\n\nexport class OutOfOrderMutation extends Error {\n  constructor(\n    clientID: string,\n    receivedMutationID: number,\n    lastMutationID: number | bigint,\n  ) {\n    super(\n      `Client ${clientID} sent mutation ID ${receivedMutationID} but expected ${lastMutationID}`,\n    );\n  }\n}\n\nfunction makeAppErrorResponse(\n  m: Mutation,\n  error: ApplicationError<ReadonlyJSONValue | undefined>,\n): MutationResponse {\n  return {\n    id: {\n      clientID: m.clientID,\n      id: m.id,\n    },\n    result: {\n      error: 'app',\n      message: error.message,\n      ...(error.details ? {details: error.details} : {}),\n    },\n  };\n}\n\n/** @deprecated Use getMutator instead */\nexport function getMutation(\n  // oxlint-disable-next-line no-explicit-any\n  mutators: AnyMutatorRegistry | CustomMutatorDefs<any>,\n  name: string,\n  // oxlint-disable-next-line no-explicit-any\n): CustomMutatorImpl<any> {\n  const path = name.split(separatorRe);\n  const mutator = getObjectAtPath(mutators, path);\n  assert(typeof mutator === 'function', `could not find mutator ${name}`);\n\n  if (isMutator(mutator)) {\n    // mutator needs to be called with {tx, args, ctx}\n    // CustomMutatorImpl is called with (tx, args, ctx)\n    return (tx, args, ctx) => mutator.fn({args, ctx, tx});\n  }\n\n  // oxlint-disable-next-line no-explicit-any\n  return mutator as CustomMutatorImpl<any>;\n}\n\nfunction getObjectAtPath(\n  obj: Record<string, unknown>,\n  path: string[],\n): unknown {\n  let current: unknown = obj;\n  for (const part of path) {\n    if (typeof current !== 'object' || current === null || !(part in current)) {\n      return undefined;\n    }\n    current = (current as Record<string, unknown>)[part];\n  }\n  return current;\n}\n\n/**\n * Processes internal cleanup mutation that deletes acknowledged mutation results\n * from the upstream database. This runs without LMID tracking since it's an\n * internal operation.\n */\nasync function processCleanupResultsMutation<\n  D extends Database<ExtractTransactionType<D>>,\n>(\n  dbProvider: D,\n  mutation: CustomMutation,\n  queryParams: Params,\n  lc: LogContext,\n): Promise<void> {\n  const parseResult = v.test(mutation.args[0], cleanupResultsArgSchema);\n  if (!parseResult.ok) {\n    // log-leak-ignore -- valita reports the failing type, not the value\n    lc.warn?.('Cleanup mutation has invalid args', parseResult.error);\n    return;\n  }\n  const args: CleanupResultsArg = parseResult.value;\n\n  // Determine clientID for transaction input based on cleanup type\n  // Note: legacy format without type field is treated as single\n  const clientID =\n    'type' in args && args.type === 'bulk' ? args.clientIDs[0] : args.clientID;\n\n  // Run in a transaction, using the hook for DB-specific operation.\n  // Note: only upstreamSchema is used by deleteMutationResults; the other\n  // fields are required by the interface but ignored for this operation.\n  await dbProvider.transaction(\n    async (_, hooks) => {\n      await hooks.deleteMutationResults(args);\n    },\n    {\n      upstreamSchema: queryParams.schema,\n      clientGroupID: args.clientGroupID,\n      clientID,\n      mutationID: 0,\n    },\n  );\n}\n\ntype DatabaseTransactionPhase = 'open' | 'execute';\nclass DatabaseTransactionError extends Error {\n  constructor(phase: DatabaseTransactionPhase, options?: ErrorOptions) {\n    super(\n      phase === 'open'\n        ? `Failed to open database transaction: ${getErrorMessage(options?.cause)}`\n        : `Database transaction failed after opening: ${getErrorMessage(options?.cause)}`,\n      options,\n    );\n    this.name = 'DatabaseTransactionError';\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;AAmGA,IAAM,0BAA0B,OAAU,OAAqC;CAC7E,IAAI;EACF,OAAO,MAAM,GAAG;CAClB,SAAS,OAAO;EACd,IACE,iBAAiB,4BACjB,iBAAiB,sBACjB,iBAAiB,iCACjB,mBAAmB,KAAK,GAExB,MAAM;EAGR,MAAM,yBAAyB,KAAK;CACtC;AACF;;;;AAKA,IAAa,wBAAwB;AAqHrC,eAAsB,oBAGpB,mBACA,cACA,gBACA,gBACA,mBACA,cACyB;CACzB,MAAM,aACJ,OAAO,sBAAsB,YAAY,aAAa,oBAClD,4BAA4B,iBAAiB,IAC7C,iCACE,mBACA,cACA,gBACA,gBACA,mBACA,YACF;CAEN,MAAM,KAAK,iBACT,WAAW,UACX,WAAW,OACb,EAAE,YAAY,eAAe;CAC7B,IAAI;CAEJ,IAAI,WAAW,SAAS,WACtB,IAAI;EACF,WAAW,MAAM,WAAW,QAAQ,KAAK;CAC3C,SAAS,OAAO;EACd,GAAG,QAAQ,6BAA6B,KAAK;EAC7C,MAAM,UAAU,8BAA8B,gBAAgB,KAAK;EACnE,MAAM,UAAU,gBAAgB,KAAK;EACrC,OAAO;GACL,MAAM;GACN,QAAQ;GACR,QAAQ;GACR;GACA,aAAa,CAAC;GACd,GAAI,UAAU,EAAC,QAAO,IAAI,CAAC;EAC7B;CACF;MAEA,WAAW,WAAW;CAGxB,IAAI,cAA4B,CAAC;CAEjC,IAAI;CACJ,IAAI;EACF,WAAW,MAAQ,UAAU,gBAAgB,aAAa;EAC1D,cAAc,SAAS,UAAU,KAAI,OAAM;GACzC,IAAI,EAAE;GACN,UAAU,EAAE;EACd,EAAE;CACJ,SAAS,OAAO;EACd,GAAG,QAAQ,6BAA6B,KAAK;EAC7C,MAAM,UAAU,8BAA8B,gBAAgB,KAAK;EACnE,MAAM,UAAU,gBAAgB,KAAK;EACrC,OAAO;GACL,MAAM;GACN,QAAQ;GACR,QAAQ;GACR;GACA;GACA,GAAI,UAAU,EAAC,QAAO,IAAI,CAAC;EAC7B;CACF;CAEA,IAAI;CACJ,IAAI;EACF,oBAAoB,MAClB,WAAW,SAAS,YAChB,OAAO,YAAY,IAAI,IAAI,WAAW,QAAQ,GAAG,EAAE,YAAY,IAC/D,WAAW,aACf,oBACA,aACF;CACF,SAAS,OAAO;EACd,GAAG,QAAQ,yCAAyC,KAAK;EACzD,MAAM,UAAU,0CAA0C,gBAAgB,KAAK;EAC/E,MAAM,UAAU,gBAAgB,KAAK;EACrC,OAAO;GACL,MAAM;GACN,QAAQ;GACR,QAAQ;GACR;GACA;GACA,GAAI,UAAU,EAAC,QAAO,IAAI,CAAC;EAC7B;CACF;CAEA,IAAI,SAAS,gBAAgB,GAQ3B,OAAO;EANL,MAAM;EACN,QAAQ;EACR,QAAQ;EACR;EACA,SAAS,6BAA6B,SAAS;CAE1C;CAGT,MAAM,YAAgC,CAAC;CACvC,IAAI,iBAAiB;CAErB,IAAI;EACF,MAAM,aAAa,IAAI,WACrB,WAAW,YACX,UACA,mBACA,EACF;EASA,KAAK,MAAM,KAAK,SAAS,WAAW;GAElC,IAAI,EAAE,SAAS,YAAY,EAAE,SAAA,wBAAwC;IACnE,GAAG,QACD,iCAAiC,EAAE,KAAK,cAAc,EAAE,SAAS,EACnE;IACA,IAAI;KACF,MAAM,8BACJ,WAAW,YACX,GACA,mBACA,EACF;KAEA;IACF,SAAS,OAAO;KACd,GAAG,OACD,iDAAiD,EAAE,YACnD,KACF;KAEA;IACF;IACA;GACF;GAEA,OAAO,EAAE,SAAS,UAAU,0BAA0B;GACtD,GAAG,QACD,wBAAwB,EAAE,KAAK,QAAQ,EAAE,GAAG,aAAa,EAAE,SAAS,EACtE;GAEA,IAAI,gBAA+B;GAEnC,MAAM,gBAA+B,OAAM,YAAW;IACpD,gBAAgB;IAChB,MAAM,SAAS,MAAM,WAAW,SAAS,GAAG,OAAO;IACnD,gBAAgB;IAChB,OAAO;GACT;GAEA,IAAI;IACF,MAAM,MAAM,MAAM,8BAChB,WAAW,QAAQ,eAAe,CAAC,CACrC;IACA,UAAU,KAAK,GAAG;IAClB,GAAG,QAAQ,aAAa,EAAE,KAAK,QAAQ,EAAE,GAAG,yBAAyB;IAErE;GACF,SAAS,OAAO;IACd,IAAI,CAAC,mBAAmB,KAAK,GAC3B,MAAM;IAGR,IAAI,kBAAkB,kBAEpB,MAAM,WAAW,6BAA6B,GAAG,KAAK;SACjD,IAAI,kBAAkB,cAE3B,GAAG,QACD,oDAAoD,EAAE,GAAG,cAAc,EAAE,YACzE,KACF;IAGF,GAAG,OACD,yCAAyC,EAAE,GAAG,cAAc,EAAE,YAC9D,KACF;IACA,UAAU,KAAK,qBAAqB,GAAG,KAAK,CAAC;IAE7C;GACF;EACF;EAEA,OAAO;GACL,MAAM;GACN,WAAW;GACX,GAAI,OAAO,WAAW,WAAW,cAC7B,EAAC,QAAQ,WAAW,OAAM,IAC1B,CAAC;EACP;CACF,SAAS,OAAO;EACd,GAAG,QAAQ,kCAAkC,KAAK;EAElD,MAAM,yBAAyB,YAAY,MAAM,cAAc;EAE/D,MAAM,UAAU,gBAAgB,KAAK;EACrC,MAAM,UAAU,gBAAgB,KAAK;EAErC,OAAO;GACL,MAAM;GACN,QAAQ;GACR,QACE,iBAAiB,qBACb,uBACA,iBAAiB,2BACf,WACA;GACR;GACA,aAAa;GACb,GAAI,UAAU,EAAC,QAAO,IAAI,CAAC;EAC7B;CACF;AACF;AAEA,SAAS,4BAEP,OAAmE;CACnE,IAAI,aAAa,OACf,OAAO;EACL,MAAM;EACN,YAAY,MAAM;EAClB,SAAS,MAAM;EACf,SAAS,MAAM;EACf,QAAQ,MAAM,UAAU;EACxB,UAAU,MAAM,YAAY;EAC5B,SAAS,MAAM,WAAW;CAC5B;CAGF,OAAO;EACL,MAAM;EACN,YAAY,MAAM;EAClB,SAAS,MAAM;EACf,UAAU,MAAM;EAChB,QAAQ,MAAM,UAAU;EACxB,aACE,MAAM,iBAAiB,kBACnB,OAAO,YAAY,MAAM,KAAK,IAC9B,MAAM;EACZ,UAAU,MAAM,YAAY;EAC5B,SAAS,MAAM,WAAW;CAC5B;AACF;AAEA,SAAS,iCAGP,YACA,SACA,gBACA,gBACA,mBACA,cACgC;CAChC,OAAO,OAAO,YAAY,YAAY,8BAA8B;CACpE,OACE,OAAO,mBAAmB,aAC1B,0CACF;CAEA,IAAI,0BAA0B,SAC5B,OAAO;EACL,MAAM;EACN;EACA;EACA,SAAS;EACT,QAAQ,KAAA;EACR,UAAW,kBAA2C;EAGtD,SAAU,qBAA6C;CACzD;CAGF,OACE,OAAO,mBAAmB,aAC1B,+BACF;CAEA,OAAO;EACL,MAAM;EACN;EACA;EACA,UAAU;EACV,QAAQ,KAAA;EACR,aACE,0BAA0B,kBACtB,OAAO,YAAY,cAAc,IACjC;EACN,UAAW,qBAA8C;EACzD,SAAS,gBAAgB;CAC3B;AACF;AASA,IAAM,aAAN,MAAgE;CAC9D;CACA;CACA;CACA;CAEA,YAAY,YAAe,KAAe,QAAgB,IAAgB;EACxE,KAAKA,cAAc;EACnB,KAAKC,OAAO;EACZ,KAAKC,UAAU;EACf,KAAKC,MAAM;CACb;;;;;;;;CASA,aACE,YACA,OACA,SACkB;EAClB,IAAI,WAAW,WAAW,GACxB,OAAO,QAAQ,QAAQ,KAAK;EAE9B,MAAM,MAAM,OAAO,gBAAkD;GACnE,IAAI;IACF,OAAQ,MAAM,YAAY,OAAO,OAAO,MAAO;GACjD,SAAS,YAAY;IACnB,KAAKA,IAAI,OACP,8CACA,UACF;IACA,OAAO;GACT;EACF;EACA,OAAO,IAAI,SAAQ,YAAW;GAC5B,IAAI,UAAU,WAAW;GACzB,KAAK,MAAM,eAAe,YACxB,IAAS,WAAW,EAAE,MAAK,QAAO;IAChC,IAAI,KACF,QAAQ,IAAI;SACP,IAAI,EAAE,YAAY,GACvB,QAAQ,KAAK;GAEjB,CAAC;EAEL,CAAC;CACH;CAEA,WAAW,OACT,UACA,OAC8B;EAC9B,IAAI,WAAyC,KAAA;EAG7C,IAAI,UAAU;EACd,SAAS;GAEP,MAAM,aAA+B,CAAC;GACtC,IAAI;IACF,MAAM,MAAM,MAAM,KAAKC,cACrB,UACA,IACA,WACA,gBAAe;KACb,WAAW,KAAK,WAAW;IAC7B,CACF;IACA,IAAI,aAAa,KAAA,GAAW;KAC1B,KAAKD,IAAI,OAEP,YAAY,SAAS,GAAG,cAAc,SAAS,SAAS,8BACxD,QACF;KACA,OAAO,qBAAqB,UAAU,QAAQ;IAChD;IAEA,OAAO;GACT,SAAS,OAAO;IACd,IAAI,iBAAiB,oBAAoB;KACvC,KAAKA,IAAI,QAAQ,KAAK;KACtB,MAAM;IACR;IAEA,IAAI,iBAAiB,+BAA+B;KAClD,KAAKA,IAAI,OAAO,KAAK;KACrB,OAAO;MACL,IAAI;OACF,UAAU,SAAS;OACnB,IAAI,SAAS;MACf;MACA,QAAQ;OACN,OAAO;OACP,SAAS,MAAM;MACjB;KACF;IACF;IAEA,IAAI,aAAa,KAAA,GAAW;KAE1B,KAAKA,IAAI,QAEP,kCAAkC,SAAS,GAAG,cAAc,SAAS,YACrE,KACF;KACA,MAAM;IACR;IAIA,MAAM,gBACJ,iBAAiB,2BACZ,MAAM,SAAS,QAChB;IAUN,IACE,UAAA,KACC,MAAM,KAAKE,aAAa,YAAY,eAAe,OAAO,GAC3D;KACA,KAAKF,IAAI,OAEP,YAAY,SAAS,GAAG,cAAc,SAAS,SAAS,qCAAqC,QAAQ,IACrG,aACF;KACA;KACA;IACF;IAGA,WAAW,yBAAyB,aAAa;IACjD,KAAKA,IAAI,OAEP,6BAA6B,SAAS,GAAG,cAAc,SAAS,SAAS,6BACzE,QACF;IACA;GACF;EACF;CACF;CAEA,MAAM,6BACJ,UACA,UAC2B;EAU3B,OAAO,MAPW,KAAKC,cACrB,gBAEM,aACN,UACA,KAAA,CACF;CAEF;CAEA,MAAMA,cACJ,UACA,IACA,UACA,mBAC2B;EAC3B,IAAI,mBAA6C;EAEjD,IAAI;GAsCF,OAAO,MArCW,KAAKJ,YAAY,YACjC,OAAO,MAAM,qBAAqB;IAEhC,mBAAmB;IAEnB,MAAM,KAAKM,iCACT,kBACA,SAAS,UACT,SAAS,EACX;IAEA,IAAI,aAAa,KAAA,GAAW;KAC1B,KAAKH,IAAI,QAEP,sBAAsB,SAAS,KAAK,QAAQ,SAAS,GAAG,EAC1D;KACA,MAAM,OAAQ,MAAM,GAAG,MAAM,SAAS,MAAM,SAAS,KAAK,EAAE;KAG5D,MAAM,KAAK;MAAC,UAAU,SAAS;MAAU,IAAI,SAAS;KAAE;KACxD,IAAI,SAAS,KAAA,GACX,OAAO;MAAC;MAAI,QAAQ,CAAC;KAAC;KAIxB,MAAM,gBAAkC;MAAC;MAAI,QAAQ,EAAC,KAAI;KAAC;KAC3D,MAAM,iBAAiB,oBAAoB,aAAa;KACxD,OAAO;IACT,OAAO;KACL,MAAM,iBAAiB,qBAAqB,UAAU,QAAQ;KAC9D,MAAM,iBAAiB,oBAAoB,cAAc;KACzD,OAAO;IACT;GACF,GACA,KAAKI,qBAAqB,UAAU,iBAAiB,CACvD;EAGF,SAAS,OAAO;GACd,IACE,mBAAmB,KAAK,KACxB,iBAAiB,sBACjB,iBAAiB,+BAEjB,MAAM;GAGR,MAAM,IAAI,yBAAyB,kBAAkB,EAAC,OAAO,MAAK,CAAC;EACrE;CACF;CAEA,qBACE,UACA,mBAC0B;EAC1B,OAAO;GACL,gBAAgB,KAAKL,QAAQ;GAC7B,eAAe,KAAKD,KAAK;GACzB,UAAU,SAAS;GACnB,YAAY,SAAS;GACrB;EACF;CACF;CAEA,MAAMK,iCACJ,kBACA,UACA,oBACA;EACA,MAAM,EAAC,mBAAkB,MAAM,iBAAiB,uBAAuB;EAEvE,IAAI,qBAAqB,gBACvB,MAAM,IAAI,8BACR,UACA,oBACA,cACF;OACK,IAAI,qBAAqB,gBAC9B,MAAM,IAAI,mBACR,UACA,oBACA,cACF;CAEJ;AACF;AAEA,IAAa,qBAAb,cAAwC,MAAM;CAC5C,YACE,UACA,oBACA,gBACA;EACA,MACE,UAAU,SAAS,oBAAoB,mBAAmB,gBAAgB,gBAC5E;CACF;AACF;AAEA,SAAS,qBACP,GACA,OACkB;CAClB,OAAO;EACL,IAAI;GACF,UAAU,EAAE;GACZ,IAAI,EAAE;EACR;EACA,QAAQ;GACN,OAAO;GACP,SAAS,MAAM;GACf,GAAI,MAAM,UAAU,EAAC,SAAS,MAAM,QAAO,IAAI,CAAC;EAClD;CACF;AACF;;AAGA,SAAgB,YAEd,UACA,MAEwB;CAExB,MAAM,UAAU,gBAAgB,UADnB,KAAK,MAAM,WACkB,CAAI;CAC9C,OAAO,OAAO,YAAY,YAAY,0BAA0B,MAAM;CAEtE,IAAI,UAAU,OAAO,GAGnB,QAAQ,IAAI,MAAM,QAAQ,QAAQ,GAAG;EAAC;EAAM;EAAK;CAAE,CAAC;CAItD,OAAO;AACT;AAEA,SAAS,gBACP,KACA,MACS;CACT,IAAI,UAAmB;CACvB,KAAK,MAAM,QAAQ,MAAM;EACvB,IAAI,OAAO,YAAY,YAAY,YAAY,QAAQ,EAAE,QAAQ,UAC/D;EAEF,UAAW,QAAoC;CACjD;CACA,OAAO;AACT;;;;;;AAOA,eAAe,8BAGb,YACA,UACA,aACA,IACe;CACf,MAAM,cAAc,KAAO,SAAS,KAAK,IAAI,uBAAuB;CACpE,IAAI,CAAC,YAAY,IAAI;EAEnB,GAAG,OAAO,qCAAqC,YAAY,KAAK;EAChE;CACF;CACA,MAAM,OAA0B,YAAY;CAI5C,MAAM,WACJ,UAAU,QAAQ,KAAK,SAAS,SAAS,KAAK,UAAU,KAAK,KAAK;CAKpE,MAAM,WAAW,YACf,OAAO,GAAG,UAAU;EAClB,MAAM,MAAM,sBAAsB,IAAI;CACxC,GACA;EACE,gBAAgB,YAAY;EAC5B,eAAe,KAAK;EACpB;EACA,YAAY;CACd,CACF;AACF;AAGA,IAAM,2BAAN,cAAuC,MAAM;CAC3C,YAAY,OAAiC,SAAwB;EACnE,MACE,UAAU,SACN,wCAAwC,gBAAgB,SAAS,KAAK,MACtE,8CAA8C,gBAAgB,SAAS,KAAK,KAChF,OACF;EACA,KAAK,OAAO;CACd;AACF"}
+{"version":3,"file":"process-mutations.js","names":["ErrorKind.PushFailed","ErrorOrigin.Server","ErrorReason.Parse","v.parse","ErrorReason.UnsupportedPushVersion","ErrorReason.OutOfOrderMutation","ErrorReason.Database","ErrorReason.Internal","v.test"],"sources":["../../../../zero-server/src/process-mutations.ts"],"sourcesContent":["import {\n  consoleLogSink,\n  type LogContext,\n  type LogLevel,\n  type LogSink,\n} from '@rocicorp/logger';\nimport {assert} from '../../shared/src/asserts.ts';\nimport {getErrorDetails, getErrorMessage} from '../../shared/src/error.ts';\nimport type {ReadonlyJSONValue} from '../../shared/src/json.ts';\nimport {promiseVoid} from '../../shared/src/resolved-promises.ts';\nimport type {MaybePromise} from '../../shared/src/types.ts';\nimport * as v from '../../shared/src/valita.ts';\nimport {MutationAlreadyProcessedError} from '../../zero-cache/src/services/mutagen/error.ts';\nimport type {ApplicationError} from '../../zero-protocol/src/application-error.ts';\nimport {\n  isApplicationError,\n  wrapWithApplicationError,\n} from '../../zero-protocol/src/application-error.ts';\nimport {ErrorKind} from '../../zero-protocol/src/error-kind.ts';\nimport {ErrorOrigin} from '../../zero-protocol/src/error-origin.ts';\nimport {ErrorReason} from '../../zero-protocol/src/error-reason.ts';\nimport type {PushFailedBody} from '../../zero-protocol/src/error.ts';\nimport {\n  mutateParamsSchema,\n  type MutateResponse,\n} from '../../zero-protocol/src/mutate-server.ts';\nimport type {MutationID} from '../../zero-protocol/src/mutation-id.ts';\nimport {\n  CLEANUP_RESULTS_MUTATION_NAME,\n  cleanupResultsArgSchema,\n  type CleanupResultsArg,\n  type CustomMutation,\n  type Mutation,\n} from '../../zero-protocol/src/mutation.ts';\nimport {\n  pushBodySchema,\n  type MutationResponse,\n  type PushBody,\n} from '../../zero-protocol/src/push.ts';\nimport type {RetryPredicate} from '../../zql/src/mutate/custom.ts';\nimport type {AnyMutatorRegistry} from '../../zql/src/mutate/mutator-registry.ts';\nimport {isMutator} from '../../zql/src/mutate/mutator.ts';\nimport type {CustomMutatorDefs, CustomMutatorImpl} from './custom.ts';\nimport {createLogContext} from './logging.ts';\nimport {separatorRe} from './push-processor.ts';\n\nexport interface TransactionProviderHooks {\n  updateClientMutationID: () => Promise<{lastMutationID: number | bigint}>;\n  writeMutationResult: (result: MutationResponse) => Promise<void>;\n  deleteMutationResults: (args: CleanupResultsArg) => Promise<void>;\n}\n\nexport interface TransactionProviderInput {\n  upstreamSchema: string;\n  clientGroupID: string;\n  clientID: string;\n  mutationID: number;\n  /**\n   * Receives each predicate the mutator registers with `tx.retryOn()`. A\n   * `Database` passes it on to the server transaction it creates.\n   */\n  addRetryPredicate?: ((shouldRetry: RetryPredicate) => void) | undefined;\n}\n\n/**\n * Defines the abstract interface for a database that PushProcessor can execute\n * transactions against.\n */\nexport interface Database<T> {\n  transaction: <R>(\n    callback: (\n      tx: T,\n      transactionHooks: TransactionProviderHooks,\n    ) => MaybePromise<R>,\n    transactionInput?: TransactionProviderInput,\n  ) => Promise<R>;\n}\n\nexport type ExtractTransactionType<D> = D extends Database<infer T> ? T : never;\nexport type Params = v.Infer<typeof mutateParamsSchema>;\n\nexport type TransactFn<D extends Database<ExtractTransactionType<D>>> = (\n  cb: TransactFnCallback<D>,\n) => Promise<MutationResponse>;\n\nexport type TransactFnCallback<D extends Database<ExtractTransactionType<D>>> =\n  (\n    tx: ExtractTransactionType<D>,\n    mutatorName: string,\n    mutatorArgs: ReadonlyJSONValue | undefined,\n  ) => Promise<ReadonlyJSONValue | void>;\n\nexport type Parsed<D extends Database<ExtractTransactionType<D>>> = {\n  transact: TransactFn<D>;\n  mutations: CustomMutation[];\n};\n\ntype MutationPhase = 'preTransaction' | 'transactionPending' | 'postCommit';\n\nconst applicationErrorWrapper = async <T>(fn: () => Promise<T>): Promise<T> => {\n  try {\n    return await fn();\n  } catch (error) {\n    if (\n      error instanceof DatabaseTransactionError ||\n      error instanceof OutOfOrderMutation ||\n      error instanceof MutationAlreadyProcessedError ||\n      isApplicationError(error)\n    ) {\n      throw error;\n    }\n\n    throw wrapWithApplicationError(error);\n  }\n};\n\n/**\n * @deprecated Use {@link handleMutateRequest} instead.\n */\nexport const handleMutationRequest = handleMutateRequest;\n\n/**\n * Parsed query params accepted by {@link handleMutateRequest} when the\n * incoming request URL has already been handled by your framework.\n */\nexport type MutateSearchParams = URLSearchParams | Record<string, string>;\n\n/**\n * Runs once per custom mutation in a `/mutate` request.\n *\n * Call `transact` to execute database work for the mutation. Throwing before\n * `transact` persists an application error; throwing after `transact` resolves\n * only affects server-side logging because the mutation is already committed.\n */\nexport type MutateRequestHandler<\n  D extends Database<ExtractTransactionType<D>>,\n> = (\n  transact: TransactFn<D>,\n  mutation: CustomMutation,\n) => Promise<MutationResponse>;\n\nexport type HandleMutateRequestArgs<\n  D extends Database<ExtractTransactionType<D>>,\n> = {\n  /** Database used to run transactions and store mutation results. */\n  dbProvider: D;\n  handler: MutateRequestHandler<D>;\n  /**\n   * Authenticated user ID. Null or undefined means the user is logged out.\n   */\n  userID: string | null | undefined;\n  /** Optional log level for request parsing and execution. */\n  logLevel?: LogLevel | undefined;\n  /**\n   * Destination for this request's log output. When omitted, logs are written\n   * with `console.log`/`info`/`warn`/`error` etc.\n   *\n   * Provide a custom {@link LogSink} to redirect logs — for example into an\n   * application's own structured logging pipeline, so that records emitted while\n   * processing a push carry the same fields as the rest of the server's logs.\n   */\n  logSink?: LogSink | undefined;\n} & (\n  | {\n      /** Fetch request containing both query params and the JSON body. */\n      request: Request;\n    }\n  | {\n      /** Parsed query params from the `/mutate` request URL. */\n      query: MutateSearchParams;\n      /** Parsed JSON body from the `/mutate` request. */\n      body: ReadonlyJSONValue;\n    }\n);\n\ntype NormalizedMutateRequestArgs<\n  D extends Database<ExtractTransactionType<D>>,\n> = {\n  readonly dbProvider: D;\n  readonly handler: MutateRequestHandler<D>;\n  // Note: semantics of undefined differ from HandleMutateRequestArgs.userID.\n  // Here, undefined means the app didn't provide a user ID - we do not know if\n  // the user is logged in or not. This is legacy behavior needed to support\n  // deprecated signatures of handleMutateRequest which did not receive userID\n  // from app.\n  readonly userID: string | null | undefined;\n  readonly logLevel: LogLevel;\n  readonly logSink: LogSink;\n} & (\n  | {\n      readonly type: 'request';\n      readonly request: Request;\n    }\n  | {\n      readonly type: 'body';\n      readonly queryParams: Record<string, string>;\n      readonly jsonBody: ReadonlyJSONValue;\n    }\n);\n\n/**\n * Process a `/mutate` request from a Fetch `Request`.\n */\nexport function handleMutateRequest<\n  D extends Database<ExtractTransactionType<D>>,\n>(input: HandleMutateRequestArgs<D>): Promise<MutateResponse>;\n\n/**\n * @deprecated Pass a single object instead:\n * `handleMutateRequest({dbProvider, handler, query, body, userID, logLevel})`.\n */\nexport function handleMutateRequest<\n  D extends Database<ExtractTransactionType<D>>,\n>(\n  dbProvider: D,\n  handler: MutateRequestHandler<D>,\n  query: MutateSearchParams,\n  body: ReadonlyJSONValue,\n  logLevel?: LogLevel,\n  logSink?: LogSink,\n): Promise<MutateResponse>;\n\n/**\n * @deprecated Pass a single object instead:\n * `handleMutateRequest({dbProvider, handler, request, userID, logLevel})`.\n */\nexport function handleMutateRequest<\n  D extends Database<ExtractTransactionType<D>>,\n>(\n  dbProvider: D,\n  handler: MutateRequestHandler<D>,\n  request: Request,\n  logLevel?: LogLevel,\n  logSink?: LogSink,\n): Promise<MutateResponse>;\n\nexport async function handleMutateRequest<\n  D extends Database<ExtractTransactionType<D>>,\n>(\n  inputOrDbProvider: HandleMutateRequestArgs<D> | D,\n  maybeHandler?: MutateRequestHandler<D> | undefined,\n  requestOrQuery?: Request | MutateSearchParams | undefined,\n  bodyOrLogLevel?: ReadonlyJSONValue | LogLevel | undefined,\n  logLevelOrLogSink?: LogLevel | LogSink | undefined,\n  maybeLogSink?: LogSink | undefined,\n): Promise<MutateResponse> {\n  const normalized =\n    typeof inputOrDbProvider === 'object' && 'handler' in inputOrDbProvider\n      ? normalizeMutateRequestInput(inputOrDbProvider)\n      : normalizeLegacyMutateRequestArgs(\n          inputOrDbProvider,\n          maybeHandler,\n          requestOrQuery,\n          bodyOrLogLevel,\n          logLevelOrLogSink,\n          maybeLogSink,\n        );\n\n  const lc = createLogContext(\n    normalized.logLevel,\n    normalized.logSink,\n  ).withContext('PushProcessor');\n  let jsonBody: unknown;\n\n  if (normalized.type === 'request') {\n    try {\n      jsonBody = await normalized.request.json();\n    } catch (error) {\n      lc.error?.('Failed to parse push body', error);\n      const message = `Failed to parse push body: ${getErrorMessage(error)}`;\n      const details = getErrorDetails(error);\n      return {\n        kind: ErrorKind.PushFailed,\n        origin: ErrorOrigin.Server,\n        reason: ErrorReason.Parse,\n        message,\n        mutationIDs: [],\n        ...(details ? {details} : {}),\n      } as const satisfies PushFailedBody;\n    }\n  } else {\n    jsonBody = normalized.jsonBody;\n  }\n\n  let mutationIDs: MutationID[] = [];\n\n  let pushBody: PushBody;\n  try {\n    pushBody = v.parse(jsonBody, pushBodySchema, 'passthrough');\n    mutationIDs = pushBody.mutations.map(m => ({\n      id: m.id,\n      clientID: m.clientID,\n    }));\n  } catch (error) {\n    lc.error?.('Failed to parse push body', error);\n    const message = `Failed to parse push body: ${getErrorMessage(error)}`;\n    const details = getErrorDetails(error);\n    return {\n      kind: ErrorKind.PushFailed,\n      origin: ErrorOrigin.Server,\n      reason: ErrorReason.Parse,\n      message,\n      mutationIDs,\n      ...(details ? {details} : {}),\n    } as const satisfies PushFailedBody;\n  }\n\n  let parsedQueryParams: Params;\n  try {\n    parsedQueryParams = v.parse(\n      normalized.type === 'request'\n        ? Object.fromEntries(new URL(normalized.request.url).searchParams)\n        : normalized.queryParams,\n      mutateParamsSchema,\n      'passthrough',\n    );\n  } catch (error) {\n    lc.error?.('Failed to parse push query parameters', error);\n    const message = `Failed to parse push query parameters: ${getErrorMessage(error)}`;\n    const details = getErrorDetails(error);\n    return {\n      kind: ErrorKind.PushFailed,\n      origin: ErrorOrigin.Server,\n      reason: ErrorReason.Parse,\n      message,\n      mutationIDs,\n      ...(details ? {details} : {}),\n    } as const satisfies PushFailedBody;\n  }\n\n  if (pushBody.pushVersion !== 1) {\n    const response = {\n      kind: ErrorKind.PushFailed,\n      origin: ErrorOrigin.Server,\n      reason: ErrorReason.UnsupportedPushVersion,\n      mutationIDs,\n      message: `Unsupported push version: ${pushBody.pushVersion}`,\n    } as const satisfies PushFailedBody;\n    return response;\n  }\n\n  const responses: MutationResponse[] = [];\n  let processedCount = 0;\n\n  try {\n    const transactor = new Transactor(\n      normalized.dbProvider,\n      pushBody,\n      parsedQueryParams,\n      lc,\n    );\n\n    // Each mutation goes through three phases:\n    //   1. Pre-transaction: user logic that runs before `transact` is called. If\n    //      this throws we still advance LMID and persist the failure result.\n    //   2. Transaction: the callback passed to `transact`, which can be retried\n    //      if it fails with an ApplicationError.\n    //   3. Post-commit: any logic that runs after `transact` resolves. Failures\n    //      here are logged but the mutation remains committed.\n    for (const m of pushBody.mutations) {\n      // Handle internal mutations (like cleanup) directly without user dispatch\n      if (m.type === 'custom' && m.name === CLEANUP_RESULTS_MUTATION_NAME) {\n        lc.debug?.(\n          `Processing internal mutation '${m.name}' (clientID=${m.clientID})`,\n        );\n        try {\n          await processCleanupResultsMutation(\n            normalized.dbProvider,\n            m,\n            parsedQueryParams,\n            lc,\n          );\n          // No response added - this is fire-and-forget\n          processedCount++;\n        } catch (error) {\n          lc.warn?.(\n            `Failed to process cleanup mutation for client ${m.clientID}`,\n            error,\n          );\n          // Don't fail the whole push for cleanup errors\n          processedCount++;\n        }\n        continue;\n      }\n\n      assert(m.type === 'custom', 'Expected custom mutation');\n      lc.debug?.(\n        `Processing mutation '${m.name}' (id=${m.id}, clientID=${m.clientID})`,\n      );\n\n      let mutationPhase: MutationPhase = 'preTransaction';\n\n      const transactProxy: TransactFn<D> = async innerCb => {\n        mutationPhase = 'transactionPending';\n        const result = await transactor.transact(m, innerCb);\n        mutationPhase = 'postCommit';\n        return result;\n      };\n\n      try {\n        const res = await applicationErrorWrapper(() =>\n          normalized.handler(transactProxy, m),\n        );\n        responses.push(res);\n        lc.debug?.(`Mutation '${m.name}' (id=${m.id}) completed successfully`);\n\n        processedCount++;\n      } catch (error) {\n        if (!isApplicationError(error)) {\n          throw error;\n        }\n\n        if (mutationPhase === 'preTransaction') {\n          // Pre-transaction\n          await transactor.persistPreTransactionFailure(m, error);\n        } else if (mutationPhase === 'postCommit') {\n          // Post-commit\n          lc.error?.(\n            `Post-commit mutation handler failed for mutation ${m.id} for client ${m.clientID}`,\n            error,\n          );\n        }\n\n        lc.warn?.(\n          `Application error processing mutation ${m.id} for client ${m.clientID}`,\n          error,\n        );\n        responses.push(makeAppErrorResponse(m, error));\n\n        processedCount++;\n      }\n    }\n\n    return {\n      kind: 'MutateResponse',\n      mutations: responses,\n      ...(typeof normalized.userID !== 'undefined'\n        ? {userID: normalized.userID}\n        : {}),\n    } as const satisfies MutateResponse;\n  } catch (error) {\n    lc.error?.('Failed to process push request', error);\n    // only include mutationIDs for mutations that were not processed\n    const unprocessedMutationIDs = mutationIDs.slice(processedCount);\n\n    const message = getErrorMessage(error);\n    const details = getErrorDetails(error);\n\n    return {\n      kind: ErrorKind.PushFailed,\n      origin: ErrorOrigin.Server,\n      reason:\n        error instanceof OutOfOrderMutation\n          ? ErrorReason.OutOfOrderMutation\n          : error instanceof DatabaseTransactionError\n            ? ErrorReason.Database\n            : ErrorReason.Internal,\n      message,\n      mutationIDs: unprocessedMutationIDs,\n      ...(details ? {details} : {}),\n    };\n  }\n}\n\nfunction normalizeMutateRequestInput<\n  D extends Database<ExtractTransactionType<D>>,\n>(input: HandleMutateRequestArgs<D>): NormalizedMutateRequestArgs<D> {\n  if ('request' in input) {\n    return {\n      type: 'request',\n      dbProvider: input.dbProvider,\n      handler: input.handler,\n      request: input.request,\n      userID: input.userID ?? null,\n      logLevel: input.logLevel ?? 'info',\n      logSink: input.logSink ?? consoleLogSink,\n    };\n  }\n\n  return {\n    type: 'body',\n    dbProvider: input.dbProvider,\n    handler: input.handler,\n    jsonBody: input.body,\n    userID: input.userID ?? null,\n    queryParams:\n      input.query instanceof URLSearchParams\n        ? Object.fromEntries(input.query)\n        : input.query,\n    logLevel: input.logLevel ?? 'info',\n    logSink: input.logSink ?? consoleLogSink,\n  };\n}\n\nfunction normalizeLegacyMutateRequestArgs<\n  D extends Database<ExtractTransactionType<D>>,\n>(\n  dbProvider: D,\n  handler: MutateRequestHandler<D> | undefined,\n  requestOrQuery: Request | MutateSearchParams | undefined,\n  bodyOrLogLevel: ReadonlyJSONValue | LogLevel | undefined,\n  logLevelOrLogSink: LogLevel | LogSink | undefined,\n  maybeLogSink: LogSink | undefined,\n): NormalizedMutateRequestArgs<D> {\n  assert(typeof handler === 'function', 'Handler function is required');\n  assert(\n    typeof requestOrQuery !== 'undefined',\n    'Request or query parameters are required',\n  );\n\n  if (requestOrQuery instanceof Request) {\n    return {\n      type: 'request',\n      dbProvider,\n      handler,\n      request: requestOrQuery,\n      userID: undefined,\n      logLevel: (bodyOrLogLevel as LogLevel | undefined) ?? 'info',\n      // The `request` overload has no `body` slot, so the sink arrives one\n      // position earlier than it does in the `query`/`body` overload.\n      logSink: (logLevelOrLogSink as LogSink | undefined) ?? consoleLogSink,\n    };\n  }\n\n  assert(\n    typeof bodyOrLogLevel !== 'undefined',\n    'JSON body cannot be undefined',\n  );\n\n  return {\n    type: 'body',\n    dbProvider,\n    handler,\n    jsonBody: bodyOrLogLevel,\n    userID: undefined,\n    queryParams:\n      requestOrQuery instanceof URLSearchParams\n        ? Object.fromEntries(requestOrQuery)\n        : requestOrQuery,\n    logLevel: (logLevelOrLogSink as LogLevel | undefined) ?? 'info',\n    logSink: maybeLogSink ?? consoleLogSink,\n  };\n}\n\n/**\n * The most times a mutator runs for one mutation within one push when its\n * `tx.retryOn()` predicate keeps answering yes: the first run and four\n * re-runs.\n */\nexport const MAX_MUTATOR_RUNS = 5;\n\nclass Transactor<D extends Database<ExtractTransactionType<D>>> {\n  readonly #dbProvider: D;\n  readonly #req: PushBody;\n  readonly #params: Params;\n  readonly #lc: LogContext;\n\n  constructor(dbProvider: D, req: PushBody, params: Params, lc: LogContext) {\n    this.#dbProvider = dbProvider;\n    this.#req = req;\n    this.#params = params;\n    this.#lc = lc;\n  }\n\n  /**\n   * Asks all of the run's `tx.retryOn()` predicates at once whether to run\n   * the mutator again. The first to answer yes decides it and the other\n   * answers are ignored; it is no once every predicate has answered no. A\n   * predicate that throws or rejects answers no: escaping here would skip\n   * recording the mutation's result, and the client would send it again.\n   */\n  #shouldRetry(\n    predicates: readonly RetryPredicate[],\n    error: unknown,\n    attempt: number,\n  ): Promise<boolean> {\n    if (predicates.length === 0) {\n      return Promise.resolve(false);\n    }\n    const ask = async (shouldRetry: RetryPredicate): Promise<boolean> => {\n      try {\n        return (await shouldRetry(error, attempt)) === true;\n      } catch (retryError) {\n        this.#lc.warn?.(\n          'retryOn predicate threw; treating it as no',\n          retryError,\n        );\n        return false;\n      }\n    };\n    return new Promise(resolve => {\n      let pending = predicates.length;\n      for (const shouldRetry of predicates) {\n        void ask(shouldRetry).then(yes => {\n          if (yes) {\n            resolve(true);\n          } else if (--pending === 0) {\n            resolve(false);\n          }\n        });\n      }\n    });\n  }\n\n  transact = async (\n    mutation: CustomMutation,\n    cb: TransactFnCallback<D>,\n  ): Promise<MutationResponse> => {\n    let appError: ApplicationError | undefined = undefined;\n    // Which run of the mutator this is: 1, then 2 for the first re-run, and so\n    // on. When a run fails, it is also the number of the re-run being decided.\n    let attempt = 1;\n    for (;;) {\n      // What this run registers with `tx.retryOn()`. Each run starts empty.\n      const predicates: RetryPredicate[] = [];\n      try {\n        const ret = await this.#transactImpl(\n          mutation,\n          cb,\n          appError,\n          shouldRetry => {\n            predicates.push(shouldRetry);\n          },\n        );\n        if (appError !== undefined) {\n          this.#lc.warn?.(\n            // log-leak-ignore -- mutation and client ids, not row data\n            `Mutation ${mutation.id} for client ${mutation.clientID} was retried after an error`,\n            appError,\n          );\n          return makeAppErrorResponse(mutation, appError);\n        }\n\n        return ret;\n      } catch (error) {\n        if (error instanceof OutOfOrderMutation) {\n          this.#lc.error?.(error);\n          throw error;\n        }\n\n        if (error instanceof MutationAlreadyProcessedError) {\n          this.#lc.warn?.(error);\n          return {\n            id: {\n              clientID: mutation.clientID,\n              id: mutation.id,\n            },\n            result: {\n              error: 'alreadyProcessed',\n              details: error.message,\n            },\n          };\n        }\n\n        if (appError !== undefined) {\n          // Retry also failed → internal error, cannot skip mutation\n          this.#lc.error?.(\n            // log-leak-ignore -- mutation and client ids, not row data\n            `Retry also failed for mutation ${mutation.id} for client ${mutation.clientID}`,\n            error,\n          );\n          throw error;\n        }\n\n        // What the mutator or the database threw, without the wrapper. This\n        // includes errors raised at COMMIT, which the mutator cannot catch.\n        const originalError =\n          error instanceof DatabaseTransactionError\n            ? (error.cause ?? error)\n            : error;\n\n        // One of the run's `tx.retryOn()` predicates says to run it again → do\n        // so, in a fresh transaction.\n        //\n        // `continue` WITHOUT setting `appError` is the whole mechanism: it is\n        // `appError` being set that makes the next `#transactImpl` skip\n        // `cb(...)`. `#transactImpl` opens a new transaction each pass, so the\n        // re-run gets a new snapshot. `#checkAndIncrementLastMutationID` re-runs\n        // safely because the failed run rolled its increment back.\n        if (\n          attempt < MAX_MUTATOR_RUNS &&\n          (await this.#shouldRetry(predicates, originalError, attempt))\n        ) {\n          this.#lc.warn?.(\n            // log-leak-ignore -- mutation and client ids, not row data\n            `Mutation ${mutation.id} for client ${mutation.clientID} failed, re-running mutator (retry ${attempt})`,\n            originalError,\n          );\n          attempt++;\n          continue;\n        }\n\n        // Not re-run → store error and retry without mutator\n        appError = wrapWithApplicationError(originalError);\n        this.#lc.warn?.(\n          // log-leak-ignore -- mutation and client ids, not row data\n          `Error processing mutation ${mutation.id} for client ${mutation.clientID}, retrying without mutator`,\n          appError,\n        );\n        continue;\n      }\n    }\n  };\n\n  async persistPreTransactionFailure(\n    mutation: CustomMutation,\n    appError: ApplicationError<ReadonlyJSONValue | undefined>,\n  ): Promise<MutationResponse> {\n    // User-land code threw before calling `transact`. We still need to bump the\n    // LMID for this mutation and persist the error so that the client knows it failed.\n    const ret = await this.#transactImpl(\n      mutation,\n      // noop callback since there's no transaction to execute\n      () => promiseVoid,\n      appError,\n      undefined,\n    );\n    return ret;\n  }\n\n  async #transactImpl(\n    mutation: CustomMutation,\n    cb: TransactFnCallback<D>,\n    appError: ApplicationError | undefined,\n    addRetryPredicate: ((shouldRetry: RetryPredicate) => void) | undefined,\n  ): Promise<MutationResponse> {\n    let transactionPhase: DatabaseTransactionPhase = 'open';\n\n    try {\n      const ret = await this.#dbProvider.transaction(\n        async (dbTx, transactionHooks) => {\n          // update the transaction phase to 'execute' after the transaction is opened\n          transactionPhase = 'execute';\n\n          await this.#checkAndIncrementLastMutationID(\n            transactionHooks,\n            mutation.clientID,\n            mutation.id,\n          );\n\n          if (appError === undefined) {\n            this.#lc.debug?.(\n              // log-leak-ignore -- mutator name and mutation id, not row data\n              `Executing mutator '${mutation.name}' (id=${mutation.id})`,\n            );\n            const data = (await cb(dbTx, mutation.name, mutation.args[0])) as\n              | ReadonlyJSONValue\n              | undefined;\n            const id = {clientID: mutation.clientID, id: mutation.id};\n            if (data === undefined) {\n              return {id, result: {}};\n            }\n            // Persisted so the data still reaches the client if the push\n            // response is lost. A mutator that returns nothing writes no row.\n            const successResult: MutationResponse = {id, result: {data}};\n            await transactionHooks.writeMutationResult(successResult);\n            return successResult;\n          } else {\n            const mutationResult = makeAppErrorResponse(mutation, appError);\n            await transactionHooks.writeMutationResult(mutationResult);\n            return mutationResult;\n          }\n        },\n        this.#getTransactionInput(mutation, addRetryPredicate),\n      );\n\n      return ret;\n    } catch (error) {\n      if (\n        isApplicationError(error) ||\n        error instanceof OutOfOrderMutation ||\n        error instanceof MutationAlreadyProcessedError\n      ) {\n        throw error;\n      }\n\n      throw new DatabaseTransactionError(transactionPhase, {cause: error});\n    }\n  }\n\n  #getTransactionInput(\n    mutation: CustomMutation,\n    addRetryPredicate: ((shouldRetry: RetryPredicate) => void) | undefined,\n  ): TransactionProviderInput {\n    return {\n      upstreamSchema: this.#params.schema,\n      clientGroupID: this.#req.clientGroupID,\n      clientID: mutation.clientID,\n      mutationID: mutation.id,\n      addRetryPredicate,\n    };\n  }\n\n  async #checkAndIncrementLastMutationID(\n    transactionHooks: TransactionProviderHooks,\n    clientID: string,\n    receivedMutationID: number,\n  ) {\n    const {lastMutationID} = await transactionHooks.updateClientMutationID();\n\n    if (receivedMutationID < lastMutationID) {\n      throw new MutationAlreadyProcessedError(\n        clientID,\n        receivedMutationID,\n        lastMutationID,\n      );\n    } else if (receivedMutationID > lastMutationID) {\n      throw new OutOfOrderMutation(\n        clientID,\n        receivedMutationID,\n        lastMutationID,\n      );\n    }\n  }\n}\n\nexport class OutOfOrderMutation extends Error {\n  constructor(\n    clientID: string,\n    receivedMutationID: number,\n    lastMutationID: number | bigint,\n  ) {\n    super(\n      `Client ${clientID} sent mutation ID ${receivedMutationID} but expected ${lastMutationID}`,\n    );\n  }\n}\n\nfunction makeAppErrorResponse(\n  m: Mutation,\n  error: ApplicationError<ReadonlyJSONValue | undefined>,\n): MutationResponse {\n  return {\n    id: {\n      clientID: m.clientID,\n      id: m.id,\n    },\n    result: {\n      error: 'app',\n      message: error.message,\n      ...(error.details ? {details: error.details} : {}),\n    },\n  };\n}\n\n/** @deprecated Use getMutator instead */\nexport function getMutation(\n  // oxlint-disable-next-line no-explicit-any\n  mutators: AnyMutatorRegistry | CustomMutatorDefs<any>,\n  name: string,\n  // oxlint-disable-next-line no-explicit-any\n): CustomMutatorImpl<any> {\n  const path = name.split(separatorRe);\n  const mutator = getObjectAtPath(mutators, path);\n  assert(typeof mutator === 'function', `could not find mutator ${name}`);\n\n  if (isMutator(mutator)) {\n    // mutator needs to be called with {tx, args, ctx}\n    // CustomMutatorImpl is called with (tx, args, ctx)\n    return (tx, args, ctx) => mutator.fn({args, ctx, tx});\n  }\n\n  // oxlint-disable-next-line no-explicit-any\n  return mutator as CustomMutatorImpl<any>;\n}\n\nfunction getObjectAtPath(\n  obj: Record<string, unknown>,\n  path: string[],\n): unknown {\n  let current: unknown = obj;\n  for (const part of path) {\n    if (typeof current !== 'object' || current === null || !(part in current)) {\n      return undefined;\n    }\n    current = (current as Record<string, unknown>)[part];\n  }\n  return current;\n}\n\n/**\n * Processes internal cleanup mutation that deletes acknowledged mutation results\n * from the upstream database. This runs without LMID tracking since it's an\n * internal operation.\n */\nasync function processCleanupResultsMutation<\n  D extends Database<ExtractTransactionType<D>>,\n>(\n  dbProvider: D,\n  mutation: CustomMutation,\n  queryParams: Params,\n  lc: LogContext,\n): Promise<void> {\n  const parseResult = v.test(mutation.args[0], cleanupResultsArgSchema);\n  if (!parseResult.ok) {\n    // log-leak-ignore -- valita reports the failing type, not the value\n    lc.warn?.('Cleanup mutation has invalid args', parseResult.error);\n    return;\n  }\n  const args: CleanupResultsArg = parseResult.value;\n\n  // Determine clientID for transaction input based on cleanup type\n  // Note: legacy format without type field is treated as single\n  const clientID =\n    'type' in args && args.type === 'bulk' ? args.clientIDs[0] : args.clientID;\n\n  // Run in a transaction, using the hook for DB-specific operation.\n  // Note: only upstreamSchema is used by deleteMutationResults; the other\n  // fields are required by the interface but ignored for this operation.\n  await dbProvider.transaction(\n    async (_, hooks) => {\n      await hooks.deleteMutationResults(args);\n    },\n    {\n      upstreamSchema: queryParams.schema,\n      clientGroupID: args.clientGroupID,\n      clientID,\n      mutationID: 0,\n    },\n  );\n}\n\ntype DatabaseTransactionPhase = 'open' | 'execute';\nclass DatabaseTransactionError extends Error {\n  constructor(phase: DatabaseTransactionPhase, options?: ErrorOptions) {\n    super(\n      phase === 'open'\n        ? `Failed to open database transaction: ${getErrorMessage(options?.cause)}`\n        : `Database transaction failed after opening: ${getErrorMessage(options?.cause)}`,\n      options,\n    );\n    this.name = 'DatabaseTransactionError';\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;AAmGA,IAAM,0BAA0B,OAAU,OAAqC;CAC7E,IAAI;EACF,OAAO,MAAM,GAAG;CAClB,SAAS,OAAO;EACd,IACE,iBAAiB,4BACjB,iBAAiB,sBACjB,iBAAiB,iCACjB,mBAAmB,KAAK,GAExB,MAAM;EAGR,MAAM,yBAAyB,KAAK;CACtC;AACF;;;;AAKA,IAAa,wBAAwB;AAqHrC,eAAsB,oBAGpB,mBACA,cACA,gBACA,gBACA,mBACA,cACyB;CACzB,MAAM,aACJ,OAAO,sBAAsB,YAAY,aAAa,oBAClD,4BAA4B,iBAAiB,IAC7C,iCACE,mBACA,cACA,gBACA,gBACA,mBACA,YACF;CAEN,MAAM,KAAK,iBACT,WAAW,UACX,WAAW,OACb,CAAC,CAAC,YAAY,eAAe;CAC7B,IAAI;CAEJ,IAAI,WAAW,SAAS,WACtB,IAAI;EACF,WAAW,MAAM,WAAW,QAAQ,KAAK;CAC3C,SAAS,OAAO;EACd,GAAG,QAAQ,6BAA6B,KAAK;EAC7C,MAAM,UAAU,8BAA8B,gBAAgB,KAAK;EACnE,MAAM,UAAU,gBAAgB,KAAK;EACrC,OAAO;GACL,MAAMA;GACN,QAAQC;GACR,QAAQC;GACR;GACA,aAAa,CAAC;GACd,GAAI,UAAU,EAAC,QAAO,IAAI,CAAC;EAC7B;CACF;MAEA,WAAW,WAAW;CAGxB,IAAI,cAA4B,CAAC;CAEjC,IAAI;CACJ,IAAI;EACF,WAAWC,MAAQ,UAAU,gBAAgB,aAAa;EAC1D,cAAc,SAAS,UAAU,KAAI,OAAM;GACzC,IAAI,EAAE;GACN,UAAU,EAAE;EACd,EAAE;CACJ,SAAS,OAAO;EACd,GAAG,QAAQ,6BAA6B,KAAK;EAC7C,MAAM,UAAU,8BAA8B,gBAAgB,KAAK;EACnE,MAAM,UAAU,gBAAgB,KAAK;EACrC,OAAO;GACL,MAAMH;GACN,QAAQC;GACR,QAAQC;GACR;GACA;GACA,GAAI,UAAU,EAAC,QAAO,IAAI,CAAC;EAC7B;CACF;CAEA,IAAI;CACJ,IAAI;EACF,oBAAoBC,MAClB,WAAW,SAAS,YAChB,OAAO,YAAY,IAAI,IAAI,WAAW,QAAQ,GAAG,CAAC,CAAC,YAAY,IAC/D,WAAW,aACf,oBACA,aACF;CACF,SAAS,OAAO;EACd,GAAG,QAAQ,yCAAyC,KAAK;EACzD,MAAM,UAAU,0CAA0C,gBAAgB,KAAK;EAC/E,MAAM,UAAU,gBAAgB,KAAK;EACrC,OAAO;GACL,MAAMH;GACN,QAAQC;GACR,QAAQC;GACR;GACA;GACA,GAAI,UAAU,EAAC,QAAO,IAAI,CAAC;EAC7B;CACF;CAEA,IAAI,SAAS,gBAAgB,GAQ3B,OAAO;EANL,MAAMF;EACN,QAAQC;EACR,QAAQG;EACR;EACA,SAAS,6BAA6B,SAAS;CAE1C;CAGT,MAAM,YAAgC,CAAC;CACvC,IAAI,iBAAiB;CAErB,IAAI;EACF,MAAM,aAAa,IAAI,WACrB,WAAW,YACX,UACA,mBACA,EACF;EASA,KAAK,MAAM,KAAK,SAAS,WAAW;GAElC,IAAI,EAAE,SAAS,YAAY,EAAE,SAAA,wBAAwC;IACnE,GAAG,QACD,iCAAiC,EAAE,KAAK,cAAc,EAAE,SAAS,EACnE;IACA,IAAI;KACF,MAAM,8BACJ,WAAW,YACX,GACA,mBACA,EACF;KAEA;IACF,SAAS,OAAO;KACd,GAAG,OACD,iDAAiD,EAAE,YACnD,KACF;KAEA;IACF;IACA;GACF;GAEA,OAAO,EAAE,SAAS,UAAU,0BAA0B;GACtD,GAAG,QACD,wBAAwB,EAAE,KAAK,QAAQ,EAAE,GAAG,aAAa,EAAE,SAAS,EACtE;GAEA,IAAI,gBAA+B;GAEnC,MAAM,gBAA+B,OAAM,YAAW;IACpD,gBAAgB;IAChB,MAAM,SAAS,MAAM,WAAW,SAAS,GAAG,OAAO;IACnD,gBAAgB;IAChB,OAAO;GACT;GAEA,IAAI;IACF,MAAM,MAAM,MAAM,8BAChB,WAAW,QAAQ,eAAe,CAAC,CACrC;IACA,UAAU,KAAK,GAAG;IAClB,GAAG,QAAQ,aAAa,EAAE,KAAK,QAAQ,EAAE,GAAG,yBAAyB;IAErE;GACF,SAAS,OAAO;IACd,IAAI,CAAC,mBAAmB,KAAK,GAC3B,MAAM;IAGR,IAAI,kBAAkB,kBAEpB,MAAM,WAAW,6BAA6B,GAAG,KAAK;SACjD,IAAI,kBAAkB,cAE3B,GAAG,QACD,oDAAoD,EAAE,GAAG,cAAc,EAAE,YACzE,KACF;IAGF,GAAG,OACD,yCAAyC,EAAE,GAAG,cAAc,EAAE,YAC9D,KACF;IACA,UAAU,KAAK,qBAAqB,GAAG,KAAK,CAAC;IAE7C;GACF;EACF;EAEA,OAAO;GACL,MAAM;GACN,WAAW;GACX,GAAI,OAAO,WAAW,WAAW,cAC7B,EAAC,QAAQ,WAAW,OAAM,IAC1B,CAAC;EACP;CACF,SAAS,OAAO;EACd,GAAG,QAAQ,kCAAkC,KAAK;EAElD,MAAM,yBAAyB,YAAY,MAAM,cAAc;EAE/D,MAAM,UAAU,gBAAgB,KAAK;EACrC,MAAM,UAAU,gBAAgB,KAAK;EAErC,OAAO;GACL,MAAMJ;GACN,QAAQC;GACR,QACE,iBAAiB,qBACbI,uBACA,iBAAiB,2BACfC,WACAC;GACR;GACA,aAAa;GACb,GAAI,UAAU,EAAC,QAAO,IAAI,CAAC;EAC7B;CACF;AACF;AAEA,SAAS,4BAEP,OAAmE;CACnE,IAAI,aAAa,OACf,OAAO;EACL,MAAM;EACN,YAAY,MAAM;EAClB,SAAS,MAAM;EACf,SAAS,MAAM;EACf,QAAQ,MAAM,UAAU;EACxB,UAAU,MAAM,YAAY;EAC5B,SAAS,MAAM,WAAW;CAC5B;CAGF,OAAO;EACL,MAAM;EACN,YAAY,MAAM;EAClB,SAAS,MAAM;EACf,UAAU,MAAM;EAChB,QAAQ,MAAM,UAAU;EACxB,aACE,MAAM,iBAAiB,kBACnB,OAAO,YAAY,MAAM,KAAK,IAC9B,MAAM;EACZ,UAAU,MAAM,YAAY;EAC5B,SAAS,MAAM,WAAW;CAC5B;AACF;AAEA,SAAS,iCAGP,YACA,SACA,gBACA,gBACA,mBACA,cACgC;CAChC,OAAO,OAAO,YAAY,YAAY,8BAA8B;CACpE,OACE,OAAO,mBAAmB,aAC1B,0CACF;CAEA,IAAI,0BAA0B,SAC5B,OAAO;EACL,MAAM;EACN;EACA;EACA,SAAS;EACT,QAAQ,KAAA;EACR,UAAW,kBAA2C;EAGtD,SAAU,qBAA6C;CACzD;CAGF,OACE,OAAO,mBAAmB,aAC1B,+BACF;CAEA,OAAO;EACL,MAAM;EACN;EACA;EACA,UAAU;EACV,QAAQ,KAAA;EACR,aACE,0BAA0B,kBACtB,OAAO,YAAY,cAAc,IACjC;EACN,UAAW,qBAA8C;EACzD,SAAS,gBAAgB;CAC3B;AACF;AASA,IAAM,aAAN,MAAgE;CAC9D;CACA;CACA;CACA;CAEA,YAAY,YAAe,KAAe,QAAgB,IAAgB;EACxE,KAAK,cAAc;EACnB,KAAK,OAAO;EACZ,KAAK,UAAU;EACf,KAAK,MAAM;CACb;;;;;;;;CASA,aACE,YACA,OACA,SACkB;EAClB,IAAI,WAAW,WAAW,GACxB,OAAO,QAAQ,QAAQ,KAAK;EAE9B,MAAM,MAAM,OAAO,gBAAkD;GACnE,IAAI;IACF,OAAQ,MAAM,YAAY,OAAO,OAAO,MAAO;GACjD,SAAS,YAAY;IACnB,KAAK,IAAI,OACP,8CACA,UACF;IACA,OAAO;GACT;EACF;EACA,OAAO,IAAI,SAAQ,YAAW;GAC5B,IAAI,UAAU,WAAW;GACzB,KAAK,MAAM,eAAe,YACxB,IAAS,WAAW,CAAC,CAAC,MAAK,QAAO;IAChC,IAAI,KACF,QAAQ,IAAI;SACP,IAAI,EAAE,YAAY,GACvB,QAAQ,KAAK;GAEjB,CAAC;EAEL,CAAC;CACH;CAEA,WAAW,OACT,UACA,OAC8B;EAC9B,IAAI,WAAyC,KAAA;EAG7C,IAAI,UAAU;EACd,SAAS;GAEP,MAAM,aAA+B,CAAC;GACtC,IAAI;IACF,MAAM,MAAM,MAAM,KAAK,cACrB,UACA,IACA,WACA,gBAAe;KACb,WAAW,KAAK,WAAW;IAC7B,CACF;IACA,IAAI,aAAa,KAAA,GAAW;KAC1B,KAAK,IAAI,OAEP,YAAY,SAAS,GAAG,cAAc,SAAS,SAAS,8BACxD,QACF;KACA,OAAO,qBAAqB,UAAU,QAAQ;IAChD;IAEA,OAAO;GACT,SAAS,OAAO;IACd,IAAI,iBAAiB,oBAAoB;KACvC,KAAK,IAAI,QAAQ,KAAK;KACtB,MAAM;IACR;IAEA,IAAI,iBAAiB,+BAA+B;KAClD,KAAK,IAAI,OAAO,KAAK;KACrB,OAAO;MACL,IAAI;OACF,UAAU,SAAS;OACnB,IAAI,SAAS;MACf;MACA,QAAQ;OACN,OAAO;OACP,SAAS,MAAM;MACjB;KACF;IACF;IAEA,IAAI,aAAa,KAAA,GAAW;KAE1B,KAAK,IAAI,QAEP,kCAAkC,SAAS,GAAG,cAAc,SAAS,YACrE,KACF;KACA,MAAM;IACR;IAIA,MAAM,gBACJ,iBAAiB,2BACZ,MAAM,SAAS,QAChB;IAUN,IACE,UAAA,KACC,MAAM,KAAK,aAAa,YAAY,eAAe,OAAO,GAC3D;KACA,KAAK,IAAI,OAEP,YAAY,SAAS,GAAG,cAAc,SAAS,SAAS,qCAAqC,QAAQ,IACrG,aACF;KACA;KACA;IACF;IAGA,WAAW,yBAAyB,aAAa;IACjD,KAAK,IAAI,OAEP,6BAA6B,SAAS,GAAG,cAAc,SAAS,SAAS,6BACzE,QACF;IACA;GACF;EACF;CACF;CAEA,MAAM,6BACJ,UACA,UAC2B;EAU3B,OAAO,MAPW,KAAK,cACrB,gBAEM,aACN,UACA,KAAA,CACF;CAEF;CAEA,MAAM,cACJ,UACA,IACA,UACA,mBAC2B;EAC3B,IAAI,mBAA6C;EAEjD,IAAI;GAsCF,OAAO,MArCW,KAAK,YAAY,YACjC,OAAO,MAAM,qBAAqB;IAEhC,mBAAmB;IAEnB,MAAM,KAAK,iCACT,kBACA,SAAS,UACT,SAAS,EACX;IAEA,IAAI,aAAa,KAAA,GAAW;KAC1B,KAAK,IAAI,QAEP,sBAAsB,SAAS,KAAK,QAAQ,SAAS,GAAG,EAC1D;KACA,MAAM,OAAQ,MAAM,GAAG,MAAM,SAAS,MAAM,SAAS,KAAK,EAAE;KAG5D,MAAM,KAAK;MAAC,UAAU,SAAS;MAAU,IAAI,SAAS;KAAE;KACxD,IAAI,SAAS,KAAA,GACX,OAAO;MAAC;MAAI,QAAQ,CAAC;KAAC;KAIxB,MAAM,gBAAkC;MAAC;MAAI,QAAQ,EAAC,KAAI;KAAC;KAC3D,MAAM,iBAAiB,oBAAoB,aAAa;KACxD,OAAO;IACT,OAAO;KACL,MAAM,iBAAiB,qBAAqB,UAAU,QAAQ;KAC9D,MAAM,iBAAiB,oBAAoB,cAAc;KACzD,OAAO;IACT;GACF,GACA,KAAK,qBAAqB,UAAU,iBAAiB,CACvD;EAGF,SAAS,OAAO;GACd,IACE,mBAAmB,KAAK,KACxB,iBAAiB,sBACjB,iBAAiB,+BAEjB,MAAM;GAGR,MAAM,IAAI,yBAAyB,kBAAkB,EAAC,OAAO,MAAK,CAAC;EACrE;CACF;CAEA,qBACE,UACA,mBAC0B;EAC1B,OAAO;GACL,gBAAgB,KAAK,QAAQ;GAC7B,eAAe,KAAK,KAAK;GACzB,UAAU,SAAS;GACnB,YAAY,SAAS;GACrB;EACF;CACF;CAEA,MAAM,iCACJ,kBACA,UACA,oBACA;EACA,MAAM,EAAC,mBAAkB,MAAM,iBAAiB,uBAAuB;EAEvE,IAAI,qBAAqB,gBACvB,MAAM,IAAI,8BACR,UACA,oBACA,cACF;OACK,IAAI,qBAAqB,gBAC9B,MAAM,IAAI,mBACR,UACA,oBACA,cACF;CAEJ;AACF;AAEA,IAAa,qBAAb,cAAwC,MAAM;CAC5C,YACE,UACA,oBACA,gBACA;EACA,MACE,UAAU,SAAS,oBAAoB,mBAAmB,gBAAgB,gBAC5E;CACF;AACF;AAEA,SAAS,qBACP,GACA,OACkB;CAClB,OAAO;EACL,IAAI;GACF,UAAU,EAAE;GACZ,IAAI,EAAE;EACR;EACA,QAAQ;GACN,OAAO;GACP,SAAS,MAAM;GACf,GAAI,MAAM,UAAU,EAAC,SAAS,MAAM,QAAO,IAAI,CAAC;EAClD;CACF;AACF;;AAGA,SAAgB,YAEd,UACA,MAEwB;CAExB,MAAM,UAAU,gBAAgB,UADnB,KAAK,MAAM,WACkB,CAAI;CAC9C,OAAO,OAAO,YAAY,YAAY,0BAA0B,MAAM;CAEtE,IAAI,UAAU,OAAO,GAGnB,QAAQ,IAAI,MAAM,QAAQ,QAAQ,GAAG;EAAC;EAAM;EAAK;CAAE,CAAC;CAItD,OAAO;AACT;AAEA,SAAS,gBACP,KACA,MACS;CACT,IAAI,UAAmB;CACvB,KAAK,MAAM,QAAQ,MAAM;EACvB,IAAI,OAAO,YAAY,YAAY,YAAY,QAAQ,EAAE,QAAQ,UAC/D;EAEF,UAAW,QAAoC;CACjD;CACA,OAAO;AACT;;;;;;AAOA,eAAe,8BAGb,YACA,UACA,aACA,IACe;CACf,MAAM,cAAcC,KAAO,SAAS,KAAK,IAAI,uBAAuB;CACpE,IAAI,CAAC,YAAY,IAAI;EAEnB,GAAG,OAAO,qCAAqC,YAAY,KAAK;EAChE;CACF;CACA,MAAM,OAA0B,YAAY;CAI5C,MAAM,WACJ,UAAU,QAAQ,KAAK,SAAS,SAAS,KAAK,UAAU,KAAK,KAAK;CAKpE,MAAM,WAAW,YACf,OAAO,GAAG,UAAU;EAClB,MAAM,MAAM,sBAAsB,IAAI;CACxC,GACA;EACE,gBAAgB,YAAY;EAC5B,eAAe,KAAK;EACpB;EACA,YAAY;CACd,CACF;AACF;AAGA,IAAM,2BAAN,cAAuC,MAAM;CAC3C,YAAY,OAAiC,SAAwB;EACnE,MACE,UAAU,SACN,wCAAwC,gBAAgB,SAAS,KAAK,MACtE,8CAA8C,gBAAgB,SAAS,KAAK,KAChF,OACF;EACA,KAAK,OAAO;CACd;AACF"}

package/out/zero-server/src/push-processor.js.map

@@ -1 +1 @@
-{"version":3,"file":"push-processor.js","names":["#dbProvider","#logLevel","#logSink","#context","#processMutation","#dispatchMutation"],"sources":["../../../../zero-server/src/push-processor.ts"],"sourcesContent":["import {consoleLogSink, type LogLevel, type LogSink} from '@rocicorp/logger';\nimport {assert} from '../../shared/src/asserts.ts';\nimport type {ReadonlyJSONValue} from '../../shared/src/json.ts';\nimport {must} from '../../shared/src/must.ts';\nimport {getValueAtPath} from '../../shared/src/object-traversal.ts';\nimport type {MutateResponse} from '../../zero-protocol/src/mutate-server.ts';\nimport type {CustomMutation} from '../../zero-protocol/src/mutation.ts';\nimport {type MutationResponse} from '../../zero-protocol/src/push.ts';\nimport {\n  type Database,\n  type ExtractTransactionType,\n  handleMutateRequest,\n  type TransactFn,\n} from '../../zero-server/src/process-mutations.ts';\nimport type {Schema} from '../../zero-types/src/schema.ts';\nimport type {Transaction} from '../../zql/src/mutate/custom.ts';\nimport type {AnyMutatorRegistry} from '../../zql/src/mutate/mutator-registry.ts';\nimport {isMutator} from '../../zql/src/mutate/mutator.ts';\nimport type {CustomMutatorDefs} from './custom.ts';\n\nexport const separatorRe = /[.|]/;\n\nexport class PushProcessor<\n  _S extends Schema,\n  D extends Database<ExtractTransactionType<D>>,\n  MD extends AnyMutatorRegistry | CustomMutatorDefs<ExtractTransactionType<D>>,\n  C = undefined,\n> {\n  readonly #dbProvider: D;\n  readonly #logLevel: LogLevel;\n  readonly #logSink: LogSink;\n  readonly #context: C;\n\n  /**\n   * @param logSink Destination for this processor's log output. When omitted,\n   *   logs are written with `console.log`/`info`/`warn`/`error` etc. Provide a\n   *   custom {@link LogSink} to redirect them into the application's own\n   *   logging pipeline.\n   */\n  constructor(\n    dbProvider: D,\n    context?: C,\n    logLevel: LogLevel = 'info',\n    logSink: LogSink = consoleLogSink,\n  ) {\n    this.#dbProvider = dbProvider;\n    this.#context = context as C;\n    this.#logLevel = logLevel;\n    this.#logSink = logSink;\n  }\n\n  /**\n   * Processes a push request from zero-cache.\n   * This function will parse the request, check the protocol version, and process each mutation in the request.\n   * - If a mutation is out of order: processing will stop and an error will be returned. The zero client will retry the mutation.\n   * - If a mutation has already been processed: it will be skipped and the processing will continue.\n   * - If a mutation receives an application error: it will be skipped, the error will be returned to the client, and processing will continue.\n   *\n   * @param mutators the custom mutators for the application\n   * @param queryString the query string from the request sent by zero-cache. This will include zero's postgres schema name and appID.\n   * @param body the body of the request sent by zero-cache as a JSON object.\n   */\n  process(\n    mutators: MD,\n    queryString: URLSearchParams | Record<string, string>,\n    body: ReadonlyJSONValue,\n  ): Promise<MutateResponse>;\n\n  /**\n   * This override gets the query string and the body from a Request object.\n   *\n   * @param mutators the custom mutators for the application\n   * @param request A `Request` object.\n   */\n  process(mutators: MD, request: Request): Promise<MutateResponse>;\n  process(\n    mutators: MD,\n    queryOrQueryString: Request | URLSearchParams | Record<string, string>,\n    body?: ReadonlyJSONValue,\n  ): Promise<MutateResponse> {\n    if (queryOrQueryString instanceof Request) {\n      return handleMutateRequest(\n        this.#dbProvider,\n        (transact, mutation) =>\n          this.#processMutation(mutators, transact, mutation),\n        queryOrQueryString,\n        this.#logLevel,\n        this.#logSink,\n      );\n    }\n    return handleMutateRequest(\n      this.#dbProvider,\n      (transact, mutation) =>\n        this.#processMutation(mutators, transact, mutation),\n      queryOrQueryString,\n      must(body, 'body is required when using query params directly'),\n      this.#logLevel,\n      this.#logSink,\n    );\n  }\n\n  #processMutation(\n    mutators: MD,\n    transact: TransactFn<D>,\n    _mutation: CustomMutation,\n  ): Promise<MutationResponse> {\n    return transact((tx, name, args) =>\n      this.#dispatchMutation(mutators, tx, name, args),\n    );\n  }\n\n  #dispatchMutation(\n    mutators: MD,\n    dbTx: ExtractTransactionType<D>,\n    key: string,\n    args: ReadonlyJSONValue | undefined,\n  ): Promise<ReadonlyJSONValue | void> {\n    // Legacy mutators used | as a separator, new mutators use .\n    const mutator = getValueAtPath(mutators, key, separatorRe);\n    assert(typeof mutator === 'function', `could not find mutator ${key}`);\n    if (isMutator(mutator)) {\n      return mutator.fn({\n        args,\n        ctx: this.#context,\n        tx: dbTx as Transaction<Schema, unknown>,\n      });\n    }\n    return mutator(dbTx, args);\n  }\n}\n"],"mappings":";;;;;;;;AAoBA,IAAa,cAAc;AAE3B,IAAa,gBAAb,MAKE;CACA;CACA;CACA;CACA;;;;;;;CAQA,YACE,YACA,SACA,WAAqB,QACrB,UAAmB,gBACnB;EACA,KAAKA,cAAc;EACnB,KAAKG,WAAW;EAChB,KAAKF,YAAY;EACjB,KAAKC,WAAW;CAClB;CA0BA,QACE,UACA,oBACA,MACyB;EACzB,IAAI,8BAA8B,SAChC,OAAO,oBACL,KAAKF,cACJ,UAAU,aACT,KAAKI,iBAAiB,UAAU,UAAU,QAAQ,GACpD,oBACA,KAAKH,WACL,KAAKC,QACP;EAEF,OAAO,oBACL,KAAKF,cACJ,UAAU,aACT,KAAKI,iBAAiB,UAAU,UAAU,QAAQ,GACpD,oBACA,KAAK,MAAM,mDAAmD,GAC9D,KAAKH,WACL,KAAKC,QACP;CACF;CAEA,iBACE,UACA,UACA,WAC2B;EAC3B,OAAO,UAAU,IAAI,MAAM,SACzB,KAAKG,kBAAkB,UAAU,IAAI,MAAM,IAAI,CACjD;CACF;CAEA,kBACE,UACA,MACA,KACA,MACmC;EAEnC,MAAM,UAAU,eAAe,UAAU,KAAK,WAAW;EACzD,OAAO,OAAO,YAAY,YAAY,0BAA0B,KAAK;EACrE,IAAI,UAAU,OAAO,GACnB,OAAO,QAAQ,GAAG;GAChB;GACA,KAAK,KAAKF;GACV,IAAI;EACN,CAAC;EAEH,OAAO,QAAQ,MAAM,IAAI;CAC3B;AACF"}
+{"version":3,"file":"push-processor.js","names":[],"sources":["../../../../zero-server/src/push-processor.ts"],"sourcesContent":["import {consoleLogSink, type LogLevel, type LogSink} from '@rocicorp/logger';\nimport {assert} from '../../shared/src/asserts.ts';\nimport type {ReadonlyJSONValue} from '../../shared/src/json.ts';\nimport {must} from '../../shared/src/must.ts';\nimport {getValueAtPath} from '../../shared/src/object-traversal.ts';\nimport type {MutateResponse} from '../../zero-protocol/src/mutate-server.ts';\nimport type {CustomMutation} from '../../zero-protocol/src/mutation.ts';\nimport {type MutationResponse} from '../../zero-protocol/src/push.ts';\nimport {\n  type Database,\n  type ExtractTransactionType,\n  handleMutateRequest,\n  type TransactFn,\n} from '../../zero-server/src/process-mutations.ts';\nimport type {Schema} from '../../zero-types/src/schema.ts';\nimport type {Transaction} from '../../zql/src/mutate/custom.ts';\nimport type {AnyMutatorRegistry} from '../../zql/src/mutate/mutator-registry.ts';\nimport {isMutator} from '../../zql/src/mutate/mutator.ts';\nimport type {CustomMutatorDefs} from './custom.ts';\n\nexport const separatorRe = /[.|]/;\n\nexport class PushProcessor<\n  _S extends Schema,\n  D extends Database<ExtractTransactionType<D>>,\n  MD extends AnyMutatorRegistry | CustomMutatorDefs<ExtractTransactionType<D>>,\n  C = undefined,\n> {\n  readonly #dbProvider: D;\n  readonly #logLevel: LogLevel;\n  readonly #logSink: LogSink;\n  readonly #context: C;\n\n  /**\n   * @param logSink Destination for this processor's log output. When omitted,\n   *   logs are written with `console.log`/`info`/`warn`/`error` etc. Provide a\n   *   custom {@link LogSink} to redirect them into the application's own\n   *   logging pipeline.\n   */\n  constructor(\n    dbProvider: D,\n    context?: C,\n    logLevel: LogLevel = 'info',\n    logSink: LogSink = consoleLogSink,\n  ) {\n    this.#dbProvider = dbProvider;\n    this.#context = context as C;\n    this.#logLevel = logLevel;\n    this.#logSink = logSink;\n  }\n\n  /**\n   * Processes a push request from zero-cache.\n   * This function will parse the request, check the protocol version, and process each mutation in the request.\n   * - If a mutation is out of order: processing will stop and an error will be returned. The zero client will retry the mutation.\n   * - If a mutation has already been processed: it will be skipped and the processing will continue.\n   * - If a mutation receives an application error: it will be skipped, the error will be returned to the client, and processing will continue.\n   *\n   * @param mutators the custom mutators for the application\n   * @param queryString the query string from the request sent by zero-cache. This will include zero's postgres schema name and appID.\n   * @param body the body of the request sent by zero-cache as a JSON object.\n   */\n  process(\n    mutators: MD,\n    queryString: URLSearchParams | Record<string, string>,\n    body: ReadonlyJSONValue,\n  ): Promise<MutateResponse>;\n\n  /**\n   * This override gets the query string and the body from a Request object.\n   *\n   * @param mutators the custom mutators for the application\n   * @param request A `Request` object.\n   */\n  process(mutators: MD, request: Request): Promise<MutateResponse>;\n  process(\n    mutators: MD,\n    queryOrQueryString: Request | URLSearchParams | Record<string, string>,\n    body?: ReadonlyJSONValue,\n  ): Promise<MutateResponse> {\n    if (queryOrQueryString instanceof Request) {\n      return handleMutateRequest(\n        this.#dbProvider,\n        (transact, mutation) =>\n          this.#processMutation(mutators, transact, mutation),\n        queryOrQueryString,\n        this.#logLevel,\n        this.#logSink,\n      );\n    }\n    return handleMutateRequest(\n      this.#dbProvider,\n      (transact, mutation) =>\n        this.#processMutation(mutators, transact, mutation),\n      queryOrQueryString,\n      must(body, 'body is required when using query params directly'),\n      this.#logLevel,\n      this.#logSink,\n    );\n  }\n\n  #processMutation(\n    mutators: MD,\n    transact: TransactFn<D>,\n    _mutation: CustomMutation,\n  ): Promise<MutationResponse> {\n    return transact((tx, name, args) =>\n      this.#dispatchMutation(mutators, tx, name, args),\n    );\n  }\n\n  #dispatchMutation(\n    mutators: MD,\n    dbTx: ExtractTransactionType<D>,\n    key: string,\n    args: ReadonlyJSONValue | undefined,\n  ): Promise<ReadonlyJSONValue | void> {\n    // Legacy mutators used | as a separator, new mutators use .\n    const mutator = getValueAtPath(mutators, key, separatorRe);\n    assert(typeof mutator === 'function', `could not find mutator ${key}`);\n    if (isMutator(mutator)) {\n      return mutator.fn({\n        args,\n        ctx: this.#context,\n        tx: dbTx as Transaction<Schema, unknown>,\n      });\n    }\n    return mutator(dbTx, args);\n  }\n}\n"],"mappings":";;;;;;;;AAoBA,IAAa,cAAc;AAE3B,IAAa,gBAAb,MAKE;CACA;CACA;CACA;CACA;;;;;;;CAQA,YACE,YACA,SACA,WAAqB,QACrB,UAAmB,gBACnB;EACA,KAAK,cAAc;EACnB,KAAK,WAAW;EAChB,KAAK,YAAY;EACjB,KAAK,WAAW;CAClB;CA0BA,QACE,UACA,oBACA,MACyB;EACzB,IAAI,8BAA8B,SAChC,OAAO,oBACL,KAAK,cACJ,UAAU,aACT,KAAK,iBAAiB,UAAU,UAAU,QAAQ,GACpD,oBACA,KAAK,WACL,KAAK,QACP;EAEF,OAAO,oBACL,KAAK,cACJ,UAAU,aACT,KAAK,iBAAiB,UAAU,UAAU,QAAQ,GACpD,oBACA,KAAK,MAAM,mDAAmD,GAC9D,KAAK,WACL,KAAK,QACP;CACF;CAEA,iBACE,UACA,UACA,WAC2B;EAC3B,OAAO,UAAU,IAAI,MAAM,SACzB,KAAK,kBAAkB,UAAU,IAAI,MAAM,IAAI,CACjD;CACF;CAEA,kBACE,UACA,MACA,KACA,MACmC;EAEnC,MAAM,UAAU,eAAe,UAAU,KAAK,WAAW;EACzD,OAAO,OAAO,YAAY,YAAY,0BAA0B,KAAK;EACrE,IAAI,UAAU,OAAO,GACnB,OAAO,QAAQ,GAAG;GAChB;GACA,KAAK,KAAK;GACV,IAAI;EACN,CAAC;EAEH,OAAO,QAAQ,MAAM,IAAI;CAC3B;AACF"}

package/out/zero-server/src/queries/process-queries.js

@@ -105,7 +105,8 @@ async function transform(args, apiName) {
 					};
 				}
 				try {
-					const ast = mapAST(asQueryInternals(finalQuery).ast, nameMapper);
+					const q = asQueryInternals(finalQuery);
+					const ast = mapAST(q.ast, nameMapper);
 					return {
 						id: req.id,
 						name: req.name,

package/out/zero-server/src/queries/process-queries.js.map

@@ -1 +1 @@
-{"version":3,"file":"process-queries.js","names":[],"sources":["../../../../../zero-server/src/queries/process-queries.ts"],"sourcesContent":["import type {LogLevel} from '@rocicorp/logger';\nimport {assert} from '../../../shared/src/asserts.ts';\nimport {getErrorDetails, getErrorMessage} from '../../../shared/src/error.ts';\nimport type {ReadonlyJSONValue} from '../../../shared/src/json.ts';\nimport type {MaybePromise} from '../../../shared/src/types.ts';\nimport * as v from '../../../shared/src/valita.ts';\nimport {mapAST} from '../../../zero-protocol/src/ast.ts';\nimport {\n  transformRequestMessageSchema,\n  type TransformRequestMessage,\n  type TransformResponseBody,\n} from '../../../zero-protocol/src/custom-queries.ts';\nimport {ErrorKind} from '../../../zero-protocol/src/error-kind.ts';\nimport {ErrorOrigin} from '../../../zero-protocol/src/error-origin.ts';\nimport {ErrorReason} from '../../../zero-protocol/src/error-reason.ts';\nimport type {QueryResponse} from '../../../zero-protocol/src/query-server.ts';\nimport {clientToServer} from '../../../zero-schema/src/name-mapper.ts';\nimport type {Schema} from '../../../zero-types/src/schema.ts';\nimport {QueryParseError} from '../../../zql/src/query/error.ts';\nimport {asQueryInternals} from '../../../zql/src/query/query-internals.ts';\nimport type {AnyQuery} from '../../../zql/src/query/query.ts';\nimport {createLogContext} from '../logging.ts';\n\n/**\n * Invokes the callback `cb` for each query in the request or JSON body.\n * The callback should return a Query or Promise<Query> that is the transformed result.\n *\n * This function will call `cb` in parallel for each query found in the request.\n *\n * If you need to limit concurrency, you can use a library like `p-limit` to wrap the `cb` function.\n * @deprecated Use {@link handleQueryRequest} instead.\n */\nexport function handleGetQueriesRequest<S extends Schema>(\n  cb: (\n    name: string,\n    args: readonly ReadonlyJSONValue[],\n  ) => MaybePromise<{query: AnyQuery} | AnyQuery>,\n  schema: S,\n  requestOrJsonBody: Request | ReadonlyJSONValue,\n  logLevel: LogLevel = 'info',\n): Promise<QueryResponse> {\n  return transform(\n    normalizeLegacyQueryRequestArgs(cb, schema, requestOrJsonBody, logLevel),\n    'getQueries',\n  );\n}\n\n/**\n * Invokes the callback `cb` for each query in the request or JSON body.\n * The callback should return a Query or Promise<Query> that is the transformed result.\n *\n * This function will call `cb` in parallel for each query found in the request.\n *\n * If you need to limit concurrency, you can use a library like `p-limit` to wrap the `cb` function.\n * @deprecated Use {@link handleQueryRequest} instead.\n */\nexport function handleTransformRequest<S extends Schema>(\n  cb: (\n    name: string,\n    args: readonly ReadonlyJSONValue[],\n  ) => MaybePromise<{query: AnyQuery} | AnyQuery>,\n  schema: S,\n  requestOrJsonBody: Request | ReadonlyJSONValue,\n  logLevel: LogLevel = 'info',\n): Promise<QueryResponse> {\n  return transform(\n    normalizeLegacyQueryRequestArgs(cb, schema, requestOrJsonBody, logLevel),\n    'transform',\n  );\n}\n\n/**\n * Parsed query params accepted by {@link handleQueryRequest} when the\n * incoming request URL has already been handled by your framework.\n */\nexport type QuerySearchParams = URLSearchParams | Record<string, string>;\n\nexport type HandleQueryRequestArgs<S extends Schema> = {\n  /** Callback that transforms each requested query into a `Query`. */\n  handler: QueryRequestHandler;\n  /** Schema used when building the returned ASTs. */\n  schema: S;\n  /**\n   * Authenticated user ID. Null or undefined means the user is logged out.\n   */\n  userID: string | null | undefined;\n  /** Optional log level for request parsing and execution. */\n  logLevel?: LogLevel | undefined;\n} & (\n  | {\n      /** Fetch request containing the `/query` JSON body. */\n      request: Request;\n    }\n  | {\n      /** Parsed query params from the `/query` request URL. */\n      query: QuerySearchParams;\n      /** Parsed JSON body from the `/query` request. */\n      body: ReadonlyJSONValue;\n    }\n);\n\ntype NormalizedQueryRequestArgs<S extends Schema> = {\n  readonly schema: S;\n  readonly handler: LegacyQueryRequestHandler;\n  // Note: semantics of undefined differ from HandleQueryRequestArgs.userID.\n  // Here, undefined means the app didn't provide a user ID - we do not know if\n  // the user is logged in or not. This is legacy behavior needed to support\n  // deprecated signatures of handleQueryRequest which did not receive userID\n  // from app.\n  readonly userID: string | null | undefined;\n  readonly logLevel: LogLevel;\n} & (\n  | {\n      readonly type: 'request';\n      readonly request: Request;\n    }\n  | {\n      readonly type: 'body';\n      readonly jsonBody: ReadonlyJSONValue;\n    }\n);\n\n/**\n * Process a `/query` request from a Fetch `Request`.\n */\nexport function handleQueryRequest<S extends Schema>(\n  input: HandleQueryRequestArgs<S>,\n): Promise<QueryResponse>;\n\n/**\n * @deprecated Pass a single object instead:\n * `handleQueryRequest({handler, schema, request, userID, logLevel})`.\n */\nexport function handleQueryRequest<S extends Schema>(\n  transformQuery: QueryRequestHandler,\n  schema: S,\n  request: Request,\n  logLevel?: LogLevel,\n): Promise<QueryResponse>;\n\n/**\n * @deprecated Pass a single object instead:\n * `handleQueryRequest({handler, schema, body, userID, logLevel})`.\n */\nexport function handleQueryRequest<S extends Schema>(\n  transformQuery: QueryRequestHandler,\n  schema: S,\n  jsonBody: ReadonlyJSONValue,\n  logLevel?: LogLevel,\n): Promise<QueryResponse>;\n\nexport function handleQueryRequest<S extends Schema>(\n  inputOrTransformQuery: HandleQueryRequestArgs<S> | QueryRequestHandler,\n  schema?: S,\n  requestOrJsonBody?: Request | ReadonlyJSONValue,\n  logLevel?: LogLevel,\n): Promise<QueryResponse> {\n  const normalized =\n    typeof inputOrTransformQuery === 'object' &&\n    'handler' in inputOrTransformQuery\n      ? normalizeQueryRequestInput(inputOrTransformQuery)\n      : normalizeLegacyQueryRequestArgs(\n          wrapQueryRequestHandler(inputOrTransformQuery),\n          schema,\n          requestOrJsonBody,\n          logLevel,\n        );\n\n  return transform(normalized, 'query');\n}\n\nfunction normalizeQueryRequestInput<S extends Schema>(\n  input: HandleQueryRequestArgs<S>,\n): NormalizedQueryRequestArgs<S> {\n  return 'request' in input\n    ? normalizeLegacyQueryRequestArgs(\n        wrapQueryRequestHandler(input.handler),\n        input.schema,\n        input.request,\n        input.logLevel,\n        input.userID ?? null,\n      )\n    : normalizeLegacyQueryRequestArgs(\n        wrapQueryRequestHandler(input.handler),\n        input.schema,\n        input.body,\n        input.logLevel,\n        input.userID ?? null,\n      );\n}\n\nfunction normalizeLegacyQueryRequestArgs<S extends Schema>(\n  handler: LegacyQueryRequestHandler,\n  schema: S | undefined,\n  requestOrJsonBody: Request | ReadonlyJSONValue | undefined,\n  logLevel: LogLevel | undefined,\n  userID?: string | null,\n): NormalizedQueryRequestArgs<S> {\n  assert(\n    typeof schema !== 'undefined',\n    'Schema must be provided when using handleQueryRequest',\n  );\n\n  if (requestOrJsonBody instanceof Request) {\n    return {\n      type: 'request',\n      handler,\n      schema,\n      request: requestOrJsonBody,\n      userID,\n      logLevel: logLevel ?? 'info',\n    };\n  }\n\n  assert(\n    typeof requestOrJsonBody !== 'undefined',\n    'JSON body cannot be undefined',\n  );\n\n  return {\n    type: 'body',\n    handler,\n    schema,\n    jsonBody: requestOrJsonBody,\n    userID,\n    logLevel: logLevel ?? 'info',\n  };\n}\n\nasync function transform<S extends Schema>(\n  args: NormalizedQueryRequestArgs<S>,\n  apiName: 'query' | 'transform' | 'getQueries',\n): Promise<QueryResponse> {\n  const lc = createLogContext(args.logLevel).withContext('TransformRequest');\n  let parsed: TransformRequestMessage;\n  let queryIDs: string[] = [];\n  try {\n    let body: ReadonlyJSONValue;\n    if (args.type === 'request') {\n      body = await args.request.json();\n    } else {\n      body = args.jsonBody;\n    }\n\n    // Strip, so a newer zero-cache can add fields to the request.\n    parsed = v.parse(body, transformRequestMessageSchema, 'strip');\n\n    queryIDs = parsed[1].map(r => r.id);\n  } catch (error) {\n    lc.error?.(`Failed to parse ${apiName} request`, error);\n\n    const message = `Failed to parse ${apiName} request: ${getErrorMessage(error)}`;\n    const details = getErrorDetails(error);\n\n    return {\n      kind: ErrorKind.TransformFailed,\n      origin: ErrorOrigin.Server,\n      reason: ErrorReason.Parse,\n      message,\n      queryIDs,\n      ...(details ? {details} : {}),\n    };\n  }\n\n  try {\n    const nameMapper = clientToServer(args.schema.tables);\n\n    const responses: TransformResponseBody = await Promise.all(\n      parsed[1].map(async req => {\n        let finalQuery: AnyQuery;\n        try {\n          const result = await args.handler(req.name, req.args);\n          finalQuery = 'query' in result ? result.query : result;\n        } catch (error) {\n          const message = getErrorMessage(error);\n          const details = getErrorDetails(error);\n\n          return {\n            error: error instanceof QueryParseError ? 'parse' : 'app',\n            id: req.id,\n            name: req.name,\n            message,\n            ...(details ? {details} : {}),\n          };\n        }\n\n        try {\n          const q = asQueryInternals(finalQuery);\n          const ast = mapAST(q.ast, nameMapper);\n\n          return {\n            id: req.id,\n            name: req.name,\n            ast,\n          };\n        } catch (error) {\n          lc.error?.('Failed to map AST', error);\n          throw error;\n        }\n      }),\n    );\n\n    return {\n      kind: 'QueryResponse',\n      queries: responses,\n      ...(typeof args.userID !== 'undefined' ? {userID: args.userID} : {}),\n    } as const satisfies QueryResponse;\n  } catch (e) {\n    const message = getErrorMessage(e);\n    const details = getErrorDetails(e);\n\n    return {\n      kind: ErrorKind.TransformFailed,\n      origin: ErrorOrigin.Server,\n      reason: ErrorReason.Internal,\n      message,\n      queryIDs,\n      ...(details ? {details} : {}),\n    };\n  }\n}\n\n/**\n * A function that transforms a query by name and arguments into a Query object.\n *\n * @param name - The name of the query (can be dot-separated for nested queries)\n * @param args - The arguments to pass to the query (can be undefined)\n * @returns A Query object\n */\nexport type QueryRequestHandler = (\n  name: string,\n  args: ReadonlyJSONValue | undefined,\n) => AnyQuery;\n\n/** @deprecated Use `QueryRequestHandler` instead. */\nexport type TransformQueryFunction = QueryRequestHandler;\n\nexport type LegacyQueryRequestHandler = (\n  name: string,\n  args: readonly ReadonlyJSONValue[],\n) => MaybePromise<{query: AnyQuery} | AnyQuery>;\n\nfunction wrapQueryRequestHandler(\n  handler: QueryRequestHandler,\n): LegacyQueryRequestHandler {\n  return (name, args) => handler(name, args[0]);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AAgCA,SAAgB,wBACd,IAIA,QACA,mBACA,WAAqB,QACG;CACxB,OAAO,UACL,gCAAgC,IAAI,QAAQ,mBAAmB,QAAQ,GACvE,YACF;AACF;;;;;;;;;;AAWA,SAAgB,uBACd,IAIA,QACA,mBACA,WAAqB,QACG;CACxB,OAAO,UACL,gCAAgC,IAAI,QAAQ,mBAAmB,QAAQ,GACvE,WACF;AACF;AAkFA,SAAgB,mBACd,uBACA,QACA,mBACA,UACwB;CAYxB,OAAO,UAVL,OAAO,0BAA0B,YACjC,aAAa,wBACT,2BAA2B,qBAAqB,IAChD,gCACE,wBAAwB,qBAAqB,GAC7C,QACA,mBACA,QACF,GAEuB,OAAO;AACtC;AAEA,SAAS,2BACP,OAC+B;CAC/B,OAAO,aAAa,QAChB,gCACE,wBAAwB,MAAM,OAAO,GACrC,MAAM,QACN,MAAM,SACN,MAAM,UACN,MAAM,UAAU,IAClB,IACA,gCACE,wBAAwB,MAAM,OAAO,GACrC,MAAM,QACN,MAAM,MACN,MAAM,UACN,MAAM,UAAU,IAClB;AACN;AAEA,SAAS,gCACP,SACA,QACA,mBACA,UACA,QAC+B;CAC/B,OACE,OAAO,WAAW,aAClB,uDACF;CAEA,IAAI,6BAA6B,SAC/B,OAAO;EACL,MAAM;EACN;EACA;EACA,SAAS;EACT;EACA,UAAU,YAAY;CACxB;CAGF,OACE,OAAO,sBAAsB,aAC7B,+BACF;CAEA,OAAO;EACL,MAAM;EACN;EACA;EACA,UAAU;EACV;EACA,UAAU,YAAY;CACxB;AACF;AAEA,eAAe,UACb,MACA,SACwB;CACxB,MAAM,KAAK,iBAAiB,KAAK,QAAQ,EAAE,YAAY,kBAAkB;CACzE,IAAI;CACJ,IAAI,WAAqB,CAAC;CAC1B,IAAI;EACF,IAAI;EACJ,IAAI,KAAK,SAAS,WAChB,OAAO,MAAM,KAAK,QAAQ,KAAK;OAE/B,OAAO,KAAK;EAId,SAAS,MAAQ,MAAM,+BAA+B,OAAO;EAE7D,WAAW,OAAO,GAAG,KAAI,MAAK,EAAE,EAAE;CACpC,SAAS,OAAO;EACd,GAAG,QAAQ,mBAAmB,QAAQ,WAAW,KAAK;EAEtD,MAAM,UAAU,mBAAmB,QAAQ,YAAY,gBAAgB,KAAK;EAC5E,MAAM,UAAU,gBAAgB,KAAK;EAErC,OAAO;GACL,MAAM;GACN,QAAQ;GACR,QAAQ;GACR;GACA;GACA,GAAI,UAAU,EAAC,QAAO,IAAI,CAAC;EAC7B;CACF;CAEA,IAAI;EACF,MAAM,aAAa,eAAe,KAAK,OAAO,MAAM;EAqCpD,OAAO;GACL,MAAM;GACN,SAAS,MArCoC,QAAQ,IACrD,OAAO,GAAG,IAAI,OAAM,QAAO;IACzB,IAAI;IACJ,IAAI;KACF,MAAM,SAAS,MAAM,KAAK,QAAQ,IAAI,MAAM,IAAI,IAAI;KACpD,aAAa,WAAW,SAAS,OAAO,QAAQ;IAClD,SAAS,OAAO;KACd,MAAM,UAAU,gBAAgB,KAAK;KACrC,MAAM,UAAU,gBAAgB,KAAK;KAErC,OAAO;MACL,OAAO,iBAAiB,kBAAkB,UAAU;MACpD,IAAI,IAAI;MACR,MAAM,IAAI;MACV;MACA,GAAI,UAAU,EAAC,QAAO,IAAI,CAAC;KAC7B;IACF;IAEA,IAAI;KAEF,MAAM,MAAM,OADF,iBAAiB,UACR,EAAE,KAAK,UAAU;KAEpC,OAAO;MACL,IAAI,IAAI;MACR,MAAM,IAAI;MACV;KACF;IACF,SAAS,OAAO;KACd,GAAG,QAAQ,qBAAqB,KAAK;KACrC,MAAM;IACR;GACF,CAAC,CACH;GAKE,GAAI,OAAO,KAAK,WAAW,cAAc,EAAC,QAAQ,KAAK,OAAM,IAAI,CAAC;EACpE;CACF,SAAS,GAAG;EACV,MAAM,UAAU,gBAAgB,CAAC;EACjC,MAAM,UAAU,gBAAgB,CAAC;EAEjC,OAAO;GACL,MAAM;GACN,QAAQ;GACR,QAAQ;GACR;GACA;GACA,GAAI,UAAU,EAAC,QAAO,IAAI,CAAC;EAC7B;CACF;AACF;AAsBA,SAAS,wBACP,SAC2B;CAC3B,QAAQ,MAAM,SAAS,QAAQ,MAAM,KAAK,EAAE;AAC9C"}
+{"version":3,"file":"process-queries.js","names":["v.parse","ErrorKind.TransformFailed","ErrorOrigin.Server","ErrorReason.Parse","ErrorReason.Internal"],"sources":["../../../../../zero-server/src/queries/process-queries.ts"],"sourcesContent":["import type {LogLevel} from '@rocicorp/logger';\nimport {assert} from '../../../shared/src/asserts.ts';\nimport {getErrorDetails, getErrorMessage} from '../../../shared/src/error.ts';\nimport type {ReadonlyJSONValue} from '../../../shared/src/json.ts';\nimport type {MaybePromise} from '../../../shared/src/types.ts';\nimport * as v from '../../../shared/src/valita.ts';\nimport {mapAST} from '../../../zero-protocol/src/ast.ts';\nimport {\n  transformRequestMessageSchema,\n  type TransformRequestMessage,\n  type TransformResponseBody,\n} from '../../../zero-protocol/src/custom-queries.ts';\nimport {ErrorKind} from '../../../zero-protocol/src/error-kind.ts';\nimport {ErrorOrigin} from '../../../zero-protocol/src/error-origin.ts';\nimport {ErrorReason} from '../../../zero-protocol/src/error-reason.ts';\nimport type {QueryResponse} from '../../../zero-protocol/src/query-server.ts';\nimport {clientToServer} from '../../../zero-schema/src/name-mapper.ts';\nimport type {Schema} from '../../../zero-types/src/schema.ts';\nimport {QueryParseError} from '../../../zql/src/query/error.ts';\nimport {asQueryInternals} from '../../../zql/src/query/query-internals.ts';\nimport type {AnyQuery} from '../../../zql/src/query/query.ts';\nimport {createLogContext} from '../logging.ts';\n\n/**\n * Invokes the callback `cb` for each query in the request or JSON body.\n * The callback should return a Query or Promise<Query> that is the transformed result.\n *\n * This function will call `cb` in parallel for each query found in the request.\n *\n * If you need to limit concurrency, you can use a library like `p-limit` to wrap the `cb` function.\n * @deprecated Use {@link handleQueryRequest} instead.\n */\nexport function handleGetQueriesRequest<S extends Schema>(\n  cb: (\n    name: string,\n    args: readonly ReadonlyJSONValue[],\n  ) => MaybePromise<{query: AnyQuery} | AnyQuery>,\n  schema: S,\n  requestOrJsonBody: Request | ReadonlyJSONValue,\n  logLevel: LogLevel = 'info',\n): Promise<QueryResponse> {\n  return transform(\n    normalizeLegacyQueryRequestArgs(cb, schema, requestOrJsonBody, logLevel),\n    'getQueries',\n  );\n}\n\n/**\n * Invokes the callback `cb` for each query in the request or JSON body.\n * The callback should return a Query or Promise<Query> that is the transformed result.\n *\n * This function will call `cb` in parallel for each query found in the request.\n *\n * If you need to limit concurrency, you can use a library like `p-limit` to wrap the `cb` function.\n * @deprecated Use {@link handleQueryRequest} instead.\n */\nexport function handleTransformRequest<S extends Schema>(\n  cb: (\n    name: string,\n    args: readonly ReadonlyJSONValue[],\n  ) => MaybePromise<{query: AnyQuery} | AnyQuery>,\n  schema: S,\n  requestOrJsonBody: Request | ReadonlyJSONValue,\n  logLevel: LogLevel = 'info',\n): Promise<QueryResponse> {\n  return transform(\n    normalizeLegacyQueryRequestArgs(cb, schema, requestOrJsonBody, logLevel),\n    'transform',\n  );\n}\n\n/**\n * Parsed query params accepted by {@link handleQueryRequest} when the\n * incoming request URL has already been handled by your framework.\n */\nexport type QuerySearchParams = URLSearchParams | Record<string, string>;\n\nexport type HandleQueryRequestArgs<S extends Schema> = {\n  /** Callback that transforms each requested query into a `Query`. */\n  handler: QueryRequestHandler;\n  /** Schema used when building the returned ASTs. */\n  schema: S;\n  /**\n   * Authenticated user ID. Null or undefined means the user is logged out.\n   */\n  userID: string | null | undefined;\n  /** Optional log level for request parsing and execution. */\n  logLevel?: LogLevel | undefined;\n} & (\n  | {\n      /** Fetch request containing the `/query` JSON body. */\n      request: Request;\n    }\n  | {\n      /** Parsed query params from the `/query` request URL. */\n      query: QuerySearchParams;\n      /** Parsed JSON body from the `/query` request. */\n      body: ReadonlyJSONValue;\n    }\n);\n\ntype NormalizedQueryRequestArgs<S extends Schema> = {\n  readonly schema: S;\n  readonly handler: LegacyQueryRequestHandler;\n  // Note: semantics of undefined differ from HandleQueryRequestArgs.userID.\n  // Here, undefined means the app didn't provide a user ID - we do not know if\n  // the user is logged in or not. This is legacy behavior needed to support\n  // deprecated signatures of handleQueryRequest which did not receive userID\n  // from app.\n  readonly userID: string | null | undefined;\n  readonly logLevel: LogLevel;\n} & (\n  | {\n      readonly type: 'request';\n      readonly request: Request;\n    }\n  | {\n      readonly type: 'body';\n      readonly jsonBody: ReadonlyJSONValue;\n    }\n);\n\n/**\n * Process a `/query` request from a Fetch `Request`.\n */\nexport function handleQueryRequest<S extends Schema>(\n  input: HandleQueryRequestArgs<S>,\n): Promise<QueryResponse>;\n\n/**\n * @deprecated Pass a single object instead:\n * `handleQueryRequest({handler, schema, request, userID, logLevel})`.\n */\nexport function handleQueryRequest<S extends Schema>(\n  transformQuery: QueryRequestHandler,\n  schema: S,\n  request: Request,\n  logLevel?: LogLevel,\n): Promise<QueryResponse>;\n\n/**\n * @deprecated Pass a single object instead:\n * `handleQueryRequest({handler, schema, body, userID, logLevel})`.\n */\nexport function handleQueryRequest<S extends Schema>(\n  transformQuery: QueryRequestHandler,\n  schema: S,\n  jsonBody: ReadonlyJSONValue,\n  logLevel?: LogLevel,\n): Promise<QueryResponse>;\n\nexport function handleQueryRequest<S extends Schema>(\n  inputOrTransformQuery: HandleQueryRequestArgs<S> | QueryRequestHandler,\n  schema?: S,\n  requestOrJsonBody?: Request | ReadonlyJSONValue,\n  logLevel?: LogLevel,\n): Promise<QueryResponse> {\n  const normalized =\n    typeof inputOrTransformQuery === 'object' &&\n    'handler' in inputOrTransformQuery\n      ? normalizeQueryRequestInput(inputOrTransformQuery)\n      : normalizeLegacyQueryRequestArgs(\n          wrapQueryRequestHandler(inputOrTransformQuery),\n          schema,\n          requestOrJsonBody,\n          logLevel,\n        );\n\n  return transform(normalized, 'query');\n}\n\nfunction normalizeQueryRequestInput<S extends Schema>(\n  input: HandleQueryRequestArgs<S>,\n): NormalizedQueryRequestArgs<S> {\n  return 'request' in input\n    ? normalizeLegacyQueryRequestArgs(\n        wrapQueryRequestHandler(input.handler),\n        input.schema,\n        input.request,\n        input.logLevel,\n        input.userID ?? null,\n      )\n    : normalizeLegacyQueryRequestArgs(\n        wrapQueryRequestHandler(input.handler),\n        input.schema,\n        input.body,\n        input.logLevel,\n        input.userID ?? null,\n      );\n}\n\nfunction normalizeLegacyQueryRequestArgs<S extends Schema>(\n  handler: LegacyQueryRequestHandler,\n  schema: S | undefined,\n  requestOrJsonBody: Request | ReadonlyJSONValue | undefined,\n  logLevel: LogLevel | undefined,\n  userID?: string | null,\n): NormalizedQueryRequestArgs<S> {\n  assert(\n    typeof schema !== 'undefined',\n    'Schema must be provided when using handleQueryRequest',\n  );\n\n  if (requestOrJsonBody instanceof Request) {\n    return {\n      type: 'request',\n      handler,\n      schema,\n      request: requestOrJsonBody,\n      userID,\n      logLevel: logLevel ?? 'info',\n    };\n  }\n\n  assert(\n    typeof requestOrJsonBody !== 'undefined',\n    'JSON body cannot be undefined',\n  );\n\n  return {\n    type: 'body',\n    handler,\n    schema,\n    jsonBody: requestOrJsonBody,\n    userID,\n    logLevel: logLevel ?? 'info',\n  };\n}\n\nasync function transform<S extends Schema>(\n  args: NormalizedQueryRequestArgs<S>,\n  apiName: 'query' | 'transform' | 'getQueries',\n): Promise<QueryResponse> {\n  const lc = createLogContext(args.logLevel).withContext('TransformRequest');\n  let parsed: TransformRequestMessage;\n  let queryIDs: string[] = [];\n  try {\n    let body: ReadonlyJSONValue;\n    if (args.type === 'request') {\n      body = await args.request.json();\n    } else {\n      body = args.jsonBody;\n    }\n\n    // Strip, so a newer zero-cache can add fields to the request.\n    parsed = v.parse(body, transformRequestMessageSchema, 'strip');\n\n    queryIDs = parsed[1].map(r => r.id);\n  } catch (error) {\n    lc.error?.(`Failed to parse ${apiName} request`, error);\n\n    const message = `Failed to parse ${apiName} request: ${getErrorMessage(error)}`;\n    const details = getErrorDetails(error);\n\n    return {\n      kind: ErrorKind.TransformFailed,\n      origin: ErrorOrigin.Server,\n      reason: ErrorReason.Parse,\n      message,\n      queryIDs,\n      ...(details ? {details} : {}),\n    };\n  }\n\n  try {\n    const nameMapper = clientToServer(args.schema.tables);\n\n    const responses: TransformResponseBody = await Promise.all(\n      parsed[1].map(async req => {\n        let finalQuery: AnyQuery;\n        try {\n          const result = await args.handler(req.name, req.args);\n          finalQuery = 'query' in result ? result.query : result;\n        } catch (error) {\n          const message = getErrorMessage(error);\n          const details = getErrorDetails(error);\n\n          return {\n            error: error instanceof QueryParseError ? 'parse' : 'app',\n            id: req.id,\n            name: req.name,\n            message,\n            ...(details ? {details} : {}),\n          };\n        }\n\n        try {\n          const q = asQueryInternals(finalQuery);\n          const ast = mapAST(q.ast, nameMapper);\n\n          return {\n            id: req.id,\n            name: req.name,\n            ast,\n          };\n        } catch (error) {\n          lc.error?.('Failed to map AST', error);\n          throw error;\n        }\n      }),\n    );\n\n    return {\n      kind: 'QueryResponse',\n      queries: responses,\n      ...(typeof args.userID !== 'undefined' ? {userID: args.userID} : {}),\n    } as const satisfies QueryResponse;\n  } catch (e) {\n    const message = getErrorMessage(e);\n    const details = getErrorDetails(e);\n\n    return {\n      kind: ErrorKind.TransformFailed,\n      origin: ErrorOrigin.Server,\n      reason: ErrorReason.Internal,\n      message,\n      queryIDs,\n      ...(details ? {details} : {}),\n    };\n  }\n}\n\n/**\n * A function that transforms a query by name and arguments into a Query object.\n *\n * @param name - The name of the query (can be dot-separated for nested queries)\n * @param args - The arguments to pass to the query (can be undefined)\n * @returns A Query object\n */\nexport type QueryRequestHandler = (\n  name: string,\n  args: ReadonlyJSONValue | undefined,\n) => AnyQuery;\n\n/** @deprecated Use `QueryRequestHandler` instead. */\nexport type TransformQueryFunction = QueryRequestHandler;\n\nexport type LegacyQueryRequestHandler = (\n  name: string,\n  args: readonly ReadonlyJSONValue[],\n) => MaybePromise<{query: AnyQuery} | AnyQuery>;\n\nfunction wrapQueryRequestHandler(\n  handler: QueryRequestHandler,\n): LegacyQueryRequestHandler {\n  return (name, args) => handler(name, args[0]);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AAgCA,SAAgB,wBACd,IAIA,QACA,mBACA,WAAqB,QACG;CACxB,OAAO,UACL,gCAAgC,IAAI,QAAQ,mBAAmB,QAAQ,GACvE,YACF;AACF;;;;;;;;;;AAWA,SAAgB,uBACd,IAIA,QACA,mBACA,WAAqB,QACG;CACxB,OAAO,UACL,gCAAgC,IAAI,QAAQ,mBAAmB,QAAQ,GACvE,WACF;AACF;AAkFA,SAAgB,mBACd,uBACA,QACA,mBACA,UACwB;CAYxB,OAAO,UAVL,OAAO,0BAA0B,YACjC,aAAa,wBACT,2BAA2B,qBAAqB,IAChD,gCACE,wBAAwB,qBAAqB,GAC7C,QACA,mBACA,QACF,GAEuB,OAAO;AACtC;AAEA,SAAS,2BACP,OAC+B;CAC/B,OAAO,aAAa,QAChB,gCACE,wBAAwB,MAAM,OAAO,GACrC,MAAM,QACN,MAAM,SACN,MAAM,UACN,MAAM,UAAU,IAClB,IACA,gCACE,wBAAwB,MAAM,OAAO,GACrC,MAAM,QACN,MAAM,MACN,MAAM,UACN,MAAM,UAAU,IAClB;AACN;AAEA,SAAS,gCACP,SACA,QACA,mBACA,UACA,QAC+B;CAC/B,OACE,OAAO,WAAW,aAClB,uDACF;CAEA,IAAI,6BAA6B,SAC/B,OAAO;EACL,MAAM;EACN;EACA;EACA,SAAS;EACT;EACA,UAAU,YAAY;CACxB;CAGF,OACE,OAAO,sBAAsB,aAC7B,+BACF;CAEA,OAAO;EACL,MAAM;EACN;EACA;EACA,UAAU;EACV;EACA,UAAU,YAAY;CACxB;AACF;AAEA,eAAe,UACb,MACA,SACwB;CACxB,MAAM,KAAK,iBAAiB,KAAK,QAAQ,CAAC,CAAC,YAAY,kBAAkB;CACzE,IAAI;CACJ,IAAI,WAAqB,CAAC;CAC1B,IAAI;EACF,IAAI;EACJ,IAAI,KAAK,SAAS,WAChB,OAAO,MAAM,KAAK,QAAQ,KAAK;OAE/B,OAAO,KAAK;EAId,SAASA,MAAQ,MAAM,+BAA+B,OAAO;EAE7D,WAAW,OAAO,EAAE,CAAC,KAAI,MAAK,EAAE,EAAE;CACpC,SAAS,OAAO;EACd,GAAG,QAAQ,mBAAmB,QAAQ,WAAW,KAAK;EAEtD,MAAM,UAAU,mBAAmB,QAAQ,YAAY,gBAAgB,KAAK;EAC5E,MAAM,UAAU,gBAAgB,KAAK;EAErC,OAAO;GACL,MAAMC;GACN,QAAQC;GACR,QAAQC;GACR;GACA;GACA,GAAI,UAAU,EAAC,QAAO,IAAI,CAAC;EAC7B;CACF;CAEA,IAAI;EACF,MAAM,aAAa,eAAe,KAAK,OAAO,MAAM;EAqCpD,OAAO;GACL,MAAM;GACN,SAAS,MArCoC,QAAQ,IACrD,OAAO,EAAE,CAAC,IAAI,OAAM,QAAO;IACzB,IAAI;IACJ,IAAI;KACF,MAAM,SAAS,MAAM,KAAK,QAAQ,IAAI,MAAM,IAAI,IAAI;KACpD,aAAa,WAAW,SAAS,OAAO,QAAQ;IAClD,SAAS,OAAO;KACd,MAAM,UAAU,gBAAgB,KAAK;KACrC,MAAM,UAAU,gBAAgB,KAAK;KAErC,OAAO;MACL,OAAO,iBAAiB,kBAAkB,UAAU;MACpD,IAAI,IAAI;MACR,MAAM,IAAI;MACV;MACA,GAAI,UAAU,EAAC,QAAO,IAAI,CAAC;KAC7B;IACF;IAEA,IAAI;KACF,MAAM,IAAI,iBAAiB,UAAU;KACrC,MAAM,MAAM,OAAO,EAAE,KAAK,UAAU;KAEpC,OAAO;MACL,IAAI,IAAI;MACR,MAAM,IAAI;MACV;KACF;IACF,SAAS,OAAO;KACd,GAAG,QAAQ,qBAAqB,KAAK;KACrC,MAAM;IACR;GACF,CAAC,CACH;GAKE,GAAI,OAAO,KAAK,WAAW,cAAc,EAAC,QAAQ,KAAK,OAAM,IAAI,CAAC;EACpE;CACF,SAAS,GAAG;EACV,MAAM,UAAU,gBAAgB,CAAC;EACjC,MAAM,UAAU,gBAAgB,CAAC;EAEjC,OAAO;GACL,MAAMF;GACN,QAAQC;GACR,QAAQE;GACR;GACA;GACA,GAAI,UAAU,EAAC,QAAO,IAAI,CAAC;EAC7B;CACF;AACF;AAsBA,SAAS,wBACP,SAC2B;CAC3B,QAAQ,MAAM,SAAS,QAAQ,MAAM,KAAK,EAAE;AAC9C"}

package/out/zero-server/src/schema.js

@@ -47,7 +47,8 @@ async function getServerSchema(queryable, schema) {
 * diffed against the `information_schema` query it replaced.
 */
 function serverSchemaQuery(schemaTablePairs) {
-	return formatPg(sql`
+	const inClause = sql.join(schemaTablePairs.map(([schema, table]) => sql`(${schema}::text, ${table}::text)`), ",");
+	const query = sql`
       SELECT
           n.nspname::text AS schema,
           cl.relname::text AS table,
@@ -101,8 +102,9 @@ function serverSchemaQuery(schemaTablePairs) {
           AND NOT pg_catalog.pg_is_other_temp_schema(n.oid)
           AND (pg_catalog.pg_has_role(cl.relowner, 'USAGE')
                OR pg_catalog.has_column_privilege(cl.oid, a.attnum, 'SELECT, INSERT, UPDATE, REFERENCES'))
-          AND (n.nspname, cl.relname) IN (${sql.join(schemaTablePairs.map(([schema, table]) => sql`(${schema}::text, ${table}::text)`), ",")})
-    `);
+          AND (n.nspname, cl.relname) IN (${inClause})
+    `;
+	return formatPg(query);
 }
 function makeSchemaIncompatibleErrorMessage(errors) {
 	if (errors.length === 0) return "No schema incompatibilities found.";
@@ -114,9 +116,7 @@ function makeSchemaIncompatibleErrorMessage(errors) {
 		case "missingColumn":
 			messages.push(`Column "${error.column}" in table "${error.table}" is defined in your zero schema but does not exist in the database.`);
 			break;
-		case "typeError":
-			messages.push(`Type mismatch for column "${error.column}" in table "${error.table}": ${error.requiredType === void 0 ? `${error.pgType} is currently unsupported in Zero. Please file a bug at https://bugs.rocicorp.dev/` : `${error.pgType} should be mapped to ${error.requiredType} in Zero not ${error.declaredType}.`}`);
-			break;
+		case "typeError": messages.push(`Type mismatch for column "${error.column}" in table "${error.table}": ${error.requiredType === void 0 ? `${error.pgType} is currently unsupported in Zero. Please file a bug at https://bugs.rocicorp.dev/` : `${error.pgType} should be mapped to ${error.requiredType} in Zero not ${error.declaredType}.`}`);
 	}
 	return [
 		"Schema incompatibility detected between your zero schema definition and the database:",
@@ -164,6 +164,6 @@ function checkSchemasAreCompatible(schema, serverSchema) {
 	return errors;
 }
 //#endregion
-export { getServerSchema };
+export { checkSchemasAreCompatible, getServerSchema, serverSchemaQuery };
 
 //# sourceMappingURL=schema.js.map

package/out/zero-server/src/schema.js.map

@@ -1 +1 @@
-{"version":3,"file":"schema.js","names":[],"sources":["../../../../zero-server/src/schema.ts"],"sourcesContent":["import {assert} from '../../shared/src/asserts.ts';\nimport type {Enum} from '../../shared/src/enum.ts';\nimport {must} from '../../shared/src/must.ts';\nimport {formatPg, sql} from '../../z2s/src/sql.ts';\nimport * as PostgresTypeClass from '../../zero-cache/src/db/postgres-type-class-enum.ts';\nimport {dataTypeToZqlValueType} from '../../zero-cache/src/types/pg-data-type.ts';\nimport type {Schema} from '../../zero-types/src/schema.ts';\nimport type {ServerSchema} from '../../zero-types/src/server-schema.ts';\nimport type {Queryable} from '../../zql/src/mutate/custom.ts';\n\ntype PostgresTypeClass = Enum<typeof PostgresTypeClass>;\n\ntype ServerSchemaRow = {\n  schema: string;\n  table: string;\n  column: string;\n  dataType: string;\n  length: string | null;\n  precision: string | null;\n  scale: string | null;\n  typtype: PostgresTypeClass;\n  typename: string;\n  elemTyptype: PostgresTypeClass | null;\n  elemTypname: string | null;\n};\n\nexport async function getServerSchema<S extends Schema>(\n  queryable: Queryable,\n  schema: S,\n): Promise<ServerSchema> {\n  const schemaTablePairs: [string, string][] = Object.values(schema.tables).map(\n    ({name, serverName}) => {\n      let schemaTablePair: [string, string] = ['public', serverName ?? name];\n      if (serverName) {\n        const firstPeriod = serverName.indexOf('.');\n        if (firstPeriod > -1) {\n          schemaTablePair = [\n            serverName.substring(0, firstPeriod),\n            serverName.substring(firstPeriod + 1, serverName.length),\n          ];\n        }\n      }\n      return schemaTablePair;\n    },\n  );\n\n  if (schemaTablePairs.length === 0) {\n    return {}; // No pairs to query for\n  }\n\n  const {text, values} = serverSchemaQuery(schemaTablePairs);\n  const results: Iterable<ServerSchemaRow> = (await queryable.query(\n    text,\n    values,\n  )) as Iterable<ServerSchemaRow>;\n\n  const serverSchema: ServerSchema = {};\n\n  for (const row of results) {\n    const tableName =\n      row.schema === 'public' ? row.table : `${row.schema}.${row.table}`;\n    let tableSchema = serverSchema[tableName];\n    if (!tableSchema) {\n      tableSchema = {};\n      serverSchema[tableName] = tableSchema;\n    }\n    const isArray = row.elemTyptype !== null;\n    const isEnum = (row.elemTyptype ?? row.typtype) === PostgresTypeClass.Enum;\n    let type = row.dataType.toLowerCase();\n    if (isArray) {\n      type = must(\n        row.elemTypname,\n        `Array column \"${row.column}\" in table \"${tableName}\" is missing element type name`,\n      );\n    } else if (isEnum) {\n      type = row.typename;\n    } else if (\n      type === 'user-defined' &&\n      dataTypeToZqlValueType(row.typename, false, false) !== undefined\n    ) {\n      type = row.typename;\n    }\n    if (\n      (type === 'bpchar' ||\n        type === 'character' ||\n        type === 'character varying' ||\n        type === 'varchar') &&\n      row.length\n    ) {\n      type = `${type}(${row.length})`;\n    } else if (type === 'character') {\n      // Without a length `character` means `character(1)`, so, like Postgres's\n      // own format_type, name an unbounded `bpchar` column `bpchar`.\n      type = 'bpchar';\n    }\n    if (\n      (type === 'numeric' || type === 'decimal') &&\n      row.precision !== null &&\n      row.scale !== null\n    ) {\n      type = `${type}(${row.precision}, ${row.scale})`;\n    }\n    tableSchema[row.column] = {\n      type,\n      isEnum,\n      isArray,\n    };\n  }\n\n  const errors = checkSchemasAreCompatible(schema, serverSchema);\n  assert(errors.length === 0, () => makeSchemaIncompatibleErrorMessage(errors));\n\n  return serverSchema;\n}\n\n/**\n * The query behind {@link getServerSchema}, exported so that it can be\n * diffed against the `information_schema` query it replaced.\n */\nexport function serverSchemaQuery(\n  schemaTablePairs: readonly (readonly [string, string])[],\n): {text: string; values: unknown[]} {\n  // Cast all inputs to text and all outputs to text to avoid\n  // any conversions customer's DBTransaction impl has on other types.\n  const inClause = sql.join(\n    schemaTablePairs.map(\n      ([schema, table]) => sql`(${schema}::text, ${table}::text)`,\n    ),\n    ',',\n  );\n  // Read pg_catalog directly instead of information_schema.columns. The view\n  // derives every column it returns through the information_schema._pg_*\n  // functions and a per column privilege check, which is several times slower\n  // than the query below. Server processes run this on the first transaction\n  // they serve, so serverless deployments pay for it on every cold start.\n  //\n  // The expressions below reproduce the parts of the view's definition that we\n  // select, down to the int4 wire types of length, precision and scale: the\n  // _pg_char_max_length / _pg_numeric_precision / _pg_numeric_scale arithmetic,\n  // the single level domain unwrap that data_type and udt_name perform (`ut`),\n  // and the view's own relkind, attnum and privilege filters. pg_type is joined\n  // by OID rather than by udt_name, which additionally avoids duplicate rows\n  // when two schemas declare a type with the same name. The types those\n  // functions special case are named through regtype, which the parser resolves\n  // while planning.\n  const query = sql`\n      SELECT\n          n.nspname::text AS schema,\n          cl.relname::text AS table,\n          a.attname::text AS column,\n          CASE\n              WHEN ut.typelem <> 0 AND ut.typlen = -1 THEN 'ARRAY'\n              WHEN nut.nspname = 'pg_catalog' THEN pg_catalog.format_type(ut.oid, NULL)\n              ELSE 'USER-DEFINED'\n          END::text AS \"dataType\",\n          CASE\n              WHEN tm.typmod = -1 THEN NULL\n              WHEN ut.oid IN ('pg_catalog.bpchar'::regtype, 'pg_catalog.varchar'::regtype) THEN tm.typmod - 4\n              WHEN ut.oid IN ('pg_catalog.bit'::regtype, 'pg_catalog.varbit'::regtype) THEN tm.typmod\n          END AS length,\n          CASE ut.oid\n              WHEN 'pg_catalog.int2'::regtype THEN 16\n              WHEN 'pg_catalog.int4'::regtype THEN 32\n              WHEN 'pg_catalog.int8'::regtype THEN 64\n              WHEN 'pg_catalog.float4'::regtype THEN 24\n              WHEN 'pg_catalog.float8'::regtype THEN 53\n              WHEN 'pg_catalog.numeric'::regtype THEN CASE WHEN tm.typmod = -1 THEN NULL ELSE ((tm.typmod - 4) >> 16) & 65535 END\n          END AS precision,\n          CASE\n              WHEN ut.oid IN ('pg_catalog.int2'::regtype, 'pg_catalog.int4'::regtype, 'pg_catalog.int8'::regtype) THEN 0\n              WHEN ut.oid = 'pg_catalog.numeric'::regtype AND tm.typmod <> -1 THEN (tm.typmod - 4) & 65535\n          END AS scale,\n          ut.typtype::text AS typtype,\n          ut.typname::text AS typename,\n          CASE WHEN ut.typelem <> 0 THEN et.typtype::text ELSE NULL END AS \"elemTyptype\",\n          CASE WHEN ut.typelem <> 0 THEN et.typname::text ELSE NULL END AS \"elemTypname\"\n      FROM\n          pg_catalog.pg_attribute a\n      JOIN\n          pg_catalog.pg_class cl ON cl.oid = a.attrelid\n      JOIN\n          pg_catalog.pg_namespace n ON n.oid = cl.relnamespace\n      JOIN\n          pg_catalog.pg_type t ON t.oid = a.atttypid\n      JOIN\n          pg_catalog.pg_type ut ON ut.oid = CASE WHEN t.typtype = 'd' THEN t.typbasetype ELSE t.oid END\n      JOIN\n          pg_catalog.pg_namespace nut ON nut.oid = ut.typnamespace\n      LEFT JOIN\n          pg_catalog.pg_type et ON et.oid = ut.typelem\n      CROSS JOIN LATERAL\n          (SELECT CASE WHEN t.typtype = 'd' THEN t.typtypmod ELSE a.atttypmod END AS typmod) tm\n      WHERE\n          a.attnum > 0\n          AND NOT a.attisdropped\n          AND cl.relkind IN ('r', 'v', 'f', 'p')\n          AND NOT pg_catalog.pg_is_other_temp_schema(n.oid)\n          AND (pg_catalog.pg_has_role(cl.relowner, 'USAGE')\n               OR pg_catalog.has_column_privilege(cl.oid, a.attnum, 'SELECT, INSERT, UPDATE, REFERENCES'))\n          AND (n.nspname, cl.relname) IN (${inClause})\n    `;\n  return formatPg(query);\n}\n\nfunction makeSchemaIncompatibleErrorMessage(\n  errors: SchemaIncompatibilityError[],\n) {\n  if (errors.length === 0) {\n    return 'No schema incompatibilities found.';\n  }\n\n  const messages: string[] = [];\n\n  for (const error of errors) {\n    switch (error.type) {\n      case 'missingTable':\n        messages.push(\n          `Table \"${error.table}\" is defined in your zero schema but does not exist in the database.`,\n        );\n        break;\n      case 'missingColumn':\n        messages.push(\n          `Column \"${error.column}\" in table \"${error.table}\" is defined in your zero schema but does not exist in the database.`,\n        );\n        break;\n      case 'typeError':\n        messages.push(\n          `Type mismatch for column \"${error.column}\" in table \"${error.table}\": ${error.requiredType === undefined ? `${error.pgType} is currently unsupported in Zero. Please file a bug at https://bugs.rocicorp.dev/` : `${error.pgType} should be mapped to ${error.requiredType} in Zero not ${error.declaredType}.`}`,\n        );\n        break;\n    }\n  }\n\n  return [\n    'Schema incompatibility detected between your zero schema definition and the database:',\n    '',\n    ...messages.map(msg => `  - ${msg}`),\n    '',\n    'Please update your schema definition to match the database or migrate your database to match the schema.',\n  ].join('\\n');\n}\n\nexport type SchemaIncompatibilityError =\n  | {\n      type: 'typeError';\n      table: string;\n      column: string;\n      pgType: string;\n      declaredType: string;\n      requiredType: string | undefined;\n    }\n  | {\n      type: 'missingColumn';\n      table: string;\n      column: string;\n    }\n  | {\n      type: 'missingTable';\n      table: string;\n    };\n\nexport function checkSchemasAreCompatible(\n  schema: Schema,\n  serverSchema: ServerSchema,\n): SchemaIncompatibilityError[] {\n  const errors: SchemaIncompatibilityError[] = [];\n  // Check that all tables in schema exist in serverSchema\n  for (const table of Object.values(schema.tables)) {\n    const serverTableName = table.serverName ?? table.name;\n\n    if (!serverSchema[serverTableName]) {\n      errors.push({\n        type: 'missingTable',\n        table: serverTableName,\n      });\n      continue;\n    }\n\n    // Check that all columns in the table exist in serverSchema\n    for (const [columnName, column] of Object.entries(table.columns)) {\n      const serverColumnName = column.serverName ?? columnName;\n\n      if (!serverSchema[serverTableName][serverColumnName]) {\n        errors.push({\n          type: 'missingColumn',\n          table: serverTableName,\n          column: serverColumnName,\n        });\n        continue;\n      }\n\n      // Check type compatibility\n      const serverColumn = serverSchema[serverTableName][serverColumnName];\n      const declaredType = column.type;\n      const pgType = serverColumn.type;\n      const requiredType = dataTypeToZqlValueType(\n        pgType,\n        serverColumn.isEnum,\n        serverColumn.isArray,\n      );\n      if (requiredType !== declaredType) {\n        errors.push({\n          type: 'typeError',\n          table: serverTableName,\n          column: serverColumnName,\n          pgType,\n          declaredType,\n          requiredType,\n        });\n      }\n    }\n  }\n\n  return errors;\n}\n"],"mappings":";;;;;AA0BA,eAAsB,gBACpB,WACA,QACuB;CACvB,MAAM,mBAAuC,OAAO,OAAO,OAAO,MAAM,EAAE,KACvE,EAAC,MAAM,iBAAgB;EACtB,IAAI,kBAAoC,CAAC,UAAU,cAAc,IAAI;EACrE,IAAI,YAAY;GACd,MAAM,cAAc,WAAW,QAAQ,GAAG;GAC1C,IAAI,cAAc,IAChB,kBAAkB,CAChB,WAAW,UAAU,GAAG,WAAW,GACnC,WAAW,UAAU,cAAc,GAAG,WAAW,MAAM,CACzD;EAEJ;EACA,OAAO;CACT,CACF;CAEA,IAAI,iBAAiB,WAAW,GAC9B,OAAO,CAAC;CAGV,MAAM,EAAC,MAAM,WAAU,kBAAkB,gBAAgB;CACzD,MAAM,UAAsC,MAAM,UAAU,MAC1D,MACA,MACF;CAEA,MAAM,eAA6B,CAAC;CAEpC,KAAK,MAAM,OAAO,SAAS;EACzB,MAAM,YACJ,IAAI,WAAW,WAAW,IAAI,QAAQ,GAAG,IAAI,OAAO,GAAG,IAAI;EAC7D,IAAI,cAAc,aAAa;EAC/B,IAAI,CAAC,aAAa;GAChB,cAAc,CAAC;GACf,aAAa,aAAa;EAC5B;EACA,MAAM,UAAU,IAAI,gBAAgB;EACpC,MAAM,UAAU,IAAI,eAAe,IAAI,aAAa;EACpD,IAAI,OAAO,IAAI,SAAS,YAAY;EACpC,IAAI,SACF,OAAO,KACL,IAAI,aACJ,iBAAiB,IAAI,OAAO,cAAc,UAAU,+BACtD;OACK,IAAI,QACT,OAAO,IAAI;OACN,IACL,SAAS,kBACT,uBAAuB,IAAI,UAAU,OAAO,KAAK,MAAM,KAAA,GAEvD,OAAO,IAAI;EAEb,KACG,SAAS,YACR,SAAS,eACT,SAAS,uBACT,SAAS,cACX,IAAI,QAEJ,OAAO,GAAG,KAAK,GAAG,IAAI,OAAO;OACxB,IAAI,SAAS,aAGlB,OAAO;EAET,KACG,SAAS,aAAa,SAAS,cAChC,IAAI,cAAc,QAClB,IAAI,UAAU,MAEd,OAAO,GAAG,KAAK,GAAG,IAAI,UAAU,IAAI,IAAI,MAAM;EAEhD,YAAY,IAAI,UAAU;GACxB;GACA;GACA;EACF;CACF;CAEA,MAAM,SAAS,0BAA0B,QAAQ,YAAY;CAC7D,OAAO,OAAO,WAAW,SAAS,mCAAmC,MAAM,CAAC;CAE5E,OAAO;AACT;;;;;AAMA,SAAgB,kBACd,kBACmC;CAgFnC,OAAO,SAAS,GAxDC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;4CArBA,IAAI,KACnB,iBAAiB,KACd,CAAC,QAAQ,WAAW,GAAG,IAAI,OAAO,UAAU,MAAM,QACrD,GACA,GAuEwC,EAAS;KAE9B;AACvB;AAEA,SAAS,mCACP,QACA;CACA,IAAI,OAAO,WAAW,GACpB,OAAO;CAGT,MAAM,WAAqB,CAAC;CAE5B,KAAK,MAAM,SAAS,QAClB,QAAQ,MAAM,MAAd;EACE,KAAK;GACH,SAAS,KACP,UAAU,MAAM,MAAM,qEACxB;GACA;EACF,KAAK;GACH,SAAS,KACP,WAAW,MAAM,OAAO,cAAc,MAAM,MAAM,qEACpD;GACA;EACF,KAAK;GACH,SAAS,KACP,6BAA6B,MAAM,OAAO,cAAc,MAAM,MAAM,KAAK,MAAM,iBAAiB,KAAA,IAAY,GAAG,MAAM,OAAO,sFAAsF,GAAG,MAAM,OAAO,uBAAuB,MAAM,aAAa,eAAe,MAAM,aAAa,IAChT;GACA;CACJ;CAGF,OAAO;EACL;EACA;EACA,GAAG,SAAS,KAAI,QAAO,OAAO,KAAK;EACnC;EACA;CACF,EAAE,KAAK,IAAI;AACb;AAqBA,SAAgB,0BACd,QACA,cAC8B;CAC9B,MAAM,SAAuC,CAAC;CAE9C,KAAK,MAAM,SAAS,OAAO,OAAO,OAAO,MAAM,GAAG;EAChD,MAAM,kBAAkB,MAAM,cAAc,MAAM;EAElD,IAAI,CAAC,aAAa,kBAAkB;GAClC,OAAO,KAAK;IACV,MAAM;IACN,OAAO;GACT,CAAC;GACD;EACF;EAGA,KAAK,MAAM,CAAC,YAAY,WAAW,OAAO,QAAQ,MAAM,OAAO,GAAG;GAChE,MAAM,mBAAmB,OAAO,cAAc;GAE9C,IAAI,CAAC,aAAa,iBAAiB,mBAAmB;IACpD,OAAO,KAAK;KACV,MAAM;KACN,OAAO;KACP,QAAQ;IACV,CAAC;IACD;GACF;GAGA,MAAM,eAAe,aAAa,iBAAiB;GACnD,MAAM,eAAe,OAAO;GAC5B,MAAM,SAAS,aAAa;GAC5B,MAAM,eAAe,uBACnB,QACA,aAAa,QACb,aAAa,OACf;GACA,IAAI,iBAAiB,cACnB,OAAO,KAAK;IACV,MAAM;IACN,OAAO;IACP,QAAQ;IACR;IACA;IACA;GACF,CAAC;EAEL;CACF;CAEA,OAAO;AACT"}
+{"version":3,"file":"schema.js","names":[],"sources":["../../../../zero-server/src/schema.ts"],"sourcesContent":["import {assert} from '../../shared/src/asserts.ts';\nimport type {Enum} from '../../shared/src/enum.ts';\nimport {must} from '../../shared/src/must.ts';\nimport {formatPg, sql} from '../../z2s/src/sql.ts';\nimport * as PostgresTypeClass from '../../zero-cache/src/db/postgres-type-class-enum.ts';\nimport {dataTypeToZqlValueType} from '../../zero-cache/src/types/pg-data-type.ts';\nimport type {Schema} from '../../zero-types/src/schema.ts';\nimport type {ServerSchema} from '../../zero-types/src/server-schema.ts';\nimport type {Queryable} from '../../zql/src/mutate/custom.ts';\n\ntype PostgresTypeClass = Enum<typeof PostgresTypeClass>;\n\ntype ServerSchemaRow = {\n  schema: string;\n  table: string;\n  column: string;\n  dataType: string;\n  length: string | null;\n  precision: string | null;\n  scale: string | null;\n  typtype: PostgresTypeClass;\n  typename: string;\n  elemTyptype: PostgresTypeClass | null;\n  elemTypname: string | null;\n};\n\nexport async function getServerSchema<S extends Schema>(\n  queryable: Queryable,\n  schema: S,\n): Promise<ServerSchema> {\n  const schemaTablePairs: [string, string][] = Object.values(schema.tables).map(\n    ({name, serverName}) => {\n      let schemaTablePair: [string, string] = ['public', serverName ?? name];\n      if (serverName) {\n        const firstPeriod = serverName.indexOf('.');\n        if (firstPeriod > -1) {\n          schemaTablePair = [\n            serverName.substring(0, firstPeriod),\n            serverName.substring(firstPeriod + 1, serverName.length),\n          ];\n        }\n      }\n      return schemaTablePair;\n    },\n  );\n\n  if (schemaTablePairs.length === 0) {\n    return {}; // No pairs to query for\n  }\n\n  const {text, values} = serverSchemaQuery(schemaTablePairs);\n  const results: Iterable<ServerSchemaRow> = (await queryable.query(\n    text,\n    values,\n  )) as Iterable<ServerSchemaRow>;\n\n  const serverSchema: ServerSchema = {};\n\n  for (const row of results) {\n    const tableName =\n      row.schema === 'public' ? row.table : `${row.schema}.${row.table}`;\n    let tableSchema = serverSchema[tableName];\n    if (!tableSchema) {\n      tableSchema = {};\n      serverSchema[tableName] = tableSchema;\n    }\n    const isArray = row.elemTyptype !== null;\n    const isEnum = (row.elemTyptype ?? row.typtype) === PostgresTypeClass.Enum;\n    let type = row.dataType.toLowerCase();\n    if (isArray) {\n      type = must(\n        row.elemTypname,\n        `Array column \"${row.column}\" in table \"${tableName}\" is missing element type name`,\n      );\n    } else if (isEnum) {\n      type = row.typename;\n    } else if (\n      type === 'user-defined' &&\n      dataTypeToZqlValueType(row.typename, false, false) !== undefined\n    ) {\n      type = row.typename;\n    }\n    if (\n      (type === 'bpchar' ||\n        type === 'character' ||\n        type === 'character varying' ||\n        type === 'varchar') &&\n      row.length\n    ) {\n      type = `${type}(${row.length})`;\n    } else if (type === 'character') {\n      // Without a length `character` means `character(1)`, so, like Postgres's\n      // own format_type, name an unbounded `bpchar` column `bpchar`.\n      type = 'bpchar';\n    }\n    if (\n      (type === 'numeric' || type === 'decimal') &&\n      row.precision !== null &&\n      row.scale !== null\n    ) {\n      type = `${type}(${row.precision}, ${row.scale})`;\n    }\n    tableSchema[row.column] = {\n      type,\n      isEnum,\n      isArray,\n    };\n  }\n\n  const errors = checkSchemasAreCompatible(schema, serverSchema);\n  assert(errors.length === 0, () => makeSchemaIncompatibleErrorMessage(errors));\n\n  return serverSchema;\n}\n\n/**\n * The query behind {@link getServerSchema}, exported so that it can be\n * diffed against the `information_schema` query it replaced.\n */\nexport function serverSchemaQuery(\n  schemaTablePairs: readonly (readonly [string, string])[],\n): {text: string; values: unknown[]} {\n  // Cast all inputs to text and all outputs to text to avoid\n  // any conversions customer's DBTransaction impl has on other types.\n  const inClause = sql.join(\n    schemaTablePairs.map(\n      ([schema, table]) => sql`(${schema}::text, ${table}::text)`,\n    ),\n    ',',\n  );\n  // Read pg_catalog directly instead of information_schema.columns. The view\n  // derives every column it returns through the information_schema._pg_*\n  // functions and a per column privilege check, which is several times slower\n  // than the query below. Server processes run this on the first transaction\n  // they serve, so serverless deployments pay for it on every cold start.\n  //\n  // The expressions below reproduce the parts of the view's definition that we\n  // select, down to the int4 wire types of length, precision and scale: the\n  // _pg_char_max_length / _pg_numeric_precision / _pg_numeric_scale arithmetic,\n  // the single level domain unwrap that data_type and udt_name perform (`ut`),\n  // and the view's own relkind, attnum and privilege filters. pg_type is joined\n  // by OID rather than by udt_name, which additionally avoids duplicate rows\n  // when two schemas declare a type with the same name. The types those\n  // functions special case are named through regtype, which the parser resolves\n  // while planning.\n  const query = sql`\n      SELECT\n          n.nspname::text AS schema,\n          cl.relname::text AS table,\n          a.attname::text AS column,\n          CASE\n              WHEN ut.typelem <> 0 AND ut.typlen = -1 THEN 'ARRAY'\n              WHEN nut.nspname = 'pg_catalog' THEN pg_catalog.format_type(ut.oid, NULL)\n              ELSE 'USER-DEFINED'\n          END::text AS \"dataType\",\n          CASE\n              WHEN tm.typmod = -1 THEN NULL\n              WHEN ut.oid IN ('pg_catalog.bpchar'::regtype, 'pg_catalog.varchar'::regtype) THEN tm.typmod - 4\n              WHEN ut.oid IN ('pg_catalog.bit'::regtype, 'pg_catalog.varbit'::regtype) THEN tm.typmod\n          END AS length,\n          CASE ut.oid\n              WHEN 'pg_catalog.int2'::regtype THEN 16\n              WHEN 'pg_catalog.int4'::regtype THEN 32\n              WHEN 'pg_catalog.int8'::regtype THEN 64\n              WHEN 'pg_catalog.float4'::regtype THEN 24\n              WHEN 'pg_catalog.float8'::regtype THEN 53\n              WHEN 'pg_catalog.numeric'::regtype THEN CASE WHEN tm.typmod = -1 THEN NULL ELSE ((tm.typmod - 4) >> 16) & 65535 END\n          END AS precision,\n          CASE\n              WHEN ut.oid IN ('pg_catalog.int2'::regtype, 'pg_catalog.int4'::regtype, 'pg_catalog.int8'::regtype) THEN 0\n              WHEN ut.oid = 'pg_catalog.numeric'::regtype AND tm.typmod <> -1 THEN (tm.typmod - 4) & 65535\n          END AS scale,\n          ut.typtype::text AS typtype,\n          ut.typname::text AS typename,\n          CASE WHEN ut.typelem <> 0 THEN et.typtype::text ELSE NULL END AS \"elemTyptype\",\n          CASE WHEN ut.typelem <> 0 THEN et.typname::text ELSE NULL END AS \"elemTypname\"\n      FROM\n          pg_catalog.pg_attribute a\n      JOIN\n          pg_catalog.pg_class cl ON cl.oid = a.attrelid\n      JOIN\n          pg_catalog.pg_namespace n ON n.oid = cl.relnamespace\n      JOIN\n          pg_catalog.pg_type t ON t.oid = a.atttypid\n      JOIN\n          pg_catalog.pg_type ut ON ut.oid = CASE WHEN t.typtype = 'd' THEN t.typbasetype ELSE t.oid END\n      JOIN\n          pg_catalog.pg_namespace nut ON nut.oid = ut.typnamespace\n      LEFT JOIN\n          pg_catalog.pg_type et ON et.oid = ut.typelem\n      CROSS JOIN LATERAL\n          (SELECT CASE WHEN t.typtype = 'd' THEN t.typtypmod ELSE a.atttypmod END AS typmod) tm\n      WHERE\n          a.attnum > 0\n          AND NOT a.attisdropped\n          AND cl.relkind IN ('r', 'v', 'f', 'p')\n          AND NOT pg_catalog.pg_is_other_temp_schema(n.oid)\n          AND (pg_catalog.pg_has_role(cl.relowner, 'USAGE')\n               OR pg_catalog.has_column_privilege(cl.oid, a.attnum, 'SELECT, INSERT, UPDATE, REFERENCES'))\n          AND (n.nspname, cl.relname) IN (${inClause})\n    `;\n  return formatPg(query);\n}\n\nfunction makeSchemaIncompatibleErrorMessage(\n  errors: SchemaIncompatibilityError[],\n) {\n  if (errors.length === 0) {\n    return 'No schema incompatibilities found.';\n  }\n\n  const messages: string[] = [];\n\n  for (const error of errors) {\n    switch (error.type) {\n      case 'missingTable':\n        messages.push(\n          `Table \"${error.table}\" is defined in your zero schema but does not exist in the database.`,\n        );\n        break;\n      case 'missingColumn':\n        messages.push(\n          `Column \"${error.column}\" in table \"${error.table}\" is defined in your zero schema but does not exist in the database.`,\n        );\n        break;\n      case 'typeError':\n        messages.push(\n          `Type mismatch for column \"${error.column}\" in table \"${error.table}\": ${error.requiredType === undefined ? `${error.pgType} is currently unsupported in Zero. Please file a bug at https://bugs.rocicorp.dev/` : `${error.pgType} should be mapped to ${error.requiredType} in Zero not ${error.declaredType}.`}`,\n        );\n        break;\n    }\n  }\n\n  return [\n    'Schema incompatibility detected between your zero schema definition and the database:',\n    '',\n    ...messages.map(msg => `  - ${msg}`),\n    '',\n    'Please update your schema definition to match the database or migrate your database to match the schema.',\n  ].join('\\n');\n}\n\nexport type SchemaIncompatibilityError =\n  | {\n      type: 'typeError';\n      table: string;\n      column: string;\n      pgType: string;\n      declaredType: string;\n      requiredType: string | undefined;\n    }\n  | {\n      type: 'missingColumn';\n      table: string;\n      column: string;\n    }\n  | {\n      type: 'missingTable';\n      table: string;\n    };\n\nexport function checkSchemasAreCompatible(\n  schema: Schema,\n  serverSchema: ServerSchema,\n): SchemaIncompatibilityError[] {\n  const errors: SchemaIncompatibilityError[] = [];\n  // Check that all tables in schema exist in serverSchema\n  for (const table of Object.values(schema.tables)) {\n    const serverTableName = table.serverName ?? table.name;\n\n    if (!serverSchema[serverTableName]) {\n      errors.push({\n        type: 'missingTable',\n        table: serverTableName,\n      });\n      continue;\n    }\n\n    // Check that all columns in the table exist in serverSchema\n    for (const [columnName, column] of Object.entries(table.columns)) {\n      const serverColumnName = column.serverName ?? columnName;\n\n      if (!serverSchema[serverTableName][serverColumnName]) {\n        errors.push({\n          type: 'missingColumn',\n          table: serverTableName,\n          column: serverColumnName,\n        });\n        continue;\n      }\n\n      // Check type compatibility\n      const serverColumn = serverSchema[serverTableName][serverColumnName];\n      const declaredType = column.type;\n      const pgType = serverColumn.type;\n      const requiredType = dataTypeToZqlValueType(\n        pgType,\n        serverColumn.isEnum,\n        serverColumn.isArray,\n      );\n      if (requiredType !== declaredType) {\n        errors.push({\n          type: 'typeError',\n          table: serverTableName,\n          column: serverColumnName,\n          pgType,\n          declaredType,\n          requiredType,\n        });\n      }\n    }\n  }\n\n  return errors;\n}\n"],"mappings":";;;;;AA0BA,eAAsB,gBACpB,WACA,QACuB;CACvB,MAAM,mBAAuC,OAAO,OAAO,OAAO,MAAM,CAAC,CAAC,KACvE,EAAC,MAAM,iBAAgB;EACtB,IAAI,kBAAoC,CAAC,UAAU,cAAc,IAAI;EACrE,IAAI,YAAY;GACd,MAAM,cAAc,WAAW,QAAQ,GAAG;GAC1C,IAAI,cAAc,IAChB,kBAAkB,CAChB,WAAW,UAAU,GAAG,WAAW,GACnC,WAAW,UAAU,cAAc,GAAG,WAAW,MAAM,CACzD;EAEJ;EACA,OAAO;CACT,CACF;CAEA,IAAI,iBAAiB,WAAW,GAC9B,OAAO,CAAC;CAGV,MAAM,EAAC,MAAM,WAAU,kBAAkB,gBAAgB;CACzD,MAAM,UAAsC,MAAM,UAAU,MAC1D,MACA,MACF;CAEA,MAAM,eAA6B,CAAC;CAEpC,KAAK,MAAM,OAAO,SAAS;EACzB,MAAM,YACJ,IAAI,WAAW,WAAW,IAAI,QAAQ,GAAG,IAAI,OAAO,GAAG,IAAI;EAC7D,IAAI,cAAc,aAAa;EAC/B,IAAI,CAAC,aAAa;GAChB,cAAc,CAAC;GACf,aAAa,aAAa;EAC5B;EACA,MAAM,UAAU,IAAI,gBAAgB;EACpC,MAAM,UAAU,IAAI,eAAe,IAAI,aAAa;EACpD,IAAI,OAAO,IAAI,SAAS,YAAY;EACpC,IAAI,SACF,OAAO,KACL,IAAI,aACJ,iBAAiB,IAAI,OAAO,cAAc,UAAU,+BACtD;OACK,IAAI,QACT,OAAO,IAAI;OACN,IACL,SAAS,kBACT,uBAAuB,IAAI,UAAU,OAAO,KAAK,MAAM,KAAA,GAEvD,OAAO,IAAI;EAEb,KACG,SAAS,YACR,SAAS,eACT,SAAS,uBACT,SAAS,cACX,IAAI,QAEJ,OAAO,GAAG,KAAK,GAAG,IAAI,OAAO;OACxB,IAAI,SAAS,aAGlB,OAAO;EAET,KACG,SAAS,aAAa,SAAS,cAChC,IAAI,cAAc,QAClB,IAAI,UAAU,MAEd,OAAO,GAAG,KAAK,GAAG,IAAI,UAAU,IAAI,IAAI,MAAM;EAEhD,YAAY,IAAI,UAAU;GACxB;GACA;GACA;EACF;CACF;CAEA,MAAM,SAAS,0BAA0B,QAAQ,YAAY;CAC7D,OAAO,OAAO,WAAW,SAAS,mCAAmC,MAAM,CAAC;CAE5E,OAAO;AACT;;;;;AAMA,SAAgB,kBACd,kBACmC;CAGnC,MAAM,WAAW,IAAI,KACnB,iBAAiB,KACd,CAAC,QAAQ,WAAW,GAAG,IAAI,OAAO,UAAU,MAAM,QACrD,GACA,GACF;CAgBA,MAAM,QAAQ,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;4CAsDyB,SAAS;;CAEnD,OAAO,SAAS,KAAK;AACvB;AAEA,SAAS,mCACP,QACA;CACA,IAAI,OAAO,WAAW,GACpB,OAAO;CAGT,MAAM,WAAqB,CAAC;CAE5B,KAAK,MAAM,SAAS,QAClB,QAAQ,MAAM,MAAd;EACE,KAAK;GACH,SAAS,KACP,UAAU,MAAM,MAAM,qEACxB;GACA;EACF,KAAK;GACH,SAAS,KACP,WAAW,MAAM,OAAO,cAAc,MAAM,MAAM,qEACpD;GACA;EACF,KAAK,aACH,SAAS,KACP,6BAA6B,MAAM,OAAO,cAAc,MAAM,MAAM,KAAK,MAAM,iBAAiB,KAAA,IAAY,GAAG,MAAM,OAAO,sFAAsF,GAAG,MAAM,OAAO,uBAAuB,MAAM,aAAa,eAAe,MAAM,aAAa,IAChT;CAEJ;CAGF,OAAO;EACL;EACA;EACA,GAAG,SAAS,KAAI,QAAO,OAAO,KAAK;EACnC;EACA;CACF,CAAC,CAAC,KAAK,IAAI;AACb;AAqBA,SAAgB,0BACd,QACA,cAC8B;CAC9B,MAAM,SAAuC,CAAC;CAE9C,KAAK,MAAM,SAAS,OAAO,OAAO,OAAO,MAAM,GAAG;EAChD,MAAM,kBAAkB,MAAM,cAAc,MAAM;EAElD,IAAI,CAAC,aAAa,kBAAkB;GAClC,OAAO,KAAK;IACV,MAAM;IACN,OAAO;GACT,CAAC;GACD;EACF;EAGA,KAAK,MAAM,CAAC,YAAY,WAAW,OAAO,QAAQ,MAAM,OAAO,GAAG;GAChE,MAAM,mBAAmB,OAAO,cAAc;GAE9C,IAAI,CAAC,aAAa,gBAAgB,CAAC,mBAAmB;IACpD,OAAO,KAAK;KACV,MAAM;KACN,OAAO;KACP,QAAQ;IACV,CAAC;IACD;GACF;GAGA,MAAM,eAAe,aAAa,gBAAgB,CAAC;GACnD,MAAM,eAAe,OAAO;GAC5B,MAAM,SAAS,aAAa;GAC5B,MAAM,eAAe,uBACnB,QACA,aAAa,QACb,aAAa,OACf;GACA,IAAI,iBAAiB,cACnB,OAAO,KAAK;IACV,MAAM;IACN,OAAO;IACP,QAAQ;IACR;IACA;IACA;GACF,CAAC;EAEL;CACF;CAEA,OAAO;AACT"}

package/out/zero-server/src/zql-database.js.map

@@ -1 +1 @@
-{"version":3,"file":"zql-database.js","names":["#crudFactory","#schema","#makeServerTransaction","#runOn"],"sources":["../../../../zero-server/src/zql-database.ts"],"sourcesContent":["import type {MaybePromise} from '../../shared/src/types.ts';\nimport {formatPg, sql} from '../../z2s/src/sql.ts';\nimport type {CleanupResultsArg} from '../../zero-protocol/src/mutation.ts';\nimport type {Schema} from '../../zero-types/src/schema.ts';\nimport type {\n  DBConnection,\n  DBTransaction,\n  Queryable,\n  RetryPredicate,\n} from '../../zql/src/mutate/custom.ts';\nimport {asQueryInternals} from '../../zql/src/query/query-internals.ts';\nimport type {\n  HumanReadable,\n  Query,\n  RunOptions,\n} from '../../zql/src/query/query.ts';\nimport {ClosableDBTransaction} from './closable-db-transaction.ts';\nimport {CRUDMutatorFactory, type TransactionImpl} from './custom.ts';\nimport {executePostgresQuery} from './pg-query-executor.ts';\nimport type {\n  Database,\n  TransactionProviderHooks,\n  TransactionProviderInput,\n} from './process-mutations.ts';\n\n/**\n * Implements a Database for use with PushProcessor that is backed by Postgres.\n *\n * This implementation also implements the same ZQL interfaces for reading and\n * writing data that the Zero client does, so that mutator functions can be\n * shared across client and server.\n */\nexport class ZQLDatabase<\n  TSchema extends Schema,\n  TWrappedTransaction,\n> implements Database<TransactionImpl<TSchema, TWrappedTransaction>> {\n  readonly connection: DBConnection<TWrappedTransaction>;\n  readonly #crudFactory: CRUDMutatorFactory<TSchema>;\n  readonly #schema: TSchema;\n\n  constructor(connection: DBConnection<TWrappedTransaction>, schema: TSchema) {\n    this.connection = connection;\n    this.#crudFactory = new CRUDMutatorFactory(schema);\n    this.#schema = schema;\n  }\n\n  transaction<R>(\n    callback: (\n      tx: TransactionImpl<TSchema, TWrappedTransaction>,\n      transactionHooks: TransactionProviderHooks,\n    ) => MaybePromise<R>,\n    transactionInput?: TransactionProviderInput,\n  ): Promise<R> {\n    // Icky hack. This is just here to have user not have to do this.\n    // These interfaces need to be factored better.\n    const {\n      upstreamSchema = '',\n      clientGroupID = '',\n      clientID = '',\n      mutationID = 0,\n      addRetryPredicate,\n    } = transactionInput ?? {};\n    return this.connection.transaction(async dbTx => {\n      // What the mutator's `tx` uses. Closed once the callback settles, so a\n      // `tx` that outlives its transaction, such as one captured by a\n      // `tx.retryOn()` predicate, throws instead of reaching a connection\n      // that may already be back in the pool.\n      const mutatorTx = new ClosableDBTransaction(dbTx);\n      try {\n        const zeroTx = await this.#makeServerTransaction(\n          mutatorTx,\n          clientID,\n          mutationID,\n          addRetryPredicate,\n        );\n\n        return await callback(zeroTx, {\n          async updateClientMutationID() {\n            const formatted = formatPg(\n              sql`INSERT INTO ${sql.ident(upstreamSchema)}.clients \n                    as current (\"clientGroupID\", \"clientID\", \"lastMutationID\")\n                        VALUES (${clientGroupID}, ${clientID}, ${1})\n                    ON CONFLICT (\"clientGroupID\", \"clientID\")\n                    DO UPDATE SET \"lastMutationID\" = current.\"lastMutationID\" + 1\n                    RETURNING \"lastMutationID\"`,\n            );\n\n            const [{lastMutationID}] = (await dbTx.query(\n              formatted.text,\n              formatted.values,\n            )) as {lastMutationID: bigint}[];\n\n            return {lastMutationID};\n          },\n\n          async writeMutationResult(result) {\n            const formatted = formatPg(\n              sql`INSERT INTO ${sql.ident(upstreamSchema)}.mutations\n                    (\"clientGroupID\", \"clientID\", \"mutationID\", \"result\")\n                VALUES (${clientGroupID}, ${result.id.clientID}, ${result.id.id}, ${JSON.stringify(\n                  result.result,\n                )}::text::json)`,\n            );\n            await dbTx.query(formatted.text, formatted.values);\n          },\n\n          async deleteMutationResults(args: CleanupResultsArg) {\n            if ('type' in args && args.type === 'bulk') {\n              // Bulk deletion: delete all mutations for multiple clients\n              const formatted = formatPg(\n                sql`DELETE FROM ${sql.ident(upstreamSchema)}.\"mutations\"\n                  WHERE \"clientGroupID\" = ${args.clientGroupID}\n                    AND \"clientID\" = ANY(${args.clientIDs})`,\n              );\n              await dbTx.query(formatted.text, formatted.values);\n            } else {\n              // Single client (explicit 'single' or legacy without type): delete up to mutation ID\n              const formatted = formatPg(\n                sql`DELETE FROM ${sql.ident(upstreamSchema)}.\"mutations\"\n                  WHERE \"clientGroupID\" = ${args.clientGroupID}\n                    AND \"clientID\" = ${args.clientID}\n                    AND \"mutationID\" <= ${args.upToMutationID}`,\n              );\n              await dbTx.query(formatted.text, formatted.values);\n            }\n          },\n        });\n      } finally {\n        mutatorTx.close();\n      }\n    });\n  }\n\n  #makeServerTransaction(\n    dbTx: DBTransaction<TWrappedTransaction>,\n    clientID: string,\n    mutationID: number,\n    addRetryPredicate: ((shouldRetry: RetryPredicate) => void) | undefined,\n  ) {\n    return this.#crudFactory.createTransaction(\n      dbTx,\n      clientID,\n      mutationID,\n      addRetryPredicate,\n    );\n  }\n\n  /**\n   * Runs a single read query.\n   *\n   * When the {@link DBConnection} implements `query`, the compiled SQL is\n   * issued as a bare statement with no `BEGIN`/`COMMIT` around it (plus a\n   * one-time server schema lookup the first time this instance touches the\n   * database). Postgres then releases the statement's locks the moment it\n   * finishes instead of waiting for a `COMMIT` round-trip, which matters in\n   * serverless environments where the process can be frozen or reclaimed\n   * between the query resolving and the `COMMIT` being sent, leaving the\n   * transaction and its locks open indefinitely.\n   *\n   * Because no transaction is opened, any per-transaction setup the adapter\n   * performs inside `transaction` does not apply to these reads. If the\n   * connection does not implement `query`, the read is wrapped in a\n   * transaction as before.\n   *\n   * If you need multiple reads to observe a consistent snapshot, use\n   * {@link transaction} and call `tx.run(...)` for each query instead.\n   */\n  run<TTable extends keyof TSchema['tables'] & string, TReturn>(\n    query: Query<TTable, TSchema, TReturn>,\n    _options?: RunOptions,\n  ): Promise<HumanReadable<TReturn>> {\n    const {connection} = this;\n    if (connection.query) {\n      // TS narrows `connection.query`, not `connection` itself.\n      return this.#runOn(connection as ReadTarget<TWrappedTransaction>, query);\n    }\n    return connection.transaction(dbTx => this.#runOn(dbTx, query));\n  }\n\n  async #runOn<TTable extends keyof TSchema['tables'] & string, TReturn>(\n    target: ReadTarget<TWrappedTransaction>,\n    query: Query<TTable, TSchema, TReturn>,\n  ): Promise<HumanReadable<TReturn>> {\n    const {ast, format} = asQueryInternals(query);\n    const serverSchema = await this.#crudFactory.getOrFetchServerSchema(target);\n    return target.runQuery\n      ? target.runQuery<TReturn>(ast, format, this.#schema, serverSchema)\n      : executePostgresQuery<TReturn>(\n          target,\n          ast,\n          format,\n          this.#schema,\n          serverSchema,\n        );\n  }\n}\n\n/**\n * Something a read can be executed against: a `DBTransaction`, or a\n * `DBConnection` that implements `query`.\n */\ntype ReadTarget<TWrappedTransaction> = Queryable & {\n  runQuery?: DBTransaction<TWrappedTransaction>['runQuery'] | undefined;\n};\n"],"mappings":";;;;;;;;;;;;;AAgCA,IAAa,cAAb,MAGqE;CACnE;CACA;CACA;CAEA,YAAY,YAA+C,QAAiB;EAC1E,KAAK,aAAa;EAClB,KAAKA,eAAe,IAAI,mBAAmB,MAAM;EACjD,KAAKC,UAAU;CACjB;CAEA,YACE,UAIA,kBACY;EAGZ,MAAM,EACJ,iBAAiB,IACjB,gBAAgB,IAChB,WAAW,IACX,aAAa,GACb,sBACE,oBAAoB,CAAC;EACzB,OAAO,KAAK,WAAW,YAAY,OAAM,SAAQ;GAK/C,MAAM,YAAY,IAAI,sBAAsB,IAAI;GAChD,IAAI;IAQF,OAAO,MAAM,SAAS,MAPD,KAAKC,uBACxB,WACA,UACA,YACA,iBACF,GAE8B;KAC5B,MAAM,yBAAyB;MAC7B,MAAM,YAAY,SAChB,GAAG,eAAe,IAAI,MAAM,cAAc,EAAE;;kCAExB,cAAc,IAAI,SAAS,IAAI,EAAE;;;+CAIvD;MAEA,MAAM,CAAC,EAAC,oBAAoB,MAAM,KAAK,MACrC,UAAU,MACV,UAAU,MACZ;MAEA,OAAO,EAAC,eAAc;KACxB;KAEA,MAAM,oBAAoB,QAAQ;MAChC,MAAM,YAAY,SAChB,GAAG,eAAe,IAAI,MAAM,cAAc,EAAE;;0BAEhC,cAAc,IAAI,OAAO,GAAG,SAAS,IAAI,OAAO,GAAG,GAAG,IAAI,KAAK,UACvE,OAAO,MACT,EAAE,cACN;MACA,MAAM,KAAK,MAAM,UAAU,MAAM,UAAU,MAAM;KACnD;KAEA,MAAM,sBAAsB,MAAyB;MACnD,IAAI,UAAU,QAAQ,KAAK,SAAS,QAAQ;OAE1C,MAAM,YAAY,SAChB,GAAG,eAAe,IAAI,MAAM,cAAc,EAAE;4CAChB,KAAK,cAAc;2CACpB,KAAK,UAAU,EAC5C;OACA,MAAM,KAAK,MAAM,UAAU,MAAM,UAAU,MAAM;MACnD,OAAO;OAEL,MAAM,YAAY,SAChB,GAAG,eAAe,IAAI,MAAM,cAAc,EAAE;4CAChB,KAAK,cAAc;uCACxB,KAAK,SAAS;0CACX,KAAK,gBACjC;OACA,MAAM,KAAK,MAAM,UAAU,MAAM,UAAU,MAAM;MACnD;KACF;IACF,CAAC;GACH,UAAU;IACR,UAAU,MAAM;GAClB;EACF,CAAC;CACH;CAEA,uBACE,MACA,UACA,YACA,mBACA;EACA,OAAO,KAAKF,aAAa,kBACvB,MACA,UACA,YACA,iBACF;CACF;;;;;;;;;;;;;;;;;;;;;CAsBA,IACE,OACA,UACiC;EACjC,MAAM,EAAC,eAAc;EACrB,IAAI,WAAW,OAEb,OAAO,KAAKG,OAAO,YAA+C,KAAK;EAEzE,OAAO,WAAW,aAAY,SAAQ,KAAKA,OAAO,MAAM,KAAK,CAAC;CAChE;CAEA,MAAMA,OACJ,QACA,OACiC;EACjC,MAAM,EAAC,KAAK,WAAU,iBAAiB,KAAK;EAC5C,MAAM,eAAe,MAAM,KAAKH,aAAa,uBAAuB,MAAM;EAC1E,OAAO,OAAO,WACV,OAAO,SAAkB,KAAK,QAAQ,KAAKC,SAAS,YAAY,IAChE,qBACE,QACA,KACA,QACA,KAAKA,SACL,YACF;CACN;AACF"}
+{"version":3,"file":"zql-database.js","names":[],"sources":["../../../../zero-server/src/zql-database.ts"],"sourcesContent":["import type {MaybePromise} from '../../shared/src/types.ts';\nimport {formatPg, sql} from '../../z2s/src/sql.ts';\nimport type {CleanupResultsArg} from '../../zero-protocol/src/mutation.ts';\nimport type {Schema} from '../../zero-types/src/schema.ts';\nimport type {\n  DBConnection,\n  DBTransaction,\n  Queryable,\n  RetryPredicate,\n} from '../../zql/src/mutate/custom.ts';\nimport {asQueryInternals} from '../../zql/src/query/query-internals.ts';\nimport type {\n  HumanReadable,\n  Query,\n  RunOptions,\n} from '../../zql/src/query/query.ts';\nimport {ClosableDBTransaction} from './closable-db-transaction.ts';\nimport {CRUDMutatorFactory, type TransactionImpl} from './custom.ts';\nimport {executePostgresQuery} from './pg-query-executor.ts';\nimport type {\n  Database,\n  TransactionProviderHooks,\n  TransactionProviderInput,\n} from './process-mutations.ts';\n\n/**\n * Implements a Database for use with PushProcessor that is backed by Postgres.\n *\n * This implementation also implements the same ZQL interfaces for reading and\n * writing data that the Zero client does, so that mutator functions can be\n * shared across client and server.\n */\nexport class ZQLDatabase<\n  TSchema extends Schema,\n  TWrappedTransaction,\n> implements Database<TransactionImpl<TSchema, TWrappedTransaction>> {\n  readonly connection: DBConnection<TWrappedTransaction>;\n  readonly #crudFactory: CRUDMutatorFactory<TSchema>;\n  readonly #schema: TSchema;\n\n  constructor(connection: DBConnection<TWrappedTransaction>, schema: TSchema) {\n    this.connection = connection;\n    this.#crudFactory = new CRUDMutatorFactory(schema);\n    this.#schema = schema;\n  }\n\n  transaction<R>(\n    callback: (\n      tx: TransactionImpl<TSchema, TWrappedTransaction>,\n      transactionHooks: TransactionProviderHooks,\n    ) => MaybePromise<R>,\n    transactionInput?: TransactionProviderInput,\n  ): Promise<R> {\n    // Icky hack. This is just here to have user not have to do this.\n    // These interfaces need to be factored better.\n    const {\n      upstreamSchema = '',\n      clientGroupID = '',\n      clientID = '',\n      mutationID = 0,\n      addRetryPredicate,\n    } = transactionInput ?? {};\n    return this.connection.transaction(async dbTx => {\n      // What the mutator's `tx` uses. Closed once the callback settles, so a\n      // `tx` that outlives its transaction, such as one captured by a\n      // `tx.retryOn()` predicate, throws instead of reaching a connection\n      // that may already be back in the pool.\n      const mutatorTx = new ClosableDBTransaction(dbTx);\n      try {\n        const zeroTx = await this.#makeServerTransaction(\n          mutatorTx,\n          clientID,\n          mutationID,\n          addRetryPredicate,\n        );\n\n        return await callback(zeroTx, {\n          async updateClientMutationID() {\n            const formatted = formatPg(\n              sql`INSERT INTO ${sql.ident(upstreamSchema)}.clients \n                    as current (\"clientGroupID\", \"clientID\", \"lastMutationID\")\n                        VALUES (${clientGroupID}, ${clientID}, ${1})\n                    ON CONFLICT (\"clientGroupID\", \"clientID\")\n                    DO UPDATE SET \"lastMutationID\" = current.\"lastMutationID\" + 1\n                    RETURNING \"lastMutationID\"`,\n            );\n\n            const [{lastMutationID}] = (await dbTx.query(\n              formatted.text,\n              formatted.values,\n            )) as {lastMutationID: bigint}[];\n\n            return {lastMutationID};\n          },\n\n          async writeMutationResult(result) {\n            const formatted = formatPg(\n              sql`INSERT INTO ${sql.ident(upstreamSchema)}.mutations\n                    (\"clientGroupID\", \"clientID\", \"mutationID\", \"result\")\n                VALUES (${clientGroupID}, ${result.id.clientID}, ${result.id.id}, ${JSON.stringify(\n                  result.result,\n                )}::text::json)`,\n            );\n            await dbTx.query(formatted.text, formatted.values);\n          },\n\n          async deleteMutationResults(args: CleanupResultsArg) {\n            if ('type' in args && args.type === 'bulk') {\n              // Bulk deletion: delete all mutations for multiple clients\n              const formatted = formatPg(\n                sql`DELETE FROM ${sql.ident(upstreamSchema)}.\"mutations\"\n                  WHERE \"clientGroupID\" = ${args.clientGroupID}\n                    AND \"clientID\" = ANY(${args.clientIDs})`,\n              );\n              await dbTx.query(formatted.text, formatted.values);\n            } else {\n              // Single client (explicit 'single' or legacy without type): delete up to mutation ID\n              const formatted = formatPg(\n                sql`DELETE FROM ${sql.ident(upstreamSchema)}.\"mutations\"\n                  WHERE \"clientGroupID\" = ${args.clientGroupID}\n                    AND \"clientID\" = ${args.clientID}\n                    AND \"mutationID\" <= ${args.upToMutationID}`,\n              );\n              await dbTx.query(formatted.text, formatted.values);\n            }\n          },\n        });\n      } finally {\n        mutatorTx.close();\n      }\n    });\n  }\n\n  #makeServerTransaction(\n    dbTx: DBTransaction<TWrappedTransaction>,\n    clientID: string,\n    mutationID: number,\n    addRetryPredicate: ((shouldRetry: RetryPredicate) => void) | undefined,\n  ) {\n    return this.#crudFactory.createTransaction(\n      dbTx,\n      clientID,\n      mutationID,\n      addRetryPredicate,\n    );\n  }\n\n  /**\n   * Runs a single read query.\n   *\n   * When the {@link DBConnection} implements `query`, the compiled SQL is\n   * issued as a bare statement with no `BEGIN`/`COMMIT` around it (plus a\n   * one-time server schema lookup the first time this instance touches the\n   * database). Postgres then releases the statement's locks the moment it\n   * finishes instead of waiting for a `COMMIT` round-trip, which matters in\n   * serverless environments where the process can be frozen or reclaimed\n   * between the query resolving and the `COMMIT` being sent, leaving the\n   * transaction and its locks open indefinitely.\n   *\n   * Because no transaction is opened, any per-transaction setup the adapter\n   * performs inside `transaction` does not apply to these reads. If the\n   * connection does not implement `query`, the read is wrapped in a\n   * transaction as before.\n   *\n   * If you need multiple reads to observe a consistent snapshot, use\n   * {@link transaction} and call `tx.run(...)` for each query instead.\n   */\n  run<TTable extends keyof TSchema['tables'] & string, TReturn>(\n    query: Query<TTable, TSchema, TReturn>,\n    _options?: RunOptions,\n  ): Promise<HumanReadable<TReturn>> {\n    const {connection} = this;\n    if (connection.query) {\n      // TS narrows `connection.query`, not `connection` itself.\n      return this.#runOn(connection as ReadTarget<TWrappedTransaction>, query);\n    }\n    return connection.transaction(dbTx => this.#runOn(dbTx, query));\n  }\n\n  async #runOn<TTable extends keyof TSchema['tables'] & string, TReturn>(\n    target: ReadTarget<TWrappedTransaction>,\n    query: Query<TTable, TSchema, TReturn>,\n  ): Promise<HumanReadable<TReturn>> {\n    const {ast, format} = asQueryInternals(query);\n    const serverSchema = await this.#crudFactory.getOrFetchServerSchema(target);\n    return target.runQuery\n      ? target.runQuery<TReturn>(ast, format, this.#schema, serverSchema)\n      : executePostgresQuery<TReturn>(\n          target,\n          ast,\n          format,\n          this.#schema,\n          serverSchema,\n        );\n  }\n}\n\n/**\n * Something a read can be executed against: a `DBTransaction`, or a\n * `DBConnection` that implements `query`.\n */\ntype ReadTarget<TWrappedTransaction> = Queryable & {\n  runQuery?: DBTransaction<TWrappedTransaction>['runQuery'] | undefined;\n};\n"],"mappings":";;;;;;;;;;;;;AAgCA,IAAa,cAAb,MAGqE;CACnE;CACA;CACA;CAEA,YAAY,YAA+C,QAAiB;EAC1E,KAAK,aAAa;EAClB,KAAK,eAAe,IAAI,mBAAmB,MAAM;EACjD,KAAK,UAAU;CACjB;CAEA,YACE,UAIA,kBACY;EAGZ,MAAM,EACJ,iBAAiB,IACjB,gBAAgB,IAChB,WAAW,IACX,aAAa,GACb,sBACE,oBAAoB,CAAC;EACzB,OAAO,KAAK,WAAW,YAAY,OAAM,SAAQ;GAK/C,MAAM,YAAY,IAAI,sBAAsB,IAAI;GAChD,IAAI;IAQF,OAAO,MAAM,SAAS,MAPD,KAAK,uBACxB,WACA,UACA,YACA,iBACF,GAE8B;KAC5B,MAAM,yBAAyB;MAC7B,MAAM,YAAY,SAChB,GAAG,eAAe,IAAI,MAAM,cAAc,EAAE;;kCAExB,cAAc,IAAI,SAAS,IAAI,EAAE;;;+CAIvD;MAEA,MAAM,CAAC,EAAC,oBAAoB,MAAM,KAAK,MACrC,UAAU,MACV,UAAU,MACZ;MAEA,OAAO,EAAC,eAAc;KACxB;KAEA,MAAM,oBAAoB,QAAQ;MAChC,MAAM,YAAY,SAChB,GAAG,eAAe,IAAI,MAAM,cAAc,EAAE;;0BAEhC,cAAc,IAAI,OAAO,GAAG,SAAS,IAAI,OAAO,GAAG,GAAG,IAAI,KAAK,UACvE,OAAO,MACT,EAAE,cACN;MACA,MAAM,KAAK,MAAM,UAAU,MAAM,UAAU,MAAM;KACnD;KAEA,MAAM,sBAAsB,MAAyB;MACnD,IAAI,UAAU,QAAQ,KAAK,SAAS,QAAQ;OAE1C,MAAM,YAAY,SAChB,GAAG,eAAe,IAAI,MAAM,cAAc,EAAE;4CAChB,KAAK,cAAc;2CACpB,KAAK,UAAU,EAC5C;OACA,MAAM,KAAK,MAAM,UAAU,MAAM,UAAU,MAAM;MACnD,OAAO;OAEL,MAAM,YAAY,SAChB,GAAG,eAAe,IAAI,MAAM,cAAc,EAAE;4CAChB,KAAK,cAAc;uCACxB,KAAK,SAAS;0CACX,KAAK,gBACjC;OACA,MAAM,KAAK,MAAM,UAAU,MAAM,UAAU,MAAM;MACnD;KACF;IACF,CAAC;GACH,UAAU;IACR,UAAU,MAAM;GAClB;EACF,CAAC;CACH;CAEA,uBACE,MACA,UACA,YACA,mBACA;EACA,OAAO,KAAK,aAAa,kBACvB,MACA,UACA,YACA,iBACF;CACF;;;;;;;;;;;;;;;;;;;;;CAsBA,IACE,OACA,UACiC;EACjC,MAAM,EAAC,eAAc;EACrB,IAAI,WAAW,OAEb,OAAO,KAAK,OAAO,YAA+C,KAAK;EAEzE,OAAO,WAAW,aAAY,SAAQ,KAAK,OAAO,MAAM,KAAK,CAAC;CAChE;CAEA,MAAM,OACJ,QACA,OACiC;EACjC,MAAM,EAAC,KAAK,WAAU,iBAAiB,KAAK;EAC5C,MAAM,eAAe,MAAM,KAAK,aAAa,uBAAuB,MAAM;EAC1E,OAAO,OAAO,WACV,OAAO,SAAkB,KAAK,QAAQ,KAAK,SAAS,YAAY,IAChE,qBACE,QACA,KACA,QACA,KAAK,SACL,YACF;CACN;AACF"}

package/out/zero-solid/src/bindings.js

@@ -1,12 +1,12 @@
-import "../../shared/src/asserts.js";
-import "../../shared/src/must.js";
-import "../../zql/src/query/query-internals.js";
-import "../../zql/src/query/query-impl.js";
-import "../../zql/src/query/query-registry.js";
-import "../../zql/src/ivm/skip-yields.js";
-import "../../zql/src/ivm/stream.js";
-import "../../zql/src/ivm/view-apply-change.js";
-import "../../zql/src/query/ttl.js";
-import "../../shared/src/deep-clone.js";
+import { assert, unreachable } from "../../shared/src/asserts.js";
+import { must } from "../../shared/src/must.js";
+import { asQueryInternals, queryInternalsTag } from "../../zql/src/query/query-internals.js";
+import { newQuery } from "../../zql/src/query/query-impl.js";
+import { addContextToQuery } from "../../zql/src/query/query-registry.js";
+import { skipYields } from "../../zql/src/ivm/skip-yields.js";
+import { consume } from "../../zql/src/ivm/stream.js";
+import { applyChange, idSymbol, refCountSymbol } from "../../zql/src/ivm/view-apply-change.js";
+import { DEFAULT_TTL_MS } from "../../zql/src/query/ttl.js";
+import { deepClone } from "../../shared/src/deep-clone.js";
 import "../../zero-client/src/client/bindings.js";
-export {};
+export { DEFAULT_TTL_MS, addContextToQuery, applyChange, asQueryInternals, assert, consume, deepClone, idSymbol, must, newQuery, queryInternalsTag, refCountSymbol, skipYields, unreachable };

package/out/zero-solid/src/mod.js

@@ -1,5 +1,5 @@
-import "./use-zero.js";
-import "./use-connection-state.js";
-import "./use-query.js";
-import "./use-zero-online.js";
-export {};
+import { ZeroProvider, createUseZero, createZero, useZero } from "./use-zero.js";
+import { useConnectionState } from "./use-connection-state.js";
+import { createQuery, useQuery } from "./use-query.js";
+import { useZeroOnline } from "./use-zero-online.js";
+export { ZeroProvider, createQuery, createUseZero, createZero, useConnectionState, useQuery, useZero, useZeroOnline };

package/out/zero-solid/src/solid-view.js

@@ -219,6 +219,6 @@ function createSolidViewFactory(setState, retry) {
 	return solidViewFactory;
 }
 //#endregion
-export { UNKNOWN, createSolidViewFactory };
+export { CACHED, COMPLETE, SolidView, UNKNOWN, createSolidViewFactory };
 
 //# sourceMappingURL=solid-view.js.map