@kubuild/ai 0.3.0

package/dist/chunk-UH24LHRY.js

@@ -0,0 +1,443 @@
+// src/core/prompt-compiler.ts
+function compileComponentCatalog(registry) {
+  if (!registry) return [];
+  const definitions = registry.list();
+  return definitions.map((def) => {
+    const props = [];
+    if (def.propFields && Array.isArray(def.propFields)) {
+      for (const field of def.propFields) {
+        props.push({
+          name: field.name,
+          type: field.type || "string",
+          defaultValue: field.defaultValue,
+          options: field.options?.map((o) => o.value),
+          description: field.description
+        });
+      }
+    }
+    return {
+      type: def.type,
+      label: def.label || def.type,
+      category: def.category || "custom",
+      description: def.description,
+      acceptsChildren: def.acceptsChildren ?? false,
+      allowedChildren: def.allowedChildren,
+      disallowedParents: def.disallowedParents,
+      defaultProps: def.defaultProps,
+      props: props.length > 0 ? props : void 0
+    };
+  });
+}
+function buildSystemPrompt(options) {
+  const { catalog, mode, prefix, stylePreference } = options;
+  const catalogSummary = catalog.map((c) => {
+    let desc = `- **${c.type}** (${c.category}): ${c.label}`;
+    if (c.description) desc += ` - ${c.description}`;
+    desc += ` | acceptsChildren: ${c.acceptsChildren}`;
+    if (c.allowedChildren && c.allowedChildren.length > 0) {
+      desc += ` | allowedChildren: [${c.allowedChildren.join(", ")}]`;
+    }
+    if (c.props && c.props.length > 0) {
+      const propList = c.props.map((p) => {
+        let pStr = `${p.name} (${p.type})`;
+        if (p.options) pStr += `[${p.options.map(String).join("|")}]`;
+        return pStr;
+      }).join(", ");
+      desc += ` | props: { ${propList} }`;
+    }
+    return desc;
+  }).join("\n");
+  let modeInstructions = "";
+  if (mode === "full-page") {
+    modeInstructions = `
+You must generate a complete, valid KUBUILD PageDocument JSON with this exact structure:
+{
+  "schema": "stora.page",
+  "version": "1.0.0",
+  "metadata": {
+    "title": "<Page Title>",
+    "description": "<Page Description>",
+    "tags": ["<tag1>", "<tag2>"],
+    "category": "<category>"
+  },
+  "document": {
+    "id": "root-page",
+    "type": "page",
+    "styles": {
+      "base": {
+        "backgroundColor": "#ffffff",
+        "fontFamily": "Inter, system-ui, sans-serif",
+        "color": "#111827",
+        "minHeight": "100vh"
+      }
+    },
+    "children": [
+      // Sections here...
+    ]
+  }
+}
+`;
+  } else if (mode === "section") {
+    modeInstructions = `
+You must generate a single valid KUBUILD section Node JSON with this structure:
+{
+  "id": "section_<unique_suffix>",
+  "type": "section",
+  "styles": {
+    "base": {
+      "paddingTop": "64px",
+      "paddingBottom": "64px",
+      "paddingLeft": "24px",
+      "paddingRight": "24px"
+    }
+  },
+  "children": [
+    // Container and child nodes...
+  ]
+}
+`;
+  } else if (mode === "refactor") {
+    modeInstructions = `
+You must refactor the provided KUBUILD Node based on the user's instructions.
+Return ONLY the updated Node JSON while maintaining structural integrity.
+`;
+  }
+  const styleGuide = stylePreference ? `
+User Style Preference: "${stylePreference}". Follow this visual aesthetic closely.` : "\nStyle with modern, clean, high-conversion design patterns (accessible contrast, harmonious spacing, beautiful typography, subtle shadows/borders).";
+  return `${prefix ? prefix + "\n\n" : ""}You are an expert web designer and KUBUILD architecture generator.
+KUBUILD is a component-driven visual website builder. Your job is to output pure, valid JSON (and NOTHING ELSE) matching the KUBUILD schema and the registered component definitions.
+
+### RULES & CONSTRAINTS:
+1. Output MUST be valid JSON only. No markdown fences, no explanatory text, no HTML tags outside html-embed.
+2. Only use component types listed in the Component Catalog below. Do NOT invent new component types.
+3. Obey \`acceptsChildren\` and \`allowedChildren\`:
+   - Layout hierarchy: \`page\` -> \`section\` -> \`container\` or \`columns\` -> content components (\`heading\`, \`paragraph\`, \`button\`, \`image\`, \`form\`, etc.).
+   - Leaf components (e.g. \`button\`, \`input\`, \`heading\`, \`text\`, \`badge\`) must have \`acceptsChildren: false\` and empty or omitted \`children\`.
+4. Every node MUST have a unique string \`id\` (e.g. "hero-section", "hero-title", "cta-btn").
+5. Responsive & Interactive State Styling:
+   - Use the \`styles\` object with breakpoint keys: \`base\` (all viewports/desktop), and optionally \`tablet\` or \`mobile\`.
+   - In \`base\`, \`tablet\`, and \`mobile\`, values MUST be CSS primitives (strings or numbers, e.g. "backgroundColor": "#3b82f6"). NEVER put nested objects or ":hover" inside "base"!
+   - For hover, active, or interactive pseudo-classes, use \`styles.states\` with pseudo-class keys (e.g. \`":hover"\`), like:
+     "styles": {
+       "base": { "backgroundColor": "#3b82f6", "color": "#ffffff" },
+       "states": {
+         ":hover": { "backgroundColor": "#2563eb" }
+       }
+     }
+6. NEVER include dangerous scripts, \`javascript:\` URIs, or malicious payload attributes.
+${styleGuide}
+
+### OUTPUT SPECIFICATION:
+${modeInstructions}
+
+### REGISTERED COMPONENT CATALOG:
+${catalogSummary || "(Standard core components: page, section, container, columns, heading, paragraph, text, button, input, textarea, select, image, video, icon, badge, code-block)"}
+`;
+}
+function buildJsonSchemaForMode(mode) {
+  const nodeSchema = {
+    type: "object",
+    required: ["id", "type"],
+    properties: {
+      id: { type: "string" },
+      type: { type: "string" },
+      props: { type: "object" },
+      styles: {
+        type: "object",
+        properties: {
+          base: { type: "object" },
+          tablet: { type: "object" },
+          mobile: { type: "object" },
+          states: { type: "object" }
+        }
+      },
+      children: {
+        type: "array",
+        items: { $ref: "#" }
+      }
+    }
+  };
+  if (mode === "full-page") {
+    return {
+      type: "object",
+      required: ["schema", "version", "document"],
+      properties: {
+        schema: { type: "string", const: "stora.page" },
+        version: { type: "string", const: "1.0.0" },
+        metadata: {
+          type: "object",
+          properties: {
+            title: { type: "string" },
+            description: { type: "string" },
+            tags: { type: "array", items: { type: "string" } },
+            category: { type: "string" }
+          }
+        },
+        document: nodeSchema
+      }
+    };
+  }
+  return nodeSchema;
+}
+
+// src/core/normalizer.ts
+import {
+  PageDocumentSchema,
+  DocumentMetadataSchema,
+  NodeSchema,
+  CURRENT_SCHEMA_VERSION,
+  SCHEMA_NAME
+} from "@kubuild/schema";
+import {
+  validateDocument,
+  validateDocumentSecurity
+} from "@kubuild/core";
+function extractJsonFromResponse(raw) {
+  let cleaned = raw.trim();
+  if (cleaned.startsWith("```")) {
+    const firstNewline = cleaned.indexOf("\n");
+    if (firstNewline !== -1) {
+      cleaned = cleaned.slice(firstNewline + 1);
+    }
+    if (cleaned.endsWith("```")) {
+      cleaned = cleaned.slice(0, cleaned.length - 3).trim();
+    }
+  }
+  const firstBrace = cleaned.indexOf("{");
+  const lastBrace = cleaned.lastIndexOf("}");
+  if (firstBrace !== -1 && lastBrace !== -1 && lastBrace > firstBrace) {
+    cleaned = cleaned.slice(firstBrace, lastBrace + 1);
+  }
+  try {
+    return JSON.parse(cleaned);
+  } catch (err) {
+    const msg = err instanceof Error ? err.message : String(err);
+    throw new Error(`Failed to parse AI response as JSON: ${msg}. Response was: ${raw.slice(0, 200)}...`);
+  }
+}
+function normalizeStyles(rawStyles) {
+  if (!rawStyles || typeof rawStyles !== "object" || Array.isArray(rawStyles)) {
+    return void 0;
+  }
+  const raw = rawStyles;
+  const result = {};
+  const states = {};
+  if (raw.states && typeof raw.states === "object" && !Array.isArray(raw.states)) {
+    for (const [stateKey, stateVal] of Object.entries(raw.states)) {
+      if (stateVal && typeof stateVal === "object" && !Array.isArray(stateVal)) {
+        states[stateKey] = { ...stateVal };
+      }
+    }
+  }
+  const knownBreakpoints = ["base", "desktop", "tablet", "mobile"];
+  const hasBreakpoint = Object.keys(raw).some((k) => knownBreakpoints.includes(k));
+  if (!hasBreakpoint) {
+    const baseStyles = {};
+    for (const [key, value] of Object.entries(raw)) {
+      if (key === "states") continue;
+      if (key.startsWith(":") || key === "hover" || key === "focus" || key === "active") {
+        const stateKey = key.startsWith(":") ? key : `:${key}`;
+        if (value && typeof value === "object" && !Array.isArray(value)) {
+          states[stateKey] = { ...states[stateKey] || {}, ...value };
+        }
+      } else if (typeof value !== "object" || value === null) {
+        baseStyles[key] = value;
+      }
+    }
+    if (Object.keys(baseStyles).length > 0) {
+      result.base = baseStyles;
+    }
+  } else {
+    for (const bp of knownBreakpoints) {
+      const bpVal = raw[bp];
+      if (bpVal && typeof bpVal === "object" && !Array.isArray(bpVal)) {
+        const cleanBp = {};
+        for (const [key, value] of Object.entries(bpVal)) {
+          if (key.startsWith(":") || key === "hover" || key === "focus" || key === "active") {
+            const stateKey = key.startsWith(":") ? key : `:${key}`;
+            if (value && typeof value === "object" && !Array.isArray(value)) {
+              states[stateKey] = { ...states[stateKey] || {}, ...value };
+            }
+          } else if (typeof value !== "object" || value === null) {
+            cleanBp[key] = value;
+          }
+        }
+        if (Object.keys(cleanBp).length > 0) {
+          result[bp] = cleanBp;
+        }
+      }
+    }
+    for (const [key, value] of Object.entries(raw)) {
+      if (knownBreakpoints.includes(key) || key === "states") continue;
+      if (key.startsWith(":") || key === "hover" || key === "focus" || key === "active") {
+        const stateKey = key.startsWith(":") ? key : `:${key}`;
+        if (value && typeof value === "object" && !Array.isArray(value)) {
+          states[stateKey] = { ...states[stateKey] || {}, ...value };
+        }
+      }
+    }
+  }
+  const cleanStates = {};
+  for (const [stateKey, stateDef] of Object.entries(states)) {
+    const cleanDef = {};
+    for (const [propKey, propVal] of Object.entries(stateDef)) {
+      if (typeof propVal !== "object" || propVal === null) {
+        cleanDef[propKey] = propVal;
+      }
+    }
+    if (Object.keys(cleanDef).length > 0) {
+      cleanStates[stateKey] = cleanDef;
+    }
+  }
+  if (Object.keys(cleanStates).length > 0) {
+    result.states = cleanStates;
+  }
+  return Object.keys(result).length > 0 ? result : void 0;
+}
+function normalizeNodeTree(node, usedIds = /* @__PURE__ */ new Set(), depth = 0) {
+  if (!node || typeof node !== "object") {
+    throw new Error(`Invalid node at depth ${depth}: must be an object`);
+  }
+  const raw = node;
+  const type = typeof raw.type === "string" && raw.type.trim() ? raw.type.trim() : "container";
+  let id = typeof raw.id === "string" && raw.id.trim() ? raw.id.trim() : "";
+  if (!id || usedIds.has(id)) {
+    let counter = 1;
+    let candidate = `${type}-${counter}`;
+    while (usedIds.has(candidate)) {
+      counter++;
+      candidate = `${type}-${counter}`;
+    }
+    id = candidate;
+  }
+  usedIds.add(id);
+  const props = raw.props && typeof raw.props === "object" && !Array.isArray(raw.props) ? { ...raw.props } : {};
+  if (type === "heading" && typeof props.level === "string") {
+    const parsedLevel = parseInt(props.level, 10);
+    if (!isNaN(parsedLevel) && parsedLevel >= 1 && parsedLevel <= 6) {
+      props.level = parsedLevel;
+    }
+  }
+  const styles = normalizeStyles(raw.styles);
+  let children = void 0;
+  if (Array.isArray(raw.children)) {
+    children = raw.children.filter((c) => c && typeof c === "object").map((c) => normalizeNodeTree(c, usedIds, depth + 1));
+  }
+  return {
+    id,
+    type,
+    props: Object.keys(props).length > 0 ? props : void 0,
+    styles,
+    children
+  };
+}
+function normalizeAndValidatePageDocument(rawJson, securityLimits) {
+  if (!rawJson || typeof rawJson !== "object") {
+    throw new Error("AI output must be a JSON object");
+  }
+  const obj = rawJson;
+  let docObject;
+  if (obj.type === "page" && !obj.document) {
+    docObject = {
+      schema: SCHEMA_NAME,
+      version: CURRENT_SCHEMA_VERSION,
+      metadata: {
+        title: typeof obj.props === "object" && obj.props !== null && obj.props.title || "AI Generated Page",
+        description: "Generated by KUBUILD AI"
+      },
+      document: obj
+    };
+  } else {
+    docObject = {
+      schema: SCHEMA_NAME,
+      version: typeof obj.version === "string" ? obj.version : CURRENT_SCHEMA_VERSION,
+      metadata: typeof obj.metadata === "object" && obj.metadata !== null ? obj.metadata : { title: "AI Generated Page", description: "Generated by KUBUILD AI" },
+      document: obj.document || obj
+    };
+  }
+  const usedIds = /* @__PURE__ */ new Set();
+  const normalizedRoot = normalizeNodeTree(docObject.document, usedIds);
+  if (normalizedRoot.type !== "page") {
+    normalizedRoot.type = "page";
+  }
+  const parsedMetadata = DocumentMetadataSchema.parse({
+    title: "AI Generated Page",
+    description: "Generated by KUBUILD AI",
+    author: "KUBUILD AI",
+    tags: ["ai-generated"],
+    category: "general",
+    version: "1.0.0",
+    ...typeof docObject.metadata === "object" && docObject.metadata !== null ? docObject.metadata : {}
+  });
+  const finalDocument = {
+    schema: SCHEMA_NAME,
+    version: CURRENT_SCHEMA_VERSION,
+    metadata: parsedMetadata,
+    document: normalizedRoot
+  };
+  const parsed = PageDocumentSchema.safeParse(finalDocument);
+  if (!parsed.success) {
+    const issues = parsed.error.issues.map((i) => `${i.path.join(".")}: ${i.message}`).join("; ");
+    throw new Error(`Document schema validation failed: ${issues}`);
+  }
+  const validationResult = validateDocument(finalDocument, { securityLimits });
+  if (!validationResult.valid) {
+    const errorMsgs = validationResult.errors.map((e) => `${e.code}: ${e.message} (${e.path})`).join("; ");
+    throw new Error(`Document validation failed: ${errorMsgs}`);
+  }
+  return parsed.data;
+}
+function normalizeAndValidateSectionNode(rawJson, securityLimits) {
+  if (!rawJson || typeof rawJson !== "object") {
+    throw new Error("AI output must be a JSON object");
+  }
+  const usedIds = /* @__PURE__ */ new Set();
+  const normalized = normalizeNodeTree(rawJson, usedIds);
+  if (normalized.type !== "section") {
+    normalized.type = "section";
+  }
+  const parsed = NodeSchema.safeParse(normalized);
+  if (!parsed.success) {
+    const issues = parsed.error.issues.map((i) => `${i.path.join(".")}: ${i.message}`).join("; ");
+    throw new Error(`Section node schema validation failed: ${issues}`);
+  }
+  const secCheck = validateDocumentSecurity({ document: normalized }, securityLimits);
+  if (!secCheck.safe) {
+    const issues = secCheck.errors.map((e) => `${e.code}: ${e.message}`).join("; ");
+    throw new Error(`Section security check failed: ${issues}`);
+  }
+  return parsed.data;
+}
+function normalizeAndValidateRefactoredNode(rawJson, originalNode, securityLimits) {
+  if (!rawJson || typeof rawJson !== "object") {
+    throw new Error("AI output must be a JSON object");
+  }
+  const usedIds = /* @__PURE__ */ new Set();
+  const normalized = normalizeNodeTree(rawJson, usedIds);
+  normalized.id = originalNode.id;
+  normalized.type = originalNode.type;
+  const parsed = NodeSchema.safeParse(normalized);
+  if (!parsed.success) {
+    const issues = parsed.error.issues.map((i) => `${i.path.join(".")}: ${i.message}`).join("; ");
+    throw new Error(`Refactored node schema validation failed: ${issues}`);
+  }
+  const secCheck = validateDocumentSecurity({ document: normalized }, securityLimits);
+  if (!secCheck.safe) {
+    const issues = secCheck.errors.map((e) => `${e.code}: ${e.message}`).join("; ");
+    throw new Error(`Refactored node security check failed: ${issues}`);
+  }
+  return parsed.data;
+}
+
+export {
+  compileComponentCatalog,
+  buildSystemPrompt,
+  buildJsonSchemaForMode,
+  extractJsonFromResponse,
+  normalizeStyles,
+  normalizeNodeTree,
+  normalizeAndValidatePageDocument,
+  normalizeAndValidateSectionNode,
+  normalizeAndValidateRefactoredNode
+};
+//# sourceMappingURL=chunk-UH24LHRY.js.map

