@dyyz1993/create-agent 2.0.0 → 2.0.1

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

@@ -9,171 +9,205 @@ 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;
 
 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|folder|directory|browse|explorer|open file|read file/i.test(lower)) {
-    return (
-      "To browse and manage files, use the **File Explorer** panel on the left sidebar. " +
-      "You can navigate directories, preview files, and open them for editing. " +
-      "Try typing a file path or asking me about a specific directory!"
-    );
-  }
-
-  if (/git|commit|branch|push|pull|merge|status|diff|log/i.test(lower)) {
-    return (
-      "The **Git Panel** gives you full source control right inside the app. " +
-      "You can view changes, stage files, commit, switch branches, push/pull, and see the commit log. " +
-      "Look for the source control icon in the sidebar to get started."
-    );
-  }
-
-  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 I'll explain the File Explorer\n" +
-      "- **Git** - Ask about version control and I'll walk you through it\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, do simple math, and explain the app's features. 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 explain features. 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|folder|directory|browse|explorer|open file|read file/i.test(lower)) {
+		return (
+			"To browse and manage files, use the **File Explorer** panel on the left sidebar. " +
+			"You can navigate directories, preview files, and open them for editing. " +
+			"Try typing a file path or asking me about a specific directory!"
+		);
+	}
+
+	if (/git|commit|branch|push|pull|merge|status|diff|log/i.test(lower)) {
+		return (
+			"The **Git Panel** gives you full source control right inside the app. " +
+			"You can view changes, stage files, commit, switch branches, push/pull, and see the commit log. " +
+			"Look for the source control icon in the sidebar to get started."
+		);
+	}
+
+	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 I'll explain the File Explorer\n" +
+			"- **Git** - Ask about version control and I'll walk you through it\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, do simple math, and explain the app's features. 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 explain features. 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/agent/src/shared/handlers/debug.ts

@@ -0,0 +1,8 @@
+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.getActiveSubscriptions() };
+	});
+}

package/templates/agent/src/shared/handlers/feed.ts

@@ -6,48 +6,47 @@ import { createLogger } from "../lib/logger";
 
 const log = createLogger("feed");
 
-// 内存存储
-const posts: FeedPost[] = [];
-
 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 register(server: RPCServer, _options: HandlerOptions): void {
-  const r: RegisterFn = (method, handler) => {
-    server.register(method, handler as (params: unknown) => Promise<unknown>);
-  };
+	const posts: FeedPost[] = [];
+
+	const r: RegisterFn = (method, handler) => {
+		server.register(method, handler as (params: unknown) => Promise<unknown>);
+	};
 
-  r("feed.post", async (params) => {
-    const category = params.category as FeedCategory;
-    const author = (params.author as string) || "anonymous";
+	r("feed.post", async (params) => {
+		const category = params.category as FeedCategory;
+		const author = (params.author as string) || "anonymous";
 
-    const post: FeedPost = {
-      id: `feed-${Date.now()}-${Math.random().toString(36).slice(2, 6)}`,
-      content: params.content,
-      category,
-      author,
-      timestamp: Date.now(),
-    };
+		const post: FeedPost = {
+			id: `feed-${Date.now()}-${Math.random().toString(36).slice(2, 6)}`,
+			content: params.content,
+			category,
+			author,
+			timestamp: Date.now(),
+		};
 
-    posts.push(post);
-    log.info(`New post: ${post.id} [${category}] by ${author}`);
+		posts.push(post);
+		log.info(`New post: ${post.id} [${category}] by ${author}`);
 
-    // 发送事件 — metadata 用于过滤
-    server.emitEvent("feed.update", post, { category, author });
+		// 发送事件 — metadata 用于过滤
+		server.emitEvent("feed.update", post, { category, author });
 
-    return { id: post.id };
-  });
+		return { id: post.id };
+	});
 
-  r("feed.list", async (params) => {
-    const limit = params.limit ?? 50;
-    let filtered = posts;
+	r("feed.list", async (params) => {
+		const limit = params.limit ?? 50;
+		let filtered = posts;
 
-    if (params.category) {
-      filtered = posts.filter((p) => p.category === params.category);
-    }
+		if (params.category) {
+			filtered = posts.filter((p) => p.category === params.category);
+		}
 
-    return { posts: filtered.slice(-limit) };
-  });
+		return { posts: filtered.slice(-limit) };
+	});
 }

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

@@ -8,3 +8,4 @@ export { register as feed } from "./feed";
 export { register as bash } from "./bash";
 export { register as rules } from "./rules";
 export { register as todo } from "./todo";
+export { register as debug } from "./debug";

package/templates/agent/src/shared/handlers/rules.ts

