@dyyz1993/create-agent 2.0.0 → 2.1.0

package/templates/chat/src/server-config.ts

@@ -3,40 +3,42 @@
  * Values are read from environment variables with sensible defaults.
  */
 
+import { randomUUID } from "crypto";
+
+if (process.env.NODE_ENV === "production" && !process.env.AUTH_TOKEN) {
+	throw new Error(
+		"[FATAL] AUTH_TOKEN environment variable must be set in production mode. " +
+			"Example: AUTH_TOKEN=$(openssl rand -hex 32) bun src/server.ts"
+	);
+}
+
 export function parseEnvInt(
-  key: string,
-  value: string | undefined,
-  defaultValue: number,
-  min: number,
-  max: number,
+	key: string,
+	value: string | undefined,
+	defaultValue: number,
+	min: number,
+	max: number
 ): number {
-  if (value === undefined || value === "") return defaultValue;
-  const parsed = parseInt(value, 10);
-  if (isNaN(parsed) || parsed < min || parsed > max) {
-    console.warn(
-      `[config] Invalid ${key}: "${value}", using default: ${defaultValue}`,
-    );
-    return defaultValue;
-  }
-  return parsed;
+	if (value === undefined || value === "") return defaultValue;
+	const parsed = parseInt(value, 10);
+	if (isNaN(parsed) || parsed < min || parsed > max) {
+		process.stderr.write(`[config] Invalid ${key}: "${value}", using default: ${defaultValue}\n`);
+		return defaultValue;
+	}
+	return parsed;
 }
 
 export const config = {
-  port: parseEnvInt("PORT", process.env.PORT, 3100, 1024, 65535),
-  authToken: process.env.AUTH_TOKEN || "pi-agent-template-token",
-  maxUploadSize: parseEnvInt(
-    "MAX_UPLOAD_SIZE",
-    process.env.MAX_UPLOAD_SIZE,
-    50 * 1024 * 1024,
-    0,
-    1024 * 1024 * 1024,
-  ),
-  logDir: process.env.LOG_DIR || "logs",
-  corsOrigin: process.env.CORS_ORIGIN || "http://localhost:5173",
+	port: parseEnvInt("PORT", process.env.PORT, 3100, 1024, 65535),
+	authToken: process.env.AUTH_TOKEN || `dev-${randomUUID()}`,
+	maxUploadSize: parseEnvInt(
+		"MAX_UPLOAD_SIZE",
+		process.env.MAX_UPLOAD_SIZE,
+		50 * 1024 * 1024,
+		0,
+		1024 * 1024 * 1024
+	),
+	logDir: process.env.LOG_DIR || "logs",
+	enableWebService: process.env.ENABLE_WEB_SERVICE === "true",
+	corsOrigin: process.env.CORS_ORIGIN || "http://localhost:5173",
 } as const;
-
-if (process.env.NODE_ENV === "production" && !process.env.AUTH_TOKEN) {
-  console.warn(
-    "[security] WARNING: Using default AUTH_TOKEN in production. Set AUTH_TOKEN environment variable.",
-  );
-}

package/templates/chat/src/server.ts

@@ -1,15 +1,7 @@
-/**
- * Web server entry point — HTTP file endpoints + WebSocket RPC gateway.
- * Port auto-negotiation: tries config.port first, increments on EADDRINUSE.
- * Writes actual port to .server-port for dev orchestration.
- */
-
-import { createServer } from "http";
 import { writeFileSync, unlinkSync, existsSync } from "fs";
 import { join, resolve, basename } from "path";
 import { config } from "./server-config";
-import { createHttpHandler } from "./gateway/http-routes";
-import { createWsHandler } from "./gateway/ws-handler";
+import { createWebServer, getLocalIP } from "./shared/lib/web-server";
 import { createLogger, configureLogDir } from "./shared/lib/logger";
 import { registerPort, unregisterPort, formatRegistryForOutput } from "./shared/lib/port-registry";
 import { discoverMethodNames } from "./shared/register-all-handlers";
