opencode-gemini-auth 1.3.10 → 1.3.11

package/src/plugin/request/prepare.ts

@@ -0,0 +1,190 @@
+import { randomUUID } from "node:crypto";
+
+import { CODE_ASSIST_HEADERS, GEMINI_CODE_ASSIST_ENDPOINT } from "../../constants";
+import { normalizeThinkingConfig } from "../request-helpers";
+import { normalizeRequestPayloadIdentifiers, normalizeWrappedIdentifiers } from "./identifiers";
+import { addThoughtSignaturesToFunctionCalls, transformOpenAIToolCalls } from "./openai";
+import { isGenerativeLanguageRequest, toRequestUrlString } from "./shared";
+
+const STREAM_ACTION = "streamGenerateContent";
+const MODEL_FALLBACKS: Record<string, string> = {
+  "gemini-2.5-flash-image": "gemini-2.5-flash",
+};
+
+/**
+ * Rewrites OpenAI-style requests into Gemini Code Assist request shape.
+ */
+export function prepareGeminiRequest(
+  input: RequestInfo,
+  init: RequestInit | undefined,
+  accessToken: string,
+  projectId: string,
+): {
+  request: RequestInfo;
+  init: RequestInit;
+  streaming: boolean;
+  requestedModel?: string;
+} {
+  const baseInit: RequestInit = { ...init };
+  const headers = new Headers(init?.headers ?? {});
+
+  if (!isGenerativeLanguageRequest(input)) {
+    return {
+      request: input,
+      init: { ...baseInit, headers },
+      streaming: false,
+    };
+  }
+
+  headers.set("Authorization", `Bearer ${accessToken}`);
+  headers.delete("x-api-key");
+
+  const match = toRequestUrlString(input).match(/\/models\/([^:]+):(\w+)/);
+  if (!match) {
+    return {
+      request: input,
+      init: { ...baseInit, headers },
+      streaming: false,
+    };
+  }
+
+  const [, rawModel = "", rawAction = ""] = match;
+  const effectiveModel = MODEL_FALLBACKS[rawModel] ?? rawModel;
+  const streaming = rawAction === STREAM_ACTION;
+  const transformedUrl = `${GEMINI_CODE_ASSIST_ENDPOINT}/v1internal:${rawAction}${
+    streaming ? "?alt=sse" : ""
+  }`;
+
+  let body = baseInit.body;
+  let requestIdentifier: string = randomUUID();
+
+  if (typeof baseInit.body === "string" && baseInit.body) {
+    const transformed = transformRequestBody(baseInit.body, projectId, effectiveModel);
+    if (transformed.body) {
+      body = transformed.body;
+      requestIdentifier = transformed.userPromptId;
+    }
+  }
+
+  if (streaming) {
+    headers.set("Accept", "text/event-stream");
+  }
+
+  headers.set("User-Agent", CODE_ASSIST_HEADERS["User-Agent"]);
+  headers.set("X-Goog-Api-Client", CODE_ASSIST_HEADERS["X-Goog-Api-Client"]);
+  headers.set("Client-Metadata", CODE_ASSIST_HEADERS["Client-Metadata"]);
+  /**
+   * Request-scoped identifier used by Gemini CLI tooling and backend traces.
+   * We keep this aligned so quota/debug triage can correlate client and server events.
+   */
+  headers.set("x-activity-request-id", requestIdentifier);
+
+  return {
+    request: transformedUrl,
+    init: {
+      ...baseInit,
+      headers,
+      body,
+    },
+    streaming,
+    requestedModel: rawModel,
+  };
+}
+
+function transformRequestBody(
+  body: string,
+  projectId: string,
+  effectiveModel: string,
+): { body?: string; userPromptId: string } {
+  const fallbackId = randomUUID();
+  try {
+    const parsedBody = JSON.parse(body) as Record<string, unknown>;
+    const isWrapped = typeof parsedBody.project === "string" && "request" in parsedBody;
+
+    if (isWrapped) {
+      const wrappedBody = {
+        ...parsedBody,
+        model: effectiveModel,
+      } as Record<string, unknown>;
+      const { userPromptId } = normalizeWrappedIdentifiers(wrappedBody);
+      return { body: JSON.stringify(wrappedBody), userPromptId };
+    }
+
+    const requestPayload = { ...parsedBody };
+    transformOpenAIToolCalls(requestPayload);
+    addThoughtSignaturesToFunctionCalls(requestPayload);
+    normalizeThinking(requestPayload);
+    normalizeSystemInstruction(requestPayload);
+    normalizeCachedContent(requestPayload);
+
+    if ("model" in requestPayload) {
+      delete requestPayload.model;
+    }
+
+    const { userPromptId } = normalizeRequestPayloadIdentifiers(requestPayload);
+    const wrappedBody = {
+      project: projectId,
+      model: effectiveModel,
+      user_prompt_id: userPromptId,
+      request: requestPayload,
+    };
+
+    return { body: JSON.stringify(wrappedBody), userPromptId };
+  } catch (error) {
+    console.error("Failed to transform Gemini request body:", error);
+    return { userPromptId: fallbackId };
+  }
+}
+
+function normalizeThinking(requestPayload: Record<string, unknown>): void {
+  const rawGenerationConfig = requestPayload.generationConfig as Record<string, unknown> | undefined;
+  const normalizedThinking = normalizeThinkingConfig(rawGenerationConfig?.thinkingConfig);
+  if (normalizedThinking) {
+    if (rawGenerationConfig) {
+      rawGenerationConfig.thinkingConfig = normalizedThinking;
+      requestPayload.generationConfig = rawGenerationConfig;
+    } else {
+      requestPayload.generationConfig = { thinkingConfig: normalizedThinking };
+    }
+    return;
+  }
+
+  if (rawGenerationConfig?.thinkingConfig) {
+    delete rawGenerationConfig.thinkingConfig;
+    requestPayload.generationConfig = rawGenerationConfig;
+  }
+}
+
+function normalizeSystemInstruction(requestPayload: Record<string, unknown>): void {
+  if ("system_instruction" in requestPayload) {
+    requestPayload.systemInstruction = requestPayload.system_instruction;
+    delete requestPayload.system_instruction;
+  }
+}
+
+function normalizeCachedContent(requestPayload: Record<string, unknown>): void {
+  const extraBody =
+    requestPayload.extra_body && typeof requestPayload.extra_body === "object"
+      ? (requestPayload.extra_body as Record<string, unknown>)
+      : undefined;
+  const cachedContentFromExtra = extraBody?.cached_content ?? extraBody?.cachedContent;
+  const cachedContent =
+    (requestPayload.cached_content as string | undefined) ??
+    (requestPayload.cachedContent as string | undefined) ??
+    (cachedContentFromExtra as string | undefined);
+
+  if (cachedContent) {
+    requestPayload.cachedContent = cachedContent;
+  }
+
+  delete requestPayload.cached_content;
+  if (!extraBody) {
+    return;
+  }
+
+  delete extraBody.cached_content;
+  delete extraBody.cachedContent;
+  if (Object.keys(extraBody).length === 0) {
+    delete requestPayload.extra_body;
+  }
+}

