Expose chat tabs to widget

Author: intitni

Core/Sources/Service/GUI/ChatTabFactory.swift

@@ -1,3 +1,4 @@
+import BuiltinExtension
 import ChatGPTChatTab
 import ChatService
 import ChatTab
@@ -9,20 +10,11 @@ import XcodeInspector
 
 #if canImport(ProChatTabs)
 import ProChatTabs
+#endif
 
 enum ChatTabFactory {
     static func chatTabBuilderCollection() -> [ChatTabBuilderCollection] {
-        func folderIfNeeded(
-            _ builders: [any ChatTabBuilder],
-            title: String
-        ) -> ChatTabBuilderCollection? {
-            if builders.count > 1 {
-                return .folder(title: title, kinds: builders.map(ChatTabKind.init))
-            }
-            if let first = builders.first { return .kind(ChatTabKind(first)) }
-            return nil
-        }
-
+        #if canImport(ProChatTabs)
         let collection = [
             folderIfNeeded(ChatGPTChatTab.chatBuilders(), title: ChatGPTChatTab.name),
             folderIfNeeded(
@@ -32,11 +24,34 @@ enum ChatTabFactory {
                 title: BrowserChatTab.name
             ),
             folderIfNeeded(TerminalChatTab.chatBuilders(), title: TerminalChatTab.name),
-        ].compactMap { $0 }
+        ]
+        #else
+        let collection = [
+            folderIfNeeded(ChatGPTChatTab.chatBuilders(), title: ChatGPTChatTab.name),
+        ]
+        #endif
 
-        return collection
+        return collection.compactMap { $0 } + chatTabsFromExtensions()
     }
 
+    private static func folderIfNeeded(
+        _ builders: [any ChatTabBuilder],
+        title: String
+    ) -> ChatTabBuilderCollection? {
+        if builders.count > 1 {
+            return .folder(title: title, kinds: builders.map(ChatTabKind.init))
+        }
+        if let first = builders.first { return .kind(ChatTabKind(first)) }
+        return nil
+    }
+
+    static func chatTabsFromExtensions() -> [ChatTabBuilderCollection] {
+        let extensions = BuiltinExtensionManager.shared.extensions
+        let chatBuilders = extensions.flatMap { $0.chatBuilders }
+        return chatBuilders.compactMap { folderIfNeeded($0.value, title: $0.key) }
+    }
+
+    #if canImport(ProChatTabs)
     static func externalDependenciesForBrowserChatTab() -> BrowserChatTab.ExternalDependency {
         .init(
             getEditorContent: {
@@ -109,28 +124,6 @@ enum ChatTabFactory {
             }
         )
     }
+    #endif
 }
 
-#else
-
-enum ChatTabFactory {
-    static func chatTabBuilderCollection() -> [ChatTabBuilderCollection] {
-        func folderIfNeeded(
-            _ builders: [any ChatTabBuilder],
-            title: String
-        ) -> ChatTabBuilderCollection? {
-            if builders.count > 1 {
-                return .folder(title: title, kinds: builders.map(ChatTabKind.init))
-            }
-            if let first = builders.first { return .kind(ChatTabKind(first)) }
-            return nil
-        }
-
-        return [
-            folderIfNeeded(ChatGPTChatTab.chatBuilders(), title: ChatGPTChatTab.name),
-        ].compactMap { $0 }
-    }
-}
-
-#endif
-

Tool/Sources/BuiltinExtension/BuiltinExtension.swift

@@ -1,3 +1,4 @@
+import ChatTab
 import CopilotForXcodeKit
 import Foundation
 import Preferences
@@ -6,8 +7,17 @@ public protocol BuiltinExtension: CopilotForXcodeExtensionCapability {
     /// An id that let the extension manager determine whether the extension is in use.
     var suggestionServiceId: BuiltInSuggestionFeatureProvider { get }
 
+    /// All chat builders provided by this extension.
+    var chatBuilders: [String: [ChatTabBuilder]] { get }
+
     /// It's usually called when the app is about to quit,
     /// you should clean up all the resources here.
     func terminate()
 }
 
+// MARK: - Default Implementation
+
+public extension BuiltinExtension {
+    var chatBuilders: [String: [ChatTabBuilder]] { [:] }
+}
+

Tool/Sources/BuiltinExtension/BuiltinExtensionManager.swift

@@ -5,7 +5,7 @@ import XcodeInspector
 
 public final class BuiltinExtensionManager {
     public static let shared: BuiltinExtensionManager = .init()
-    private(set) var extensions: [any BuiltinExtension] = []
+    public private(set) var extensions: [any BuiltinExtension] = []
 
     private var cancellable: Set<AnyCancellable> = []