Support Codeium sign in

Author: intitni

Core/Sources/CodeiumService/CodeiumKeyGenerator.swift

@@ -1,8 +1,68 @@
-//
-//  File.swift
-//  
-//
-//  Created by Shangxin Guo on 2023/5/10.
-//
-
 import Foundation
+import KeychainAccess
+
+public final class CodeiumAuthService {
+    public init() {}
+
+    let codeiumKeyKey = "codeiumKey"
+    let keychain: Keychain = {
+        let info = Bundle.main.infoDictionary
+        return Keychain(
+            service: info?["KEYCHAIN_SERVICE"] as! String,
+            accessGroup: info?["KEYCHAIN_GROUP"] as! String
+        )
+    }()
+
+    var key: String? { try? keychain.getString(codeiumKeyKey) }
+
+    public var isSignedIn: Bool { return key != nil }
+
+    public func signIn(token: String) async throws {
+        let key = try await generate(token: token)
+        let info = Bundle.main.infoDictionary
+        let keychain = Keychain(
+            service: info?["KEYCHAIN_SERVICE"] as! String,
+            accessGroup: info?["KEYCHAIN_GROUP"] as! String
+        )
+        try keychain.set(key, key: codeiumKeyKey)
+    }
+
+    public func signOut() async throws {
+        try keychain.remove(codeiumKeyKey)
+    }
+
+    struct GenerateKeyRequestBody: Codable {
+        var firebase_id_token: String
+    }
+
+    struct GenerateKeyResponseBody: Codable {
+        var api_key: String
+    }
+
+    struct GenerateKeyErrorResponseBody: Codable, Error, LocalizedError {
+        var detail: String
+        var errorDescription: String? { detail }
+    }
+
+    func generate(token: String) async throws -> String {
+        var request = URLRequest(url: URL(string: "https://api.codeium.com/register_user/")!)
+        request.httpMethod = "POST"
+        request.addValue("application/json", forHTTPHeaderField: "Content-Type")
+        let requestBody = GenerateKeyRequestBody(firebase_id_token: token)
+        let requestData = try JSONEncoder().encode(requestBody)
+        request.httpBody = requestData
+        let (data, _) = try await URLSession.shared.data(for: request)
+        do {
+            let response = try JSONDecoder().decode(GenerateKeyResponseBody.self, from: data)
+            return response.api_key
+        } catch {
+            if let response = try? JSONDecoder()
+                .decode(GenerateKeyErrorResponseBody.self, from: data)
+            {
+                throw response
+            }
+            throw error
+        }
+    }
+}
+

Core/Sources/CodeiumService/CodeiumModels.swift

