GitHubCopilotRequest.swift

import Foundation
import JSONRPC
import LanguageServerProtocol
import SuggestionBasic
import XcodeInspector

struct GitHubCopilotDoc: Codable {
    var source: String
    var tabSize: Int
    var indentSize: Int
    var insertSpaces: Bool
    var path: String
    var uri: String
    var relativePath: String
    var languageId: CodeLanguage
    var position: Position
    var version: Int = 0
}

protocol GitHubCopilotRequestType {
    associatedtype Response: Codable
    var request: ClientRequest { get }
}

public struct GitHubCopilotCodeSuggestion: Codable, Equatable {
    public init(
        text: String,
        position: CursorPosition,
        uuid: String,
        range: CursorRange,
        displayText: String
    ) {
        self.text = text
        self.position = position
        self.uuid = uuid
        self.range = range
        self.displayText = displayText
    }

    /// The new code to be inserted and the original code on the first line.
    public var text: String
    /// The position of the cursor before generating the completion.
    public var position: CursorPosition
    /// An id.
    public var uuid: String
    /// The range of the original code that should be replaced.
    public var range: CursorRange
    /// The new code to be inserted.
    public var displayText: String
}

enum GitHubCopilotChatSource: String, Codable {
    case panel
    case inline
}

enum GitHubCopilotRequest {
    struct SetEditorInfo: GitHubCopilotRequestType {
        let xcodeVersion: String

        struct Response: Codable {}

        var networkProxy: JSONValue? {
            let host = UserDefaults.shared.value(for: \.gitHubCopilotProxyHost)
            if host.isEmpty { return nil }
            var port = UserDefaults.shared.value(for: \.gitHubCopilotProxyPort)
            if port.isEmpty { port = "80" }
            let username = UserDefaults.shared.value(for: \.gitHubCopilotProxyUsername)
            if username.isEmpty {
                return .hash([
                    "host": .string(host),
                    "port": .number(Double(Int(port) ?? 80)),
                    "rejectUnauthorized": .bool(UserDefaults.shared
                        .value(for: \.gitHubCopilotUseStrictSSL)),
                ])
            } else {
                return .hash([
                    "host": .string(host),
                    "port": .number(Double(Int(port) ?? 80)),
                    "rejectUnauthorized": .bool(UserDefaults.shared
                        .value(for: \.gitHubCopilotUseStrictSSL)),
                    "username": .string(username),
                    "password": .string(UserDefaults.shared
                        .value(for: \.gitHubCopilotProxyPassword)),

                ])
            }
        }

        var http: JSONValue? {
            var dict: [String: JSONValue] = [:]
            let host = UserDefaults.shared.value(for: \.gitHubCopilotProxyHost)
            if host.isEmpty { return nil }
            var port = UserDefaults.shared.value(for: \.gitHubCopilotProxyPort)
            if port.isEmpty { port = "80" }
            let username = UserDefaults.shared.value(for: \.gitHubCopilotProxyUsername)
            let password = UserDefaults.shared.value(for: \.gitHubCopilotProxyPassword)
            let strictSSL = UserDefaults.shared.value(for: \.gitHubCopilotUseStrictSSL)

            let url = if !username.isEmpty {
                "http://\(username):\(password)@\(host):\(port)"
            } else {
                "http://\(host):\(port)"
            }

            dict["proxy"] = .string(url)
            dict["proxyStrictSSL"] = .bool(strictSSL)

            if dict.isEmpty { return nil }

            return .hash(dict)
        }

        var editorConfiguration: JSONValue? {
            var dict: [String: JSONValue] = [:]
            dict["http"] = http

            let enterpriseURI = UserDefaults.shared.value(for: \.gitHubCopilotEnterpriseURI)
            if !enterpriseURI.isEmpty {
                dict["github-enterprise"] = .hash([
                    "uri": .string(enterpriseURI),
                ])
            }

            if dict.isEmpty { return nil }
            return .hash(dict)
        }

        var authProvider: JSONValue? {
            var dict: [String: JSONValue] = [:]
            let enterpriseURI = UserDefaults.shared.value(for: \.gitHubCopilotEnterpriseURI)
            if !enterpriseURI.isEmpty {
                dict["url"] = .string(enterpriseURI)
            }

            if dict.isEmpty { return nil }
            return .hash(dict)
        }

        var request: ClientRequest {
            let pretendToBeVSCode = UserDefaults.shared
                .value(for: \.gitHubCopilotPretendIDEToBeVSCode)
            var dict: [String: JSONValue] = [
                "editorInfo": pretendToBeVSCode ? .hash([
                    "name": "vscode",
                    "version": "1.99.3",
                ]) : .hash([
                    "name": "Xcode",
                    "version": .string(xcodeVersion),
                ]),
                "editorPluginInfo": .hash([
                    "name": "Copilot for Xcode",
                    "version": .string(Bundle.main
                        .infoDictionary?["CFBundleShortVersionString"] as? String ?? ""),
                ]),
            ]

            dict["editorConfiguration"] = editorConfiguration
            dict["authProvider"] = authProvider
            dict["networkProxy"] = networkProxy

            return .custom("setEditorInfo", .hash(dict))
        }
    }

