@dyyz1993/create-agent 2.0.0 → 2.1.0

package/templates/chat/src/mainview/hooks/use-input-history.ts

@@ -4,81 +4,90 @@ const HISTORY_KEY = "pi-input-history";
 const MAX_ITEMS = 10;
 
 function getStorageKey(sessionId: string): string {
-  return `${HISTORY_KEY}:${sessionId}`;
+	return `${HISTORY_KEY}:${sessionId}`;
 }
 
 function readHistory(sessionId: string): string[] {
-  try {
-    const raw = localStorage.getItem(getStorageKey(sessionId));
-    if (!raw) return [];
-    const parsed = JSON.parse(raw);
-    if (Array.isArray(parsed)) return parsed.slice(0, MAX_ITEMS);
-  } catch { /* ignore */ }
-  return [];
+	try {
+		const raw = localStorage.getItem(getStorageKey(sessionId));
+		if (!raw) return [];
+		const parsed = JSON.parse(raw);
+		if (Array.isArray(parsed)) return parsed.slice(0, MAX_ITEMS);
+	} catch {
+		/* ignore */
+	}
+	return [];
 }
 
 function writeHistory(sessionId: string, items: string[]) {
-  try {
-    localStorage.setItem(getStorageKey(sessionId), JSON.stringify(items.slice(0, MAX_ITEMS)));
-  } catch { /* ignore */ }
+	try {
+		localStorage.setItem(getStorageKey(sessionId), JSON.stringify(items.slice(0, MAX_ITEMS)));
+	} catch {
+		/* ignore */
+	}
 }
 
 export function useInputHistory(sessionId: string) {
-  const historyRef = useRef<string[]>(readHistory(sessionId));
-  const indexRef = useRef(-1);
-  const [, forceUpdate] = useState(0);
+	const historyRef = useRef<string[]>(readHistory(sessionId));
+	const indexRef = useRef(-1);
+	const [, forceUpdate] = useState(0);
 
-  const hasPrev = historyRef.current.length > 0 && indexRef.current < historyRef.current.length - 1;
-  const hasNext = indexRef.current > 0;
+	const hasPrev = historyRef.current.length > 0 && indexRef.current < historyRef.current.length - 1;
+	const hasNext = indexRef.current > 0;
 
-  const saveToHistory = useCallback((text: string) => {
-    const trimmed = text.trim();
-    if (!trimmed) return;
-    const h = historyRef.current;
-    const filtered = h.filter((item) => item !== trimmed);
-    const updated = [trimmed, ...filtered].slice(0, MAX_ITEMS);
-    historyRef.current = updated;
-    writeHistory(sessionId, updated);
-    indexRef.current = -1;
-    forceUpdate((n) => n + 1);
-  }, [sessionId]);
+	const saveToHistory = useCallback(
+		(text: string) => {
+			const trimmed = text.trim();
+			if (!trimmed) return;
+			const h = historyRef.current;
+			const filtered = h.filter((item) => item !== trimmed);
+			const updated = [trimmed, ...filtered].slice(0, MAX_ITEMS);
+			historyRef.current = updated;
+			writeHistory(sessionId, updated);
+			indexRef.current = -1;
+			forceUpdate((n) => n + 1);
+		},
+		[sessionId]
+	);
 
-  const navigatePrev = useCallback((): string | null => {
-    const h = historyRef.current;
-    if (h.length === 0) return null;
-    const nextIdx = Math.min(indexRef.current + 1, h.length - 1);
-    indexRef.current = nextIdx;
-    forceUpdate((n) => n + 1);
-    return h[nextIdx];
-  }, []);
+	const navigatePrev = useCallback((): string | null => {
+		const h = historyRef.current;
+		if (h.length === 0) return null;
+		const nextIdx = Math.min(indexRef.current + 1, h.length - 1);
+		indexRef.current = nextIdx;
+		forceUpdate((n) => n + 1);
+		return h[nextIdx] ?? null;
+	}, []);
 
-  const navigateNext = useCallback((): string | null => {
-    const h = historyRef.current;
-    if (h.length === 0) return null;
-    const nextIdx = indexRef.current - 1;
-    if (nextIdx < 0) {
-      indexRef.current = -1;
-      forceUpdate((n) => n + 1);
-      return "";
-    }
-    indexRef.current = nextIdx;
-    forceUpdate((n) => n + 1);
-    return h[nextIdx];
-  }, []);
+	const navigateNext = useCallback((): string | null => {
+		const h = historyRef.current;
+		if (h.length === 0) return null;
+		const nextIdx = indexRef.current - 1;
+		if (nextIdx < 0) {
+			indexRef.current = -1;
+			forceUpdate((n) => n + 1);
+			return "";
+		}
+		indexRef.current = nextIdx;
+		forceUpdate((n) => n + 1);
+		return h[nextIdx] ?? null;
+	}, []);
 
-  const clearHistory = useCallback(() => {
-    historyRef.current = [];
-    indexRef.current = -1;
-    try {
-      localStorage.removeItem(getStorageKey(sessionId));
-    } catch { /* ignore */ }
-    forceUpdate((n) => n + 1);
-  }, [sessionId]);
+	const clearHistory = useCallback(() => {
+		historyRef.current = [];
+		indexRef.current = -1;
+		try {
+			localStorage.removeItem(getStorageKey(sessionId));
+		} catch {
+			/* ignore */
+		}
+		forceUpdate((n) => n + 1);
+	}, [sessionId]);
 
-  const resetIndex = useCallback(() => {
-    indexRef.current = -1;
-    forceUpdate((n) => n + 1);
-  }, []);
+	const resetIndex = useCallback(() => {
+		indexRef.current = -1;
+		forceUpdate((n) => n + 1);
+	}, []);
 
-  return { saveToHistory, navigatePrev, navigateNext, clearHistory, resetIndex, hasPrev, hasNext };
+	return { saveToHistory, navigatePrev, navigateNext, clearHistory, resetIndex, hasPrev, hasNext };
 }

