Add CCA session support: models, bridge protocol, service layer, sidebar UI, and tests

Copilot · PureWeen · Copilot · commit 3d32ac6e3be9 · 2026-02-14T14:29:52.000Z
Co-authored-by: PureWeen &lt;5375137+PureWeen@users.noreply.github.com&gt;
diff --git a/PolyPilot.Tests/BridgeMessageTests.cs b/PolyPilot.Tests/BridgeMessageTests.cs
@@ -142,6 +142,7 @@ public void ClientToServer_TypeConstants_AreCorrect()
         Assert.Equal("get_sessions", BridgeMessageTypes.GetSessions);
         Assert.Equal("get_history", BridgeMessageTypes.GetHistory);
         Assert.Equal("get_persisted_sessions", BridgeMessageTypes.GetPersistedSessions);
+        Assert.Equal("get_cca_sessions", BridgeMessageTypes.GetCcaSessions);
         Assert.Equal("send_message", BridgeMessageTypes.SendMessage);
         Assert.Equal("create_session", BridgeMessageTypes.CreateSession);
         Assert.Equal("resume_session", BridgeMessageTypes.ResumeSession);
@@ -152,6 +153,12 @@ public void ClientToServer_TypeConstants_AreCorrect()
         Assert.Equal("list_directories", BridgeMessageTypes.ListDirectories);
     }
 
+    [Fact]
+    public void CcaSessionsList_TypeConstant_IsCorrect()
+    {
+        Assert.Equal("cca_sessions", BridgeMessageTypes.CcaSessionsList);
+    }
+
     [Fact]
     public void DirectoriesList_TypeConstant_IsCorrect()
     {
@@ -436,4 +443,86 @@ public void AttentionNeededPayload_AllReasons_RoundTrip(AttentionReason reason)
 
         Assert.Equal(reason, restored!.Reason);
     }
+
+    [Fact]
+    public void CcaSessionsPayload_RoundTrip()
+    {
+        var payload = new CcaSessionsPayload
+        {
+            Sessions = new List<CcaSessionSummary>
+            {
+                new()
+                {
+                    SessionId = "cca-guid-1",
+                    Summary = "Fix login bug",
+                    StartTime = new DateTime(2025, 6, 15, 10, 0, 0, DateTimeKind.Utc),
+                    ModifiedTime = new DateTime(2025, 6, 15, 11, 0, 0, DateTimeKind.Utc),
+                    Repository = "owner/repo",
+                    Branch = "copilot/fix-123",
+                    WorkingDirectory = "/home/runner/work/repo"
+                },
+                new()
+                {
+                    SessionId = "cca-guid-2",
+                    Summary = "Add tests for API",
+                    StartTime = new DateTime(2025, 6, 15, 12, 0, 0, DateTimeKind.Utc),
+                    ModifiedTime = new DateTime(2025, 6, 15, 13, 0, 0, DateTimeKind.Utc),
+                    Repository = "owner/other-repo",
+                    Branch = "copilot/add-tests",
+                }
+            }
+        };
+        var msg = BridgeMessage.Create(BridgeMessageTypes.CcaSessionsList, payload);
+        var json = msg.Serialize();
+        var restored = BridgeMessage.Deserialize(json)!.GetPayload<CcaSessionsPayload>();
+
+        Assert.NotNull(restored);
+        Assert.Equal(2, restored!.Sessions.Count);
+        Assert.Equal("cca-guid-1", restored.Sessions[0].SessionId);
+        Assert.Equal("Fix login bug", restored.Sessions[0].Summary);
+        Assert.Equal("owner/repo", restored.Sessions[0].Repository);
+        Assert.Equal("copilot/fix-123", restored.Sessions[0].Branch);
+        Assert.Equal("/home/runner/work/repo", restored.Sessions[0].WorkingDirectory);
+        Assert.Equal("cca-guid-2", restored.Sessions[1].SessionId);
+        Assert.Equal("Add tests for API", restored.Sessions[1].Summary);
+        Assert.Null(restored.Sessions[1].WorkingDirectory);
+    }
+
+    [Fact]
+    public void CcaSessionSummary_NullOptionalFields_RoundTrip()
+    {
+        var payload = new CcaSessionsPayload
+        {
+            Sessions = new List<CcaSessionSummary>
+            {
+                new()
+                {
+                    SessionId = "cca-minimal",
+                    StartTime = DateTime.UtcNow,
+                    ModifiedTime = DateTime.UtcNow
+                }
+            }
+        };
+        var msg = BridgeMessage.Create(BridgeMessageTypes.CcaSessionsList, payload);
+        var json = msg.Serialize();
+        var restored = BridgeMessage.Deserialize(json)!.GetPayload<CcaSessionsPayload>();
+
+        Assert.Single(restored!.Sessions);
+        Assert.Equal("cca-minimal", restored.Sessions[0].SessionId);
+        Assert.Null(restored.Sessions[0].Summary);
+        Assert.Null(restored.Sessions[0].Repository);
+        Assert.Null(restored.Sessions[0].Branch);
+        Assert.Null(restored.Sessions[0].WorkingDirectory);
+    }
+
+    [Fact]
+    public void CcaSessionsPayload_EmptyList_RoundTrip()
+    {
+        var payload = new CcaSessionsPayload { Sessions = new List<CcaSessionSummary>() };
+        var msg = BridgeMessage.Create(BridgeMessageTypes.CcaSessionsList, payload);
+        var restored = BridgeMessage.Deserialize(msg.Serialize())!.GetPayload<CcaSessionsPayload>();
+
+        Assert.NotNull(restored);
+        Assert.Empty(restored!.Sessions);
+    }
 }
