Merge branch 'feature/github-copilot-installation-manager' into develop

Author: intitni

Core/Sources/GitHubCopilotService/GitHubCopilotInstallationManager.swift

@@ -0,0 +1,96 @@
+import Foundation
+import Terminal
+
+public struct GitHubCopilotInstallationManager {
+    private static var isInstalling = false
+
+    public init() {}
+
+    public enum InstallationStatus {
+        case notInstalled
+        case installed
+    }
+
+    public func checkInstallation() -> InstallationStatus {
+        guard let urls = try? GitHubCopilotBaseService.createFoldersIfNeeded()
+        else { return .notInstalled }
+        let executableFolderURL = urls.executableURL
+        let binaryURL = executableFolderURL.appendingPathComponent("copilot")
+
+        if !FileManager.default.fileExists(atPath: binaryURL.path) {
+            return .notInstalled
+        }
+
+        return .installed
+    }
+
+    public enum InstallationStep {
+        case downloading
+        case uninstalling
+        case decompressing
+        case done
+    }
+
+    public enum Error: Swift.Error, LocalizedError {
+        case isInstalling
+        case failedToFindLanguageServer
+
+        public var errorDescription: String? {
+            switch self {
+            case .isInstalling:
+                return "Language server is installing."
+            case .failedToFindLanguageServer:
+                return "Failed to find language server. Please open an issue on GitHub."
+            }
+        }
+    }
+
+    public func installLatestVersion() -> AsyncThrowingStream<InstallationStep, Swift.Error> {
+        AsyncThrowingStream<InstallationStep, Swift.Error> { continuation in
+            Task {
+                guard !GitHubCopilotInstallationManager.isInstalling else {
+                    continuation.finish(throwing: Error.isInstalling)
+                    return
+                }
+                GitHubCopilotInstallationManager.isInstalling = true
+                defer { GitHubCopilotInstallationManager.isInstalling = false }
+                do {
+                    continuation.yield(.downloading)
+                    let urls = try GitHubCopilotBaseService.createFoldersIfNeeded()
+                    let executable = Bundle.main.bundleURL.appendingPathComponent("Contents/Applications/CopilotForXcodeExtensionService.app/Contents/Resources/copilot")
+                    guard FileManager.default.fileExists(atPath: executable.path) else {
+                        throw Error.failedToFindLanguageServer
+                    }
+                    
+                    let targetURL = urls.executableURL.appendingPathComponent("copilot")
+
+                    try FileManager.default.copyItem(
+                        at: executable,
+                        to: targetURL
+                    )
+
+                    // update permission 755
+                    try FileManager.default.setAttributes(
+                        [.posixPermissions: 0o755],
+                        ofItemAtPath: targetURL.path
+                    )
+
+                    continuation.yield(.done)
+                    continuation.finish()
+                } catch {
+                    continuation.finish(throwing: error)
+                }
+            }
+        }
+    }
+
+    public func uninstall() async throws {
+        guard let urls = try? GitHubCopilotBaseService.createFoldersIfNeeded()
+        else { return }
+        let executableFolderURL = urls.executableURL
+        let binaryURL = executableFolderURL.appendingPathComponent("copilot")
+        if FileManager.default.fileExists(atPath: binaryURL.path) {
+            try FileManager.default.removeItem(at: binaryURL)
+        }
+    }
+}

Core/Sources/HostApp/AccountSettings/CopilotView.swift

@@ -16,10 +16,69 @@ struct CopilotView: View {
 
         init() {}
     }
+    
+    class ViewModel: ObservableObject {
+        let installationManager = GitHubCopilotInstallationManager()
+        
+        @Published var installationStatus: GitHubCopilotInstallationManager.InstallationStatus
+        @Published var installationStep: GitHubCopilotInstallationManager.InstallationStep?
+        
+        init() {
+            installationStatus = installationManager.checkInstallation()
+        }
+        
+        init(
+            installationStatus: GitHubCopilotInstallationManager.InstallationStatus,
+            installationStep: GitHubCopilotInstallationManager.InstallationStep?
+        ) {
+            assert(isPreview)
+            self.installationStatus = installationStatus
+            self.installationStep = installationStep
+        }
+        
+        func refreshInstallationStatus() {
+            Task { @MainActor in
+                installationStatus = installationManager.checkInstallation()
+            }
+        }
+        
+        func install() async throws {
+            defer { refreshInstallationStatus() }
+            do {
+                for try await step in installationManager.installLatestVersion() {
+                    Task { @MainActor in
+                        self.installationStep = step
+                    }
+                }
+                Task {
+                    try await Task.sleep(nanoseconds: 1_000_000_000)
+                    Task { @MainActor in
+                        self.installationStep = nil
+                    }
+                }
+            } catch {
+                Task { @MainActor in
+                    installationStep = nil
+                }
+                throw error
+            }
+        }
+
+        func uninstall() {
+            Task {
+                defer { refreshInstallationStatus() }
+                try await installationManager.uninstall()
+                Task { @MainActor in
+                    CopilotView.copilotAuthService = nil
+                }
+            }
+        }
+    }
 
     @Environment(\.openURL) var openURL
     @Environment(\.toast) var toast
     @StateObject var settings = Settings()
+    @StateObject var viewModel = ViewModel()
 
     @State var status: GitHubCopilotAccountStatus?
     @State var userCode: String?
@@ -33,6 +92,30 @@ struct CopilotView: View {
         Self.copilotAuthService = service
         return service
     }
+    
+    var installButton: some View {
+        Button(action: {
+            Task {
+                do {
+                    try await viewModel.install()
+                } catch {
+                    toast(Text(error.localizedDescription), .error)
+                }
+            }
+        }) {
+            Text("Install")
+        }
+        .disabled(viewModel.installationStep != nil)
+    }
+
+    var uninstallButton: some View {
+        Button(action: {
+            viewModel.uninstall()
+        }) {
+            Text("Uninstall")
+        }
+        .disabled(viewModel.installationStep != nil)
+    }
 
     var body: some View {
         HStack {
@@ -66,7 +149,15 @@ struct CopilotView: View {
                 .foregroundColor(.secondary)
 
                 VStack(alignment: .leading) {
-                    Text("Language Server Version: \(version ?? "Loading..")")
+                    HStack {
+                        Text("Language Server Version: \(version ?? "Loading..")")
+                        switch viewModel.installationStatus {
+                        case .notInstalled:
+                            installButton
+                        case .installed:
+                            uninstallButton
+                        }
+                    }
                     Text("Status: \(status?.description ?? "Loading..")")
 
                     HStack(alignment: .center) {