opencode-antigravity-auth 1.0.2 → 1.0.3

package/src/plugin/debug.ts

@@ -1,198 +0,0 @@
-import { createWriteStream } from "node:fs";
-import { join } from "node:path";
-import { cwd, env } from "node:process";
-
-const DEBUG_FLAG = env.OPENCODE_ANTIGRAVITY_DEBUG ?? "";
-const MAX_BODY_PREVIEW_CHARS = 12000;
-const debugEnabled = DEBUG_FLAG.trim() === "1";
-const logFilePath = debugEnabled ? defaultLogFilePath() : undefined;
-const logWriter = createLogWriter(logFilePath);
-
-export interface AntigravityDebugContext {
-  id: string;
-  streaming: boolean;
-  startedAt: number;
-}
-
-interface AntigravityDebugRequestMeta {
-  originalUrl: string;
-  resolvedUrl: string;
-  method?: string;
-  headers?: HeadersInit;
-  body?: BodyInit | null;
-  streaming: boolean;
-  projectId?: string;
-}
-
-interface AntigravityDebugResponseMeta {
-  body?: string;
-  note?: string;
-  error?: unknown;
-  headersOverride?: HeadersInit;
-}
-
-let requestCounter = 0;
-
-/**
- * Begins a debug trace for an Antigravity request, logging request metadata when debugging is enabled.
- */
-export function startAntigravityDebugRequest(meta: AntigravityDebugRequestMeta): AntigravityDebugContext | null {
-  if (!debugEnabled) {
-    return null;
-  }
-
-  const id = `ANTIGRAVITY-${++requestCounter}`;
-  const method = meta.method ?? "GET";
-  logDebug(`[Antigravity Debug ${id}] ${method} ${meta.resolvedUrl}`);
-  if (meta.originalUrl && meta.originalUrl !== meta.resolvedUrl) {
-    logDebug(`[Antigravity Debug ${id}] Original URL: ${meta.originalUrl}`);
-  }
-  if (meta.projectId) {
-    logDebug(`[Antigravity Debug ${id}] Project: ${meta.projectId}`);
-  }
-  logDebug(`[Antigravity Debug ${id}] Streaming: ${meta.streaming ? "yes" : "no"}`);
-  logDebug(`[Antigravity Debug ${id}] Headers: ${JSON.stringify(maskHeaders(meta.headers))}`);
-  const bodyPreview = formatBodyPreview(meta.body);
-  if (bodyPreview) {
-    logDebug(`[Antigravity Debug ${id}] Body Preview: ${bodyPreview}`);
-  }
-
-  return { id, streaming: meta.streaming, startedAt: Date.now() };
-}
-
-/**
- * Logs response details for a previously started debug trace when debugging is enabled.
- */
-export function logAntigravityDebugResponse(
-  context: AntigravityDebugContext | null | undefined,
-  response: Response,
-  meta: AntigravityDebugResponseMeta = {},
-): void {
-  if (!debugEnabled || !context) {
-    return;
-  }
-
-  const durationMs = Date.now() - context.startedAt;
-  logDebug(
-    `[Antigravity Debug ${context.id}] Response ${response.status} ${response.statusText} (${durationMs}ms)`,
-  );
-  logDebug(
-    `[Antigravity Debug ${context.id}] Response Headers: ${JSON.stringify(
-      maskHeaders(meta.headersOverride ?? response.headers),
-    )}`,
-  );
-
-  if (meta.note) {
-    logDebug(`[Antigravity Debug ${context.id}] Note: ${meta.note}`);
-  }
-
-  if (meta.error) {
-    logDebug(`[Antigravity Debug ${context.id}] Error: ${formatError(meta.error)}`);
-  }
-
-  if (meta.body) {
-    logDebug(
-      `[Antigravity Debug ${context.id}] Response Body Preview: ${truncateForLog(meta.body)}`,
-    );
-  }
-}
-
-/**
- * Obscures sensitive headers and returns a plain object for logging.
- */
-function maskHeaders(headers?: HeadersInit | Headers): Record<string, string> {
-  if (!headers) {
-    return {};
-  }
-
-  const result: Record<string, string> = {};
-  const parsed = headers instanceof Headers ? headers : new Headers(headers);
-  parsed.forEach((value, key) => {
-    if (key.toLowerCase() === "authorization") {
-      result[key] = "[redacted]";
-    } else {
-      result[key] = value;
-    }
-  });
-  return result;
-}
-
-/**
- * Produces a short, type-aware preview of a request/response body for logs.
- */
-function formatBodyPreview(body?: BodyInit | null): string | undefined {
-  if (body == null) {
-    return undefined;
-  }
-
-  if (typeof body === "string") {
-    return truncateForLog(body);
-  }
-
-  if (body instanceof URLSearchParams) {
-    return truncateForLog(body.toString());
-  }
-
-  if (typeof Blob !== "undefined" && body instanceof Blob) {
-    return `[Blob size=${body.size}]`;
-  }
-
-  if (typeof FormData !== "undefined" && body instanceof FormData) {
-    return "[FormData payload omitted]";
-  }
-
-  return `[${body.constructor?.name ?? typeof body} payload omitted]`;
-}
-
-/**
- * Truncates long strings to a fixed preview length for logging.
- */
-function truncateForLog(text: string): string {
-  if (text.length <= MAX_BODY_PREVIEW_CHARS) {
-    return text;
-  }
-  return `${text.slice(0, MAX_BODY_PREVIEW_CHARS)}... (truncated ${text.length - MAX_BODY_PREVIEW_CHARS} chars)`;
-}
-
-/**
- * Writes a single debug line using the configured writer.
- */
-function logDebug(line: string): void {
-  logWriter(line);
-}
-
-/**
- * Converts unknown error-like values into printable strings.
- */
-function formatError(error: unknown): string {
-  if (error instanceof Error) {
-    return error.stack ?? error.message;
-  }
-  try {
-    return JSON.stringify(error);
-  } catch {
-    return String(error);
-  }
-}
-
-/**
- * Builds a timestamped log file path in the current working directory.
- */
-function defaultLogFilePath(): string {
-  const timestamp = new Date().toISOString().replace(/[:.]/g, "-");
-  return join(cwd(), `antigravity-debug-${timestamp}.log`);
-}
-
-/**
- * Creates a line writer that appends to a file when provided.
- */
-function createLogWriter(filePath?: string): (line: string) => void {
-  if (!filePath) {
-    return () => {};
-  }
-
-  const stream = createWriteStream(filePath, { flags: "a" });
-  return (line: string) => {
-    stream.write(`${line}\n`);
-  };
-}

