@nodii/dev 0.0.0-stage → 0.3.1

package/dist/hooks.js

@@ -0,0 +1,153 @@
+import { CliError, hubRequest } from "./core.js";
+
+const FIGMA_PREFIXES = [
+  "mcp__figma__",
+  "mcp__claude_ai_Figma__",
+  "mcp__figma-dev-mode__",
+];
+const READ_EXACT = new Set([
+  "whoami",
+  "get_metadata",
+  "get_screenshot",
+  "get_design_context",
+  "get_variable_defs",
+  "get_libraries",
+  "search_design_system",
+  "get_code_connect_map",
+  "get_code_connect_suggestions",
+  "list_file_components_for_code_connect",
+  "get_figjam",
+  "get_figma_skill",
+  "create_session",
+  "get_context_for_code_connect",
+  "get_motion_context",
+  "download_assets",
+]);
+const suffix = (name) =>
+  FIGMA_PREFIXES.find((p) => name.startsWith(p))
+    ? name.slice(FIGMA_PREFIXES.find((p) => name.startsWith(p)).length)
+    : null;
+export const isFigmaTool = (name) => suffix(name) !== null;
+export const isReadTool = (name) => {
+  const s = suffix(name);
+  return (
+    s !== null &&
+    (READ_EXACT.has(s) || s.startsWith("list_") || s.startsWith("get_"))
+  );
+};
+
+export function extractFileKey(input) {
+  if (!input || typeof input !== "object") return null;
+  for (const key of ["fileKey", "file_key"])
+    if (typeof input[key] === "string" && input[key]) return input[key];
+  for (const key of ["url", "figmaUrl", "figma_url"])
+    if (typeof input[key] === "string") {
+      const match = input[key].match(
+        /figma\.com\/(?:design|file|board)\/([^/?#]+)/,
+      );
+      if (match) return match[1];
+    }
+  for (const value of Object.values(input)) {
+    const found = extractFileKey(value);
+    if (found) return found;
+  }
+  return null;
+}
+
+export function extractNodeIds(value, out = new Set()) {
+  if (!value || typeof value !== "object") return [...out];
+  for (const [key, item] of Object.entries(value)) {
+    if (/^(node_?ids?|id)$/i.test(key))
+      for (const v of Array.isArray(item) ? item : [item])
+        if (
+          typeof v === "string" &&
+          /^(?:\d+[:;-]\d+|[A-Za-z0-9_-]{8,})$/.test(v)
+        )
+          out.add(v);
+    if (typeof item === "object") extractNodeIds(item, out);
+  }
+  return [...out];
+}
+
+export async function preHook(payload, request = hubRequest) {
+  const name = payload.tool_name ?? "";
+  if (!isFigmaTool(name) || isReadTool(name))
+    return { allowed: true, ignored: true };
+  const fileKey = extractFileKey(payload.tool_input);
+  if (!fileKey)
+    throw new CliError(
+      "FIGMA_FILE_KEY_MISSING",
+      "Blocked Figma write: no fileKey/file_key/Figma URL was found.",
+      2,
+    );
+  const config = (await import("./core.js")).readConfig;
+  const { agentId } = await config();
+  let response;
+  try {
+    response = await request(
+      "GET",
+      `/api/v1/design/files/${encodeURIComponent(fileKey)}/lock/check?agentId=${encodeURIComponent(agentId)}`,
+    );
+  } catch (error) {
+    throw new CliError(
+      "FIGMA_CHECK_FAILED",
+      `Blocked Figma write: ${error.message}`,
+      2,
+    );
+  }
+  if (!response.data?.allowed)
+    throw new CliError(
+      "FIGMA_LOCK_DENIED",
+      response.data?.reason ?? "Blocked Figma write: lock not held.",
+      2,
+      response.data,
+    );
+  return { allowed: true, fileKey, changeSetId: response.data.changeSetId };
+}
+
+export async function postHook(payload, request = hubRequest) {
+  const name = payload.tool_name ?? "";
+  if (!isFigmaTool(name) || isReadTool(name))
+    return { logged: false, ignored: true };
+  const fileKey = extractFileKey(payload.tool_input);
+  if (!fileKey) return { logged: false, reason: "no file key" };
+  const { agentId } = await (await import("./core.js")).readConfig();
+  const checked = await request(
+    "GET",
+    `/api/v1/design/files/${encodeURIComponent(fileKey)}/lock/check?agentId=${encodeURIComponent(agentId)}`,
+  );
+  if (!checked.data?.allowed || !checked.data.changeSetId)
+    return { logged: false, reason: checked.data?.reason ?? "lock denied" };
+  const nodeIds = extractNodeIds({
+    input: payload.tool_input,
+    response: payload.tool_response,
+  });
+  if (!nodeIds.length) nodeIds.push("unknown");
+  await request(
+    "POST",
+    `/api/v1/design/change-sets/${encodeURIComponent(checked.data.changeSetId)}/writes`,
+    { tool: name, nodeIds },
+  );
+  return { logged: true, changeSetId: checked.data.changeSetId, nodeIds };
+}
+
+export const settingsSnippet = () => ({
+  hooks: {
+    PreToolUse: [
+      {
+        matcher: "^mcp__(figma|claude_ai_Figma|figma-dev-mode)__.*$",
+        hooks: [
+          { type: "command", command: "nodii-dev hook claude-figma-pre" },
+        ],
+      },
+    ],
+    PostToolUse: [
+      {
+        matcher: "^mcp__(figma|claude_ai_Figma|figma-dev-mode)__.*$",
+        hooks: [
+          { type: "command", command: "nodii-dev hook claude-figma-post" },
+        ],
+      },
+    ],
+  },
+});