package/dist/chunk-UH24LHRY.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/core/prompt-compiler.ts","../src/core/normalizer.ts"],"sourcesContent":["import type {\n  AiCompiledComponentSpec,\n  AiCompiledComponentProp,\n  ComponentRegistryLike,\n  ComponentDefinitionLike,\n  AiGenerationMode,\n} from '../types';\n\nexport function compileComponentCatalog(\n  registry?: ComponentRegistryLike,\n): AiCompiledComponentSpec[] {\n  if (!registry) return [];\n\n  const definitions = registry.list();\n  return definitions.map((def: ComponentDefinitionLike) => {\n    const props: AiCompiledComponentProp[] = [];\n\n    if (def.propFields && Array.isArray(def.propFields)) {\n      for (const field of def.propFields) {\n        props.push({\n          name: field.name,\n          type: field.type || 'string',\n          defaultValue: field.defaultValue,\n          options: field.options?.map((o) => o.value),\n          description: field.description,\n        });\n      }\n    }\n\n    return {\n      type: def.type,\n      label: def.label || def.type,\n      category: def.category || 'custom',\n      description: def.description,\n      acceptsChildren: def.acceptsChildren ?? false,\n      allowedChildren: def.allowedChildren,\n      disallowedParents: def.disallowedParents,\n      defaultProps: def.defaultProps,\n      props: props.length > 0 ? props : undefined,\n    };\n  });\n}\n\nexport function buildSystemPrompt(options: {\n  catalog: AiCompiledComponentSpec[];\n  mode: AiGenerationMode;\n  prefix?: string;\n  stylePreference?: string;\n}): string {\n  const { catalog, mode, prefix, stylePreference } = options;\n\n  const catalogSummary = catalog\n    .map((c) => {\n      let desc = `- **${c.type}** (${c.category}): ${c.label}`;\n      if (c.description) desc += ` - ${c.description}`;\n      desc += ` | acceptsChildren: ${c.acceptsChildren}`;\n      if (c.allowedChildren && c.allowedChildren.length > 0) {\n        desc += ` | allowedChildren: [${c.allowedChildren.join(', ')}]`;\n      }\n      if (c.props && c.props.length > 0) {\n        const propList = c.props\n          .map((p) => {\n            let pStr = `${p.name} (${p.type})`;\n            if (p.options) pStr += `[${p.options.map(String).join('|')}]`;\n            return pStr;\n          })\n          .join(', ');\n        desc += ` | props: { ${propList} }`;\n      }\n      return desc;\n    })\n    .join('\\n');\n\n  let modeInstructions = '';\n  if (mode === 'full-page') {\n    modeInstructions = `\nYou must generate a complete, valid KUBUILD PageDocument JSON with this exact structure:\n{\n  \"schema\": \"stora.page\",\n  \"version\": \"1.0.0\",\n  \"metadata\": {\n    \"title\": \"<Page Title>\",\n    \"description\": \"<Page Description>\",\n    \"tags\": [\"<tag1>\", \"<tag2>\"],\n    \"category\": \"<category>\"\n  },\n  \"document\": {\n    \"id\": \"root-page\",\n    \"type\": \"page\",\n    \"styles\": {\n      \"base\": {\n        \"backgroundColor\": \"#ffffff\",\n        \"fontFamily\": \"Inter, system-ui, sans-serif\",\n        \"color\": \"#111827\",\n        \"minHeight\": \"100vh\"\n      }\n    },\n    \"children\": [\n      // Sections here...\n    ]\n  }\n}\n`;\n  } else if (mode === 'section') {\n    modeInstructions = `\nYou must generate a single valid KUBUILD section Node JSON with this structure:\n{\n  \"id\": \"section_<unique_suffix>\",\n  \"type\": \"section\",\n  \"styles\": {\n    \"base\": {\n      \"paddingTop\": \"64px\",\n      \"paddingBottom\": \"64px\",\n      \"paddingLeft\": \"24px\",\n      \"paddingRight\": \"24px\"\n    }\n  },\n  \"children\": [\n    // Container and child nodes...\n  ]\n}\n`;\n  } else if (mode === 'refactor') {\n    modeInstructions = `\nYou must refactor the provided KUBUILD Node based on the user's instructions.\nReturn ONLY the updated Node JSON while maintaining structural integrity.\n`;\n  }\n\n  const styleGuide = stylePreference\n    ? `\\nUser Style Preference: \"${stylePreference}\". Follow this visual aesthetic closely.`\n    : '\\nStyle with modern, clean, high-conversion design patterns (accessible contrast, harmonious spacing, beautiful typography, subtle shadows/borders).';\n\n  return `${prefix ? prefix + '\\n\\n' : ''}You are an expert web designer and KUBUILD architecture generator.\nKUBUILD is a component-driven visual website builder. Your job is to output pure, valid JSON (and NOTHING ELSE) matching the KUBUILD schema and the registered component definitions.\n\n### RULES & CONSTRAINTS:\n1. Output MUST be valid JSON only. No markdown fences, no explanatory text, no HTML tags outside html-embed.\n2. Only use component types listed in the Component Catalog below. Do NOT invent new component types.\n3. Obey \\`acceptsChildren\\` and \\`allowedChildren\\`:\n   - Layout hierarchy: \\`page\\` -> \\`section\\` -> \\`container\\` or \\`columns\\` -> content components (\\`heading\\`, \\`paragraph\\`, \\`button\\`, \\`image\\`, \\`form\\`, etc.).\n   - Leaf components (e.g. \\`button\\`, \\`input\\`, \\`heading\\`, \\`text\\`, \\`badge\\`) must have \\`acceptsChildren: false\\` and empty or omitted \\`children\\`.\n4. Every node MUST have a unique string \\`id\\` (e.g. \"hero-section\", \"hero-title\", \"cta-btn\").\n5. Responsive & Interactive State Styling:\n   - Use the \\`styles\\` object with breakpoint keys: \\`base\\` (all viewports/desktop), and optionally \\`tablet\\` or \\`mobile\\`.\n   - In \\`base\\`, \\`tablet\\`, and \\`mobile\\`, values MUST be CSS primitives (strings or numbers, e.g. \"backgroundColor\": \"#3b82f6\"). NEVER put nested objects or \":hover\" inside \"base\"!\n   - For hover, active, or interactive pseudo-classes, use \\`styles.states\\` with pseudo-class keys (e.g. \\`\":hover\"\\`), like:\n     \"styles\": {\n       \"base\": { \"backgroundColor\": \"#3b82f6\", \"color\": \"#ffffff\" },\n       \"states\": {\n         \":hover\": { \"backgroundColor\": \"#2563eb\" }\n       }\n     }\n6. NEVER include dangerous scripts, \\`javascript:\\` URIs, or malicious payload attributes.\n${styleGuide}\n\n### OUTPUT SPECIFICATION:\n${modeInstructions}\n\n### REGISTERED COMPONENT CATALOG:\n${catalogSummary || '(Standard core components: page, section, container, columns, heading, paragraph, text, button, input, textarea, select, image, video, icon, badge, code-block)'}\n`;\n}\n\nexport function buildJsonSchemaForMode(mode: AiGenerationMode): Record<string, unknown> {\n  const nodeSchema: Record<string, unknown> = {\n    type: 'object',\n    required: ['id', 'type'],\n    properties: {\n      id: { type: 'string' },\n      type: { type: 'string' },\n      props: { type: 'object' },\n      styles: {\n        type: 'object',\n        properties: {\n          base: { type: 'object' },\n          tablet: { type: 'object' },\n          mobile: { type: 'object' },\n          states: { type: 'object' },\n        },\n      },\n      children: {\n        type: 'array',\n        items: { $ref: '#' },\n      },\n    },\n  };\n\n  if (mode === 'full-page') {\n    return {\n      type: 'object',\n      required: ['schema', 'version', 'document'],\n      properties: {\n        schema: { type: 'string', const: 'stora.page' },\n        version: { type: 'string', const: '1.0.0' },\n        metadata: {\n          type: 'object',\n          properties: {\n            title: { type: 'string' },\n            description: { type: 'string' },\n            tags: { type: 'array', items: { type: 'string' } },\n            category: { type: 'string' },\n          },\n        },\n        document: nodeSchema,\n      },\n    };\n  }\n\n  return nodeSchema;\n}\n","import {\n  PageDocument,\n  PageDocumentSchema,\n  DocumentMetadata,\n  DocumentMetadataSchema,\n  Node,\n  NodeSchema,\n  CURRENT_SCHEMA_VERSION,\n  SCHEMA_NAME,\n} from '@kubuild/schema';\nimport {\n  validateDocument,\n  validateDocumentSecurity,\n  DocumentSecurityLimits,\n} from '@kubuild/core';\n\nexport function extractJsonFromResponse(raw: string): unknown {\n  let cleaned = raw.trim();\n  // Strip Markdown code blocks if present (```json ... ``` or ``` ...)\n  if (cleaned.startsWith('```')) {\n    const firstNewline = cleaned.indexOf('\\n');\n    if (firstNewline !== -1) {\n      cleaned = cleaned.slice(firstNewline + 1);\n    }\n    if (cleaned.endsWith('```')) {\n      cleaned = cleaned.slice(0, cleaned.length - 3).trim();\n    }\n  }\n\n  // Find the outermost JSON object bounds if there is extraneous text\n  const firstBrace = cleaned.indexOf('{');\n  const lastBrace = cleaned.lastIndexOf('}');\n  if (firstBrace !== -1 && lastBrace !== -1 && lastBrace > firstBrace) {\n    cleaned = cleaned.slice(firstBrace, lastBrace + 1);\n  }\n\n  try {\n    return JSON.parse(cleaned);\n  } catch (err: unknown) {\n    const msg = err instanceof Error ? err.message : String(err);\n    throw new Error(`Failed to parse AI response as JSON: ${msg}. Response was: ${raw.slice(0, 200)}...`);\n  }\n}\n\nexport function normalizeStyles(rawStyles: unknown): Node['styles'] {\n  if (!rawStyles || typeof rawStyles !== 'object' || Array.isArray(rawStyles)) {\n    return undefined;\n  }\n\n  const raw = rawStyles as Record<string, unknown>;\n  const result: Record<string, unknown> = {};\n  const states: Record<string, Record<string, unknown>> = {};\n\n  // Existing states\n  if (raw.states && typeof raw.states === 'object' && !Array.isArray(raw.states)) {\n    for (const [stateKey, stateVal] of Object.entries(raw.states as Record<string, unknown>)) {\n      if (stateVal && typeof stateVal === 'object' && !Array.isArray(stateVal)) {\n        states[stateKey] = { ...(stateVal as Record<string, unknown>) };\n      }\n    }\n  }\n\n  const knownBreakpoints = ['base', 'desktop', 'tablet', 'mobile'];\n  const hasBreakpoint = Object.keys(raw).some((k) => knownBreakpoints.includes(k));\n\n  if (!hasBreakpoint) {\n    // If AI output flat CSS properties without 'base' wrapper\n    const baseStyles: Record<string, unknown> = {};\n    for (const [key, value] of Object.entries(raw)) {\n      if (key === 'states') continue;\n      if (key.startsWith(':') || key === 'hover' || key === 'focus' || key === 'active') {\n        const stateKey = key.startsWith(':') ? key : `:${key}`;\n        if (value && typeof value === 'object' && !Array.isArray(value)) {\n          states[stateKey] = { ...(states[stateKey] || {}), ...(value as Record<string, unknown>) };\n        }\n      } else if (typeof value !== 'object' || value === null) {\n        baseStyles[key] = value;\n      }\n    }\n    if (Object.keys(baseStyles).length > 0) {\n      result.base = baseStyles;\n    }\n  } else {\n    for (const bp of knownBreakpoints) {\n      const bpVal = raw[bp];\n      if (bpVal && typeof bpVal === 'object' && !Array.isArray(bpVal)) {\n        const cleanBp: Record<string, unknown> = {};\n        for (const [key, value] of Object.entries(bpVal as Record<string, unknown>)) {\n          if (key.startsWith(':') || key === 'hover' || key === 'focus' || key === 'active') {\n            const stateKey = key.startsWith(':') ? key : `:${key}`;\n            if (value && typeof value === 'object' && !Array.isArray(value)) {\n              states[stateKey] = { ...(states[stateKey] || {}), ...(value as Record<string, unknown>) };\n            }\n          } else if (typeof value !== 'object' || value === null) {\n            cleanBp[key] = value;\n          }\n        }\n        if (Object.keys(cleanBp).length > 0) {\n          result[bp] = cleanBp;\n        }\n      }\n    }\n\n    // Also extract any pseudo-classes at top level\n    for (const [key, value] of Object.entries(raw)) {\n      if (knownBreakpoints.includes(key) || key === 'states') continue;\n      if (key.startsWith(':') || key === 'hover' || key === 'focus' || key === 'active') {\n        const stateKey = key.startsWith(':') ? key : `:${key}`;\n        if (value && typeof value === 'object' && !Array.isArray(value)) {\n          states[stateKey] = { ...(states[stateKey] || {}), ...(value as Record<string, unknown>) };\n        }\n      }\n    }\n  }\n\n  // Clean and sanitize states: each state can only have primitives\n  const cleanStates: Record<string, Record<string, unknown>> = {};\n  for (const [stateKey, stateDef] of Object.entries(states)) {\n    const cleanDef: Record<string, unknown> = {};\n    for (const [propKey, propVal] of Object.entries(stateDef)) {\n      if (typeof propVal !== 'object' || propVal === null) {\n        cleanDef[propKey] = propVal;\n      }\n    }\n    if (Object.keys(cleanDef).length > 0) {\n      cleanStates[stateKey] = cleanDef;\n    }\n  }\n\n  if (Object.keys(cleanStates).length > 0) {\n    result.states = cleanStates;\n  }\n\n  return Object.keys(result).length > 0 ? (result as Node['styles']) : undefined;\n}\n\nexport function normalizeNodeTree(\n  node: unknown,\n  usedIds: Set<string> = new Set(),\n  depth = 0,\n): Node {\n  if (!node || typeof node !== 'object') {\n    throw new Error(`Invalid node at depth ${depth}: must be an object`);\n  }\n\n  const raw = node as Record<string, unknown>;\n  const type = typeof raw.type === 'string' && raw.type.trim() ? raw.type.trim() : 'container';\n\n  // Ensure deterministic, unique ID\n  let id = typeof raw.id === 'string' && raw.id.trim() ? raw.id.trim() : '';\n  if (!id || usedIds.has(id)) {\n    let counter = 1;\n    let candidate = `${type}-${counter}`;\n    while (usedIds.has(candidate)) {\n      counter++;\n      candidate = `${type}-${counter}`;\n    }\n    id = candidate;\n  }\n  usedIds.add(id);\n\n  // Normalize props\n  const props =\n    raw.props && typeof raw.props === 'object' && !Array.isArray(raw.props)\n      ? { ...(raw.props as Record<string, unknown>) }\n      : {};\n\n  // Fix common LLM prop quirks:\n  // e.g. heading level passed as string \"1\" instead of number 1\n  if (type === 'heading' && typeof props.level === 'string') {\n    const parsedLevel = parseInt(props.level, 10);\n    if (!isNaN(parsedLevel) && parsedLevel >= 1 && parsedLevel <= 6) {\n      props.level = parsedLevel;\n    }\n  }\n\n  // Normalize styles\n  const styles = normalizeStyles(raw.styles);\n\n  // Normalize children recursively\n  let children: Node[] | undefined = undefined;\n  if (Array.isArray(raw.children)) {\n    children = raw.children\n      .filter((c) => c && typeof c === 'object')\n      .map((c) => normalizeNodeTree(c, usedIds, depth + 1));\n  }\n\n  return {\n    id,\n    type,\n    props: Object.keys(props).length > 0 ? props : undefined,\n    styles,\n    children,\n  };\n}\n\nexport function normalizeAndValidatePageDocument(\n  rawJson: unknown,\n  securityLimits?: DocumentSecurityLimits,\n): PageDocument {\n  if (!rawJson || typeof rawJson !== 'object') {\n    throw new Error('AI output must be a JSON object');\n  }\n\n  const obj = rawJson as Record<string, unknown>;\n\n  // Case 1: AI returned directly a page node instead of PageDocument\n  let docObject: Record<string, unknown>;\n  if (obj.type === 'page' && !obj.document) {\n    docObject = {\n      schema: SCHEMA_NAME,\n      version: CURRENT_SCHEMA_VERSION,\n      metadata: {\n        title: (typeof obj.props === 'object' && obj.props !== null && (obj.props as Record<string, unknown>).title) || 'AI Generated Page',\n        description: 'Generated by KUBUILD AI',\n      },\n      document: obj,\n    };\n  } else {\n    docObject = {\n      schema: SCHEMA_NAME,\n      version: typeof obj.version === 'string' ? obj.version : CURRENT_SCHEMA_VERSION,\n      metadata: typeof obj.metadata === 'object' && obj.metadata !== null\n        ? obj.metadata\n        : { title: 'AI Generated Page', description: 'Generated by KUBUILD AI' },\n      document: obj.document || obj,\n    };\n  }\n\n  // Normalize the node tree and ensure IDs\n  const usedIds = new Set<string>();\n  const normalizedRoot = normalizeNodeTree(docObject.document, usedIds);\n\n  if (normalizedRoot.type !== 'page') {\n    normalizedRoot.type = 'page';\n  }\n\n  const parsedMetadata = DocumentMetadataSchema.parse({\n    title: 'AI Generated Page',\n    description: 'Generated by KUBUILD AI',\n    author: 'KUBUILD AI',\n    tags: ['ai-generated'],\n    category: 'general',\n    version: '1.0.0',\n    ...(typeof docObject.metadata === 'object' && docObject.metadata !== null\n      ? (docObject.metadata as Record<string, unknown>)\n      : {}),\n  });\n\n  const finalDocument: PageDocument = {\n    schema: SCHEMA_NAME,\n    version: CURRENT_SCHEMA_VERSION,\n    metadata: parsedMetadata,\n    document: normalizedRoot as PageDocument['document'],\n  };\n\n  // 1. Zod Schema parse\n  const parsed = PageDocumentSchema.safeParse(finalDocument);\n  if (!parsed.success) {\n    const issues = parsed.error.issues.map((i) => `${i.path.join('.')}: ${i.message}`).join('; ');\n    throw new Error(`Document schema validation failed: ${issues}`);\n  }\n\n  // 2. Core document and security validation\n  const validationResult = validateDocument(finalDocument, { securityLimits });\n  if (!validationResult.valid) {\n    const errorMsgs = validationResult.errors.map((e) => `${e.code}: ${e.message} (${e.path})`).join('; ');\n    throw new Error(`Document validation failed: ${errorMsgs}`);\n  }\n\n  return parsed.data;\n}\n\nexport function normalizeAndValidateSectionNode(\n  rawJson: unknown,\n  securityLimits?: DocumentSecurityLimits,\n): Node {\n  if (!rawJson || typeof rawJson !== 'object') {\n    throw new Error('AI output must be a JSON object');\n  }\n\n  const usedIds = new Set<string>();\n  const normalized = normalizeNodeTree(rawJson, usedIds);\n\n  if (normalized.type !== 'section') {\n    // If AI wrapped it or returned another root, enforce section\n    normalized.type = 'section';\n  }\n\n  const parsed = NodeSchema.safeParse(normalized);\n  if (!parsed.success) {\n    const issues = parsed.error.issues.map((i) => `${i.path.join('.')}: ${i.message}`).join('; ');\n    throw new Error(`Section node schema validation failed: ${issues}`);\n  }\n\n  // Security check\n  const secCheck = validateDocumentSecurity({ document: normalized }, securityLimits);\n  if (!secCheck.safe) {\n    const issues = secCheck.errors.map((e) => `${e.code}: ${e.message}`).join('; ');\n    throw new Error(`Section security check failed: ${issues}`);\n  }\n\n  return parsed.data;\n}\n\nexport function normalizeAndValidateRefactoredNode(\n  rawJson: unknown,\n  originalNode: Node,\n  securityLimits?: DocumentSecurityLimits,\n): Node {\n  if (!rawJson || typeof rawJson !== 'object') {\n    throw new Error('AI output must be a JSON object');\n  }\n\n  const usedIds = new Set<string>();\n  const normalized = normalizeNodeTree(rawJson, usedIds);\n\n  // Preserve original ID and type if AI changed them accidentally\n  normalized.id = originalNode.id;\n  normalized.type = originalNode.type;\n\n  const parsed = NodeSchema.safeParse(normalized);\n  if (!parsed.success) {\n    const issues = parsed.error.issues.map((i) => `${i.path.join('.')}: ${i.message}`).join('; ');\n    throw new Error(`Refactored node schema validation failed: ${issues}`);\n  }\n\n  const secCheck = validateDocumentSecurity({ document: normalized }, securityLimits);\n  if (!secCheck.safe) {\n    const issues = secCheck.errors.map((e) => `${e.code}: ${e.message}`).join('; ');\n    throw new Error(`Refactored node security check failed: ${issues}`);\n  }\n\n  return parsed.data;\n}\n"],"mappings":";AAQO,SAAS,wBACd,UAC2B;AAC3B,MAAI,CAAC,SAAU,QAAO,CAAC;AAEvB,QAAM,cAAc,SAAS,KAAK;AAClC,SAAO,YAAY,IAAI,CAAC,QAAiC;AACvD,UAAM,QAAmC,CAAC;AAE1C,QAAI,IAAI,cAAc,MAAM,QAAQ,IAAI,UAAU,GAAG;AACnD,iBAAW,SAAS,IAAI,YAAY;AAClC,cAAM,KAAK;AAAA,UACT,MAAM,MAAM;AAAA,UACZ,MAAM,MAAM,QAAQ;AAAA,UACpB,cAAc,MAAM;AAAA,UACpB,SAAS,MAAM,SAAS,IAAI,CAAC,MAAM,EAAE,KAAK;AAAA,UAC1C,aAAa,MAAM;AAAA,QACrB,CAAC;AAAA,MACH;AAAA,IACF;AAEA,WAAO;AAAA,MACL,MAAM,IAAI;AAAA,MACV,OAAO,IAAI,SAAS,IAAI;AAAA,MACxB,UAAU,IAAI,YAAY;AAAA,MAC1B,aAAa,IAAI;AAAA,MACjB,iBAAiB,IAAI,mBAAmB;AAAA,MACxC,iBAAiB,IAAI;AAAA,MACrB,mBAAmB,IAAI;AAAA,MACvB,cAAc,IAAI;AAAA,MAClB,OAAO,MAAM,SAAS,IAAI,QAAQ;AAAA,IACpC;AAAA,EACF,CAAC;AACH;AAEO,SAAS,kBAAkB,SAKvB;AACT,QAAM,EAAE,SAAS,MAAM,QAAQ,gBAAgB,IAAI;AAEnD,QAAM,iBAAiB,QACpB,IAAI,CAAC,MAAM;AACV,QAAI,OAAO,OAAO,EAAE,IAAI,OAAO,EAAE,QAAQ,MAAM,EAAE,KAAK;AACtD,QAAI,EAAE,YAAa,SAAQ,MAAM,EAAE,WAAW;AAC9C,YAAQ,uBAAuB,EAAE,eAAe;AAChD,QAAI,EAAE,mBAAmB,EAAE,gBAAgB,SAAS,GAAG;AACrD,cAAQ,wBAAwB,EAAE,gBAAgB,KAAK,IAAI,CAAC;AAAA,IAC9D;AACA,QAAI,EAAE,SAAS,EAAE,MAAM,SAAS,GAAG;AACjC,YAAM,WAAW,EAAE,MAChB,IAAI,CAAC,MAAM;AACV,YAAI,OAAO,GAAG,EAAE,IAAI,KAAK,EAAE,IAAI;AAC/B,YAAI,EAAE,QAAS,SAAQ,IAAI,EAAE,QAAQ,IAAI,MAAM,EAAE,KAAK,GAAG,CAAC;AAC1D,eAAO;AAAA,MACT,CAAC,EACA,KAAK,IAAI;AACZ,cAAQ,eAAe,QAAQ;AAAA,IACjC;AACA,WAAO;AAAA,EACT,CAAC,EACA,KAAK,IAAI;AAEZ,MAAI,mBAAmB;AACvB,MAAI,SAAS,aAAa;AACxB,uBAAmB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EA4BrB,WAAW,SAAS,WAAW;AAC7B,uBAAmB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBrB,WAAW,SAAS,YAAY;AAC9B,uBAAmB;AAAA;AAAA;AAAA;AAAA,EAIrB;AAEA,QAAM,aAAa,kBACf;AAAA,0BAA6B,eAAe,6CAC5C;AAEJ,SAAO,GAAG,SAAS,SAAS,SAAS,EAAE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAqBvC,UAAU;AAAA;AAAA;AAAA,EAGV,gBAAgB;AAAA;AAAA;AAAA,EAGhB,kBAAkB,iKAAiK;AAAA;AAErL;AAEO,SAAS,uBAAuB,MAAiD;AACtF,QAAM,aAAsC;AAAA,IAC1C,MAAM;AAAA,IACN,UAAU,CAAC,MAAM,MAAM;AAAA,IACvB,YAAY;AAAA,MACV,IAAI,EAAE,MAAM,SAAS;AAAA,MACrB,MAAM,EAAE,MAAM,SAAS;AAAA,MACvB,OAAO,EAAE,MAAM,SAAS;AAAA,MACxB,QAAQ;AAAA,QACN,MAAM;AAAA,QACN,YAAY;AAAA,UACV,MAAM,EAAE,MAAM,SAAS;AAAA,UACvB,QAAQ,EAAE,MAAM,SAAS;AAAA,UACzB,QAAQ,EAAE,MAAM,SAAS;AAAA,UACzB,QAAQ,EAAE,MAAM,SAAS;AAAA,QAC3B;AAAA,MACF;AAAA,MACA,UAAU;AAAA,QACR,MAAM;AAAA,QACN,OAAO,EAAE,MAAM,IAAI;AAAA,MACrB;AAAA,IACF;AAAA,EACF;AAEA,MAAI,SAAS,aAAa;AACxB,WAAO;AAAA,MACL,MAAM;AAAA,MACN,UAAU,CAAC,UAAU,WAAW,UAAU;AAAA,MAC1C,YAAY;AAAA,QACV,QAAQ,EAAE,MAAM,UAAU,OAAO,aAAa;AAAA,QAC9C,SAAS,EAAE,MAAM,UAAU,OAAO,QAAQ;AAAA,QAC1C,UAAU;AAAA,UACR,MAAM;AAAA,UACN,YAAY;AAAA,YACV,OAAO,EAAE,MAAM,SAAS;AAAA,YACxB,aAAa,EAAE,MAAM,SAAS;AAAA,YAC9B,MAAM,EAAE,MAAM,SAAS,OAAO,EAAE,MAAM,SAAS,EAAE;AAAA,YACjD,UAAU,EAAE,MAAM,SAAS;AAAA,UAC7B;AAAA,QACF;AAAA,QACA,UAAU;AAAA,MACZ;AAAA,IACF;AAAA,EACF;AAEA,SAAO;AACT;;;AClNA;AAAA,EAEE;AAAA,EAEA;AAAA,EAEA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP;AAAA,EACE;AAAA,EACA;AAAA,OAEK;AAEA,SAAS,wBAAwB,KAAsB;AAC5D,MAAI,UAAU,IAAI,KAAK;AAEvB,MAAI,QAAQ,WAAW,KAAK,GAAG;AAC7B,UAAM,eAAe,QAAQ,QAAQ,IAAI;AACzC,QAAI,iBAAiB,IAAI;AACvB,gBAAU,QAAQ,MAAM,eAAe,CAAC;AAAA,IAC1C;AACA,QAAI,QAAQ,SAAS,KAAK,GAAG;AAC3B,gBAAU,QAAQ,MAAM,GAAG,QAAQ,SAAS,CAAC,EAAE,KAAK;AAAA,IACtD;AAAA,EACF;AAGA,QAAM,aAAa,QAAQ,QAAQ,GAAG;AACtC,QAAM,YAAY,QAAQ,YAAY,GAAG;AACzC,MAAI,eAAe,MAAM,cAAc,MAAM,YAAY,YAAY;AACnE,cAAU,QAAQ,MAAM,YAAY,YAAY,CAAC;AAAA,EACnD;AAEA,MAAI;AACF,WAAO,KAAK,MAAM,OAAO;AAAA,EAC3B,SAAS,KAAc;AACrB,UAAM,MAAM,eAAe,QAAQ,IAAI,UAAU,OAAO,GAAG;AAC3D,UAAM,IAAI,MAAM,wCAAwC,GAAG,mBAAmB,IAAI,MAAM,GAAG,GAAG,CAAC,KAAK;AAAA,EACtG;AACF;AAEO,SAAS,gBAAgB,WAAoC;AAClE,MAAI,CAAC,aAAa,OAAO,cAAc,YAAY,MAAM,QAAQ,SAAS,GAAG;AAC3E,WAAO;AAAA,EACT;AAEA,QAAM,MAAM;AACZ,QAAM,SAAkC,CAAC;AACzC,QAAM,SAAkD,CAAC;AAGzD,MAAI,IAAI,UAAU,OAAO,IAAI,WAAW,YAAY,CAAC,MAAM,QAAQ,IAAI,MAAM,GAAG;AAC9E,eAAW,CAAC,UAAU,QAAQ,KAAK,OAAO,QAAQ,IAAI,MAAiC,GAAG;AACxF,UAAI,YAAY,OAAO,aAAa,YAAY,CAAC,MAAM,QAAQ,QAAQ,GAAG;AACxE,eAAO,QAAQ,IAAI,EAAE,GAAI,SAAqC;AAAA,MAChE;AAAA,IACF;AAAA,EACF;AAEA,QAAM,mBAAmB,CAAC,QAAQ,WAAW,UAAU,QAAQ;AAC/D,QAAM,gBAAgB,OAAO,KAAK,GAAG,EAAE,KAAK,CAAC,MAAM,iBAAiB,SAAS,CAAC,CAAC;AAE/E,MAAI,CAAC,eAAe;AAElB,UAAM,aAAsC,CAAC;AAC7C,eAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,GAAG,GAAG;AAC9C,UAAI,QAAQ,SAAU;AACtB,UAAI,IAAI,WAAW,GAAG,KAAK,QAAQ,WAAW,QAAQ,WAAW,QAAQ,UAAU;AACjF,cAAM,WAAW,IAAI,WAAW,GAAG,IAAI,MAAM,IAAI,GAAG;AACpD,YAAI,SAAS,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,KAAK,GAAG;AAC/D,iBAAO,QAAQ,IAAI,EAAE,GAAI,OAAO,QAAQ,KAAK,CAAC,GAAI,GAAI,MAAkC;AAAA,QAC1F;AAAA,MACF,WAAW,OAAO,UAAU,YAAY,UAAU,MAAM;AACtD,mBAAW,GAAG,IAAI;AAAA,MACpB;AAAA,IACF;AACA,QAAI,OAAO,KAAK,UAAU,EAAE,SAAS,GAAG;AACtC,aAAO,OAAO;AAAA,IAChB;AAAA,EACF,OAAO;AACL,eAAW,MAAM,kBAAkB;AACjC,YAAM,QAAQ,IAAI,EAAE;AACpB,UAAI,SAAS,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,KAAK,GAAG;AAC/D,cAAM,UAAmC,CAAC;AAC1C,mBAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,KAAgC,GAAG;AAC3E,cAAI,IAAI,WAAW,GAAG,KAAK,QAAQ,WAAW,QAAQ,WAAW,QAAQ,UAAU;AACjF,kBAAM,WAAW,IAAI,WAAW,GAAG,IAAI,MAAM,IAAI,GAAG;AACpD,gBAAI,SAAS,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,KAAK,GAAG;AAC/D,qBAAO,QAAQ,IAAI,EAAE,GAAI,OAAO,QAAQ,KAAK,CAAC,GAAI,GAAI,MAAkC;AAAA,YAC1F;AAAA,UACF,WAAW,OAAO,UAAU,YAAY,UAAU,MAAM;AACtD,oBAAQ,GAAG,IAAI;AAAA,UACjB;AAAA,QACF;AACA,YAAI,OAAO,KAAK,OAAO,EAAE,SAAS,GAAG;AACnC,iBAAO,EAAE,IAAI;AAAA,QACf;AAAA,MACF;AAAA,IACF;AAGA,eAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,GAAG,GAAG;AAC9C,UAAI,iBAAiB,SAAS,GAAG,KAAK,QAAQ,SAAU;AACxD,UAAI,IAAI,WAAW,GAAG,KAAK,QAAQ,WAAW,QAAQ,WAAW,QAAQ,UAAU;AACjF,cAAM,WAAW,IAAI,WAAW,GAAG,IAAI,MAAM,IAAI,GAAG;AACpD,YAAI,SAAS,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,KAAK,GAAG;AAC/D,iBAAO,QAAQ,IAAI,EAAE,GAAI,OAAO,QAAQ,KAAK,CAAC,GAAI,GAAI,MAAkC;AAAA,QAC1F;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAGA,QAAM,cAAuD,CAAC;AAC9D,aAAW,CAAC,UAAU,QAAQ,KAAK,OAAO,QAAQ,MAAM,GAAG;AACzD,UAAM,WAAoC,CAAC;AAC3C,eAAW,CAAC,SAAS,OAAO,KAAK,OAAO,QAAQ,QAAQ,GAAG;AACzD,UAAI,OAAO,YAAY,YAAY,YAAY,MAAM;AACnD,iBAAS,OAAO,IAAI;AAAA,MACtB;AAAA,IACF;AACA,QAAI,OAAO,KAAK,QAAQ,EAAE,SAAS,GAAG;AACpC,kBAAY,QAAQ,IAAI;AAAA,IAC1B;AAAA,EACF;AAEA,MAAI,OAAO,KAAK,WAAW,EAAE,SAAS,GAAG;AACvC,WAAO,SAAS;AAAA,EAClB;AAEA,SAAO,OAAO,KAAK,MAAM,EAAE,SAAS,IAAK,SAA4B;AACvE;AAEO,SAAS,kBACd,MACA,UAAuB,oBAAI,IAAI,GAC/B,QAAQ,GACF;AACN,MAAI,CAAC,QAAQ,OAAO,SAAS,UAAU;AACrC,UAAM,IAAI,MAAM,yBAAyB,KAAK,qBAAqB;AAAA,EACrE;AAEA,QAAM,MAAM;AACZ,QAAM,OAAO,OAAO,IAAI,SAAS,YAAY,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,IAAI;AAGjF,MAAI,KAAK,OAAO,IAAI,OAAO,YAAY,IAAI,GAAG,KAAK,IAAI,IAAI,GAAG,KAAK,IAAI;AACvE,MAAI,CAAC,MAAM,QAAQ,IAAI,EAAE,GAAG;AAC1B,QAAI,UAAU;AACd,QAAI,YAAY,GAAG,IAAI,IAAI,OAAO;AAClC,WAAO,QAAQ,IAAI,SAAS,GAAG;AAC7B;AACA,kBAAY,GAAG,IAAI,IAAI,OAAO;AAAA,IAChC;AACA,SAAK;AAAA,EACP;AACA,UAAQ,IAAI,EAAE;AAGd,QAAM,QACJ,IAAI,SAAS,OAAO,IAAI,UAAU,YAAY,CAAC,MAAM,QAAQ,IAAI,KAAK,IAClE,EAAE,GAAI,IAAI,MAAkC,IAC5C,CAAC;AAIP,MAAI,SAAS,aAAa,OAAO,MAAM,UAAU,UAAU;AACzD,UAAM,cAAc,SAAS,MAAM,OAAO,EAAE;AAC5C,QAAI,CAAC,MAAM,WAAW,KAAK,eAAe,KAAK,eAAe,GAAG;AAC/D,YAAM,QAAQ;AAAA,IAChB;AAAA,EACF;AAGA,QAAM,SAAS,gBAAgB,IAAI,MAAM;AAGzC,MAAI,WAA+B;AACnC,MAAI,MAAM,QAAQ,IAAI,QAAQ,GAAG;AAC/B,eAAW,IAAI,SACZ,OAAO,CAAC,MAAM,KAAK,OAAO,MAAM,QAAQ,EACxC,IAAI,CAAC,MAAM,kBAAkB,GAAG,SAAS,QAAQ,CAAC,CAAC;AAAA,EACxD;AAEA,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA,OAAO,OAAO,KAAK,KAAK,EAAE,SAAS,IAAI,QAAQ;AAAA,IAC/C;AAAA,IACA;AAAA,EACF;AACF;AAEO,SAAS,iCACd,SACA,gBACc;AACd,MAAI,CAAC,WAAW,OAAO,YAAY,UAAU;AAC3C,UAAM,IAAI,MAAM,iCAAiC;AAAA,EACnD;AAEA,QAAM,MAAM;AAGZ,MAAI;AACJ,MAAI,IAAI,SAAS,UAAU,CAAC,IAAI,UAAU;AACxC,gBAAY;AAAA,MACV,QAAQ;AAAA,MACR,SAAS;AAAA,MACT,UAAU;AAAA,QACR,OAAQ,OAAO,IAAI,UAAU,YAAY,IAAI,UAAU,QAAS,IAAI,MAAkC,SAAU;AAAA,QAChH,aAAa;AAAA,MACf;AAAA,MACA,UAAU;AAAA,IACZ;AAAA,EACF,OAAO;AACL,gBAAY;AAAA,MACV,QAAQ;AAAA,MACR,SAAS,OAAO,IAAI,YAAY,WAAW,IAAI,UAAU;AAAA,MACzD,UAAU,OAAO,IAAI,aAAa,YAAY,IAAI,aAAa,OAC3D,IAAI,WACJ,EAAE,OAAO,qBAAqB,aAAa,0BAA0B;AAAA,MACzE,UAAU,IAAI,YAAY;AAAA,IAC5B;AAAA,EACF;AAGA,QAAM,UAAU,oBAAI,IAAY;AAChC,QAAM,iBAAiB,kBAAkB,UAAU,UAAU,OAAO;AAEpE,MAAI,eAAe,SAAS,QAAQ;AAClC,mBAAe,OAAO;AAAA,EACxB;AAEA,QAAM,iBAAiB,uBAAuB,MAAM;AAAA,IAClD,OAAO;AAAA,IACP,aAAa;AAAA,IACb,QAAQ;AAAA,IACR,MAAM,CAAC,cAAc;AAAA,IACrB,UAAU;AAAA,IACV,SAAS;AAAA,IACT,GAAI,OAAO,UAAU,aAAa,YAAY,UAAU,aAAa,OAChE,UAAU,WACX,CAAC;AAAA,EACP,CAAC;AAED,QAAM,gBAA8B;AAAA,IAClC,QAAQ;AAAA,IACR,SAAS;AAAA,IACT,UAAU;AAAA,IACV,UAAU;AAAA,EACZ;AAGA,QAAM,SAAS,mBAAmB,UAAU,aAAa;AACzD,MAAI,CAAC,OAAO,SAAS;AACnB,UAAM,SAAS,OAAO,MAAM,OAAO,IAAI,CAAC,MAAM,GAAG,EAAE,KAAK,KAAK,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,KAAK,IAAI;AAC5F,UAAM,IAAI,MAAM,sCAAsC,MAAM,EAAE;AAAA,EAChE;AAGA,QAAM,mBAAmB,iBAAiB,eAAe,EAAE,eAAe,CAAC;AAC3E,MAAI,CAAC,iBAAiB,OAAO;AAC3B,UAAM,YAAY,iBAAiB,OAAO,IAAI,CAAC,MAAM,GAAG,EAAE,IAAI,KAAK,EAAE,OAAO,KAAK,EAAE,IAAI,GAAG,EAAE,KAAK,IAAI;AACrG,UAAM,IAAI,MAAM,+BAA+B,SAAS,EAAE;AAAA,EAC5D;AAEA,SAAO,OAAO;AAChB;AAEO,SAAS,gCACd,SACA,gBACM;AACN,MAAI,CAAC,WAAW,OAAO,YAAY,UAAU;AAC3C,UAAM,IAAI,MAAM,iCAAiC;AAAA,EACnD;AAEA,QAAM,UAAU,oBAAI,IAAY;AAChC,QAAM,aAAa,kBAAkB,SAAS,OAAO;AAErD,MAAI,WAAW,SAAS,WAAW;AAEjC,eAAW,OAAO;AAAA,EACpB;AAEA,QAAM,SAAS,WAAW,UAAU,UAAU;AAC9C,MAAI,CAAC,OAAO,SAAS;AACnB,UAAM,SAAS,OAAO,MAAM,OAAO,IAAI,CAAC,MAAM,GAAG,EAAE,KAAK,KAAK,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,KAAK,IAAI;AAC5F,UAAM,IAAI,MAAM,0CAA0C,MAAM,EAAE;AAAA,EACpE;AAGA,QAAM,WAAW,yBAAyB,EAAE,UAAU,WAAW,GAAG,cAAc;AAClF,MAAI,CAAC,SAAS,MAAM;AAClB,UAAM,SAAS,SAAS,OAAO,IAAI,CAAC,MAAM,GAAG,EAAE,IAAI,KAAK,EAAE,OAAO,EAAE,EAAE,KAAK,IAAI;AAC9E,UAAM,IAAI,MAAM,kCAAkC,MAAM,EAAE;AAAA,EAC5D;AAEA,SAAO,OAAO;AAChB;AAEO,SAAS,mCACd,SACA,cACA,gBACM;AACN,MAAI,CAAC,WAAW,OAAO,YAAY,UAAU;AAC3C,UAAM,IAAI,MAAM,iCAAiC;AAAA,EACnD;AAEA,QAAM,UAAU,oBAAI,IAAY;AAChC,QAAM,aAAa,kBAAkB,SAAS,OAAO;AAGrD,aAAW,KAAK,aAAa;AAC7B,aAAW,OAAO,aAAa;AAE/B,QAAM,SAAS,WAAW,UAAU,UAAU;AAC9C,MAAI,CAAC,OAAO,SAAS;AACnB,UAAM,SAAS,OAAO,MAAM,OAAO,IAAI,CAAC,MAAM,GAAG,EAAE,KAAK,KAAK,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,KAAK,IAAI;AAC5F,UAAM,IAAI,MAAM,6CAA6C,MAAM,EAAE;AAAA,EACvE;AAEA,QAAM,WAAW,yBAAyB,EAAE,UAAU,WAAW,GAAG,cAAc;AAClF,MAAI,CAAC,SAAS,MAAM;AAClB,UAAM,SAAS,SAAS,OAAO,IAAI,CAAC,MAAM,GAAG,EAAE,IAAI,KAAK,EAAE,OAAO,EAAE,EAAE,KAAK,IAAI;AAC9E,UAAM,IAAI,MAAM,0CAA0C,MAAM,EAAE;AAAA,EACpE;AAEA,SAAO,OAAO;AAChB;","names":[]}