package/src/plugin/request/response.ts

@@ -0,0 +1,191 @@
+import { logGeminiDebugResponse, type GeminiDebugContext } from "../debug";
+import {
+  enhanceGeminiErrorResponse,
+  extractUsageMetadata,
+  parseGeminiApiBody,
+  rewriteGeminiPreviewAccessError,
+  type GeminiApiBody,
+} from "../request-helpers";
+import { injectResponseIdFromTrace } from "./shared";
+
+/**
+ * Normalizes Gemini responses, preserving request metadata and usage counters.
+ */
+export async function transformGeminiResponse(
+  response: Response,
+  streaming: boolean,
+  debugContext?: GeminiDebugContext | null,
+  requestedModel?: string,
+): Promise<Response> {
+  const contentType = response.headers.get("content-type") ?? "";
+  const isJsonResponse = contentType.includes("application/json");
+  const isEventStreamResponse = contentType.includes("text/event-stream");
+
+  if (!isJsonResponse && !isEventStreamResponse) {
+    logGeminiDebugResponse(debugContext, response, {
+      note: "Non-JSON response (body omitted)",
+    });
+    return response;
+  }
+
+  try {
+    const headers = new Headers(response.headers);
+
+    if (streaming && response.ok && isEventStreamResponse && response.body) {
+      logGeminiDebugResponse(debugContext, response, {
+        note: "Streaming SSE payload (body omitted)",
+        headersOverride: headers,
+      });
+
+      return new Response(transformStreamingPayloadStream(response.body), {
+        status: response.status,
+        statusText: response.statusText,
+        headers,
+      });
+    }
+
+    const text = await response.text();
+    const init = {
+      status: response.status,
+      statusText: response.statusText,
+      headers,
+    };
+
+    const parsed: GeminiApiBody | null = !streaming || !isEventStreamResponse ? parseGeminiApiBody(text) : null;
+    const enhanced = !response.ok && parsed ? enhanceGeminiErrorResponse(parsed, response.status) : null;
+    if (enhanced?.retryAfterMs) {
+      const retryAfterSec = Math.ceil(enhanced.retryAfterMs / 1000).toString();
+      headers.set("Retry-After", retryAfterSec);
+      headers.set("retry-after-ms", String(enhanced.retryAfterMs));
+    }
+
+    const previewPatched = parsed
+      ? rewriteGeminiPreviewAccessError(enhanced?.body ?? parsed, response.status, requestedModel)
+      : null;
+
+    const effectiveBodyRaw = previewPatched ?? enhanced?.body ?? parsed ?? undefined;
+    const effectiveBody =
+      effectiveBodyRaw && typeof effectiveBodyRaw === "object"
+        ? injectResponseIdFromTrace(effectiveBodyRaw as Record<string, unknown>)
+        : effectiveBodyRaw;
+
+    attachUsageHeaders(headers, effectiveBody);
+
+    logGeminiDebugResponse(debugContext, response, {
+      body: text,
+      note: streaming ? "Streaming SSE payload (buffered)" : undefined,
+      headersOverride: headers,
+    });
+
+    if (!parsed) {
+      return new Response(text, init);
+    }
+
+    if (effectiveBody && typeof effectiveBody === "object" && "response" in effectiveBody) {
+      return new Response(JSON.stringify((effectiveBody as { response: unknown }).response), init);
+    }
+    if (previewPatched) {
+      return new Response(JSON.stringify(previewPatched), init);
+    }
+
+    return new Response(text, init);
+  } catch (error) {
+    logGeminiDebugResponse(debugContext, response, {
+      error,
+      note: "Failed to transform Gemini response",
+    });
+    console.error("Failed to transform Gemini response:", error);
+    return response;
+  }
+}
+
+function attachUsageHeaders(headers: Headers, effectiveBody: unknown): void {
+  if (!effectiveBody || typeof effectiveBody !== "object") {
+    return;
+  }
+  const usage = extractUsageMetadata(effectiveBody as GeminiApiBody);
+  if (usage?.cachedContentTokenCount === undefined) {
+    return;
+  }
+
+  headers.set("x-gemini-cached-content-token-count", String(usage.cachedContentTokenCount));
+  if (usage.totalTokenCount !== undefined) {
+    headers.set("x-gemini-total-token-count", String(usage.totalTokenCount));
+  }
+  if (usage.promptTokenCount !== undefined) {
+    headers.set("x-gemini-prompt-token-count", String(usage.promptTokenCount));
+  }
+  if (usage.candidatesTokenCount !== undefined) {
+    headers.set("x-gemini-candidates-token-count", String(usage.candidatesTokenCount));
+  }
+}
+
+function transformStreamingPayloadStream(stream: ReadableStream<Uint8Array>): ReadableStream<Uint8Array> {
+  const decoder = new TextDecoder();
+  const encoder = new TextEncoder();
+  let buffer = "";
+  let reader: ReadableStreamDefaultReader<Uint8Array> | null = null;
+
+  return new ReadableStream<Uint8Array>({
+    start(controller) {
+      reader = stream.getReader();
+      const pump = (): void => {
+        reader!
+          .read()
+          .then(({ done, value }) => {
+            if (done) {
+              buffer += decoder.decode();
+              if (buffer.length > 0) {
+                controller.enqueue(encoder.encode(transformStreamingLine(buffer)));
+              }
+              controller.close();
+              return;
+            }
+
+            buffer += decoder.decode(value, { stream: true });
+            let newlineIndex = buffer.indexOf("\n");
+            while (newlineIndex !== -1) {
+              const line = buffer.slice(0, newlineIndex);
+              buffer = buffer.slice(newlineIndex + 1);
+              const hasCr = line.endsWith("\r");
+              const rawLine = hasCr ? line.slice(0, -1) : line;
+              const transformed = transformStreamingLine(rawLine);
+              controller.enqueue(encoder.encode(`${transformed}${hasCr ? "\r\n" : "\n"}`));
+              newlineIndex = buffer.indexOf("\n");
+            }
+            pump();
+          })
+          .catch((error) => {
+            controller.error(error);
+          });
+      };
+      pump();
+    },
+    cancel(reason) {
+      if (reader) {
+        reader.cancel(reason).catch(() => {});
+      }
+    },
+  });
+}
+
+function transformStreamingLine(line: string): string {
+  if (!line.startsWith("data:")) {
+    return line;
+  }
+  const json = line.slice(5).trim();
+  if (!json) {
+    return line;
+  }
+
+  try {
+    const parsed = JSON.parse(json) as Record<string, unknown>;
+    const patched = injectResponseIdFromTrace(parsed);
+    if (patched.response !== undefined) {
+      return `data: ${JSON.stringify(patched.response)}`;
+    }
+  } catch {
+    return line;
+  }
+  return line;
+}

