opencode-kilocode-auth 1.0.5 → 1.0.7

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "opencode-kilocode-auth",
   "module": "index.ts",
-  "version": "1.0.5",
+  "version": "1.0.7",
   "author": "ported from Kilo Code",
   "description": "OpenCode plugin for Kilo Code authentication with support for various models including Giga Potato",
   "files": [

package/src/plugin.ts

@@ -2,11 +2,15 @@ import type { Hooks, PluginInput } from "@opencode-ai/plugin"
 import {
   KILOCODE_PROVIDER_ID,
   DEFAULT_MODEL_ID,
+  DEVICE_AUTH_POLL_INTERVAL_MS,
 } from "./constants"
 import {
+  initiateDeviceAuth,
+  pollDeviceAuth,
   getKilocodeProfile,
   getKilocodeDefaultModel,
   getKilocodeModels,
+  openBrowserUrl,
   getApiUrl,
 } from "./kilocode/auth"
 
@@ -76,7 +80,7 @@ export async function KilocodeAuthPlugin(input: PluginInput): Promise<Hooks> {
 
             const currentKilocodeAuth = currentAuth as KilocodeAuth
             const currentToken = currentKilocodeAuth.type === "api"
-              ? currentKilocodeAuth.apiKey
+              ? currentKilocodeAuth.key
               : currentKilocodeAuth.refresh
 
             if (!currentToken) {
@@ -99,18 +103,22 @@ export async function KilocodeAuthPlugin(input: PluginInput): Promise<Hooks> {
               }
             }
 
-            // Set Kilo Code authorization and required headers
+            // EXACT headers from Kilo Code VSCode extension (DEFAULT_HEADERS + customRequestOptions)
+            // Authorization is set by OpenAI SDK internally
             headers.set("Authorization", `Bearer ${currentToken}`)
-            headers.set("x-api-key", currentToken) // Kilo Code uses both
+
+            // DEFAULT_HEADERS from kilocode/src/api/providers/constants.ts
             headers.set("HTTP-Referer", "https://kilocode.ai")
             headers.set("X-Title", "Kilo Code")
             headers.set("X-KiloCode-Version", "4.151.0")
             headers.set("User-Agent", "Kilo-Code/4.151.0")
+
+            // customRequestOptions from KilocodeOpenrouterHandler
             headers.set("X-KiloCode-EditorName", "Visual Studio Code 1.96.0")
 
-            // Add organization header if present
+            // Organization header (exact casing from headers.ts)
             if (currentKilocodeAuth.organizationId) {
-              headers.set("X-KILOCODE-ORGANIZATIONID", currentKilocodeAuth.organizationId)
+              headers.set("X-KiloCode-OrganizationId", currentKilocodeAuth.organizationId)
             }
 
             // Rewrite URL to Kilo Code endpoint
@@ -139,7 +147,72 @@ export async function KilocodeAuthPlugin(input: PluginInput): Promise<Hooks> {
       },
 
       methods: [
-        // API key is the primary method - get it from https://app.kilo.ai
+        {
+          type: "oauth",
+          label: "Login with Kilo Code (Device Auth)",
+          async authorize() {
+            const authData = await initiateDeviceAuth()
+            const { code, verificationUrl, expiresIn } = authData
+
+            openBrowserUrl(verificationUrl)
+
+            return {
+              url: verificationUrl,
+              instructions: `Enter code: ${code}`,
+              method: "auto" as const,
+
+              async callback() {
+                const maxAttempts = Math.ceil((expiresIn * 1000) / DEVICE_AUTH_POLL_INTERVAL_MS)
+                let attempt = 0
+
+                while (attempt < maxAttempts) {
+                  await new Promise((resolve) => setTimeout(resolve, DEVICE_AUTH_POLL_INTERVAL_MS))
+
+                  try {
+                    const pollResult = await pollDeviceAuth(code)
+
+                    if (pollResult.status === "approved" && pollResult.token) {
+                      let organizationId: string | undefined
+                      try {
+                        const profile = await getKilocodeProfile(pollResult.token)
+                        if (profile.organizations && profile.organizations.length > 0) {
+                          organizationId = profile.organizations[0].id
+                        }
+                      } catch {
+                        // Continue without organization
+                      }
+
+                      const model = await getKilocodeDefaultModel(pollResult.token, organizationId)
+
+                      return {
+                        type: "success" as const,
+                        refresh: pollResult.token,
+                        access: pollResult.token,
+                        expires: Date.now() + 365 * 24 * 60 * 60 * 1000,
+                        organizationId,
+                        model,
+                      } as any
+                    }
+
+                    if (pollResult.status === "denied") {
+                      return { type: "failed" as const }
+                    }
+
+                    if (pollResult.status === "expired") {
+                      return { type: "failed" as const }
+                    }
+                  } catch {
+                    // Continue polling on error
+                  }
+
+                  attempt++
+                }
+
+                return { type: "failed" as const }
+              },
+            }
+          },
+        },
         {
           type: "api",
           label: "Enter API Key (get from app.kilo.ai)",
@@ -147,13 +220,6 @@ export async function KilocodeAuthPlugin(input: PluginInput): Promise<Hooks> {
       ],
     },
 
-    // Add custom headers for Kilo Code requests
-    "chat.headers": async (input, output) => {
-      if (input.model.providerID !== KILOCODE_PROVIDER_ID) return
-
-      output.headers["X-KILOCODE-EDITORNAME"] = "opencode"
-      output.headers["X-KILOCODE-SESSIONID"] = input.sessionID
-    },
   }
 }