fix(codegen): Adds special cases for AuthStatus to re-enable codegen (#5854)

Adds a ProtocolAuthStatus type to get around issues of using a boolean as a discriminator. Also refactors the code generation into separate classes for each language type and leaves all SCIP logic in Codegen.

## Test plan
Regenerated all three language types and verified that code changes were
just due to updates in the underlying types since disabling.

Author: jamesmcnamara

.github/workflows/agent-bindings.yml

@@ -0,0 +1,38 @@
+# Please post in #team-cody-clients if you need help getting this CI check to pass.
+# Worst-case: feel free to disable this workflow here https://github.com/sourcegraph/cody/actions/workflows/agent-bindings.yml
+name: agent-bindings
+on:
+
+  pull_request:
+    paths:
+      - '**.ts'
+      - '**.tsx'
+      - '**.js'
+
+jobs:
+  kotlin:
+    if: github.repository == 'sourcegraph/cody'
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: pnpm/action-setup@fe02b34f77f8bc703788d5817da081398fad5dd2 # SECURITY: pin third-party action hashes
+        id: pnpm-install
+        with:
+          version: 8.6.7
+          run_install: false
+      - name: Get pnpm store directory
+        id: pnpm-cache
+        shell: bash
+        run: |
+          echo "STORE_PATH=$(pnpm store path)" >> $GITHUB_OUTPUT
+      - uses: actions/cache@v3
+        name: Setup pnpm cache
+        with:
+          path: ${{ steps.pnpm-cache.outputs.STORE_PATH }}
+          key: ${{ runner.os }}-pnpm-store-${{ hashFiles('**/pnpm-lock.yaml') }}
+          restore-keys: |
+            ${{ runner.os }}-pnpm-store-
+      - run: pnpm install --frozen-lockfile
+      - run: pnpm generate-agent-kotlin-bindings
+      - run: ./agent/scripts/error-if-diff.sh
+      - run: ./agent/scripts/compile-bindings-if-diff.sh

agent/bindings/kotlin/lib/src/main/kotlin/com/sourcegraph/cody/agent/protocol_generated/CodyAgentServer.kt

@@ -120,10 +120,12 @@ interface CodyAgentServer {
   fun testing_autocomplete_awaitPendingVisibilityTimeout(params: Null?): CompletableFuture<CompletionItemID?>
   @JsonRequest("testing/autocomplete/setCompletionVisibilityDelay")
   fun testing_autocomplete_setCompletionVisibilityDelay(params: Testing_Autocomplete_SetCompletionVisibilityDelayParams): CompletableFuture<Null?>
+  @JsonRequest("testing/autocomplete/providerConfig")
+  fun testing_autocomplete_providerConfig(params: Null?): CompletableFuture<Testing_Autocomplete_ProviderConfigResult>
   @JsonRequest("extensionConfiguration/change")
-  fun extensionConfiguration_change(params: ExtensionConfiguration): CompletableFuture<AuthStatus?>
+  fun extensionConfiguration_change(params: ExtensionConfiguration): CompletableFuture<ProtocolAuthStatus?>
   @JsonRequest("extensionConfiguration/status")
-  fun extensionConfiguration_status(params: Null?): CompletableFuture<AuthStatus?>
+  fun extensionConfiguration_status(params: Null?): CompletableFuture<ProtocolAuthStatus?>
   @JsonRequest("extensionConfiguration/getSettingsSchema")
   fun extensionConfiguration_getSettingsSchema(params: Null?): CompletableFuture<String>
   @JsonRequest("textDocument/change")

agent/bindings/kotlin/lib/src/main/kotlin/com/sourcegraph/cody/agent/protocol_generated/CodyLLMSiteConfiguration.kt

@@ -16,6 +16,7 @@ object Constants {
   const val agentic = "agentic"
   const val ask = "ask"
   const val assistant = "assistant"
+  const val authenticated = "authenticated"
   const val autocomplete = "autocomplete"
   const val balanced = "balanced"
   const val byok = "byok"
@@ -41,7 +42,6 @@ object Constants {
   const val function = "function"
   const val gateway = "gateway"
   const val history = "history"
-  const val `https-example-com` = "https://example.com"
   const val human = "human"
   const val ignore = "ignore"
   const val indentation = "indentation"
@@ -84,6 +84,7 @@ object Constants {
   const val terminal = "terminal"
   const val tree = "tree"
   const val `tree-sitter` = "tree-sitter"
+  const val unauthenticated = "unauthenticated"
   const val unified = "unified"
   const val use = "use"
   const val user = "user"

agent/bindings/kotlin/lib/src/main/kotlin/com/sourcegraph/cody/agent/protocol_generated/Constants.kt

@@ -1,9 +1,8 @@
 @file:Suppress("FunctionName", "ClassName", "unused", "EnumEntryName", "UnusedImport")
 package com.sourcegraph.cody.agent.protocol_generated;
 
-data class Chat_RestoreParams(
-  val modelID: String? = null,
-  val messages: List<SerializedChatMessage>,
-  val chatID: String,
+data class OrganizationsParams(
+  val name: String,
+  val id: String,
 )
 

agent/bindings/kotlin/lib/src/main/kotlin/com/sourcegraph/cody/agent/protocol_generated/Chat_RestoreParams.kt renamed to agent/bindings/kotlin/lib/src/main/kotlin/com/sourcegraph/cody/agent/protocol_generated/OrganizationsParams.kt

@@ -8,31 +8,23 @@ import com.google.gson.JsonDeserializer;
 import com.google.gson.JsonElement;
 import java.lang.reflect.Type;
 
-sealed class AuthStatus {
+sealed class ProtocolAuthStatus {
   companion object {
-    val deserializer: JsonDeserializer<AuthStatus> =
+    val deserializer: JsonDeserializer<ProtocolAuthStatus> =
       JsonDeserializer { element: JsonElement, _: Type, context: JsonDeserializationContext ->
-          if (element.getAsJsonObject().get("username") == null) {
-              context.deserialize<UnauthenticatedAuthStatus>(element, UnauthenticatedAuthStatus::class.java)
-          } else {
-              context.deserialize<AuthenticatedAuthStatus>(element, AuthenticatedAuthStatus::class.java)
-          }
+        when (element.getAsJsonObject().get("status").getAsString()) {
+          "authenticated" -> context.deserialize<ProtocolAuthenticatedAuthStatus>(element, ProtocolAuthenticatedAuthStatus::class.java)
+          "unauthenticated" -> context.deserialize<ProtocolUnauthenticatedAuthStatus>(element, ProtocolUnauthenticatedAuthStatus::class.java)
+          else -> throw Exception("Unknown discriminator ${element}")
+        }
       }
   }
 }
 
-data class UnauthenticatedAuthStatus(
-  val endpoint: String,
+data class ProtocolAuthenticatedAuthStatus(
+  val status: StatusEnum, // Oneof: authenticated
   val authenticated: Boolean,
-  val showNetworkError: Boolean? = null,
-  val showInvalidAccessTokenError: Boolean? = null,
-  val pendingValidation: Boolean,
-) : AuthStatus() {
-}
-
-data class AuthenticatedAuthStatus(
   val endpoint: String,
-  val authenticated: Boolean,
   val username: String,
   val isFireworksTracingEnabled: Boolean? = null,
   val hasVerifiedEmail: Boolean? = null,
@@ -41,6 +33,25 @@ data class AuthenticatedAuthStatus(
   val displayName: String? = null,
   val avatarURL: String? = null,
   val pendingValidation: Boolean,
-) : AuthStatus() {
+  val organizations: List<OrganizationsParams>? = null,
+) : ProtocolAuthStatus() {
+
+  enum class StatusEnum {
+    @SerializedName("authenticated") Authenticated,
+  }
+}
+
+data class ProtocolUnauthenticatedAuthStatus(
+  val status: StatusEnum, // Oneof: unauthenticated
+  val authenticated: Boolean,
+  val endpoint: String,
+  val showNetworkError: Boolean? = null,
+  val showInvalidAccessTokenError: Boolean? = null,
+  val pendingValidation: Boolean,
+) : ProtocolAuthStatus() {
+
+  enum class StatusEnum {
+    @SerializedName("unauthenticated") Unauthenticated,
+  }
 }
 

agent/bindings/kotlin/lib/src/main/kotlin/com/sourcegraph/cody/agent/protocol_generated/AuthStatus.kt renamed to agent/bindings/kotlin/lib/src/main/kotlin/com/sourcegraph/cody/agent/protocol_generated/ProtocolAuthStatus.kt

@@ -3,9 +3,9 @@ package com.sourcegraph.cody.agent.protocol_generated;
 
 object ProtocolTypeAdapters {
   fun register(gson: com.google.gson.GsonBuilder) {
-    gson.registerTypeAdapter(AuthStatus::class.java, AuthStatus.deserializer)
     gson.registerTypeAdapter(ContextItem::class.java, ContextItem.deserializer)
     gson.registerTypeAdapter(CustomCommandResult::class.java, CustomCommandResult.deserializer)
+    gson.registerTypeAdapter(ProtocolAuthStatus::class.java, ProtocolAuthStatus.deserializer)
     gson.registerTypeAdapter(TextEdit::class.java, TextEdit.deserializer)
     gson.registerTypeAdapter(WorkspaceEditOperation::class.java, WorkspaceEditOperation.deserializer)
   }

agent/bindings/kotlin/lib/src/main/kotlin/com/sourcegraph/cody/agent/protocol_generated/ProtocolTypeAdapters.kt

@@ -10,12 +10,18 @@ data class SerializedChatMessage(
   val speaker: SpeakerEnum, // Oneof: human, assistant, system
   val text: String? = null,
   val model: String? = null,
+  val intent: IntentEnum? = null, // Oneof: search, chat
 ) {
 
   enum class SpeakerEnum {
     @SerializedName("human") Human,
     @SerializedName("assistant") Assistant,
     @SerializedName("system") System,
   }
+
+  enum class IntentEnum {
+    @SerializedName("search") Search,
+    @SerializedName("chat") Chat,
+  }
 }
 

agent/bindings/kotlin/lib/src/main/kotlin/com/sourcegraph/cody/agent/protocol_generated/SerializedChatMessage.kt

@@ -4,6 +4,6 @@ package com.sourcegraph.cody.agent.protocol_generated;
 data class ServerInfo(
   val name: String,
   val authenticated: Boolean? = null,
-  val authStatus: AuthStatus? = null,
+  val authStatus: ProtocolAuthStatus? = null,
 )
 

agent/bindings/kotlin/lib/src/main/kotlin/com/sourcegraph/cody/agent/protocol_generated/ServerInfo.kt

@@ -0,0 +1,9 @@
+@file:Suppress("FunctionName", "ClassName", "unused", "EnumEntryName", "UnusedImport")
+package com.sourcegraph.cody.agent.protocol_generated;
+
+data class Testing_Autocomplete_ProviderConfigResult(
+  val id: String,
+  val legacyModel: String,
+  val configSource: String,
+)
+