91 lines
2.0 KiB
Go
91 lines
2.0 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"skillmgr/internal/config"
|
|
"skillmgr/internal/repo"
|
|
"skillmgr/internal/types"
|
|
)
|
|
|
|
// validateGitURL 验证 Git URL 格式
|
|
func validateGitURL(url string) error {
|
|
if url == "" {
|
|
return fmt.Errorf("URL 不能为空")
|
|
}
|
|
// 支持 https://, http://, git@, git:// 协议
|
|
validPrefixes := []string{"https://", "http://", "git@", "git://"}
|
|
for _, prefix := range validPrefixes {
|
|
if strings.HasPrefix(url, prefix) {
|
|
return nil
|
|
}
|
|
}
|
|
return fmt.Errorf("无效的 Git URL 格式，必须以 https://, http://, git@ 或 git:// 开头")
|
|
}
|
|
|
|
var addCmd = &cobra.Command{
|
|
Use: "add <url>",
|
|
Short: "添加源仓库",
|
|
Long: `添加一个 git 仓库作为 skills/commands 的源。
|
|
|
|
示例:
|
|
skillmgr add https://github.com/user/skills
|
|
skillmgr add https://github.com/user/skills --name my-skills
|
|
skillmgr add https://github.com/user/skills --branch main`,
|
|
Args: cobra.ExactArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
url := args[0]
|
|
|
|
// 验证 URL 格式
|
|
if err := validateGitURL(url); err != nil {
|
|
return err
|
|
}
|
|
|
|
name, _ := cmd.Flags().GetString("name")
|
|
branch, _ := cmd.Flags().GetString("branch")
|
|
|
|
// 如果没有指定名称，从 URL 生成
|
|
if name == "" {
|
|
name = repo.URLToPathName(url)
|
|
}
|
|
|
|
if branch == "" {
|
|
branch = "main" // 默认分支
|
|
}
|
|
|
|
// Clone 仓库
|
|
fmt.Printf("正在克隆仓库 %s...\n", url)
|
|
repoPath, err := repo.CloneOrPull(url, branch)
|
|
if err != nil {
|
|
return fmt.Errorf("克隆仓库失败: %w", err)
|
|
}
|
|
|
|
// 保存到配置
|
|
repository := types.Repository{
|
|
Name: name,
|
|
URL: url,
|
|
Branch: branch,
|
|
AddedAt: time.Now(),
|
|
}
|
|
|
|
if err := config.AddRepository(repository); err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Printf("✓ 仓库 '%s' 添加成功\n", name)
|
|
fmt.Printf(" 缓存路径: %s\n", repoPath)
|
|
|
|
return nil
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
addCmd.Flags().String("name", "", "仓库别名")
|
|
addCmd.Flags().String("branch", "main", "克隆的分支")
|
|
rootCmd.AddCommand(addCmd)
|
|
}
|