package/src/plugin/request/shared.ts

@@ -0,0 +1,72 @@
+/**
+ * Returns a URL string for supported RequestInfo inputs.
+ */
+export function toRequestUrlString(value: RequestInfo): string {
+  if (typeof value === "string") {
+    return value;
+  }
+  if (value instanceof URL) {
+    return value.toString();
+  }
+  const candidate = (value as Request).url;
+  if (candidate) {
+    return candidate;
+  }
+  return value.toString();
+}
+
+/**
+ * Detects Gemini/Generative Language API requests by URL.
+ */
+export function isGenerativeLanguageRequest(input: RequestInfo): input is string {
+  return toRequestUrlString(input).includes("generativelanguage.googleapis.com");
+}
+
+export function isRecord(value: unknown): value is Record<string, unknown> {
+  return !!value && typeof value === "object";
+}
+
+export function readString(value: unknown): string | undefined {
+  if (typeof value !== "string") {
+    return undefined;
+  }
+  const trimmed = value.trim();
+  return trimmed.length > 0 ? trimmed : undefined;
+}
+
+export function pickString(...values: unknown[]): string | undefined {
+  for (const value of values) {
+    const str = readString(value);
+    if (str) {
+      return str;
+    }
+  }
+  return undefined;
+}
+
+/**
+ * Preserves Cloud Code trace identity for downstream clients by mapping traceId -> responseId.
+ */
+export function injectResponseIdFromTrace<T extends Record<string, unknown>>(body: T): T {
+  const traceId = readString(body.traceId);
+  if (!traceId) {
+    return body;
+  }
+
+  const response = body.response;
+  if (!isRecord(response)) {
+    return body;
+  }
+
+  if (readString(response.responseId)) {
+    return body;
+  }
+
+  return {
+    ...body,
+    response: {
+      ...response,
+      responseId: traceId,
+    },
+  };
+}