Make auto completion dynamic to registered plugins

Author: intitni

Core/Sources/SuggestionWidget/ChatProvider.swift

@@ -7,6 +7,7 @@ public final class ChatProvider: ObservableObject {
     let id = UUID()
     @Published public var history: [ChatMessage] = []
     @Published public var isReceivingMessage = false
+    public var pluginIdentifiers: [String] = []
     public var systemPrompt = ""
     public var extraSystemPrompt = ""
     public var onMessageSend: (String) -> Void
@@ -23,6 +24,7 @@ public final class ChatProvider: ObservableObject {
     public init(
         history: [ChatMessage] = [],
         isReceivingMessage: Bool = false,
+        pluginIdentifiers: [String] = [],
         onMessageSend: @escaping (String) -> Void = { _ in },
         onStop: @escaping () -> Void = {},
         onClear: @escaping () -> Void = {},
@@ -36,6 +38,7 @@ public final class ChatProvider: ObservableObject {
     ) {
         self.history = history
         self.isReceivingMessage = isReceivingMessage
+        self.pluginIdentifiers = pluginIdentifiers
         self.onMessageSend = onMessageSend
         self.onStop = onStop
         self.onClear = onClear
@@ -59,6 +62,7 @@ public final class ChatProvider: ObservableObject {
     public func triggerCustomCommand(_ command: CustomCommand) {
         onRunCustomCommand(command)
     }
+
     public func setAsExtraPrompt(id: MessageID) { onSetAsExtraPrompt(id) }
 }
 

Core/Sources/SuggestionWidget/SuggestionPanelContent/ChatPanel.swift

@@ -355,29 +355,7 @@ struct ChatPanelInputArea: View {
                     text: $typedMessage,
                     font: .systemFont(ofSize: 14),
                     onSubmit: { submitText() },
-                    completions: { text, _, range in
-                        if text.isEmpty { return [] }
-                        let availableFeatures = [
-                            "/run",
-                            "/airun",
-                            "/math",
-                            "/search",
-                            "/shortcut",
-                            "/exit",
-                            "@selection",
-                            "@file",
-                        ]
-                        return availableFeatures
-                            .filter { $0.hasPrefix(text) && $0 != text }
-                            .compactMap {
-                                guard let index = $0.index(
-                                    $0.startIndex,
-                                    offsetBy: range.location,
-                                    limitedBy: $0.endIndex
-                                ) else { return nil }
-                                return String($0[index...])
-                            }
-                    }
+                    completions: chatAutoCompletion
                 )
                 .padding(.top, 1)
                 .padding(.bottom, -1)
@@ -420,6 +398,26 @@ struct ChatPanelInputArea: View {
         chat.send(typedMessage)
         typedMessage = ""
     }
+    
+    func chatAutoCompletion(text: String, proposed: [String], range: NSRange) -> [String] {
+        if text.isEmpty { return [] }
+        let plugins = chat.pluginIdentifiers.map { "/\($0)" }
+        let availableFeatures = plugins + [
+            "/exit",
+            "@selection",
+            "@file",
+        ]
+        return availableFeatures
+            .filter { $0.hasPrefix(text) && $0 != text }
+            .compactMap {
+                guard let index = $0.index(
+                    $0.startIndex,
+                    offsetBy: range.location,
+                    limitedBy: $0.endIndex
+                ) else { return nil }
+                return String($0[index...])
+            }
+    }
 }
 
 struct ChatContextMenu: View {
@@ -700,3 +698,6 @@ struct ChatPanel_Light_Preview: PreviewProvider {
     }
 }
 
+
+
+