package/templates/chat/src/mainview/lib/api-client.ts

@@ -1,6 +1,13 @@
 import { createTypedClient, WebSocketTransport, IPCTransport } from "@dyyz1993/rpc-core";
-import type { TypedClient, MethodParams, MethodResult, EventPayload, EventMetadata } from "@dyyz1993/rpc-core";
+import type {
+	TypedClient,
+	MethodParams,
+	MethodResult,
+	EventPayload,
+	EventMetadata,
+} from "@dyyz1993/rpc-core";
 import type { RPCMethods, RPCEvents } from "../../shared/rpc-schema";
+import { rpcCache, CACHEABLE_METHODS } from "./rpc-cache";
 
 /**
  * Token 来源优先级：
@@ -9,166 +16,183 @@ import type { RPCMethods, RPCEvents } from "../../shared/rpc-schema";
  * 3. 默认值（开发用）
  */
 function resolveAuthToken(): string {
-  if (typeof window !== "undefined") {
-    const fromQuery = new URLSearchParams(window.location.search).get("token");
-    if (fromQuery) return fromQuery;
-    const fromStorage = localStorage.getItem("rpc-auth-token");
-    if (fromStorage) return fromStorage;
-  }
-  return "pi-agent-template-token";
+	if (typeof window !== "undefined") {
+		const fromQuery = new URLSearchParams(window.location.search).get("token");
+		if (fromQuery) return fromQuery;
+		const fromStorage = localStorage.getItem("rpc-auth-token");
+		if (fromStorage) return fromStorage;
+	}
+	return "pi-agent-template-token";
 }
 
 const AUTH_TOKEN = resolveAuthToken();
 
 class APIClientImpl {
-  private client: TypedClient<RPCMethods, RPCEvents> | null = null;
-  private initPromise: Promise<void> | null = null;
-  private _transport: "ipc" | "websocket" = "websocket";
-  private _baseUrl: string | null = null;
-  private wsTransport: WebSocketTransport | null = null;
-
-  /**
-   * 桌面端同步初始化：通过 executeJavascript 接收 + __electrobunBunBridge 发送
-   */
-  initSyncForDesktop(): void {
-    if (this.client) return;
-
-    const ipcTransport = new IPCTransport();
-    this._transport = "ipc";
-    this._baseUrl = null; // 桌面端不走 HTTP
-    this.client = createTypedClient<RPCMethods, RPCEvents>(ipcTransport);
-    this.setupElectrobunBridge(ipcTransport);
-  }
-
-  /**
-   * 异步初始化：Web 端使用（连接 WebSocket）
-   */
-  async initialize(): Promise<void> {
-    // 已连接且 transport 正常 → 直接返回
-    if (this.client && (this._transport === "ipc" || this.wsTransport?.isConnected())) {
-      return;
-    }
-
-    // client 存在但 WS 断开了 → 清理后重连
-    if (this.client && this.wsTransport && !this.wsTransport.isConnected()) {
-      this.wsTransport.close();
-      this.wsTransport = null;
-      this.client = null;
-      this.initPromise = null;
-    }
-
-    if (this.initPromise) return this.initPromise;
-
-    this.initPromise = (async () => {
-      const env = this.detectEnvironment();
-
-      if (env === "electrobun") {
-        // 不应走到这里，桌面端应通过 initSyncForDesktop() 初始化
-        this.initSyncForDesktop();
-      } else {
-        this._transport = "websocket";
-        const wsUrl = this.getWebSocketUrl();
-        this.wsTransport = new WebSocketTransport(wsUrl);
-        await this.wsTransport.connect();
-        this.client = createTypedClient<RPCMethods, RPCEvents>(this.wsTransport);
-
-        const wsUrlObj = new URL(wsUrl);
-        this._baseUrl = `http://${wsUrlObj.host}`;
-      }
-    })();
-
-    return this.initPromise;
-  }
-
-  private detectEnvironment(): "electrobun" | "browser" {
-    if (typeof window === "undefined") return "browser";
-    if (window.__electrobunBunBridge) return "electrobun";
-    return "browser";
-  }
-
-  private getWebSocketUrl(): string {
-    if (typeof window === "undefined") return `ws://localhost:3100?token=${AUTH_TOKEN}`;
-    const customUrl = (
-      new URLSearchParams(window.location.search).get("ws") ||
-      localStorage.getItem("rpc-websocket-url")
-    );
-    if (customUrl) return customUrl.includes("token=") ? customUrl : `${customUrl}?token=${AUTH_TOKEN}`;
-    return `ws://${window.location.host}/ws?token=${AUTH_TOKEN}`;
-  }
-
-  /**
-   * 桌面端 IPC 桥接：
-   * - Bun → Browser: 通过 executeJavascript 调用 window.__piAgentIPC()
-   * - Browser → Bun: 通过 __electrobunBunBridge.postMessage 发送 Electrobun 消息格式
-   */
-  private setupElectrobunBridge(ipcTransport: IPCTransport): void {
-    if (typeof window === "undefined") return;
-
-    const win = window;
-
-    // 1. 注册接收函数：Bun 通过 executeJavascript 调用此函数发送消息到 Browser
-    win.__piAgentIPC = (msg: unknown) => {
-      ipcTransport.simulateMessage(msg);
-    };
-
-    // 2. 覆写 send：将 RPC-core 消息包装成 Electrobun 消息格式，通过原生桥接发送
-    const bridge = win.__electrobunBunBridge;
-
-    if (bridge) {
-      ipcTransport.send = async (message: unknown) => {
-        // 包装成 Electrobun message packet，bun 端 defineRPC 注册了 "rpc-message" handler
-        const electrobunPacket = {
-          type: "message",
-          id: "rpc-message",
-          payload: JSON.stringify(message),
-        };
-        bridge.postMessage(JSON.stringify(electrobunPacket));
-      };
-    }
-  }
-
-  getTransport(): "ipc" | "websocket" {
-    return this._transport;
-  }
-
-  /** Web 端 HTTP 基础 URL（如 http://localhost:3100），桌面端返回 null */
-  getBaseUrl(): string | null {
-    return this._baseUrl;
-  }
-
-  /** 获取当前 auth token（用于需要直接调 HTTP 的场景） */
-  getAuthToken(): string {
-    return AUTH_TOKEN;
-  }
-
-  async call<K extends keyof RPCMethods>(
-    method: K,
-    params: MethodParams<RPCMethods, K>
-  ): Promise<MethodResult<RPCMethods, K>> {
-    await this.initialize();
-    return this.client!.call(method, params);
-  }
-
-  async subscribe<K extends keyof RPCEvents>(
-    eventType: K,
-    handler: (payload: EventPayload<RPCEvents[K]>, metadata: EventMetadata<RPCEvents[K]>) => void,
-    filter?: Record<string, unknown>
-  ): Promise<string> {
-    await this.initialize();
-    return this.client!.subscribe(eventType, handler, filter);
-  }
-
-  unsubscribe(subscriptionId: string): void {
-    this.client?.unsubscribe(subscriptionId);
-  }
-
-  isConnected(): boolean {
-    return this.client !== null;
-  }
-
-  close(): void {
-    this.client?.close();
-  }
+	private client: TypedClient<RPCMethods, RPCEvents> | null = null;
+	private initPromise: Promise<void> | null = null;
+	private _transport: "ipc" | "websocket" = "websocket";
+	private _baseUrl: string | null = null;
+	private wsTransport: WebSocketTransport | null = null;
+
+	/**
+	 * 桌面端同步初始化：通过 executeJavascript 接收 + __electrobunBunBridge 发送
+	 */
+	initSyncForDesktop(): void {
+		if (this.client) return;
+
+		const ipcTransport = new IPCTransport();
+		this._transport = "ipc";
+		this._baseUrl = null;
+		this.client = createTypedClient<RPCMethods, RPCEvents>(ipcTransport);
+		this.setupElectrobunBridge(ipcTransport);
+	}
+
+	/**
+	 * 异步初始化：Web 端使用（连接 WebSocket）
+	 */
+	async initialize(): Promise<void> {
+		// 已连接且 transport 正常 → 直接返回
+		if (this.client && (this._transport === "ipc" || this.wsTransport?.isConnected())) {
+			return;
+		}
+
+		// client 存在但 WS 断开了 → 清理后重连
+		if (this.client && this.wsTransport && !this.wsTransport.isConnected()) {
+			this.wsTransport.close();
+			this.wsTransport = null;
+			this.client = null;
+			this.initPromise = null;
+		}
+
+		if (this.initPromise) return this.initPromise;
+
+		this.initPromise = (async () => {
+			const env = this.detectEnvironment();
+
+			if (env === "electrobun") {
+				// 不应走到这里，桌面端应通过 initSyncForDesktop() 初始化
+				this.initSyncForDesktop();
+			} else {
+				this._transport = "websocket";
+				const wsUrl = this.getWebSocketUrl();
+				this.wsTransport = new WebSocketTransport({
+					url: wsUrl,
+					authToken: AUTH_TOKEN,
+					reconnect: true,
+				} as never);
+				await this.wsTransport.connect();
+				this.client = createTypedClient<RPCMethods, RPCEvents>(this.wsTransport);
+
+				const wsUrlObj = new URL(wsUrl);
+				this._baseUrl = `http://${wsUrlObj.host}`;
+			}
+		})();
+
+		return this.initPromise;
+	}
+
+	private detectEnvironment(): "electrobun" | "browser" {
+		if (typeof window === "undefined") return "browser";
+		if (window.__electrobunBunBridge) return "electrobun";
+		return "browser";
+	}
+
+	private getWebSocketUrl(): string {
+		if (typeof window === "undefined") return `ws://localhost:3100?token=${AUTH_TOKEN}`;
+		const customUrl =
+			new URLSearchParams(window.location.search).get("ws") ||
+			localStorage.getItem("rpc-websocket-url");
+		if (customUrl)
+			return customUrl.includes("token=") ? customUrl : `${customUrl}?token=${AUTH_TOKEN}`;
+		return `ws://${window.location.host}/ws?token=${AUTH_TOKEN}`;
+	}
+
+	/**
+	 * 桌面端 IPC 桥接：
+	 * - Bun → Browser: 通过 executeJavascript 调用 window.__piAgentIPC()
+	 * - Browser → Bun: 通过 __electrobunBunBridge.postMessage 发送 Electrobun 消息格式
+	 */
+	private setupElectrobunBridge(ipcTransport: IPCTransport): void {
+		if (typeof window === "undefined") return;
+
+		const win = window;
+
+		// 1. 注册接收函数：Bun 通过 executeJavascript 调用此函数发送消息到 Browser
+		win.__piAgentIPC = (msg: unknown) => {
+			ipcTransport.simulateMessage(msg);
+		};
+
+		// 2. 覆写 send：将 RPC-core 消息包装成 Electrobun 消息格式，通过原生桥接发送
+		const bridge = win.__electrobunBunBridge;
+
+		if (bridge) {
+			ipcTransport.send = async (message: unknown) => {
+				// 包装成 Electrobun message packet，bun 端 defineRPC 注册了 "rpc-message" handler
+				const electrobunPacket = {
+					type: "message",
+					id: "rpc-message",
+					payload: JSON.stringify(message),
+				};
+				bridge.postMessage(JSON.stringify(electrobunPacket));
+			};
+		}
+	}
+
+	getTransport(): "ipc" | "websocket" {
+		return this._transport;
+	}
+
+	/** Web 端 HTTP 基础 URL（如 http://localhost:3100），桌面端返回 null */
+	getBaseUrl(): string | null {
+		return this._baseUrl;
+	}
+
+	/** 获取当前 auth token（用于需要直接调 HTTP 的场景） */
+	getAuthToken(): string {
+		return AUTH_TOKEN;
+	}
+
+	async call<K extends keyof RPCMethods>(
+		method: K,
+		params: MethodParams<RPCMethods, K>
+	): Promise<MethodResult<RPCMethods, K>> {
+		const methodStr = method as string;
+		const ttl = CACHEABLE_METHODS[methodStr];
+		if (ttl !== undefined) {
+			const cached = rpcCache.get<MethodResult<RPCMethods, K>>(methodStr, params, ttl);
+			if (cached !== null) return cached;
+		}
+
+		await this.initialize();
+		const result = await this.client!.call(method, params);
+
+		if (ttl !== undefined) {
+			rpcCache.set(methodStr, params, result);
+		}
+
+		return result;
+	}
+
+	async subscribe<K extends keyof RPCEvents>(
+		eventType: K,
+		handler: (payload: EventPayload<RPCEvents[K]>, metadata: EventMetadata<RPCEvents[K]>) => void,
+		filter?: Record<string, unknown>
+	): Promise<string> {
+		await this.initialize();
+		return this.client!.subscribe(eventType, handler, filter);
+	}
+
+	unsubscribe(subscriptionId: string): void {
+		this.client?.unsubscribe(subscriptionId);
+	}
+
+	isConnected(): boolean {
+		return this.client !== null;
+	}
+
+	close(): void {
+		this.client?.close();
+	}
 }
 
 export const apiClient = new APIClientImpl();