@@ -22,68 +14,58 @@ const PROJECT_ROOT = resolve(import.meta.dir, "..");
 const PROJECT_NAME = basename(PROJECT_ROOT);
 
 function cleanupPortFile() {
-  try { if (existsSync(PORT_FILE)) unlinkSync(PORT_FILE); } catch {}
-  unregisterPort(PROJECT_ROOT);
+	try {
+		if (existsSync(PORT_FILE)) unlinkSync(PORT_FILE);
+	} catch {}
+	unregisterPort(PROJECT_ROOT);
 }
 
 function writePortFile(port: number) {
-  writeFileSync(PORT_FILE, String(port), "utf-8");
+	writeFileSync(PORT_FILE, String(port), "utf-8");
 }
 
 process.on("exit", cleanupPortFile);
-process.on("SIGINT", () => { cleanupPortFile(); process.exit(0); });
-process.on("SIGTERM", () => { cleanupPortFile(); process.exit(0); });
-
-const httpServer = createServer();
-const wss = createWsHandler(httpServer, { config });
+process.on("SIGINT", () => {
+	cleanupPortFile();
+	process.exit(0);
+});
+process.on("SIGTERM", () => {
+	cleanupPortFile();
+	process.exit(0);
+});
 
-httpServer.on("request", createHttpHandler({
-  config,
-  getWebSocketClientCount: () => wss.clients.size,
-}));
+async function start() {
+	const { port, close } = await createWebServer({
+		port: config.port,
+		authToken: config.authToken,
+		maxUploadSize: config.maxUploadSize,
+		corsOrigin: config.corsOrigin,
+	});
 
-function checkPort(port: number): Promise<number> {
-  return new Promise((resolve, reject) => {
-    const testServer = createServer();
-    testServer.once("error", (err: NodeJS.ErrnoException) => {
-      testServer.close();
-      reject(err);
-    });
-    testServer.once("listening", () => {
-      testServer.close();
-      resolve(port);
-    });
-    testServer.listen(port);
-  });
-}
+	process.on("SIGINT", () => {
+		close();
+		cleanupPortFile();
+		process.exit(0);
+	});
+	process.on("SIGTERM", () => {
+		close();
+		cleanupPortFile();
+		process.exit(0);
+	});
 
-async function findAvailablePort(startPort: number, maxRetries: number = 10): Promise<number> {
-  for (let port = startPort; port < startPort + maxRetries; port++) {
-    try {
-      await checkPort(port);
-      return port;
-    } catch {
-      log.info(`Port ${port} in use, trying ${port + 1}...`);
-    }
-  }
-  throw new Error(`No available port found after ${maxRetries} retries`);
-}
+	writePortFile(port);
+	registerPort(PROJECT_ROOT, port, PROJECT_NAME);
 
-async function start() {
-  const port = await findAvailablePort(config.port);
-  httpServer.listen(port, () => {
-    writePortFile(port);
-    registerPort(PROJECT_ROOT, port, PROJECT_NAME);
-    log.info(`HTTP + WebSocket server running on http://localhost:${port}`);
-    log.info(`WebSocket: ws://localhost:${port}/ws (auth required)`);
-    log.info(`Available RPC methods: ${discoverMethodNames().join(", ")}`);
-    log.info("File endpoints: GET /file/{path}, GET /info/{path}");
-    // eslint-disable-next-line no-console
-    console.log("\n" + formatRegistryForOutput() + "\n");
-  });
+	const localIp = getLocalIP();
+	log.info(`HTTP + WebSocket server running on http://localhost:${port}`);
+	log.info(`Local network access: http://${localIp}:${port}`);
+	log.info(`WebSocket: ws://localhost:${port}/ws (auth required)`);
+	log.info(`Available RPC methods: ${discoverMethodNames().join(", ")}`);
+	log.info("File endpoints: GET /file/{path}, GET /info/{path}");
+	log.info(formatRegistryForOutput());
 }
 
 start().catch((err) => {
-  log.error("Server failed to start", { error: err.message });
-  process.exit(1);
+	log.error("Server failed to start", { error: err.message });
+	process.exit(1);
 });

package/templates/chat/src/shared/handlers/chat.ts

@@ -9,178 +9,212 @@ import { createLogger } from "../lib/logger";
 
 const log = createLogger("chat");
 
-function getStoragePath(): string {
-  const dir = join(homedir(), ".pi-agent");
-  return join(dir, "chat-history.json");
+type Platform = "desktop" | "web";
+
+export function getStoragePathFor(platform: Platform): string {
+	const dir = join(homedir(), ".pi-agent");
+	if (platform === "desktop") {
+		return join(dir, "chat-history-desktop.json");
+	}
+	const sessionId = Date.now().toString(36) + "-" + Math.random().toString(36).slice(2, 8);
+	return join(dir, `chat-history-web-${sessionId}.json`);
 }
 
 type ChatMessage = { id: string; role: "user" | "assistant"; content: string; timestamp: number };
 
-async function loadMessages(): Promise<ChatMessage[]> {
-  const filePath = getStoragePath();
-  try {
-    if (!existsSync(filePath)) {
-      log.info(`No history file at ${filePath}`);
-      return [];
-    }
-    const raw = await readFile(filePath, "utf-8");
-    const msgs = JSON.parse(raw) as ChatMessage[];
-    log.info(`Loaded ${msgs.length} messages from ${filePath}`);
-    return msgs;
-  } catch (err) {
-    log.error("Failed to load history", { error: err });
-    return [];
-  }
-}
-
-async function saveMessages(messages: ChatMessage[]): Promise<void> {
-  const filePath = getStoragePath();
-  const dir = dirname(filePath);
-  try {
-    if (!existsSync(dir)) {
-      await mkdir(dir, { recursive: true });
-    }
-    await writeFile(filePath, JSON.stringify(messages, null, 2), "utf-8");
-    log.info(`Saved ${messages.length} messages to ${filePath}`);
-  } catch (err) {
-    log.error("Failed to save history", { error: err });
-  }
+function createMessageStore(filePath: string) {
+	async function load(): Promise<ChatMessage[]> {
+		try {
+			if (!existsSync(filePath)) {
+				log.info(`No history file at ${filePath}`);
+				return [];
+			}
+			const raw = await readFile(filePath, "utf-8");
+			const msgs = JSON.parse(raw) as ChatMessage[];
+			log.info(`Loaded ${msgs.length} messages from ${filePath}`);
+			return msgs;
+		} catch (err) {
+			log.error("Failed to load history", { error: err });
+			return [];
+		}
+	}
+
+	async function save(messages: ChatMessage[]): Promise<void> {
+		const dir = dirname(filePath);
+		try {
+			if (!existsSync(dir)) {
+				await mkdir(dir, { recursive: true });
+			}
+			await writeFile(filePath, JSON.stringify(messages, null, 2), "utf-8");
+			log.info(`Saved ${messages.length} messages to ${filePath}`);
+		} catch (err) {
+			log.error("Failed to save history", { error: err });
+		}
+	}
+
+	return { load, save };
 }
 
 type RegisterFn = <K extends keyof RPCMethods & string>(
-  method: K,
-  handler: (params: MethodParams<RPCMethods, K>) => Promise<MethodResult<RPCMethods, K>>,
+	method: K,
+	handler: (params: MethodParams<RPCMethods, K>) => Promise<MethodResult<RPCMethods, K>>
 ) => void;
 
 export function generateReply(input: string): string {
-  const lower = input.toLowerCase().trim();
-
-  if (/^(hi|hello|hey|howdy|hola|yo|sup)\b/i.test(lower)) {
-    const greetings = [
-      "Hey there! How can I help you today?",
-      "Hello! Great to see you. What would you like to know?",
-      "Hi! I'm your desktop assistant. Ask me anything!",
-    ];
-    return greetings[Math.floor(Math.random() * greetings.length)];
-  }
-
-  if (/what('?s| is) the (time|date|day)|current (time|date)|what time|today'?s date/i.test(lower)) {
-    const now = new Date();
-    const date = now.toLocaleDateString("en-US", { weekday: "long", year: "numeric", month: "long", day: "numeric" });
-    const time = now.toLocaleTimeString("en-US", { hour: "2-digit", minute: "2-digit" });
-    return `It's currently **${time}** on **${date}**.`;
-  }
-
-  const mathMatch = lower.match(
-    /(?:what(?:'s| is)\s+)?(\d+(?:\.\d+)?)\s*([+\-*/x×÷^])\s*(\d+(?:\.\d+)?)/,
-  );
-  if (mathMatch) {
-    const a = parseFloat(mathMatch[1]);
-    const op = mathMatch[2];
-    const b = parseFloat(mathMatch[3]);
-    let result: number;
-    switch (op) {
-      case "+": result = a + b; break;
-      case "-": result = a - b; break;
-      case "*": case "x": case "×": result = a * b; break;
-      case "/": case "÷": result = b !== 0 ? a / b : NaN; break;
-      case "^": result = Math.pow(a, b); break;
-      default: result = NaN;
-    }
-    if (isNaN(result)) {
-      return "Hmm, I couldn't calculate that. Did you try dividing by zero?";
-    }
-    const niceResult = Number.isInteger(result) ? result.toString() : result.toFixed(4).replace(/0+$/, "").replace(/\.$/, "");
-    return `That would be **${niceResult}**! Need help with anything else?`;
-  }
-
-  if (/file|files|browse|explorer|directory|folder|open file/i.test(lower)) {
-    return (
-      "I can help you explore files! While I can't directly browse files in this demo, " +
-      "here's what you can do:\n\n" +
-      "- Use the **Explorer** panel to browse your project files\n" +
-      "- Click any file to preview its contents\n" +
-      "- Use the search feature to find specific files\n\n" +
-      "Try asking about **time** or **math** for a live demo!"
-    );
-  }
-
-  if (/git|commit|branch|status|diff|push|pull/i.test(lower)) {
-    return (
-      "Git is a powerful version control system! Here's what I know:\n\n" +
-      "- **git status** - Check your current changes\n" +
-      "- **git branch** - See or switch branches\n" +
-      "- **git log** - View commit history\n" +
-      "- **git diff** - See what changed\n\n" +
-      "Check out the **Source Control** panel for a visual Git interface! " +
-      "Or try asking me about **time** or **math**."
-    );
-  }
-
-  if (/^(help|commands|what can you|what do you|capabilities|features)/i.test(lower)) {
-    return (
-      "Here's what I can help with:\n\n" +
-      "- **Greetings** - Say hi and I'll say hi back!\n" +
-      "- **Time & Date** - Ask \"what time is it?\" or \"what's today's date?\"\n" +
-      "- **Math** - Give me an expression like \"12 * 8\" or \"what is 100 / 4\"\n" +
-      "- **Files** - Ask about file browsing and the explorer\n" +
-      "- **Git** - Ask about version control commands\n" +
-      "- **Help** - Show this message anytime!\n\n" +
-      "This is a demo assistant - try different things to see what sticks!"
-    );
-  }
-
-  const defaults = [
-    "That's an interesting question! I'm a demo assistant, so my knowledge is limited - but try asking about **time**, **math**, **files**, or **git**.",
-    "I wish I could help with that! For now I can answer questions about time, math, files, and git. Type **help** to see what I can do.",
-    "Hmm, I'm not sure about that one. But I *can* do math, tell you the time, and talk about files and git. Give it a shot!",
-  ];
-  return defaults[Math.floor(Math.random() * defaults.length)];
+	const lower = input.toLowerCase().trim();
+
+	if (/^(hi|hello|hey|howdy|hola|yo|sup)\b/i.test(lower)) {
+		const greetings = [
+			"Hey there! How can I help you today?",
+			"Hello! Great to see you. What would you like to know?",
+			"Hi! I'm your desktop assistant. Ask me anything!",
+		];
+		return greetings[Math.floor(Math.random() * greetings.length)]!;
+	}
+
+	if (
+		/what('?s| is) the (time|date|day)|current (time|date)|what time|today'?s date/i.test(lower)
+	) {
+		const now = new Date();
+		const date = now.toLocaleDateString("en-US", {
+			weekday: "long",
+			year: "numeric",
+			month: "long",
+			day: "numeric",
+		});
+		const time = now.toLocaleTimeString("en-US", { hour: "2-digit", minute: "2-digit" });
+		return `It's currently **${time}** on **${date}**.`;
+	}
+
+	const mathMatch = lower.match(
+		/(?:what(?:'s| is)\s+)?(\d+(?:\.\d+)?)\s*([+\-*/x×÷^])\s*(\d+(?:\.\d+)?)/
+	);
+	if (mathMatch) {
+		const a = parseFloat(mathMatch[1]!);
+		const op = mathMatch[2]!;
+		const b = parseFloat(mathMatch[3]!);
+		let result: number;
+		switch (op) {
+			case "+":
+				result = a + b;
+				break;
+			case "-":
+				result = a - b;
+				break;
+			case "*":
+			case "x":
+			case "×":
+				result = a * b;
+				break;
+			case "/":
+			case "÷":
+				result = b !== 0 ? a / b : NaN;
+				break;
+			case "^":
+				result = Math.pow(a, b);
+				break;
+			default:
+				result = NaN;
+		}
+		if (isNaN(result)) {
+			return "Hmm, I couldn't calculate that. Did you try dividing by zero?";
+		}
+		const niceResult = Number.isInteger(result)
+			? result.toString()
+			: result.toFixed(4).replace(/0+$/, "").replace(/\.$/, "");
+		return `That would be **${niceResult}**! Need help with anything else?`;
+	}
+
+	if (/file|files|browse|explorer|directory|folder|open file/i.test(lower)) {
+		return (
+			"I can help you explore files! While I can't directly browse files in this demo, " +
+			"here's what you can do:\n\n" +
+			"- Use the **Explorer** panel to browse your project files\n" +
+			"- Click any file to preview its contents\n" +
+			"- Use the search feature to find specific files\n\n" +
+			"Try asking about **time** or **math** for a live demo!"
+		);
+	}
+
+	if (/git|commit|branch|status|diff|push|pull/i.test(lower)) {
+		return (
+			"Git is a powerful version control system! Here's what I know:\n\n" +
+			"- **git status** - Check your current changes\n" +
+			"- **git branch** - See or switch branches\n" +
+			"- **git log** - View commit history\n" +
+			"- **git diff** - See what changed\n\n" +
+			"Check out the **Source Control** panel for a visual Git interface! " +
+			"Or try asking me about **time** or **math**."
+		);
+	}
+
+	if (/^(help|commands|what can you|what do you|capabilities|features)/i.test(lower)) {
+		return (
+			"Here's what I can help with:\n\n" +
+			"- **Greetings** - Say hi and I'll say hi back!\n" +
+			'- **Time & Date** - Ask "what time is it?" or "what\'s today\'s date?"\n' +
+			'- **Math** - Give me an expression like "12 * 8" or "what is 100 / 4"\n' +
+			"- **Files** - Ask about file browsing and the explorer\n" +
+			"- **Git** - Ask about version control commands\n" +
+			"- **Help** - Show this message anytime!\n\n" +
+			"This is a demo assistant - try different things to see what sticks!"
+		);
+	}
+
+	const defaults = [
+		"That's an interesting question! I'm a demo assistant, so my knowledge is limited - but try asking about **time**, **math**, **files**, or **git**.",
+		"I wish I could help with that! For now I can answer questions about time, math, files, and git. Type **help** to see what I can do.",
+		"Hmm, I'm not sure about that one. But I *can* do math, tell you the time, and talk about files and git. Give it a shot!",
+	];
+	return defaults[Math.floor(Math.random() * defaults.length)]!;
 }
 
-export function register(server: RPCServer, _options: HandlerOptions): void {
-  const r: RegisterFn = (method, handler) => {
-    server.register(method, handler as (params: unknown) => Promise<unknown>);
-  };
-
-  r("chat.list", async (params) => {
-    const all = await loadMessages();
-    const limit = params.limit ?? 50;
-    const messages = all.slice(-limit);
-    return {
-      messages,
-      hasMore: all.length > limit,
-    };
-  });
-
-  r("chat.send", async (params) => {
-    const all = await loadMessages();
-
-    const userMsg: ChatMessage = {
-      id: `msg-${Date.now()}-user`,
-      role: "user",
-      content: params.content,
-      timestamp: Date.now(),
-    };
-    all.push(userMsg);
-
-    server.emitEvent("chat.message", userMsg, { role: userMsg.role });
-
-    const thinkDelay = 200 + Math.floor(Math.random() * 300);
-    await new Promise((r) => setTimeout(r, thinkDelay));
-
-    const reply: ChatMessage = {
-      id: `msg-${Date.now()}-assistant`,
-      role: "assistant",
-      content: generateReply(params.content),
-      timestamp: Date.now(),
-    };
-    all.push(reply);
-
-    server.emitEvent("chat.message", reply, { role: reply.role });
-
-    await saveMessages(all);
-
-    return { ok: true };
-  });
+export function register(server: RPCServer, options: HandlerOptions): void {
+	const storagePath = getStoragePathFor(options.platform);
+	const store = createMessageStore(storagePath);
+
+	const r: RegisterFn = (method, handler) => {
+		server.register(method, handler as (params: unknown) => Promise<unknown>);
+	};
+
+	r("chat.list", async (params) => {
+		const all = await store.load();
+		const limit = params.limit ?? 50;
+		const messages = all.slice(-limit);
+		return {
+			messages,
+			hasMore: all.length > limit,
+		};
+	});
+
+	r("chat.send", async (params) => {
+		const all = await store.load();
+
+		const userMsg: ChatMessage = {
+			id: `msg-${Date.now()}-user`,
+			role: "user",
+			content: params.content,
+			timestamp: Date.now(),
+		};
+		all.push(userMsg);
+
+		server.emitEvent("chat.message", userMsg, { role: userMsg.role });
+
+		const thinkDelay = 200 + Math.floor(Math.random() * 300);
+		await new Promise((r) => setTimeout(r, thinkDelay));
+
+		const reply: ChatMessage = {
+			id: `msg-${Date.now()}-assistant`,
+			role: "assistant",
+			content: generateReply(params.content),
+			timestamp: Date.now(),
+		};
+		all.push(reply);
+
+		server.emitEvent("chat.message", reply, { role: reply.role });
+
+		await store.save(all);
+
+		return { ok: true };
+	});
 }

package/templates/chat/src/shared/handlers/debug.ts

@@ -0,0 +1,12 @@
+import type { RPCServer } from "@dyyz1993/rpc-core";
+import type { HandlerOptions } from "../rpc-schema";
+
+export function register(server: RPCServer, _options: HandlerOptions): void {
+	server.register("debug.subscriptions", async () => {
+		return {
+			subscriptions: (
+				server as unknown as { getActiveSubscriptions(): unknown }
+			).getActiveSubscriptions(),
+		};
+	});
+}

package/templates/chat/src/shared/handlers/index.ts

@@ -1,2 +1,3 @@
 export { register as system } from "./system";
 export { register as chat } from "./chat";
+export { register as debug } from "./debug";