@@ -4,42 +4,42 @@ import type { RPCMethods, HandlerOptions } from "../rpc-schema";
 import type { Rule } from "../modules/rules";
 
 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;
 
-const rules: Rule[] = [];
-let ruleIdCounter = 1;
-
 export function register(server: RPCServer, _options: HandlerOptions): void {
-  const r: RegisterFn = (method, handler) => {
-    server.register(method, handler as (params: unknown) => Promise<unknown>);
-  };
+	const rules: Rule[] = [];
+	let ruleIdCounter = 1;
+
+	const r: RegisterFn = (method, handler) => {
+		server.register(method, handler as (params: unknown) => Promise<unknown>);
+	};
 
-  r("rules.list", async () => ({ rules }));
+	r("rules.list", async () => ({ rules }));
 
-  r("rules.add", async (params) => {
-    const rule: Rule = {
-      id: `rule-${ruleIdCounter++}`,
-      name: params.name,
-      pattern: params.pattern,
-      enabled: true,
-    };
-    rules.push(rule);
-    return { rule };
-  });
+	r("rules.add", async (params) => {
+		const rule: Rule = {
+			id: `rule-${ruleIdCounter++}`,
+			name: params.name,
+			pattern: params.pattern,
+			enabled: true,
+		};
+		rules.push(rule);
+		return { rule };
+	});
 
-  r("rules.toggle", async (params) => {
-    const rule = rules.find((r) => r.id === params.id);
-    if (!rule) throw new Error(`Rule ${params.id} not found`);
-    rule.enabled = !rule.enabled;
-    return { rule };
-  });
+	r("rules.toggle", async (params) => {
+		const rule = rules.find((r) => r.id === params.id);
+		if (!rule) throw new Error(`Rule ${params.id} not found`);
+		rule.enabled = !rule.enabled;
+		return { rule };
+	});
 
-  r("rules.remove", async (params) => {
-    const idx = rules.findIndex((r) => r.id === params.id);
-    if (idx === -1) return { success: false };
-    rules.splice(idx, 1);
-    return { success: true };
-  });
+	r("rules.remove", async (params) => {
+		const idx = rules.findIndex((r) => r.id === params.id);
+		if (idx === -1) return { success: false };
+		rules.splice(idx, 1);
+		return { success: true };
+	});
 }

package/templates/agent/src/shared/handlers/todo.ts

@@ -4,42 +4,42 @@ import type { RPCMethods, HandlerOptions } from "../rpc-schema";
 import type { TodoItem, TodoStatus } from "../modules/todo";
 
 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;
 
-const items: TodoItem[] = [];
-let todoIdCounter = 1;
-
 export function register(server: RPCServer, _options: HandlerOptions): void {
-  const r: RegisterFn = (method, handler) => {
-    server.register(method, handler as (params: unknown) => Promise<unknown>);
-  };
+	const items: TodoItem[] = [];
+	let todoIdCounter = 1;
+
+	const r: RegisterFn = (method, handler) => {
+		server.register(method, handler as (params: unknown) => Promise<unknown>);
+	};
 
-  r("todo.list", async () => ({ items }));
+	r("todo.list", async () => ({ items }));
 
-  r("todo.add", async (params) => {
-    const item: TodoItem = {
-      id: `todo-${todoIdCounter++}`,
-      content: params.content,
-      status: "pending",
-      createdAt: Date.now(),
-    };
-    items.push(item);
-    return { item };
-  });
+	r("todo.add", async (params) => {
+		const item: TodoItem = {
+			id: `todo-${todoIdCounter++}`,
+			content: params.content,
+			status: "pending",
+			createdAt: Date.now(),
+		};
+		items.push(item);
+		return { item };
+	});
 
-  r("todo.update", async (params) => {
-    const item = items.find((i) => i.id === params.id);
-    if (!item) throw new Error(`Todo ${params.id} not found`);
-    item.status = params.status as TodoStatus;
-    return { item };
-  });
+	r("todo.update", async (params) => {
+		const item = items.find((i) => i.id === params.id);
+		if (!item) throw new Error(`Todo ${params.id} not found`);
+		item.status = params.status as TodoStatus;
+		return { item };
+	});
 
-  r("todo.remove", async (params) => {
-    const idx = items.findIndex((i) => i.id === params.id);
-    if (idx === -1) return { success: false };
-    items.splice(idx, 1);
-    return { success: true };
-  });
+	r("todo.remove", async (params) => {
+		const idx = items.findIndex((i) => i.id === params.id);
+		if (idx === -1) return { success: false };
+		items.splice(idx, 1);
+		return { success: true };
+	});
 }