@@ -110,7 +110,7 @@ struct CodeiumEditorOptions: Codable {
 struct Metadata: Codable {
     var ide_name: String
     var ide_version: String
-    var extension_name: String
+//    var extension_name: String
     var extension_version: String
     var api_key: String
 

Core/Sources/CodeiumService/CodeiumService.swift

@@ -31,8 +31,6 @@ enum CodeiumError: Error, LocalizedError {
     }
 }
 
-let token = ""
-
 public class CodeiumSuggestionService {
     static let sessionId = UUID().uuidString
     let projectRootURL: URL
@@ -45,6 +43,10 @@ public class CodeiumSuggestionService {
     let languageServerURL: URL
     let supportURL: URL
 
+    let authService = CodeiumAuthService()
+    
+    var xcodeVersion = "14.0"
+
     init(designatedServer: CodeiumLSP) {
         projectRootURL = URL(fileURLWithPath: "/")
         server = designatedServer
@@ -62,12 +64,16 @@ public class CodeiumSuggestionService {
         guard FileManager.default.fileExists(atPath: languageServerURL.path) else {
             throw CodeiumError.languageServerNotInstalled
         }
-        try setupServerIfNeeded()
+        Task {
+            try await setupServerIfNeeded()
+        }
     }
 
     @discardableResult
-    func setupServerIfNeeded() throws -> CodeiumLSP {
+    func setupServerIfNeeded() async throws -> CodeiumLSP {
         if let server { return server }
+        let metadata = try getMetadata()
+        xcodeVersion = (try? await getXcodeVersion()) ?? xcodeVersion
         let tempFolderURL = FileManager.default.temporaryDirectory
         let managerDirectoryURL = tempFolderURL
             .appendingPathComponent("com.intii.CopilotForXcode")
@@ -94,9 +100,8 @@ public class CodeiumSuggestionService {
 
         server.launchHandler = { [weak self] in
             guard let self else { return }
-            let metadata = self.getMetadata()
             self.onServiceLaunched()
-            self.heartbeatTask = Task { [weak self] in
+            self.heartbeatTask = Task { [weak self, metadata] in
                 while true {
                     try Task.checkCancellation()
                     _ = try? await self?.server?.sendRequest(
@@ -151,13 +156,18 @@ public class CodeiumSuggestionService {
 }
 
 extension CodeiumSuggestionService {
-    func getMetadata() -> Metadata {
-        Metadata(
-            ide_name: "jetbrains",
-            ide_version: "14.3",
-            extension_name: "Copilot for Xcode",
+    func getMetadata() throws -> Metadata {
+        guard let key = authService.key else {
+            struct E: Error, LocalizedError {
+                var errorDescription: String? { "Codeium not signed in." }
+            }
+            throw E()
+        }
+        return Metadata(
+            ide_name: "xcode",
+            ide_version: xcodeVersion,
             extension_version: "14.0.0",
-            api_key: token,
+            api_key: key,
             session_id: CodeiumSuggestionService.sessionId,
             request_id: requestCounter
         )
@@ -195,7 +205,7 @@ extension CodeiumSuggestionService: CodeiumSuggestionServiceType {
         let relativePath = getRelativePath(of: fileURL)
 
         let request = CodeiumRequest.GetCompletion(requestBody: .init(
-            metadata: getMetadata(),
+            metadata: try getMetadata(),
             document: .init(
                 absolute_path: fileURL.path,
                 relative_path: relativePath,
@@ -221,7 +231,7 @@ extension CodeiumSuggestionService: CodeiumSuggestionServiceType {
                 }
         ))
 
-        let result = try await (try setupServerIfNeeded()).sendRequest(request)
+        let result = try await (try await setupServerIfNeeded()).sendRequest(request)
 
         return result.completionItems?.filter { item in
             if ignoreSpaceOnlySuggestions {
@@ -279,3 +289,36 @@ extension CodeiumSuggestionService: CodeiumSuggestionServiceType {
     }
 }
 
+func getXcodeVersion() async throws -> String {
+    let task = Process()
+    task.launchPath = "/usr/bin/xcodebuild"
+    task.arguments = ["-version"]
+    let outpipe = Pipe()
+    task.standardOutput = outpipe
+    task.standardError = Pipe()
+    return try await withUnsafeThrowingContinuation { continuation in
+        do {
+            task.terminationHandler = { _ in
+                do {
+                    if let data = try outpipe.fileHandleForReading.readToEnd(),
+                       let content = String(data: data, encoding: .utf8)
+                    {
+                        let firstLine = content.split(separator: "\n").first ?? ""
+                        var version = firstLine.replacingOccurrences(of: "Xcode ", with: "")
+                        if version.isEmpty {
+                            version = "14.0"
+                        }
+                        continuation.resume(returning: version)
+                        return
+                    }
+                    continuation.resume(returning: "")
+                } catch {
+                    continuation.resume(throwing: error)
+                }
+            }
+            try task.run()
+        } catch {
+            continuation.resume(throwing: error)
+        }
+    }
+}

Core/Sources/HostApp/AccountSettings/CodeiumView.swift

@@ -1,8 +1,110 @@
-//
-//  File.swift
-//  
-//
-//  Created by Shangxin Guo on 2023/5/10.
-//
-
+import CodeiumService
 import Foundation
+import SwiftUI
+
+struct CodeiumView: View {
+    class ViewModel: ObservableObject {
+        let codeiumAuthService = CodeiumAuthService()
+        @Published var isSignedIn: Bool
+        
+        init() {
+            isSignedIn = codeiumAuthService.isSignedIn
+        }
+        
+        func generateAuthURL() -> URL {
+            return URL(
+                string: "https://www.codeium.com/profile?response_type=token&redirect_uri=show-auth-token&state=\(UUID().uuidString)&scope=openid%20profile%20email&redirect_parameters_type=query"
+            )!
+        }
+        
+        func signIn(token: String) async throws {
+            try await codeiumAuthService.signIn(token: token)
+            Task { @MainActor in isSignedIn = true }
+            
+        }
+        
+        func signOut() async throws {
+            try await codeiumAuthService.signOut()
+            Task { @MainActor in isSignedIn = false }
+        }
+    }
+    
+    @StateObject var viewModel = ViewModel()
+    @State var isSignInPanelPresented = false
+
+    var body: some View {
+        Form {
+            if viewModel.isSignedIn {
+                Button(action: {
+                    viewModel.isSignedIn = false
+                }) {
+                    Text("Sign Out")
+                }
+            } else {
+                Button(action: {
+                    isSignInPanelPresented = true
+                }) {
+                    Text("Sign In")
+                }
+            }
+        }
+        .sheet(isPresented: $isSignInPanelPresented) {
+            CodeiumSignInView(viewModel: viewModel, isPresented: $isSignInPanelPresented)
+        }
+    }
+}
+
+struct CodeiumSignInView: View {
+    let viewModel: CodeiumView.ViewModel
+    @Binding var isPresented: Bool
+    @Environment(\.openURL) var openURL
+    @Environment(\.toast) var toast
+    @State var isGeneratingKey = false
+    @State var token = ""
+
+    var body: some View {
+        VStack {
+            Text(
+                "You will be redirected to codeium.com. Please paste the generated token below and click the \"Sign In\" button."
+            )
+
+            TextEditor(text: $token)
+                .font(Font.system(.body, design: .monospaced))
+                .padding(4)
+                .frame(minHeight: 120)
+                .multilineTextAlignment(.leading)
+                .overlay(
+                    RoundedRectangle(cornerRadius: 4)
+                        .stroke(Color(nsColor: .separatorColor), lineWidth: 1)
+                )
+
+            Button(action: {
+                isGeneratingKey = true
+                Task {
+                    do {
+                        try await viewModel.signIn(token: token)
+                        isGeneratingKey = false
+                        isPresented  = false
+                    } catch {
+                        isGeneratingKey = false
+                        toast(Text(error.localizedDescription), .error)
+                    }
+                }
+            }) {
+                Text(isGeneratingKey ? "Signing In.." : "Sign In")
+            }.disabled(isGeneratingKey)
+        }
+        .padding()
+        .onAppear {
+            openURL(viewModel.generateAuthURL())
+        }
+    }
+}
+
+struct CodeiumView_Previews: PreviewProvider {
+    static var previews: some View {
+        CodeiumView()
+            .frame(width: 600, height: 500)
+    }
+}
+

Core/Sources/HostApp/ServiceView.swift

@@ -14,9 +14,18 @@ struct ServiceView: View {
             )
 
             ScrollView {
-                OpenAIView().padding()
+                CodeiumView().padding()
             }.sidebarItem(
                 tag: 1,
+                title: "Codeium",
+                subtitle: "Suggestion",
+                image: "globe"
+            )
+            
+            ScrollView {
+                OpenAIView().padding()
+            }.sidebarItem(
+                tag: 2,
                 title: "OpenAI",
                 subtitle: "Chat, Prompt to Code",
                 image: "globe"