704 lines
19 KiB
Go
704 lines
19 KiB
Go
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"`
|
||
Pushed bool `json:"pushed"`
|
||
}
|
||
|
||
// 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)
|
||
}
|
||
logs := parseCommitLog(out)
|
||
|
||
// 获取当前分支名,查询未推送的提交
|
||
branch, brErr := s.gitClient.Branch(path)
|
||
if brErr == nil {
|
||
branch = strings.TrimSpace(branch)
|
||
unpushedOut, upErr := s.gitClient.UnpushedCommits(path, branch)
|
||
unpushedSet := make(map[string]bool)
|
||
if upErr == nil {
|
||
for _, h := range strings.Split(strings.TrimSpace(unpushedOut), "\n") {
|
||
if h != "" {
|
||
unpushedSet[h] = true
|
||
}
|
||
}
|
||
}
|
||
for i := range logs {
|
||
logs[i].Pushed = !unpushedSet[logs[i].Hash]
|
||
}
|
||
}
|
||
|
||
return logs, nil
|
||
}
|
||
|
||
// RevertCommit 撤回指定提交(生成一个反向提交)
|
||
func (s *AppService) RevertCommit(path, hash string) error {
|
||
if _, err := os.Stat(path); os.IsNotExist(err) {
|
||
return fmt.Errorf("项目路径不存在: %s", path)
|
||
}
|
||
if strings.TrimSpace(hash) == "" {
|
||
return fmt.Errorf("提交哈希不能为空")
|
||
}
|
||
_, err := s.gitClient.RevertCommit(path, strings.TrimSpace(hash))
|
||
return err
|
||
}
|
||
|
||
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
|
||
}
|