Merge branch 'feature/codeium-enterprise' into develop

Author: intitni

Core/Sources/CodeiumService/CodeiumAuthService.swift

@@ -34,7 +34,14 @@ public final class CodeiumAuthService {
     }
 
     func generate(token: String) async throws -> String {
-        var request = URLRequest(url: URL(string: "https://api.codeium.com/register_user/")!)
+        var registerUserUrl = URL(string: "https://api.codeium.com/register_user/")
+        let apiUrl = UserDefaults.shared.value(for: \.codeiumApiUrl)
+        if UserDefaults.shared.value(for: \.codeiumEnterpriseMode), apiUrl != "" {
+            registerUserUrl =
+                URL(string: apiUrl + "/exa.api_server_pb.ApiServerService/RegisterUser")
+        }
+
+        var request = URLRequest(url: registerUserUrl!)
         request.httpMethod = "POST"
         request.addValue("application/json", forHTTPHeaderField: "Content-Type")
         let requestBody = GenerateKeyRequestBody(firebase_id_token: token)

Core/Sources/CodeiumService/CodeiumLanguageServer.swift

@@ -42,13 +42,23 @@ final class CodeiumLanguageServer {
 
         process.executableURL = languageServerExecutableURL
 
+        let isEnterpriseMode = UserDefaults.shared.value(for: \.codeiumEnterpriseMode)
+        var apiServerUrl = "https://server.codeium.com"
+        if isEnterpriseMode, UserDefaults.shared.value(for: \.codeiumApiUrl) != "" {
+            apiServerUrl = UserDefaults.shared.value(for: \.codeiumApiUrl)
+        }
+
         process.arguments = [
             "--api_server_url",
-            "https://server.codeium.com",
+            apiServerUrl,
             "--manager_dir",
             managerDirectoryURL.path,
         ]
 
+        if isEnterpriseMode {
+            process.arguments?.append("--enterprise_mode")
+        }
+
         process.currentDirectoryURL = supportURL
 
         process.terminationHandler = { [weak self] task in
@@ -120,7 +130,7 @@ final class CodeiumLanguageServer {
         self.port = port
         launchHandler?()
     }
-    
+
     func terminate() {
         process.terminationHandler = nil
         if process.isRunning {

Core/Sources/HostApp/AccountSettings/CodeiumView.swift

@@ -10,6 +10,9 @@ struct CodeiumView: View {
         @Published var installationStatus: CodeiumInstallationManager.InstallationStatus
         @Published var installationStep: CodeiumInstallationManager.InstallationStep?
         @AppStorage(\.codeiumVerboseLog) var codeiumVerboseLog
+        @AppStorage(\.codeiumEnterpriseMode) var codeiumEnterpriseMode
+        @AppStorage(\.codeiumPortalUrl) var codeiumPortalUrl
+        @AppStorage(\.codeiumApiUrl) var codeiumApiUrl
 
         init() {
             isSignedIn = codeiumAuthService.isSignedIn
@@ -28,6 +31,13 @@ struct CodeiumView: View {
         }
 
         func generateAuthURL() -> URL {
+            if codeiumEnterpriseMode && (codeiumPortalUrl != "") {
+                return URL(
+                    string: codeiumPortalUrl +
+                        "/profile?response_type=token&redirect_uri=show-auth-token&state=\(UUID().uuidString)&scope=openid%20profile%20email&redirect_parameters_type=query"
+                )!
+            }
+
             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"
             )!
@@ -149,10 +159,10 @@ struct CodeiumView: View {
                         updateButton
                     }
                 }
-                
+
                 if viewModel.isSignedIn {
                     Text("Status: Signed In")
-                    
+
                     Button(action: {
                         Task {
                             do {
@@ -166,7 +176,7 @@ struct CodeiumView: View {
                     }
                 } else {
                     Text("Status: Not Signed In")
-                    
+
                     Button(action: {
                         isSignInPanelPresented = true
                     }) {
@@ -197,9 +207,23 @@ struct CodeiumView: View {
                     }
                 }
             }
-            
+
+            Divider()
+
+            Form {
+                Toggle("Codeium Enterprise Mode", isOn: $viewModel.codeiumEnterpriseMode)
+                TextField("Codeium Portal URL", text: $viewModel.codeiumPortalUrl)
+                TextField("Codeium API URL", text: $viewModel.codeiumApiUrl)
+            }
+            .padding(8)
+            .frame(maxWidth: .infinity, alignment: .leading)
+            .overlay {
+                RoundedRectangle(cornerRadius: 8)
+                    .stroke(Color(nsColor: .separatorColor), style: .init(lineWidth: 1))
+            }
+
             Divider()
-            
+
             Form {
                 Toggle("Verbose Log", isOn: $viewModel.codeiumVerboseLog)
             }
@@ -233,13 +257,13 @@ struct CodeiumSignInView: View {
 
             HStack {
                 Spacer()
-                
+
                 Button(action: {
                     isPresented = false
                 }) {
                     Text("Cancel")
                 }
-                
+
                 Button(action: {
                     isGeneratingKey = true
                     Task {

Tool/Sources/Preferences/Keys.swift

@@ -189,6 +189,18 @@ public extension UserDefaultPreferenceKeys {
     var codeiumVerboseLog: PreferenceKey<Bool> {
         .init(defaultValue: false, key: "CodeiumVerboseLog")
     }
+
+    var codeiumEnterpriseMode: PreferenceKey<Bool> {
+        .init(defaultValue: false, key: "CodeiumEnterpriseMode")
+    }
+
+    var codeiumPortalUrl: PreferenceKey<String> {
+        .init(defaultValue: "", key: "CodeiumPortalUrl")
+    }
+
+    var codeiumApiUrl: PreferenceKey<String> {
+        .init(defaultValue: "", key: "CodeiumApiUrl")
+    }
 }
 
 // MARK: - Prompt to Code
@@ -217,7 +229,7 @@ public extension UserDefaultPreferenceKeys {
     var realtimeSuggestionToggle: PreferenceKey<Bool> {
         .init(defaultValue: true, key: "RealtimeSuggestionToggle")
     }
-    
+
     var suggestionDisplayCompactMode: PreferenceKey<Bool> {
         .init(defaultValue: false, key: "SuggestionDisplayCompactMode")
     }
@@ -249,7 +261,7 @@ public extension UserDefaultPreferenceKeys {
     var realtimeSuggestionDebounce: PreferenceKey<Double> {
         .init(defaultValue: 0, key: "RealtimeSuggestionDebounce")
     }
-    
+
     var acceptSuggestionWithTab: PreferenceKey<Bool> {
         .init(defaultValue: false, key: "AcceptSuggestionWithTab")
     }
@@ -409,13 +421,16 @@ public extension UserDefaultPreferenceKeys {
     var enableXcodeInspectorDebugMenu: FeatureFlag {
         .init(defaultValue: false, key: "FeatureFlag-EnableXcodeInspectorDebugMenu")
     }
-    
+
     var disableFunctionCalling: FeatureFlag {
         .init(defaultValue: false, key: "FeatureFlag-DisableFunctionCalling")
     }
-    
+
     var disableGitHubCopilotSettingsAutoRefreshOnAppear: FeatureFlag {
-        .init(defaultValue: false, key: "FeatureFlag-DisableGitHubCopilotSettingsAutoRefreshOnAppear")
+        .init(
+            defaultValue: false,
+            key: "FeatureFlag-DisableGitHubCopilotSettingsAutoRefreshOnAppear"
+        )
     }
 }