Add ChatService

Author: intitni

Core/Package.swift

@@ -86,6 +86,7 @@ let package = Package(
                 "SuggestionWidget",
                 "AXExtension",
                 "Logger",
+                "ChatService",
                 .product(name: "AsyncAlgorithms", package: "swift-async-algorithms"),
             ]
         ),
@@ -149,5 +150,6 @@ let package = Package(
         .target(name: "Preferences"),
         .target(name: "ChatPlugins", dependencies: ["OpenAIService", "Environment"]),
         .target(name: "Terminal"),
+        .target(name: "ChatService", dependencies: ["OpenAIService", "ChatPlugins", "Environment"]),
     ]
 )

Core/Sources/ChatService/ChatService.swift

@@ -0,0 +1,74 @@
+import ChatPlugins
+import Foundation
+import OpenAIService
+
+public final class ChatService: ObservableObject {
+    let chatGPTService: ChatGPTServiceType
+    let plugins = registerPlugins(
+        TerminalChatPlugin.self
+    )
+    var runningPlugin: ChatPlugin?
+
+    public init(chatGPTService: ChatGPTServiceType) {
+        self.chatGPTService = chatGPTService
+    }
+    
+    deinit {
+        print("deinit")
+    }
+
+    public func send(content: String) async throws {
+        // look for the prefix of content, see if there is something like /command.
+        // If there is, then we need to find the plugin that can handle this command.
+        // If there is no such plugin, then we just send the message to the GPT service.
+        let regex = try NSRegularExpression(pattern: #"^\/([a-zA-Z0-9]+)"#)
+        let matches = regex.matches(in: content, range: NSRange(content.startIndex..., in: content))
+        if let match = matches.first {
+            let command = String(content[Range(match.range(at: 1), in: content)!])
+            if let pluginType = plugins[command] {
+                let plugin = pluginType.init(inside: chatGPTService, delegate: self)
+                await plugin.send(content: String(content.dropFirst(command.count + 1)))
+            }
+        } else {
+            _ = try await chatGPTService.send(content: content, summary: nil)
+        }
+    }
+
+    public func stopReceivingMessage() async {
+        if let runningPlugin {
+            await runningPlugin.cancel()
+        }
+        await chatGPTService.stopReceivingMessage()
+    }
+
+    public func clearHistory() async {
+        if let runningPlugin {
+            await runningPlugin.cancel()
+        }
+        await chatGPTService.clearHistory()
+    }
+}
+
+extension ChatService: ChatPluginDelegate {
+    public func pluginDidStart(_ plugin: ChatPlugin) {
+        Task {
+            await chatGPTService.markReceivingMessage(true)
+        }
+        runningPlugin = plugin
+    }
+    
+    public func pluginDidEnd(_ plugin: ChatPlugin) {
+        Task {
+            await chatGPTService.markReceivingMessage(false)
+        }
+        runningPlugin = nil
+    }
+}
+
+func registerPlugins(_ plugins: ChatPlugin.Type...) -> [String: ChatPlugin.Type] {
+    var all = [String: ChatPlugin.Type]()
+    for plugin in plugins {
+        all[plugin.command] = plugin
+    }
+    return all
+}

Core/Sources/OpenAIService/ChatGPTService.swift

@@ -7,6 +7,7 @@ public protocol ChatGPTServiceType {
     func clearHistory() async
     func mutateSystemPrompt(_ newPrompt: String) async
     func mutateHistory(_ mutate: (inout [ChatMessage]) -> Void) async
+    func markReceivingMessage(_ receiving: Bool) async
 }
 
 public enum ChatGPTServiceError: Error, LocalizedError {
@@ -166,6 +167,7 @@ public actor ChatGPTService: ChatGPTServiceType, ObservableObject {
     }
 
     public func clearHistory() {
+        stopReceivingMessage()
         history = []
     }
 
@@ -176,13 +178,17 @@ public actor ChatGPTService: ChatGPTServiceType, ObservableObject {
     public func mutateHistory(_ mutate: (inout [ChatMessage]) -> Void) async {
         mutate(&history)
     }
+
+    public func markReceivingMessage(_ receiving: Bool) {
+        isReceivingMessage = receiving
+    }
 }
 
 extension ChatGPTService {
     func changeBuildCompletionStreamAPI(_ builder: @escaping CompletionStreamAPIBuilder) {
         buildCompletionStreamAPI = builder
     }
-    
+
     func changeUUIDGenerator(_ generator: @escaping () -> String) {
         uuidGenerator = generator
     }