shapecraft 1.0.0

package/packages/core/src/shape/transforms/sql/from-sql.ts

@@ -0,0 +1,210 @@
+import { R } from "../../../common/result";
+import { annotate, shapes } from "../../shape";
+import { resolveDialect, SQLOptions } from "./options";
+import { decodeCheck } from "./parser/check-decoder";
+import { parseCreateDomain } from "./parser/create-domain";
+import { parseCreateTable, ParsedTable } from "./parser/create-table";
+import { parseCreateType } from "./parser/create-type";
+import { decodeDefault } from "./parser/default-decoder";
+import { LexError, tokenize } from "./parser/lexer";
+import {
+  parsePgType,
+  Registries,
+  resolvePgType,
+  ResolvedDomain,
+} from "./parser/pg-types";
+import { classifyStatement, splitStatements } from "./parser/statements";
+import { ParseError } from "./parser/cursor";
+
+const newRegistries = (): Registries => ({
+  enums: new Map(),
+  composites: new Map(),
+  domains: new Map(),
+});
+
+const registerDomain = (
+  registries: Registries,
+  raw: ReturnType<typeof parseCreateDomain>,
+): void => {
+  const baseSpec = parsePgType(raw.baseTypeTokens);
+  const resolved = resolvePgType(baseSpec, registries);
+  let shape = resolved.shape;
+  for (const chkTokens of raw.checks) {
+    const dec = decodeCheck(chkTokens, {
+      columnName: "value",
+      isInteger: shape.type === "number",
+      isString: shape.type === "string",
+      isArray: shape.type === "array",
+    });
+    if (dec.fullyDecoded) {
+      const u = dec.updates;
+      if (u.min !== undefined) shape = annotate.as(shape, { min: u.min });
+      if (u.max !== undefined) shape = annotate.as(shape, { max: u.max });
+      if (u.pattern !== undefined)
+        shape = annotate.as(shape, { pattern: u.pattern });
+    }
+    // undecoded domain checks are dropped — they tend to reference VALUE
+    // rather than a specific column, so the column-oriented check-decoder
+    // wouldn't be able to round-trip them onto a column anyway.
+  }
+  // Add domain-name format hint
+  const qualified =
+    raw.schema !== undefined ? `${raw.schema}.${raw.name}` : raw.name;
+  shape = annotate.as(shape, { format: `pgtype:${qualified}` });
+  if (raw.defaultTokens !== undefined) {
+    const dec = decodeDefault(raw.defaultTokens, null);
+    if (dec.kind === "literal") {
+      shape = annotate.as(shape, { default: dec.value });
+    } else if (dec.sql.length > 0) {
+      shape = annotate.as(shape, { defaultExpr: dec.sql });
+    }
+  }
+  if (raw.notNull) {
+    // domain says NOT NULL; column nullability is still column-level
+    // (the column's NOT NULL applies). We leave the domain's shape alone.
+  }
+  const def: ResolvedDomain = { name: raw.name, shape };
+  if (raw.schema !== undefined) def.schema = raw.schema;
+  registries.domains.set(qualified, def);
+  registries.domains.set(raw.name, def);
+};
+
+const assemble = (
+  parsedTables: ParsedTable[],
+): R.Result<shapes.Shape, string[]> => {
+  if (parsedTables.length === 0) {
+    return R.err(["no CREATE TABLE statements found"]);
+  }
+
+  // For a single table return the mapping directly
+  if (parsedTables.length === 1) {
+    const t = parsedTables[0]!;
+    return R.ok(makeMapping(t));
+  }
+
+  // For multiple tables, build a module
+  const tables: Record<string, shapes.ShapeMapping> = {};
+  const usedKeys = new Set<string>();
+  for (const t of parsedTables) {
+    let key = t.name;
+    if (usedKeys.has(key)) {
+      key = t.schema !== undefined ? `${t.schema}_${t.name}` : `${t.name}_2`;
+      // ensure uniqueness even with the prefix
+      let suffix = 2;
+      let cand = key;
+      while (usedKeys.has(cand)) {
+        cand = `${key}_${suffix}`;
+        suffix += 1;
+      }
+      key = cand;
+    }
+    usedKeys.add(key);
+    tables[key] = makeMapping(t);
+  }
+  return R.ok(shapes.module(tables));
+};
+
+const makeMapping = (t: ParsedTable): shapes.ShapeMapping => {
+  const rec: Record<string, shapes.Shape> = {};
+  for (const c of t.columns) rec[c.key] = c.shape;
+  let m: shapes.ShapeMapping = shapes.mapping(rec);
+  m = annotate.titled(m, t.mappingAnno.title) as shapes.ShapeMapping;
+  if (t.mappingAnno.schema !== undefined) {
+    m = annotate.as(m, { schema: t.mappingAnno.schema }) as shapes.ShapeMapping;
+  }
+  if (t.mappingAnno.check !== undefined) {
+    m = annotate.as(m, { check: t.mappingAnno.check }) as shapes.ShapeMapping;
+  }
+  if (t.mappingAnno.description !== undefined) {
+    m = annotate.described(m, t.mappingAnno.description) as shapes.ShapeMapping;
+  }
+  if (t.uniqueConstraints !== undefined && t.uniqueConstraints.length > 0) {
+    m = annotate.as(m, {
+      uniqueConstraints: t.uniqueConstraints,
+    }) as shapes.ShapeMapping;
+  }
+  return m;
+};
+
+export const fromSQL = (
+  sql: string,
+  opts?: SQLOptions,
+): R.Result<shapes.Shape, string[]> => {
+  const dialect = resolveDialect(opts);
+  let tokens;
+  try {
+    tokens = tokenize(sql);
+  } catch (e) {
+    if (e instanceof LexError) return R.err([`lex error: ${e.message}`]);
+    return R.err([`lex error: ${String(e)}`]);
+  }
+
+  const stmts = splitStatements(tokens).map(classifyStatement);
+  const registries = newRegistries();
+  const tableStmts: typeof stmts = [];
+  const errors: string[] = [];
+
+  // First pass: types and domains
+  for (const s of stmts) {
+    if (s.kind === "createType") {
+      try {
+        const parsed = parseCreateType(s.tokens);
+        if (parsed === null) continue;
+        if (parsed.kind === "enum") {
+          const def = parsed.def;
+          const qualified =
+            def.schema !== undefined ? `${def.schema}.${def.name}` : def.name;
+          registries.enums.set(qualified, def);
+          registries.enums.set(def.name, def);
+        } else {
+          const def = parsed.def;
+          const qualified =
+            def.schema !== undefined ? `${def.schema}.${def.name}` : def.name;
+          registries.composites.set(qualified, def);
+          registries.composites.set(def.name, def);
+        }
+      } catch (e) {
+        errors.push(formatParserError(e, "CREATE TYPE"));
+      }
+      continue;
+    }
+    if (s.kind === "createDomain") {
+      try {
+        const parsed = parseCreateDomain(s.tokens);
+        registerDomain(registries, parsed);
+      } catch (e) {
+        errors.push(formatParserError(e, "CREATE DOMAIN"));
+      }
+      continue;
+    }
+    if (s.kind === "createTable") {
+      tableStmts.push(s);
+      continue;
+    }
+    // createTablePartitionOf and skip — ignored
+  }
+
+  // Second pass: tables
+  const parsedTables: ParsedTable[] = [];
+  for (const s of tableStmts) {
+    try {
+      const parsed = parseCreateTable(s.tokens, { registries, dialect });
+      parsedTables.push(parsed);
+    } catch (e) {
+      errors.push(formatParserError(e, "CREATE TABLE"));
+    }
+  }
+
+  if (errors.length > 0 && parsedTables.length === 0) {
+    return R.err(errors);
+  }
+
+  return assemble(parsedTables);
+};
+
+const formatParserError = (e: unknown, ctx: string): string => {
+  if (e instanceof ParseError) return `${ctx}: ${e.message}`;
+  if (e instanceof LexError) return `${ctx}: ${e.message}`;
+  if (e instanceof Error) return `${ctx}: ${e.message}`;
+  return `${ctx}: ${String(e)}`;
+};

