fix(codeagent): restore worktree when recovering workspace (#131)

Author: wwcchh0123

internal/workspace/manager.go

@@ -215,6 +215,10 @@ func (m *Manager) recoverExistingWorkspaces() {
 					m.mutex.Lock()
 					if m.repoManagers[orgRepoPath] == nil {
 						repoManager := NewRepoManager(repoPath, remoteURL)
+						// 恢复 worktrees
+						if err := repoManager.RestoreWorktrees(); err != nil {
+							log.Warnf("Failed to restore worktrees for %s: %v", orgRepoPath, err)
+						}
 						m.repoManagers[orgRepoPath] = repoManager
 						log.Infof("Created repo manager for %s", orgRepoPath)
 					}

internal/workspace/repo_manager.go

@@ -10,6 +10,8 @@ import (
 	"sync"
 	"time"
 
+	"strconv"
+
 	"github.com/qiniu/x/log"
 )
 
@@ -441,6 +443,7 @@ func (r *RepoManager) CreateWorktreeWithName(worktreeName string, branch string,
 	return worktree, nil
 }
 
+// 注册单个 worktree 到内存
 func (r *RepoManager) RegisterWorktree(prNumber int, worktree *WorktreeInfo) {
 	r.mutex.Lock()
 	defer r.mutex.Unlock()
@@ -550,3 +553,24 @@ func (r *RepoManager) EnsureMainRepositoryUpToDate() error {
 	}
 	return r.updateMainRepository()
 }
+
+// RestoreWorktrees 扫描磁盘上的 worktree 并注册到内存
+func (r *RepoManager) RestoreWorktrees() error {
+	worktrees, err := r.ListWorktrees()
+	if err != nil {
+		return err
+	}
+	for _, wt := range worktrees {
+		// 只处理含 -pr- 的 worktree 目录
+		base := filepath.Base(wt.Worktree)
+		if strings.Contains(base, "-pr-") {
+			parts := strings.Split(base, "-pr-")
+			numberParts := strings.Split(parts[1], "-")
+			prNumber, err := strconv.Atoi(numberParts[0])
+			if err == nil {
+				r.RegisterWorktree(prNumber, wt)
+			}
+		}
+	}
+	return nil
+}