MyRustTools 是一个基于 Tauri + React + TypeScript + Rust 的桌面应用程序，采用模块化架构设计，支持 YOLO 目标检测、爬虫管理、RPA 自动化等功能扩展。
- 前端: React 18 + TypeScript + Vite + Tailwind CSS + Zustand
- 后端: Rust + Tauri 2.x + Serde
- UI组件: Lucide React 图标库
- 状态管理: Zustand (前端全局状态)
- 构建工具: Vite (前端) + Cargo (后端)
- 包管理: npm/pnpm (前端) + Cargo (后端)
core/
├── api/ # 全局 API 调用层
│ ├── index.ts # API 统一导出
│ ├── types.ts # 共享类型定义
│ ├── common.ts # 通用功能 (文件对话框、版本检查等)
│ ├── project.ts # 项目管理 API
│ ├── dataset.ts # 数据集管理 API
│ ├── annotation.ts # 标注功能 API
│ ├── training.ts # 训练功能 API
│ ├── model.ts # 模型管理 API
│ ├── device.ts # 设备管理 API
│ ├── video.ts # 视频处理 API
│ └── settings.ts # 设置管理 API
├── components/ # 全局共享组件
│ └── layout/ # 通用布局组件
│ ├── AppShell.tsx # 主应用外壳（接收 sidebar 和 children 作为 props）
│ └── TitleBar.tsx # 自定义标题栏
├── stores/ # 全局状态管理
│ ├── routerStore.ts # 路由状态 (模块切换、页面导航)
│ ├── workspaceStore.ts # 工作区状态
│ ├── trainingStore.ts # 训练状态
│ └── settingsStore.ts # 设置状态
└── styles/ # 全局样式
├── index.css # 全局样式 + 标题栏样式
└── hub.css # Hub 页面专用样式
核心层职责:
- 提供统一的 API 调用接口
- 管理全局状态 (路由、设置、工作区)
- 定义共享的 TypeScript 类型
- 实现基础的 UI 布局组件
- 处理跨模块的通用功能
modules/
├── types.ts # 模块类型定义
├── registry.ts # 模块注册中心 (单例模式)
├── hub/ # Hub 首页模块
│ └── HubPage.tsx
└── yolo/ # YOLO 检测模块
├── manifest.ts # 模块清单 (元数据)
├── pages/ # 模块页面
│ ├── AnnotationPage.tsx # 标注页面
│ ├── TrainingPage.tsx # 训练页面
│ ├── ResultsPage.tsx # 结果页面
│ ├── VideoPage.tsx # 视频页面
│ ├── DevicePage.tsx # 设备页面
│ └── ToolsPage.tsx # 工具页面
└── components/ # 模块专用组件
├── layout/ # 模块专用布局
│ ├── ActivityBar.tsx # YOLO 活动栏
│ ├── Sidebar.tsx # YOLO 侧边栏
│ └── StatusBar.tsx # YOLO 状态栏
├── NewProjectModal.tsx # 新建项目弹窗（YOLO 专用）
├── HelpModal.tsx # 帮助弹窗（YOLO 专用）
├── TrainingPanel.tsx # 训练面板
└── ModelConvertModal.tsx # 模型转换弹窗
模块系统设计:
- 模块清单 (ModuleManifest): 定义模块的 id、名称、图标、描述、版本、排序、能力列表
- 模块注册中心 (ModuleRegistry): 单例模式，管理模块的注册、卸载、查询
- 模块能力 (CapabilityType): 'annotation' | 'training' | 'inference' | 'crawling' | 'automation' | 'taskflow'
- 当前状态: 模块注册中心已实现，但模块热插拔机制尚未完全实现 (YOLO模块直接在App.tsx中注册)
shared/
├── components/ # 共享 UI 组件
│ ├── ui/ # 基础 UI 组件
│ │ ├── Button.tsx
│ │ └── Modal.tsx
│ └── pages/ # 共享页面组件
│ └── HubPage.tsx # Hub 首页
└── stores/ # 共享状态 (如有需要)
core/
├── commands/ # Tauri 命令层
│ ├── mod.rs # 命令模块导出
│ ├── file.rs # 文件操作命令 (read/write/delete/list等)
│ ├── project.rs # 项目最近列表管理
│ ├── settings.rs # 设置加载/保存
│ └── watcher.rs # 文件监视命令
└── models/ # 数据模型
└── mod.rs # 模型导出
核心层职责:
- 定义统一的错误处理和响应格式
- 提供基础的文件系统操作
- 实现设置持久化
- 处理 Tauri 命令的基础设施
modules/
└── yolo/ # YOLO 检测/训练模块
├── mod.rs # 模块入口
├── commands/ # 模块命令
│ ├── mod.rs # 项目管理 (create/open/update_classes等)
│ ├── train.rs # 训练命令 (start/stop/pause/resume等)
│ ├── video.rs # 视频推理命令 (load/inference/extract等)
│ ├── desktop.rs # 桌面捕获命令 (capture/start/stop等)
│ ├── device.rs # 设备管理命令 (list/stats/set_default)
│ ├── env.rs # 环境检测 (Python/Rust/依赖安装)
│ └── model_conversion.rs # 模型转换 (format检测/优化/简化)
├── services/ # 模块服务 (18个)
│ ├── mod.rs # 服务导出
│ ├── trainer.rs # Python训练服务封装
│ ├── burn_trainer.rs # 纯Rust Burn训练器 ⭐
│ ├── yolo_dataset.rs # YOLO数据集加载器 ⭐
│ ├── yolo_loss.rs # YOLO损失函数 (CIoU/Focal/DFL) ⭐
│ ├── inference_engine.rs # ONNX推理引擎 (tract-onnx)
│ ├── yolo_inference_core.rs # 简化版推理核心
│ ├── video_inference.rs # 视频推理服务
│ ├── desktop_capture.rs # 桌面捕获服务 (xcap)
│ ├── async_capture.rs # 异步捕获优化
│ ├── model_converter.rs # 模型格式转换
│ ├── model_optimizer.rs # ONNX模型优化
│ ├── device.rs # 设备检测服务
│ ├── video.rs # 视频处理服务
│ ├── env.rs # 环境检测服务
│ ├── desktop_performance_test.rs # 性能测试
│ └── scrap_burn_final.rs # 修复线程安全的最终版本
└── models/
├── mod.rs # 模型导出
└── training.rs # 训练数据结构
已实现命令统计 (50+ commands):
| 类别 | 命令数 | 示例 |
|---|---|---|
| 项目管理 | 6 | project_create, project_open, update_classes |
| 标注功能 | 4 | load_annotation, save_annotation, import_dataset |
| 训练功能 | 8 | training_start/stop/pause/resume, model_list/delete |
| 环境检测 | 7 | check_python_env, check_rust_env, install_python_deps |
| 视频推理 | 8 | video_load, rust_video_inference_start/stop |
| 桌面捕获 | 8 | desktop_capture_start/stop, get_monitors |
| 设备管理 | 3 | device_list, device_stats, device_set_default |
| 模型转换 | 8 | detect_format, optimize_onnx, simplify_onnx |
| 文件操作 | 9 | read/write/delete file, list_directory, copy_file |
| 设置 | 2 | settings_load, settings_save |
| 文件监视 | 2 | start_watch, stop_watch |
模块命令设计:
- 每个命令使用
#[tauri::command]宏标记 - 返回
Result<T, String>统一错误处理 - 命令命名:
{module}_{action}(如project_create)
shared/
└── utils/
├── mod.rs
└── path.rs # 路径处理工具
- 路由状态 (routerStore): 管理模块切换、页面导航、路由参数
- 工作区状态 (workspaceStore): 管理当前项目、文件状态
- 训练状态 (trainingStore): 管理训练进程、日志、结果
- 设置状态 (settingsStore): 管理应用配置
- 使用 Zustand 的
create函数创建 store - 支持订阅模式监听状态变化
- 模块间状态解耦，避免直接依赖
- 统一导出:
core/api/index.ts导出所有 API 函数 - 类型安全: 所有 API 调用都有完整的 TypeScript 类型
- 错误处理: 统一的错误处理和用户提示
- 模块化: 按功能划分 API 文件 (project.ts, training.ts 等)
- 命令模式: 每个功能对应一个 Tauri 命令
- 序列化: 使用 Serde 进行 JSON 序列化/反序列化
- 异步处理: 支持
async fn的异步命令 - 错误传播: 统一的
Result<T, String>错误处理
doc/
├── 0-index.md # 模块清单索引
├── 进行中/ # 开发中功能文档
│ └── {序号}-{功能名}-{日期}.md
├── 已完成/ # 已完成功能
│ ├── 已完成项目/ # 已完成但不维护
│ └── 正在维护/ # 正在维护的功能
└── 归档/ # 稳定功能
- 索引驱动: 所有功能必须在
0-index.md中登记 - 状态追踪: 明确标注功能状态 (进行中/已完成/归档)
- 进度管理: 进行中功能需标注完成百分比
- 更新日志: 记录每次文档更新
约束: core/ 目录下的代码禁止直接修改
理由: 核心基础设施应保持稳定，避免模块开发时意外破坏基础功能
例外: 只有在进行重大架构重构时，且经过充分讨论后才能修改
替代方案: 通过扩展机制 (如模块系统) 实现新功能
约束: 新增/移除模块不得修改核心代码 实现方式:
- 模块通过
ModuleRegistry.register()注册 - 模块清单 (manifest.ts) 定义模块元数据
- 路由系统自动发现已注册模块 当前状态: 模块注册中心已实现，但热插拔机制尚未完全实现 (YOLO模块直接在App.tsx中注册) 目标: 实现完全的模块热插拔，支持动态加载/卸载模块
约束: 前后端接口必须遵循统一标准 前端 API:
- 函数命名:
camelCase - 返回类型:
Promise<Result<T>> - 错误处理: 统一的错误提示 后端命令:
- 命令命名:
{module}_{action}(snake_case) - 返回类型:
Result<T, String> - 序列化: 使用 Serde derive
约束: 状态变更必须通过 store 方法 禁止: 直接修改 store 状态对象 要求: 使用 store 提供的 action 方法 理由: 确保状态变更可追踪和调试
前端文件命名:
- 组件:
PascalCase.tsx - 工具函数:
camelCase.ts - 样式:
kebab-case.css - API 文件:
功能名.ts
后端文件命名:
- 模块入口:
mod.rs - 命令文件:
{功能}_commands.rs - 服务文件:
{功能}_service.rs - 模型文件:
{类型}.rs
要求: 任何新功能开发前必须创建文档
文档位置: doc/进行中/{序号}-{功能名}-{日期}.md
文档内容:
- 功能描述
- 任务分解
- 进度跟踪
- 相关文件列表
步骤:
- 在
doc/进行中/创建功能文档 - 定义模块清单 (manifest.ts)
- 实现模块页面和组件
- 实现后端命令和服务
- 在模块入口注册模块 (未来: 通过注册中心自动注册)
- 更新路由配置
- 测试模块功能
- 更新文档状态为"已完成"
触发条件: 所有代码修改后必须进行审查 审查内容:
- 遵循架构约束
- 代码质量检查
- 安全漏洞检查
- 性能优化建议 工具: 使用 code-reviewer agent
单元测试: 每个模块的核心功能必须有单元测试 集成测试: API 调用和状态管理需要集成测试 E2E 测试: 关键用户流程需要端到端测试 覆盖率: 最低 80% 测试覆盖率
已规划模块:
- crawler: 爬虫管理模块
- rpa: RPA 自动化模块
- taskflow: 任务流模块
扩展原则:
- 每个新模块遵循相同目录结构
- 复用核心基础设施
- 通过注册中心自动集成
- 保持前后端接口一致性
允许扩展:
- 前端: 新 UI 组件库、状态管理库
- 后端: 新 Rust crate、性能优化库 约束:
- 必须经过技术评估
- 不能破坏现有架构
- 需要更新相关文档
前端: 所有 API 调用必须有错误处理
后端: 所有命令返回 Result<T, String>
用户体验: 错误信息要用户友好，不能暴露内部实现
启动时间: 应用启动时间 < 3秒 内存使用: 正常使用 < 500MB 响应时间: UI 操作响应 < 100ms 监控: 定期检查性能指标
输入验证: 所有用户输入必须验证 文件操作: 路径遍历防护 敏感数据: 不在日志中记录敏感信息 依赖安全: 定期更新依赖，检查安全漏洞
- ✅ 核心基础设施 (core/) 架构设计
- ✅ 模块注册中心 (ModuleRegistry) 实现
- ✅ 全局状态管理 (Zustand stores)
- ✅ 前端 API 层设计
- ✅ 后端项目管理命令实现
- ✅ 自定义标题栏功能
- ✅ 基础 UI 布局组件
- 🔄 YOLO 模块页面实现 (AnnotationPage, TrainingPage 等)
- 🔄 后端训练/推理命令开发
- 🔄 模块热插拔机制完善
- 🔄 文档管理系统
- ❌ 后端训练服务和推理服务
- ❌ 完整的模块热插拔支持
- ❌ 爬虫/RPA模块框架
- ❌ 单元测试和集成测试
- ❌ E2E 测试套件
- Phase 1: 完善 YOLO 模块功能 (当前重点)
- Phase 2: 实现完全模块热插拔
- Phase 3: 添加新模块 (crawler, rpa)
- Phase 4: 测试和质量保证完善
MIT License