package/src/plugin/project.ts

@@ -1,334 +0,0 @@
-import {
-  ANTIGRAVITY_HEADERS,
-  ANTIGRAVITY_ENDPOINT_FALLBACKS,
-  ANTIGRAVITY_LOAD_ENDPOINTS,
-  ANTIGRAVITY_DEFAULT_PROJECT_ID,
-} from "../constants";
-import { formatRefreshParts, parseRefreshParts } from "./auth";
-import type {
-  OAuthAuthDetails,
-  PluginClient,
-  ProjectContextResult,
-} from "./types";
-
-const projectContextResultCache = new Map<string, ProjectContextResult>();
-const projectContextPendingCache = new Map<string, Promise<ProjectContextResult>>();
-
-const CODE_ASSIST_METADATA = {
-  ideType: "IDE_UNSPECIFIED",
-  platform: "PLATFORM_UNSPECIFIED",
-  pluginType: "GEMINI",
-} as const;
-
-interface AntigravityUserTier {
-  id?: string;
-  isDefault?: boolean;
-  userDefinedCloudaicompanionProject?: boolean;
-}
-
-interface LoadCodeAssistPayload {
-  cloudaicompanionProject?: string;
-  currentTier?: {
-    id?: string;
-  };
-  allowedTiers?: AntigravityUserTier[];
-}
-
-interface OnboardUserPayload {
-  done?: boolean;
-  response?: {
-    cloudaicompanionProject?: {
-      id?: string;
-    };
-  };
-}
-
-class ProjectIdRequiredError extends Error {
-  /**
-   * Error raised when a required Google Cloud project is missing during Antigravity onboarding.
-   */
-  constructor() {
-    super(
-      "Google Antigravity requires a Google Cloud project. Enable the Antigravity API on a project you control, rerun `opencode auth login`, and supply that project ID when prompted.",
-    );
-  }
-}
-
-/**
- * Builds metadata headers required by the Code Assist API.
- */
-function buildMetadata(projectId?: string): Record<string, string> {
-  const metadata: Record<string, string> = {
-    ideType: CODE_ASSIST_METADATA.ideType,
-    platform: CODE_ASSIST_METADATA.platform,
-    pluginType: CODE_ASSIST_METADATA.pluginType,
-  };
-  if (projectId) {
-    metadata.duetProject = projectId;
-  }
-  return metadata;
-}
-
-/**
- * Selects the default tier ID from the allowed tiers list.
- */
-function getDefaultTierId(allowedTiers?: AntigravityUserTier[]): string | undefined {
-  if (!allowedTiers || allowedTiers.length === 0) {
-    return undefined;
-  }
-  for (const tier of allowedTiers) {
-    if (tier?.isDefault) {
-      return tier.id;
-    }
-  }
-  return allowedTiers[0]?.id;
-}
-
-/**
- * Promise-based delay utility.
- */
-function wait(ms: number): Promise<void> {
-  return new Promise(function (resolve) {
-    setTimeout(resolve, ms);
-  });
-}
-
-/**
- * Extracts the cloudaicompanion project id from loadCodeAssist responses.
- */
-function extractManagedProjectId(payload: LoadCodeAssistPayload | null): string | undefined {
-  if (!payload) {
-    return undefined;
-  }
-  if (typeof payload.cloudaicompanionProject === "string") {
-    return payload.cloudaicompanionProject;
-  }
-  if (payload.cloudaicompanionProject && typeof payload.cloudaicompanionProject.id === "string") {
-    return payload.cloudaicompanionProject.id;
-  }
-  return undefined;
-}
-
-/**
- * Generates a cache key for project context based on refresh token.
- */
-function getCacheKey(auth: OAuthAuthDetails): string | undefined {
-  const refresh = auth.refresh?.trim();
-  return refresh ? refresh : undefined;
-}
-
-/**
- * Clears cached project context results and pending promises, globally or for a refresh key.
- */
-export function invalidateProjectContextCache(refresh?: string): void {
-  if (!refresh) {
-    projectContextPendingCache.clear();
-    projectContextResultCache.clear();
-    return;
-  }
-  projectContextPendingCache.delete(refresh);
-  projectContextResultCache.delete(refresh);
-}
-
-/**
- * Loads managed project information for the given access token and optional project.
- */
-export async function loadManagedProject(
-  accessToken: string,
-  projectId?: string,
-): Promise<LoadCodeAssistPayload | null> {
-  const metadata = buildMetadata(projectId);
-  const requestBody: Record<string, unknown> = { metadata };
-
-  const loadHeaders: Record<string, string> = {
-    "Content-Type": "application/json",
-    Authorization: `Bearer ${accessToken}`,
-    "User-Agent": "google-api-nodejs-client/9.15.1",
-    "X-Goog-Api-Client": "google-cloud-sdk vscode_cloudshelleditor/0.1",
-    "Client-Metadata": ANTIGRAVITY_HEADERS["Client-Metadata"],
-  };
-
-  const loadEndpoints = Array.from(
-    new Set<string>([...ANTIGRAVITY_LOAD_ENDPOINTS, ...ANTIGRAVITY_ENDPOINT_FALLBACKS]),
-  );
-
-  for (const baseEndpoint of loadEndpoints) {
-    try {
-      const response = await fetch(
-        `${baseEndpoint}/v1internal:loadCodeAssist`,
-        {
-          method: "POST",
-          headers: loadHeaders,
-          body: JSON.stringify(requestBody),
-        },
-      );
-
-      if (!response.ok) {
-        continue;
-      }
-
-      return (await response.json()) as LoadCodeAssistPayload;
-    } catch (error) {
-      console.error(`Failed to load Antigravity managed project via ${baseEndpoint}:`, error);
-      continue;
-    }
-  }
-
-  return null;
-}
-
-
-/**
- * Onboards a managed project for the user, optionally retrying until completion.
- */
-export async function onboardManagedProject(
-  accessToken: string,
-  tierId: string,
-  projectId?: string,
-  attempts = 10,
-  delayMs = 5000,
-): Promise<string | undefined> {
-  const metadata = buildMetadata(projectId);
-  const requestBody: Record<string, unknown> = {
-    tierId,
-    metadata,
-  };
-
-  if (tierId !== "FREE") {
-    if (!projectId) {
-      throw new ProjectIdRequiredError();
-    }
-    requestBody.cloudaicompanionProject = projectId;
-  }
-
-  for (const baseEndpoint of ANTIGRAVITY_ENDPOINT_FALLBACKS) {
-    for (let attempt = 0; attempt < attempts; attempt += 1) {
-      try {
-        const response = await fetch(
-          `${baseEndpoint}/v1internal:onboardUser`,
-          {
-            method: "POST",
-            headers: {
-              "Content-Type": "application/json",
-              Authorization: `Bearer ${accessToken}`,
-              ...ANTIGRAVITY_HEADERS,
-            },
-            body: JSON.stringify(requestBody),
-          },
-        );
-
-        if (!response.ok) {
-          break;
-        }
-
-        const payload = (await response.json()) as OnboardUserPayload;
-        const managedProjectId = payload.response?.cloudaicompanionProject?.id;
-        if (payload.done && managedProjectId) {
-          return managedProjectId;
-        }
-        if (payload.done && projectId) {
-          return projectId;
-        }
-      } catch (error) {
-        console.error(
-          `Failed to onboard Antigravity managed project via ${baseEndpoint}:`,
-          error,
-        );
-        break;
-      }
-
-      await wait(delayMs);
-    }
-  }
-
-  return undefined;
-}
-
-/**
- * Resolves an effective project ID for the current auth state, caching results per refresh token.
- */
-export async function ensureProjectContext(
-  auth: OAuthAuthDetails,
-  client: PluginClient,
-  providerId: string,
-): Promise<ProjectContextResult> {
-  const accessToken = auth.access;
-  if (!accessToken) {
-    return { auth, effectiveProjectId: "" };
-  }
-
-  const cacheKey = getCacheKey(auth);
-  if (cacheKey) {
-    const cached = projectContextResultCache.get(cacheKey);
-    if (cached) {
-      return cached;
-    }
-    const pending = projectContextPendingCache.get(cacheKey);
-    if (pending) {
-      return pending;
-    }
-  }
-
-  const resolveContext = async (): Promise<ProjectContextResult> => {
-    const parts = parseRefreshParts(auth.refresh);
-    if (parts.managedProjectId) {
-      return { auth, effectiveProjectId: parts.managedProjectId };
-    }
-
-    const fallbackProjectId = ANTIGRAVITY_DEFAULT_PROJECT_ID;
-    const persistManagedProject = async (managedProjectId: string): Promise<ProjectContextResult> => {
-      const updatedAuth: OAuthAuthDetails = {
-        ...auth,
-        refresh: formatRefreshParts({
-          refreshToken: parts.refreshToken,
-          projectId: parts.projectId,
-          managedProjectId,
-        }),
-      };
-
-      await client.auth.set({
-        path: { id: providerId },
-        body: updatedAuth,
-      });
-
-      return { auth: updatedAuth, effectiveProjectId: managedProjectId };
-    };
-
-    // Try to resolve a managed project from Antigravity if possible.
-    const loadPayload = await loadManagedProject(accessToken, parts.projectId ?? fallbackProjectId);
-    const resolvedManagedProjectId = extractManagedProjectId(loadPayload);
-
-    if (resolvedManagedProjectId) {
-      return persistManagedProject(resolvedManagedProjectId);
-    }
-
-    if (parts.projectId) {
-      return { auth, effectiveProjectId: parts.projectId };
-    }
-
-    // No project id present in auth; fall back to the hardcoded id for requests.
-    return { auth, effectiveProjectId: fallbackProjectId };
-  };
-
-  if (!cacheKey) {
-    return resolveContext();
-  }
-
-  const promise = resolveContext()
-    .then((result) => {
-      const nextKey = getCacheKey(result.auth) ?? cacheKey;
-      projectContextPendingCache.delete(cacheKey);
-      projectContextResultCache.set(nextKey, result);
-      if (nextKey !== cacheKey) {
-        projectContextResultCache.delete(cacheKey);
-      }
-      return result;
-    })
-    .catch((error) => {
-      projectContextPendingCache.delete(cacheKey);
-      throw error;
-    });
-
-  projectContextPendingCache.set(cacheKey, promise);
-  return promise;
-}