    struct GetVersion: GitHubCopilotRequestType {
        struct Response: Codable {
            var version: String
        }

        var request: ClientRequest {
            .custom("getVersion", .hash([:]))
        }
    }

    struct CheckStatus: GitHubCopilotRequestType {
        struct Response: Codable {
            var status: GitHubCopilotAccountStatus
        }

        var request: ClientRequest {
            .custom("checkStatus", .hash([:]))
        }
    }

    struct SignInInitiate: GitHubCopilotRequestType {
        struct Response: Codable {
            var verificationUri: String
            var status: String
            var userCode: String
            var expiresIn: Int
            var interval: Int
        }

        var request: ClientRequest {
            .custom("signInInitiate", .hash([:]))
        }
    }

    struct SignInConfirm: GitHubCopilotRequestType {
        struct Response: Codable {
            var status: GitHubCopilotAccountStatus
            var user: String
        }

        var userCode: String

        var request: ClientRequest {
            .custom("signInConfirm", .hash([
                "userCode": .string(userCode),
            ]))
        }
    }

    struct SignOut: GitHubCopilotRequestType {
        struct Response: Codable {
            var status: GitHubCopilotAccountStatus
        }

        var request: ClientRequest {
            .custom("signOut", .hash([:]))
        }
    }

    struct GetCompletions: GitHubCopilotRequestType {
        struct Response: Codable {
            var completions: [GitHubCopilotCodeSuggestion]
        }

        var doc: GitHubCopilotDoc

        var request: ClientRequest {
            let data = (try? JSONEncoder().encode(doc)) ?? Data()
            let dict = (try? JSONDecoder().decode(JSONValue.self, from: data)) ?? .hash([:])
            return .custom("getCompletions", .hash([
                "doc": dict,
            ]))
        }
    }

    struct GetCompletionsCycling: GitHubCopilotRequestType {
        struct Response: Codable {
            var completions: [GitHubCopilotCodeSuggestion]
        }

        var doc: GitHubCopilotDoc

        var request: ClientRequest {
            let data = (try? JSONEncoder().encode(doc)) ?? Data()
            let dict = (try? JSONDecoder().decode(JSONValue.self, from: data)) ?? .hash([:])
            return .custom("getCompletionsCycling", .hash([
                "doc": dict,
            ]))
        }
    }

    struct InlineCompletion: GitHubCopilotRequestType {
        struct Response: Codable {
            var items: [InlineCompletionItem]
        }

        struct InlineCompletionItem: Codable {
            var insertText: String
            var filterText: String?
            var range: Range?
            var command: Command?

            struct Range: Codable {
                var start: Position
                var end: Position
            }

            struct Command: Codable {
                var title: String
                var command: String
                var arguments: [String]?
            }
        }

        var doc: Input

        struct Input: Codable {
            var textDocument: _TextDocument; struct _TextDocument: Codable {
                var uri: String
                var version: Int
            }

            var position: Position
            var formattingOptions: FormattingOptions
            var context: _Context; struct _Context: Codable {
                enum TriggerKind: Int, Codable {
                    case invoked = 1
                    case automatic = 2
                }

                var triggerKind: TriggerKind
            }
        }

        var request: ClientRequest {
            let data = (try? JSONEncoder().encode(doc)) ?? Data()
            let dict = (try? JSONDecoder().decode(JSONValue.self, from: data)) ?? .hash([:])
            return .custom("textDocument/inlineCompletion", dict)
        }
    }

    struct GetPanelCompletions: GitHubCopilotRequestType {
        struct Response: Codable {
            var completions: [GitHubCopilotCodeSuggestion]
        }

        var doc: GitHubCopilotDoc

        var request: ClientRequest {
            let data = (try? JSONEncoder().encode(doc)) ?? Data()
            let dict = (try? JSONDecoder().decode(JSONValue.self, from: data)) ?? .hash([:])
            return .custom("getPanelCompletions", .hash([
                "doc": dict,
            ]))
        }
    }

    struct NotifyAccepted: GitHubCopilotRequestType {
        struct Response: Codable {}

        var completionUUID: String

        var request: ClientRequest {
            .custom("notifyAccepted", .hash([
                "uuid": .string(completionUUID),
            ]))
        }
    }

    struct NotifyRejected: GitHubCopilotRequestType {
        struct Response: Codable {}