package/templates/chat/src/mainview/lib/rpc-cache.ts

@@ -0,0 +1,84 @@
+interface CacheEntry<T> {
+	data: T;
+	timestamp: number;
+}
+
+class RpcCache {
+	private cache = new Map<string, CacheEntry<unknown>>();
+	private defaultTTL: number;
+	private pruneTimer: ReturnType<typeof setInterval> | null = null;
+
+	constructor(defaultTTL = 2000) {
+		this.defaultTTL = defaultTTL;
+	}
+
+	private key(method: string, params: unknown): string {
+		return `${method}:${JSON.stringify(params)}`;
+	}
+
+	get<T>(method: string, params: unknown, ttl?: number): T | null {
+		const k = this.key(method, params);
+		const entry = this.cache.get(k);
+		if (!entry) return null;
+
+		const effectiveTTL = ttl ?? this.defaultTTL;
+		if (Date.now() - entry.timestamp > effectiveTTL) {
+			this.cache.delete(k);
+			return null;
+		}
+
+		return entry.data as T;
+	}
+
+	set<T>(method: string, params: unknown, data: T): void {
+		const k = this.key(method, params);
+		this.cache.set(k, { data, timestamp: Date.now() });
+		if (!this.pruneTimer) this.startPruneTimer();
+	}
+
+	invalidate(method: string): void {
+		const prefix = `${method}:`;
+		for (const key of this.cache.keys()) {
+			if (key.startsWith(prefix)) {
+				this.cache.delete(key);
+			}
+		}
+	}
+
+	invalidateAll(methods: string[]): void {
+		for (const m of methods) {
+			this.invalidate(m);
+		}
+	}
+
+	clear(): void {
+		this.cache.clear();
+		if (this.pruneTimer) {
+			clearInterval(this.pruneTimer);
+			this.pruneTimer = null;
+		}
+	}
+
+	prune(): void {
+		const threshold = this.defaultTTL * 10;
+		const now = Date.now();
+		for (const [key, entry] of this.cache) {
+			if (now - entry.timestamp > threshold) {
+				this.cache.delete(key);
+			}
+		}
+		if (this.cache.size === 0 && this.pruneTimer) {
+			clearInterval(this.pruneTimer);
+			this.pruneTimer = null;
+		}
+	}
+
+	private startPruneTimer(): void {
+		if (this.pruneTimer) return;
+		this.pruneTimer = setInterval(() => this.prune(), this.defaultTTL * 5);
+	}
+}
+
+export const CACHEABLE_METHODS: Record<string, number> = {};
+
+export const rpcCache = new RpcCache(2000);

package/templates/chat/src/mainview/main.tsx

@@ -1,20 +1,24 @@
 import "./lib/i18n";
 import { StrictMode } from "react";
+import type { ReactNode } from "react";
 import { createRoot } from "react-dom/client";
 import { apiClient } from "./lib/api-client";
+import { ErrorBoundary as _EB } from "@shared/components/ErrorBoundary";
 import "./index.css";
 import App from "./App";
 
+const ErrorBoundary = _EB as unknown as React.FC<{ children: ReactNode }>;
+
 const isElectrobun = typeof window !== "undefined" && !!window.__electrobunBunBridge;
 
 if (isElectrobun) {
-  // 桌面端：同步初始化 IPC（Electrobun 要求桥接在页面加载时同步设置）
-  apiClient.initSyncForDesktop();
+	apiClient.initSyncForDesktop();
 }
-// Web 端 WS 初始化由 App.tsx 统一管理，避免竞争
 
 createRoot(document.getElementById("root")!).render(
-  <StrictMode>
-    <App />
-  </StrictMode>
+	<StrictMode>
+		<ErrorBoundary>
+			<App />
+		</ErrorBoundary>
+	</StrictMode>
 );