@rocicorp/zero 1.11.0 → 1.11.1

package/out/zero-cache/src/services/change-source/pg/schema/index-predicate.js

@@ -190,7 +190,7 @@ var Parser = class {
 		if (!this.#take(type)) throw new UnsupportedPredicateError("unsupported-syntax");
 	}
 };
-var unsupportedOperatorWords = new Set([
+var unsupportedOperatorWords = /* @__PURE__ */ new Set([
 	"ALL",
 	"BETWEEN",
 	"ILIKE",
@@ -351,7 +351,7 @@ function resolveLiteral(expression, column) {
 	if (literal.type !== "literal" || literal.value === null) throw new UnsupportedPredicateError("unsupported-syntax");
 	switch (column.category) {
 		case "boolean":
-			validateLiteralCast(cast, new Set(["bool"]));
+			validateLiteralCast(cast, /* @__PURE__ */ new Set(["bool"]));
 			if (literal.literalType !== "boolean") throw new UnsupportedPredicateError("unsupported-type");
 			return {
 				type: "boolean",
@@ -368,7 +368,7 @@ function resolveLiteral(expression, column) {
 				value: canonicalInteger(String(literal.value), column.rank)
 			};
 		case "text":
-			validateLiteralCast(cast, new Set(["text", "varchar"]));
+			validateLiteralCast(cast, /* @__PURE__ */ new Set(["text", "varchar"]));
 			if (literal.literalType !== "string") throw new UnsupportedPredicateError("unsupported-type");
 			return {
 				type: "string",
@@ -382,7 +382,7 @@ function resolveLiteral(expression, column) {
 				value: String(literal.value)
 			};
 		case "uuid": {
-			validateLiteralCast(cast, new Set(["uuid"]));
+			validateLiteralCast(cast, /* @__PURE__ */ new Set(["uuid"]));
 			const value = String(literal.value).toLowerCase();
 			if (literal.literalType !== "string" || !UUID_RE.test(value)) throw new UnsupportedPredicateError("unsupported-type");
 			return {
@@ -529,14 +529,15 @@ function readQuoted(sql, start, quote, escapes) {
 	let value = "";
 	for (let i = start + 1; i < sql.length; i++) {
 		const char = sql[i];
-		if (char === quote) if (sql[i + 1] === quote) {
-			value += quote;
-			i++;
-		} else return {
-			value,
-			end: i + 1
-		};
-		else if (escapes && char === "\\") {
+		if (char === quote) {
+			if (sql[i + 1] === quote) {
+				value += quote;
+				i++;
+			} else return {
+				value,
+				end: i + 1
+			};
+		} else if (escapes && char === "\\") {
 			const escape = sql[++i];
 			if (escape === void 0) throw new UnsupportedPredicateError("unsupported-syntax");
 			if (escape === "x") {

package/out/zero-cache/src/services/change-source/pg/schema/index-predicate.js.map

@@ -1 +1 @@
-{"version":3,"file":"index-predicate.js","names":["#tokens","#parseOr","#parseAnd","#take","#parseNot","#parseComparison","#parseOperand","#position","#parseOperandList","#takeIdentifier","#parseLiteralArray","#expectIdentifier","#parseCastName","#parsePrimary"],"sources":["../../../../../../../../zero-cache/src/services/change-source/pg/schema/index-predicate.ts"],"sourcesContent":["import * as PostgresTypeClass from '../../../../db/postgres-type-class-enum.ts';\nimport type {\n  IndexPredicate,\n  IndexPredicateValue,\n  PublishedTableSpec,\n} from '../../../../db/specs.ts';\n\nconst UUID_RE =\n  /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/;\nconst INTEGER_RE = /^[+-]?\\d+$/;\nconst WHITESPACE_RE = /\\s/;\nconst OPERATOR_TOKEN_RE = /^(?:::|<>|!=|<=|>=|=|<|>)/;\nconst UNSUPPORTED_OPERATOR_TOKEN_RE = /^(?:!~\\*|!~|~\\*|~)/;\nconst INTEGER_TOKEN_RE = /^[+-]?\\d+/;\nconst WORD_TOKEN_RE = /^[A-Za-z_][A-Za-z0-9_$]*/;\nconst SIMPLE_ESCAPES: Readonly<Record<string, string>> = {\n  b: '\\b',\n  f: '\\f',\n  n: '\\n',\n  r: '\\r',\n  t: '\\t',\n  v: '\\v',\n};\n\nexport const unsupportedPredicateReasons = [\n  'unpublished-column',\n  'unsupported-type',\n  'unsupported-operator',\n  'unsupported-function',\n  'unsupported-cast',\n  'unsupported-syntax',\n  'non-deterministic-collation',\n] as const;\n\nexport type UnsupportedPredicateReason =\n  (typeof unsupportedPredicateReasons)[number];\n\nexport type PredicateTranslation =\n  | {readonly supported: true; readonly predicate: IndexPredicate}\n  | {readonly supported: false; readonly reason: UnsupportedPredicateReason};\n\ntype PredicateColumnSpec = PublishedTableSpec['columns'][string] & {\n  readonly collationIsDeterministic?: boolean | null | undefined;\n};\n\ntype PredicateTableSpec = Omit<PublishedTableSpec, 'columns'> & {\n  readonly columns: Readonly<Record<string, PredicateColumnSpec>>;\n};\n\ntype Expr =\n  | {readonly type: 'column'; readonly name: string}\n  | {\n      readonly type: 'literal';\n      readonly value: boolean | string | null;\n      readonly literalType: 'boolean' | 'integer' | 'string' | 'null';\n    }\n  | {readonly type: 'cast'; readonly expression: Expr; readonly cast: string}\n  | {readonly type: 'not'; readonly expression: Expr}\n  | {\n      readonly type: 'and' | 'or';\n      readonly expressions: readonly Expr[];\n    }\n  | {\n      readonly type: 'comparison';\n      readonly left: Expr;\n      readonly op: '=' | '<>' | '!=' | '<' | '<=' | '>' | '>=';\n      readonly right: Expr;\n    }\n  | {\n      readonly type: 'null-test';\n      readonly expression: Expr;\n      readonly not: boolean;\n    };\n\ntype Token =\n  | {readonly type: 'identifier'; readonly value: string}\n  | {\n      readonly type: 'literal';\n      readonly value: boolean | string | null;\n      readonly literalType: 'boolean' | 'integer' | 'string' | 'null';\n    }\n  | {\n      readonly type:\n        | '('\n        | ')'\n        | '['\n        | ']'\n        | ','\n        | '.'\n        | '::'\n        | '='\n        | '<>'\n        | '!='\n        | '<'\n        | '<='\n        | '>'\n        | '>='\n        | 'AND'\n        | 'OR'\n        | 'NOT'\n        | 'IN'\n        | 'IS'\n        | 'EOF';\n    }\n  | {readonly type: 'unsupported'; readonly value: string};\n\nclass UnsupportedPredicateError extends Error {\n  readonly reason: UnsupportedPredicateReason;\n\n  constructor(reason: UnsupportedPredicateReason) {\n    super(reason);\n    this.reason = reason;\n  }\n}\n\nexport function translatePostgresIndexPredicate(\n  sql: string,\n  table: PredicateTableSpec,\n): PredicateTranslation {\n  try {\n    const parser = new Parser(tokenize(sql));\n    const expression = parser.parse();\n    parser.expect('EOF');\n    return {supported: true, predicate: normalize(expression, table)};\n  } catch (error) {\n    return {\n      supported: false,\n      reason:\n        error instanceof UnsupportedPredicateError\n          ? error.reason\n          : 'unsupported-syntax',\n    };\n  }\n}\n\nclass Parser {\n  readonly #tokens: readonly Token[];\n  #position = 0;\n\n  constructor(tokens: readonly Token[]) {\n    this.#tokens = tokens;\n  }\n\n  parse(): Expr {\n    return this.#parseOr();\n  }\n\n  #parseOr(): Expr {\n    const expressions = [this.#parseAnd()];\n    while (this.#take('OR')) expressions.push(this.#parseAnd());\n    return expressions.length === 1\n      ? expressions[0]\n      : {type: 'or', expressions};\n  }\n\n  #parseAnd(): Expr {\n    const expressions = [this.#parseNot()];\n    while (this.#take('AND')) expressions.push(this.#parseNot());\n    return expressions.length === 1\n      ? expressions[0]\n      : {type: 'and', expressions};\n  }\n\n  #parseNot(): Expr {\n    if (this.#take('NOT')) {\n      return {type: 'not', expression: this.#parseNot()};\n    }\n    return this.#parseComparison();\n  }\n\n  #parseComparison(): Expr {\n    const left = this.#parseOperand();\n    if (this.#take('IS')) {\n      const not = this.#take('NOT');\n      const token = this.#tokens[this.#position++];\n      if (token?.type !== 'literal' || token.literalType !== 'null') {\n        throw new UnsupportedPredicateError('unsupported-operator');\n      }\n      return {type: 'null-test', expression: left, not};\n    }\n    if (this.#take('IN')) {\n      this.expect('(');\n      const values = this.#parseOperandList(')');\n      this.expect(')');\n      return comparisons(left, values);\n    }\n    const op = this.#tokens[this.#position]?.type;\n    if (isComparison(op)) {\n      this.#position++;\n      if (op === '=' && this.#takeIdentifier('ANY')) {\n        return comparisons(left, this.#parseLiteralArray());\n      }\n      return {\n        type: 'comparison',\n        left,\n        op,\n        right: this.#parseOperand(),\n      };\n    }\n    const next = this.#tokens[this.#position];\n    if (next?.type === 'identifier' && next.value.toUpperCase() === 'COLLATE') {\n      throw new UnsupportedPredicateError('non-deterministic-collation');\n    }\n    if (\n      next?.type === 'unsupported' &&\n      UNSUPPORTED_OPERATOR_TOKEN_RE.test(next.value)\n    ) {\n      throw new UnsupportedPredicateError('unsupported-operator');\n    }\n    if (\n      next?.type === 'identifier' &&\n      unsupportedOperatorWords.has(next.value.toUpperCase())\n    ) {\n      throw new UnsupportedPredicateError('unsupported-operator');\n    }\n    return left;\n  }\n\n  #parseLiteralArray(): readonly Expr[] {\n    this.expect('(');\n    this.#expectIdentifier('ARRAY');\n    this.expect('[');\n    const values = this.#parseOperandList(']');\n    this.expect(']');\n    if (this.#take('::')) {\n      this.#parseCastName();\n      this.expect('[');\n      this.expect(']');\n    }\n    this.expect(')');\n    return values;\n  }\n\n  #parseOperandList(end: ')' | ']'): readonly Expr[] {\n    const values: Expr[] = [];\n    if (this.#tokens[this.#position]?.type === end) return values;\n    values.push(this.#parseOperand());\n    while (this.#take(',')) values.push(this.#parseOperand());\n    return values;\n  }\n\n  #parseOperand(): Expr {\n    let expression = this.#parsePrimary();\n    while (this.#take('::')) {\n      expression = {\n        type: 'cast',\n        expression,\n        cast: this.#parseCastName(),\n      };\n    }\n    return expression;\n  }\n\n  #parsePrimary(): Expr {\n    if (this.#take('(')) {\n      const expression = this.#parseOr();\n      this.expect(')');\n      return expression;\n    }\n    const token = this.#tokens[this.#position++];\n    if (token?.type === 'literal') {\n      return {\n        type: 'literal',\n        value: token.value,\n        literalType: token.literalType,\n      };\n    }\n    if (token?.type === 'identifier') {\n      if (this.#tokens[this.#position]?.type === '(') {\n        throw new UnsupportedPredicateError('unsupported-function');\n      }\n      if (this.#tokens[this.#position]?.type === '.') {\n        throw new UnsupportedPredicateError('unsupported-syntax');\n      }\n      if (token.value.toUpperCase() === 'COLLATE') {\n        throw new UnsupportedPredicateError('non-deterministic-collation');\n      }\n      return {type: 'column', name: token.value};\n    }\n    throw new UnsupportedPredicateError('unsupported-syntax');\n  }\n\n  #parseCastName(): string {\n    const first = this.#expectIdentifier();\n    const parts = [first];\n    let separator = '.';\n    while (this.#take('.')) parts.push(this.#expectIdentifier());\n    const next = this.#tokens[this.#position];\n    if (\n      first.toLowerCase() === 'character' &&\n      next?.type === 'identifier' &&\n      next.value.toLowerCase() === 'varying'\n    ) {\n      parts.push(this.#expectIdentifier());\n      separator = ' ';\n    }\n    if (this.#tokens[this.#position]?.type === '(') {\n      throw new UnsupportedPredicateError('unsupported-cast');\n    }\n    return parts.join(separator);\n  }\n\n  #expectIdentifier(expected?: string): string {\n    const token = this.#tokens[this.#position++];\n    if (\n      token?.type !== 'identifier' ||\n      (expected !== undefined && token.value.toUpperCase() !== expected)\n    ) {\n      throw new UnsupportedPredicateError('unsupported-cast');\n    }\n    return token.value;\n  }\n\n  #takeIdentifier(expected: string): boolean {\n    const token = this.#tokens[this.#position];\n    if (\n      token?.type !== 'identifier' ||\n      token.value.toUpperCase() !== expected\n    ) {\n      return false;\n    }\n    this.#position++;\n    return true;\n  }\n\n  #take(type: Token['type']): boolean {\n    if (this.#tokens[this.#position]?.type !== type) return false;\n    this.#position++;\n    return true;\n  }\n\n  expect(type: Token['type']): void {\n    if (!this.#take(type)) {\n      throw new UnsupportedPredicateError('unsupported-syntax');\n    }\n  }\n}\n\nconst unsupportedOperatorWords = new Set([\n  'ALL',\n  'BETWEEN',\n  'ILIKE',\n  'LIKE',\n  'SIMILAR',\n]);\n\nfunction comparisons(left: Expr, values: readonly Expr[]): Expr {\n  if (values.length === 0) {\n    throw new UnsupportedPredicateError('unsupported-syntax');\n  }\n  const expressions = values.map(\n    right => ({type: 'comparison', left, op: '=', right}) as const,\n  );\n  return expressions.length === 1 ? expressions[0] : {type: 'or', expressions};\n}\n\nfunction normalize(\n  expression: Expr,\n  table: PredicateTableSpec,\n): IndexPredicate {\n  switch (expression.type) {\n    case 'and':\n    case 'or': {\n      const conditions = expression.expressions.map(expr =>\n        normalize(expr, table),\n      );\n      return {\n        type: expression.type,\n        conditions: conditions.flatMap(condition =>\n          condition.type === expression.type\n            ? condition.conditions\n            : [condition],\n        ),\n      };\n    }\n    case 'not': {\n      const booleanColumn = resolveBareBooleanColumn(\n        expression.expression,\n        table,\n      );\n      if (booleanColumn) {\n        return booleanComparison(booleanColumn, false);\n      }\n      return {type: 'not', condition: normalize(expression.expression, table)};\n    }\n    case 'column': {\n      const column = resolveColumn(expression, table);\n      if (column.category !== 'boolean') {\n        throw new UnsupportedPredicateError('unsupported-type');\n      }\n      return booleanComparison(column.name, true);\n    }\n    case 'cast': {\n      const column = resolveColumnOperand(expression, table);\n      if (column.category !== 'boolean') {\n        throw new UnsupportedPredicateError('unsupported-type');\n      }\n      return booleanComparison(column.name, true);\n    }\n    case 'null-test': {\n      const column = resolveNullTestColumn(expression.expression, table);\n      return {\n        type: 'null-test',\n        column,\n        op: expression.not ? 'IS NOT NULL' : 'IS NULL',\n      };\n    }\n    case 'comparison':\n      return normalizeComparison(expression, table);\n    case 'literal':\n      throw new UnsupportedPredicateError('unsupported-syntax');\n  }\n}\n\nfunction resolveNullTestColumn(\n  expression: Expr,\n  table: PredicateTableSpec,\n): string {\n  if (expression.type === 'cast') {\n    return resolveColumnOperand(expression, table).name;\n  }\n  if (expression.type === 'column' && !table.columns[expression.name]) {\n    throw new UnsupportedPredicateError('unpublished-column');\n  }\n  if (expression.type === 'column') return expression.name;\n  throw new UnsupportedPredicateError('unsupported-syntax');\n}\n\nfunction resolveBareBooleanColumn(\n  expression: Expr,\n  table: PredicateTableSpec,\n): string | undefined {\n  if (expression.type !== 'column' && expression.type !== 'cast') {\n    return undefined;\n  }\n  const column = resolveColumnOperand(expression, table);\n  return column.category === 'boolean' ? column.name : undefined;\n}\n\nfunction booleanComparison(column: string, value: boolean): IndexPredicate {\n  return {\n    type: 'comparison',\n    column,\n    op: '=',\n    value: {type: 'boolean', value},\n  };\n}\n\ntype ColumnCategory =\n  | {readonly category: 'boolean'; readonly name: string}\n  | {\n      readonly category: 'integer';\n      readonly name: string;\n      readonly rank: 0 | 1 | 2;\n    }\n  | {readonly category: 'text'; readonly name: string}\n  | {\n      readonly category: 'enum';\n      readonly name: string;\n      readonly enumName: string;\n    }\n  | {readonly category: 'uuid'; readonly name: string};\n\nfunction normalizeComparison(\n  expression: Extract<Expr, {type: 'comparison'}>,\n  table: PredicateTableSpec,\n): IndexPredicate {\n  const column = resolveColumnOperand(expression.left, table);\n  const op = expression.op === '!=' ? '<>' : expression.op;\n  if (column.category !== 'integer' && op !== '=' && op !== '<>') {\n    throw new UnsupportedPredicateError('unsupported-operator');\n  }\n  const value = resolveLiteral(expression.right, column);\n  return {type: 'comparison', column: column.name, op, value};\n}\n\nfunction resolveColumnOperand(\n  expression: Expr,\n  table: PredicateTableSpec,\n): ColumnCategory {\n  if (expression.type === 'column') return resolveColumn(expression, table);\n  if (expression.type !== 'cast') {\n    throw new UnsupportedPredicateError('unsupported-syntax');\n  }\n  const column = resolveColumnOperand(expression.expression, table);\n  const cast = canonicalTypeName(expression.cast);\n  switch (column.category) {\n    case 'integer': {\n      const rank = integerRank(cast);\n      if (rank === undefined || rank < column.rank) {\n        throw new UnsupportedPredicateError('unsupported-cast');\n      }\n      return {...column, rank};\n    }\n    case 'text':\n      if (cast !== 'text' && cast !== 'varchar') {\n        throw new UnsupportedPredicateError('unsupported-cast');\n      }\n      return column;\n    case 'boolean':\n      if (cast !== 'bool') {\n        throw new UnsupportedPredicateError('unsupported-cast');\n      }\n      return column;\n    case 'uuid':\n      if (cast !== 'uuid') {\n        throw new UnsupportedPredicateError('unsupported-cast');\n      }\n      return column;\n    case 'enum':\n      if (lastTypeName(expression.cast) !== column.enumName.toLowerCase()) {\n        throw new UnsupportedPredicateError('unsupported-cast');\n      }\n      return column;\n  }\n}\n\nfunction resolveColumn(\n  expression: Extract<Expr, {type: 'column'}>,\n  table: PredicateTableSpec,\n): ColumnCategory {\n  const spec = table.columns[expression.name];\n  if (!spec) throw new UnsupportedPredicateError('unpublished-column');\n  const dataType = canonicalTypeName(spec.dataType);\n  if (dataType === 'bool') return {category: 'boolean', name: expression.name};\n  const rank = integerRank(dataType);\n  if (rank !== undefined) {\n    return {category: 'integer', name: expression.name, rank};\n  }\n  if (dataType === 'text' || dataType === 'varchar') {\n    if (spec.collationIsDeterministic === false) {\n      throw new UnsupportedPredicateError('non-deterministic-collation');\n    }\n    return {category: 'text', name: expression.name};\n  }\n  if (dataType === 'uuid') return {category: 'uuid', name: expression.name};\n  if (spec.pgTypeClass === PostgresTypeClass.Enum) {\n    return {\n      category: 'enum',\n      name: expression.name,\n      enumName: spec.dataType.toLowerCase(),\n    };\n  }\n  throw new UnsupportedPredicateError('unsupported-type');\n}\n\nfunction resolveLiteral(\n  expression: Expr,\n  column: ColumnCategory,\n): IndexPredicateValue {\n  let literal = expression;\n  let cast: string | undefined;\n  while (literal.type === 'cast') {\n    cast = literal.cast;\n    literal = literal.expression;\n  }\n  if (literal.type !== 'literal' || literal.value === null) {\n    throw new UnsupportedPredicateError('unsupported-syntax');\n  }\n\n  switch (column.category) {\n    case 'boolean': {\n      validateLiteralCast(cast, new Set(['bool']));\n      if (literal.literalType !== 'boolean') {\n        throw new UnsupportedPredicateError('unsupported-type');\n      }\n      return {type: 'boolean', value: literal.value as boolean};\n    }\n    case 'integer': {\n      if (cast !== undefined) {\n        const castRank = integerRank(canonicalTypeName(cast));\n        if (castRank === undefined || castRank > column.rank) {\n          throw new UnsupportedPredicateError('unsupported-cast');\n        }\n      }\n      if (\n        literal.literalType !== 'integer' &&\n        !(literal.literalType === 'string' && cast !== undefined)\n      ) {\n        throw new UnsupportedPredicateError('unsupported-type');\n      }\n      return {\n        type: 'integer',\n        value: canonicalInteger(String(literal.value), column.rank),\n      };\n    }\n    case 'text':\n      validateLiteralCast(cast, new Set(['text', 'varchar']));\n      if (literal.literalType !== 'string') {\n        throw new UnsupportedPredicateError('unsupported-type');\n      }\n      return {type: 'string', value: String(literal.value)};\n    case 'enum':\n      if (\n        cast !== undefined &&\n        lastTypeName(cast) !== column.enumName.toLowerCase()\n      ) {\n        throw new UnsupportedPredicateError('unsupported-cast');\n      }\n      if (literal.literalType !== 'string') {\n        throw new UnsupportedPredicateError('unsupported-type');\n      }\n      return {type: 'string', value: String(literal.value)};\n    case 'uuid': {\n      validateLiteralCast(cast, new Set(['uuid']));\n      const value = String(literal.value).toLowerCase();\n      if (literal.literalType !== 'string' || !UUID_RE.test(value)) {\n        throw new UnsupportedPredicateError('unsupported-type');\n      }\n      return {type: 'string', value};\n    }\n  }\n}\n\nfunction validateLiteralCast(\n  cast: string | undefined,\n  allowed: ReadonlySet<string>,\n): void {\n  if (cast !== undefined && !allowed.has(canonicalTypeName(cast))) {\n    throw new UnsupportedPredicateError('unsupported-cast');\n  }\n}\n\nfunction canonicalInteger(value: string, rank: 0 | 1 | 2): string {\n  if (!INTEGER_RE.test(value)) {\n    throw new UnsupportedPredicateError('unsupported-type');\n  }\n  let integer: bigint;\n  try {\n    integer = BigInt(value);\n  } catch {\n    throw new UnsupportedPredicateError('unsupported-type');\n  }\n  const bounds = [\n    [-32768n, 32767n],\n    [-2147483648n, 2147483647n],\n    [-9223372036854775808n, 9223372036854775807n],\n  ] as const;\n  if (integer < bounds[rank][0] || integer > bounds[rank][1]) {\n    throw new UnsupportedPredicateError('unsupported-type');\n  }\n  return integer.toString();\n}\n\nfunction canonicalTypeName(type: string): string {\n  const name = lastTypeName(type).replace(/\\s+/g, ' ');\n  switch (name) {\n    case 'boolean':\n      return 'bool';\n    case 'smallint':\n      return 'int2';\n    case 'integer':\n      return 'int4';\n    case 'bigint':\n      return 'int8';\n    case 'character varying':\n      return 'varchar';\n    default:\n      return name;\n  }\n}\n\nfunction lastTypeName(type: string): string {\n  return type.split('.').at(-1)?.toLowerCase() ?? '';\n}\n\nfunction integerRank(type: string): 0 | 1 | 2 | undefined {\n  switch (type) {\n    case 'int2':\n      return 0;\n    case 'int4':\n      return 1;\n    case 'int8':\n      return 2;\n    default:\n      return undefined;\n  }\n}\n\nfunction isComparison(\n  type: Token['type'] | undefined,\n): type is '=' | '<>' | '!=' | '<' | '<=' | '>' | '>=' {\n  return (\n    type === '=' ||\n    type === '<>' ||\n    type === '!=' ||\n    type === '<' ||\n    type === '<=' ||\n    type === '>' ||\n    type === '>='\n  );\n}\n\nfunction tokenize(sql: string): readonly Token[] {\n  const tokens: Token[] = [];\n  for (let i = 0; i < sql.length;) {\n    const char = sql[i];\n    if (WHITESPACE_RE.test(char)) {\n      i++;\n      continue;\n    }\n    if (char === '\"') {\n      const result = readQuoted(sql, i, '\"', false);\n      tokens.push({type: 'identifier', value: result.value});\n      i = result.end;\n      continue;\n    }\n    const escapeString = (char === 'E' || char === 'e') && sql[i + 1] === \"'\";\n    if (char === \"'\" || escapeString) {\n      const result = readQuoted(\n        sql,\n        i + (escapeString ? 1 : 0),\n        \"'\",\n        escapeString,\n      );\n      tokens.push({\n        type: 'literal',\n        literalType: 'string',\n        value: result.value,\n      });\n      i = result.end;\n      continue;\n    }\n    const operator = OPERATOR_TOKEN_RE.exec(sql.slice(i))?.[0];\n    if (operator) {\n      tokens.push({\n        type: operator as '::' | '=' | '<>' | '!=' | '<' | '<=' | '>' | '>=',\n      });\n      i += operator.length;\n      continue;\n    }\n    const unsupportedOperator = UNSUPPORTED_OPERATOR_TOKEN_RE.exec(\n      sql.slice(i),\n    )?.[0];\n    if (unsupportedOperator) {\n      tokens.push({type: 'unsupported', value: unsupportedOperator});\n      i += unsupportedOperator.length;\n      continue;\n    }\n    if (\n      char === '(' ||\n      char === ')' ||\n      char === '[' ||\n      char === ']' ||\n      char === ',' ||\n      char === '.'\n    ) {\n      tokens.push({type: char});\n      i++;\n      continue;\n    }\n    const integer = INTEGER_TOKEN_RE.exec(sql.slice(i))?.[0];\n    if (integer) {\n      tokens.push({type: 'literal', literalType: 'integer', value: integer});\n      i += integer.length;\n      continue;\n    }\n    const word = WORD_TOKEN_RE.exec(sql.slice(i))?.[0];\n    if (word) {\n      const keyword = word.toUpperCase();\n      if (unsupportedOperatorWords.has(keyword)) {\n        throw new UnsupportedPredicateError('unsupported-operator');\n      }\n      if (keyword === 'TRUE' || keyword === 'FALSE') {\n        tokens.push({\n          type: 'literal',\n          literalType: 'boolean',\n          value: keyword === 'TRUE',\n        });\n      } else if (keyword === 'NULL') {\n        tokens.push({type: 'literal', literalType: 'null', value: null});\n      } else if (\n        keyword === 'AND' ||\n        keyword === 'OR' ||\n        keyword === 'NOT' ||\n        keyword === 'IN' ||\n        keyword === 'IS'\n      ) {\n        tokens.push({type: keyword});\n      } else {\n        tokens.push({type: 'identifier', value: word.toLowerCase()});\n      }\n      i += word.length;\n      continue;\n    }\n    tokens.push({type: 'unsupported', value: char});\n    i++;\n  }\n  tokens.push({type: 'EOF'});\n  return tokens;\n}\n\nfunction readQuoted(\n  sql: string,\n  start: number,\n  quote: '\"' | \"'\",\n  escapes: boolean,\n): {readonly value: string; readonly end: number} {\n  let value = '';\n  for (let i = start + 1; i < sql.length; i++) {\n    const char = sql[i];\n    if (char === quote) {\n      if (sql[i + 1] === quote) {\n        value += quote;\n        i++;\n      } else {\n        return {value, end: i + 1};\n      }\n    } else if (escapes && char === '\\\\') {\n      const escape = sql[++i];\n      if (escape === undefined) {\n        throw new UnsupportedPredicateError('unsupported-syntax');\n      }\n      if (escape === 'x') {\n        const hex = readEscapeDigits(sql, i + 1, 2, 16, 1);\n        value += String.fromCharCode(hex.value);\n        i = hex.end - 1;\n      } else if (escape === 'u' || escape === 'U') {\n        const hex = readEscapeDigits(sql, i + 1, escape === 'u' ? 4 : 8, 16);\n        try {\n          value += String.fromCodePoint(hex.value);\n        } catch {\n          throw new UnsupportedPredicateError('unsupported-syntax');\n        }\n        i = hex.end - 1;\n      } else if (escape >= '0' && escape <= '7') {\n        let octal = escape;\n        while (\n          octal.length < 3 &&\n          sql[i + 1] !== undefined &&\n          sql[i + 1] >= '0' &&\n          sql[i + 1] <= '7'\n        ) {\n          octal += sql[++i];\n        }\n        value += String.fromCharCode(Number.parseInt(octal, 8));\n      } else if (escape === '\\n') {\n        // A backslash-newline pair is a line continuation.\n      } else if (escape === '\\r' && sql[i + 1] === '\\n') {\n        i++;\n      } else {\n        value += SIMPLE_ESCAPES[escape] ?? escape;\n      }\n    } else {\n      value += char;\n    }\n  }\n  throw new UnsupportedPredicateError('unsupported-syntax');\n}\n\nfunction readEscapeDigits(\n  sql: string,\n  start: number,\n  length: number,\n  radix: 16,\n  minLength = length,\n): {readonly value: number; readonly end: number} {\n  let end = start;\n  while (end < start + length) {\n    const digit = sql[end];\n    if (\n      digit === undefined ||\n      !(\n        (digit >= '0' && digit <= '9') ||\n        (digit.toLowerCase() >= 'a' && digit.toLowerCase() <= 'f')\n      )\n    ) {\n      break;\n    }\n    end++;\n  }\n  const digits = sql.slice(start, end);\n  if (digits.length < minLength) {\n    throw new UnsupportedPredicateError('unsupported-syntax');\n  }\n  return {value: Number.parseInt(digits, radix), end};\n}\n"],"mappings":";AAOA,IAAM,UACJ;AACF,IAAM,aAAa;AACnB,IAAM,gBAAgB;AACtB,IAAM,oBAAoB;AAC1B,IAAM,gCAAgC;AACtC,IAAM,mBAAmB;AACzB,IAAM,gBAAgB;AACtB,IAAM,iBAAmD;CACvD,GAAG;CACH,GAAG;CACH,GAAG;CACH,GAAG;CACH,GAAG;CACH,GAAG;AACL;AAoFA,IAAM,4BAAN,cAAwC,MAAM;CAC5C;CAEA,YAAY,QAAoC;EAC9C,MAAM,MAAM;EACZ,KAAK,SAAS;CAChB;AACF;AAEA,SAAgB,gCACd,KACA,OACsB;CACtB,IAAI;EACF,MAAM,SAAS,IAAI,OAAO,SAAS,GAAG,CAAC;EACvC,MAAM,aAAa,OAAO,MAAM;EAChC,OAAO,OAAO,KAAK;EACnB,OAAO;GAAC,WAAW;GAAM,WAAW,UAAU,YAAY,KAAK;EAAC;CAClE,SAAS,OAAO;EACd,OAAO;GACL,WAAW;GACX,QACE,iBAAiB,4BACb,MAAM,SACN;EACR;CACF;AACF;AAEA,IAAM,SAAN,MAAa;CACX;CACA,YAAY;CAEZ,YAAY,QAA0B;EACpC,KAAKA,UAAU;CACjB;CAEA,QAAc;EACZ,OAAO,KAAKC,SAAS;CACvB;CAEA,WAAiB;EACf,MAAM,cAAc,CAAC,KAAKC,UAAU,CAAC;EACrC,OAAO,KAAKC,MAAM,IAAI,GAAG,YAAY,KAAK,KAAKD,UAAU,CAAC;EAC1D,OAAO,YAAY,WAAW,IAC1B,YAAY,KACZ;GAAC,MAAM;GAAM;EAAW;CAC9B;CAEA,YAAkB;EAChB,MAAM,cAAc,CAAC,KAAKE,UAAU,CAAC;EACrC,OAAO,KAAKD,MAAM,KAAK,GAAG,YAAY,KAAK,KAAKC,UAAU,CAAC;EAC3D,OAAO,YAAY,WAAW,IAC1B,YAAY,KACZ;GAAC,MAAM;GAAO;EAAW;CAC/B;CAEA,YAAkB;EAChB,IAAI,KAAKD,MAAM,KAAK,GAClB,OAAO;GAAC,MAAM;GAAO,YAAY,KAAKC,UAAU;EAAC;EAEnD,OAAO,KAAKC,iBAAiB;CAC/B;CAEA,mBAAyB;EACvB,MAAM,OAAO,KAAKC,cAAc;EAChC,IAAI,KAAKH,MAAM,IAAI,GAAG;GACpB,MAAM,MAAM,KAAKA,MAAM,KAAK;GAC5B,MAAM,QAAQ,KAAKH,QAAQ,KAAKO;GAChC,IAAI,OAAO,SAAS,aAAa,MAAM,gBAAgB,QACrD,MAAM,IAAI,0BAA0B,sBAAsB;GAE5D,OAAO;IAAC,MAAM;IAAa,YAAY;IAAM;GAAG;EAClD;EACA,IAAI,KAAKJ,MAAM,IAAI,GAAG;GACpB,KAAK,OAAO,GAAG;GACf,MAAM,SAAS,KAAKK,kBAAkB,GAAG;GACzC,KAAK,OAAO,GAAG;GACf,OAAO,YAAY,MAAM,MAAM;EACjC;EACA,MAAM,KAAK,KAAKR,QAAQ,KAAKO,YAAY;EACzC,IAAI,aAAa,EAAE,GAAG;GACpB,KAAKA;GACL,IAAI,OAAO,OAAO,KAAKE,gBAAgB,KAAK,GAC1C,OAAO,YAAY,MAAM,KAAKC,mBAAmB,CAAC;GAEpD,OAAO;IACL,MAAM;IACN;IACA;IACA,OAAO,KAAKJ,cAAc;GAC5B;EACF;EACA,MAAM,OAAO,KAAKN,QAAQ,KAAKO;EAC/B,IAAI,MAAM,SAAS,gBAAgB,KAAK,MAAM,YAAY,MAAM,WAC9D,MAAM,IAAI,0BAA0B,6BAA6B;EAEnE,IACE,MAAM,SAAS,iBACf,8BAA8B,KAAK,KAAK,KAAK,GAE7C,MAAM,IAAI,0BAA0B,sBAAsB;EAE5D,IACE,MAAM,SAAS,gBACf,yBAAyB,IAAI,KAAK,MAAM,YAAY,CAAC,GAErD,MAAM,IAAI,0BAA0B,sBAAsB;EAE5D,OAAO;CACT;CAEA,qBAAsC;EACpC,KAAK,OAAO,GAAG;EACf,KAAKI,kBAAkB,OAAO;EAC9B,KAAK,OAAO,GAAG;EACf,MAAM,SAAS,KAAKH,kBAAkB,GAAG;EACzC,KAAK,OAAO,GAAG;EACf,IAAI,KAAKL,MAAM,IAAI,GAAG;GACpB,KAAKS,eAAe;GACpB,KAAK,OAAO,GAAG;GACf,KAAK,OAAO,GAAG;EACjB;EACA,KAAK,OAAO,GAAG;EACf,OAAO;CACT;CAEA,kBAAkB,KAAiC;EACjD,MAAM,SAAiB,CAAC;EACxB,IAAI,KAAKZ,QAAQ,KAAKO,YAAY,SAAS,KAAK,OAAO;EACvD,OAAO,KAAK,KAAKD,cAAc,CAAC;EAChC,OAAO,KAAKH,MAAM,GAAG,GAAG,OAAO,KAAK,KAAKG,cAAc,CAAC;EACxD,OAAO;CACT;CAEA,gBAAsB;EACpB,IAAI,aAAa,KAAKO,cAAc;EACpC,OAAO,KAAKV,MAAM,IAAI,GACpB,aAAa;GACX,MAAM;GACN;GACA,MAAM,KAAKS,eAAe;EAC5B;EAEF,OAAO;CACT;CAEA,gBAAsB;EACpB,IAAI,KAAKT,MAAM,GAAG,GAAG;GACnB,MAAM,aAAa,KAAKF,SAAS;GACjC,KAAK,OAAO,GAAG;GACf,OAAO;EACT;EACA,MAAM,QAAQ,KAAKD,QAAQ,KAAKO;EAChC,IAAI,OAAO,SAAS,WAClB,OAAO;GACL,MAAM;GACN,OAAO,MAAM;GACb,aAAa,MAAM;EACrB;EAEF,IAAI,OAAO,SAAS,cAAc;GAChC,IAAI,KAAKP,QAAQ,KAAKO,YAAY,SAAS,KACzC,MAAM,IAAI,0BAA0B,sBAAsB;GAE5D,IAAI,KAAKP,QAAQ,KAAKO,YAAY,SAAS,KACzC,MAAM,IAAI,0BAA0B,oBAAoB;GAE1D,IAAI,MAAM,MAAM,YAAY,MAAM,WAChC,MAAM,IAAI,0BAA0B,6BAA6B;GAEnE,OAAO;IAAC,MAAM;IAAU,MAAM,MAAM;GAAK;EAC3C;EACA,MAAM,IAAI,0BAA0B,oBAAoB;CAC1D;CAEA,iBAAyB;EACvB,MAAM,QAAQ,KAAKI,kBAAkB;EACrC,MAAM,QAAQ,CAAC,KAAK;EACpB,IAAI,YAAY;EAChB,OAAO,KAAKR,MAAM,GAAG,GAAG,MAAM,KAAK,KAAKQ,kBAAkB,CAAC;EAC3D,MAAM,OAAO,KAAKX,QAAQ,KAAKO;EAC/B,IACE,MAAM,YAAY,MAAM,eACxB,MAAM,SAAS,gBACf,KAAK,MAAM,YAAY,MAAM,WAC7B;GACA,MAAM,KAAK,KAAKI,kBAAkB,CAAC;GACnC,YAAY;EACd;EACA,IAAI,KAAKX,QAAQ,KAAKO,YAAY,SAAS,KACzC,MAAM,IAAI,0BAA0B,kBAAkB;EAExD,OAAO,MAAM,KAAK,SAAS;CAC7B;CAEA,kBAAkB,UAA2B;EAC3C,MAAM,QAAQ,KAAKP,QAAQ,KAAKO;EAChC,IACE,OAAO,SAAS,gBACf,aAAa,KAAA,KAAa,MAAM,MAAM,YAAY,MAAM,UAEzD,MAAM,IAAI,0BAA0B,kBAAkB;EAExD,OAAO,MAAM;CACf;CAEA,gBAAgB,UAA2B;EACzC,MAAM,QAAQ,KAAKP,QAAQ,KAAKO;EAChC,IACE,OAAO,SAAS,gBAChB,MAAM,MAAM,YAAY,MAAM,UAE9B,OAAO;EAET,KAAKA;EACL,OAAO;CACT;CAEA,MAAM,MAA8B;EAClC,IAAI,KAAKP,QAAQ,KAAKO,YAAY,SAAS,MAAM,OAAO;EACxD,KAAKA;EACL,OAAO;CACT;CAEA,OAAO,MAA2B;EAChC,IAAI,CAAC,KAAKJ,MAAM,IAAI,GAClB,MAAM,IAAI,0BAA0B,oBAAoB;CAE5D;AACF;AAEA,IAAM,2BAA2B,IAAI,IAAI;CACvC;CACA;CACA;CACA;CACA;AACF,CAAC;AAED,SAAS,YAAY,MAAY,QAA+B;CAC9D,IAAI,OAAO,WAAW,GACpB,MAAM,IAAI,0BAA0B,oBAAoB;CAE1D,MAAM,cAAc,OAAO,KACzB,WAAU;EAAC,MAAM;EAAc;EAAM,IAAI;EAAK;CAAK,EACrD;CACA,OAAO,YAAY,WAAW,IAAI,YAAY,KAAK;EAAC,MAAM;EAAM;CAAW;AAC7E;AAEA,SAAS,UACP,YACA,OACgB;CAChB,QAAQ,WAAW,MAAnB;EACE,KAAK;EACL,KAAK,MAAM;GACT,MAAM,aAAa,WAAW,YAAY,KAAI,SAC5C,UAAU,MAAM,KAAK,CACvB;GACA,OAAO;IACL,MAAM,WAAW;IACjB,YAAY,WAAW,SAAQ,cAC7B,UAAU,SAAS,WAAW,OAC1B,UAAU,aACV,CAAC,SAAS,CAChB;GACF;EACF;EACA,KAAK,OAAO;GACV,MAAM,gBAAgB,yBACpB,WAAW,YACX,KACF;GACA,IAAI,eACF,OAAO,kBAAkB,eAAe,KAAK;GAE/C,OAAO;IAAC,MAAM;IAAO,WAAW,UAAU,WAAW,YAAY,KAAK;GAAC;EACzE;EACA,KAAK,UAAU;GACb,MAAM,SAAS,cAAc,YAAY,KAAK;GAC9C,IAAI,OAAO,aAAa,WACtB,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO,kBAAkB,OAAO,MAAM,IAAI;EAC5C;EACA,KAAK,QAAQ;GACX,MAAM,SAAS,qBAAqB,YAAY,KAAK;GACrD,IAAI,OAAO,aAAa,WACtB,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO,kBAAkB,OAAO,MAAM,IAAI;EAC5C;EACA,KAAK,aAEH,OAAO;GACL,MAAM;GACN,QAHa,sBAAsB,WAAW,YAAY,KAG1D;GACA,IAAI,WAAW,MAAM,gBAAgB;EACvC;EAEF,KAAK,cACH,OAAO,oBAAoB,YAAY,KAAK;EAC9C,KAAK,WACH,MAAM,IAAI,0BAA0B,oBAAoB;CAC5D;AACF;AAEA,SAAS,sBACP,YACA,OACQ;CACR,IAAI,WAAW,SAAS,QACtB,OAAO,qBAAqB,YAAY,KAAK,EAAE;CAEjD,IAAI,WAAW,SAAS,YAAY,CAAC,MAAM,QAAQ,WAAW,OAC5D,MAAM,IAAI,0BAA0B,oBAAoB;CAE1D,IAAI,WAAW,SAAS,UAAU,OAAO,WAAW;CACpD,MAAM,IAAI,0BAA0B,oBAAoB;AAC1D;AAEA,SAAS,yBACP,YACA,OACoB;CACpB,IAAI,WAAW,SAAS,YAAY,WAAW,SAAS,QACtD;CAEF,MAAM,SAAS,qBAAqB,YAAY,KAAK;CACrD,OAAO,OAAO,aAAa,YAAY,OAAO,OAAO,KAAA;AACvD;AAEA,SAAS,kBAAkB,QAAgB,OAAgC;CACzE,OAAO;EACL,MAAM;EACN;EACA,IAAI;EACJ,OAAO;GAAC,MAAM;GAAW;EAAK;CAChC;AACF;AAiBA,SAAS,oBACP,YACA,OACgB;CAChB,MAAM,SAAS,qBAAqB,WAAW,MAAM,KAAK;CAC1D,MAAM,KAAK,WAAW,OAAO,OAAO,OAAO,WAAW;CACtD,IAAI,OAAO,aAAa,aAAa,OAAO,OAAO,OAAO,MACxD,MAAM,IAAI,0BAA0B,sBAAsB;CAE5D,MAAM,QAAQ,eAAe,WAAW,OAAO,MAAM;CACrD,OAAO;EAAC,MAAM;EAAc,QAAQ,OAAO;EAAM;EAAI;CAAK;AAC5D;AAEA,SAAS,qBACP,YACA,OACgB;CAChB,IAAI,WAAW,SAAS,UAAU,OAAO,cAAc,YAAY,KAAK;CACxE,IAAI,WAAW,SAAS,QACtB,MAAM,IAAI,0BAA0B,oBAAoB;CAE1D,MAAM,SAAS,qBAAqB,WAAW,YAAY,KAAK;CAChE,MAAM,OAAO,kBAAkB,WAAW,IAAI;CAC9C,QAAQ,OAAO,UAAf;EACE,KAAK,WAAW;GACd,MAAM,OAAO,YAAY,IAAI;GAC7B,IAAI,SAAS,KAAA,KAAa,OAAO,OAAO,MACtC,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;IAAC,GAAG;IAAQ;GAAI;EACzB;EACA,KAAK;GACH,IAAI,SAAS,UAAU,SAAS,WAC9B,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;EACT,KAAK;GACH,IAAI,SAAS,QACX,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;EACT,KAAK;GACH,IAAI,SAAS,QACX,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;EACT,KAAK;GACH,IAAI,aAAa,WAAW,IAAI,MAAM,OAAO,SAAS,YAAY,GAChE,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;CACX;AACF;AAEA,SAAS,cACP,YACA,OACgB;CAChB,MAAM,OAAO,MAAM,QAAQ,WAAW;CACtC,IAAI,CAAC,MAAM,MAAM,IAAI,0BAA0B,oBAAoB;CACnE,MAAM,WAAW,kBAAkB,KAAK,QAAQ;CAChD,IAAI,aAAa,QAAQ,OAAO;EAAC,UAAU;EAAW,MAAM,WAAW;CAAI;CAC3E,MAAM,OAAO,YAAY,QAAQ;CACjC,IAAI,SAAS,KAAA,GACX,OAAO;EAAC,UAAU;EAAW,MAAM,WAAW;EAAM;CAAI;CAE1D,IAAI,aAAa,UAAU,aAAa,WAAW;EACjD,IAAI,KAAK,6BAA6B,OACpC,MAAM,IAAI,0BAA0B,6BAA6B;EAEnE,OAAO;GAAC,UAAU;GAAQ,MAAM,WAAW;EAAI;CACjD;CACA,IAAI,aAAa,QAAQ,OAAO;EAAC,UAAU;EAAQ,MAAM,WAAW;CAAI;CACxE,IAAI,KAAK,gBAAgB,KACvB,OAAO;EACL,UAAU;EACV,MAAM,WAAW;EACjB,UAAU,KAAK,SAAS,YAAY;CACtC;CAEF,MAAM,IAAI,0BAA0B,kBAAkB;AACxD;AAEA,SAAS,eACP,YACA,QACqB;CACrB,IAAI,UAAU;CACd,IAAI;CACJ,OAAO,QAAQ,SAAS,QAAQ;EAC9B,OAAO,QAAQ;EACf,UAAU,QAAQ;CACpB;CACA,IAAI,QAAQ,SAAS,aAAa,QAAQ,UAAU,MAClD,MAAM,IAAI,0BAA0B,oBAAoB;CAG1D,QAAQ,OAAO,UAAf;EACE,KAAK;GACH,oBAAoB,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC;GAC3C,IAAI,QAAQ,gBAAgB,WAC1B,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;IAAC,MAAM;IAAW,OAAO,QAAQ;GAAgB;EAE1D,KAAK;GACH,IAAI,SAAS,KAAA,GAAW;IACtB,MAAM,WAAW,YAAY,kBAAkB,IAAI,CAAC;IACpD,IAAI,aAAa,KAAA,KAAa,WAAW,OAAO,MAC9C,MAAM,IAAI,0BAA0B,kBAAkB;GAE1D;GACA,IACE,QAAQ,gBAAgB,aACxB,EAAE,QAAQ,gBAAgB,YAAY,SAAS,KAAA,IAE/C,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;IACL,MAAM;IACN,OAAO,iBAAiB,OAAO,QAAQ,KAAK,GAAG,OAAO,IAAI;GAC5D;EAEF,KAAK;GACH,oBAAoB,MAAM,IAAI,IAAI,CAAC,QAAQ,SAAS,CAAC,CAAC;GACtD,IAAI,QAAQ,gBAAgB,UAC1B,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;IAAC,MAAM;IAAU,OAAO,OAAO,QAAQ,KAAK;GAAC;EACtD,KAAK;GACH,IACE,SAAS,KAAA,KACT,aAAa,IAAI,MAAM,OAAO,SAAS,YAAY,GAEnD,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,IAAI,QAAQ,gBAAgB,UAC1B,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;IAAC,MAAM;IAAU,OAAO,OAAO,QAAQ,KAAK;GAAC;EACtD,KAAK,QAAQ;GACX,oBAAoB,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC;GAC3C,MAAM,QAAQ,OAAO,QAAQ,KAAK,EAAE,YAAY;GAChD,IAAI,QAAQ,gBAAgB,YAAY,CAAC,QAAQ,KAAK,KAAK,GACzD,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;IAAC,MAAM;IAAU;GAAK;EAC/B;CACF;AACF;AAEA,SAAS,oBACP,MACA,SACM;CACN,IAAI,SAAS,KAAA,KAAa,CAAC,QAAQ,IAAI,kBAAkB,IAAI,CAAC,GAC5D,MAAM,IAAI,0BAA0B,kBAAkB;AAE1D;AAEA,SAAS,iBAAiB,OAAe,MAAyB;CAChE,IAAI,CAAC,WAAW,KAAK,KAAK,GACxB,MAAM,IAAI,0BAA0B,kBAAkB;CAExD,IAAI;CACJ,IAAI;EACF,UAAU,OAAO,KAAK;CACxB,QAAQ;EACN,MAAM,IAAI,0BAA0B,kBAAkB;CACxD;CACA,MAAM,SAAS;EACb,CAAC,CAAC,QAAQ,MAAM;EAChB,CAAC,CAAC,aAAa,WAAW;EAC1B,CAAC,CAAC,sBAAsB,oBAAoB;CAC9C;CACA,IAAI,UAAU,OAAO,MAAM,MAAM,UAAU,OAAO,MAAM,IACtD,MAAM,IAAI,0BAA0B,kBAAkB;CAExD,OAAO,QAAQ,SAAS;AAC1B;AAEA,SAAS,kBAAkB,MAAsB;CAC/C,MAAM,OAAO,aAAa,IAAI,EAAE,QAAQ,QAAQ,GAAG;CACnD,QAAQ,MAAR;EACE,KAAK,WACH,OAAO;EACT,KAAK,YACH,OAAO;EACT,KAAK,WACH,OAAO;EACT,KAAK,UACH,OAAO;EACT,KAAK,qBACH,OAAO;EACT,SACE,OAAO;CACX;AACF;AAEA,SAAS,aAAa,MAAsB;CAC1C,OAAO,KAAK,MAAM,GAAG,EAAE,GAAG,EAAE,GAAG,YAAY,KAAK;AAClD;AAEA,SAAS,YAAY,MAAqC;CACxD,QAAQ,MAAR;EACE,KAAK,QACH,OAAO;EACT,KAAK,QACH,OAAO;EACT,KAAK,QACH,OAAO;EACT,SACE;CACJ;AACF;AAEA,SAAS,aACP,MACqD;CACrD,OACE,SAAS,OACT,SAAS,QACT,SAAS,QACT,SAAS,OACT,SAAS,QACT,SAAS,OACT,SAAS;AAEb;AAEA,SAAS,SAAS,KAA+B;CAC/C,MAAM,SAAkB,CAAC;CACzB,KAAK,IAAI,IAAI,GAAG,IAAI,IAAI,SAAS;EAC/B,MAAM,OAAO,IAAI;EACjB,IAAI,cAAc,KAAK,IAAI,GAAG;GAC5B;GACA;EACF;EACA,IAAI,SAAS,MAAK;GAChB,MAAM,SAAS,WAAW,KAAK,GAAG,MAAK,KAAK;GAC5C,OAAO,KAAK;IAAC,MAAM;IAAc,OAAO,OAAO;GAAK,CAAC;GACrD,IAAI,OAAO;GACX;EACF;EACA,MAAM,gBAAgB,SAAS,OAAO,SAAS,QAAQ,IAAI,IAAI,OAAO;EACtE,IAAI,SAAS,OAAO,cAAc;GAChC,MAAM,SAAS,WACb,KACA,KAAK,eAAe,IAAI,IACxB,KACA,YACF;GACA,OAAO,KAAK;IACV,MAAM;IACN,aAAa;IACb,OAAO,OAAO;GAChB,CAAC;GACD,IAAI,OAAO;GACX;EACF;EACA,MAAM,WAAW,kBAAkB,KAAK,IAAI,MAAM,CAAC,CAAC,IAAI;EACxD,IAAI,UAAU;GACZ,OAAO,KAAK,EACV,MAAM,SACR,CAAC;GACD,KAAK,SAAS;GACd;EACF;EACA,MAAM,sBAAsB,8BAA8B,KACxD,IAAI,MAAM,CAAC,CACb,IAAI;EACJ,IAAI,qBAAqB;GACvB,OAAO,KAAK;IAAC,MAAM;IAAe,OAAO;GAAmB,CAAC;GAC7D,KAAK,oBAAoB;GACzB;EACF;EACA,IACE,SAAS,OACT,SAAS,OACT,SAAS,OACT,SAAS,OACT,SAAS,OACT,SAAS,KACT;GACA,OAAO,KAAK,EAAC,MAAM,KAAI,CAAC;GACxB;GACA;EACF;EACA,MAAM,UAAU,iBAAiB,KAAK,IAAI,MAAM,CAAC,CAAC,IAAI;EACtD,IAAI,SAAS;GACX,OAAO,KAAK;IAAC,MAAM;IAAW,aAAa;IAAW,OAAO;GAAO,CAAC;GACrE,KAAK,QAAQ;GACb;EACF;EACA,MAAM,OAAO,cAAc,KAAK,IAAI,MAAM,CAAC,CAAC,IAAI;EAChD,IAAI,MAAM;GACR,MAAM,UAAU,KAAK,YAAY;GACjC,IAAI,yBAAyB,IAAI,OAAO,GACtC,MAAM,IAAI,0BAA0B,sBAAsB;GAE5D,IAAI,YAAY,UAAU,YAAY,SACpC,OAAO,KAAK;IACV,MAAM;IACN,aAAa;IACb,OAAO,YAAY;GACrB,CAAC;QACI,IAAI,YAAY,QACrB,OAAO,KAAK;IAAC,MAAM;IAAW,aAAa;IAAQ,OAAO;GAAI,CAAC;QAC1D,IACL,YAAY,SACZ,YAAY,QACZ,YAAY,SACZ,YAAY,QACZ,YAAY,MAEZ,OAAO,KAAK,EAAC,MAAM,QAAO,CAAC;QAE3B,OAAO,KAAK;IAAC,MAAM;IAAc,OAAO,KAAK,YAAY;GAAC,CAAC;GAE7D,KAAK,KAAK;GACV;EACF;EACA,OAAO,KAAK;GAAC,MAAM;GAAe,OAAO;EAAI,CAAC;EAC9C;CACF;CACA,OAAO,KAAK,EAAC,MAAM,MAAK,CAAC;CACzB,OAAO;AACT;AAEA,SAAS,WACP,KACA,OACA,OACA,SACgD;CAChD,IAAI,QAAQ;CACZ,KAAK,IAAI,IAAI,QAAQ,GAAG,IAAI,IAAI,QAAQ,KAAK;EAC3C,MAAM,OAAO,IAAI;EACjB,IAAI,SAAS,OACX,IAAI,IAAI,IAAI,OAAO,OAAO;GACxB,SAAS;GACT;EACF,OACE,OAAO;GAAC;GAAO,KAAK,IAAI;EAAC;OAEtB,IAAI,WAAW,SAAS,MAAM;GACnC,MAAM,SAAS,IAAI,EAAE;GACrB,IAAI,WAAW,KAAA,GACb,MAAM,IAAI,0BAA0B,oBAAoB;GAE1D,IAAI,WAAW,KAAK;IAClB,MAAM,MAAM,iBAAiB,KAAK,IAAI,GAAG,GAAG,IAAI,CAAC;IACjD,SAAS,OAAO,aAAa,IAAI,KAAK;IACtC,IAAI,IAAI,MAAM;GAChB,OAAO,IAAI,WAAW,OAAO,WAAW,KAAK;IAC3C,MAAM,MAAM,iBAAiB,KAAK,IAAI,GAAG,WAAW,MAAM,IAAI,GAAG,EAAE;IACnE,IAAI;KACF,SAAS,OAAO,cAAc,IAAI,KAAK;IACzC,QAAQ;KACN,MAAM,IAAI,0BAA0B,oBAAoB;IAC1D;IACA,IAAI,IAAI,MAAM;GAChB,OAAO,IAAI,UAAU,OAAO,UAAU,KAAK;IACzC,IAAI,QAAQ;IACZ,OACE,MAAM,SAAS,KACf,IAAI,IAAI,OAAO,KAAA,KACf,IAAI,IAAI,MAAM,OACd,IAAI,IAAI,MAAM,KAEd,SAAS,IAAI,EAAE;IAEjB,SAAS,OAAO,aAAa,OAAO,SAAS,OAAO,CAAC,CAAC;GACxD,OAAO,IAAI,WAAW,MAAM,CAE5B,OAAO,IAAI,WAAW,QAAQ,IAAI,IAAI,OAAO,MAC3C;QAEA,SAAS,eAAe,WAAW;EAEvC,OACE,SAAS;CAEb;CACA,MAAM,IAAI,0BAA0B,oBAAoB;AAC1D;AAEA,SAAS,iBACP,KACA,OACA,QACA,OACA,YAAY,QACoC;CAChD,IAAI,MAAM;CACV,OAAO,MAAM,QAAQ,QAAQ;EAC3B,MAAM,QAAQ,IAAI;EAClB,IACE,UAAU,KAAA,KACV,EACG,SAAS,OAAO,SAAS,OACzB,MAAM,YAAY,KAAK,OAAO,MAAM,YAAY,KAAK,MAGxD;EAEF;CACF;CACA,MAAM,SAAS,IAAI,MAAM,OAAO,GAAG;CACnC,IAAI,OAAO,SAAS,WAClB,MAAM,IAAI,0BAA0B,oBAAoB;CAE1D,OAAO;EAAC,OAAO,OAAO,SAAS,QAAQ,KAAK;EAAG;CAAG;AACpD"}
+{"version":3,"file":"index-predicate.js","names":[],"sources":["../../../../../../../../zero-cache/src/services/change-source/pg/schema/index-predicate.ts"],"sourcesContent":["import * as PostgresTypeClass from '../../../../db/postgres-type-class-enum.ts';\nimport type {\n  IndexPredicate,\n  IndexPredicateValue,\n  PublishedTableSpec,\n} from '../../../../db/specs.ts';\n\nconst UUID_RE =\n  /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/;\nconst INTEGER_RE = /^[+-]?\\d+$/;\nconst WHITESPACE_RE = /\\s/;\nconst OPERATOR_TOKEN_RE = /^(?:::|<>|!=|<=|>=|=|<|>)/;\nconst UNSUPPORTED_OPERATOR_TOKEN_RE = /^(?:!~\\*|!~|~\\*|~)/;\nconst INTEGER_TOKEN_RE = /^[+-]?\\d+/;\nconst WORD_TOKEN_RE = /^[A-Za-z_][A-Za-z0-9_$]*/;\nconst SIMPLE_ESCAPES: Readonly<Record<string, string>> = {\n  b: '\\b',\n  f: '\\f',\n  n: '\\n',\n  r: '\\r',\n  t: '\\t',\n  v: '\\v',\n};\n\nexport const unsupportedPredicateReasons = [\n  'unpublished-column',\n  'unsupported-type',\n  'unsupported-operator',\n  'unsupported-function',\n  'unsupported-cast',\n  'unsupported-syntax',\n  'non-deterministic-collation',\n] as const;\n\nexport type UnsupportedPredicateReason =\n  (typeof unsupportedPredicateReasons)[number];\n\nexport type PredicateTranslation =\n  | {readonly supported: true; readonly predicate: IndexPredicate}\n  | {readonly supported: false; readonly reason: UnsupportedPredicateReason};\n\ntype PredicateColumnSpec = PublishedTableSpec['columns'][string] & {\n  readonly collationIsDeterministic?: boolean | null | undefined;\n};\n\ntype PredicateTableSpec = Omit<PublishedTableSpec, 'columns'> & {\n  readonly columns: Readonly<Record<string, PredicateColumnSpec>>;\n};\n\ntype Expr =\n  | {readonly type: 'column'; readonly name: string}\n  | {\n      readonly type: 'literal';\n      readonly value: boolean | string | null;\n      readonly literalType: 'boolean' | 'integer' | 'string' | 'null';\n    }\n  | {readonly type: 'cast'; readonly expression: Expr; readonly cast: string}\n  | {readonly type: 'not'; readonly expression: Expr}\n  | {\n      readonly type: 'and' | 'or';\n      readonly expressions: readonly Expr[];\n    }\n  | {\n      readonly type: 'comparison';\n      readonly left: Expr;\n      readonly op: '=' | '<>' | '!=' | '<' | '<=' | '>' | '>=';\n      readonly right: Expr;\n    }\n  | {\n      readonly type: 'null-test';\n      readonly expression: Expr;\n      readonly not: boolean;\n    };\n\ntype Token =\n  | {readonly type: 'identifier'; readonly value: string}\n  | {\n      readonly type: 'literal';\n      readonly value: boolean | string | null;\n      readonly literalType: 'boolean' | 'integer' | 'string' | 'null';\n    }\n  | {\n      readonly type:\n        | '('\n        | ')'\n        | '['\n        | ']'\n        | ','\n        | '.'\n        | '::'\n        | '='\n        | '<>'\n        | '!='\n        | '<'\n        | '<='\n        | '>'\n        | '>='\n        | 'AND'\n        | 'OR'\n        | 'NOT'\n        | 'IN'\n        | 'IS'\n        | 'EOF';\n    }\n  | {readonly type: 'unsupported'; readonly value: string};\n\nclass UnsupportedPredicateError extends Error {\n  readonly reason: UnsupportedPredicateReason;\n\n  constructor(reason: UnsupportedPredicateReason) {\n    super(reason);\n    this.reason = reason;\n  }\n}\n\nexport function translatePostgresIndexPredicate(\n  sql: string,\n  table: PredicateTableSpec,\n): PredicateTranslation {\n  try {\n    const parser = new Parser(tokenize(sql));\n    const expression = parser.parse();\n    parser.expect('EOF');\n    return {supported: true, predicate: normalize(expression, table)};\n  } catch (error) {\n    return {\n      supported: false,\n      reason:\n        error instanceof UnsupportedPredicateError\n          ? error.reason\n          : 'unsupported-syntax',\n    };\n  }\n}\n\nclass Parser {\n  readonly #tokens: readonly Token[];\n  #position = 0;\n\n  constructor(tokens: readonly Token[]) {\n    this.#tokens = tokens;\n  }\n\n  parse(): Expr {\n    return this.#parseOr();\n  }\n\n  #parseOr(): Expr {\n    const expressions = [this.#parseAnd()];\n    while (this.#take('OR')) expressions.push(this.#parseAnd());\n    return expressions.length === 1\n      ? expressions[0]\n      : {type: 'or', expressions};\n  }\n\n  #parseAnd(): Expr {\n    const expressions = [this.#parseNot()];\n    while (this.#take('AND')) expressions.push(this.#parseNot());\n    return expressions.length === 1\n      ? expressions[0]\n      : {type: 'and', expressions};\n  }\n\n  #parseNot(): Expr {\n    if (this.#take('NOT')) {\n      return {type: 'not', expression: this.#parseNot()};\n    }\n    return this.#parseComparison();\n  }\n\n  #parseComparison(): Expr {\n    const left = this.#parseOperand();\n    if (this.#take('IS')) {\n      const not = this.#take('NOT');\n      const token = this.#tokens[this.#position++];\n      if (token?.type !== 'literal' || token.literalType !== 'null') {\n        throw new UnsupportedPredicateError('unsupported-operator');\n      }\n      return {type: 'null-test', expression: left, not};\n    }\n    if (this.#take('IN')) {\n      this.expect('(');\n      const values = this.#parseOperandList(')');\n      this.expect(')');\n      return comparisons(left, values);\n    }\n    const op = this.#tokens[this.#position]?.type;\n    if (isComparison(op)) {\n      this.#position++;\n      if (op === '=' && this.#takeIdentifier('ANY')) {\n        return comparisons(left, this.#parseLiteralArray());\n      }\n      return {\n        type: 'comparison',\n        left,\n        op,\n        right: this.#parseOperand(),\n      };\n    }\n    const next = this.#tokens[this.#position];\n    if (next?.type === 'identifier' && next.value.toUpperCase() === 'COLLATE') {\n      throw new UnsupportedPredicateError('non-deterministic-collation');\n    }\n    if (\n      next?.type === 'unsupported' &&\n      UNSUPPORTED_OPERATOR_TOKEN_RE.test(next.value)\n    ) {\n      throw new UnsupportedPredicateError('unsupported-operator');\n    }\n    if (\n      next?.type === 'identifier' &&\n      unsupportedOperatorWords.has(next.value.toUpperCase())\n    ) {\n      throw new UnsupportedPredicateError('unsupported-operator');\n    }\n    return left;\n  }\n\n  #parseLiteralArray(): readonly Expr[] {\n    this.expect('(');\n    this.#expectIdentifier('ARRAY');\n    this.expect('[');\n    const values = this.#parseOperandList(']');\n    this.expect(']');\n    if (this.#take('::')) {\n      this.#parseCastName();\n      this.expect('[');\n      this.expect(']');\n    }\n    this.expect(')');\n    return values;\n  }\n\n  #parseOperandList(end: ')' | ']'): readonly Expr[] {\n    const values: Expr[] = [];\n    if (this.#tokens[this.#position]?.type === end) return values;\n    values.push(this.#parseOperand());\n    while (this.#take(',')) values.push(this.#parseOperand());\n    return values;\n  }\n\n  #parseOperand(): Expr {\n    let expression = this.#parsePrimary();\n    while (this.#take('::')) {\n      expression = {\n        type: 'cast',\n        expression,\n        cast: this.#parseCastName(),\n      };\n    }\n    return expression;\n  }\n\n  #parsePrimary(): Expr {\n    if (this.#take('(')) {\n      const expression = this.#parseOr();\n      this.expect(')');\n      return expression;\n    }\n    const token = this.#tokens[this.#position++];\n    if (token?.type === 'literal') {\n      return {\n        type: 'literal',\n        value: token.value,\n        literalType: token.literalType,\n      };\n    }\n    if (token?.type === 'identifier') {\n      if (this.#tokens[this.#position]?.type === '(') {\n        throw new UnsupportedPredicateError('unsupported-function');\n      }\n      if (this.#tokens[this.#position]?.type === '.') {\n        throw new UnsupportedPredicateError('unsupported-syntax');\n      }\n      if (token.value.toUpperCase() === 'COLLATE') {\n        throw new UnsupportedPredicateError('non-deterministic-collation');\n      }\n      return {type: 'column', name: token.value};\n    }\n    throw new UnsupportedPredicateError('unsupported-syntax');\n  }\n\n  #parseCastName(): string {\n    const first = this.#expectIdentifier();\n    const parts = [first];\n    let separator = '.';\n    while (this.#take('.')) parts.push(this.#expectIdentifier());\n    const next = this.#tokens[this.#position];\n    if (\n      first.toLowerCase() === 'character' &&\n      next?.type === 'identifier' &&\n      next.value.toLowerCase() === 'varying'\n    ) {\n      parts.push(this.#expectIdentifier());\n      separator = ' ';\n    }\n    if (this.#tokens[this.#position]?.type === '(') {\n      throw new UnsupportedPredicateError('unsupported-cast');\n    }\n    return parts.join(separator);\n  }\n\n  #expectIdentifier(expected?: string): string {\n    const token = this.#tokens[this.#position++];\n    if (\n      token?.type !== 'identifier' ||\n      (expected !== undefined && token.value.toUpperCase() !== expected)\n    ) {\n      throw new UnsupportedPredicateError('unsupported-cast');\n    }\n    return token.value;\n  }\n\n  #takeIdentifier(expected: string): boolean {\n    const token = this.#tokens[this.#position];\n    if (\n      token?.type !== 'identifier' ||\n      token.value.toUpperCase() !== expected\n    ) {\n      return false;\n    }\n    this.#position++;\n    return true;\n  }\n\n  #take(type: Token['type']): boolean {\n    if (this.#tokens[this.#position]?.type !== type) return false;\n    this.#position++;\n    return true;\n  }\n\n  expect(type: Token['type']): void {\n    if (!this.#take(type)) {\n      throw new UnsupportedPredicateError('unsupported-syntax');\n    }\n  }\n}\n\nconst unsupportedOperatorWords = new Set([\n  'ALL',\n  'BETWEEN',\n  'ILIKE',\n  'LIKE',\n  'SIMILAR',\n]);\n\nfunction comparisons(left: Expr, values: readonly Expr[]): Expr {\n  if (values.length === 0) {\n    throw new UnsupportedPredicateError('unsupported-syntax');\n  }\n  const expressions = values.map(\n    right => ({type: 'comparison', left, op: '=', right}) as const,\n  );\n  return expressions.length === 1 ? expressions[0] : {type: 'or', expressions};\n}\n\nfunction normalize(\n  expression: Expr,\n  table: PredicateTableSpec,\n): IndexPredicate {\n  switch (expression.type) {\n    case 'and':\n    case 'or': {\n      const conditions = expression.expressions.map(expr =>\n        normalize(expr, table),\n      );\n      return {\n        type: expression.type,\n        conditions: conditions.flatMap(condition =>\n          condition.type === expression.type\n            ? condition.conditions\n            : [condition],\n        ),\n      };\n    }\n    case 'not': {\n      const booleanColumn = resolveBareBooleanColumn(\n        expression.expression,\n        table,\n      );\n      if (booleanColumn) {\n        return booleanComparison(booleanColumn, false);\n      }\n      return {type: 'not', condition: normalize(expression.expression, table)};\n    }\n    case 'column': {\n      const column = resolveColumn(expression, table);\n      if (column.category !== 'boolean') {\n        throw new UnsupportedPredicateError('unsupported-type');\n      }\n      return booleanComparison(column.name, true);\n    }\n    case 'cast': {\n      const column = resolveColumnOperand(expression, table);\n      if (column.category !== 'boolean') {\n        throw new UnsupportedPredicateError('unsupported-type');\n      }\n      return booleanComparison(column.name, true);\n    }\n    case 'null-test': {\n      const column = resolveNullTestColumn(expression.expression, table);\n      return {\n        type: 'null-test',\n        column,\n        op: expression.not ? 'IS NOT NULL' : 'IS NULL',\n      };\n    }\n    case 'comparison':\n      return normalizeComparison(expression, table);\n    case 'literal':\n      throw new UnsupportedPredicateError('unsupported-syntax');\n  }\n}\n\nfunction resolveNullTestColumn(\n  expression: Expr,\n  table: PredicateTableSpec,\n): string {\n  if (expression.type === 'cast') {\n    return resolveColumnOperand(expression, table).name;\n  }\n  if (expression.type === 'column' && !table.columns[expression.name]) {\n    throw new UnsupportedPredicateError('unpublished-column');\n  }\n  if (expression.type === 'column') return expression.name;\n  throw new UnsupportedPredicateError('unsupported-syntax');\n}\n\nfunction resolveBareBooleanColumn(\n  expression: Expr,\n  table: PredicateTableSpec,\n): string | undefined {\n  if (expression.type !== 'column' && expression.type !== 'cast') {\n    return undefined;\n  }\n  const column = resolveColumnOperand(expression, table);\n  return column.category === 'boolean' ? column.name : undefined;\n}\n\nfunction booleanComparison(column: string, value: boolean): IndexPredicate {\n  return {\n    type: 'comparison',\n    column,\n    op: '=',\n    value: {type: 'boolean', value},\n  };\n}\n\ntype ColumnCategory =\n  | {readonly category: 'boolean'; readonly name: string}\n  | {\n      readonly category: 'integer';\n      readonly name: string;\n      readonly rank: 0 | 1 | 2;\n    }\n  | {readonly category: 'text'; readonly name: string}\n  | {\n      readonly category: 'enum';\n      readonly name: string;\n      readonly enumName: string;\n    }\n  | {readonly category: 'uuid'; readonly name: string};\n\nfunction normalizeComparison(\n  expression: Extract<Expr, {type: 'comparison'}>,\n  table: PredicateTableSpec,\n): IndexPredicate {\n  const column = resolveColumnOperand(expression.left, table);\n  const op = expression.op === '!=' ? '<>' : expression.op;\n  if (column.category !== 'integer' && op !== '=' && op !== '<>') {\n    throw new UnsupportedPredicateError('unsupported-operator');\n  }\n  const value = resolveLiteral(expression.right, column);\n  return {type: 'comparison', column: column.name, op, value};\n}\n\nfunction resolveColumnOperand(\n  expression: Expr,\n  table: PredicateTableSpec,\n): ColumnCategory {\n  if (expression.type === 'column') return resolveColumn(expression, table);\n  if (expression.type !== 'cast') {\n    throw new UnsupportedPredicateError('unsupported-syntax');\n  }\n  const column = resolveColumnOperand(expression.expression, table);\n  const cast = canonicalTypeName(expression.cast);\n  switch (column.category) {\n    case 'integer': {\n      const rank = integerRank(cast);\n      if (rank === undefined || rank < column.rank) {\n        throw new UnsupportedPredicateError('unsupported-cast');\n      }\n      return {...column, rank};\n    }\n    case 'text':\n      if (cast !== 'text' && cast !== 'varchar') {\n        throw new UnsupportedPredicateError('unsupported-cast');\n      }\n      return column;\n    case 'boolean':\n      if (cast !== 'bool') {\n        throw new UnsupportedPredicateError('unsupported-cast');\n      }\n      return column;\n    case 'uuid':\n      if (cast !== 'uuid') {\n        throw new UnsupportedPredicateError('unsupported-cast');\n      }\n      return column;\n    case 'enum':\n      if (lastTypeName(expression.cast) !== column.enumName.toLowerCase()) {\n        throw new UnsupportedPredicateError('unsupported-cast');\n      }\n      return column;\n  }\n}\n\nfunction resolveColumn(\n  expression: Extract<Expr, {type: 'column'}>,\n  table: PredicateTableSpec,\n): ColumnCategory {\n  const spec = table.columns[expression.name];\n  if (!spec) throw new UnsupportedPredicateError('unpublished-column');\n  const dataType = canonicalTypeName(spec.dataType);\n  if (dataType === 'bool') return {category: 'boolean', name: expression.name};\n  const rank = integerRank(dataType);\n  if (rank !== undefined) {\n    return {category: 'integer', name: expression.name, rank};\n  }\n  if (dataType === 'text' || dataType === 'varchar') {\n    if (spec.collationIsDeterministic === false) {\n      throw new UnsupportedPredicateError('non-deterministic-collation');\n    }\n    return {category: 'text', name: expression.name};\n  }\n  if (dataType === 'uuid') return {category: 'uuid', name: expression.name};\n  if (spec.pgTypeClass === PostgresTypeClass.Enum) {\n    return {\n      category: 'enum',\n      name: expression.name,\n      enumName: spec.dataType.toLowerCase(),\n    };\n  }\n  throw new UnsupportedPredicateError('unsupported-type');\n}\n\nfunction resolveLiteral(\n  expression: Expr,\n  column: ColumnCategory,\n): IndexPredicateValue {\n  let literal = expression;\n  let cast: string | undefined;\n  while (literal.type === 'cast') {\n    cast = literal.cast;\n    literal = literal.expression;\n  }\n  if (literal.type !== 'literal' || literal.value === null) {\n    throw new UnsupportedPredicateError('unsupported-syntax');\n  }\n\n  switch (column.category) {\n    case 'boolean': {\n      validateLiteralCast(cast, new Set(['bool']));\n      if (literal.literalType !== 'boolean') {\n        throw new UnsupportedPredicateError('unsupported-type');\n      }\n      return {type: 'boolean', value: literal.value as boolean};\n    }\n    case 'integer': {\n      if (cast !== undefined) {\n        const castRank = integerRank(canonicalTypeName(cast));\n        if (castRank === undefined || castRank > column.rank) {\n          throw new UnsupportedPredicateError('unsupported-cast');\n        }\n      }\n      if (\n        literal.literalType !== 'integer' &&\n        !(literal.literalType === 'string' && cast !== undefined)\n      ) {\n        throw new UnsupportedPredicateError('unsupported-type');\n      }\n      return {\n        type: 'integer',\n        value: canonicalInteger(String(literal.value), column.rank),\n      };\n    }\n    case 'text':\n      validateLiteralCast(cast, new Set(['text', 'varchar']));\n      if (literal.literalType !== 'string') {\n        throw new UnsupportedPredicateError('unsupported-type');\n      }\n      return {type: 'string', value: String(literal.value)};\n    case 'enum':\n      if (\n        cast !== undefined &&\n        lastTypeName(cast) !== column.enumName.toLowerCase()\n      ) {\n        throw new UnsupportedPredicateError('unsupported-cast');\n      }\n      if (literal.literalType !== 'string') {\n        throw new UnsupportedPredicateError('unsupported-type');\n      }\n      return {type: 'string', value: String(literal.value)};\n    case 'uuid': {\n      validateLiteralCast(cast, new Set(['uuid']));\n      const value = String(literal.value).toLowerCase();\n      if (literal.literalType !== 'string' || !UUID_RE.test(value)) {\n        throw new UnsupportedPredicateError('unsupported-type');\n      }\n      return {type: 'string', value};\n    }\n  }\n}\n\nfunction validateLiteralCast(\n  cast: string | undefined,\n  allowed: ReadonlySet<string>,\n): void {\n  if (cast !== undefined && !allowed.has(canonicalTypeName(cast))) {\n    throw new UnsupportedPredicateError('unsupported-cast');\n  }\n}\n\nfunction canonicalInteger(value: string, rank: 0 | 1 | 2): string {\n  if (!INTEGER_RE.test(value)) {\n    throw new UnsupportedPredicateError('unsupported-type');\n  }\n  let integer: bigint;\n  try {\n    integer = BigInt(value);\n  } catch {\n    throw new UnsupportedPredicateError('unsupported-type');\n  }\n  const bounds = [\n    [-32768n, 32767n],\n    [-2147483648n, 2147483647n],\n    [-9223372036854775808n, 9223372036854775807n],\n  ] as const;\n  if (integer < bounds[rank][0] || integer > bounds[rank][1]) {\n    throw new UnsupportedPredicateError('unsupported-type');\n  }\n  return integer.toString();\n}\n\nfunction canonicalTypeName(type: string): string {\n  const name = lastTypeName(type).replace(/\\s+/g, ' ');\n  switch (name) {\n    case 'boolean':\n      return 'bool';\n    case 'smallint':\n      return 'int2';\n    case 'integer':\n      return 'int4';\n    case 'bigint':\n      return 'int8';\n    case 'character varying':\n      return 'varchar';\n    default:\n      return name;\n  }\n}\n\nfunction lastTypeName(type: string): string {\n  return type.split('.').at(-1)?.toLowerCase() ?? '';\n}\n\nfunction integerRank(type: string): 0 | 1 | 2 | undefined {\n  switch (type) {\n    case 'int2':\n      return 0;\n    case 'int4':\n      return 1;\n    case 'int8':\n      return 2;\n    default:\n      return undefined;\n  }\n}\n\nfunction isComparison(\n  type: Token['type'] | undefined,\n): type is '=' | '<>' | '!=' | '<' | '<=' | '>' | '>=' {\n  return (\n    type === '=' ||\n    type === '<>' ||\n    type === '!=' ||\n    type === '<' ||\n    type === '<=' ||\n    type === '>' ||\n    type === '>='\n  );\n}\n\nfunction tokenize(sql: string): readonly Token[] {\n  const tokens: Token[] = [];\n  for (let i = 0; i < sql.length;) {\n    const char = sql[i];\n    if (WHITESPACE_RE.test(char)) {\n      i++;\n      continue;\n    }\n    if (char === '\"') {\n      const result = readQuoted(sql, i, '\"', false);\n      tokens.push({type: 'identifier', value: result.value});\n      i = result.end;\n      continue;\n    }\n    const escapeString = (char === 'E' || char === 'e') && sql[i + 1] === \"'\";\n    if (char === \"'\" || escapeString) {\n      const result = readQuoted(\n        sql,\n        i + (escapeString ? 1 : 0),\n        \"'\",\n        escapeString,\n      );\n      tokens.push({\n        type: 'literal',\n        literalType: 'string',\n        value: result.value,\n      });\n      i = result.end;\n      continue;\n    }\n    const operator = OPERATOR_TOKEN_RE.exec(sql.slice(i))?.[0];\n    if (operator) {\n      tokens.push({\n        type: operator as '::' | '=' | '<>' | '!=' | '<' | '<=' | '>' | '>=',\n      });\n      i += operator.length;\n      continue;\n    }\n    const unsupportedOperator = UNSUPPORTED_OPERATOR_TOKEN_RE.exec(\n      sql.slice(i),\n    )?.[0];\n    if (unsupportedOperator) {\n      tokens.push({type: 'unsupported', value: unsupportedOperator});\n      i += unsupportedOperator.length;\n      continue;\n    }\n    if (\n      char === '(' ||\n      char === ')' ||\n      char === '[' ||\n      char === ']' ||\n      char === ',' ||\n      char === '.'\n    ) {\n      tokens.push({type: char});\n      i++;\n      continue;\n    }\n    const integer = INTEGER_TOKEN_RE.exec(sql.slice(i))?.[0];\n    if (integer) {\n      tokens.push({type: 'literal', literalType: 'integer', value: integer});\n      i += integer.length;\n      continue;\n    }\n    const word = WORD_TOKEN_RE.exec(sql.slice(i))?.[0];\n    if (word) {\n      const keyword = word.toUpperCase();\n      if (unsupportedOperatorWords.has(keyword)) {\n        throw new UnsupportedPredicateError('unsupported-operator');\n      }\n      if (keyword === 'TRUE' || keyword === 'FALSE') {\n        tokens.push({\n          type: 'literal',\n          literalType: 'boolean',\n          value: keyword === 'TRUE',\n        });\n      } else if (keyword === 'NULL') {\n        tokens.push({type: 'literal', literalType: 'null', value: null});\n      } else if (\n        keyword === 'AND' ||\n        keyword === 'OR' ||\n        keyword === 'NOT' ||\n        keyword === 'IN' ||\n        keyword === 'IS'\n      ) {\n        tokens.push({type: keyword});\n      } else {\n        tokens.push({type: 'identifier', value: word.toLowerCase()});\n      }\n      i += word.length;\n      continue;\n    }\n    tokens.push({type: 'unsupported', value: char});\n    i++;\n  }\n  tokens.push({type: 'EOF'});\n  return tokens;\n}\n\nfunction readQuoted(\n  sql: string,\n  start: number,\n  quote: '\"' | \"'\",\n  escapes: boolean,\n): {readonly value: string; readonly end: number} {\n  let value = '';\n  for (let i = start + 1; i < sql.length; i++) {\n    const char = sql[i];\n    if (char === quote) {\n      if (sql[i + 1] === quote) {\n        value += quote;\n        i++;\n      } else {\n        return {value, end: i + 1};\n      }\n    } else if (escapes && char === '\\\\') {\n      const escape = sql[++i];\n      if (escape === undefined) {\n        throw new UnsupportedPredicateError('unsupported-syntax');\n      }\n      if (escape === 'x') {\n        const hex = readEscapeDigits(sql, i + 1, 2, 16, 1);\n        value += String.fromCharCode(hex.value);\n        i = hex.end - 1;\n      } else if (escape === 'u' || escape === 'U') {\n        const hex = readEscapeDigits(sql, i + 1, escape === 'u' ? 4 : 8, 16);\n        try {\n          value += String.fromCodePoint(hex.value);\n        } catch {\n          throw new UnsupportedPredicateError('unsupported-syntax');\n        }\n        i = hex.end - 1;\n      } else if (escape >= '0' && escape <= '7') {\n        let octal = escape;\n        while (\n          octal.length < 3 &&\n          sql[i + 1] !== undefined &&\n          sql[i + 1] >= '0' &&\n          sql[i + 1] <= '7'\n        ) {\n          octal += sql[++i];\n        }\n        value += String.fromCharCode(Number.parseInt(octal, 8));\n      } else if (escape === '\\n') {\n        // A backslash-newline pair is a line continuation.\n      } else if (escape === '\\r' && sql[i + 1] === '\\n') {\n        i++;\n      } else {\n        value += SIMPLE_ESCAPES[escape] ?? escape;\n      }\n    } else {\n      value += char;\n    }\n  }\n  throw new UnsupportedPredicateError('unsupported-syntax');\n}\n\nfunction readEscapeDigits(\n  sql: string,\n  start: number,\n  length: number,\n  radix: 16,\n  minLength = length,\n): {readonly value: number; readonly end: number} {\n  let end = start;\n  while (end < start + length) {\n    const digit = sql[end];\n    if (\n      digit === undefined ||\n      !(\n        (digit >= '0' && digit <= '9') ||\n        (digit.toLowerCase() >= 'a' && digit.toLowerCase() <= 'f')\n      )\n    ) {\n      break;\n    }\n    end++;\n  }\n  const digits = sql.slice(start, end);\n  if (digits.length < minLength) {\n    throw new UnsupportedPredicateError('unsupported-syntax');\n  }\n  return {value: Number.parseInt(digits, radix), end};\n}\n"],"mappings":";AAOA,IAAM,UACJ;AACF,IAAM,aAAa;AACnB,IAAM,gBAAgB;AACtB,IAAM,oBAAoB;AAC1B,IAAM,gCAAgC;AACtC,IAAM,mBAAmB;AACzB,IAAM,gBAAgB;AACtB,IAAM,iBAAmD;CACvD,GAAG;CACH,GAAG;CACH,GAAG;CACH,GAAG;CACH,GAAG;CACH,GAAG;AACL;AAoFA,IAAM,4BAAN,cAAwC,MAAM;CAC5C;CAEA,YAAY,QAAoC;EAC9C,MAAM,MAAM;EACZ,KAAK,SAAS;CAChB;AACF;AAEA,SAAgB,gCACd,KACA,OACsB;CACtB,IAAI;EACF,MAAM,SAAS,IAAI,OAAO,SAAS,GAAG,CAAC;EACvC,MAAM,aAAa,OAAO,MAAM;EAChC,OAAO,OAAO,KAAK;EACnB,OAAO;GAAC,WAAW;GAAM,WAAW,UAAU,YAAY,KAAK;EAAC;CAClE,SAAS,OAAO;EACd,OAAO;GACL,WAAW;GACX,QACE,iBAAiB,4BACb,MAAM,SACN;EACR;CACF;AACF;AAEA,IAAM,SAAN,MAAa;CACX;CACA,YAAY;CAEZ,YAAY,QAA0B;EACpC,KAAK,UAAU;CACjB;CAEA,QAAc;EACZ,OAAO,KAAK,SAAS;CACvB;CAEA,WAAiB;EACf,MAAM,cAAc,CAAC,KAAK,UAAU,CAAC;EACrC,OAAO,KAAK,MAAM,IAAI,GAAG,YAAY,KAAK,KAAK,UAAU,CAAC;EAC1D,OAAO,YAAY,WAAW,IAC1B,YAAY,KACZ;GAAC,MAAM;GAAM;EAAW;CAC9B;CAEA,YAAkB;EAChB,MAAM,cAAc,CAAC,KAAK,UAAU,CAAC;EACrC,OAAO,KAAK,MAAM,KAAK,GAAG,YAAY,KAAK,KAAK,UAAU,CAAC;EAC3D,OAAO,YAAY,WAAW,IAC1B,YAAY,KACZ;GAAC,MAAM;GAAO;EAAW;CAC/B;CAEA,YAAkB;EAChB,IAAI,KAAK,MAAM,KAAK,GAClB,OAAO;GAAC,MAAM;GAAO,YAAY,KAAK,UAAU;EAAC;EAEnD,OAAO,KAAK,iBAAiB;CAC/B;CAEA,mBAAyB;EACvB,MAAM,OAAO,KAAK,cAAc;EAChC,IAAI,KAAK,MAAM,IAAI,GAAG;GACpB,MAAM,MAAM,KAAK,MAAM,KAAK;GAC5B,MAAM,QAAQ,KAAK,QAAQ,KAAK;GAChC,IAAI,OAAO,SAAS,aAAa,MAAM,gBAAgB,QACrD,MAAM,IAAI,0BAA0B,sBAAsB;GAE5D,OAAO;IAAC,MAAM;IAAa,YAAY;IAAM;GAAG;EAClD;EACA,IAAI,KAAK,MAAM,IAAI,GAAG;GACpB,KAAK,OAAO,GAAG;GACf,MAAM,SAAS,KAAK,kBAAkB,GAAG;GACzC,KAAK,OAAO,GAAG;GACf,OAAO,YAAY,MAAM,MAAM;EACjC;EACA,MAAM,KAAK,KAAK,QAAQ,KAAK,UAAU,EAAE;EACzC,IAAI,aAAa,EAAE,GAAG;GACpB,KAAK;GACL,IAAI,OAAO,OAAO,KAAK,gBAAgB,KAAK,GAC1C,OAAO,YAAY,MAAM,KAAK,mBAAmB,CAAC;GAEpD,OAAO;IACL,MAAM;IACN;IACA;IACA,OAAO,KAAK,cAAc;GAC5B;EACF;EACA,MAAM,OAAO,KAAK,QAAQ,KAAK;EAC/B,IAAI,MAAM,SAAS,gBAAgB,KAAK,MAAM,YAAY,MAAM,WAC9D,MAAM,IAAI,0BAA0B,6BAA6B;EAEnE,IACE,MAAM,SAAS,iBACf,8BAA8B,KAAK,KAAK,KAAK,GAE7C,MAAM,IAAI,0BAA0B,sBAAsB;EAE5D,IACE,MAAM,SAAS,gBACf,yBAAyB,IAAI,KAAK,MAAM,YAAY,CAAC,GAErD,MAAM,IAAI,0BAA0B,sBAAsB;EAE5D,OAAO;CACT;CAEA,qBAAsC;EACpC,KAAK,OAAO,GAAG;EACf,KAAK,kBAAkB,OAAO;EAC9B,KAAK,OAAO,GAAG;EACf,MAAM,SAAS,KAAK,kBAAkB,GAAG;EACzC,KAAK,OAAO,GAAG;EACf,IAAI,KAAK,MAAM,IAAI,GAAG;GACpB,KAAK,eAAe;GACpB,KAAK,OAAO,GAAG;GACf,KAAK,OAAO,GAAG;EACjB;EACA,KAAK,OAAO,GAAG;EACf,OAAO;CACT;CAEA,kBAAkB,KAAiC;EACjD,MAAM,SAAiB,CAAC;EACxB,IAAI,KAAK,QAAQ,KAAK,UAAU,EAAE,SAAS,KAAK,OAAO;EACvD,OAAO,KAAK,KAAK,cAAc,CAAC;EAChC,OAAO,KAAK,MAAM,GAAG,GAAG,OAAO,KAAK,KAAK,cAAc,CAAC;EACxD,OAAO;CACT;CAEA,gBAAsB;EACpB,IAAI,aAAa,KAAK,cAAc;EACpC,OAAO,KAAK,MAAM,IAAI,GACpB,aAAa;GACX,MAAM;GACN;GACA,MAAM,KAAK,eAAe;EAC5B;EAEF,OAAO;CACT;CAEA,gBAAsB;EACpB,IAAI,KAAK,MAAM,GAAG,GAAG;GACnB,MAAM,aAAa,KAAK,SAAS;GACjC,KAAK,OAAO,GAAG;GACf,OAAO;EACT;EACA,MAAM,QAAQ,KAAK,QAAQ,KAAK;EAChC,IAAI,OAAO,SAAS,WAClB,OAAO;GACL,MAAM;GACN,OAAO,MAAM;GACb,aAAa,MAAM;EACrB;EAEF,IAAI,OAAO,SAAS,cAAc;GAChC,IAAI,KAAK,QAAQ,KAAK,UAAU,EAAE,SAAS,KACzC,MAAM,IAAI,0BAA0B,sBAAsB;GAE5D,IAAI,KAAK,QAAQ,KAAK,UAAU,EAAE,SAAS,KACzC,MAAM,IAAI,0BAA0B,oBAAoB;GAE1D,IAAI,MAAM,MAAM,YAAY,MAAM,WAChC,MAAM,IAAI,0BAA0B,6BAA6B;GAEnE,OAAO;IAAC,MAAM;IAAU,MAAM,MAAM;GAAK;EAC3C;EACA,MAAM,IAAI,0BAA0B,oBAAoB;CAC1D;CAEA,iBAAyB;EACvB,MAAM,QAAQ,KAAK,kBAAkB;EACrC,MAAM,QAAQ,CAAC,KAAK;EACpB,IAAI,YAAY;EAChB,OAAO,KAAK,MAAM,GAAG,GAAG,MAAM,KAAK,KAAK,kBAAkB,CAAC;EAC3D,MAAM,OAAO,KAAK,QAAQ,KAAK;EAC/B,IACE,MAAM,YAAY,MAAM,eACxB,MAAM,SAAS,gBACf,KAAK,MAAM,YAAY,MAAM,WAC7B;GACA,MAAM,KAAK,KAAK,kBAAkB,CAAC;GACnC,YAAY;EACd;EACA,IAAI,KAAK,QAAQ,KAAK,UAAU,EAAE,SAAS,KACzC,MAAM,IAAI,0BAA0B,kBAAkB;EAExD,OAAO,MAAM,KAAK,SAAS;CAC7B;CAEA,kBAAkB,UAA2B;EAC3C,MAAM,QAAQ,KAAK,QAAQ,KAAK;EAChC,IACE,OAAO,SAAS,gBACf,aAAa,KAAA,KAAa,MAAM,MAAM,YAAY,MAAM,UAEzD,MAAM,IAAI,0BAA0B,kBAAkB;EAExD,OAAO,MAAM;CACf;CAEA,gBAAgB,UAA2B;EACzC,MAAM,QAAQ,KAAK,QAAQ,KAAK;EAChC,IACE,OAAO,SAAS,gBAChB,MAAM,MAAM,YAAY,MAAM,UAE9B,OAAO;EAET,KAAK;EACL,OAAO;CACT;CAEA,MAAM,MAA8B;EAClC,IAAI,KAAK,QAAQ,KAAK,UAAU,EAAE,SAAS,MAAM,OAAO;EACxD,KAAK;EACL,OAAO;CACT;CAEA,OAAO,MAA2B;EAChC,IAAI,CAAC,KAAK,MAAM,IAAI,GAClB,MAAM,IAAI,0BAA0B,oBAAoB;CAE5D;AACF;AAEA,IAAM,2CAA2B,IAAI,IAAI;CACvC;CACA;CACA;CACA;CACA;AACF,CAAC;AAED,SAAS,YAAY,MAAY,QAA+B;CAC9D,IAAI,OAAO,WAAW,GACpB,MAAM,IAAI,0BAA0B,oBAAoB;CAE1D,MAAM,cAAc,OAAO,KACzB,WAAU;EAAC,MAAM;EAAc;EAAM,IAAI;EAAK;CAAK,EACrD;CACA,OAAO,YAAY,WAAW,IAAI,YAAY,KAAK;EAAC,MAAM;EAAM;CAAW;AAC7E;AAEA,SAAS,UACP,YACA,OACgB;CAChB,QAAQ,WAAW,MAAnB;EACE,KAAK;EACL,KAAK,MAAM;GACT,MAAM,aAAa,WAAW,YAAY,KAAI,SAC5C,UAAU,MAAM,KAAK,CACvB;GACA,OAAO;IACL,MAAM,WAAW;IACjB,YAAY,WAAW,SAAQ,cAC7B,UAAU,SAAS,WAAW,OAC1B,UAAU,aACV,CAAC,SAAS,CAChB;GACF;EACF;EACA,KAAK,OAAO;GACV,MAAM,gBAAgB,yBACpB,WAAW,YACX,KACF;GACA,IAAI,eACF,OAAO,kBAAkB,eAAe,KAAK;GAE/C,OAAO;IAAC,MAAM;IAAO,WAAW,UAAU,WAAW,YAAY,KAAK;GAAC;EACzE;EACA,KAAK,UAAU;GACb,MAAM,SAAS,cAAc,YAAY,KAAK;GAC9C,IAAI,OAAO,aAAa,WACtB,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO,kBAAkB,OAAO,MAAM,IAAI;EAC5C;EACA,KAAK,QAAQ;GACX,MAAM,SAAS,qBAAqB,YAAY,KAAK;GACrD,IAAI,OAAO,aAAa,WACtB,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO,kBAAkB,OAAO,MAAM,IAAI;EAC5C;EACA,KAAK,aAEH,OAAO;GACL,MAAM;GACN,QAHa,sBAAsB,WAAW,YAAY,KAG1D;GACA,IAAI,WAAW,MAAM,gBAAgB;EACvC;EAEF,KAAK,cACH,OAAO,oBAAoB,YAAY,KAAK;EAC9C,KAAK,WACH,MAAM,IAAI,0BAA0B,oBAAoB;CAC5D;AACF;AAEA,SAAS,sBACP,YACA,OACQ;CACR,IAAI,WAAW,SAAS,QACtB,OAAO,qBAAqB,YAAY,KAAK,CAAC,CAAC;CAEjD,IAAI,WAAW,SAAS,YAAY,CAAC,MAAM,QAAQ,WAAW,OAC5D,MAAM,IAAI,0BAA0B,oBAAoB;CAE1D,IAAI,WAAW,SAAS,UAAU,OAAO,WAAW;CACpD,MAAM,IAAI,0BAA0B,oBAAoB;AAC1D;AAEA,SAAS,yBACP,YACA,OACoB;CACpB,IAAI,WAAW,SAAS,YAAY,WAAW,SAAS,QACtD;CAEF,MAAM,SAAS,qBAAqB,YAAY,KAAK;CACrD,OAAO,OAAO,aAAa,YAAY,OAAO,OAAO,KAAA;AACvD;AAEA,SAAS,kBAAkB,QAAgB,OAAgC;CACzE,OAAO;EACL,MAAM;EACN;EACA,IAAI;EACJ,OAAO;GAAC,MAAM;GAAW;EAAK;CAChC;AACF;AAiBA,SAAS,oBACP,YACA,OACgB;CAChB,MAAM,SAAS,qBAAqB,WAAW,MAAM,KAAK;CAC1D,MAAM,KAAK,WAAW,OAAO,OAAO,OAAO,WAAW;CACtD,IAAI,OAAO,aAAa,aAAa,OAAO,OAAO,OAAO,MACxD,MAAM,IAAI,0BAA0B,sBAAsB;CAE5D,MAAM,QAAQ,eAAe,WAAW,OAAO,MAAM;CACrD,OAAO;EAAC,MAAM;EAAc,QAAQ,OAAO;EAAM;EAAI;CAAK;AAC5D;AAEA,SAAS,qBACP,YACA,OACgB;CAChB,IAAI,WAAW,SAAS,UAAU,OAAO,cAAc,YAAY,KAAK;CACxE,IAAI,WAAW,SAAS,QACtB,MAAM,IAAI,0BAA0B,oBAAoB;CAE1D,MAAM,SAAS,qBAAqB,WAAW,YAAY,KAAK;CAChE,MAAM,OAAO,kBAAkB,WAAW,IAAI;CAC9C,QAAQ,OAAO,UAAf;EACE,KAAK,WAAW;GACd,MAAM,OAAO,YAAY,IAAI;GAC7B,IAAI,SAAS,KAAA,KAAa,OAAO,OAAO,MACtC,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;IAAC,GAAG;IAAQ;GAAI;EACzB;EACA,KAAK;GACH,IAAI,SAAS,UAAU,SAAS,WAC9B,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;EACT,KAAK;GACH,IAAI,SAAS,QACX,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;EACT,KAAK;GACH,IAAI,SAAS,QACX,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;EACT,KAAK;GACH,IAAI,aAAa,WAAW,IAAI,MAAM,OAAO,SAAS,YAAY,GAChE,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;CACX;AACF;AAEA,SAAS,cACP,YACA,OACgB;CAChB,MAAM,OAAO,MAAM,QAAQ,WAAW;CACtC,IAAI,CAAC,MAAM,MAAM,IAAI,0BAA0B,oBAAoB;CACnE,MAAM,WAAW,kBAAkB,KAAK,QAAQ;CAChD,IAAI,aAAa,QAAQ,OAAO;EAAC,UAAU;EAAW,MAAM,WAAW;CAAI;CAC3E,MAAM,OAAO,YAAY,QAAQ;CACjC,IAAI,SAAS,KAAA,GACX,OAAO;EAAC,UAAU;EAAW,MAAM,WAAW;EAAM;CAAI;CAE1D,IAAI,aAAa,UAAU,aAAa,WAAW;EACjD,IAAI,KAAK,6BAA6B,OACpC,MAAM,IAAI,0BAA0B,6BAA6B;EAEnE,OAAO;GAAC,UAAU;GAAQ,MAAM,WAAW;EAAI;CACjD;CACA,IAAI,aAAa,QAAQ,OAAO;EAAC,UAAU;EAAQ,MAAM,WAAW;CAAI;CACxE,IAAI,KAAK,gBAAgB,KACvB,OAAO;EACL,UAAU;EACV,MAAM,WAAW;EACjB,UAAU,KAAK,SAAS,YAAY;CACtC;CAEF,MAAM,IAAI,0BAA0B,kBAAkB;AACxD;AAEA,SAAS,eACP,YACA,QACqB;CACrB,IAAI,UAAU;CACd,IAAI;CACJ,OAAO,QAAQ,SAAS,QAAQ;EAC9B,OAAO,QAAQ;EACf,UAAU,QAAQ;CACpB;CACA,IAAI,QAAQ,SAAS,aAAa,QAAQ,UAAU,MAClD,MAAM,IAAI,0BAA0B,oBAAoB;CAG1D,QAAQ,OAAO,UAAf;EACE,KAAK;GACH,oBAAoB,sBAAM,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC;GAC3C,IAAI,QAAQ,gBAAgB,WAC1B,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;IAAC,MAAM;IAAW,OAAO,QAAQ;GAAgB;EAE1D,KAAK;GACH,IAAI,SAAS,KAAA,GAAW;IACtB,MAAM,WAAW,YAAY,kBAAkB,IAAI,CAAC;IACpD,IAAI,aAAa,KAAA,KAAa,WAAW,OAAO,MAC9C,MAAM,IAAI,0BAA0B,kBAAkB;GAE1D;GACA,IACE,QAAQ,gBAAgB,aACxB,EAAE,QAAQ,gBAAgB,YAAY,SAAS,KAAA,IAE/C,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;IACL,MAAM;IACN,OAAO,iBAAiB,OAAO,QAAQ,KAAK,GAAG,OAAO,IAAI;GAC5D;EAEF,KAAK;GACH,oBAAoB,sBAAM,IAAI,IAAI,CAAC,QAAQ,SAAS,CAAC,CAAC;GACtD,IAAI,QAAQ,gBAAgB,UAC1B,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;IAAC,MAAM;IAAU,OAAO,OAAO,QAAQ,KAAK;GAAC;EACtD,KAAK;GACH,IACE,SAAS,KAAA,KACT,aAAa,IAAI,MAAM,OAAO,SAAS,YAAY,GAEnD,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,IAAI,QAAQ,gBAAgB,UAC1B,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;IAAC,MAAM;IAAU,OAAO,OAAO,QAAQ,KAAK;GAAC;EACtD,KAAK,QAAQ;GACX,oBAAoB,sBAAM,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC;GAC3C,MAAM,QAAQ,OAAO,QAAQ,KAAK,CAAC,CAAC,YAAY;GAChD,IAAI,QAAQ,gBAAgB,YAAY,CAAC,QAAQ,KAAK,KAAK,GACzD,MAAM,IAAI,0BAA0B,kBAAkB;GAExD,OAAO;IAAC,MAAM;IAAU;GAAK;EAC/B;CACF;AACF;AAEA,SAAS,oBACP,MACA,SACM;CACN,IAAI,SAAS,KAAA,KAAa,CAAC,QAAQ,IAAI,kBAAkB,IAAI,CAAC,GAC5D,MAAM,IAAI,0BAA0B,kBAAkB;AAE1D;AAEA,SAAS,iBAAiB,OAAe,MAAyB;CAChE,IAAI,CAAC,WAAW,KAAK,KAAK,GACxB,MAAM,IAAI,0BAA0B,kBAAkB;CAExD,IAAI;CACJ,IAAI;EACF,UAAU,OAAO,KAAK;CACxB,QAAQ;EACN,MAAM,IAAI,0BAA0B,kBAAkB;CACxD;CACA,MAAM,SAAS;EACb,CAAC,CAAC,QAAQ,MAAM;EAChB,CAAC,CAAC,aAAa,WAAW;EAC1B,CAAC,CAAC,sBAAsB,oBAAoB;CAC9C;CACA,IAAI,UAAU,OAAO,KAAK,CAAC,MAAM,UAAU,OAAO,KAAK,CAAC,IACtD,MAAM,IAAI,0BAA0B,kBAAkB;CAExD,OAAO,QAAQ,SAAS;AAC1B;AAEA,SAAS,kBAAkB,MAAsB;CAC/C,MAAM,OAAO,aAAa,IAAI,CAAC,CAAC,QAAQ,QAAQ,GAAG;CACnD,QAAQ,MAAR;EACE,KAAK,WACH,OAAO;EACT,KAAK,YACH,OAAO;EACT,KAAK,WACH,OAAO;EACT,KAAK,UACH,OAAO;EACT,KAAK,qBACH,OAAO;EACT,SACE,OAAO;CACX;AACF;AAEA,SAAS,aAAa,MAAsB;CAC1C,OAAO,KAAK,MAAM,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,YAAY,KAAK;AAClD;AAEA,SAAS,YAAY,MAAqC;CACxD,QAAQ,MAAR;EACE,KAAK,QACH,OAAO;EACT,KAAK,QACH,OAAO;EACT,KAAK,QACH,OAAO;EACT,SACE;CACJ;AACF;AAEA,SAAS,aACP,MACqD;CACrD,OACE,SAAS,OACT,SAAS,QACT,SAAS,QACT,SAAS,OACT,SAAS,QACT,SAAS,OACT,SAAS;AAEb;AAEA,SAAS,SAAS,KAA+B;CAC/C,MAAM,SAAkB,CAAC;CACzB,KAAK,IAAI,IAAI,GAAG,IAAI,IAAI,SAAS;EAC/B,MAAM,OAAO,IAAI;EACjB,IAAI,cAAc,KAAK,IAAI,GAAG;GAC5B;GACA;EACF;EACA,IAAI,SAAS,MAAK;GAChB,MAAM,SAAS,WAAW,KAAK,GAAG,MAAK,KAAK;GAC5C,OAAO,KAAK;IAAC,MAAM;IAAc,OAAO,OAAO;GAAK,CAAC;GACrD,IAAI,OAAO;GACX;EACF;EACA,MAAM,gBAAgB,SAAS,OAAO,SAAS,QAAQ,IAAI,IAAI,OAAO;EACtE,IAAI,SAAS,OAAO,cAAc;GAChC,MAAM,SAAS,WACb,KACA,KAAK,eAAe,IAAI,IACxB,KACA,YACF;GACA,OAAO,KAAK;IACV,MAAM;IACN,aAAa;IACb,OAAO,OAAO;GAChB,CAAC;GACD,IAAI,OAAO;GACX;EACF;EACA,MAAM,WAAW,kBAAkB,KAAK,IAAI,MAAM,CAAC,CAAC,CAAC,GAAG;EACxD,IAAI,UAAU;GACZ,OAAO,KAAK,EACV,MAAM,SACR,CAAC;GACD,KAAK,SAAS;GACd;EACF;EACA,MAAM,sBAAsB,8BAA8B,KACxD,IAAI,MAAM,CAAC,CACb,CAAC,GAAG;EACJ,IAAI,qBAAqB;GACvB,OAAO,KAAK;IAAC,MAAM;IAAe,OAAO;GAAmB,CAAC;GAC7D,KAAK,oBAAoB;GACzB;EACF;EACA,IACE,SAAS,OACT,SAAS,OACT,SAAS,OACT,SAAS,OACT,SAAS,OACT,SAAS,KACT;GACA,OAAO,KAAK,EAAC,MAAM,KAAI,CAAC;GACxB;GACA;EACF;EACA,MAAM,UAAU,iBAAiB,KAAK,IAAI,MAAM,CAAC,CAAC,CAAC,GAAG;EACtD,IAAI,SAAS;GACX,OAAO,KAAK;IAAC,MAAM;IAAW,aAAa;IAAW,OAAO;GAAO,CAAC;GACrE,KAAK,QAAQ;GACb;EACF;EACA,MAAM,OAAO,cAAc,KAAK,IAAI,MAAM,CAAC,CAAC,CAAC,GAAG;EAChD,IAAI,MAAM;GACR,MAAM,UAAU,KAAK,YAAY;GACjC,IAAI,yBAAyB,IAAI,OAAO,GACtC,MAAM,IAAI,0BAA0B,sBAAsB;GAE5D,IAAI,YAAY,UAAU,YAAY,SACpC,OAAO,KAAK;IACV,MAAM;IACN,aAAa;IACb,OAAO,YAAY;GACrB,CAAC;QACI,IAAI,YAAY,QACrB,OAAO,KAAK;IAAC,MAAM;IAAW,aAAa;IAAQ,OAAO;GAAI,CAAC;QAC1D,IACL,YAAY,SACZ,YAAY,QACZ,YAAY,SACZ,YAAY,QACZ,YAAY,MAEZ,OAAO,KAAK,EAAC,MAAM,QAAO,CAAC;QAE3B,OAAO,KAAK;IAAC,MAAM;IAAc,OAAO,KAAK,YAAY;GAAC,CAAC;GAE7D,KAAK,KAAK;GACV;EACF;EACA,OAAO,KAAK;GAAC,MAAM;GAAe,OAAO;EAAI,CAAC;EAC9C;CACF;CACA,OAAO,KAAK,EAAC,MAAM,MAAK,CAAC;CACzB,OAAO;AACT;AAEA,SAAS,WACP,KACA,OACA,OACA,SACgD;CAChD,IAAI,QAAQ;CACZ,KAAK,IAAI,IAAI,QAAQ,GAAG,IAAI,IAAI,QAAQ,KAAK;EAC3C,MAAM,OAAO,IAAI;EACjB,IAAI,SAAS,OAAO;GAClB,IAAI,IAAI,IAAI,OAAO,OAAO;IACxB,SAAS;IACT;GACF,OACE,OAAO;IAAC;IAAO,KAAK,IAAI;GAAC;EAE7B,OAAO,IAAI,WAAW,SAAS,MAAM;GACnC,MAAM,SAAS,IAAI,EAAE;GACrB,IAAI,WAAW,KAAA,GACb,MAAM,IAAI,0BAA0B,oBAAoB;GAE1D,IAAI,WAAW,KAAK;IAClB,MAAM,MAAM,iBAAiB,KAAK,IAAI,GAAG,GAAG,IAAI,CAAC;IACjD,SAAS,OAAO,aAAa,IAAI,KAAK;IACtC,IAAI,IAAI,MAAM;GAChB,OAAO,IAAI,WAAW,OAAO,WAAW,KAAK;IAC3C,MAAM,MAAM,iBAAiB,KAAK,IAAI,GAAG,WAAW,MAAM,IAAI,GAAG,EAAE;IACnE,IAAI;KACF,SAAS,OAAO,cAAc,IAAI,KAAK;IACzC,QAAQ;KACN,MAAM,IAAI,0BAA0B,oBAAoB;IAC1D;IACA,IAAI,IAAI,MAAM;GAChB,OAAO,IAAI,UAAU,OAAO,UAAU,KAAK;IACzC,IAAI,QAAQ;IACZ,OACE,MAAM,SAAS,KACf,IAAI,IAAI,OAAO,KAAA,KACf,IAAI,IAAI,MAAM,OACd,IAAI,IAAI,MAAM,KAEd,SAAS,IAAI,EAAE;IAEjB,SAAS,OAAO,aAAa,OAAO,SAAS,OAAO,CAAC,CAAC;GACxD,OAAO,IAAI,WAAW,MAAM,CAE5B,OAAO,IAAI,WAAW,QAAQ,IAAI,IAAI,OAAO,MAC3C;QAEA,SAAS,eAAe,WAAW;EAEvC,OACE,SAAS;CAEb;CACA,MAAM,IAAI,0BAA0B,oBAAoB;AAC1D;AAEA,SAAS,iBACP,KACA,OACA,QACA,OACA,YAAY,QACoC;CAChD,IAAI,MAAM;CACV,OAAO,MAAM,QAAQ,QAAQ;EAC3B,MAAM,QAAQ,IAAI;EAClB,IACE,UAAU,KAAA,KACV,EACG,SAAS,OAAO,SAAS,OACzB,MAAM,YAAY,KAAK,OAAO,MAAM,YAAY,KAAK,MAGxD;EAEF;CACF;CACA,MAAM,SAAS,IAAI,MAAM,OAAO,GAAG;CACnC,IAAI,OAAO,SAAS,WAClB,MAAM,IAAI,0BAA0B,oBAAoB;CAE1D,OAAO;EAAC,OAAO,OAAO,SAAS,QAAQ,KAAK;EAAG;CAAG;AACpD"}

package/out/zero-cache/src/services/change-source/pg/schema/init.js.map

@@ -1 +1 @@
-{"version":3,"file":"init.js","names":[],"sources":["../../../../../../../../zero-cache/src/services/change-source/pg/schema/init.ts"],"sourcesContent":["import type {LogContext} from '@rocicorp/logger';\nimport {assert} from '../../../../../../shared/src/asserts.ts';\nimport * as v from '../../../../../../shared/src/valita.ts';\nimport {\n  runSchemaMigrations,\n  type IncrementalMigrationMap,\n  type Migration,\n} from '../../../../db/migration.ts';\nimport type {PostgresDB} from '../../../../types/pg.ts';\nimport {upstreamSchema, type ShardConfig} from '../../../../types/shards.ts';\nimport {id} from '../../../../types/sql.ts';\nimport {AutoResetSignal} from '../../../change-streamer/schema/tables.ts';\nimport {publishedSchema} from './published.ts';\nimport * as ReplicaStage from './replica-stage-enum.ts';\nimport {\n  getMutationsTableDefinition,\n  legacyReplicationSlot,\n  metadataPublicationName,\n  setupTablesAndReplication,\n  setupTriggers,\n} from './shard.ts';\n\n/**\n * Ensures that a shard is set up for initial sync.\n */\nexport async function ensureShardSchema(\n  lc: LogContext,\n  db: PostgresDB,\n  shard: ShardConfig,\n  installPartialIndexTriggers = true,\n): Promise<void> {\n  const initialSetup: Migration = {\n    migrateSchema: (lc, tx) =>\n      setupTablesAndReplication(lc, tx, shard, installPartialIndexTriggers),\n    minSafeVersion: 1,\n  };\n  await runSchemaMigrations(\n    lc,\n    `upstream-shard-${shard.appID}`,\n    upstreamSchema(shard),\n    db,\n    initialSetup,\n    // The incremental migration of any existing replicas will be replaced by\n    // the incoming replica being synced, so the replicaVersion here is\n    // unnecessary.\n    getIncrementalMigrations(shard, installPartialIndexTriggers),\n  );\n}\n\nfunction getIncrementalMigrations(\n  shard: ShardConfig,\n  installPartialIndexTriggers: boolean,\n): IncrementalMigrationMap {\n  const shardConfigTable = `${upstreamSchema(shard)}.shardConfig`;\n\n  return {\n    6: {\n      migrateSchema: () => {\n        throw new AutoResetSignal('resetting to upgrade shard schema');\n      },\n    },\n\n    // v7: Updates the DDL event trigger protocol to v2, and adds support for\n    //     ALTER SCHEMA x RENAME TO y\n    //     (migration logic subsumbed by v23)\n\n    // v8: Adds support for non-disruptive resyncs, which tracks multiple\n    //     replicas with different slot names.\n    8: {\n      migrateSchema: async (lc, sql) => {\n        const legacyShardConfigSchema = v.object({\n          replicaVersion: v.string().nullable(),\n          initialSchema: publishedSchema.nullable(),\n        });\n        const result = await sql`\n          SELECT \"replicaVersion\", \"initialSchema\" FROM ${sql(shardConfigTable)}`;\n        assert(\n          result.length === 1,\n          () => `Expected exactly one shardConfig row, got ${result.length}`,\n        );\n        const {replicaVersion, initialSchema} = v.parse(\n          result[0],\n          legacyShardConfigSchema,\n          'passthrough',\n        );\n\n        await Promise.all([\n          sql`\n          CREATE TABLE ${sql(upstreamSchema(shard))}.replicas (\n            \"slot\"          TEXT PRIMARY KEY,\n            \"version\"       TEXT NOT NULL,\n            \"initialSchema\" JSON NOT NULL\n          );\n          `,\n          sql`\n          INSERT INTO ${sql(upstreamSchema(shard))}.replicas ${sql({\n            slot: legacyReplicationSlot(shard),\n            version: replicaVersion,\n            initialSchema,\n          })}\n          `,\n          sql`\n          ALTER TABLE ${sql(shardConfigTable)} DROP \"replicaVersion\", DROP \"initialSchema\"\n          `,\n        ]);\n        lc.info?.(`Upgraded schema to support non-disruptive resyncs`);\n      },\n    },\n\n    // v9: Fixes field ordering of compound indexes. This incremental migration\n    // only fixes indexes resulting from new schema changes. A full resync is\n    // required to fix existing indexes.\n    //\n    // The migration has been subsumed by the identical logic for migrating\n    // to v12 (i.e. a trigger upgrade).\n\n    // Adds the `mutations` table used to track mutation results.\n    10: {\n      migrateSchema: async (lc, sql) => {\n        await sql.unsafe(/*sql*/ `\n          ${getMutationsTableDefinition(upstreamSchema(shard))}\n          ALTER PUBLICATION ${id(metadataPublicationName(shard.appID, shard.shardNum))} ADD TABLE ${id(upstreamSchema(shard))}.\"mutations\";\n        `);\n        lc.info?.('Upgraded schema with new mutations table');\n      },\n    },\n\n    // v11: Formerly dropped the schemaVersions table, but restored in the v13\n    // migration for rollback safety.\n\n    // v12: Upgrade DDL trigger to query schemaOID, needed information for auto-backfill.\n    // (subsumed by v14)\n\n    // Recreates the legacy schemaVersions table that was prematurely dropped\n    // in the (former) v11 migration. It needs to remain present for at least one\n    // release in order to be rollback safe.\n    //\n    // TODO: Drop the table once a release that no longer reads the table has\n    // been rolled out.\n    13: {\n      migrateSchema: async (_, sql) => {\n        await sql`\n          CREATE TABLE IF NOT EXISTS ${sql(upstreamSchema(shard))}.\"schemaVersions\" (\n            \"minSupportedVersion\" INT4,\n            \"maxSupportedVersion\" INT4,\n            \"lock\" BOOL PRIMARY KEY DEFAULT true CHECK (lock)\n        );`;\n        await sql`\n          INSERT INTO ${sql(upstreamSchema(shard))}.\"schemaVersions\" \n            (\"lock\", \"minSupportedVersion\", \"maxSupportedVersion\")\n            VALUES (true, 1, 1)\n            ON CONFLICT DO NOTHING;\n        `;\n      },\n    },\n\n    // v14: Upgrade DDL trigger to log more info to PG logs.\n    // (subsumed by v16)\n\n    // Add initialSyncContext column to replicas table.\n    15: {\n      migrateSchema: async (_, sql) => {\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas\n            ADD COLUMN \"initialSyncContext\" JSON,\n            ADD COLUMN \"subscriberContext\" JSON\n        `;\n      },\n    },\n\n    // v16: Upgrade DDL trigger to fire on all ALTER TABLE statements\n    // to catch the *removal* of a table from the published set.\n    // v17 (1.0.0): Upgrade DDL triggers to support the COMMENT ON PUBLICATION hook for\n    // working around the lack of event trigger support for PUBLICATION\n    // changes in supabase.\n    // This also adds forwards-compatible support for hierarchical logical\n    // message prefixes and unknown ddl event types.\n    // v18: Pure refactoring of event trigger code.\n    // v19 (1.4.0): Correctly handle concurrently issued DDL statements.\n    // v20 (1.4.0): Handle nested DDL triggers\n    // v21 (1.5.0): Handle cross-transaction DDL operations (i.e. concurrent\n    // index operations), and support manual invocation of update_schemas().\n\n    22: {\n      migrateSchema: async (_, sql) => {\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas \n            ALTER \"initialSchema\" DROP NOT NULL;\n        `;\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas \n            DROP CONSTRAINT replicas_pkey;\n        `;\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas \n            ADD COLUMN id TEXT PRIMARY KEY DEFAULT replace(gen_random_uuid()::text, '-', '');\n        `;\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas \n            ADD COLUMN rank BIGSERIAL;\n        `;\n      },\n    },\n\n    // v23 (1.6.0): Event triggers: Handle the case where current_query()\n    // returns NULL. Add more logging to debug unexpected messages\n    23: {\n      migrateSchema: async (lc, sql) => {\n        const [{publications}] = await sql<{publications: string[]}[]>`\n          SELECT publications FROM ${sql(shardConfigTable)}`;\n        await setupTriggers(lc, sql, {...shard, publications});\n        lc.info?.(`Upgraded DDL event triggers`);\n      },\n    },\n\n    // v24 (1.10.0):\n    // - adds the \"backupPath\" column for per-replica backups.\n    // - adds the \"generation\" column to replace \"version\".\n    24: {\n      migrateSchema: async (_, sql) => {\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas \n            ADD COLUMN \"backupPath\" TEXT;\n        `;\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas \n            ADD COLUMN \"generation\" TEXT;\n        `;\n      },\n\n      migrateData: async (_, sql) => {\n        // Backfill the \"generation\" column to eventually replace \"version\"\n        await sql`\n          UPDATE ${sql(upstreamSchema(shard))}.replicas SET \"generation\" = \"version\";\n        `;\n      },\n    },\n\n    // v25 (1.10.0):\n    // - adds the \"backupV5\" column to signal the current semantics\n    //   of the replication slot's confirmed_flush_lsn with respect to what\n    //   has been backed up.\n    25: {\n      migrateSchema: async (_, sql) => {\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas \n            ADD COLUMN \"backupV5\" BOOL DEFAULT false;\n        `;\n      },\n    },\n\n    // v27 (1.11.0):\n    // - make `generation` NOT NULL and stop reading `version`\n    27: {\n      migrateSchema: async (_, sql) => {\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas ALTER \"generation\" SET NOT NULL;\n        `;\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas ALTER \"version\" DROP NOT NULL;\n        `;\n      },\n    },\n\n    28: {\n      migrateSchema: async (_, sql) => {\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas ADD \"epoch\" INT4 DEFAULT 0;\n        `;\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas ADD \"stage\" INT4 DEFAULT 0;  -- ReplicaStage.InitialSync\n        `;\n\n        const metapub = metadataPublicationName(shard.appID, shard.shardNum);\n        await sql`\n          ALTER PUBLICATION ${sql(metapub)} ADD TABLE ${sql(upstreamSchema(shard))}.replicas (id);\n        `;\n        // Make sure supabase gets the ALTER PUBLICATION change\n        await sql`\n          SELECT ${sql(upstreamSchema(shard))}.update_schemas();\n        `;\n      },\n\n      migrateData: async (_, sql) => {\n        await sql`\n          UPDATE ${sql(upstreamSchema(shard))}.replicas SET \"stage\" = ${ReplicaStage.Replicate}\n            WHERE \"subscriberContext\" IS NOT NULL;\n        `;\n      },\n    },\n\n    // v30: Adds the \"pgChangeLog\" column, which records whether the\n    // replication-manager that took over a replica's slot acks it only after\n    // storing transactions in the PG change log (RMv1.5), in which case a\n    // slot position past the change log's head reflects acked keepalives\n    // rather than missing transactions.\n    //\n    // The column defaults to false (i.e. unknown), so that a slot is assumed\n    // to be possibly ahead of the change log until it is taken over by a\n    // replication-manager that sets it. Existing replicas that have been taken\n    // over are backfilled as true (see migrateData). Note that a replication-manager that\n    // predates this column does not reset it when taking over a slot, so\n    // once the column is set, RMv2 (in which acks are not gated on the PG\n    // change log) must run at least this version.\n    30: {\n      migrateSchema: async (_, sql) => {\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas \n            ADD COLUMN IF NOT EXISTS \"pgChangeLog\" BOOL DEFAULT false;\n        `;\n      },\n\n      // Existing replicas whose slots have been taken over (i.e. that have a\n      // subscriberContext) were acked by RMv1 or RMv1.5, which store\n      // transactions in the PG change log before acking them. This assumes\n      // that no RMv2 is running when the migration is run. Replicas that have\n      // not been taken over (e.g. fork destinations) remain unknown, as their\n      // slots have not been acked.\n      migrateData: async (_, sql) => {\n        await sql`\n          UPDATE ${sql(upstreamSchema(shard))}.replicas SET \"pgChangeLog\" = true\n            WHERE \"subscriberContext\" IS NOT NULL;\n        `;\n      },\n    },\n\n    // Note: While this is conditional, it always has to be bumped to the last\n    // version. (It was v29, and is re-run for shards at v29 when it is bumped,\n    // which is harmless as setupTriggers() is idempotent.)\n    ...(installPartialIndexTriggers\n      ? {\n          // v31: Upgrade the DDL event triggers to include partial indexes in\n          // schema snapshots. Note that setupTriggers() also refreshes the\n          // stored \"publishedSchema\" so that the change in format does not\n          // manifest as a spurious schema change.\n          31: {\n            migrateSchema: async (lc, sql) => {\n              const [{publications}] = await sql<{publications: string[]}[]>`\n                SELECT publications FROM ${sql(shardConfigTable)}`;\n              await setupTriggers(lc, sql, {...shard, publications});\n              lc.info?.(`Upgraded DDL event triggers`);\n            },\n          },\n        }\n      : {}),\n  };\n}\n\n// Referenced in tests.\nexport const CURRENT_SCHEMA_VERSION = Object.keys(\n  getIncrementalMigrations(\n    {\n      appID: 'unused',\n      shardNum: 0,\n      publications: ['foo'],\n    },\n    true,\n  ),\n).reduce((prev, curr) => Math.max(prev, parseInt(curr)), 0);\n"],"mappings":";;;;;;;;;;;;;AAyBA,eAAsB,kBACpB,IACA,IACA,OACA,8BAA8B,MACf;CAMf,MAAM,oBACJ,IACA,kBAAkB,MAAM,SACxB,eAAe,KAAK,GACpB,IACA;EATA,gBAAgB,IAAI,OAClB,0BAA0B,IAAI,IAAI,OAAO,2BAA2B;EACtE,gBAAgB;CAOhB,GAIA,yBAAyB,OAAO,2BAA2B,CAC7D;AACF;AAEA,SAAS,yBACP,OACA,6BACyB;CACzB,MAAM,mBAAmB,GAAG,eAAe,KAAK,EAAE;CAElD,OAAO;EACL,GAAG,EACD,qBAAqB;GACnB,MAAM,IAAI,gBAAgB,mCAAmC;EAC/D,EACF;EAQA,GAAG,EACD,eAAe,OAAO,IAAI,QAAQ;GAChC,MAAM,0BAA0B,eAAE,OAAO;IACvC,gBAAgB,eAAE,OAAO,EAAE,SAAS;IACpC,eAAe,gBAAgB,SAAS;GAC1C,CAAC;GACD,MAAM,SAAS,MAAM,GAAG;0DAC0B,IAAI,gBAAgB;GACtE,OACE,OAAO,WAAW,SACZ,6CAA6C,OAAO,QAC5D;GACA,MAAM,EAAC,gBAAgB,kBAAiB,MACtC,OAAO,IACP,yBACA,aACF;GAEA,MAAM,QAAQ,IAAI;IAChB,GAAG;yBACY,IAAI,eAAe,KAAK,CAAC,EAAE;;;;;;IAM1C,GAAG;wBACW,IAAI,eAAe,KAAK,CAAC,EAAE,YAAY,IAAI;KACvD,MAAM,sBAAsB,KAAK;KACjC,SAAS;KACT;IACF,CAAC,EAAE;;IAEH,GAAG;wBACW,IAAI,gBAAgB,EAAE;;GAEtC,CAAC;GACD,GAAG,OAAO,mDAAmD;EAC/D,EACF;EAUA,IAAI,EACF,eAAe,OAAO,IAAI,QAAQ;GAChC,MAAM,IAAI,OAAe;YACrB,4BAA4B,eAAe,KAAK,CAAC,EAAE;8BACjC,GAAG,wBAAwB,MAAM,OAAO,MAAM,QAAQ,CAAC,EAAE,aAAa,GAAG,eAAe,KAAK,CAAC,EAAE;SACrH;GACD,GAAG,OAAO,0CAA0C;EACtD,EACF;EAcA,IAAI,EACF,eAAe,OAAO,GAAG,QAAQ;GAC/B,MAAM,GAAG;uCACsB,IAAI,eAAe,KAAK,CAAC,EAAE;;;;;GAK1D,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;;;EAK7C,EACF;EAMA,IAAI,EACF,eAAe,OAAO,GAAG,QAAQ;GAC/B,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;;EAI7C,EACF;EAeA,IAAI,EACF,eAAe,OAAO,GAAG,QAAQ;GAC/B,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;GAG3C,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;GAG3C,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;GAG3C,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;EAG7C,EACF;EAIA,IAAI,EACF,eAAe,OAAO,IAAI,QAAQ;GAChC,MAAM,CAAC,EAAC,kBAAiB,MAAM,GAA+B;qCACjC,IAAI,gBAAgB;GACjD,MAAM,cAAc,IAAI,KAAK;IAAC,GAAG;IAAO;GAAY,CAAC;GACrD,GAAG,OAAO,6BAA6B;EACzC,EACF;EAKA,IAAI;GACF,eAAe,OAAO,GAAG,QAAQ;IAC/B,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;IAG3C,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;GAG7C;GAEA,aAAa,OAAO,GAAG,QAAQ;IAE7B,MAAM,GAAG;mBACE,IAAI,eAAe,KAAK,CAAC,EAAE;;GAExC;EACF;EAMA,IAAI,EACF,eAAe,OAAO,GAAG,QAAQ;GAC/B,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;EAG7C,EACF;EAIA,IAAI,EACF,eAAe,OAAO,GAAG,QAAQ;GAC/B,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;GAE3C,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;EAE7C,EACF;EAEA,IAAI;GACF,eAAe,OAAO,GAAG,QAAQ;IAC/B,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;IAE3C,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;IAI3C,MAAM,GAAG;8BACa,IAFN,wBAAwB,MAAM,OAAO,MAAM,QAEjC,CAAO,EAAE,aAAa,IAAI,eAAe,KAAK,CAAC,EAAE;;IAG3E,MAAM,GAAG;mBACE,IAAI,eAAe,KAAK,CAAC,EAAE;;GAExC;GAEA,aAAa,OAAO,GAAG,QAAQ;IAC7B,MAAM,GAAG;mBACE,IAAI,eAAe,KAAK,CAAC,EAAE,0BAA0B,EAAuB;;;GAGzF;EACF;EAeA,IAAI;GACF,eAAe,OAAO,GAAG,QAAQ;IAC/B,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;GAG7C;GAQA,aAAa,OAAO,GAAG,QAAQ;IAC7B,MAAM,GAAG;mBACE,IAAI,eAAe,KAAK,CAAC,EAAE;;;GAGxC;EACF;EAKA,GAAI,8BACA,EAKE,IAAI,EACF,eAAe,OAAO,IAAI,QAAQ;GAChC,MAAM,CAAC,EAAC,kBAAiB,MAAM,GAA+B;2CACjC,IAAI,gBAAgB;GACjD,MAAM,cAAc,IAAI,KAAK;IAAC,GAAG;IAAO;GAAY,CAAC;GACrD,GAAG,OAAO,6BAA6B;EACzC,EACF,EACF,IACA,CAAC;CACP;AACF;AAGsC,OAAO,KAC3C,yBACE;CACE,OAAO;CACP,UAAU;CACV,cAAc,CAAC,KAAK;AACtB,GACA,IACF,CACF,EAAE,QAAQ,MAAM,SAAS,KAAK,IAAI,MAAM,SAAS,IAAI,CAAC,GAAG,CAAC"}
+{"version":3,"file":"init.js","names":["v","v.parse"],"sources":["../../../../../../../../zero-cache/src/services/change-source/pg/schema/init.ts"],"sourcesContent":["import type {LogContext} from '@rocicorp/logger';\nimport {assert} from '../../../../../../shared/src/asserts.ts';\nimport * as v from '../../../../../../shared/src/valita.ts';\nimport {\n  runSchemaMigrations,\n  type IncrementalMigrationMap,\n  type Migration,\n} from '../../../../db/migration.ts';\nimport type {PostgresDB} from '../../../../types/pg.ts';\nimport {upstreamSchema, type ShardConfig} from '../../../../types/shards.ts';\nimport {id} from '../../../../types/sql.ts';\nimport {AutoResetSignal} from '../../../change-streamer/schema/tables.ts';\nimport {publishedSchema} from './published.ts';\nimport * as ReplicaStage from './replica-stage-enum.ts';\nimport {\n  getMutationsTableDefinition,\n  legacyReplicationSlot,\n  metadataPublicationName,\n  setupTablesAndReplication,\n  setupTriggers,\n} from './shard.ts';\n\n/**\n * Ensures that a shard is set up for initial sync.\n */\nexport async function ensureShardSchema(\n  lc: LogContext,\n  db: PostgresDB,\n  shard: ShardConfig,\n  installPartialIndexTriggers = true,\n): Promise<void> {\n  const initialSetup: Migration = {\n    migrateSchema: (lc, tx) =>\n      setupTablesAndReplication(lc, tx, shard, installPartialIndexTriggers),\n    minSafeVersion: 1,\n  };\n  await runSchemaMigrations(\n    lc,\n    `upstream-shard-${shard.appID}`,\n    upstreamSchema(shard),\n    db,\n    initialSetup,\n    // The incremental migration of any existing replicas will be replaced by\n    // the incoming replica being synced, so the replicaVersion here is\n    // unnecessary.\n    getIncrementalMigrations(shard, installPartialIndexTriggers),\n  );\n}\n\nfunction getIncrementalMigrations(\n  shard: ShardConfig,\n  installPartialIndexTriggers: boolean,\n): IncrementalMigrationMap {\n  const shardConfigTable = `${upstreamSchema(shard)}.shardConfig`;\n\n  return {\n    6: {\n      migrateSchema: () => {\n        throw new AutoResetSignal('resetting to upgrade shard schema');\n      },\n    },\n\n    // v7: Updates the DDL event trigger protocol to v2, and adds support for\n    //     ALTER SCHEMA x RENAME TO y\n    //     (migration logic subsumbed by v23)\n\n    // v8: Adds support for non-disruptive resyncs, which tracks multiple\n    //     replicas with different slot names.\n    8: {\n      migrateSchema: async (lc, sql) => {\n        const legacyShardConfigSchema = v.object({\n          replicaVersion: v.string().nullable(),\n          initialSchema: publishedSchema.nullable(),\n        });\n        const result = await sql`\n          SELECT \"replicaVersion\", \"initialSchema\" FROM ${sql(shardConfigTable)}`;\n        assert(\n          result.length === 1,\n          () => `Expected exactly one shardConfig row, got ${result.length}`,\n        );\n        const {replicaVersion, initialSchema} = v.parse(\n          result[0],\n          legacyShardConfigSchema,\n          'passthrough',\n        );\n\n        await Promise.all([\n          sql`\n          CREATE TABLE ${sql(upstreamSchema(shard))}.replicas (\n            \"slot\"          TEXT PRIMARY KEY,\n            \"version\"       TEXT NOT NULL,\n            \"initialSchema\" JSON NOT NULL\n          );\n          `,\n          sql`\n          INSERT INTO ${sql(upstreamSchema(shard))}.replicas ${sql({\n            slot: legacyReplicationSlot(shard),\n            version: replicaVersion,\n            initialSchema,\n          })}\n          `,\n          sql`\n          ALTER TABLE ${sql(shardConfigTable)} DROP \"replicaVersion\", DROP \"initialSchema\"\n          `,\n        ]);\n        lc.info?.(`Upgraded schema to support non-disruptive resyncs`);\n      },\n    },\n\n    // v9: Fixes field ordering of compound indexes. This incremental migration\n    // only fixes indexes resulting from new schema changes. A full resync is\n    // required to fix existing indexes.\n    //\n    // The migration has been subsumed by the identical logic for migrating\n    // to v12 (i.e. a trigger upgrade).\n\n    // Adds the `mutations` table used to track mutation results.\n    10: {\n      migrateSchema: async (lc, sql) => {\n        await sql.unsafe(/*sql*/ `\n          ${getMutationsTableDefinition(upstreamSchema(shard))}\n          ALTER PUBLICATION ${id(metadataPublicationName(shard.appID, shard.shardNum))} ADD TABLE ${id(upstreamSchema(shard))}.\"mutations\";\n        `);\n        lc.info?.('Upgraded schema with new mutations table');\n      },\n    },\n\n    // v11: Formerly dropped the schemaVersions table, but restored in the v13\n    // migration for rollback safety.\n\n    // v12: Upgrade DDL trigger to query schemaOID, needed information for auto-backfill.\n    // (subsumed by v14)\n\n    // Recreates the legacy schemaVersions table that was prematurely dropped\n    // in the (former) v11 migration. It needs to remain present for at least one\n    // release in order to be rollback safe.\n    //\n    // TODO: Drop the table once a release that no longer reads the table has\n    // been rolled out.\n    13: {\n      migrateSchema: async (_, sql) => {\n        await sql`\n          CREATE TABLE IF NOT EXISTS ${sql(upstreamSchema(shard))}.\"schemaVersions\" (\n            \"minSupportedVersion\" INT4,\n            \"maxSupportedVersion\" INT4,\n            \"lock\" BOOL PRIMARY KEY DEFAULT true CHECK (lock)\n        );`;\n        await sql`\n          INSERT INTO ${sql(upstreamSchema(shard))}.\"schemaVersions\" \n            (\"lock\", \"minSupportedVersion\", \"maxSupportedVersion\")\n            VALUES (true, 1, 1)\n            ON CONFLICT DO NOTHING;\n        `;\n      },\n    },\n\n    // v14: Upgrade DDL trigger to log more info to PG logs.\n    // (subsumed by v16)\n\n    // Add initialSyncContext column to replicas table.\n    15: {\n      migrateSchema: async (_, sql) => {\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas\n            ADD COLUMN \"initialSyncContext\" JSON,\n            ADD COLUMN \"subscriberContext\" JSON\n        `;\n      },\n    },\n\n    // v16: Upgrade DDL trigger to fire on all ALTER TABLE statements\n    // to catch the *removal* of a table from the published set.\n    // v17 (1.0.0): Upgrade DDL triggers to support the COMMENT ON PUBLICATION hook for\n    // working around the lack of event trigger support for PUBLICATION\n    // changes in supabase.\n    // This also adds forwards-compatible support for hierarchical logical\n    // message prefixes and unknown ddl event types.\n    // v18: Pure refactoring of event trigger code.\n    // v19 (1.4.0): Correctly handle concurrently issued DDL statements.\n    // v20 (1.4.0): Handle nested DDL triggers\n    // v21 (1.5.0): Handle cross-transaction DDL operations (i.e. concurrent\n    // index operations), and support manual invocation of update_schemas().\n\n    22: {\n      migrateSchema: async (_, sql) => {\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas \n            ALTER \"initialSchema\" DROP NOT NULL;\n        `;\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas \n            DROP CONSTRAINT replicas_pkey;\n        `;\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas \n            ADD COLUMN id TEXT PRIMARY KEY DEFAULT replace(gen_random_uuid()::text, '-', '');\n        `;\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas \n            ADD COLUMN rank BIGSERIAL;\n        `;\n      },\n    },\n\n    // v23 (1.6.0): Event triggers: Handle the case where current_query()\n    // returns NULL. Add more logging to debug unexpected messages\n    23: {\n      migrateSchema: async (lc, sql) => {\n        const [{publications}] = await sql<{publications: string[]}[]>`\n          SELECT publications FROM ${sql(shardConfigTable)}`;\n        await setupTriggers(lc, sql, {...shard, publications});\n        lc.info?.(`Upgraded DDL event triggers`);\n      },\n    },\n\n    // v24 (1.10.0):\n    // - adds the \"backupPath\" column for per-replica backups.\n    // - adds the \"generation\" column to replace \"version\".\n    24: {\n      migrateSchema: async (_, sql) => {\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas \n            ADD COLUMN \"backupPath\" TEXT;\n        `;\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas \n            ADD COLUMN \"generation\" TEXT;\n        `;\n      },\n\n      migrateData: async (_, sql) => {\n        // Backfill the \"generation\" column to eventually replace \"version\"\n        await sql`\n          UPDATE ${sql(upstreamSchema(shard))}.replicas SET \"generation\" = \"version\";\n        `;\n      },\n    },\n\n    // v25 (1.10.0):\n    // - adds the \"backupV5\" column to signal the current semantics\n    //   of the replication slot's confirmed_flush_lsn with respect to what\n    //   has been backed up.\n    25: {\n      migrateSchema: async (_, sql) => {\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas \n            ADD COLUMN \"backupV5\" BOOL DEFAULT false;\n        `;\n      },\n    },\n\n    // v27 (1.11.0):\n    // - make `generation` NOT NULL and stop reading `version`\n    27: {\n      migrateSchema: async (_, sql) => {\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas ALTER \"generation\" SET NOT NULL;\n        `;\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas ALTER \"version\" DROP NOT NULL;\n        `;\n      },\n    },\n\n    28: {\n      migrateSchema: async (_, sql) => {\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas ADD \"epoch\" INT4 DEFAULT 0;\n        `;\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas ADD \"stage\" INT4 DEFAULT 0;  -- ReplicaStage.InitialSync\n        `;\n\n        const metapub = metadataPublicationName(shard.appID, shard.shardNum);\n        await sql`\n          ALTER PUBLICATION ${sql(metapub)} ADD TABLE ${sql(upstreamSchema(shard))}.replicas (id);\n        `;\n        // Make sure supabase gets the ALTER PUBLICATION change\n        await sql`\n          SELECT ${sql(upstreamSchema(shard))}.update_schemas();\n        `;\n      },\n\n      migrateData: async (_, sql) => {\n        await sql`\n          UPDATE ${sql(upstreamSchema(shard))}.replicas SET \"stage\" = ${ReplicaStage.Replicate}\n            WHERE \"subscriberContext\" IS NOT NULL;\n        `;\n      },\n    },\n\n    // v30: Adds the \"pgChangeLog\" column, which records whether the\n    // replication-manager that took over a replica's slot acks it only after\n    // storing transactions in the PG change log (RMv1.5), in which case a\n    // slot position past the change log's head reflects acked keepalives\n    // rather than missing transactions.\n    //\n    // The column defaults to false (i.e. unknown), so that a slot is assumed\n    // to be possibly ahead of the change log until it is taken over by a\n    // replication-manager that sets it. Existing replicas that have been taken\n    // over are backfilled as true (see migrateData). Note that a replication-manager that\n    // predates this column does not reset it when taking over a slot, so\n    // once the column is set, RMv2 (in which acks are not gated on the PG\n    // change log) must run at least this version.\n    30: {\n      migrateSchema: async (_, sql) => {\n        await sql`\n          ALTER TABLE ${sql(upstreamSchema(shard))}.replicas \n            ADD COLUMN IF NOT EXISTS \"pgChangeLog\" BOOL DEFAULT false;\n        `;\n      },\n\n      // Existing replicas whose slots have been taken over (i.e. that have a\n      // subscriberContext) were acked by RMv1 or RMv1.5, which store\n      // transactions in the PG change log before acking them. This assumes\n      // that no RMv2 is running when the migration is run. Replicas that have\n      // not been taken over (e.g. fork destinations) remain unknown, as their\n      // slots have not been acked.\n      migrateData: async (_, sql) => {\n        await sql`\n          UPDATE ${sql(upstreamSchema(shard))}.replicas SET \"pgChangeLog\" = true\n            WHERE \"subscriberContext\" IS NOT NULL;\n        `;\n      },\n    },\n\n    // Note: While this is conditional, it always has to be bumped to the last\n    // version. (It was v29, and is re-run for shards at v29 when it is bumped,\n    // which is harmless as setupTriggers() is idempotent.)\n    ...(installPartialIndexTriggers\n      ? {\n          // v31: Upgrade the DDL event triggers to include partial indexes in\n          // schema snapshots. Note that setupTriggers() also refreshes the\n          // stored \"publishedSchema\" so that the change in format does not\n          // manifest as a spurious schema change.\n          31: {\n            migrateSchema: async (lc, sql) => {\n              const [{publications}] = await sql<{publications: string[]}[]>`\n                SELECT publications FROM ${sql(shardConfigTable)}`;\n              await setupTriggers(lc, sql, {...shard, publications});\n              lc.info?.(`Upgraded DDL event triggers`);\n            },\n          },\n        }\n      : {}),\n  };\n}\n\n// Referenced in tests.\nexport const CURRENT_SCHEMA_VERSION = Object.keys(\n  getIncrementalMigrations(\n    {\n      appID: 'unused',\n      shardNum: 0,\n      publications: ['foo'],\n    },\n    true,\n  ),\n).reduce((prev, curr) => Math.max(prev, parseInt(curr)), 0);\n"],"mappings":";;;;;;;;;;;;;AAyBA,eAAsB,kBACpB,IACA,IACA,OACA,8BAA8B,MACf;CAMf,MAAM,oBACJ,IACA,kBAAkB,MAAM,SACxB,eAAe,KAAK,GACpB,IACA;EATA,gBAAgB,IAAI,OAClB,0BAA0B,IAAI,IAAI,OAAO,2BAA2B;EACtE,gBAAgB;CAOhB,GAIA,yBAAyB,OAAO,2BAA2B,CAC7D;AACF;AAEA,SAAS,yBACP,OACA,6BACyB;CACzB,MAAM,mBAAmB,GAAG,eAAe,KAAK,EAAE;CAElD,OAAO;EACL,GAAG,EACD,qBAAqB;GACnB,MAAM,IAAI,gBAAgB,mCAAmC;EAC/D,EACF;EAQA,GAAG,EACD,eAAe,OAAO,IAAI,QAAQ;GAChC,MAAM,0BAA0BA,eAAE,OAAO;IACvC,gBAAgBA,eAAE,OAAO,CAAC,CAAC,SAAS;IACpC,eAAe,gBAAgB,SAAS;GAC1C,CAAC;GACD,MAAM,SAAS,MAAM,GAAG;0DAC0B,IAAI,gBAAgB;GACtE,OACE,OAAO,WAAW,SACZ,6CAA6C,OAAO,QAC5D;GACA,MAAM,EAAC,gBAAgB,kBAAiBC,MACtC,OAAO,IACP,yBACA,aACF;GAEA,MAAM,QAAQ,IAAI;IAChB,GAAG;yBACY,IAAI,eAAe,KAAK,CAAC,EAAE;;;;;;IAM1C,GAAG;wBACW,IAAI,eAAe,KAAK,CAAC,EAAE,YAAY,IAAI;KACvD,MAAM,sBAAsB,KAAK;KACjC,SAAS;KACT;IACF,CAAC,EAAE;;IAEH,GAAG;wBACW,IAAI,gBAAgB,EAAE;;GAEtC,CAAC;GACD,GAAG,OAAO,mDAAmD;EAC/D,EACF;EAUA,IAAI,EACF,eAAe,OAAO,IAAI,QAAQ;GAChC,MAAM,IAAI,OAAe;YACrB,4BAA4B,eAAe,KAAK,CAAC,EAAE;8BACjC,GAAG,wBAAwB,MAAM,OAAO,MAAM,QAAQ,CAAC,EAAE,aAAa,GAAG,eAAe,KAAK,CAAC,EAAE;SACrH;GACD,GAAG,OAAO,0CAA0C;EACtD,EACF;EAcA,IAAI,EACF,eAAe,OAAO,GAAG,QAAQ;GAC/B,MAAM,GAAG;uCACsB,IAAI,eAAe,KAAK,CAAC,EAAE;;;;;GAK1D,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;;;EAK7C,EACF;EAMA,IAAI,EACF,eAAe,OAAO,GAAG,QAAQ;GAC/B,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;;EAI7C,EACF;EAeA,IAAI,EACF,eAAe,OAAO,GAAG,QAAQ;GAC/B,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;GAG3C,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;GAG3C,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;GAG3C,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;EAG7C,EACF;EAIA,IAAI,EACF,eAAe,OAAO,IAAI,QAAQ;GAChC,MAAM,CAAC,EAAC,kBAAiB,MAAM,GAA+B;qCACjC,IAAI,gBAAgB;GACjD,MAAM,cAAc,IAAI,KAAK;IAAC,GAAG;IAAO;GAAY,CAAC;GACrD,GAAG,OAAO,6BAA6B;EACzC,EACF;EAKA,IAAI;GACF,eAAe,OAAO,GAAG,QAAQ;IAC/B,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;IAG3C,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;GAG7C;GAEA,aAAa,OAAO,GAAG,QAAQ;IAE7B,MAAM,GAAG;mBACE,IAAI,eAAe,KAAK,CAAC,EAAE;;GAExC;EACF;EAMA,IAAI,EACF,eAAe,OAAO,GAAG,QAAQ;GAC/B,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;EAG7C,EACF;EAIA,IAAI,EACF,eAAe,OAAO,GAAG,QAAQ;GAC/B,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;GAE3C,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;EAE7C,EACF;EAEA,IAAI;GACF,eAAe,OAAO,GAAG,QAAQ;IAC/B,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;IAE3C,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;IAI3C,MAAM,GAAG;8BACa,IAFN,wBAAwB,MAAM,OAAO,MAAM,QAEjC,CAAO,EAAE,aAAa,IAAI,eAAe,KAAK,CAAC,EAAE;;IAG3E,MAAM,GAAG;mBACE,IAAI,eAAe,KAAK,CAAC,EAAE;;GAExC;GAEA,aAAa,OAAO,GAAG,QAAQ;IAC7B,MAAM,GAAG;mBACE,IAAI,eAAe,KAAK,CAAC,EAAE,0BAA0B,EAAuB;;;GAGzF;EACF;EAeA,IAAI;GACF,eAAe,OAAO,GAAG,QAAQ;IAC/B,MAAM,GAAG;wBACO,IAAI,eAAe,KAAK,CAAC,EAAE;;;GAG7C;GAQA,aAAa,OAAO,GAAG,QAAQ;IAC7B,MAAM,GAAG;mBACE,IAAI,eAAe,KAAK,CAAC,EAAE;;;GAGxC;EACF;EAKA,GAAI,8BACA,EAKE,IAAI,EACF,eAAe,OAAO,IAAI,QAAQ;GAChC,MAAM,CAAC,EAAC,kBAAiB,MAAM,GAA+B;2CACjC,IAAI,gBAAgB;GACjD,MAAM,cAAc,IAAI,KAAK;IAAC,GAAG;IAAO;GAAY,CAAC;GACrD,GAAG,OAAO,6BAA6B;EACzC,EACF,EACF,IACA,CAAC;CACP;AACF;AAGsC,OAAO,KAC3C,yBACE;CACE,OAAO;CACP,UAAU;CACV,cAAc,CAAC,KAAK;AACtB,GACA,IACF,CACF,CAAC,CAAC,QAAQ,MAAM,SAAS,KAAK,IAAI,MAAM,SAAS,IAAI,CAAC,GAAG,CAAC"}

package/out/zero-cache/src/services/change-source/pg/schema/published.js

@@ -287,6 +287,6 @@ async function getPublicationInfo(sql, publications) {
 	return info;
 }
 //#endregion
-export { getPublicationInfo, publishedSchema, publishedSchemaQuery, warnForSkippedIndexes };
+export { getPublicationInfo, getSkippedIndexDiagnostics, publishedSchema, publishedSchemaQuery, warnForSkippedIndexes };
 
 //# sourceMappingURL=published.js.map

package/out/zero-cache/src/services/change-source/pg/schema/published.js.map

@@ -1 +1 @@
-{"version":3,"file":"published.js","names":[],"sources":["../../../../../../../../zero-cache/src/services/change-source/pg/schema/published.ts"],"sourcesContent":["import type {LogContext} from '@rocicorp/logger';\nimport {literal} from 'pg-format';\nimport type postgres from 'postgres';\nimport {assert} from '../../../../../../shared/src/asserts.ts';\nimport {BigIntJSON} from '../../../../../../shared/src/bigint-json.ts';\nimport {equals} from '../../../../../../shared/src/set-utils.ts';\nimport * as v from '../../../../../../shared/src/valita.ts';\nimport {computeZqlSpecsFromLiteSpecs} from '../../../../db/lite-tables.ts';\nimport {\n  mapPostgresToLite,\n  mapPostgresToLiteIndex,\n} from '../../../../db/pg-to-lite.ts';\nimport {\n  publishedColumnSpec,\n  publishedIndexSpec,\n  publishedTableSpec,\n  type PublishedIndexSpec,\n} from '../../../../db/specs.ts';\nimport {getOrCreateCounter} from '../../../../observability/metrics.ts';\nimport {liteTableName} from '../../../../types/names.ts';\nimport {\n  translatePostgresIndexPredicate,\n  type UnsupportedPredicateReason,\n} from './index-predicate.ts';\n\nexport function publishedSchemaQuery(\n  publications: readonly string[],\n  includePartialIndexes = true,\n) {\n  // Notes:\n  // * There's a bug in PG15 in which generated columns are incorrectly\n  //   included in pg_publication_tables.attnames, (even though the generated\n  //   column values are not be included in the replication stream).\n  //   The WHERE condition `attgenerated = ''` fixes this by explicitly excluding\n  //   generated columns from the list.\n  return (\n    /*sql*/ `\nWITH published_columns AS (SELECT \n  pc.oid::int8 AS \"oid\",\n  nspname AS \"schema\",\n  pc.relnamespace::int8 AS \"schemaOID\" ,\n  pc.relname AS \"name\", \n  pc.relreplident AS \"replicaIdentity\",\n  attnum AS \"pos\", \n  attname AS \"col\", \n  pt.typname AS \"type\", \n  atttypid::int8 AS \"typeOID\", \n  pt.typtype,\n  elem_pt.typtype AS \"elemTyptype\",\n  coll.collisdeterministic AS \"collationIsDeterministic\",\n  NULLIF(atttypmod, -1) AS \"maxLen\", \n  attndims \"arrayDims\", \n  attnotnull AS \"notNull\",\n  pg_get_expr(pd.adbin, pd.adrelid) as \"dflt\",\n  NULLIF(ARRAY_POSITION(conkey, attnum), -1) AS \"keyPos\", \n  pb.rowfilter as \"rowFilter\",\n  pb.pubname as \"publication\"\nFROM pg_attribute\nJOIN pg_class pc ON pc.oid = attrelid\nJOIN pg_namespace pns ON pns.oid = relnamespace\nJOIN pg_type pt ON atttypid = pt.oid\nLEFT JOIN pg_type elem_pt ON elem_pt.oid = pt.typelem\nLEFT JOIN pg_collation coll ON coll.oid = pg_attribute.attcollation\nJOIN pg_publication_tables as pb ON \n  pb.schemaname = nspname AND \n  pb.tablename = pc.relname AND\n  attname = ANY(pb.attnames)\nLEFT JOIN pg_constraint pk ON pk.contype = 'p' AND pk.connamespace = relnamespace AND pk.conrelid = attrelid\nLEFT JOIN pg_attrdef pd ON pd.adrelid = attrelid AND pd.adnum = attnum\nWHERE pb.pubname IN (${literal(publications)}) AND \n      (current_setting('server_version_num')::int >= 160000 OR attgenerated = '')\nORDER BY nspname, pc.relname),\n\ntables AS (SELECT json_build_object(\n  'oid', \"oid\",\n  'schema', \"schema\", \n  'schemaOID', \"schemaOID\",\n  'name', \"name\", \n  'replicaIdentity', \"replicaIdentity\",\n  'columns', json_object_agg(\n    DISTINCT\n    col,\n    jsonb_build_object(\n      'pos', \"pos\",\n      'dataType', CASE WHEN \"arrayDims\" = 0 \n                       THEN \"type\" \n                       ELSE substring(\"type\" from 2) || repeat('[]', \"arrayDims\") END,\n      'pgTypeClass', \"typtype\",\n      'elemPgTypeClass', \"elemTyptype\",\n      'typeOID', \"typeOID\",\n      'collationIsDeterministic', \"collationIsDeterministic\",\n      -- https://stackoverflow.com/a/52376230\n      'characterMaximumLength', CASE WHEN \"typeOID\" = 1043 OR \"typeOID\" = 1042 \n                                     THEN \"maxLen\" - 4 \n                                     ELSE \"maxLen\" END,\n      'notNull', \"notNull\",\n      'dflt', \"dflt\"\n    )\n  ),\n  'primaryKey', ARRAY( SELECT json_object_keys(\n    json_strip_nulls(\n      json_object_agg(\n        DISTINCT \"col\", \"keyPos\" ORDER BY \"keyPos\"\n      )\n    )\n  )),\n  'publications', json_object_agg(\n    DISTINCT \n    \"publication\", \n    jsonb_build_object('rowFilter', \"rowFilter\")\n  )\n) AS \"table\" FROM published_columns \n  GROUP BY \"schema\", \"schemaOID\", \"name\", \"oid\", \"replicaIdentity\"),\n  ` +\n    // Note: pg_attribute contains column names for tables and for indexes.\n    // However, the latter does not get updated when a column in a table is\n    // renamed.\n    //\n    // https://www.postgresql.org/message-id/5860814f-c91d-4ab0-b771-ded90d7b9c55%40www.fastmail.com\n    //\n    // To address this, the pg_attribute rows are looked up for the index's\n    // table rather than the index itself, using the pg_index.indkey array\n    // to determine the set and order of columns to include.\n    //\n    // Notes:\n    // * The first bit of indoption is 1 for DESC and 0 for ASC:\n    //   https://github.com/postgres/postgres/blob/4e1fad37872e49a711adad5d9870516e5c71a375/src/include/catalog/pg_index.h#L89\n    // * pg_index.indkey is an int2vector which is 0-based instead of 1-based.\n    // * The additional check for attgenerated is required for the aforementioned\n    //   (in publishedTableQuery) bug in PG15 in which generated columns are\n    //   incorrectly included in pg_publication_tables.attnames\n    /*sql*/ `\n  indexed_columns AS (SELECT\n      pg_indexes.schemaname as \"schema\",\n      pg_indexes.tablename as \"tableName\",\n      pg_indexes.indexname as \"name\",\n      index_column.name as \"col\",\n      CASE WHEN pg_index.indoption[index_column.pos-1] & 1 = 1 THEN 'DESC' ELSE 'ASC' END as \"dir\",\n      -- Partial indexes are always reported as non-unique. Nothing on the\n      -- replica relies on the uniqueness of a partial index (it cannot serve\n      -- as a row key), and reporting it as non-unique keeps the ddl event\n      -- format compatible with older versions of the zero-cache, which\n      -- ignore \"predicateSQL\" and would otherwise create a full UNIQUE index\n      -- on the replica.\n      (pg_index.indisunique AND pg_index.indpred IS NULL) as \"unique\",\n      pg_index.indisprimary as \"isPrimaryKey\",\n      pg_index.indisreplident as \"isReplicaIdentity\",\n      pg_index.indimmediate as \"isImmediate\",\n      pg_get_expr(pg_index.indpred, pg_index.indrelid, false) as \"predicateSQL\"\n    FROM pg_indexes\n    JOIN pg_namespace ON pg_indexes.schemaname = pg_namespace.nspname\n    JOIN pg_class pc ON\n      pc.relname = pg_indexes.indexname\n      AND pc.relnamespace = pg_namespace.oid\n    JOIN pg_publication_tables as pb ON \n      pb.schemaname = pg_indexes.schemaname AND \n      pb.tablename = pg_indexes.tablename\n    JOIN pg_index ON pg_index.indexrelid = pc.oid\n    JOIN LATERAL (\n      SELECT array_agg(attname) as attnames, array_agg(attgenerated != '') as generated FROM pg_attribute\n        WHERE attrelid = pg_index.indrelid\n          AND attnum = ANY( (pg_index.indkey::smallint[] )[:pg_index.indnkeyatts - 1] )\n    ) as indexed ON true\n    JOIN LATERAL (\n      SELECT pg_attribute.attname as name, col.index_pos as pos\n        FROM UNNEST( (pg_index.indkey::smallint[])[:pg_index.indnkeyatts - 1] ) \n          WITH ORDINALITY as col(table_pos, index_pos)\n        JOIN pg_attribute ON attrelid = pg_index.indrelid AND attnum = col.table_pos\n    ) AS index_column ON true\n    LEFT JOIN pg_constraint ON pg_constraint.conindid = pc.oid\n    WHERE pb.pubname IN (${literal(publications)})\n      AND pg_index.indexprs IS NULL\n      AND (${includePartialIndexes} OR pg_index.indpred IS NULL)\n      AND (pg_constraint.contype IS NULL OR pg_constraint.contype IN ('p', 'u'))\n      AND indexed.attnames <@ pb.attnames\n      AND (current_setting('server_version_num')::int >= 160000 OR false = ALL(indexed.generated))\n    ORDER BY\n      pg_indexes.schemaname,\n      pg_indexes.tablename,\n      pg_indexes.indexname,\n      index_column.pos ASC),\n  \n    indexes AS (SELECT json_build_object(\n      'schema', \"schema\",\n      'tableName', \"tableName\",\n      'name', \"name\",\n      'unique', \"unique\",\n      'isPrimaryKey', \"isPrimaryKey\",\n      'isReplicaIdentity', \"isReplicaIdentity\",\n      'isImmediate', \"isImmediate\",\n      'predicateSQL', \"predicateSQL\",\n      'columns', json_object_agg(\"col\", \"dir\")\n    ) AS index FROM indexed_columns \n      GROUP BY \"schema\", \"tableName\", \"name\", \"unique\", \n         \"isPrimaryKey\", \"isReplicaIdentity\", \"isImmediate\", \"predicateSQL\")\n\n    SELECT json_build_object(\n      'tables', COALESCE((SELECT json_agg(\"table\") FROM tables), '[]'::json),\n      'indexes', COALESCE((SELECT json_agg(\"index\") FROM indexes), '[]'::json)\n    ) as \"publishedSchema\"\n  `\n  );\n}\n\nconst rawPublishedColumnSpec = publishedColumnSpec.extend({\n  collationIsDeterministic: v.boolean().nullable().optional(),\n});\n\nconst rawPublishedTableSpec = publishedTableSpec.extend({\n  columns: v.record(rawPublishedColumnSpec),\n});\n\nconst rawPublishedIndexSpec = publishedIndexSpec.extend({\n  predicateSQL: v.string().nullable().optional(),\n});\n\nexport type SkippedIndexDiagnostic = {\n  readonly schema: string;\n  readonly index: string;\n  readonly predicate: string;\n  readonly reason: UnsupportedPredicateReason;\n};\n\nconst skippedIndexDiagnostics = new WeakMap<\n  object,\n  readonly SkippedIndexDiagnostic[]\n>();\nexport function getSkippedIndexDiagnostics(\n  schema: PublishedSchema,\n): readonly SkippedIndexDiagnostic[] {\n  return skippedIndexDiagnostics.get(schema) ?? [];\n}\n\nexport function warnForSkippedIndexes(\n  lc: LogContext,\n  schema: PublishedSchema,\n  warned: Set<string> = new Set(),\n): void {\n  for (const diagnostic of getSkippedIndexDiagnostics(schema)) {\n    const key = JSON.stringify(diagnostic);\n    if (warned.has(key)) continue;\n    warned.add(key);\n    lc.warn?.('Skipping unsupported PostgreSQL partial index', diagnostic);\n    getOrCreateCounter('replication', 'skipped_index', {\n      description: 'Number of PostgreSQL indexes skipped during replication.',\n      unit: '{index}',\n    }).add(1, {reason: diagnostic.reason, partial: true});\n  }\n}\n\nexport const publishedSchema = v\n  .object({\n    tables: v.array(rawPublishedTableSpec),\n    indexes: v.array(rawPublishedIndexSpec),\n  })\n  .map(({tables: rawTables, indexes: rawIndexes}) => {\n    const diagnostics: SkippedIndexDiagnostic[] = [];\n    const indexes: PublishedIndexSpec[] = [];\n    for (const rawIndex of rawIndexes) {\n      const {predicateSQL, ...index} = rawIndex;\n      if (predicateSQL !== null && predicateSQL !== undefined) {\n        const table = rawTables.find(\n          candidate =>\n            candidate.schema === index.schema &&\n            candidate.name === index.tableName,\n        );\n        const translation = table\n          ? translatePostgresIndexPredicate(predicateSQL, table)\n          : {supported: false as const, reason: 'unpublished-column' as const};\n        if (!translation.supported) {\n          diagnostics.push({\n            schema: index.schema,\n            index: index.name,\n            predicate: predicateSQL,\n            reason: translation.reason,\n          });\n          continue;\n        }\n        indexes.push({...index, predicate: translation.predicate});\n      } else {\n        indexes.push(index);\n      }\n    }\n\n    const tables = rawTables.map(table => ({\n      ...table,\n      columns: Object.fromEntries(\n        Object.entries(table.columns).map(\n          ([name, {collationIsDeterministic: _, ...column}]) => [name, column],\n        ),\n      ),\n    }));\n    const zqlSpecs = computeZqlSpecsFromLiteSpecs(\n      tables.map(t => mapPostgresToLite(t)),\n      indexes.map(mapPostgresToLiteIndex),\n      {includeBackfillingColumns: true},\n    );\n    const result = {\n      indexes,\n\n      // Denormalize the schema such that each `table` includes the\n      // `replicaIdentityColumns` corresponding to the table's\n      // replica identity and associated primary key or index.\n      tables: tables.map(table => {\n        const replicaIdentityColumns: string[] = [];\n        switch (table.replicaIdentity) {\n          case 'd':\n            replicaIdentityColumns.push(...(table.primaryKey ?? []));\n            break;\n          case 'i':\n            replicaIdentityColumns.push(\n              ...Object.keys(\n                indexes.find(\n                  ind =>\n                    ind.schema === table.schema &&\n                    ind.tableName === table.name &&\n                    ind.isReplicaIdentity,\n                )?.columns ?? {},\n              ),\n            );\n            break;\n          case 'f': {\n            // For the key columns of replica identity \"full\", use the columns\n            // that the replicator/change-processor will end up using\n            // (in #getKey()) as the row key.\n            const zqlTable = zqlSpecs.get(liteTableName(table));\n            // Note: There zql spec will be absent if the table is not synced,\n            //       e.g. if it has no suitable unique indexes.\n            if (zqlTable) {\n              replicaIdentityColumns.push(...zqlTable.tableSpec.primaryKey);\n            }\n            break;\n          }\n        }\n        return {\n          ...table,\n          replicaIdentityColumns,\n        };\n      }),\n    };\n    skippedIndexDiagnostics.set(result, diagnostics);\n    return result;\n  });\n\nexport type PublishedSchema = v.Infer<typeof publishedSchema>;\n\nexport type PublishedTableWithReplicaIdentity =\n  PublishedSchema['tables'][number];\n\nconst publicationSchema = v.object({\n  pubname: v.string(),\n  pubinsert: v.boolean(),\n  pubupdate: v.boolean(),\n  pubdelete: v.boolean(),\n  pubtruncate: v.boolean(),\n});\n\nconst publicationsResultSchema = v.array(publicationSchema);\n\nexport type PublicationInfo = PublishedSchema & {\n  publications: v.Infer<typeof publicationsResultSchema>;\n};\n\n/**\n * Retrieves published tables and columns.\n */\nexport async function getPublicationInfo(\n  sql: postgres.Sql,\n  publications: string[],\n): Promise<PublicationInfo> {\n  const result = await sql.unsafe(/*sql*/ `\n  SELECT \n    schemaname AS \"schema\",\n    tablename AS \"table\", \n    json_object_agg(pubname, attnames) AS \"publications\"\n    FROM pg_publication_tables pb\n    WHERE pb.pubname IN (${literal(publications)})\n    GROUP BY schemaname, tablename;\n\n  SELECT ${Object.keys(publicationSchema.shape).join(\n    ',',\n  )} FROM pg_publication pb\n    WHERE pb.pubname IN (${literal(publications)})\n    ORDER BY pubname;\n\n  ${publishedSchemaQuery(publications)};\n`);\n\n  // The first query is used to check that tables in multiple publications\n  // always publish the same set of columns.\n  const publishedColumns = result[0] as {\n    schema: string;\n    table: string;\n    publications: Record<string, string[]>;\n  }[];\n  for (const {table, publications} of publishedColumns) {\n    let expected: Set<string>;\n    Object.entries(publications).forEach(([_, columns], i) => {\n      const cols = new Set(columns);\n      if (i === 0) {\n        expected = cols;\n      } else if (!equals(expected, cols)) {\n        throw new Error(\n          `Table ${table} is exported with different columns: [${[\n            ...expected,\n          ]}] vs [${[...cols]}]`,\n        );\n      }\n    });\n  }\n\n  assert(\n    result[2][0].publishedSchema,\n    () => `Invalid publishedSchema result ${BigIntJSON.stringify(result[2])}`,\n  );\n\n  const schema = v.parse(result[2][0].publishedSchema, publishedSchema);\n  const info = {\n    publications: v.parse(result[1], publicationsResultSchema),\n    ...schema,\n  };\n  skippedIndexDiagnostics.set(info, getSkippedIndexDiagnostics(schema));\n  return info;\n}\n"],"mappings":";;;;;;;;;;;;AAyBA,SAAgB,qBACd,cACA,wBAAwB,MACxB;CAOA,OACU;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;uBAiCW,QAAQ,YAAY,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;2BAqGlB,QAAQ,YAAY,EAAE;;aAEpC,sBAAsB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA8BnC;AAEA,IAAM,yBAAyB,oBAAoB,OAAO,EACxD,0BAA0B,eAAE,QAAQ,EAAE,SAAS,EAAE,SAAS,EAC5D,CAAC;AAED,IAAM,wBAAwB,mBAAmB,OAAO,EACtD,SAAS,eAAE,OAAO,sBAAsB,EAC1C,CAAC;AAED,IAAM,wBAAwB,mBAAmB,OAAO,EACtD,cAAc,eAAE,OAAO,EAAE,SAAS,EAAE,SAAS,EAC/C,CAAC;AASD,IAAM,0CAA0B,IAAI,QAGlC;AACF,SAAgB,2BACd,QACmC;CACnC,OAAO,wBAAwB,IAAI,MAAM,KAAK,CAAC;AACjD;AAEA,SAAgB,sBACd,IACA,QACA,yBAAsB,IAAI,IAAI,GACxB;CACN,KAAK,MAAM,cAAc,2BAA2B,MAAM,GAAG;EAC3D,MAAM,MAAM,KAAK,UAAU,UAAU;EACrC,IAAI,OAAO,IAAI,GAAG,GAAG;EACrB,OAAO,IAAI,GAAG;EACd,GAAG,OAAO,iDAAiD,UAAU;EACrE,mBAAmB,eAAe,iBAAiB;GACjD,aAAa;GACb,MAAM;EACR,CAAC,EAAE,IAAI,GAAG;GAAC,QAAQ,WAAW;GAAQ,SAAS;EAAI,CAAC;CACtD;AACF;AAEA,IAAa,kBAAkB,eAC5B,OAAO;CACN,QAAQ,eAAE,MAAM,qBAAqB;CACrC,SAAS,eAAE,MAAM,qBAAqB;AACxC,CAAC,EACA,KAAK,EAAC,QAAQ,WAAW,SAAS,iBAAgB;CACjD,MAAM,cAAwC,CAAC;CAC/C,MAAM,UAAgC,CAAC;CACvC,KAAK,MAAM,YAAY,YAAY;EACjC,MAAM,EAAC,cAAc,GAAG,UAAS;EACjC,IAAI,iBAAiB,QAAQ,iBAAiB,KAAA,GAAW;GACvD,MAAM,QAAQ,UAAU,MACtB,cACE,UAAU,WAAW,MAAM,UAC3B,UAAU,SAAS,MAAM,SAC7B;GACA,MAAM,cAAc,QAChB,gCAAgC,cAAc,KAAK,IACnD;IAAC,WAAW;IAAgB,QAAQ;GAA6B;GACrE,IAAI,CAAC,YAAY,WAAW;IAC1B,YAAY,KAAK;KACf,QAAQ,MAAM;KACd,OAAO,MAAM;KACb,WAAW;KACX,QAAQ,YAAY;IACtB,CAAC;IACD;GACF;GACA,QAAQ,KAAK;IAAC,GAAG;IAAO,WAAW,YAAY;GAAS,CAAC;EAC3D,OACE,QAAQ,KAAK,KAAK;CAEtB;CAEA,MAAM,SAAS,UAAU,KAAI,WAAU;EACrC,GAAG;EACH,SAAS,OAAO,YACd,OAAO,QAAQ,MAAM,OAAO,EAAE,KAC3B,CAAC,MAAM,EAAC,0BAA0B,GAAG,GAAG,cAAa,CAAC,MAAM,MAAM,CACrE,CACF;CACF,EAAE;CACF,MAAM,WAAW,6BACf,OAAO,KAAI,MAAK,kBAAkB,CAAC,CAAC,GACpC,QAAQ,IAAI,sBAAsB,GAClC,EAAC,2BAA2B,KAAI,CAClC;CACA,MAAM,SAAS;EACb;EAKA,QAAQ,OAAO,KAAI,UAAS;GAC1B,MAAM,yBAAmC,CAAC;GAC1C,QAAQ,MAAM,iBAAd;IACE,KAAK;KACH,uBAAuB,KAAK,GAAI,MAAM,cAAc,CAAC,CAAE;KACvD;IACF,KAAK;KACH,uBAAuB,KACrB,GAAG,OAAO,KACR,QAAQ,MACN,QACE,IAAI,WAAW,MAAM,UACrB,IAAI,cAAc,MAAM,QACxB,IAAI,iBACR,GAAG,WAAW,CAAC,CACjB,CACF;KACA;IACF,KAAK,KAAK;KAIR,MAAM,WAAW,SAAS,IAAI,cAAc,KAAK,CAAC;KAGlD,IAAI,UACF,uBAAuB,KAAK,GAAG,SAAS,UAAU,UAAU;KAE9D;IACF;GACF;GACA,OAAO;IACL,GAAG;IACH;GACF;EACF,CAAC;CACH;CACA,wBAAwB,IAAI,QAAQ,WAAW;CAC/C,OAAO;AACT,CAAC;AAOH,IAAM,oBAAoB,eAAE,OAAO;CACjC,SAAS,eAAE,OAAO;CAClB,WAAW,eAAE,QAAQ;CACrB,WAAW,eAAE,QAAQ;CACrB,WAAW,eAAE,QAAQ;CACrB,aAAa,eAAE,QAAQ;AACzB,CAAC;AAED,IAAM,2BAA2B,eAAE,MAAM,iBAAiB;;;;AAS1D,eAAsB,mBACpB,KACA,cAC0B;CAC1B,MAAM,SAAS,MAAM,IAAI,OAAe;;;;;;2BAMf,QAAQ,YAAY,EAAE;;;WAGtC,OAAO,KAAK,kBAAkB,KAAK,EAAE,KAC5C,GACF,EAAE;2BACuB,QAAQ,YAAY,EAAE;;;IAG7C,qBAAqB,YAAY,EAAE;CACtC;CAIC,MAAM,mBAAmB,OAAO;CAKhC,KAAK,MAAM,EAAC,OAAO,kBAAiB,kBAAkB;EACpD,IAAI;EACJ,OAAO,QAAQ,YAAY,EAAE,SAAS,CAAC,GAAG,UAAU,MAAM;GACxD,MAAM,OAAO,IAAI,IAAI,OAAO;GAC5B,IAAI,MAAM,GACR,WAAW;QACN,IAAI,CAAC,OAAO,UAAU,IAAI,GAC/B,MAAM,IAAI,MACR,SAAS,MAAM,wCAAwC,CACrD,GAAG,QACL,EAAE,QAAQ,CAAC,GAAG,IAAI,EAAE,EACtB;EAEJ,CAAC;CACH;CAEA,OACE,OAAO,GAAG,GAAG,uBACP,kCAAkC,WAAW,UAAU,OAAO,EAAE,GACxE;CAEA,MAAM,SAAS,MAAQ,OAAO,GAAG,GAAG,iBAAiB,eAAe;CACpE,MAAM,OAAO;EACX,cAAc,MAAQ,OAAO,IAAI,wBAAwB;EACzD,GAAG;CACL;CACA,wBAAwB,IAAI,MAAM,2BAA2B,MAAM,CAAC;CACpE,OAAO;AACT"}
+{"version":3,"file":"published.js","names":["v","v.parse"],"sources":["../../../../../../../../zero-cache/src/services/change-source/pg/schema/published.ts"],"sourcesContent":["import type {LogContext} from '@rocicorp/logger';\nimport {literal} from 'pg-format';\nimport type postgres from 'postgres';\nimport {assert} from '../../../../../../shared/src/asserts.ts';\nimport {BigIntJSON} from '../../../../../../shared/src/bigint-json.ts';\nimport {equals} from '../../../../../../shared/src/set-utils.ts';\nimport * as v from '../../../../../../shared/src/valita.ts';\nimport {computeZqlSpecsFromLiteSpecs} from '../../../../db/lite-tables.ts';\nimport {\n  mapPostgresToLite,\n  mapPostgresToLiteIndex,\n} from '../../../../db/pg-to-lite.ts';\nimport {\n  publishedColumnSpec,\n  publishedIndexSpec,\n  publishedTableSpec,\n  type PublishedIndexSpec,\n} from '../../../../db/specs.ts';\nimport {getOrCreateCounter} from '../../../../observability/metrics.ts';\nimport {liteTableName} from '../../../../types/names.ts';\nimport {\n  translatePostgresIndexPredicate,\n  type UnsupportedPredicateReason,\n} from './index-predicate.ts';\n\nexport function publishedSchemaQuery(\n  publications: readonly string[],\n  includePartialIndexes = true,\n) {\n  // Notes:\n  // * There's a bug in PG15 in which generated columns are incorrectly\n  //   included in pg_publication_tables.attnames, (even though the generated\n  //   column values are not be included in the replication stream).\n  //   The WHERE condition `attgenerated = ''` fixes this by explicitly excluding\n  //   generated columns from the list.\n  return (\n    /*sql*/ `\nWITH published_columns AS (SELECT \n  pc.oid::int8 AS \"oid\",\n  nspname AS \"schema\",\n  pc.relnamespace::int8 AS \"schemaOID\" ,\n  pc.relname AS \"name\", \n  pc.relreplident AS \"replicaIdentity\",\n  attnum AS \"pos\", \n  attname AS \"col\", \n  pt.typname AS \"type\", \n  atttypid::int8 AS \"typeOID\", \n  pt.typtype,\n  elem_pt.typtype AS \"elemTyptype\",\n  coll.collisdeterministic AS \"collationIsDeterministic\",\n  NULLIF(atttypmod, -1) AS \"maxLen\", \n  attndims \"arrayDims\", \n  attnotnull AS \"notNull\",\n  pg_get_expr(pd.adbin, pd.adrelid) as \"dflt\",\n  NULLIF(ARRAY_POSITION(conkey, attnum), -1) AS \"keyPos\", \n  pb.rowfilter as \"rowFilter\",\n  pb.pubname as \"publication\"\nFROM pg_attribute\nJOIN pg_class pc ON pc.oid = attrelid\nJOIN pg_namespace pns ON pns.oid = relnamespace\nJOIN pg_type pt ON atttypid = pt.oid\nLEFT JOIN pg_type elem_pt ON elem_pt.oid = pt.typelem\nLEFT JOIN pg_collation coll ON coll.oid = pg_attribute.attcollation\nJOIN pg_publication_tables as pb ON \n  pb.schemaname = nspname AND \n  pb.tablename = pc.relname AND\n  attname = ANY(pb.attnames)\nLEFT JOIN pg_constraint pk ON pk.contype = 'p' AND pk.connamespace = relnamespace AND pk.conrelid = attrelid\nLEFT JOIN pg_attrdef pd ON pd.adrelid = attrelid AND pd.adnum = attnum\nWHERE pb.pubname IN (${literal(publications)}) AND \n      (current_setting('server_version_num')::int >= 160000 OR attgenerated = '')\nORDER BY nspname, pc.relname),\n\ntables AS (SELECT json_build_object(\n  'oid', \"oid\",\n  'schema', \"schema\", \n  'schemaOID', \"schemaOID\",\n  'name', \"name\", \n  'replicaIdentity', \"replicaIdentity\",\n  'columns', json_object_agg(\n    DISTINCT\n    col,\n    jsonb_build_object(\n      'pos', \"pos\",\n      'dataType', CASE WHEN \"arrayDims\" = 0 \n                       THEN \"type\" \n                       ELSE substring(\"type\" from 2) || repeat('[]', \"arrayDims\") END,\n      'pgTypeClass', \"typtype\",\n      'elemPgTypeClass', \"elemTyptype\",\n      'typeOID', \"typeOID\",\n      'collationIsDeterministic', \"collationIsDeterministic\",\n      -- https://stackoverflow.com/a/52376230\n      'characterMaximumLength', CASE WHEN \"typeOID\" = 1043 OR \"typeOID\" = 1042 \n                                     THEN \"maxLen\" - 4 \n                                     ELSE \"maxLen\" END,\n      'notNull', \"notNull\",\n      'dflt', \"dflt\"\n    )\n  ),\n  'primaryKey', ARRAY( SELECT json_object_keys(\n    json_strip_nulls(\n      json_object_agg(\n        DISTINCT \"col\", \"keyPos\" ORDER BY \"keyPos\"\n      )\n    )\n  )),\n  'publications', json_object_agg(\n    DISTINCT \n    \"publication\", \n    jsonb_build_object('rowFilter', \"rowFilter\")\n  )\n) AS \"table\" FROM published_columns \n  GROUP BY \"schema\", \"schemaOID\", \"name\", \"oid\", \"replicaIdentity\"),\n  ` +\n    // Note: pg_attribute contains column names for tables and for indexes.\n    // However, the latter does not get updated when a column in a table is\n    // renamed.\n    //\n    // https://www.postgresql.org/message-id/5860814f-c91d-4ab0-b771-ded90d7b9c55%40www.fastmail.com\n    //\n    // To address this, the pg_attribute rows are looked up for the index's\n    // table rather than the index itself, using the pg_index.indkey array\n    // to determine the set and order of columns to include.\n    //\n    // Notes:\n    // * The first bit of indoption is 1 for DESC and 0 for ASC:\n    //   https://github.com/postgres/postgres/blob/4e1fad37872e49a711adad5d9870516e5c71a375/src/include/catalog/pg_index.h#L89\n    // * pg_index.indkey is an int2vector which is 0-based instead of 1-based.\n    // * The additional check for attgenerated is required for the aforementioned\n    //   (in publishedTableQuery) bug in PG15 in which generated columns are\n    //   incorrectly included in pg_publication_tables.attnames\n    /*sql*/ `\n  indexed_columns AS (SELECT\n      pg_indexes.schemaname as \"schema\",\n      pg_indexes.tablename as \"tableName\",\n      pg_indexes.indexname as \"name\",\n      index_column.name as \"col\",\n      CASE WHEN pg_index.indoption[index_column.pos-1] & 1 = 1 THEN 'DESC' ELSE 'ASC' END as \"dir\",\n      -- Partial indexes are always reported as non-unique. Nothing on the\n      -- replica relies on the uniqueness of a partial index (it cannot serve\n      -- as a row key), and reporting it as non-unique keeps the ddl event\n      -- format compatible with older versions of the zero-cache, which\n      -- ignore \"predicateSQL\" and would otherwise create a full UNIQUE index\n      -- on the replica.\n      (pg_index.indisunique AND pg_index.indpred IS NULL) as \"unique\",\n      pg_index.indisprimary as \"isPrimaryKey\",\n      pg_index.indisreplident as \"isReplicaIdentity\",\n      pg_index.indimmediate as \"isImmediate\",\n      pg_get_expr(pg_index.indpred, pg_index.indrelid, false) as \"predicateSQL\"\n    FROM pg_indexes\n    JOIN pg_namespace ON pg_indexes.schemaname = pg_namespace.nspname\n    JOIN pg_class pc ON\n      pc.relname = pg_indexes.indexname\n      AND pc.relnamespace = pg_namespace.oid\n    JOIN pg_publication_tables as pb ON \n      pb.schemaname = pg_indexes.schemaname AND \n      pb.tablename = pg_indexes.tablename\n    JOIN pg_index ON pg_index.indexrelid = pc.oid\n    JOIN LATERAL (\n      SELECT array_agg(attname) as attnames, array_agg(attgenerated != '') as generated FROM pg_attribute\n        WHERE attrelid = pg_index.indrelid\n          AND attnum = ANY( (pg_index.indkey::smallint[] )[:pg_index.indnkeyatts - 1] )\n    ) as indexed ON true\n    JOIN LATERAL (\n      SELECT pg_attribute.attname as name, col.index_pos as pos\n        FROM UNNEST( (pg_index.indkey::smallint[])[:pg_index.indnkeyatts - 1] ) \n          WITH ORDINALITY as col(table_pos, index_pos)\n        JOIN pg_attribute ON attrelid = pg_index.indrelid AND attnum = col.table_pos\n    ) AS index_column ON true\n    LEFT JOIN pg_constraint ON pg_constraint.conindid = pc.oid\n    WHERE pb.pubname IN (${literal(publications)})\n      AND pg_index.indexprs IS NULL\n      AND (${includePartialIndexes} OR pg_index.indpred IS NULL)\n      AND (pg_constraint.contype IS NULL OR pg_constraint.contype IN ('p', 'u'))\n      AND indexed.attnames <@ pb.attnames\n      AND (current_setting('server_version_num')::int >= 160000 OR false = ALL(indexed.generated))\n    ORDER BY\n      pg_indexes.schemaname,\n      pg_indexes.tablename,\n      pg_indexes.indexname,\n      index_column.pos ASC),\n  \n    indexes AS (SELECT json_build_object(\n      'schema', \"schema\",\n      'tableName', \"tableName\",\n      'name', \"name\",\n      'unique', \"unique\",\n      'isPrimaryKey', \"isPrimaryKey\",\n      'isReplicaIdentity', \"isReplicaIdentity\",\n      'isImmediate', \"isImmediate\",\n      'predicateSQL', \"predicateSQL\",\n      'columns', json_object_agg(\"col\", \"dir\")\n    ) AS index FROM indexed_columns \n      GROUP BY \"schema\", \"tableName\", \"name\", \"unique\", \n         \"isPrimaryKey\", \"isReplicaIdentity\", \"isImmediate\", \"predicateSQL\")\n\n    SELECT json_build_object(\n      'tables', COALESCE((SELECT json_agg(\"table\") FROM tables), '[]'::json),\n      'indexes', COALESCE((SELECT json_agg(\"index\") FROM indexes), '[]'::json)\n    ) as \"publishedSchema\"\n  `\n  );\n}\n\nconst rawPublishedColumnSpec = publishedColumnSpec.extend({\n  collationIsDeterministic: v.boolean().nullable().optional(),\n});\n\nconst rawPublishedTableSpec = publishedTableSpec.extend({\n  columns: v.record(rawPublishedColumnSpec),\n});\n\nconst rawPublishedIndexSpec = publishedIndexSpec.extend({\n  predicateSQL: v.string().nullable().optional(),\n});\n\nexport type SkippedIndexDiagnostic = {\n  readonly schema: string;\n  readonly index: string;\n  readonly predicate: string;\n  readonly reason: UnsupportedPredicateReason;\n};\n\nconst skippedIndexDiagnostics = new WeakMap<\n  object,\n  readonly SkippedIndexDiagnostic[]\n>();\nexport function getSkippedIndexDiagnostics(\n  schema: PublishedSchema,\n): readonly SkippedIndexDiagnostic[] {\n  return skippedIndexDiagnostics.get(schema) ?? [];\n}\n\nexport function warnForSkippedIndexes(\n  lc: LogContext,\n  schema: PublishedSchema,\n  warned: Set<string> = new Set(),\n): void {\n  for (const diagnostic of getSkippedIndexDiagnostics(schema)) {\n    const key = JSON.stringify(diagnostic);\n    if (warned.has(key)) continue;\n    warned.add(key);\n    lc.warn?.('Skipping unsupported PostgreSQL partial index', diagnostic);\n    getOrCreateCounter('replication', 'skipped_index', {\n      description: 'Number of PostgreSQL indexes skipped during replication.',\n      unit: '{index}',\n    }).add(1, {reason: diagnostic.reason, partial: true});\n  }\n}\n\nexport const publishedSchema = v\n  .object({\n    tables: v.array(rawPublishedTableSpec),\n    indexes: v.array(rawPublishedIndexSpec),\n  })\n  .map(({tables: rawTables, indexes: rawIndexes}) => {\n    const diagnostics: SkippedIndexDiagnostic[] = [];\n    const indexes: PublishedIndexSpec[] = [];\n    for (const rawIndex of rawIndexes) {\n      const {predicateSQL, ...index} = rawIndex;\n      if (predicateSQL !== null && predicateSQL !== undefined) {\n        const table = rawTables.find(\n          candidate =>\n            candidate.schema === index.schema &&\n            candidate.name === index.tableName,\n        );\n        const translation = table\n          ? translatePostgresIndexPredicate(predicateSQL, table)\n          : {supported: false as const, reason: 'unpublished-column' as const};\n        if (!translation.supported) {\n          diagnostics.push({\n            schema: index.schema,\n            index: index.name,\n            predicate: predicateSQL,\n            reason: translation.reason,\n          });\n          continue;\n        }\n        indexes.push({...index, predicate: translation.predicate});\n      } else {\n        indexes.push(index);\n      }\n    }\n\n    const tables = rawTables.map(table => ({\n      ...table,\n      columns: Object.fromEntries(\n        Object.entries(table.columns).map(\n          ([name, {collationIsDeterministic: _, ...column}]) => [name, column],\n        ),\n      ),\n    }));\n    const zqlSpecs = computeZqlSpecsFromLiteSpecs(\n      tables.map(t => mapPostgresToLite(t)),\n      indexes.map(mapPostgresToLiteIndex),\n      {includeBackfillingColumns: true},\n    );\n    const result = {\n      indexes,\n\n      // Denormalize the schema such that each `table` includes the\n      // `replicaIdentityColumns` corresponding to the table's\n      // replica identity and associated primary key or index.\n      tables: tables.map(table => {\n        const replicaIdentityColumns: string[] = [];\n        switch (table.replicaIdentity) {\n          case 'd':\n            replicaIdentityColumns.push(...(table.primaryKey ?? []));\n            break;\n          case 'i':\n            replicaIdentityColumns.push(\n              ...Object.keys(\n                indexes.find(\n                  ind =>\n                    ind.schema === table.schema &&\n                    ind.tableName === table.name &&\n                    ind.isReplicaIdentity,\n                )?.columns ?? {},\n              ),\n            );\n            break;\n          case 'f': {\n            // For the key columns of replica identity \"full\", use the columns\n            // that the replicator/change-processor will end up using\n            // (in #getKey()) as the row key.\n            const zqlTable = zqlSpecs.get(liteTableName(table));\n            // Note: There zql spec will be absent if the table is not synced,\n            //       e.g. if it has no suitable unique indexes.\n            if (zqlTable) {\n              replicaIdentityColumns.push(...zqlTable.tableSpec.primaryKey);\n            }\n            break;\n          }\n        }\n        return {\n          ...table,\n          replicaIdentityColumns,\n        };\n      }),\n    };\n    skippedIndexDiagnostics.set(result, diagnostics);\n    return result;\n  });\n\nexport type PublishedSchema = v.Infer<typeof publishedSchema>;\n\nexport type PublishedTableWithReplicaIdentity =\n  PublishedSchema['tables'][number];\n\nconst publicationSchema = v.object({\n  pubname: v.string(),\n  pubinsert: v.boolean(),\n  pubupdate: v.boolean(),\n  pubdelete: v.boolean(),\n  pubtruncate: v.boolean(),\n});\n\nconst publicationsResultSchema = v.array(publicationSchema);\n\nexport type PublicationInfo = PublishedSchema & {\n  publications: v.Infer<typeof publicationsResultSchema>;\n};\n\n/**\n * Retrieves published tables and columns.\n */\nexport async function getPublicationInfo(\n  sql: postgres.Sql,\n  publications: string[],\n): Promise<PublicationInfo> {\n  const result = await sql.unsafe(/*sql*/ `\n  SELECT \n    schemaname AS \"schema\",\n    tablename AS \"table\", \n    json_object_agg(pubname, attnames) AS \"publications\"\n    FROM pg_publication_tables pb\n    WHERE pb.pubname IN (${literal(publications)})\n    GROUP BY schemaname, tablename;\n\n  SELECT ${Object.keys(publicationSchema.shape).join(\n    ',',\n  )} FROM pg_publication pb\n    WHERE pb.pubname IN (${literal(publications)})\n    ORDER BY pubname;\n\n  ${publishedSchemaQuery(publications)};\n`);\n\n  // The first query is used to check that tables in multiple publications\n  // always publish the same set of columns.\n  const publishedColumns = result[0] as {\n    schema: string;\n    table: string;\n    publications: Record<string, string[]>;\n  }[];\n  for (const {table, publications} of publishedColumns) {\n    let expected: Set<string>;\n    Object.entries(publications).forEach(([_, columns], i) => {\n      const cols = new Set(columns);\n      if (i === 0) {\n        expected = cols;\n      } else if (!equals(expected, cols)) {\n        throw new Error(\n          `Table ${table} is exported with different columns: [${[\n            ...expected,\n          ]}] vs [${[...cols]}]`,\n        );\n      }\n    });\n  }\n\n  assert(\n    result[2][0].publishedSchema,\n    () => `Invalid publishedSchema result ${BigIntJSON.stringify(result[2])}`,\n  );\n\n  const schema = v.parse(result[2][0].publishedSchema, publishedSchema);\n  const info = {\n    publications: v.parse(result[1], publicationsResultSchema),\n    ...schema,\n  };\n  skippedIndexDiagnostics.set(info, getSkippedIndexDiagnostics(schema));\n  return info;\n}\n"],"mappings":";;;;;;;;;;;;AAyBA,SAAgB,qBACd,cACA,wBAAwB,MACxB;CAOA,OACU;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;uBAiCW,QAAQ,YAAY,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;2BAqGlB,QAAQ,YAAY,EAAE;;aAEpC,sBAAsB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA8BnC;AAEA,IAAM,yBAAyB,oBAAoB,OAAO,EACxD,0BAA0BA,eAAE,QAAQ,CAAC,CAAC,SAAS,CAAC,CAAC,SAAS,EAC5D,CAAC;AAED,IAAM,wBAAwB,mBAAmB,OAAO,EACtD,SAASA,eAAE,OAAO,sBAAsB,EAC1C,CAAC;AAED,IAAM,wBAAwB,mBAAmB,OAAO,EACtD,cAAcA,eAAE,OAAO,CAAC,CAAC,SAAS,CAAC,CAAC,SAAS,EAC/C,CAAC;AASD,IAAM,0CAA0B,IAAI,QAGlC;AACF,SAAgB,2BACd,QACmC;CACnC,OAAO,wBAAwB,IAAI,MAAM,KAAK,CAAC;AACjD;AAEA,SAAgB,sBACd,IACA,QACA,yBAAsB,IAAI,IAAI,GACxB;CACN,KAAK,MAAM,cAAc,2BAA2B,MAAM,GAAG;EAC3D,MAAM,MAAM,KAAK,UAAU,UAAU;EACrC,IAAI,OAAO,IAAI,GAAG,GAAG;EACrB,OAAO,IAAI,GAAG;EACd,GAAG,OAAO,iDAAiD,UAAU;EACrE,mBAAmB,eAAe,iBAAiB;GACjD,aAAa;GACb,MAAM;EACR,CAAC,CAAC,CAAC,IAAI,GAAG;GAAC,QAAQ,WAAW;GAAQ,SAAS;EAAI,CAAC;CACtD;AACF;AAEA,IAAa,kBAAkBA,eAC5B,OAAO;CACN,QAAQA,eAAE,MAAM,qBAAqB;CACrC,SAASA,eAAE,MAAM,qBAAqB;AACxC,CAAC,CAAC,CACD,KAAK,EAAC,QAAQ,WAAW,SAAS,iBAAgB;CACjD,MAAM,cAAwC,CAAC;CAC/C,MAAM,UAAgC,CAAC;CACvC,KAAK,MAAM,YAAY,YAAY;EACjC,MAAM,EAAC,cAAc,GAAG,UAAS;EACjC,IAAI,iBAAiB,QAAQ,iBAAiB,KAAA,GAAW;GACvD,MAAM,QAAQ,UAAU,MACtB,cACE,UAAU,WAAW,MAAM,UAC3B,UAAU,SAAS,MAAM,SAC7B;GACA,MAAM,cAAc,QAChB,gCAAgC,cAAc,KAAK,IACnD;IAAC,WAAW;IAAgB,QAAQ;GAA6B;GACrE,IAAI,CAAC,YAAY,WAAW;IAC1B,YAAY,KAAK;KACf,QAAQ,MAAM;KACd,OAAO,MAAM;KACb,WAAW;KACX,QAAQ,YAAY;IACtB,CAAC;IACD;GACF;GACA,QAAQ,KAAK;IAAC,GAAG;IAAO,WAAW,YAAY;GAAS,CAAC;EAC3D,OACE,QAAQ,KAAK,KAAK;CAEtB;CAEA,MAAM,SAAS,UAAU,KAAI,WAAU;EACrC,GAAG;EACH,SAAS,OAAO,YACd,OAAO,QAAQ,MAAM,OAAO,CAAC,CAAC,KAC3B,CAAC,MAAM,EAAC,0BAA0B,GAAG,GAAG,cAAa,CAAC,MAAM,MAAM,CACrE,CACF;CACF,EAAE;CACF,MAAM,WAAW,6BACf,OAAO,KAAI,MAAK,kBAAkB,CAAC,CAAC,GACpC,QAAQ,IAAI,sBAAsB,GAClC,EAAC,2BAA2B,KAAI,CAClC;CACA,MAAM,SAAS;EACb;EAKA,QAAQ,OAAO,KAAI,UAAS;GAC1B,MAAM,yBAAmC,CAAC;GAC1C,QAAQ,MAAM,iBAAd;IACE,KAAK;KACH,uBAAuB,KAAK,GAAI,MAAM,cAAc,CAAC,CAAE;KACvD;IACF,KAAK;KACH,uBAAuB,KACrB,GAAG,OAAO,KACR,QAAQ,MACN,QACE,IAAI,WAAW,MAAM,UACrB,IAAI,cAAc,MAAM,QACxB,IAAI,iBACR,CAAC,EAAE,WAAW,CAAC,CACjB,CACF;KACA;IACF,KAAK,KAAK;KAIR,MAAM,WAAW,SAAS,IAAI,cAAc,KAAK,CAAC;KAGlD,IAAI,UACF,uBAAuB,KAAK,GAAG,SAAS,UAAU,UAAU;KAE9D;IACF;GACF;GACA,OAAO;IACL,GAAG;IACH;GACF;EACF,CAAC;CACH;CACA,wBAAwB,IAAI,QAAQ,WAAW;CAC/C,OAAO;AACT,CAAC;AAOH,IAAM,oBAAoBA,eAAE,OAAO;CACjC,SAASA,eAAE,OAAO;CAClB,WAAWA,eAAE,QAAQ;CACrB,WAAWA,eAAE,QAAQ;CACrB,WAAWA,eAAE,QAAQ;CACrB,aAAaA,eAAE,QAAQ;AACzB,CAAC;AAED,IAAM,2BAA2BA,eAAE,MAAM,iBAAiB;;;;AAS1D,eAAsB,mBACpB,KACA,cAC0B;CAC1B,MAAM,SAAS,MAAM,IAAI,OAAe;;;;;;2BAMf,QAAQ,YAAY,EAAE;;;WAGtC,OAAO,KAAK,kBAAkB,KAAK,CAAC,CAAC,KAC5C,GACF,EAAE;2BACuB,QAAQ,YAAY,EAAE;;;IAG7C,qBAAqB,YAAY,EAAE;CACtC;CAIC,MAAM,mBAAmB,OAAO;CAKhC,KAAK,MAAM,EAAC,OAAO,kBAAiB,kBAAkB;EACpD,IAAI;EACJ,OAAO,QAAQ,YAAY,CAAC,CAAC,SAAS,CAAC,GAAG,UAAU,MAAM;GACxD,MAAM,OAAO,IAAI,IAAI,OAAO;GAC5B,IAAI,MAAM,GACR,WAAW;QACN,IAAI,CAAC,OAAO,UAAU,IAAI,GAC/B,MAAM,IAAI,MACR,SAAS,MAAM,wCAAwC,CACrD,GAAG,QACL,EAAE,QAAQ,CAAC,GAAG,IAAI,EAAE,EACtB;EAEJ,CAAC;CACH;CAEA,OACE,OAAO,EAAE,CAAC,EAAE,CAAC,uBACP,kCAAkC,WAAW,UAAU,OAAO,EAAE,GACxE;CAEA,MAAM,SAASC,MAAQ,OAAO,EAAE,CAAC,EAAE,CAAC,iBAAiB,eAAe;CACpE,MAAM,OAAO;EACX,cAAcA,MAAQ,OAAO,IAAI,wBAAwB;EACzD,GAAG;CACL;CACA,wBAAwB,IAAI,MAAM,2BAA2B,MAAM,CAAC;CACpE,OAAO;AACT"}

package/out/zero-cache/src/services/change-source/pg/schema/replica-stage-enum.js.map

@@ -1 +1 @@
-{"version":3,"file":"replica-stage-enum.js","names":[],"sources":["../../../../../../../../zero-cache/src/services/change-source/pg/schema/replica-stage-enum.ts"],"sourcesContent":["import * as v from '../../../../../../shared/src/valita.ts';\n\n// Note: The stages are ordered by restore availability:\n// * InitialSync is the first to create the generation, by definition.\n// * Replicate has already restored and thus can be restored from\n// * Restore is in the process of awaiting the backup\nexport const InitialSync = 0;\nexport const Replicate = 1;\nexport const Restore = 2;\n\nexport type InitialSync = typeof InitialSync;\nexport type Replicate = typeof Replicate;\nexport type Restore = typeof Restore;\n\nexport const replicaStageSchema = v.literalUnion(\n  InitialSync,\n  Replicate,\n  Restore,\n);\n\nexport type ReplicaStage = v.Infer<typeof replicaStageSchema>;\n"],"mappings":";AAcA,IAAa,qBAAqB,aAAA,GAAA,GAAA,CAIlC"}
+{"version":3,"file":"replica-stage-enum.js","names":["v.literalUnion"],"sources":["../../../../../../../../zero-cache/src/services/change-source/pg/schema/replica-stage-enum.ts"],"sourcesContent":["import * as v from '../../../../../../shared/src/valita.ts';\n\n// Note: The stages are ordered by restore availability:\n// * InitialSync is the first to create the generation, by definition.\n// * Replicate has already restored and thus can be restored from\n// * Restore is in the process of awaiting the backup\nexport const InitialSync = 0;\nexport const Replicate = 1;\nexport const Restore = 2;\n\nexport type InitialSync = typeof InitialSync;\nexport type Replicate = typeof Replicate;\nexport type Restore = typeof Restore;\n\nexport const replicaStageSchema = v.literalUnion(\n  InitialSync,\n  Replicate,\n  Restore,\n);\n\nexport type ReplicaStage = v.Infer<typeof replicaStageSchema>;\n"],"mappings":";AAcA,IAAa,qBAAqBA,aAAAA,GAAAA,GAAAA,CAIlC"}

package/out/zero-cache/src/services/change-source/pg/schema/shard.js

@@ -168,8 +168,9 @@ function shardSetup(shardConfig, metadataPublication) {
 function dropShard(appID, shardID) {
 	const schema = `${appID}_${shardID}`;
 	const metadataPublication = metadataPublicationName(appID, shardID);
+	const defaultPublication = defaultPublicationName(appID, shardID);
 	return `
-    DROP PUBLICATION IF EXISTS ${id(defaultPublicationName(appID, shardID))};
+    DROP PUBLICATION IF EXISTS ${id(defaultPublication)};
     DROP PUBLICATION IF EXISTS ${id(metadataPublication)};
     DROP SCHEMA IF EXISTS ${id(schema)} CASCADE;
   `;
@@ -296,7 +297,7 @@ async function getReplicaAtVersion(lc, sql, shard, replicaVersion, id = null, co
 	return parse(result[0], replicaSchema, "passthrough");
 }
 async function getActiveReplicas(lc, sql, shard) {
-	const replicas = parse(await sql`
+	const results = await sql`
     SELECT
       replicas."id",
       replicas."rank",
@@ -317,7 +318,8 @@ async function getActiveReplicas(lc, sql, shard) {
         AND slots.restart_lsn IS NOT NULL
         AND slots.wal_status IS DISTINCT FROM 'lost'
       ORDER BY generation DESC, confirmed_flush_lsn DESC;
-  `, valita_exports.array(replicaStateSchema), "passthrough");
+  `;
+	const replicas = parse(results, valita_exports.array(replicaStateSchema), "passthrough");
 	lc.info?.(`current replicas`, { replicas });
 	return replicas;
 }
@@ -333,7 +335,7 @@ async function getActiveReplicas(lc, sql, shard) {
 */
 async function getRestoreCandidates(sql, shard, epoch) {
 	const schema = sql(upstreamSchema(shard));
-	return parse(await sql`
+	const results = await sql`
     SELECT
       replicas."id",
       replicas."rank",
@@ -368,7 +370,8 @@ async function getRestoreCandidates(sql, shard, epoch) {
         stage ASC,                 -- ordered by backup availalibility
         active DESC,               -- prefer active slots (fork over resume)
         confirmed_flush_lsn DESC;  -- prefer replicas further ahead
-  `, valita_exports.array(replicaStateSchema), "passthrough");
+  `;
+	return parse(results, valita_exports.array(replicaStateSchema), "passthrough");
 }
 async function getReplicaState(sql, shard, id) {
 	const results = await sql`
@@ -476,6 +479,6 @@ function replicaIdentitiesForTablesWithoutPrimaryKeys(pubs) {
 	} };
 }
 //#endregion
-export { SHARD_CONFIG_TABLE, createReplica, dropShard, ensureGlobalTables, getActiveReplicas, getInternalShardConfig, getMutationsTableDefinition, getReplicaAtVersion, getReplicaState, getRestoreCandidates, initInitialSyncReplica, initRestoreReplica, internalPublicationPrefix, legacyReplicationSlot, metadataPublicationName, replicaIdentitiesForTablesWithoutPrimaryKeys, replicationSlotExpression, replicationSlotPrefix, setupTablesAndReplication, setupTriggers, validatePublications };
+export { SHARD_CONFIG_TABLE, createReplica, dropShard, ensureGlobalTables, getActiveReplicas, getClientsTableDefinition, getInternalShardConfig, getMutationsTableDefinition, getReplicaAtVersion, getReplicaState, getRestoreCandidates, initInitialSyncReplica, initRestoreReplica, internalPublicationPrefix, legacyReplicationSlot, metadataPublicationName, replicaIdentitiesForTablesWithoutPrimaryKeys, replicationSlotExpression, replicationSlotPrefix, setupTablesAndReplication, setupTriggers, shardSetup, validatePublicationName, validatePublications };
 
 //# sourceMappingURL=shard.js.map