Fix Figma runner endpoint persistence (#93)

Author: chubes4

plugins/figma-to-wordpress/src/code.ts

@@ -3,6 +3,8 @@ import type { NormalizedAsset, NormalizedDocument, NormalizedSceneNode, PluginTo
 
 figma.showUI(__html__, { width: 420, height: 420, themeColors: true });
 
+const runnerEndpointStorageKey = "figma-to-wordpress-runner-endpoint";
+
 function postToUi(message: PluginToUiMessage) {
   figma.ui.postMessage(message);
 }
@@ -196,6 +198,17 @@ figma.ui.onmessage = async (message: UiToPluginMessage) => {
     return;
   }
 
+  if (message.type === "get-runner-endpoint") {
+    const endpoint = await figma.clientStorage.getAsync(runnerEndpointStorageKey);
+    postToUi({ type: "runner-endpoint", endpoint: typeof endpoint === "string" ? endpoint : null });
+    return;
+  }
+
+  if (message.type === "set-runner-endpoint") {
+    await figma.clientStorage.setAsync(runnerEndpointStorageKey, message.endpoint);
+    return;
+  }
+
   if (message.type === "open-wordpress") {
     figma.openExternal(message.url);
     return;

plugins/figma-to-wordpress/src/types.ts

@@ -80,6 +80,10 @@ export type PluginToUiMessage =
       selection: NormalizedSelection | null;
       artifact: GeneratedArtifact | null;
     }
+  | {
+      type: "runner-endpoint";
+      endpoint: string | null;
+    }
   | {
       type: "error";
       message: string;
@@ -89,6 +93,13 @@ export type UiToPluginMessage =
   | {
       type: "refresh-document";
     }
+  | {
+      type: "get-runner-endpoint";
+    }
+  | {
+      type: "set-runner-endpoint";
+      endpoint: string;
+    }
   | {
       type: "open-wordpress";
       url: string;

plugins/figma-to-wordpress/src/ui.ts

@@ -10,7 +10,6 @@ const playgroundButton = document.querySelector<HTMLButtonElement>("#playground"
 const runnerEndpointInput = document.querySelector<HTMLInputElement>("#runner-endpoint");
 
 const defaultRunnerEndpoint = "http://localhost:8882/wp-json/static-site-importer/v1/import-figma";
-const runnerEndpointStorageKey = "figma-to-wordpress-runner-endpoint";
 
 function log(message: string, details?: unknown) {
   if (typeof details === "undefined") {
@@ -54,7 +53,7 @@ function runnerEndpoint(): string {
 function persistRunnerEndpoint() {
   const endpoint = runnerEndpoint();
 
-  localStorage.setItem(runnerEndpointStorageKey, endpoint);
+  sendToPlugin({ type: "set-runner-endpoint", endpoint });
 }
 
 async function openPlayground() {
@@ -134,6 +133,13 @@ window.onmessage = (event: MessageEvent<{ pluginMessage?: PluginToUiMessage }>)
     return;
   }
 
+  if (message.type === "runner-endpoint") {
+    if (runnerEndpointInput && message.endpoint) {
+      runnerEndpointInput.value = message.endpoint;
+    }
+    return;
+  }
+
   if (message.type === "error") {
     setStatus(message.message);
   }
@@ -144,8 +150,9 @@ playgroundButton?.addEventListener("click", () => void openPlayground());
 runnerEndpointInput?.addEventListener("change", persistRunnerEndpoint);
 
 if (runnerEndpointInput) {
-  runnerEndpointInput.value = localStorage.getItem(runnerEndpointStorageKey) || defaultRunnerEndpoint;
+  runnerEndpointInput.value = defaultRunnerEndpoint;
 }
 
 log("UI loaded.", { endpoint: runnerEndpoint() });
+sendToPlugin({ type: "get-runner-endpoint" });
 sendToPlugin({ type: "refresh-document" });

plugins/figma-to-wordpress/src/wordpress-runner.ts

@@ -17,6 +17,7 @@ export interface WordPressRunnerResponse {
   preview_session?: Record<string, unknown>;
   materialization?: Record<string, unknown>;
   error?: WordPressRunnerError;
+  message?: string;
 }
 
 const runnerTimeoutMs = 15000;
@@ -48,11 +49,11 @@ export async function createWordPressPreview(endpoint: string, artifact: Generat
   const data = await response.json().catch(() => null) as WordPressRunnerResponse | null;
 
   if (!response.ok) {
-    throw new Error(data?.error?.message || `WordPress runner failed with HTTP ${response.status}.`);
+    throw new Error(data?.error?.message || data?.message || `WordPress runner failed with HTTP ${response.status}.`);
   }
 
   if (!data?.success || !data.open_url) {
-    throw new Error(data?.error?.message || "WordPress runner did not return a Playground URL.");
+    throw new Error(data?.error?.message || data?.message || "WordPress runner did not return a Playground URL.");
   }
 
   return data;