fix(codeagent): fixed the issue where deleting workspaces failed due to worktree not being registered in memory (#128)

Author: wwcchh0123

internal/workspace/manager.go

@@ -511,6 +511,13 @@ func (m *Manager) MoveIssueToPR(ws *models.Workspace, prNumber int) error {
 
 	// 更新工作空间路径
 	ws.Path = newWorktreePath
+
+	// 移动之后，注册worktree到内存中
+	worktree := &WorktreeInfo{
+		Worktree: ws.Path,
+		Branch:   ws.Branch,
+	}
+	repoManager.RegisterWorktree(prNumber, worktree)
 	return nil
 }
 
@@ -558,6 +565,9 @@ func (m *Manager) CreateWorkspaceFromPR(pr *github.PullRequest) *models.Workspac
 		return nil
 	}
 
+	// 注册worktree 到内存中
+	repoManager.RegisterWorktree(pr.GetNumber(), worktree)
+
 	// 创建 session 目录
 	suffix := strings.TrimPrefix(filepath.Base(worktree.Worktree), fmt.Sprintf("%s-pr-%d-", repo, pr.GetNumber()))
 	sessionPath, err := m.CreateSessionPath(filepath.Dir(repoManager.GetRepoPath()), repo, pr.GetNumber(), suffix)

internal/workspace/repo_manager.go

@@ -421,6 +421,12 @@ func (r *RepoManager) CreateWorktreeWithName(worktreeName string, branch string,
 	return worktree, nil
 }
 
+func (r *RepoManager) RegisterWorktree(prNumber int, worktree *WorktreeInfo) {
+	r.mutex.Lock()
+	defer r.mutex.Unlock()
+	r.worktrees[prNumber] = worktree
+}
+
 // GetRepoPath 获取仓库路径
 func (r *RepoManager) GetRepoPath() string {
 	return r.repoPath