gig-gitignore工具实战开发（三）：gig add基础实现
gig-gitignore工具实战开发（三）：gig add基础实现
✨ 前言: 在上一篇我们已经设计了多源模板系统，本篇我们来进行实战。先实现gig add命令基础代码。
一、项目初始化
暂略，主要包含如下工作
- go mod init 项目初始化
- 添加cobra、viper等相关库
- 配置初始化等等
参考依赖
require (github.com/manifoldco/promptui v0.9.0github.com/nicksnyder/go-i18n/v2 v2.6.0github.com/rivo/tview v0.0.0-20240307173318-e804876934a1github.com/sergi/go-diff v1.4.0github.com/spf13/afero v1.12.0github.com/spf13/cobra v1.9.1github.com/spf13/viper v1.20.1golang.org/x/text v0.23.0gopkg.in/yaml.v3 v3.0.1
)
二、add命令实现
添加add.go
命令行中调用
cobra-cli add add
添加参数
func init() {rootCmd.AddCommand(addCmd)addCmd.Flags().BoolVar(&yesFlag, "yes", false, "自动确认所有交互，无需人工输入")addCmd.Flags().BoolVarP(&updateFlag, "update", "u", false, "强制更新github模板")
}
- yes参数可以忽略交互，适合自动化流程
- update主要针对github模板，默认不进行更新
业务逻辑处理
根据传入参数不同，分别进行处理
func(cmd *cobra.Command, args []string) {var languages []stringif len(args) == 0 {detected := utils.DetectionUtil.DetectProjectTypes(models.AppConfig.Detection.FileMap)if len(detected) > 0 {languages = promptUserForTypes(detected)}if len(languages) == 0 {fmt.Println("未检测到项目类型，请手动输入语言参数。例如：gig add python")os.Exit(1)}} else {languages = args}switch genType {case "ai":result, err := utils.AI.StreamChat("add", strings.Join(languages, ", "))if err != nil {fmt.Println("AI 生成失败:", err)os.Exit(1)}cleanedContent := cleanupAIOutput(result)updateGitignoreWithContent(cleanedContent)case "github":handleAdd(languages, "github", updateFlag)case "local":handleAdd(languages, "local", false)case "api":result, err := getGitignoreFromAPI(languages)if err != nil {fmt.Println("API 获取失败:", err)os.Exit(1)}updateGitignoreWithContent(result)default:fmt.Printf("Error: Invalid type '%s'. Please use 'local', 'github', 'ai' 或 'api'.\n", genType)}},
}
其中代码需要进行识别项目语言，我们进行如下工具封装,根据核心特征进行判断。由于项目繁多，不可能包括所有，用户可以自行进行核心特征拓展。
// 自动识别项目类型（递归遍历，仅根据配置文件匹配，特殊目录优化）
func (d Detection) DetectProjectTypes(fileMap map[string]string) []string {fmt.Println("typeMap:", fileMap)if len(fileMap) == 0 {// 配置为空时兜底return nil}langSet := make(map[string]struct{})// 特殊目录与类型的映射（可扩展）specialDirs := map[string]string{"node_modules": "node",// 可按需添加：// ".venv": "python",// "vendor": "go",// "target": "rust",}filepath.Walk(".", func(path string, info os.FileInfo, err error) error {if err != nil {return nil // 忽略错误}if info.IsDir() {base := filepath.Base(path)// 跳过隐藏目录（如.git、.idea等）if strings.HasPrefix(base, ".") && path != "." {return filepath.SkipDir}// 特殊目录：遇到直接判定类型并跳过递归if lang, ok := specialDirs[base]; ok {langSet[lang] = struct{}{}return filepath.SkipDir}return nil}name := info.Name()if lang, ok := fileMap[name]; ok {langSet[lang] = struct{}{}}return nil})var langs []stringfor lang := range langSet {langs = append(langs, lang)}return langs
}
识别出的语言有可能有误，这时候用户可以进行修正
func promptUserForTypes(detected []string) []string {if yesFlag {fmt.Printf("检测到以下项目类型：%s\n", strings.Join(detected, ", "))fmt.Println("已自动确认项目类型 (--yes)")return detected}fmt.Printf("检测到以下项目类型：%s\n", strings.Join(detected, ", "))fmt.Print("是否确认？（直接回车确认，或输入新的类型列表，如：go,python）：")input, _ := bufio.NewReader(os.Stdin).ReadString('\n')input = strings.TrimSpace(input)if input == "" {return detected}var result []stringfor _, t := range strings.Split(input, ",") {t = strings.TrimSpace(t)if t != "" {result = append(result, t)}}return result
}