diff --git a/PolyPilot/Components/Layout/SessionSidebar.razor b/PolyPilot/Components/Layout/SessionSidebar.razor
@@ -353,6 +353,66 @@ else
                     }
                 }
             }
+
+            @if (ccaSessions.Any())
+            {
+                <div class="sidebar-divider"></div>
+                <div class="section-header" @onclick="ToggleCcaSessions" style="cursor: pointer;">
+                    <span class="section-header-label">☁️ CCA Sessions (@ccaSessions.Count)</span>
+                    <span class="toggle-icon">@(showCcaSessions ? "▼" : "▶")</span>
+                </div>
+                
+                @if (showCcaSessions)
+                {
+                    @foreach (var cca in ccaSessions.Take(30))
+                    {
+                        var isOpen = IsSessionOpen(cca.SessionId);
+                        <div class="session-item persisted @(isOpen ? "already-open" : "") @(confirmCcaResumeId == cca.SessionId ? "confirm-active" : "")" 
+                             @onclick="async () => { if (isOpen) await SwitchToOpenSession(cca.SessionId); else ConfirmCcaResume(cca); }"
+                             title="@(isOpen ? "Click to switch to this session" : GetCcaTooltip(cca))">
+                            <div class="session-info">
+                                <span class="session-name persisted-name">@(cca.Summary ?? "CCA Session")</span>
+                                <div class="session-meta-row">
+                                    <div class="session-meta-left">
+                                        <span class="session-meta-time">@cca.ModifiedTime.ToString("MMM dd") @cca.ModifiedTime.ToShortTimeString()</span>
+                                        @if (!string.IsNullOrEmpty(cca.Repository))
+                                        {
+                                            <span class="session-meta-dir">@cca.Repository</span>
+                                        }
+                                        @if (!string.IsNullOrEmpty(cca.Branch))
+                                        {
+                                            <span class="session-meta-dir">🌿 @cca.Branch</span>
+                                        }
+                                    </div>
+                                    @if (isOpen)
+                                    {
+                                        <span class="open-badge">● Open</span>
+                                    }
+                                    else if (confirmCcaResumeId != cca.SessionId)
+                                    {
+                                        <span class="resume-icon">Connect</span>
+                                    }
+                                </div>
+                                @if (!isOpen && confirmCcaResumeId == cca.SessionId)
+                                {
+                                    <div class="resume-confirm-wrap">
+                                        <div class="resume-confirm">
+                                            <button class="btn-resume-yes" @onclick="() => ConnectToCcaSession(cca)" @onclick:stopPropagation="true" disabled="@isCreating">
+                                                @if (isCreating) { <span>Connecting<span class="dots"><span>.</span><span>.</span><span>.</span></span></span> } else { <span>▶ Connect</span> }
+                                            </button>
+                                            <button class="btn-resume-no" @onclick="CancelCcaResume" @onclick:stopPropagation="true">✕</button>
+                                        </div>
+                                        @if (!string.IsNullOrEmpty(ccaResumeError))
+                                        {
+                                            <div class="resume-error">⚠ @ccaResumeError</div>
+                                        }
+                                    </div>
+                                }
+                            </div>
+                        </div>
+                    }
+                }
+            }
         </div>
     };
 
