opencode-kilocode-auth 1.0.1 → 1.0.3

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "opencode-kilocode-auth",
   "module": "index.ts",
-  "version": "1.0.1",
+  "version": "1.0.3",
   "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

@@ -1,28 +1,23 @@
 import type { Hooks, PluginInput } from "@opencode-ai/plugin"
 import {
   KILOCODE_PROVIDER_ID,
-  KILOCODE_API_BASE_URL,
-  DEFAULT_HEADERS,
   DEFAULT_MODEL_ID,
-  DEVICE_AUTH_POLL_INTERVAL_MS,
 } from "./constants"
 import {
-  initiateDeviceAuth,
-  pollDeviceAuth,
   getKilocodeProfile,
   getKilocodeDefaultModel,
   getKilocodeModels,
-  openBrowserUrl,
   getApiUrl,
 } from "./kilocode/auth"
 
 interface KilocodeAuth {
-  type: "oauth"
-  refresh: string // token
+  type: "oauth" | "api"
+  refresh?: string // token (for oauth)
   access?: string
   expires?: number
   organizationId?: string
   model?: string
+  apiKey?: string // for api type
 }
 
 /**
@@ -40,10 +35,13 @@ export async function KilocodeAuthPlugin(input: PluginInput): Promise<Hooks> {
 
       async loader(getAuth, provider) {
         const auth = await getAuth()
-        if (!auth || auth.type !== "oauth") return {}
+        if (!auth) return {}
 
         const kilocodeAuth = auth as KilocodeAuth
-        const token = kilocodeAuth.refresh
+        // Support both OAuth (refresh token) and API key authentication
+        const token = kilocodeAuth.type === "api"
+          ? kilocodeAuth.apiKey
+          : kilocodeAuth.refresh
 
         if (!token) return {}
 
@@ -72,12 +70,18 @@ export async function KilocodeAuthPlugin(input: PluginInput): Promise<Hooks> {
 
           async fetch(requestInput: RequestInfo | URL, init?: RequestInit) {
             const currentAuth = await getAuth()
-            if (!currentAuth || currentAuth.type !== "oauth") {
+            if (!currentAuth) {
               return fetch(requestInput, init)
             }
 
             const currentKilocodeAuth = currentAuth as KilocodeAuth
-            const currentToken = currentKilocodeAuth.refresh
+            const currentToken = currentKilocodeAuth.type === "api"
+              ? currentKilocodeAuth.apiKey
+              : currentKilocodeAuth.refresh
+
+            if (!currentToken) {
+              return fetch(requestInput, init)
+            }
 
             // Build headers
             const headers = new Headers()
@@ -95,9 +99,13 @@ export async function KilocodeAuthPlugin(input: PluginInput): Promise<Hooks> {
               }
             }
 
-            // Set Kilo Code authorization
+            // Set Kilo Code authorization and required headers
             headers.set("Authorization", `Bearer ${currentToken}`)
-            headers.set("X-KILOCODE-EDITORNAME", "opencode")
+            headers.set("HTTP-Referer", "https://kilocode.ai")
+            headers.set("X-Title", "Kilo Code")
+            headers.set("X-KiloCode-Version", "3.16.3")
+            headers.set("User-Agent", "Kilo-Code/3.16.3")
+            headers.set("X-KiloCode-EditorName", "Visual Studio Code 1.96.0")
 
             // Add organization header if present
             if (currentKilocodeAuth.organizationId) {
@@ -130,81 +138,10 @@ export async function KilocodeAuthPlugin(input: PluginInput): Promise<Hooks> {
       },
 
       methods: [
-        {
-          type: "oauth",
-          label: "Login with Kilo Code",
-          async authorize() {
-            // Initiate device auth flow
-            const authData = await initiateDeviceAuth()
-            const { code, verificationUrl, expiresIn } = authData
-
-            // Open browser for user
-            openBrowserUrl(verificationUrl)
-
-            return {
-              url: verificationUrl,
-              instructions: `Enter code: ${code}`,
-              method: "auto" as const,
-
-              async callback() {
-                // Poll for authorization
-                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) {
-                      // Get profile for organization
-                      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
-                      }
-
-                      // Get default model
-                      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, // 1 year (tokens don't expire normally)
-                        // Store extra info in the auth
-                        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 }
-              },
-            }
-          },
-        },
+        // API key is the primary method - get it from https://app.kilo.ai
         {
           type: "api",
-          label: "Enter Kilo Code API Token",
+          label: "Enter API Key (get from app.kilo.ai)",
         },
       ],
     },
@@ -221,8 +158,6 @@ export async function KilocodeAuthPlugin(input: PluginInput): Promise<Hooks> {
 
 // Export utilities for external use
 export {
-  initiateDeviceAuth,
-  pollDeviceAuth,
   getKilocodeProfile,
   getKilocodeDefaultModel,
   getKilocodeModels,