package/packages/core/src/shape/transforms/sql/index.ts

@@ -0,0 +1,3 @@
+export * from "./transform";
+export * from "./from-sql";
+export * from "./options";

package/packages/core/src/shape/transforms/sql/options.ts

@@ -0,0 +1,6 @@
+export type SQLDialect = "postgres" | "sqlite";
+
+export type SQLOptions = { dialect?: SQLDialect };
+
+export const resolveDialect = (opts?: SQLOptions): SQLDialect =>
+  opts?.dialect ?? "postgres";

package/packages/core/src/shape/transforms/sql/parser/check-decoder.ts

@@ -0,0 +1,457 @@
+import { Token } from "./lexer";
+
+export type CheckUpdate = {
+  min?: number;
+  max?: number;
+  pattern?: string;
+  uniqueItems?: boolean;
+  unionLiterals?: string[];
+  // SQLite-specific hints — interpreted by the column-meta applier
+  isBooleanInt?: boolean; // `col IN (0, 1)` on an integer column
+  isJsonValid?: boolean; // `json_valid(col)` (column is JSON-encoded)
+  tupleLength?: number; // `json_array_length(col) = N` together with isJsonValid
+};
+
+export type DecodedCheck = {
+  updates: CheckUpdate;
+  // True if the matcher consumed the whole expression. False means we couldn't
+  // fully decode and the caller should keep the raw text on `anno.check`.
+  fullyDecoded: boolean;
+};
+
+// Re-emit a token stream back to readable SQL. Used when we couldn't decode
+// and want to preserve the raw expression on `anno.check`.
+export const checkExprToSql = (tokens: Token[]): string => {
+  const parts: string[] = [];
+  let prev: Token | undefined;
+  for (const t of tokens) {
+    let s: string;
+    if (t.kind === "string") s = `'${t.value.replace(/'/g, "''")}'`;
+    else if (t.kind === "ident" && t.quoted)
+      s = `"${t.value.replace(/"/g, '""')}"`;
+    else s = t.value;
+    const needsSpace =
+      prev !== undefined &&
+      !(t.kind === "punct" && (t.value === "," || t.value === ")")) &&
+      !(prev.kind === "punct" && prev.value === "(") &&
+      !(t.kind === "punct" && t.value === "(" && prev.kind === "ident") &&
+      !(prev.kind === "punct" && prev.value === ".") &&
+      !(t.kind === "punct" && t.value === ".");
+    parts.push(needsSpace ? " " + s : s);
+    prev = t;
+  }
+  return parts.join("");
+};
+
+const isColRef = (t: Token | undefined, col: string): boolean =>
+  t !== undefined &&
+  t.kind === "ident" &&
+  ((!t.quoted && t.value === col.toLowerCase()) ||
+    (t.quoted === true && t.value === col));
+
+const numLit = (t: Token | undefined): number | null => {
+  if (t === undefined) return null;
+  if (t.kind === "number") {
+    const n = Number(t.value);
+    if (isFinite(n)) return n;
+    return null;
+  }
+  return null;
+};
+
+// Try to extract a numeric literal possibly with a leading sign
+const numLitWithSign = (
+  tokens: Token[],
+  start: number,
+): { value: number; consumed: number } | null => {
+  const t = tokens[start];
+  if (t === undefined) return null;
+  if (
+    t.kind === "op" &&
+    (t.value === "-" || t.value === "+") &&
+    tokens[start + 1]?.kind === "number"
+  ) {
+    const n = Number(`${t.value}${tokens[start + 1]!.value}`);
+    if (isFinite(n)) return { value: n, consumed: 2 };
+    return null;
+  }
+  const n = numLit(t);
+  if (n !== null) return { value: n, consumed: 1 };
+  return null;
+};
+
+// Strip wrapping parens until the body no longer starts with them
+const stripParens = (tokens: Token[]): Token[] => {
+  let body = tokens;
+  while (
+    body.length >= 2 &&
+    body[0]?.kind === "punct" &&
+    body[0]?.value === "(" &&
+    body[body.length - 1]?.kind === "punct" &&
+    body[body.length - 1]?.value === ")"
+  ) {
+    // ensure these are matched
+    let depth = 0;
+    let balanced = true;
+    for (let i = 0; i < body.length; i += 1) {
+      const t = body[i]!;
+      if (t.kind === "punct" && t.value === "(") depth += 1;
+      else if (t.kind === "punct" && t.value === ")") {
+        depth -= 1;
+        if (depth === 0 && i !== body.length - 1) {
+          balanced = false;
+          break;
+        }
+      }
+    }
+    if (!balanced) break;
+    body = body.slice(1, -1);
+  }
+  return body;
+};
+
+type CheckCtx = {
+  columnName: string;
+  // Hint for how to interpret bounds (e.g. integer columns can decode `> N` → min=N+1)
+  isInteger: boolean;
+  // Whether the column is a string-typed column eligible for `IN ('...')` → union conversion
+  isString: boolean;
+  // Whether the column is an array-typed column eligible for cardinality/uniqueItems
+  isArray: boolean;
+};
+
+// Some pg checks wrap a real predicate in `<col> IS NULL OR <real>`. That
+// pattern lets a CHECK on a nullable column not fail when the value is NULL.
+// We peel that off and decode the real predicate.
+const peelIsNullOr = (tokens: Token[], ctx: CheckCtx): Token[] | null => {
+  // Look for: <col> IS NULL OR <rest>
+  if (tokens.length < 5) return null;
+  if (!isColRef(tokens[0], ctx.columnName)) return null;
+  if (tokens[1]?.kind !== "ident" || tokens[1].value !== "is") return null;
+  if (tokens[2]?.kind !== "ident" || tokens[2].value !== "null") return null;
+  if (tokens[3]?.kind !== "ident" || tokens[3].value !== "or") return null;
+  return stripParens(tokens.slice(4));
+};
+
+const tryDecode = (tokens: Token[], ctx: CheckCtx): CheckUpdate | null => {
+  if (tokens.length === 0) return null;
+
+  // <col> <op> <num>
+  if (isColRef(tokens[0], ctx.columnName)) {
+    const opTok = tokens[1];
+    const op = opTok?.kind === "op" ? opTok.value : null;
+    if (op === ">=") {
+      const n = numLitWithSign(tokens, 2);
+      if (n !== null && n.consumed === tokens.length - 2)
+        return { min: n.value };
+    }
+    if (op === ">") {
+      const n = numLitWithSign(tokens, 2);
+      if (n !== null && n.consumed === tokens.length - 2 && ctx.isInteger)
+        return { min: n.value + 1 };
+    }
+    if (op === "<=") {
+      const n = numLitWithSign(tokens, 2);
+      if (n !== null && n.consumed === tokens.length - 2)
+        return { max: n.value };
+    }
+    if (op === "<") {
+      const n = numLitWithSign(tokens, 2);
+      if (n !== null && n.consumed === tokens.length - 2 && ctx.isInteger)
+        return { max: n.value - 1 };
+    }
+    if (op === "~") {
+      const next = tokens[2];
+      if (next?.kind === "string" && tokens.length === 3) {
+        return { pattern: next.value };
+      }
+    }
+
+    // <col> REGEXP '<pat>' (SQLite). REGEXP is parsed as an ident keyword.
+    if (
+      tokens[1]?.kind === "ident" &&
+      tokens[1].value === "regexp" &&
+      tokens[2]?.kind === "string" &&
+      tokens.length === 3
+    ) {
+      return { pattern: tokens[2].value };
+    }
+
+    // <col> BETWEEN <num> AND <num>
+    if (tokens[1]?.kind === "ident" && tokens[1].value === "between") {
+      const lo = numLitWithSign(tokens, 2);
+      if (lo !== null) {
+        const andIdx = 2 + lo.consumed;
+        if (
+          tokens[andIdx]?.kind === "ident" &&
+          tokens[andIdx].value === "and"
+        ) {
+          const hi = numLitWithSign(tokens, andIdx + 1);
+          if (hi !== null && andIdx + 1 + hi.consumed === tokens.length)
+            return { min: lo.value, max: hi.value };
+        }
+      }
+    }
+
+    // <col> IN (0, 1) → boolean hint (SQLite encodes booleans as INTEGER).
+    if (
+      ctx.isInteger &&
+      tokens[1]?.kind === "ident" &&
+      tokens[1].value === "in" &&
+      tokens[2]?.kind === "punct" &&
+      tokens[2].value === "(" &&
+      tokens.length === 7 &&
+      tokens[6]?.kind === "punct" &&
+      tokens[6].value === ")"
+    ) {
+      const a = numLitWithSign(tokens, 3);
+      const comma = tokens[3 + (a?.consumed ?? 0)];
+      const b = numLitWithSign(tokens, 3 + (a?.consumed ?? 0) + 1);
+      if (
+        a !== null &&
+        b !== null &&
+        comma?.kind === "punct" &&
+        comma.value === "," &&
+        ((a.value === 0 && b.value === 1) || (a.value === 1 && b.value === 0))
+      ) {
+        return { isBooleanInt: true };
+      }
+    }
+
+    // <col> IN ('a', 'b', ...) → union of string literals
+    if (
+      ctx.isString &&
+      tokens[1]?.kind === "ident" &&
+      tokens[1].value === "in" &&
+      tokens[2]?.kind === "punct" &&
+      tokens[2].value === "("
+    ) {
+      const lits: string[] = [];
+      let i = 3;
+      let ok = true;
+      while (i < tokens.length) {
+        const t = tokens[i]!;
+        if (t.kind === "punct" && t.value === ")") {
+          i += 1;
+          break;
+        }
+        if (t.kind !== "string") {
+          ok = false;
+          break;
+        }
+        lits.push(t.value);
+        i += 1;
+        const nxt = tokens[i];
+        if (nxt?.kind === "punct" && nxt.value === ",") {
+          i += 1;
+          continue;
+        }
+        if (nxt?.kind === "punct" && nxt.value === ")") {
+          i += 1;
+          break;
+        }
+        ok = false;
+        break;
+      }
+      if (ok && i === tokens.length && lits.length > 0)
+        return { unionLiterals: lits };
+    }
+  }
+
+  // char_length(<col>) >= N  /  length(<col>) >= N  → string min
+  // char_length(<col>) <= N → string max
+  if (
+    tokens[0]?.kind === "ident" &&
+    (tokens[0].value === "char_length" || tokens[0].value === "length") &&
+    tokens[1]?.kind === "punct" &&
+    tokens[1].value === "(" &&
+    isColRef(tokens[2], ctx.columnName) &&
+    tokens[3]?.kind === "punct" &&
+    tokens[3].value === ")"
+  ) {
+    const op = tokens[4]?.kind === "op" ? tokens[4].value : null;
+    if (op === ">=") {
+      const n = numLitWithSign(tokens, 5);
+      if (n !== null && 5 + n.consumed === tokens.length)
+        return { min: n.value };
+    }
+    if (op === "<=") {
+      const n = numLitWithSign(tokens, 5);
+      if (n !== null && 5 + n.consumed === tokens.length)
+        return { max: n.value };
+    }
+  }
+
+  // cardinality(<col>) >= N / <= N → array min/max (postgres)
+  // json_array_length(<col>) >= N / <= N → array min/max (sqlite)
+  if (
+    tokens[0]?.kind === "ident" &&
+    (tokens[0].value === "cardinality" ||
+      tokens[0].value === "json_array_length") &&
+    tokens[1]?.kind === "punct" &&
+    tokens[1].value === "(" &&
+    isColRef(tokens[2], ctx.columnName) &&
+    tokens[3]?.kind === "punct" &&
+    tokens[3].value === ")"
+  ) {
+    const op = tokens[4]?.kind === "op" ? tokens[4].value : null;
+    if (op === ">=") {
+      const n = numLitWithSign(tokens, 5);
+      if (n !== null && 5 + n.consumed === tokens.length)
+        return { min: n.value };
+    }
+    if (op === "<=") {
+      const n = numLitWithSign(tokens, 5);
+      if (n !== null && 5 + n.consumed === tokens.length)
+        return { max: n.value };
+    }
+  }
+
+  // json_valid(<col>) → isJsonValid (SQLite JSON-encoded column).
+  if (
+    tokens[0]?.kind === "ident" &&
+    tokens[0].value === "json_valid" &&
+    tokens[1]?.kind === "punct" &&
+    tokens[1].value === "(" &&
+    isColRef(tokens[2], ctx.columnName) &&
+    tokens[3]?.kind === "punct" &&
+    tokens[3].value === ")" &&
+    tokens.length === 4
+  ) {
+    return { isJsonValid: true };
+  }
+
+  // json_valid(<col>) AND json_array_length(<col>) = N → tuple of length N.
+  if (
+    tokens[0]?.kind === "ident" &&
+    tokens[0].value === "json_valid" &&
+    tokens[1]?.kind === "punct" &&
+    tokens[1].value === "(" &&
+    isColRef(tokens[2], ctx.columnName) &&
+    tokens[3]?.kind === "punct" &&
+    tokens[3].value === ")" &&
+    tokens[4]?.kind === "ident" &&
+    tokens[4].value === "and" &&
+    tokens[5]?.kind === "ident" &&
+    tokens[5].value === "json_array_length" &&
+    tokens[6]?.kind === "punct" &&
+    tokens[6].value === "(" &&
+    isColRef(tokens[7], ctx.columnName) &&
+    tokens[8]?.kind === "punct" &&
+    tokens[8].value === ")" &&
+    tokens[9]?.kind === "op" &&
+    tokens[9].value === "="
+  ) {
+    const n = numLitWithSign(tokens, 10);
+    if (n !== null && 10 + n.consumed === tokens.length) {
+      return { isJsonValid: true, tupleLength: n.value };
+    }
+  }
+
+  // shapecraft_array_is_unique(<col>) → uniqueItems
+  if (
+    tokens[0]?.kind === "ident" &&
+    tokens[0].value === "shapecraft_array_is_unique" &&
+    tokens[1]?.kind === "punct" &&
+    tokens[1].value === "(" &&
+    isColRef(tokens[2], ctx.columnName) &&
+    tokens[3]?.kind === "punct" &&
+    tokens[3].value === ")" &&
+    tokens.length === 4
+  ) {
+    return { uniqueItems: true };
+  }
+
+  // SQLite uniqueItems pattern:
+  //   (SELECT COUNT(*) FROM json_each(<col>))
+  //     = (SELECT COUNT(DISTINCT value) FROM json_each(<col>))
+  if (matchSqliteUniqueItems(tokens, ctx)) {
+    return { uniqueItems: true };
+  }
+
+  return null;
+};
+
+// Match the exact token shape:
+//   ( SELECT COUNT ( * ) FROM json_each ( <col> ) )
+//   = ( SELECT COUNT ( DISTINCT value ) FROM json_each ( <col> ) )
+const matchSqliteUniqueItems = (tokens: Token[], ctx: CheckCtx): boolean => {
+  const isIdent = (t: Token | undefined, v: string): boolean =>
+    t !== undefined && t.kind === "ident" && t.value === v;
+  const isPunct = (t: Token | undefined, v: string): boolean =>
+    t !== undefined && t.kind === "punct" && t.value === v;
+  const isOp = (t: Token | undefined, v: string): boolean =>
+    t !== undefined && t.kind === "op" && t.value === v;
+
+  // Skim past wrapping parens (peeled by stripParens already, but the
+  // body itself is two paren-wrapped subqueries joined by `=`).
+  let i = 0;
+  if (!isPunct(tokens[i], "(")) return false;
+  i += 1;
+  if (!isIdent(tokens[i], "select")) return false;
+  i += 1;
+  if (!isIdent(tokens[i], "count")) return false;
+  i += 1;
+  if (!isPunct(tokens[i], "(")) return false;
+  i += 1;
+  if (!isOp(tokens[i], "*")) return false;
+  i += 1;
+  if (!isPunct(tokens[i], ")")) return false;
+  i += 1;
+  if (!isIdent(tokens[i], "from")) return false;
+  i += 1;
+  if (!isIdent(tokens[i], "json_each")) return false;
+  i += 1;
+  if (!isPunct(tokens[i], "(")) return false;
+  i += 1;
+  if (!isColRef(tokens[i], ctx.columnName)) return false;
+  i += 1;
+  if (!isPunct(tokens[i], ")")) return false;
+  i += 1;
+  if (!isPunct(tokens[i], ")")) return false;
+  i += 1;
+  if (!isOp(tokens[i], "=")) return false;
+  i += 1;
+  if (!isPunct(tokens[i], "(")) return false;
+  i += 1;
+  if (!isIdent(tokens[i], "select")) return false;
+  i += 1;
+  if (!isIdent(tokens[i], "count")) return false;
+  i += 1;
+  if (!isPunct(tokens[i], "(")) return false;
+  i += 1;
+  if (!isIdent(tokens[i], "distinct")) return false;
+  i += 1;
+  if (!isIdent(tokens[i], "value")) return false;
+  i += 1;
+  if (!isPunct(tokens[i], ")")) return false;
+  i += 1;
+  if (!isIdent(tokens[i], "from")) return false;
+  i += 1;
+  if (!isIdent(tokens[i], "json_each")) return false;
+  i += 1;
+  if (!isPunct(tokens[i], "(")) return false;
+  i += 1;
+  if (!isColRef(tokens[i], ctx.columnName)) return false;
+  i += 1;
+  if (!isPunct(tokens[i], ")")) return false;
+  i += 1;
+  if (!isPunct(tokens[i], ")")) return false;
+  i += 1;
+  return i === tokens.length;
+};
+
+export const decodeCheck = (
+  rawTokens: Token[],
+  ctx: CheckCtx,
+): DecodedCheck => {
+  let body = stripParens(rawTokens);
+
+  // try peel IS NULL OR …
+  const peeled = peelIsNullOr(body, ctx);
+  if (peeled !== null) body = peeled;
+
+  const updates = tryDecode(body, ctx);
+  if (updates !== null) return { updates, fullyDecoded: true };
+  return { updates: {}, fullyDecoded: false };
+};