feat: 添加 Agent 会话归档和搜索功能

apps/electron/src/main/ipc.ts

@@ -5,7 +5,7 @@
  */
 
 import { ipcMain, nativeTheme, shell, dialog, BrowserWindow } from 'electron'
-import { IPC_CHANNELS, CHANNEL_IPC_CHANNELS, CHAT_IPC_CHANNELS, AGENT_IPC_CHANNELS, ENVIRONMENT_IPC_CHANNELS, PROXY_IPC_CHANNELS, GITHUB_RELEASE_IPC_CHANNELS } from '@proma/shared'
+import { IPC_CHANNELS, CHANNEL_IPC_CHANNELS, CHAT_IPC_CHANNELS, AGENT_IPC_CHANNELS, ENVIRONMENT_IPC_CHANNELS, PROXY_IPC_CHANNELS, GITHUB_RELEASE_IPC_CHANNELS, USAGE_IPC_CHANNELS } from '@proma/shared'
 import { USER_PROFILE_IPC_CHANNELS, SETTINGS_IPC_CHANNELS } from '../types'
 import type {
   RuntimeStatus,
@@ -44,6 +44,9 @@ import type {
   SystemProxyDetectResult,
   GitHubRelease,
   GitHubReleaseListOptions,
+  UsageStats,
+  ConversationUsage,
+  UsageSettings,
 } from '@proma/shared'
 import type { UserProfile, AppSettings } from '../types'
 import { getRuntimeStatus, getGitRepoStatus } from './lib/runtime-init'
@@ -109,6 +112,12 @@ import {
   listReleases as listGitHubReleases,
   getReleaseByTag,
 } from './lib/github-release-service'
+import {
+  getUsageStats,
+  getConversationUsage,
+  getUsageSettings,
+  updateUsageSettings,
+} from './lib/usage-service'
 
 /**
  * 注册 IPC 处理器
@@ -541,6 +550,17 @@ export function registerIpcHandlers(): void {
     }
   )
 
+  // 切换 Agent 会话归档状态
+  ipcMain.handle(
+    AGENT_IPC_CHANNELS.TOGGLE_ARCHIVE,
+    async (_, id: string): Promise<AgentSessionMeta> => {
+      const sessions = listAgentSessions()
+      const current = sessions.find((s) => s.id === id)
+      if (!current) throw new Error(`会话不存在: ${id}`)
+      return updateAgentSessionMeta(id, { archived: !current.archived })
+    }
+  )
+
   // ===== Agent 工作区管理相关 =====
 
   // 确保默认工作区存在
@@ -860,6 +880,40 @@ export function registerIpcHandlers(): void {
     }
   )
 
+  // ===== 使用统计相关 =====
+
+  // 获取使用量统计总览
+  ipcMain.handle(
+    USAGE_IPC_CHANNELS.GET_USAGE_STATS,
+    async (_, days: number = 30): Promise<UsageStats> => {
+      return getUsageStats(days)
+    }
+  )
+
+  // 获取指定对话的使用量详情
+  ipcMain.handle(
+    USAGE_IPC_CHANNELS.GET_CONVERSATION_USAGE,
+    async (_, conversationId: string): Promise<ConversationUsage | null> => {
+      return getConversationUsage(conversationId)
+    }
+  )
+
+  // 获取使用统计设置
+  ipcMain.handle(
+    USAGE_IPC_CHANNELS.GET_USAGE_SETTINGS,
+    async (): Promise<UsageSettings> => {
+      return getUsageSettings()
+    }
+  )
+
+  // 更新使用统计设置
+  ipcMain.handle(
+    USAGE_IPC_CHANNELS.UPDATE_USAGE_SETTINGS,
+    async (_, settings: UsageSettings): Promise<UsageSettings> => {
+      return updateUsageSettings(settings)
+    }
+  )
+
   console.log('[IPC] IPC 处理器注册完成')
 
   // 注册更新 IPC 处理器

apps/electron/src/main/lib/agent-session-manager.ts

@@ -159,7 +159,7 @@ export function appendAgentMessage(id: string, message: AgentMessage): void {
  */
 export function updateAgentSessionMeta(
   id: string,
-  updates: Partial<Pick<AgentSessionMeta, 'title' | 'channelId' | 'sdkSessionId' | 'workspaceId'>>,
+  updates: Partial<Pick<AgentSessionMeta, 'title' | 'channelId' | 'sdkSessionId' | 'workspaceId' | 'archived'>>,
 ): AgentSessionMeta {
   const index = readIndex()
   const idx = index.sessions.findIndex((s) => s.id === id)

apps/electron/src/preload/index.ts

@@ -6,7 +6,7 @@
  */
 
 import { contextBridge, ipcRenderer } from 'electron'
-import { IPC_CHANNELS, CHANNEL_IPC_CHANNELS, CHAT_IPC_CHANNELS, AGENT_IPC_CHANNELS, ENVIRONMENT_IPC_CHANNELS, PROXY_IPC_CHANNELS, GITHUB_RELEASE_IPC_CHANNELS } from '@proma/shared'
+import { IPC_CHANNELS, CHANNEL_IPC_CHANNELS, CHAT_IPC_CHANNELS, AGENT_IPC_CHANNELS, ENVIRONMENT_IPC_CHANNELS, PROXY_IPC_CHANNELS, GITHUB_RELEASE_IPC_CHANNELS, USAGE_IPC_CHANNELS } from '@proma/shared'
 import { USER_PROFILE_IPC_CHANNELS, SETTINGS_IPC_CHANNELS } from '../types'
 import type {
   RuntimeStatus,
@@ -50,6 +50,9 @@ import type {
   SystemProxyDetectResult,
   GitHubRelease,
   GitHubReleaseListOptions,
+  UsageStats,
+  ConversationUsage,
+  UsageSettings,
 } from '@proma/shared'
 import type { UserProfile, AppSettings } from '../types'
 
@@ -239,6 +242,9 @@ export interface ElectronAPI {
   /** 删除 Agent 会话 */
   deleteAgentSession: (id: string) => Promise<void>
 
+  /** 切换 Agent 会话归档状态 */
+  toggleArchiveAgentSession: (id: string) => Promise<AgentSessionMeta>
+
   /** 生成 Agent 会话标题 */
   generateAgentTitle: (input: AgentGenerateTitleInput) => Promise<string | null>
 
@@ -361,6 +367,20 @@ export interface ElectronAPI {
   // 工作区文件变化通知
   onCapabilitiesChanged: (callback: () => void) => () => void
   onWorkspaceFilesChanged: (callback: () => void) => () => void
+
+  // ===== 使用统计相关 =====
+
+  /** 获取使用量统计总览 */
+  getUsageStats: (days?: number) => Promise<UsageStats>
+
+  /** 获取指定对话的使用量详情 */
+  getConversationUsage: (conversationId: string) => Promise<ConversationUsage | null>
+
+  /** 获取使用统计设置 */
+  getUsageSettings: () => Promise<UsageSettings>
+
+  /** 更新使用统计设置 */
+  updateUsageSettings: (settings: UsageSettings) => Promise<UsageSettings>
 }
 
 /**
@@ -594,6 +614,10 @@ const electronAPI: ElectronAPI = {
     return ipcRenderer.invoke(AGENT_IPC_CHANNELS.DELETE_SESSION, id)
   },
 
+  toggleArchiveAgentSession: (id: string) => {
+    return ipcRenderer.invoke(AGENT_IPC_CHANNELS.TOGGLE_ARCHIVE, id)
+  },
+
   generateAgentTitle: (input: AgentGenerateTitleInput) => {
     return ipcRenderer.invoke(AGENT_IPC_CHANNELS.GENERATE_TITLE, input)
   },
@@ -754,6 +778,23 @@ const electronAPI: ElectronAPI = {
   getReleaseByTag: (tag) => {
     return ipcRenderer.invoke(GITHUB_RELEASE_IPC_CHANNELS.GET_RELEASE_BY_TAG, tag)
   },
+
+  // 使用统计
+  getUsageStats: (days?: number) => {
+    return ipcRenderer.invoke(USAGE_IPC_CHANNELS.GET_USAGE_STATS, days)
+  },
+
+  getConversationUsage: (conversationId: string) => {
+    return ipcRenderer.invoke(USAGE_IPC_CHANNELS.GET_CONVERSATION_USAGE, conversationId)
+  },
+
+  getUsageSettings: () => {
+    return ipcRenderer.invoke(USAGE_IPC_CHANNELS.GET_USAGE_SETTINGS)
+  },
+
+  updateUsageSettings: (settings: UsageSettings) => {
+    return ipcRenderer.invoke(USAGE_IPC_CHANNELS.UPDATE_USAGE_SETTINGS, settings)
+  },
 }
 
 // 将 API 暴露到渲染进程的 window 对象上

apps/electron/src/renderer/atoms/agent-atoms.ts

@@ -444,6 +444,55 @@ export const currentAgentSessionDraftAtom = atom(
   }
 )
 
+// ===== 会话归档与搜索 =====
+
+/** 归档视图切换 */
+export const showArchivedSessionsAtom = atom<boolean>(false)
+
+/** 会话搜索关键词 */
+export const sessionSearchKeywordAtom = atom<string>('')
+
+/** 获取活跃会话（按工作区过滤 + 排除归档） */
+export const activeAgentSessionsAtom = atom((get) => {
+  const sessions = get(agentSessionsAtom)
+  const currentWorkspaceId = get(currentAgentWorkspaceIdAtom)
+  return sessions.filter(
+    (s) => s.workspaceId === currentWorkspaceId && !s.archived,
+  )
+})
+
+/** 获取归档会话 */
+export const archivedAgentSessionsAtom = atom((get) => {
+  const sessions = get(agentSessionsAtom)
+  const currentWorkspaceId = get(currentAgentWorkspaceIdAtom)
+  return sessions.filter(
+    (s) => s.workspaceId === currentWorkspaceId && s.archived,
+  )
+})
+
+/** 过滤后的会话列表（结合工作区、归档状态、搜索关键词） */
+export const filteredAgentSessionsAtom = atom((get) => {
+  const sessions = get(agentSessionsAtom)
+  const currentWorkspaceId = get(currentAgentWorkspaceIdAtom)
+  const showArchived = get(showArchivedSessionsAtom)
+  const keyword = get(sessionSearchKeywordAtom)
+
+  return sessions.filter((s) => {
+    // 工作区筛选
+    if (s.workspaceId !== currentWorkspaceId) return false
+
+    // 归档状态筛选
+    if (showArchived !== !!s.archived) return false
+
+    // 关键词搜索
+    if (keyword.trim()) {
+      return s.title.toLowerCase().includes(keyword.toLowerCase())
+    }
+
+    return true
+  })
+})
+
 // ===== 后台任务管理 =====
 
 /**