package/dist/client/ai-client.d.ts

@@ -0,0 +1,59 @@
+import type { PageDocument, Node } from '@kubuild/schema';
+import type { AiGeneratePageRequest, AiGenerateSectionRequest, AiRefactorNodeRequest, AiChatRequest, AiChatResponse, AiChatMessage, AiStreamCallbacks } from '../types';
+export interface AiClientOptions {
+    endpoint: string;
+    headers?: Record<string, string> | (() => Promise<Record<string, string>> | Record<string, string>);
+    fetch?: typeof fetch;
+}
+export declare class KubuildAiClient {
+    private endpoint;
+    private headers?;
+    private fetchFn;
+    constructor(options: AiClientOptions);
+    private resolveHeaders;
+    private sendRequest;
+    /**
+     * Standard single-pass full page generator (non-streaming).
+     */
+    generatePage(params: AiGeneratePageRequest, options?: {
+        signal?: AbortSignal;
+    }): Promise<PageDocument>;
+    /**
+     * Low-level SSE frame consumer shared by every streaming method on this client
+     * (`streamPage`, `chatStream`) — one transport implementation, reused rather than
+     * duplicated per stream "flavor" (STORA-515).
+     *
+     * Reads `res.body`, decodes `data: <json>\n\n` frames into `AiStreamEvent`s, and
+     * invokes `onEvent` for each one in arrival order. If `onEvent` throws (the
+     * conventional way callers signal a fatal `error` event), the throw propagates out of
+     * this method after the reader lock is released.
+     */
+    private consumeSse;
+    /**
+     * Progressive section streamer (SSE).
+     * Calls callbacks on each event (status, metadata, section, complete).
+     */
+    streamPage(params: AiGeneratePageRequest, callbacks?: AiStreamCallbacks, options?: {
+        signal?: AbortSignal;
+    }): Promise<PageDocument>;
+    /**
+     * Token-level streaming chat (STORA-515). Calls `onChatChunk` for every partial chunk
+     * as it arrives and resolves with the fully assembled assistant `AiChatMessage` once
+     * the stream's terminal `chat-complete` event is received. Reuses the exact same SSE
+     * transport as `streamPage` via `consumeSse` — no separate parsing path.
+     */
+    chatStream(params: AiChatRequest, callbacks?: Pick<AiStreamCallbacks, 'onChatChunk' | 'onChatComplete' | 'onError'>, options?: {
+        signal?: AbortSignal;
+    }): Promise<AiChatMessage>;
+    generateSection(params: AiGenerateSectionRequest, options?: {
+        signal?: AbortSignal;
+    }): Promise<Node>;
+    refactorNode(params: AiRefactorNodeRequest, options?: {
+        signal?: AbortSignal;
+    }): Promise<Node>;
+    chat(params: AiChatRequest, options?: {
+        signal?: AbortSignal;
+    }): Promise<AiChatResponse>;
+}
+export declare function createAiClient(options: AiClientOptions): KubuildAiClient;
+//# sourceMappingURL=ai-client.d.ts.map