@@ -373,6 +433,10 @@ else
     private bool showDirectoryPicker;
     private List<AgentSessionInfo> sessions = new();
     private List<PersistedSessionInfo> persistedSessions = new();
+    private List<CcaSessionSummary> ccaSessions = new();
+    private bool showCcaSessions = false;
+    private string? confirmCcaResumeId = null;
+    private string? ccaResumeError = null;
     private bool showAddRepo = false;
     private string newRepoUrl = "";
     private string? addRepoError = null;
@@ -531,6 +595,70 @@ else
         showPersistedSessions = !showPersistedSessions;
     }
 
+    private void ToggleCcaSessions()
+    {
+        showCcaSessions = !showCcaSessions;
+        if (showCcaSessions && ccaSessions.Count == 0)
+        {
+            _ = LoadCcaSessionsAsync();
+        }
+    }
+
+    private async Task LoadCcaSessionsAsync()
+    {
+        ccaSessions = await CopilotService.GetCcaSessionsAsync();
+        await InvokeAsync(StateHasChanged);
+    }
+
+    private void ConfirmCcaResume(CcaSessionSummary cca)
+    {
+        ccaResumeError = null;
+        confirmCcaResumeId = confirmCcaResumeId == cca.SessionId ? null : cca.SessionId;
+    }
+
+    private void CancelCcaResume()
+    {
+        confirmCcaResumeId = null;
+        ccaResumeError = null;
+    }
+
+    private async Task ConnectToCcaSession(CcaSessionSummary cca)
+    {
+        isCreating = true;
+        ccaResumeError = null;
+        try
+        {
+            var displayName = cca.Summary ?? "CCA Session";
+            if (displayName.Length > 30) displayName = displayName[..27] + "...";
+            var sessionInfo = await CopilotService.ResumeSessionAsync(cca.SessionId, displayName, workingDirectory: cca.WorkingDirectory);
+            CopilotService.SwitchSession(sessionInfo.Name);
+            confirmCcaResumeId = null;
+            // Refresh CCA list since the session is now active
+            _ = LoadCcaSessionsAsync();
+        }
+        catch (Exception ex)
+        {
+            ccaResumeError = $"Failed to connect: {ex.Message}";
+            Console.WriteLine($"Error connecting to CCA session: {ex.Message}");
+        }
+        finally
+        {
+            isCreating = false;
+        }
+    }
+
+    private string GetCcaTooltip(CcaSessionSummary cca)
+    {
+        var tooltip = $"CCA Session: {cca.SessionId}\n";
+        if (!string.IsNullOrEmpty(cca.Repository))
+            tooltip += $"Repository: {cca.Repository}\n";
+        if (!string.IsNullOrEmpty(cca.Branch))
+            tooltip += $"Branch: {cca.Branch}\n";
+        if (!string.IsNullOrEmpty(cca.Summary))
+            tooltip += $"\n{cca.Summary}";
+        return tooltip;
+    }
+
     private async Task HandleCreateSession((string Name, string Model, string Directory, string? WorktreeId, string? InitialPrompt) args)
     {
         if (isCreating) return;
diff --git a/PolyPilot/Models/BridgeMessages.cs b/PolyPilot/Models/BridgeMessages.cs
@@ -72,6 +72,7 @@ public static class BridgeMessageTypes
     public const string GetSessions = "get_sessions";
     public const string GetHistory = "get_history";
     public const string GetPersistedSessions = "get_persisted_sessions";
+    public const string GetCcaSessions = "get_cca_sessions";
     public const string SendMessage = "send_message";
     public const string CreateSession = "create_session";
     public const string ResumeSession = "resume_session";
@@ -84,6 +85,7 @@ public static class BridgeMessageTypes
 
     // Server → Client (response)
     public const string DirectoriesList = "directories_list";
+    public const string CcaSessionsList = "cca_sessions";
 }
 
 // --- Server → Client payloads ---
@@ -281,3 +283,21 @@ public class AttentionNeededPayload
     public AttentionReason Reason { get; set; }
     public string Summary { get; set; } = "";
 }
