Add auto complete for chat plugins and scopes

Author: intitni

Core/Sources/SuggestionWidget/CustomTextEditor.swift

@@ -8,19 +8,25 @@ struct CustomTextEditor: NSViewRepresentable {
     @Binding var text: String
     let font: NSFont
     let onSubmit: () -> Void
+    var completions: (_ text: String, _ words: [String], _ range: NSRange)
+        -> [String] = { _, _, _ in
+            []
+        }
 
     func makeNSView(context: Context) -> NSScrollView {
+        context.coordinator.completions = completions
         let textView = (context.coordinator.theTextView.documentView as! NSTextView)
         textView.delegate = context.coordinator
         textView.string = text
         textView.font = font
         textView.allowsUndo = true
         textView.drawsBackground = false
-       
+
         return context.coordinator.theTextView
     }
 
     func updateNSView(_ nsView: NSScrollView, context: Context) {
+        context.coordinator.completions = completions
         let textView = (context.coordinator.theTextView.documentView as! NSTextView)
         guard textView.string != text else { return }
         textView.string = text
@@ -32,6 +38,7 @@ extension CustomTextEditor {
         var view: CustomTextEditor
         var theTextView = NSTextView.scrollableTextView()
         var affectedCharRange: NSRange?
+        var completions: (String, [String], _ range: NSRange) -> [String] = { _, _, _ in [] }
 
         init(_ view: CustomTextEditor) {
             self.view = view
@@ -43,13 +50,14 @@ extension CustomTextEditor {
             }
 
             view.text = textView.string
+            textView.complete(nil)
         }
 
         func textView(_ textView: NSTextView, doCommandBy commandSelector: Selector) -> Bool {
             if commandSelector == #selector(NSTextView.insertNewline(_:)) {
                 if let event = NSApplication.shared.currentEvent,
                    !event.modifierFlags.contains(.shift),
-                   event.keyCode == 36
+                   event.keyCode == 36 // enter
                 {
                     view.onSubmit()
                     return true
@@ -66,6 +74,16 @@ extension CustomTextEditor {
         ) -> Bool {
             return true
         }
+
+        func textView(
+            _ textView: NSTextView,
+            completions words: [String],
+            forPartialWordRange charRange: NSRange,
+            indexOfSelectedItem index: UnsafeMutablePointer<Int>?
+        ) -> [String] {
+            index?.pointee = -1
+            return completions(textView.textStorage?.string ?? "", words, charRange)
+        }
     }
 }
 

Core/Sources/SuggestionWidget/SuggestionPanelContent/ChatPanel.swift

@@ -88,7 +88,7 @@ struct ChatPanelMessages: View {
                     }
                 }
                 .listItemTint(.clear)
-                
+
                 Instruction()
 
                 Spacer()
@@ -350,7 +350,28 @@ struct ChatPanelInputArea: View {
                 CustomTextEditor(
                     text: $typedMessage,
                     font: .systemFont(ofSize: 14),
-                    onSubmit: { submitText() }
+                    onSubmit: { submitText() },
+                    completions: { text, _, range in
+                        if text.isEmpty { return [] }
+                        let availableFeatures = [
+                            "/run",
+                            "/airun",
+                            "/math",
+                            "/search",
+                            "@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...])
+                            }
+                    }
                 )
                 .padding(.top, 1)
                 .padding(.bottom, -1)