60 lines
1.6 KiB
Go
60 lines
1.6 KiB
Go
package cmd
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/unclesam-ly/git-air/internal/git"
|
|
"github.com/unclesam-ly/git-air/internal/ui"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
const preCommitScript = `#!/bin/sh
|
|
# git-air pre-commit hook
|
|
echo "[git-air] 正在执行 Commit 前代码自动化评审..."
|
|
git-air --staged
|
|
`
|
|
|
|
var hookCmd = &cobra.Command{
|
|
Use: "hook",
|
|
Short: "管理 Git Pre-commit 自动审查钩子",
|
|
}
|
|
|
|
var hookInstallCmd = &cobra.Command{
|
|
Use: "install",
|
|
Short: "在当前仓库一键安装 pre-commit 钩子",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
repoRoot, err := git.GetRepoRoot(cmd.Context())
|
|
if err != nil {
|
|
ui.PrintError("请在 Git 仓库根目录下执行此命令")
|
|
return
|
|
}
|
|
hookPath := filepath.Join(repoRoot, ".git", "hooks", "pre-commit")
|
|
if err := os.WriteFile(hookPath, []byte(preCommitScript), 0755); err != nil {
|
|
ui.PrintError("写入 hook 文件失败: %v", err)
|
|
return
|
|
}
|
|
ui.PrintSuccess("已成功安装 pre-commit 钩子！每次执行 git commit 时将自动触发评审。")
|
|
},
|
|
}
|
|
|
|
var hookUninstallCmd = &cobra.Command{
|
|
Use: "uninstall",
|
|
Short: "移除当前仓库的 pre-commit 钩子",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
repoRoot, err := git.GetRepoRoot(cmd.Context())
|
|
if err != nil {
|
|
ui.PrintError("请在 Git 仓库根目录下执行此命令")
|
|
return
|
|
}
|
|
hookPath := filepath.Join(repoRoot, ".git", "hooks", "pre-commit")
|
|
_ = os.Remove(hookPath)
|
|
ui.PrintSuccess("已成功卸载 pre-commit 钩子！")
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
hookCmd.AddCommand(hookInstallCmd)
|
|
hookCmd.AddCommand(hookUninstallCmd)
|
|
}
|