        var completionUUIDs: [String]

        var request: ClientRequest {
            .custom("notifyRejected", .hash([
                "uuids": .array(completionUUIDs.map(JSONValue.string)),
            ]))
        }
    }

    struct ConversationCreate: GitHubCopilotRequestType {
        struct Response: Codable {
            var conversationId: String
            var turnId: String
        }

        struct RequestBody: Codable {
            public struct Reference: Codable, Equatable, Hashable {
                public var type: String = "file"
                public let uri: String
                public let position: Position?
                public let visibleRange: SuggestionBasic.CursorRange?
                public let selection: SuggestionBasic.CursorRange?
                public let openedAt: String?
                public let activeAt: String?
            }

            enum ConversationSource: String, Codable {
                case panel, inline
            }

            enum ConversationMode: String, Codable {
                case agent = "Agent"
            }

            struct ConversationTurn: Codable {
                var request: String
                var response: String?
                var turnId: String?
            }

            var workDoneToken: String
            var turns: [ConversationTurn]
            var capabilities: Capabilities
            var textDocument: GitHubCopilotDoc?
            var references: [Reference]?
            var computeSuggestions: Bool?
            var source: ConversationSource?
            var workspaceFolder: String?
            var workspaceFolders: [WorkspaceFolder]?
            var ignoredSkills: [String]?
            var model: String?
            var chatMode: ConversationMode?
            var userLanguage: String?

            struct Capabilities: Codable {
                var skills: [String]
                var allSkills: Bool?
            }
        }

        let requestBody: RequestBody

        var request: ClientRequest {
            let data = (try? JSONEncoder().encode(requestBody)) ?? Data()
            let dict = (try? JSONDecoder().decode(JSONValue.self, from: data)) ?? .hash([:])
            return .custom("conversation/create", dict)
        }
    }

    struct ConversationTurn: GitHubCopilotRequestType {
        struct Response: Codable {}

        struct RequestBody: Codable {
            var workDoneToken: String
            var conversationId: String
            var message: String
            var textDocument: GitHubCopilotDoc?
            var ignoredSkills: [String]?
            var references: [ConversationCreate.RequestBody.Reference]?
            var model: String?
            var workspaceFolder: String?
            var workspaceFolders: [WorkspaceFolder]?
            var chatMode: String?
        }

        let requestBody: RequestBody

        var request: ClientRequest {
            let data = (try? JSONEncoder().encode(requestBody)) ?? Data()
            let dict = (try? JSONDecoder().decode(JSONValue.self, from: data)) ?? .hash([:])
            return .custom("conversation/turn", dict)
        }
    }

    struct ConversationTurnDelete: GitHubCopilotRequestType {
        struct Response: Codable {}

        struct RequestBody: Codable {
            var conversationId: String
            var turnId: String
            var options: [String: String]?
            var source: GitHubCopilotChatSource?
        }

        let requestBody: RequestBody

        var request: ClientRequest {
            let data = (try? JSONEncoder().encode(requestBody)) ?? Data()
            let dict = (try? JSONDecoder().decode(JSONValue.self, from: data)) ?? .hash([:])
            return .custom("conversation/turnDelete", dict)
        }
    }

    struct ConversationDestroy: GitHubCopilotRequestType {
        struct Response: Codable {}

        struct RequestBody: Codable {
            var conversationId: String
            var options: [String: String]?
        }

        let requestBody: RequestBody

        var request: ClientRequest {
            let data = (try? JSONEncoder().encode(requestBody)) ?? Data()
            let dict = (try? JSONDecoder().decode(JSONValue.self, from: data)) ?? .hash([:])
            return .custom("conversation/destroy", dict)
        }
    }

    struct CopilotModels: GitHubCopilotRequestType {
        typealias Response = [GitHubCopilotModel]

        var request: ClientRequest {
            .custom("copilot/models", .hash([:]))
        }
    }
}

public struct GitHubCopilotModel: Codable, Equatable {
    public let modelFamily: String
    public let modelName: String
    public let id: String
//            public let modelPolicy: CopilotModelPolicy?
    public let scopes: [GitHubCopilotPromptTemplateScope]
    public let preview: Bool
    public let isChatDefault: Bool
    public let isChatFallback: Bool
//            public let capabilities: CopilotModelCapabilities
//            public let billing: CopilotModelBilling?
}

public struct GitHubCopilotLLMModel: Equatable, Decodable, Identifiable {
    public var id: String { modelId }
    public var modelId: String
    public var familyName: String
    public var contextWindow: Int
}

public enum GitHubCopilotPromptTemplateScope: String, Codable, Equatable {
    case chatPanel = "chat-panel"
    case editPanel = "edit-panel"
    case agentPanel = "agent-panel"
    case editor
    case inline
    case completion
}