ChatWithSelection.swift

import Client
import CopilotModel
import Foundation
import XcodeKit

class ChatWithSelectionCommand: NSObject, XCSourceEditorCommand, CommandType {
    var name: String { "Chat with Selection" }

    func perform(
        with invocation: XCSourceEditorCommandInvocation,
        completionHandler: @escaping (Error?) -> Void
    ) {
        Task {
            do {
                let service = try getService()
                if let content = try await service.chatWithSelection(
                    editorContent: .init(invocation)
                ) {
                    invocation.accept(content)
                }
                completionHandler(nil)
            } catch is CancellationError {
                completionHandler(nil)
            } catch {
                completionHandler(error)
            }
        }
    }
}