+
+// --- CCA (Copilot Coding Agent) session payloads ---
+
+public class CcaSessionsPayload
+{
+    public List<CcaSessionSummary> Sessions { get; set; } = new();
+}
+
+public class CcaSessionSummary
+{
+    public string SessionId { get; set; } = "";
+    public string? Summary { get; set; }
+    public DateTime StartTime { get; set; }
+    public DateTime ModifiedTime { get; set; }
+    public string? Repository { get; set; }
+    public string? Branch { get; set; }
+    public string? WorkingDirectory { get; set; }
+}
diff --git a/PolyPilot/Services/CopilotService.Persistence.cs b/PolyPilot/Services/CopilotService.Persistence.cs
@@ -218,6 +218,46 @@ public IEnumerable<PersistedSessionInfo> GetPersistedSessions()
             .OrderByDescending(s => s.LastModified);
     }
 
+    /// <summary>
+    /// Gets CCA (Copilot Coding Agent) sessions from the Copilot server.
+    /// These are cloud-based sessions running in GitHub Actions.
+    /// </summary>
+    public async Task<List<CcaSessionSummary>> GetCcaSessionsAsync(CancellationToken cancellationToken = default)
+    {
+        // In remote mode, return CCA sessions from the bridge
+        if (IsRemoteMode)
+        {
+            return _bridgeClient.CcaSessions;
+        }
+
+        if (!IsInitialized || _client == null)
+            return new List<CcaSessionSummary>();
+
+        try
+        {
+            var sessions = await _client.ListSessionsAsync(cancellationToken: cancellationToken);
+            return sessions
+                .Where(s => s.IsRemote)
+                .Select(s => new CcaSessionSummary
+                {
+                    SessionId = s.SessionId,
+                    Summary = s.Summary,
+                    StartTime = s.StartTime,
+                    ModifiedTime = s.ModifiedTime,
+                    Repository = s.Context?.Repository,
+                    Branch = s.Context?.Branch,
+                    WorkingDirectory = s.Context?.Cwd,
+                })
+                .OrderByDescending(s => s.ModifiedTime)
+                .ToList();
+        }
+        catch (Exception ex)
+        {
+            Debug($"Failed to list CCA sessions: {ex.Message}");
+            return new List<CcaSessionSummary>();
+        }
+    }
+
     private static bool IsResumableSessionDirectory(DirectoryInfo di)
     {
         var eventsFile = Path.Combine(di.FullName, "events.jsonl");
diff --git a/PolyPilot/Services/WsBridgeClient.cs b/PolyPilot/Services/WsBridgeClient.cs
@@ -24,6 +24,7 @@ public class WsBridgeClient : IDisposable
     public string? ActiveSessionName { get; private set; }
     public Dictionary<string, List<ChatMessage>> SessionHistories { get; } = new();
     public List<PersistedSessionSummary> PersistedSessions { get; private set; } = new();
+    public List<CcaSessionSummary> CcaSessions { get; private set; } = new();
     public string? GitHubAvatarUrl { get; private set; }
     public string? GitHubLogin { get; private set; }
 
@@ -193,6 +194,9 @@ await SendAsync(BridgeMessage.Create(BridgeMessageTypes.AbortSession,
     public async Task SendOrganizationCommandAsync(OrganizationCommandPayload cmd, CancellationToken ct = default) =>
         await SendAsync(BridgeMessage.Create(BridgeMessageTypes.OrganizationCommand, cmd), ct);
 
+    public async Task RequestCcaSessionsAsync(CancellationToken ct = default) =>
+        await SendAsync(new BridgeMessage { Type = BridgeMessageTypes.GetCcaSessions }, ct);
+
     private TaskCompletionSource<DirectoriesListPayload>? _dirListTcs;
 
     public async Task<DirectoriesListPayload> ListDirectoriesAsync(string? path = null, CancellationToken ct = default)
@@ -370,6 +374,16 @@ private void HandleServerMessage(string json)
                 }
                 break;
 
+            case BridgeMessageTypes.CcaSessionsList:
+                var ccaSessions = msg.GetPayload<CcaSessionsPayload>();
+                if (ccaSessions != null)
+                {
+                    CcaSessions = ccaSessions.Sessions;
+                    Console.WriteLine($"[WsBridgeClient] Got {CcaSessions.Count} CCA sessions");
+                    OnStateChanged?.Invoke();
+                }
+                break;
+
             case BridgeMessageTypes.ToolStarted:
                 var toolStart = msg.GetPayload<ToolStartedPayload>();
                 if (toolStart != null)
diff --git a/PolyPilot/Services/WsBridgeServer.cs b/PolyPilot/Services/WsBridgeServer.cs
