openuispec 0.2.3 → 0.2.4

package/mcp-server/screenshot-shared.ts

@@ -0,0 +1,200 @@
+/**
+ * Shared utilities for platform screenshot tools (web, android, ios).
+ */
+
+import { existsSync, readFileSync, writeFileSync, readdirSync } from "node:fs";
+import { join, resolve, relative } from "node:path";
+import { createHash } from "node:crypto";
+import YAML from "yaml";
+import { findProjectDir } from "../drift/index.js";
+
+// ── shared result type ──────────────────────────────────────────────
+
+export interface ScreenshotResult {
+  content: Array<{ type: "text"; text: string } | { type: "image"; data: string; mimeType: string }>;
+  isError?: true;
+}
+
+// ── manifest loading ────────────────────────────────────────────────
+
+export interface ManifestInfo {
+  projectDir: string;
+  projectRoot: string;
+  manifest: any;
+  projectName: string;
+}
+
+export function loadManifest(projectCwd: string): ManifestInfo {
+  const projectDir = findProjectDir(projectCwd);
+  const manifestPath = join(projectDir, "openuispec.yaml");
+  const manifest = YAML.parse(readFileSync(manifestPath, "utf-8"));
+  const projectName = manifest.project?.name ?? "app";
+  const projectRoot = resolve(projectDir, "..");
+  return { projectDir, projectRoot, manifest, projectName };
+}
+
+// ── generic platform app directory discovery ────────────────────────
+
+function tryFindProjectDir(cwd: string): string | null {
+  if (existsSync(join(cwd, "openuispec.yaml"))) {
+    return cwd;
+  }
+  try { return findProjectDir(cwd); } catch { return null; }
+}
+
+export function findPlatformAppDir(
+  projectCwd: string,
+  platform: "web" | "android" | "ios",
+  existsCheck: (dir: string) => boolean,
+  directDir?: string,
+): string {
+  const label = platform.charAt(0).toUpperCase() + platform.slice(1);
+
+  // If a direct project dir is provided, use it without manifest lookup
+  if (directDir) {
+    const resolved = resolve(directDir);
+    if (existsCheck(resolved)) return resolved;
+    throw new Error(`${label} project not found at provided path: ${resolved}`);
+  }
+
+  // Check if projectCwd has an openuispec.yaml — if so, use manifest-based discovery
+  const manifestDir = tryFindProjectDir(projectCwd);
+
+  if (manifestDir) {
+    const manifestPath = join(manifestDir, "openuispec.yaml");
+    const manifest = YAML.parse(readFileSync(manifestPath, "utf-8"));
+    const projectName = manifest.project?.name ?? "app";
+    const projectRoot = resolve(manifestDir, "..");
+
+    // Check custom output_dir first
+    const customDir = manifest.generation?.output_dir?.[platform];
+    if (customDir) {
+      const resolved = resolve(manifestDir, customDir);
+      if (existsCheck(resolved)) return resolved;
+    }
+
+    // Default: generated/<platform>/<project-name>/
+    const defaultDir = join(projectRoot, "generated", platform, projectName);
+    if (existsCheck(defaultDir)) return defaultDir;
+
+    throw new Error(
+      `${label} app not found. Checked:\n` +
+      (customDir ? `  - ${resolve(manifestDir, customDir)}\n` : "") +
+      `  - ${defaultDir}\n` +
+      `Generate the ${platform} target first, then try again.`,
+    );
+  }
+
+  // No manifest — treat projectCwd itself as the platform project dir
+  const resolved = resolve(projectCwd);
+  if (existsCheck(resolved)) return resolved;
+
+  throw new Error(
+    `${label} project not found at ${resolved}. ` +
+    `Provide a project_dir parameter or create an openuispec.yaml manifest.`,
+  );
+}
+
+// ── file hashing / caching ──────────────────────────────────────────
+
+export function hashContent(content: string): string {
+  return createHash("md5").update(content).digest("hex");
+}
+
+export function loadHashes(dir: string, hashFile: string): Record<string, string> {
+  const hashPath = join(dir, hashFile);
+  try {
+    return JSON.parse(readFileSync(hashPath, "utf-8"));
+  } catch {
+    return {};
+  }
+}
+
+export function saveHashes(dir: string, hashFile: string, hashes: Record<string, string>): void {
+  writeFileSync(join(dir, hashFile), JSON.stringify(hashes, null, 2));
+}
+
+// ── screen name filter ──────────────────────────────────────────────
+
+export function matchesScreenFilter(screenName: string, filter: string): boolean {
+  const normalizedFilter = filter.toLowerCase().replace(/_/g, "");
+  const normalizedScreen = screenName.toLowerCase().replace(/_/g, "");
+  return normalizedScreen.includes(normalizedFilter) || normalizedFilter.includes(normalizedScreen);
+}
+
+// ── recursive file walker ───────────────────────────────────────────
+
+export function walkFiles(
+  dir: string,
+  ext: string,
+  skipDirs: string[] = [],
+): string[] {
+  const results: string[] = [];
+  if (!existsSync(dir)) return results;
+
+  const walk = (d: string) => {
+    for (const entry of readdirSync(d, { withFileTypes: true })) {
+      const fullPath = join(d, entry.name);
+      if (entry.isDirectory()) {
+        if (!skipDirs.includes(entry.name)) walk(fullPath);
+      } else if (entry.name.endsWith(ext)) {
+        results.push(fullPath);
+      }
+    }
+  };
+
+  walk(dir);
+  return results;
+}
+
+// ── PNG snapshot collector ──────────────────────────────────────────
+
+export function collectPngSnapshots(
+  dirs: string[],
+  rootDir: string,
+  screenFilter: string | undefined,
+  nameExtractor: (filename: string) => string,
+): Array<{ screen: string; path: string; data: string }> {
+  const snapshots: Array<{ screen: string; path: string; data: string }> = [];
+  const seen = new Set<string>();
+
+  for (const dir of dirs) {
+    const pngs = walkFiles(dir, ".png");
+    for (const fullPath of pngs) {
+      const filename = fullPath.split("/").pop()!;
+      const screen = nameExtractor(filename);
+
+      if (screenFilter && !matchesScreenFilter(screen, screenFilter)) continue;
+
+      const key = relative(rootDir, fullPath);
+      if (seen.has(key)) continue;
+      seen.add(key);
+
+      const data = readFileSync(fullPath).toString("base64");
+      snapshots.push({ screen, path: key, data });
+    }
+  }
+
+  return snapshots;
+}
+
+// ── screenshot response builder ─────────────────────────────────────
+
+export function buildScreenshotResponse(
+  snapshots: Array<{ screen: string; path: string; data: string }>,
+  metadataFn: (snapshot: { screen: string; path: string }) => Record<string, unknown>,
+): ScreenshotResult {
+  if (snapshots.length === 0) {
+    return {
+      content: [{ type: "text", text: "No screenshots generated. Check build output." }],
+      isError: true,
+    };
+  }
+
+  const content: ScreenshotResult["content"] = [];
+  for (const snapshot of snapshots) {
+    content.push({ type: "image" as const, data: snapshot.data, mimeType: "image/png" });
+    content.push({ type: "text" as const, text: JSON.stringify(metadataFn(snapshot), null, 2) });
+  }
+  return { content };
+}