package/dist/client/ai-client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-client.d.ts","sourceRoot":"","sources":["../../src/client/ai-client.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,iBAAiB,CAAC;AAC1D,OAAO,KAAK,EACV,qBAAqB,EACrB,wBAAwB,EACxB,qBAAqB,EACrB,aAAa,EACb,cAAc,EACd,aAAa,EAEb,iBAAiB,EAElB,MAAM,UAAU,CAAC;AAElB,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EACJ,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GACtB,CAAC,MAAM,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;IACrE,KAAK,CAAC,EAAE,OAAO,KAAK,CAAC;CACtB;AAED,qBAAa,eAAe;IAC1B,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,OAAO,CAAC,CAA6B;IAC7C,OAAO,CAAC,OAAO,CAAe;gBAElB,OAAO,EAAE,eAAe;YAStB,cAAc;YAQd,WAAW;IA6BzB;;OAEG;IACG,YAAY,CAChB,MAAM,EAAE,qBAAqB,EAC7B,OAAO,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,WAAW,CAAA;KAAE,GACjC,OAAO,CAAC,YAAY,CAAC;IAUxB;;;;;;;;;OASG;YACW,UAAU;IA0CxB;;;OAGG;IACG,UAAU,CACd,MAAM,EAAE,qBAAqB,EAC7B,SAAS,CAAC,EAAE,iBAAiB,EAC7B,OAAO,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,WAAW,CAAA;KAAE,GACjC,OAAO,CAAC,YAAY,CAAC;IAmDxB;;;;;OAKG;IACG,UAAU,CACd,MAAM,EAAE,aAAa,EACrB,SAAS,CAAC,EAAE,IAAI,CAAC,iBAAiB,EAAE,aAAa,GAAG,gBAAgB,GAAG,SAAS,CAAC,EACjF,OAAO,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,WAAW,CAAA;KAAE,GACjC,OAAO,CAAC,aAAa,CAAC;IA+CnB,eAAe,CACnB,MAAM,EAAE,wBAAwB,EAChC,OAAO,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,WAAW,CAAA;KAAE,GACjC,OAAO,CAAC,IAAI,CAAC;IAUV,YAAY,CAChB,MAAM,EAAE,qBAAqB,EAC7B,OAAO,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,WAAW,CAAA;KAAE,GACjC,OAAO,CAAC,IAAI,CAAC;IAUV,IAAI,CACR,MAAM,EAAE,aAAa,EACrB,OAAO,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,WAAW,CAAA;KAAE,GACjC,OAAO,CAAC,cAAc,CAAC;CAS3B;AAED,wBAAgB,cAAc,CAAC,OAAO,EAAE,eAAe,GAAG,eAAe,CAExE"}