Support downloading Codeium language server

Author: intitni

Core/Package.swift

@@ -127,7 +127,7 @@ let package = Package(
                 "GitHubCopilotService",
                 "CodeiumService",
                 "SuggestionModel",
-                "LaunchAgentManager"
+                "LaunchAgentManager",
             ]
         ),
 
@@ -256,7 +256,13 @@ let package = Package(
 
         .target(
             name: "CodeiumService",
-            dependencies: ["LanguageClient", "SuggestionModel", "Preferences", "KeychainAccess"]
+            dependencies: [
+                "LanguageClient",
+                "SuggestionModel",
+                "Preferences",
+                "KeychainAccess",
+                "Terminal"
+            ]
         ),
     ]
 )

Core/Sources/CodeiumService/CodeiumAuthService.swift

@@ -7,8 +7,8 @@ public final class CodeiumAuthService {
     let keychain: Keychain = {
         let info = Bundle.main.infoDictionary
         return Keychain(
-            service: info?["BUNDLE_IDENTIFIER_BASE"] as! String,
-            accessGroup: "\(info?["APP_ID_PREFIX"] as! String)\(info?["BUNDLE_IDENTIFIER_BASE"] as! String).Shared"
+            service: info?["BUNDLE_IDENTIFIER_BASE"] as? String ?? "",
+            accessGroup: "\(info?["APP_ID_PREFIX"] as? String ?? "")\(info?["BUNDLE_IDENTIFIER_BASE"] as? String ?? "").Shared"
         )
     }()
 
@@ -19,10 +19,6 @@ public final class CodeiumAuthService {
     public func signIn(token: String) async throws {
         let key = try await generate(token: token)
         let info = Bundle.main.infoDictionary
-        let keychain = Keychain(
-            service: info?["BUNDLE_IDENTIFIER_BASE"] as! String,
-            accessGroup: "\(info?["APP_ID_PREFIX"] as! String)\(info?["BUNDLE_IDENTIFIER_BASE"] as! String).Shared"
-        )
         try keychain.set(key, key: codeiumKeyKey)
     }
 

Core/Sources/CodeiumService/CodeiumInstallationManager.swift

@@ -0,0 +1,131 @@
+import Foundation
+import Terminal
+
+public struct CodeiumInstallationManager {
+    private static var isInstalling = false
+    let latestSupportedVersion: String = "1.2.9"
+
+    public init() {}
+
+    public enum InstallationStatus {
+        case notInstalled
+        case installed(String)
+        case outdated(current: String, latest: String)
+        case unsupported(current: String, latest: String)
+    }
+
+    public func checkInstallation() -> InstallationStatus {
+        guard let urls = try? CodeiumSuggestionService.createFoldersIfNeeded()
+        else { return .notInstalled }
+        let executableFolderURL = urls.executableURL
+        let binaryURL = executableFolderURL.appendingPathComponent("language_server")
+        let versionFileURL = executableFolderURL.appendingPathComponent("version")
+
+        if !FileManager.default.fileExists(atPath: binaryURL.path) {
+            return .notInstalled
+        }
+
+        if FileManager.default.fileExists(atPath: versionFileURL.path),
+           let versionData = try? Data(contentsOf: versionFileURL),
+           let version = String(data: versionData, encoding: .utf8)
+        {
+            switch version.compare(latestSupportedVersion) {
+            case .orderedAscending:
+                return .outdated(current: version, latest: latestSupportedVersion)
+            case .orderedSame:
+                return .installed(version)
+            case .orderedDescending:
+                return .unsupported(current: version, latest: latestSupportedVersion)
+            }
+        }
+
+        return .outdated(current: "Unknown", latest: latestSupportedVersion)
+    }
+
+    public enum InstallationStep {
+        case downloading
+        case uninstalling
+        case decompressing
+        case done
+    }
+
+    public func installLatestVersion() -> AsyncThrowingStream<InstallationStep, Error> {
+        AsyncThrowingStream<InstallationStep, Error> { continuation in
+            Task {
+                guard !CodeiumInstallationManager.isInstalling else {
+                    continuation.finish(throwing: CodeiumError.languageServiceIsInstalling)
+                    return
+                }
+                CodeiumInstallationManager.isInstalling = true
+                defer { CodeiumInstallationManager.isInstalling = false }
+                do {
+                    continuation.yield(.downloading)
+                    let urls = try CodeiumSuggestionService.createFoldersIfNeeded()
+                    let urlString =
+                        "https://github.com/Exafunction/codeium/releases/download/language-server-v\(latestSupportedVersion)/language_server_macos_\(isAppleSilicon() ? "arm" : "x64").gz"
+                    guard let url = URL(string: urlString) else { return }
+
+                    // download
+                    let (fileURL, _) = try await URLSession.shared.download(from: url)
+                    let targetURL = urls.executableURL.appendingPathComponent("language_server")
+                        .appendingPathExtension("gz")
+                    try FileManager.default.copyItem(at: fileURL, to: targetURL)
+                    defer { try? FileManager.default.removeItem(at: targetURL) }
+
+                    continuation.yield(.uninstalling)
+                    try await uninstall()
+
+                    continuation.yield(.decompressing)
+                    // extract file
+                    let terminal = Terminal()
+                    _ = try await terminal.runCommand(
+                        "/usr/bin/gunzip",
+                        arguments: [targetURL.path],
+                        environment: [:]
+                    )
+
+                    // update permission 755
+                    try FileManager.default.setAttributes(
+                        [.posixPermissions: 0o755],
+                        ofItemAtPath: targetURL.deletingPathExtension().path
+                    )
+
+                    // create version file
+                    let data = latestSupportedVersion.data(using: .utf8)
+                    FileManager.default.createFile(
+                        atPath: urls.executableURL.appendingPathComponent("version").path,
+                        contents: data
+                    )
+
+                    continuation.yield(.done)
+                    continuation.finish()
+                } catch {
+                    continuation.finish(throwing: error)
+                }
+            }
+        }
+    }
+
+    public func uninstall() async throws {
+        guard let urls = try? CodeiumSuggestionService.createFoldersIfNeeded()
+        else { return }
+        let executableFolderURL = urls.executableURL
+        let binaryURL = executableFolderURL.appendingPathComponent("language_server")
+        let versionFileURL = executableFolderURL.appendingPathComponent("version")
+        if FileManager.default.fileExists(atPath: binaryURL.path) {
+            try FileManager.default.removeItem(at: binaryURL)
+        }
+        if FileManager.default.fileExists(atPath: versionFileURL.path) {
+            try FileManager.default.removeItem(at: versionFileURL)
+        }
+    }
+}
+
+func isAppleSilicon() -> Bool {
+    var result = false
+    #if arch(arm64)
+    result = true
+    #endif
+    return result
+}
+

Core/Sources/CodeiumService/CodeiumLanguageServer.swift

@@ -106,7 +106,9 @@ final class CodeiumLanguageServer {
 
     deinit {
         process.terminationHandler = nil
-        process.terminate()
+        if process.isRunning {
+            process.terminate()
+        }
         transport.close()
     }
 
@@ -141,6 +143,7 @@ extension CodeiumLanguageServer: CodeiumLSP {
             do {
                 let error = try JSONDecoder().decode(CodeiumResponseError.self, from: data)
                 Logger.codeium.error(error.message)
+                throw CancellationError()
             } catch {
                 Logger.codeium.error(error.localizedDescription)
                 throw error

Core/Sources/CodeiumService/CodeiumService.swift

@@ -22,11 +22,17 @@ public protocol CodeiumSuggestionServiceType {
 
 enum CodeiumError: Error, LocalizedError {
     case languageServerNotInstalled
+    case languageServerOutdated
+    case languageServiceIsInstalling
 
     var errorDescription: String? {
         switch self {
         case .languageServerNotInstalled:
-            return "Language server is not installed."
+            return "Language server is not installed. Please install it in the host app."
+        case .languageServerOutdated:
+            return "Language server is outdated. Please update it in the host app."
+        case .languageServiceIsInstalling:
+            return "Language service is installing. Please wait."
         }
     }
 }
@@ -44,7 +50,7 @@ public class CodeiumSuggestionService {
     let supportURL: URL
 
     let authService = CodeiumAuthService()
-    
+
     var xcodeVersion = "14.0"
 
     init(designatedServer: CodeiumLSP) {
@@ -61,9 +67,6 @@ public class CodeiumSuggestionService {
         let urls = try CodeiumSuggestionService.createFoldersIfNeeded()
         languageServerURL = urls.executableURL.appendingPathComponent("language_server")
         supportURL = urls.supportURL
-        guard FileManager.default.fileExists(atPath: languageServerURL.path) else {
-            throw CodeiumError.languageServerNotInstalled
-        }
         Task {
             try await setupServerIfNeeded()
         }
@@ -72,6 +75,18 @@ public class CodeiumSuggestionService {
     @discardableResult
     func setupServerIfNeeded() async throws -> CodeiumLSP {
         if let server { return server }
+        
+        let binaryManager = CodeiumInstallationManager()
+        let installationStatus = binaryManager.checkInstallation()
+        switch installationStatus {
+        case .installed, .unsupported:
+            break
+        case .notInstalled:
+            throw CodeiumError.languageServerNotInstalled
+        case .outdated:
+            throw CodeiumError.languageServerOutdated
+        }
+        
         let metadata = try getMetadata()
         xcodeVersion = (try? await getXcodeVersion()) ?? xcodeVersion
         let tempFolderURL = FileManager.default.temporaryDirectory
@@ -112,8 +127,8 @@ public class CodeiumSuggestionService {
             }
         }
 
-        server.start()
         self.server = server
+        server.start()
         return server
     }
 
@@ -322,3 +337,4 @@ func getXcodeVersion() async throws -> String {
         }
     }
 }
+