完成基本功能

2026-03-09 16:46:15 +08:00
parent d8851e7698
commit f55dd8ca03
57 changed files with 16098 additions and 6 deletions
--- a/internal/app/service.go
+++ b/internal/app/service.go
@@ -0,0 +1,670 @@
+package app
+
+import (
+	"fmt"
+	"log"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"github.com/wailsapp/wails/v3/pkg/application"
+	"github.com/zhuy1228/GitPilot/config"
+	"github.com/zhuy1228/GitPilot/internal/git"
+)
+
+// AppService 应用服务，暴露给前端调用
+type AppService struct {
+	app       *application.App
+	config    *config.AppConfig
+	gitClient *git.GitClient
+}
+
+func (a *AppService) SetApplication(app *application.App) {
+	a.app = app
+}
+
+func New() *AppService {
+	cfg, err := config.LoadConfig()
+	if err != nil {
+		log.Printf("加载配置失败: %v, 使用默认配置", err)
+		cfg = &config.AppConfig{
+			Platforms: make(map[string]config.Platform),
+			Settings:  config.Settings{Concurrency: 6, NetworkCheck: true, LogLevel: "info"},
+		}
+	}
+	return &AppService{
+		config:    cfg,
+		gitClient: git.NewGitClient(),
+	}
+}
+
+// SelectDirectory 打开系统文件夹选择器，返回选中的路径
+func (s *AppService) SelectDirectory() (string, error) {
+	if s.app == nil {
+		return "", fmt.Errorf("应用未初始化")
+	}
+	path, err := s.app.Dialog.OpenFile().
+		CanChooseDirectories(true).
+		CanChooseFiles(false).
+		SetTitle("选择项目文件夹").
+		PromptForSingleSelection()
+	if err != nil {
+		return "", fmt.Errorf("打开文件夹选择器失败: %w", err)
+	}
+	return path, nil
+}
+
+// --- 平台/项目 树形结构 ---
+
+// TreeNode 前端侧边栏树节点
+type TreeNode struct {
+	Key      string     `json:"key"`
+	Label    string     `json:"label"`
+	Type     string     `json:"type"` // platform, user, project
+	Path     string     `json:"path,omitempty"`
+	Children []TreeNode `json:"children,omitempty"`
+}
+
+// GetProjectTree 获取项目树，供前端侧边栏渲染
+func (s *AppService) GetProjectTree() []TreeNode {
+	var tree []TreeNode
+	for platformName, platform := range s.config.Platforms {
+		platformNode := TreeNode{
+			Key:   platformName,
+			Label: platformName,
+			Type:  "platform",
+		}
+		for _, user := range platform.Users {
+			userNode := TreeNode{
+				Key:   platformName + "/" + user.Username,
+				Label: user.Username,
+				Type:  "user",
+			}
+			for _, proj := range user.Projects {
+				userNode.Children = append(userNode.Children, TreeNode{
+					Key:   platformName + "/" + user.Username + "/" + proj.Name,
+					Label: proj.Name,
+					Type:  "project",
+					Path:  proj.Path,
+				})
+			}
+			platformNode.Children = append(platformNode.Children, userNode)
+		}
+		tree = append(tree, platformNode)
+	}
+	return tree
+}
+
+// --- 项目管理 ---
+
+// AddProject 添加项目到指定平台/用户下
+func (s *AppService) AddProject(platform, username, name, path string) error {
+	p, ok := s.config.Platforms[platform]
+	if !ok {
+		return fmt.Errorf("平台 %s 不存在", platform)
+	}
+	for i, user := range p.Users {
+		if user.Username == username {
+			// 检查重复
+			for _, proj := range user.Projects {
+				if proj.Name == name {
+					return fmt.Errorf("项目 %s 已存在", name)
+				}
+			}
+			s.config.Platforms[platform].Users[i].Projects = append(
+				s.config.Platforms[platform].Users[i].Projects,
+				config.Project{Name: name, Path: path},
+			)
+			return config.SaveConfig(s.config)
+		}
+	}
+	return fmt.Errorf("用户 %s 不存在于平台 %s", username, platform)
+}
+
+// RemoveProject 从指定平台/用户下删除项目
+func (s *AppService) RemoveProject(platform, username, name string) error {
+	p, ok := s.config.Platforms[platform]
+	if !ok {
+		return fmt.Errorf("平台 %s 不存在", platform)
+	}
+	for i, user := range p.Users {
+		if user.Username == username {
+			projects := user.Projects
+			for j, proj := range projects {
+				if proj.Name == name {
+					s.config.Platforms[platform].Users[i].Projects = append(projects[:j], projects[j+1:]...)
+					return config.SaveConfig(s.config)
+				}
+			}
+			return fmt.Errorf("项目 %s 不存在", name)
+		}
+	}
+	return fmt.Errorf("用户 %s 不存在于平台 %s", username, platform)
+}
+
+// --- 平台管理 ---
+
+// AddPlatform 添加新平台
+func (s *AppService) AddPlatform(name, baseURL string) error {
+	if name == "" {
+		return fmt.Errorf("平台名称不能为空")
+	}
+	if _, ok := s.config.Platforms[name]; ok {
+		return fmt.Errorf("平台 %s 已存在", name)
+	}
+	s.config.Platforms[name] = config.Platform{
+		BaseURL: baseURL,
+		Users:   []config.User{},
+	}
+	return config.SaveConfig(s.config)
+}
+
+// UpdatePlatform 修改平台信息（base_url）
+func (s *AppService) UpdatePlatform(name, baseURL string) error {
+	p, ok := s.config.Platforms[name]
+	if !ok {
+		return fmt.Errorf("平台 %s 不存在", name)
+	}
+	p.BaseURL = baseURL
+	s.config.Platforms[name] = p
+	return config.SaveConfig(s.config)
+}
+
+// RemovePlatform 删除平台
+func (s *AppService) RemovePlatform(name string) error {
+	if _, ok := s.config.Platforms[name]; !ok {
+		return fmt.Errorf("平台 %s 不存在", name)
+	}
+	delete(s.config.Platforms, name)
+	return config.SaveConfig(s.config)
+}
+
+// --- 用户管理 ---
+
+// AddUser 添加用户到指定平台
+func (s *AppService) AddUser(platform, username, token string) error {
+	p, ok := s.config.Platforms[platform]
+	if !ok {
+		return fmt.Errorf("平台 %s 不存在", platform)
+	}
+	if username == "" {
+		return fmt.Errorf("用户名不能为空")
+	}
+	for _, user := range p.Users {
+		if user.Username == username {
+			return fmt.Errorf("用户 %s 已存在于平台 %s", username, platform)
+		}
+	}
+	p.Users = append(p.Users, config.User{
+		Username: username,
+		Token:    token,
+		Projects: []config.Project{},
+	})
+	s.config.Platforms[platform] = p
+	return config.SaveConfig(s.config)
+}
+
+// UpdateUser 修改用户信息（用户名、token）
+func (s *AppService) UpdateUser(platform, oldUsername, newUsername, token string) error {
+	p, ok := s.config.Platforms[platform]
+	if !ok {
+		return fmt.Errorf("平台 %s 不存在", platform)
+	}
+	if newUsername == "" {
+		return fmt.Errorf("新用户名不能为空")
+	}
+	for i, user := range p.Users {
+		if user.Username == oldUsername {
+			// 如果改了用户名，检查新名字是否冲突
+			if oldUsername != newUsername {
+				for _, u := range p.Users {
+					if u.Username == newUsername {
+						return fmt.Errorf("用户 %s 已存在于平台 %s", newUsername, platform)
+					}
+				}
+			}
+			s.config.Platforms[platform].Users[i].Username = newUsername
+			s.config.Platforms[platform].Users[i].Token = token
+			return config.SaveConfig(s.config)
+		}
+	}
+	return fmt.Errorf("用户 %s 不存在于平台 %s", oldUsername, platform)
+}
+
+// RemoveUser 从平台删除用户
+func (s *AppService) RemoveUser(platform, username string) error {
+	p, ok := s.config.Platforms[platform]
+	if !ok {
+		return fmt.Errorf("平台 %s 不存在", platform)
+	}
+	for i, user := range p.Users {
+		if user.Username == username {
+			s.config.Platforms[platform] = config.Platform{
+				BaseURL: p.BaseURL,
+				Users:   append(p.Users[:i], p.Users[i+1:]...),
+			}
+			return config.SaveConfig(s.config)
+		}
+	}
+	return fmt.Errorf("用户 %s 不存在于平台 %s", username, platform)
+}
+
+// GetUserInfo 获取用户信息
+func (s *AppService) GetUserInfo(platform, username string) (*UserInfo, error) {
+	p, ok := s.config.Platforms[platform]
+	if !ok {
+		return nil, fmt.Errorf("平台 %s 不存在", platform)
+	}
+	for _, user := range p.Users {
+		if user.Username == username {
+			return &UserInfo{
+				Username: user.Username,
+				Token:    user.Token,
+			}, nil
+		}
+	}
+	return nil, fmt.Errorf("用户 %s 不存在于平台 %s", username, platform)
+}
+
+// GetPlatformInfo 获取平台信息
+func (s *AppService) GetPlatformInfo(name string) (*PlatformInfo, error) {
+	p, ok := s.config.Platforms[name]
+	if !ok {
+		return nil, fmt.Errorf("平台 %s 不存在", name)
+	}
+	return &PlatformInfo{
+		Name:    name,
+		BaseURL: p.BaseURL,
+	}, nil
+}
+
+// --- Git 操作 ---
+
+// ProjectStatus 项目状态信息
+type ProjectStatus struct {
+	Branch       string     `json:"branch"`
+	RemoteURL    string     `json:"remoteUrl"`
+	ChangedFiles []FileInfo `json:"changedFiles"`
+}
+
+// FileInfo 文件信息
+type FileInfo struct {
+	Status     string `json:"status"`
+	StatusText string `json:"statusText"`
+	FilePath   string `json:"filePath"`
+	Staged     bool   `json:"staged"`
+}
+
+// UserInfo 用户信息
+type UserInfo struct {
+	Username string `json:"username"`
+	Token    string `json:"token"`
+}
+
+// PlatformInfo 平台信息
+type PlatformInfo struct {
+	Name    string `json:"name"`
+	BaseURL string `json:"baseUrl"`
+}
+
+// GetProjectStatus 获取项目 git 状态
+func (s *AppService) GetProjectStatus(path string) (*ProjectStatus, error) {
+	// 先校验路径是否存在
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		return nil, fmt.Errorf("项目路径不存在: %s", path)
+	}
+
+	branch, err := s.gitClient.Branch(path)
+	if err != nil {
+		return nil, fmt.Errorf("获取分支失败: %w", err)
+	}
+
+	remoteURL, _ := s.gitClient.RemoteURL(path)
+
+	return &ProjectStatus{
+		Branch:       strings.TrimSpace(branch),
+		RemoteURL:    remoteURL,
+		ChangedFiles: []FileInfo{},
+	}, nil
+}
+
+// GetProjectChangedFiles 获取项目变更文件列表（可能较慢）
+func (s *AppService) GetProjectChangedFiles(path string) ([]FileInfo, error) {
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		return nil, fmt.Errorf("项目路径不存在: %s", path)
+	}
+
+	changes, err := s.gitClient.ChangedFiles(path)
+	if err != nil {
+		return nil, fmt.Errorf("获取变更文件失败: %w", err)
+	}
+
+	var files []FileInfo
+	for _, c := range changes {
+		files = append(files, FileInfo{
+			Status:     c.Status,
+			StatusText: c.StatusText(),
+			FilePath:   c.FilePath,
+			Staged:     c.Staged,
+		})
+	}
+
+	return files, nil
+}
+
+// GetFileContent 获取文件内容
+func (s *AppService) GetFileContent(projectPath, filePath string) (string, error) {
+	fullPath := filepath.Join(projectPath, filePath)
+
+	// 检查文件大小，超过 2MB 不读取
+	info, err := os.Stat(fullPath)
+	if err != nil {
+		return "", fmt.Errorf("读取文件失败: %w", err)
+	}
+	if info.Size() > 2*1024*1024 {
+		return "", fmt.Errorf("文件过大 (%.1f MB)，不支持预览", float64(info.Size())/1024/1024)
+	}
+
+	data, err := os.ReadFile(fullPath)
+	if err != nil {
+		return "", fmt.Errorf("读取文件失败: %w", err)
+	}
+
+	// 检测二进制文件：取前 8KB 检查是否含 NUL 字节
+	checkLen := len(data)
+	if checkLen > 8192 {
+		checkLen = 8192
+	}
+	for _, b := range data[:checkLen] {
+		if b == 0 {
+			return "", fmt.Errorf("二进制文件，不支持预览")
+		}
+	}
+
+	return string(data), nil
+}
+
+// GetFileDiff 获取文件 diff
+func (s *AppService) GetFileDiff(projectPath, filePath string) (string, error) {
+	return s.gitClient.DiffFile(projectPath, filePath)
+}
+
+// StageFiles 暂存指定文件
+func (s *AppService) StageFiles(path string, files []string) error {
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		return fmt.Errorf("项目路径不存在: %s", path)
+	}
+	if len(files) == 0 {
+		return fmt.Errorf("未指定文件")
+	}
+	_, err := s.gitClient.Add(path, files...)
+	return err
+}
+
+// UnstageFiles 取消暂存指定文件
+func (s *AppService) UnstageFiles(path string, files []string) error {
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		return fmt.Errorf("项目路径不存在: %s", path)
+	}
+	if len(files) == 0 {
+		return fmt.Errorf("未指定文件")
+	}
+	_, err := s.gitClient.Reset(path, files...)
+	return err
+}
+
+// StageAll 暂存所有变更文件
+func (s *AppService) StageAll(path string) error {
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		return fmt.Errorf("项目路径不存在: %s", path)
+	}
+	_, err := s.gitClient.Add(path, ".")
+	return err
+}
+
+// UnstageAll 取消暂存所有文件
+func (s *AppService) UnstageAll(path string) error {
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		return fmt.Errorf("项目路径不存在: %s", path)
+	}
+	_, err := s.gitClient.Reset(path, ".")
+	return err
+}
+
+// CommitChanges 提交已暂存的更改
+func (s *AppService) CommitChanges(path, message string) error {
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		return fmt.Errorf("项目路径不存在: %s", path)
+	}
+	if strings.TrimSpace(message) == "" {
+		return fmt.Errorf("提交信息不能为空")
+	}
+	_, err := s.gitClient.Commit(path, message)
+	return err
+}
+
+// DiscardFiles 丢弃工作区指定文件的更改
+func (s *AppService) DiscardFiles(path string, files []string) error {
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		return fmt.Errorf("项目路径不存在: %s", path)
+	}
+	if len(files) == 0 {
+		return fmt.Errorf("未指定文件")
+	}
+
+	// 对每个文件判断：未跟踪的用 clean 删除，已跟踪的用 checkout 还原
+	changes, err := s.gitClient.ChangedFiles(path)
+	if err != nil {
+		return fmt.Errorf("获取变更状态失败: %w", err)
+	}
+	untrackedMap := make(map[string]bool)
+	for _, c := range changes {
+		if c.Status == "?" {
+			untrackedMap[c.FilePath] = true
+		}
+	}
+
+	var trackedFiles, untrackedFiles []string
+	for _, f := range files {
+		if untrackedMap[f] {
+			untrackedFiles = append(untrackedFiles, f)
+		} else {
+			trackedFiles = append(trackedFiles, f)
+		}
+	}
+	if len(trackedFiles) > 0 {
+		if _, err := s.gitClient.Restore(path, trackedFiles...); err != nil {
+			return fmt.Errorf("还原文件失败: %w", err)
+		}
+	}
+	if len(untrackedFiles) > 0 {
+		if _, err := s.gitClient.CleanFiles(path, untrackedFiles...); err != nil {
+			return fmt.Errorf("删除未跟踪文件失败: %w", err)
+		}
+	}
+	return nil
+}
+
+// GetFileDiffStaged 获取已暂存文件的 diff
+func (s *AppService) GetFileDiffStaged(projectPath, filePath string) (string, error) {
+	return s.gitClient.DiffStagedFile(projectPath, filePath)
+}
+
+// PullProject 拉取项目（当前分支）
+func (s *AppService) PullProject(path string) (string, error) {
+	branch, err := s.gitClient.Branch(path)
+	if err != nil {
+		return "", fmt.Errorf("获取当前分支失败: %w", err)
+	}
+	return s.gitClient.Run(path, "pull", "origin", strings.TrimSpace(branch))
+}
+
+// PushProject 推送项目（当前分支）
+func (s *AppService) PushProject(path string) (string, error) {
+	branch, err := s.gitClient.Branch(path)
+	if err != nil {
+		return "", fmt.Errorf("获取当前分支失败: %w", err)
+	}
+	return s.gitClient.Run(path, "push", "origin", strings.TrimSpace(branch))
+}
+
+// GetCommitDiff 获取指定提交的 diff
+func (s *AppService) GetCommitDiff(path, hash string) (string, error) {
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		return "", fmt.Errorf("项目路径不存在: %s", path)
+	}
+	return s.gitClient.CommitShow(path, hash)
+}
+
+// CommitFileInfo 提交中的文件变更信息
+type CommitFileInfo struct {
+	Status   string `json:"status"`
+	FilePath string `json:"filePath"`
+}
+
+// GetCommitFiles 获取指定提交中变更的文件列表
+func (s *AppService) GetCommitFiles(path, hash string) ([]CommitFileInfo, error) {
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		return nil, fmt.Errorf("项目路径不存在: %s", path)
+	}
+	out, err := s.gitClient.CommitFiles(path, hash)
+	if err != nil {
+		return nil, fmt.Errorf("获取提交文件列表失败: %w", err)
+	}
+	var files []CommitFileInfo
+	for _, line := range strings.Split(strings.TrimSpace(out), "\n") {
+		if strings.TrimSpace(line) == "" {
+			continue
+		}
+		parts := strings.Fields(line)
+		if len(parts) < 2 {
+			continue
+		}
+		files = append(files, CommitFileInfo{
+			Status:   parts[0],
+			FilePath: parts[len(parts)-1],
+		})
+	}
+	return files, nil
+}
+
+// GetCommitFileDiff 获取指定提交中某个文件的 diff
+func (s *AppService) GetCommitFileDiff(path, hash, filePath string) (string, error) {
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		return "", fmt.Errorf("项目路径不存在: %s", path)
+	}
+	return s.gitClient.CommitFileDiff(path, hash, filePath)
+}
+
+// FetchProject 拉取远程信息
+func (s *AppService) FetchProject(path string) (string, error) {
+	return s.gitClient.Fetch(path)
+}
+
+// CommitLog 提交记录
+type CommitLog struct {
+	Hash      string `json:"hash"`
+	ShortHash string `json:"shortHash"`
+	Author    string `json:"author"`
+	Email     string `json:"email"`
+	Timestamp int64  `json:"timestamp"`
+	Message   string `json:"message"`
+}
+
+// BranchInfo 分支信息
+type BranchInfo struct {
+	Name    string `json:"name"`
+	Current bool   `json:"current"`
+}
+
+// GetCommitLog 获取提交历史
+func (s *AppService) GetCommitLog(path string, count int) ([]CommitLog, error) {
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		return nil, fmt.Errorf("项目路径不存在: %s", path)
+	}
+	if count <= 0 {
+		count = 50
+	}
+	out, err := s.gitClient.Log(path, count)
+	if err != nil {
+		return nil, fmt.Errorf("获取提交历史失败: %w", err)
+	}
+	return parseCommitLog(out), nil
+}
+
+func parseCommitLog(output string) []CommitLog {
+	var logs []CommitLog
+	entries := strings.Split(output, "---END---")
+	for _, entry := range entries {
+		entry = strings.TrimSpace(entry)
+		if entry == "" {
+			continue
+		}
+		lines := strings.SplitN(entry, "\n", 6)
+		if len(lines) < 6 {
+			continue
+		}
+		var ts int64
+		fmt.Sscanf(lines[4], "%d", &ts)
+		logs = append(logs, CommitLog{
+			Hash:      lines[0],
+			ShortHash: lines[1],
+			Author:    lines[2],
+			Email:     lines[3],
+			Timestamp: ts,
+			Message:   lines[5],
+		})
+	}
+	return logs
+}
+
+// GetBranches 获取所有本地分支
+func (s *AppService) GetBranches(path string) ([]BranchInfo, error) {
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		return nil, fmt.Errorf("项目路径不存在: %s", path)
+	}
+	out, err := s.gitClient.BranchList(path)
+	if err != nil {
+		return nil, fmt.Errorf("获取分支列表失败: %w", err)
+	}
+	var branches []BranchInfo
+	for _, line := range strings.Split(strings.TrimSpace(out), "\n") {
+		if line == "" {
+			continue
+		}
+		parts := strings.SplitN(line, "\t", 2)
+		name := parts[0]
+		current := len(parts) > 1 && strings.TrimSpace(parts[1]) == "*"
+		branches = append(branches, BranchInfo{Name: name, Current: current})
+	}
+	return branches, nil
+}
+
+// ResetProject 版本回滚（git reset）
+// mode: "hard"(丢弃所有更改), "soft"(保留更改到暂存区), "mixed"(保留更改到工作区)
+func (s *AppService) ResetProject(path, hash, mode string) error {
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		return fmt.Errorf("项目路径不存在: %s", path)
+	}
+	if strings.TrimSpace(hash) == "" {
+		return fmt.Errorf("提交哈希不能为空")
+	}
+	allowed := map[string]bool{"hard": true, "soft": true, "mixed": true}
+	if !allowed[mode] {
+		mode = "hard"
+	}
+	_, err := s.gitClient.ResetToCommit(path, strings.TrimSpace(hash), mode)
+	return err
+}
+
+// SwitchBranch 切换分支
+func (s *AppService) SwitchBranch(path, branch string) error {
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		return fmt.Errorf("项目路径不存在: %s", path)
+	}
+	if strings.TrimSpace(branch) == "" {
+		return fmt.Errorf("分支名不能为空")
+	}
+	_, err := s.gitClient.Checkout(path, strings.TrimSpace(branch))
+	return err
+}