e1f83ae469
## 主要改进 ### 架构重构 - 将单体 provider.go 拆分为多个专门的模块文件 - 实现了清晰的职责分离和模块化设计 - 遵循 SOLID 原则，提高代码可维护性 ### 新增功能 - **验证规则系统**: 实现了完整的 provider 验证框架 - **报告生成器**: 支持多种格式的验证报告 (JSON/HTML/Markdown/Text) - **解析器优化**: 重新设计了解析流程，提高性能和可扩展性 - **错误处理**: 增强了错误处理和诊断能力 ### 修复关键 Bug - 修复 @provider(job) 注解缺失 __job 注入参数的问题 - 统一了 job 和 cronjob 模式的处理逻辑 - 确保了 provider 生成的正确性和一致性 ### 代码质量提升 - 添加了完整的测试套件 - 引入了 golangci-lint 代码质量检查 - 优化了代码格式和结构 - 增加了详细的文档和规范 ### 文件结构优化 ``` pkg/ast/provider/ ├── types.go # 类型定义 ├── parser.go # 解析器实现 ├── validator.go # 验证规则 ├── report_generator.go # 报告生成 ├── renderer.go # 渲染器 ├── comment_parser.go # 注解解析 ├── modes.go # 模式定义 ├── errors.go # 错误处理 └── validator_test.go # 测试文件 ``` ### 兼容性 - 保持向后兼容性 - 支持现有的所有 provider 模式 - 优化了 API 设计和用户体验 This completes the implementation of T025-T029 tasks following TDD principles, including validation rules implementation and critical bug fixes.
71 lines
2.0 KiB
Go
71 lines
2.0 KiB
Go
package cmd
|
||
|
||
import (
|
||
"fmt"
|
||
"os"
|
||
"os/exec"
|
||
"path/filepath"
|
||
|
||
log "github.com/sirupsen/logrus"
|
||
"github.com/spf13/cobra"
|
||
)
|
||
|
||
func CommandBuf(root *cobra.Command) {
|
||
cmd := &cobra.Command{
|
||
Use: "buf",
|
||
Short: "run buf commands",
|
||
Long: `在指定目录执行 buf generate。若本机未安装 buf，将自动 go install github.com/bufbuild/buf/cmd/buf@v1.48.0。
|
||
|
||
Flags:
|
||
- --dir 执行目录（默认 .）
|
||
- --dry-run 仅打印将要执行的命令
|
||
|
||
说明：
|
||
- 运行前会检查 buf.yaml 是否存在，如不存在会给出提示但仍尝试执行
|
||
- 成功后输出生成结果日志`,
|
||
RunE: commandBufE,
|
||
}
|
||
|
||
cmd.Flags().String("dir", ".", "Directory to run buf from")
|
||
cmd.Flags().Bool("dry-run", false, "Preview buf command without executing")
|
||
|
||
root.AddCommand(cmd)
|
||
}
|
||
|
||
func commandBufE(cmd *cobra.Command, args []string) error {
|
||
dir := cmd.Flag("dir").Value.String()
|
||
dryRun, _ := cmd.Flags().GetBool("dry-run")
|
||
if _, err := exec.LookPath("buf"); err != nil {
|
||
log.Warn("buf 命令不存在，正在安装 buf...")
|
||
log.Info("go install github.com/bufbuild/buf/cmd/buf@v1.48.0")
|
||
installCmd := exec.Command("go", "install", "github.com/bufbuild/buf/cmd/buf@v1.48.0")
|
||
if err := installCmd.Run(); err != nil {
|
||
return fmt.Errorf("安装 buf 失败: %v", err)
|
||
}
|
||
log.Info("buf 安装成功")
|
||
|
||
if _, err := exec.LookPath("buf"); err != nil {
|
||
return fmt.Errorf("buf 命令不存在，请检查 $PATH")
|
||
}
|
||
}
|
||
|
||
// preflight: ensure buf.yaml exists
|
||
if _, err := os.Stat(filepath.Join(dir, "buf.yaml")); err != nil {
|
||
log.Warnf("未找到 %s，buf generate 可能失败", filepath.Join(dir, "buf.yaml"))
|
||
}
|
||
|
||
log.Info("buf 命令已存在，正在运行 buf generate...")
|
||
log.Info("PROTOBUF GUIDE: https://buf.build/docs/best-practices/style-guide/")
|
||
if dryRun {
|
||
log.Infof("[dry-run] (cd %s && buf generate)", dir)
|
||
return nil
|
||
}
|
||
generateCmd := exec.Command("buf", "generate")
|
||
generateCmd.Dir = dir
|
||
if err := generateCmd.Run(); err != nil {
|
||
return fmt.Errorf("运行 buf generate 失败: %v", err)
|
||
}
|
||
log.Info("buf generate 运行成功")
|
||
return nil
|
||
}
|