import { FolkShape } from "./folk-shape"; import { css, html } from "./tags"; const styles = css` :host { background: white; border-radius: 8px; box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1); min-width: 380px; min-height: 460px; } .header { display: flex; align-items: center; justify-content: space-between; padding: 8px 12px; background: linear-gradient(135deg, #059669, #34d399); color: white; border-radius: 8px 8px 0 0; font-size: 12px; font-weight: 600; cursor: move; } .header-title { display: flex; align-items: center; gap: 6px; } .header-actions { display: flex; gap: 4px; } .header-actions button { background: transparent; border: none; color: white; cursor: pointer; padding: 2px 6px; border-radius: 4px; font-size: 14px; } .header-actions button:hover { background: rgba(255, 255, 255, 0.2); } .content { display: flex; flex-direction: column; height: calc(100% - 36px); overflow: hidden; } .prompt-area { padding: 12px; border-bottom: 1px solid #e2e8f0; } .prompt-input { width: 100%; padding: 10px 12px; border: 2px solid #e2e8f0; border-radius: 8px; font-size: 13px; resize: none; outline: none; font-family: inherit; } .prompt-input:focus { border-color: #059669; } .component-input { width: 100%; padding: 8px 10px; border: 2px solid #e2e8f0; border-radius: 6px; font-size: 12px; outline: none; margin-top: 8px; font-family: inherit; } .component-input:focus { border-color: #059669; } .controls { display: flex; gap: 8px; margin-top: 8px; } .generate-btn { flex: 1; padding: 8px 16px; background: linear-gradient(135deg, #059669, #34d399); color: white; border: none; border-radius: 6px; font-size: 13px; font-weight: 600; cursor: pointer; transition: opacity 0.2s; } .generate-btn:hover { opacity: 0.9; } .generate-btn:disabled { opacity: 0.5; cursor: not-allowed; } .tabs { display: flex; border-bottom: 1px solid #e2e8f0; } .tab { flex: 1; padding: 8px; text-align: center; font-size: 12px; font-weight: 600; color: #64748b; border: none; background: none; cursor: pointer; border-bottom: 2px solid transparent; transition: all 0.2s; } .tab:hover { color: #059669; } .tab.active { color: #059669; border-bottom-color: #059669; } .preview-area { flex: 1; overflow: auto; padding: 12px; } .preview-area img { max-width: 100%; border-radius: 8px; box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1); } .drc-results { font-size: 12px; color: #334155; } .drc-results .pass { color: #059669; font-weight: 600; } .drc-results .fail { color: #ef4444; font-weight: 600; } .export-row { display: flex; gap: 8px; padding: 8px 12px; border-top: 1px solid #e2e8f0; flex-wrap: wrap; } .export-btn { padding: 6px 12px; background: #334155; color: white; border: none; border-radius: 6px; font-size: 12px; cursor: pointer; text-decoration: none; } .export-btn:hover { background: #475569; } .export-btn.primary { background: #059669; } .export-btn.primary:hover { background: #047857; } .placeholder { flex: 1; display: flex; flex-direction: column; align-items: center; justify-content: center; color: #94a3b8; text-align: center; gap: 8px; } .placeholder-icon { font-size: 48px; opacity: 0.5; } .loading { display: flex; flex-direction: column; align-items: center; justify-content: center; padding: 24px; gap: 12px; } .spinner { width: 32px; height: 32px; border: 3px solid #e2e8f0; border-top-color: #059669; border-radius: 50%; animation: spin 1s linear infinite; } @keyframes spin { to { transform: rotate(360deg); } } .error { color: #ef4444; padding: 12px; background: #fef2f2; border-radius: 6px; font-size: 13px; } `; declare global { interface HTMLElementTagNameMap { "folk-kicad": FolkKiCAD; } } export class FolkKiCAD extends FolkShape { static override tagName = "folk-kicad"; static { const sheet = new CSSStyleSheet(); const parentRules = Array.from(FolkShape.styles.cssRules) .map((r) => r.cssText) .join("\n"); const childRules = Array.from(styles.cssRules) .map((r) => r.cssText) .join("\n"); sheet.replaceSync(`${parentRules}\n${childRules}`); this.styles = sheet; } #isLoading = false; #error: string | null = null; #schematicSvg: string | null = null; #boardSvg: string | null = null; #gerberUrl: string | null = null; #bomUrl: string | null = null; #pdfUrl: string | null = null; #drcResults: any = null; #activeTab: "schematic" | "board" | "drc" = "schematic"; #promptInput: HTMLTextAreaElement | null = null; #componentInput: HTMLInputElement | null = null; #generateBtn: HTMLButtonElement | null = null; #previewArea: HTMLElement | null = null; #exportRow: HTMLElement | null = null; override createRenderRoot() { const root = super.createRenderRoot(); const wrapper = document.createElement("div"); wrapper.innerHTML = html`
🔌 KiCAD PCB
🔌 Describe a PCB design and click Generate
`; const slot = root.querySelector("slot"); const containerDiv = slot?.parentElement as HTMLElement; if (containerDiv) { containerDiv.replaceWith(wrapper); } this.#promptInput = wrapper.querySelector(".prompt-input"); this.#componentInput = wrapper.querySelector(".component-input"); this.#generateBtn = wrapper.querySelector(".generate-btn"); this.#previewArea = wrapper.querySelector(".preview-area"); this.#exportRow = wrapper.querySelector(".export-row"); const closeBtn = wrapper.querySelector(".close-btn") as HTMLButtonElement; // Tab switching wrapper.querySelectorAll(".tab").forEach((tab) => { tab.addEventListener("click", (e) => { e.stopPropagation(); const tabName = (tab as HTMLElement).dataset.tab as "schematic" | "board" | "drc"; this.#activeTab = tabName; wrapper.querySelectorAll(".tab").forEach((t) => t.classList.remove("active")); tab.classList.add("active"); this.#renderPreview(); }); }); this.#generateBtn?.addEventListener("click", (e) => { e.stopPropagation(); this.#generate(); }); this.#promptInput?.addEventListener("keydown", (e) => { if (e.key === "Enter" && !e.shiftKey) { e.preventDefault(); this.#generate(); } }); this.#promptInput?.addEventListener("pointerdown", (e) => e.stopPropagation()); this.#componentInput?.addEventListener("pointerdown", (e) => e.stopPropagation()); closeBtn.addEventListener("click", (e) => { e.stopPropagation(); this.dispatchEvent(new CustomEvent("close")); }); return root; } async #generate() { const prompt = this.#promptInput?.value.trim(); if (!prompt || this.#isLoading) return; const components = this.#componentInput?.value .split(",") .map((c) => c.trim()) .filter(Boolean) || []; this.#isLoading = true; this.#error = null; if (this.#generateBtn) this.#generateBtn.disabled = true; if (this.#previewArea) { this.#previewArea.innerHTML = '
Generating PCB design...
'; } try { // Step 1: Create project const createRes = await fetch("/api/kicad/create_project", { method: "POST", headers: { "Content-Type": "application/json" }, body: JSON.stringify({ prompt, components }), }); if (!createRes.ok) { const err = await createRes.json().catch(() => ({ error: createRes.statusText })); throw new Error(err.error || "PCB generation failed"); } const data = await createRes.json(); this.#schematicSvg = data.schematic_svg || null; this.#boardSvg = data.board_svg || null; this.#gerberUrl = data.gerber_url || null; this.#bomUrl = data.bom_url || null; this.#pdfUrl = data.pdf_url || null; this.#drcResults = data.drc || null; this.#renderPreview(); this.#showExports(); } catch (err) { this.#error = err instanceof Error ? err.message : "Generation failed"; if (this.#previewArea) { this.#previewArea.innerHTML = `
${this.#escapeHtml(this.#error)}
`; } } finally { this.#isLoading = false; if (this.#generateBtn) this.#generateBtn.disabled = false; } } #renderPreview() { if (!this.#previewArea) return; switch (this.#activeTab) { case "schematic": if (this.#schematicSvg) { this.#previewArea.innerHTML = `Schematic`; } else { this.#previewArea.innerHTML = '
\u{1F4CB}Schematic will appear here
'; } break; case "board": if (this.#boardSvg) { this.#previewArea.innerHTML = `Board Layout`; } else { this.#previewArea.innerHTML = '
\u{1F4DF}Board layout will appear here
'; } break; case "drc": if (this.#drcResults) { const violations = this.#drcResults.violations || []; const passed = violations.length === 0; this.#previewArea.innerHTML = `

${passed ? '\u2705 DRC Passed' : `\u274C ${violations.length} Violation(s)` }

${violations.map((v: any) => `

\u2022 ${this.#escapeHtml(v.message || v)}

`).join("")}
`; } else { this.#previewArea.innerHTML = '
\u2705DRC results will appear here
'; } break; } } #showExports() { if (this.#exportRow && (this.#gerberUrl || this.#bomUrl || this.#pdfUrl)) { this.#exportRow.style.display = "flex"; const gerberLink = this.#exportRow.querySelector(".gerber-dl") as HTMLAnchorElement; const bomLink = this.#exportRow.querySelector(".bom-dl") as HTMLAnchorElement; const pdfLink = this.#exportRow.querySelector(".pdf-dl") as HTMLAnchorElement; if (gerberLink) gerberLink.href = this.#gerberUrl || "#"; if (bomLink) bomLink.href = this.#bomUrl || "#"; if (pdfLink) pdfLink.href = this.#pdfUrl || "#"; } } #escapeHtml(text: string): string { const div = document.createElement("div"); div.textContent = text; return div.innerHTML; } override toJSON() { return { ...super.toJSON(), type: "folk-kicad", schematicSvg: this.#schematicSvg, boardSvg: this.#boardSvg, gerberUrl: this.#gerberUrl, bomUrl: this.#bomUrl, pdfUrl: this.#pdfUrl, }; } }