shapecraft 1.0.0

package/packages/core/src/shape/transforms/sql/parser/create-domain.ts

@@ -0,0 +1,105 @@
+import { TokenCursor } from "./cursor";
+import { Token } from "./lexer";
+
+export type DomainDef = {
+  schema?: string;
+  name: string;
+  baseTypeTokens: Token[];
+  checks: Token[][];
+  defaultTokens?: Token[];
+  notNull: boolean;
+};
+
+const qualifiedName = (parts: string[]): { schema?: string; name: string } => {
+  if (parts.length === 1) return { name: parts[0]! };
+  return { schema: parts[0]!, name: parts.slice(1).join(".") };
+};
+
+// Read until we hit a top-level keyword that ends a clause, or end of input.
+// Returns the consumed tokens (without the terminator).
+const readUntilClauseEnd = (c: TokenCursor): Token[] => {
+  const out: Token[] = [];
+  let depth = 0;
+  while (!c.done()) {
+    const t = c.peek();
+    if (t.kind === "punct" && t.value === "(") depth += 1;
+    if (t.kind === "punct" && t.value === ")") {
+      if (depth === 0) break;
+      depth -= 1;
+    }
+    if (depth === 0 && t.kind === "ident" && !t.quoted) {
+      const v = t.value;
+      if (
+        v === "default" ||
+        v === "check" ||
+        v === "constraint" ||
+        v === "collate" ||
+        v === "not" ||
+        v === "null"
+      )
+        break;
+    }
+    out.push(t);
+    c.consume();
+  }
+  return out;
+};
+
+export const parseCreateDomain = (tokens: Token[]): DomainDef => {
+  const c = new TokenCursor(tokens);
+  c.expectIdent("create");
+  if (c.consumeIdent("or")) c.expectIdent("replace");
+  c.expectIdent("domain");
+  if (c.consumeIdent("if")) {
+    c.expectIdent("not");
+    c.expectIdent("exists");
+  }
+
+  const nameParts = c.readDottedName();
+  const qn = qualifiedName(nameParts);
+  c.consumeIdent("as"); // AS is optional
+
+  const baseTypeTokens = readUntilClauseEnd(c);
+
+  const def: DomainDef = {
+    name: qn.name,
+    baseTypeTokens,
+    checks: [],
+    notNull: false,
+  };
+  if (qn.schema !== undefined) def.schema = qn.schema;
+
+  while (!c.done()) {
+    if (c.consumeIdent("default")) {
+      def.defaultTokens = readUntilClauseEnd(c);
+      continue;
+    }
+    if (c.consumeIdent("not")) {
+      c.consumeIdent("null");
+      def.notNull = true;
+      continue;
+    }
+    if (c.consumeIdent("null")) {
+      // explicit NULL allowed
+      continue;
+    }
+    if (c.consumeIdent("constraint")) {
+      // skip the constraint name
+      c.expectAnyIdent();
+      continue;
+    }
+    if (c.consumeIdent("check")) {
+      const inner = c.skipParens();
+      def.checks.push(inner);
+      continue;
+    }
+    if (c.consumeIdent("collate")) {
+      c.expectAnyIdent();
+      continue;
+    }
+    // unknown: skip one token to avoid infinite loop
+    c.consume();
+  }
+
+  return def;
+};