heroshot 0.17.0 → 0.19.0

package/dist/server-BklhdmjN.js

@@ -0,0 +1,319 @@
+#!/usr/bin/env node
+import { _ as loadConfig, b as screenshotSchema, g as getConfigPath, i as filterScreenshots, r as sync, t as generateSnippets, v as saveConfig, x as generateUid } from "./snippet-ZcZmuryj.js";
+import { z } from "zod";
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+//#region src/mcp/schemas/add.ts
+/**
+* Add tool input schema.
+* Wraps the existing screenshotSchema from heroshot.
+*/
+const screenshotInputSchema = screenshotSchema.omit({ id: true });
+const addOptionsSchema = z.object({
+	configPath: z.string().optional().describe("Path to config file (default: .heroshot/config.json in cwd)"),
+	screenshot: screenshotInputSchema.describe("Screenshot definition to add")
+});
+//#endregion
+//#region src/mcp/schemas/list.ts
+/**
+* List tool input schema.
+*/
+const listOptionsSchema = z.object({
+	configPath: z.string().optional().describe("Path to config file (default: .heroshot/config.json in cwd)"),
+	filter: z.string().optional().describe("Filter screenshots by id, name, or filename (case-insensitive substring match)")
+});
+//#endregion
+//#region src/mcp/schemas/remove.ts
+/**
+* Remove tool input schema.
+*/
+const removeOptionsSchema = z.object({
+	configPath: z.string().optional().describe("Path to config file (default: .heroshot/config.json in cwd)"),
+	id: z.string().min(1).describe("ID of the screenshot to remove")
+});
+//#endregion
+//#region src/mcp/schemas/snippet.ts
+/**
+* Snippet tool input schema.
+*/
+const snippetOptionsSchema = z.object({
+	configPath: z.string().optional().describe("Path to config file (default: .heroshot/config.json in cwd)"),
+	filter: z.string().optional().describe("Filter screenshots by id, name, or filename (case-insensitive substring match)"),
+	pathPrefix: z.string().optional().describe("Custom path prefix for images (default: ./heroshots/)")
+});
+//#endregion
+//#region src/mcp/schemas/sync.ts
+/**
+* Sync tool input schema.
+*/
+const syncOptionsSchema = z.object({
+	configPath: z.string().optional().describe("Path to config file (default: .heroshot/config.json in cwd)"),
+	filter: z.string().optional().describe("Filter screenshots by id, name, or filename (case-insensitive substring match)"),
+	clean: z.boolean().optional().describe("Delete stale files in output directory"),
+	workers: z.number().int().min(1).optional().describe("Number of parallel capture workers"),
+	sessionKey: z.string().optional().describe("Encrypted session key for authenticated screenshots")
+});
+//#endregion
+//#region src/mcp/tools/add.ts
+/**
+* Add tool handler.
+* Adds a new screenshot definition to config.
+*/
+function addHandler(input) {
+	try {
+		const configPath = input.configPath ?? getConfigPath();
+		const config = loadConfig(configPath);
+		const screenshotInput = {
+			...input.screenshot,
+			id: generateUid()
+		};
+		const screenshot = screenshotSchema.parse(screenshotInput);
+		const existingByName = config.screenshots.find((s) => s.name.toLowerCase() === screenshot.name.toLowerCase());
+		if (existingByName) return {
+			success: false,
+			error: `Screenshot with name "${screenshot.name}" already exists (id: ${existingByName.id})`
+		};
+		config.screenshots.push(screenshot);
+		saveConfig(configPath, config);
+		return {
+			success: true,
+			id: screenshot.id,
+			screenshot: {
+				id: screenshot.id,
+				name: screenshot.name,
+				url: screenshot.url
+			}
+		};
+	} catch (error) {
+		return {
+			success: false,
+			error: error instanceof Error ? error.message : String(error)
+		};
+	}
+}
+//#endregion
+//#region src/mcp/tools/list.ts
+/**
+* List tool handler.
+* Lists screenshots defined in config.
+*/
+function listHandler(input) {
+	try {
+		const screenshots = filterScreenshots(loadConfig(input.configPath ?? getConfigPath()).screenshots, input.filter);
+		return {
+			success: true,
+			count: screenshots.length,
+			screenshots: screenshots.map((s) => ({
+				id: s.id,
+				name: s.name,
+				url: s.url,
+				selector: s.selector
+			}))
+		};
+	} catch {
+		return {
+			success: false,
+			count: 0,
+			screenshots: [],
+			error: "Failed to load config"
+		};
+	}
+}
+//#endregion
+//#region src/mcp/tools/remove.ts
+/**
+* Remove tool handler.
+* Removes a screenshot definition from config.
+*/
+function removeHandler(input) {
+	try {
+		const configPath = input.configPath ?? getConfigPath();
+		const config = loadConfig(configPath);
+		const index = config.screenshots.findIndex((s) => s.id === input.id);
+		if (index === -1) return {
+			success: false,
+			error: `Screenshot with id "${input.id}" not found`
+		};
+		const [removed] = config.screenshots.splice(index, 1);
+		if (!removed) return {
+			success: false,
+			error: `Screenshot with id "${input.id}" not found`
+		};
+		saveConfig(configPath, config);
+		return {
+			success: true,
+			removed: {
+				id: removed.id,
+				name: removed.name
+			}
+		};
+	} catch (error) {
+		return {
+			success: false,
+			error: error instanceof Error ? error.message : String(error)
+		};
+	}
+}
+//#endregion
+//#region src/mcp/tools/snippet.ts
+/**
+* Snippet tool handler.
+* Generates markdown/HTML snippets for screenshots.
+*/
+function snippetHandler(input) {
+	try {
+		const results = generateSnippets(loadConfig(input.configPath ?? getConfigPath()), input.filter, { pathPrefix: input.pathPrefix });
+		return {
+			success: true,
+			count: results.length,
+			snippets: results.map((r) => ({
+				id: r.screenshot.id,
+				name: r.screenshot.name,
+				snippet: r.snippet
+			}))
+		};
+	} catch (error) {
+		return {
+			success: false,
+			count: 0,
+			snippets: [],
+			error: error instanceof Error ? error.message : String(error)
+		};
+	}
+}
+//#endregion
+//#region src/mcp/tools/sync.ts
+/**
+* Sync tool handler.
+* Captures screenshots defined in config.
+*/
+function resolveSessionKey(input) {
+	return input.sessionKey ?? process.env["HEROSHOT_SESSION_KEY"];
+}
+async function syncHandler(input) {
+	try {
+		const configPath = input.configPath ?? getConfigPath();
+		const sessionKey = resolveSessionKey(input);
+		const result = await sync({
+			configPath,
+			filter: input.filter,
+			clean: input.clean,
+			workers: input.workers,
+			sessionKey
+		});
+		return {
+			success: result.failed === 0,
+			total: result.total,
+			captured: result.success,
+			failed: result.failed,
+			results: result.results,
+			staleFiles: result.staleFiles,
+			deletedFiles: result.deletedFiles
+		};
+	} catch (error) {
+		return {
+			success: false,
+			total: 0,
+			captured: 0,
+			failed: 0,
+			results: [],
+			error: error instanceof Error ? error.message : String(error)
+		};
+	}
+}
+//#endregion
+//#region src/mcp/tools/definitions.ts
+function formatResult(result) {
+	return {
+		content: [{
+			type: "text",
+			text: JSON.stringify(result, null, 2)
+		}],
+		isError: !result.success
+	};
+}
+async function syncToolHandler(input) {
+	return formatResult(await syncHandler(input));
+}
+async function addToolHandler(input) {
+	return formatResult(addHandler(input));
+}
+async function listToolHandler(input) {
+	return formatResult(listHandler(input));
+}
+async function snippetToolHandler(input) {
+	return formatResult(snippetHandler(input));
+}
+async function removeToolHandler(input) {
+	return formatResult(removeHandler(input));
+}
+const tools = [
+	{
+		name: "heroshot_sync",
+		description: "Capture all screenshots defined in the heroshot config. Optionally filter by id/name/filename pattern, delete stale files, or run with multiple workers.",
+		inputSchema: syncOptionsSchema,
+		handler: syncToolHandler
+	},
+	{
+		name: "heroshot_add",
+		description: "Add a new screenshot definition to the heroshot config. Requires at least name and url. Optional: selector for element capture, actions for pre-capture steps.",
+		inputSchema: addOptionsSchema,
+		handler: addToolHandler
+	},
+	{
+		name: "heroshot_list",
+		description: "List all screenshots defined in the heroshot config. Returns id, name, url, and selector for each screenshot.",
+		inputSchema: listOptionsSchema,
+		handler: listToolHandler
+	},
+	{
+		name: "heroshot_snippet",
+		description: "Generate markdown/HTML snippets for embedding screenshots in documentation. Supports light/dark mode with <picture> elements.",
+		inputSchema: snippetOptionsSchema,
+		handler: snippetToolHandler
+	},
+	{
+		name: "heroshot_remove",
+		description: "Remove a screenshot definition from the heroshot config by its ID.",
+		inputSchema: removeOptionsSchema,
+		handler: removeToolHandler
+	}
+];
+//#endregion
+//#region src/mcp/utils/zodToMcp.ts
+/**
+* Utilities for converting Zod schemas to MCP tool definitions.
+*/
+/**
+* Convert a ToolDefinition to MCP tool format.
+* Uses z.toJSONSchema() for automatic schema conversion.
+*/
+function toMcpTool(definition) {
+	return {
+		name: definition.name,
+		description: definition.description,
+		inputSchema: z.toJSONSchema(definition.inputSchema)
+	};
+}
+//#endregion
+//#region src/mcp/server.ts
+/**
+* Heroshot MCP Server
+*
+* MCP server for screenshot automation. Tools are auto-derived from Zod schemas
+* using z.toJSONSchema() for low maintenance.
+*/
+async function startMcpServer() {
+	const server = new McpServer({
+		name: "heroshot",
+		version: "0.1.0"
+	});
+	for (const tool of tools) {
+		const mcpTool = toMcpTool(tool);
+		server.registerTool(tool.name, mcpTool.inputSchema, async (input) => tool.handler(input));
+	}
+	const transport = new StdioServerTransport();
+	await server.connect(transport);
+}
+//#endregion
+export { startMcpServer as t };

package/dist/server-CqRfIFJa.js

@@ -0,0 +1,3 @@
+#!/usr/bin/env node
+import { t as startMcpServer } from "./server-BklhdmjN.js";
+export { startMcpServer };