package/mcp-server/screenshot.ts

@@ -6,7 +6,7 @@
  */
 
 import { spawn, type ChildProcess, execSync } from "node:child_process";
-import { existsSync, readFileSync } from "node:fs";
+import { existsSync, readFileSync, mkdirSync, writeFileSync } from "node:fs";
 import { join, resolve } from "node:path";
 import { createServer, type AddressInfo } from "node:net";
 import YAML from "yaml";
@@ -21,6 +21,7 @@ export interface ScreenshotOptions {
   wait_for?: number;
   full_page?: boolean;
   selector?: string;
+  output_dir?: string;
 }
 
 export interface ScreenshotResult {
@@ -180,6 +181,7 @@ export async function takeScreenshot(
     wait_for = 1000,
     full_page = false,
     selector,
+    output_dir,
   } = options;
 
   // 1. Find and start
@@ -223,6 +225,17 @@ export async function takeScreenshot(
 
     const base64 = buffer.toString("base64");
 
+    // Save to output_dir if specified
+    let savedPath: string | undefined;
+    if (output_dir) {
+      const outDir = resolve(webDir, output_dir);
+      mkdirSync(outDir, { recursive: true });
+      const routeSlug = route.replace(/^\//, "").replace(/\//g, "_") || "index";
+      const themeLabel = theme ?? "default";
+      savedPath = join(outDir, `${routeSlug}_${themeLabel}.png`);
+      writeFileSync(savedPath, buffer);
+    }
+
     return {
       content: [
         { type: "image" as const, data: base64, mimeType: "image/png" },
@@ -235,6 +248,7 @@ export async function takeScreenshot(
             theme: theme ?? "default",
             full_page,
             selector: selector ?? null,
+            path: savedPath ?? null,
           }, null, 2),
         },
       ],

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "openuispec",
-  "version": "0.2.3",
+  "version": "0.2.4",
   "license": "MIT",
   "type": "module",
   "description": "A semantic UI specification format for AI-native, platform-native app development",
@@ -42,7 +42,8 @@
     "drift:snapshot": "tsx drift/index.ts --snapshot --target",
     "prepare:target": "tsx prepare/index.ts",
     "status": "tsx status/index.ts",
-    "postinstall": "echo \"\\n  ✓ openuispec installed — if upgrading, run: openuispec update-rules\\n\""
+    "postinstall": "echo \"\\n  ✓ openuispec installed — if upgrading, run: openuispec update-rules\\n\"",
+    "cloc": "cloc --exclude-dir=$(tr '\n' ',' < .cloc-ignore) ."
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.27.1",

package/examples/todo-orbit/generated/ios/Todo Orbit/TodoOrbit.xcodeproj/xcshareddata/xcschemes/TodoOrbit.xcscheme

@@ -1,79 +0,0 @@
-<?xml version="1.0" encoding="UTF-8"?>
-<Scheme
-   LastUpgradeVersion = "2600"
-   version = "1.7">
-   <BuildAction
-      parallelizeBuildables = "YES"
-      buildImplicitDependencies = "YES"
-      buildArchitectures = "Automatic">
-      <BuildActionEntries>
-         <BuildActionEntry
-            buildForTesting = "YES"
-            buildForRunning = "YES"
-            buildForProfiling = "YES"
-            buildForArchiving = "YES"
-            buildForAnalyzing = "YES">
-            <BuildableReference
-               BuildableIdentifier = "primary"
-               BlueprintIdentifier = "E00000010000000000000000"
-               BuildableName = "TodoOrbit.app"
-               BlueprintName = "TodoOrbit"
-               ReferencedContainer = "container:TodoOrbit.xcodeproj">
-            </BuildableReference>
-         </BuildActionEntry>
-      </BuildActionEntries>
-   </BuildAction>
-   <TestAction
-      buildConfiguration = "Debug"
-      selectedDebuggerIdentifier = "Xcode.DebuggerFoundation.Debugger.LLDB"
-      selectedLauncherIdentifier = "Xcode.DebuggerFoundation.Launcher.LLDB"
-      shouldUseLaunchSchemeArgsEnv = "YES">
-      <Testables>
-      </Testables>
-   </TestAction>
-   <LaunchAction
-      buildConfiguration = "Debug"
-      selectedDebuggerIdentifier = "Xcode.DebuggerFoundation.Debugger.LLDB"
-      selectedLauncherIdentifier = "Xcode.DebuggerFoundation.Launcher.LLDB"
-      launchStyle = "0"
-      useCustomWorkingDirectory = "NO"
-      ignoresPersistentStateOnLaunch = "NO"
-      debugDocumentVersioning = "YES"
-      debugServiceExtension = "internal"
-      allowLocationSimulation = "YES">
-      <BuildableProductRunnable
-         runnableDebuggingMode = "0">
-         <BuildableReference
-            BuildableIdentifier = "primary"
-            BlueprintIdentifier = "E00000010000000000000000"
-            BuildableName = "TodoOrbit.app"
-            BlueprintName = "TodoOrbit"
-            ReferencedContainer = "container:TodoOrbit.xcodeproj">
-         </BuildableReference>
-      </BuildableProductRunnable>
-   </LaunchAction>
-   <ProfileAction
-      buildConfiguration = "Release"
-      shouldUseLaunchSchemeArgsEnv = "YES"
-      savedToolIdentifier = ""
-      useCustomWorkingDirectory = "NO"
-      debugDocumentVersioning = "YES">
-      <BuildableProductRunnable
-         runnableDebuggingMode = "0">
-         <BuildableReference
-            BuildableIdentifier = "primary"
-            BlueprintIdentifier = "E00000010000000000000000"
-            BuildableName = "TodoOrbit.app"
-            BlueprintName = "TodoOrbit"
-            ReferencedContainer = "container:TodoOrbit.xcodeproj">
-         </BuildableReference>
-      </BuildableProductRunnable>
-   </ProfileAction>
-   <AnalyzeAction
-      buildConfiguration = "Debug">
-   </AnalyzeAction>
-   <ArchiveAction
-      buildConfiguration = "Release"
-      revealArchiveInOrganizer = "YES">
-   </ArchiveAction>
-</Scheme>