feat: add pkce

This commit is contained in:
Frank
2025-01-15 08:20:20 +11:00
parent 6b15f827d8
commit a102ba520b
3 changed files with 47 additions and 5 deletions

View File

@@ -1273,6 +1273,33 @@ export class ClineProvider implements vscode.WebviewViewProvider {
return cacheDir return cacheDir
} }
async handleGlamaCallback(code: string) {
let apiKey: string
try {
const response = await axios.post("https://glama.ai/api/gateway/v1/auth/exchange-code", { code })
if (response.data && response.data.apiKey) {
apiKey = response.data.apiKey
} else {
throw new Error("Invalid response from Glama API")
}
} catch (error) {
console.error("Error exchanging code for API key:", error)
throw error
}
const glama: ApiProvider = "glama"
await this.updateGlobalState("apiProvider", glama)
await this.storeSecret("glamaApiKey", apiKey)
await this.postStateToWebview()
if (this.cline) {
this.cline.api = buildApiHandler({
apiProvider: glama,
glamaApiKey: apiKey,
})
}
// await this.postMessageToWebview({ type: "action", action: "settingsButtonClicked" }) // bad ux if user is on welcome
}
async readGlamaModels(): Promise<Record<string, ModelInfo> | undefined> { async readGlamaModels(): Promise<Record<string, ModelInfo> | undefined> {
const glamaModelsFilePath = path.join( const glamaModelsFilePath = path.join(
await this.ensureCacheDirectoryExists(), await this.ensureCacheDirectoryExists(),

View File

@@ -139,6 +139,14 @@ export function activate(context: vscode.ExtensionContext) {
return return
} }
switch (path) { switch (path) {
case "/glama": {
const code = query.get("code")
if (code) {
await visibleProvider.handleGlamaCallback(code)
}
break
}
case "/openrouter": { case "/openrouter": {
const code = query.get("code") const code = query.get("code")
if (code) { if (code) {

View File

@@ -209,11 +209,12 @@ const ApiOptions = ({ apiErrorMessage, modelIdErrorMessage }: ApiOptionsProps) =
<span style={{ fontWeight: 500 }}>Glama API Key</span> <span style={{ fontWeight: 500 }}>Glama API Key</span>
</VSCodeTextField> </VSCodeTextField>
{!apiConfiguration?.glamaApiKey && ( {!apiConfiguration?.glamaApiKey && (
<VSCodeLink <VSCodeButtonLink
href="https://glama.ai/settings/api-keys" href={getGlamaAuthUrl(uriScheme)}
style={{ display: "inline", fontSize: "inherit" }}> style={{ margin: "5px 0 0 0" }}
You can get an Glama API key by signing up here. appearance="secondary">
</VSCodeLink> Get Glama API Key
</VSCodeButtonLink>
)} )}
<p <p
style={{ style={{
@@ -739,6 +740,12 @@ const ApiOptions = ({ apiErrorMessage, modelIdErrorMessage }: ApiOptionsProps) =
) )
} }
export function getGlamaAuthUrl(uriScheme?: string) {
const callbackUrl = `${uriScheme || "vscode"}://saoudrizwan.claude-dev/glama`
return `https://glama.ai/oauth/authorize?callback_url=${encodeURIComponent(callbackUrl)}`
}
export function getOpenRouterAuthUrl(uriScheme?: string) { export function getOpenRouterAuthUrl(uriScheme?: string) {
return `https://openrouter.ai/auth?callback_url=${uriScheme || "vscode"}://saoudrizwan.claude-dev/openrouter` return `https://openrouter.ai/auth?callback_url=${uriScheme || "vscode"}://saoudrizwan.claude-dev/openrouter`
} }