refactor(prompt): extract file path logic and implement auto-import on first launch (#214)
- Extract prompt file path logic to dedicated prompt_files module - Refactor PromptService to use centralized path resolution - Implement auto-import for existing prompt files on first startup - Add comprehensive unit tests for auto-import functionality
This commit is contained in:
@@ -41,6 +41,7 @@ pub struct PromptRoot {
|
||||
|
||||
use crate::config::{copy_file, get_app_config_dir, get_app_config_path, write_json_file};
|
||||
use crate::error::AppError;
|
||||
use crate::prompt_files::prompt_file_path;
|
||||
use crate::provider::ProviderManager;
|
||||
|
||||
/// 应用类型
|
||||
@@ -122,8 +123,12 @@ impl MultiAppConfig {
|
||||
let config_path = get_app_config_path();
|
||||
|
||||
if !config_path.exists() {
|
||||
log::info!("配置文件不存在,创建新的多应用配置");
|
||||
return Ok(Self::default());
|
||||
log::info!("配置文件不存在,创建新的多应用配置并自动导入提示词");
|
||||
// 使用新的方法,支持自动导入提示词
|
||||
let config = Self::default_with_auto_import()?;
|
||||
// 立即保存到磁盘
|
||||
config.save()?;
|
||||
return Ok(config);
|
||||
}
|
||||
|
||||
// 尝试读取文件
|
||||
@@ -213,4 +218,253 @@ impl MultiAppConfig {
|
||||
AppType::Gemini => &mut self.mcp.gemini,
|
||||
}
|
||||
}
|
||||
|
||||
/// 创建默认配置并自动导入已存在的提示词文件
|
||||
fn default_with_auto_import() -> Result<Self, AppError> {
|
||||
log::info!("首次启动,创建默认配置并检测提示词文件");
|
||||
|
||||
let mut config = Self::default();
|
||||
|
||||
// 为每个应用尝试自动导入提示词
|
||||
Self::auto_import_prompt_if_exists(&mut config, AppType::Claude)?;
|
||||
Self::auto_import_prompt_if_exists(&mut config, AppType::Codex)?;
|
||||
Self::auto_import_prompt_if_exists(&mut config, AppType::Gemini)?;
|
||||
|
||||
Ok(config)
|
||||
}
|
||||
|
||||
/// 检查并自动导入单个应用的提示词文件
|
||||
fn auto_import_prompt_if_exists(
|
||||
config: &mut Self,
|
||||
app: AppType,
|
||||
) -> Result<(), AppError> {
|
||||
let file_path = prompt_file_path(&app)?;
|
||||
|
||||
// 检查文件是否存在
|
||||
if !file_path.exists() {
|
||||
log::debug!("提示词文件不存在,跳过自动导入: {file_path:?}");
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
// 读取文件内容
|
||||
let content = match std::fs::read_to_string(&file_path) {
|
||||
Ok(c) => c,
|
||||
Err(e) => {
|
||||
log::warn!("读取提示词文件失败: {file_path:?}, 错误: {e}");
|
||||
return Ok(()); // 失败时不中断,继续处理其他应用
|
||||
}
|
||||
};
|
||||
|
||||
// 检查内容是否为空
|
||||
if content.trim().is_empty() {
|
||||
log::debug!("提示词文件内容为空,跳过导入: {file_path:?}");
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
log::info!("发现提示词文件,自动导入: {file_path:?}");
|
||||
|
||||
// 创建提示词对象
|
||||
let timestamp = std::time::SystemTime::now()
|
||||
.duration_since(std::time::UNIX_EPOCH)
|
||||
.unwrap()
|
||||
.as_secs() as i64;
|
||||
|
||||
let id = format!("auto-imported-{timestamp}");
|
||||
let prompt = crate::prompt::Prompt {
|
||||
id: id.clone(),
|
||||
name: format!(
|
||||
"初始提示词 {}",
|
||||
chrono::Local::now().format("%Y-%m-%d %H:%M")
|
||||
),
|
||||
content,
|
||||
description: Some("首次启动时自动导入".to_string()),
|
||||
enabled: true, // 自动启用
|
||||
created_at: Some(timestamp),
|
||||
updated_at: Some(timestamp),
|
||||
};
|
||||
|
||||
// 插入到对应的应用配置中
|
||||
let prompts = match app {
|
||||
AppType::Claude => &mut config.prompts.claude.prompts,
|
||||
AppType::Codex => &mut config.prompts.codex.prompts,
|
||||
AppType::Gemini => &mut config.prompts.gemini.prompts,
|
||||
};
|
||||
|
||||
prompts.insert(id, prompt);
|
||||
|
||||
log::info!("自动导入完成: {}", app.as_str());
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use serial_test::serial;
|
||||
use std::env;
|
||||
use std::fs;
|
||||
use tempfile::TempDir;
|
||||
|
||||
struct TempHome {
|
||||
dir: TempDir,
|
||||
original_home: Option<String>,
|
||||
original_userprofile: Option<String>,
|
||||
}
|
||||
|
||||
impl TempHome {
|
||||
fn new() -> Self {
|
||||
let dir = TempDir::new().expect("failed to create temp home");
|
||||
let original_home = env::var("HOME").ok();
|
||||
let original_userprofile = env::var("USERPROFILE").ok();
|
||||
|
||||
env::set_var("HOME", dir.path());
|
||||
env::set_var("USERPROFILE", dir.path());
|
||||
|
||||
Self {
|
||||
dir,
|
||||
original_home,
|
||||
original_userprofile,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Drop for TempHome {
|
||||
fn drop(&mut self) {
|
||||
match &self.original_home {
|
||||
Some(value) => env::set_var("HOME", value),
|
||||
None => env::remove_var("HOME"),
|
||||
}
|
||||
|
||||
match &self.original_userprofile {
|
||||
Some(value) => env::set_var("USERPROFILE", value),
|
||||
None => env::remove_var("USERPROFILE"),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn write_prompt_file(app: AppType, content: &str) {
|
||||
let path = crate::prompt_files::prompt_file_path(&app).expect("prompt path");
|
||||
if let Some(parent) = path.parent() {
|
||||
fs::create_dir_all(parent).expect("create parent dir");
|
||||
}
|
||||
fs::write(path, content).expect("write prompt");
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn auto_imports_existing_prompt_when_config_missing() {
|
||||
let _home = TempHome::new();
|
||||
write_prompt_file(AppType::Claude, "# hello");
|
||||
|
||||
let config = MultiAppConfig::load().expect("load config");
|
||||
|
||||
assert_eq!(config.prompts.claude.prompts.len(), 1);
|
||||
let prompt = config
|
||||
.prompts
|
||||
.claude
|
||||
.prompts
|
||||
.values()
|
||||
.next()
|
||||
.expect("prompt exists");
|
||||
assert!(prompt.enabled);
|
||||
assert_eq!(prompt.content, "# hello");
|
||||
|
||||
let config_path = crate::config::get_app_config_path();
|
||||
assert!(
|
||||
config_path.exists(),
|
||||
"auto import should persist config to disk"
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn skips_empty_prompt_files_during_import() {
|
||||
let _home = TempHome::new();
|
||||
write_prompt_file(AppType::Claude, " \n ");
|
||||
|
||||
let config = MultiAppConfig::load().expect("load config");
|
||||
assert!(
|
||||
config.prompts.claude.prompts.is_empty(),
|
||||
"empty files must be ignored"
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn auto_import_happens_only_once() {
|
||||
let _home = TempHome::new();
|
||||
write_prompt_file(AppType::Claude, "first version");
|
||||
|
||||
let first = MultiAppConfig::load().expect("load config");
|
||||
assert_eq!(first.prompts.claude.prompts.len(), 1);
|
||||
let claude_prompt = first
|
||||
.prompts
|
||||
.claude
|
||||
.prompts
|
||||
.values()
|
||||
.next()
|
||||
.expect("prompt exists")
|
||||
.content
|
||||
.clone();
|
||||
assert_eq!(claude_prompt, "first version");
|
||||
|
||||
// 覆盖文件内容,但保留 config.json
|
||||
write_prompt_file(AppType::Claude, "second version");
|
||||
let second = MultiAppConfig::load().expect("load config again");
|
||||
|
||||
assert_eq!(second.prompts.claude.prompts.len(), 1);
|
||||
let prompt = second
|
||||
.prompts
|
||||
.claude
|
||||
.prompts
|
||||
.values()
|
||||
.next()
|
||||
.expect("prompt exists");
|
||||
assert_eq!(
|
||||
prompt.content, "first version",
|
||||
"should not re-import when config already exists"
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn auto_imports_gemini_prompt_on_first_launch() {
|
||||
let _home = TempHome::new();
|
||||
write_prompt_file(AppType::Gemini, "# Gemini Prompt\n\nTest content");
|
||||
|
||||
let config = MultiAppConfig::load().expect("load config");
|
||||
|
||||
assert_eq!(config.prompts.gemini.prompts.len(), 1);
|
||||
let prompt = config
|
||||
.prompts
|
||||
.gemini
|
||||
.prompts
|
||||
.values()
|
||||
.next()
|
||||
.expect("gemini prompt exists");
|
||||
assert!(prompt.enabled, "gemini prompt should be enabled");
|
||||
assert_eq!(prompt.content, "# Gemini Prompt\n\nTest content");
|
||||
assert_eq!(prompt.description, Some("首次启动时自动导入".to_string()));
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[serial]
|
||||
fn auto_imports_all_three_apps_prompts() {
|
||||
let _home = TempHome::new();
|
||||
write_prompt_file(AppType::Claude, "# Claude prompt");
|
||||
write_prompt_file(AppType::Codex, "# Codex prompt");
|
||||
write_prompt_file(AppType::Gemini, "# Gemini prompt");
|
||||
|
||||
let config = MultiAppConfig::load().expect("load config");
|
||||
|
||||
// 验证所有三个应用的提示词都被导入
|
||||
assert_eq!(config.prompts.claude.prompts.len(), 1);
|
||||
assert_eq!(config.prompts.codex.prompts.len(), 1);
|
||||
assert_eq!(config.prompts.gemini.prompts.len(), 1);
|
||||
|
||||
// 验证所有提示词都被启用
|
||||
assert!(config.prompts.claude.prompts.values().next().unwrap().enabled);
|
||||
assert!(config.prompts.codex.prompts.values().next().unwrap().enabled);
|
||||
assert!(config.prompts.gemini.prompts.values().next().unwrap().enabled);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user