feat: add GitHub Copilot integration

0xtbug · 0xtbug · commit cdfb2f87a365 · 2026-02-09T13:21:50.000+07:00
This commit introduces comprehensive support for GitHub Copilot, including:

- **UI Integration**:
 - Adds a new `copilot.png` icon for GitHub Copilot.
 - Updates `ProvidersPage` to display GitHub Copilot as an available provider.
 - Implements the device code flow for Copilot authentication, showing the user code and verification URL.
 - Adds Copilot-specific fields to `ProviderState` for managing the device flow.
 - Updates `getProviderIconPath` to correctly display the Copilot icon.

- **Quota Management**:
 - Extends `QuotaPage` to recognize and display GitHub Copilot quota information.
 - Adds `copilot` to `getProviderType` and `ProviderSection` for proper categorization.
 - Implements `fetchCopilot` in `quotaApi` to retrieve Copilot entitlement details.
 - Adds `COPILOT_ENTITLEMENT_URL` and `COPILOT_HEADERS` for API requests.
 - Introduces `parseCopilotQuota` to interpret the Copilot entitlement response, including plan type and usage snapshots.

- **API Enhancements**:
 - Adds `copilot` to `OAuthProvider` type.
 - Updates `oauthApi.startAuth` to use a `AUTH_URL_PROVIDER_MAP` for mapping `copilot` to `github` endpoint.
 - Extends `OAuthStartResponse` with `user_code` and `verification_uri` for device flow.

- **General**:
 - Bumps package version to `1.0.8`.
 - Adds `copilot` to the `PROVIDERS` list in `src/types/index.ts`.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "zero-limit",
-  "version": "1.0.7",
+  "version": "1.0.8",
   "private": true,
   "type": "module",
   "scripts": {
diff --git a/public/copilot/copilot.png b/public/copilot/copilot.png
diff --git a/src/pages/ProvidersPage.tsx b/src/pages/ProvidersPage.tsx
@@ -47,6 +47,11 @@ interface ProviderState {
   url?: string;
   state?: string;
   error?: string;
+  // Copilot device flow specific
+  userCode?: string;
+  deviceCode?: string;
+  expiresIn?: number;
+  interval?: number;
 }
 
 // --- Helpers ---
@@ -79,6 +84,7 @@ function getProviderIconPath(providerId: string): string {
   if (id.includes('gemini')) return '/gemini/gemini.png';
   if (id.includes('codex') || id.includes('openai')) return '/openai/openai.png';
   if (id.includes('kiro')) return '/kiro/kiro.png';
+  if (id.includes('copilot') || id.includes('github')) return '/copilot/copilot.png';
   return '';
 }
 
@@ -231,6 +237,69 @@ export function ProvidersPage() {
         return;
       }
 
+      // Copilot uses device code flow via backend
+      if (providerId === 'copilot') {
+        try {
+          const response = await oauthApi.startAuth('copilot');
+          const url = response.url || response.verification_uri;
+          const state = response.state;
+          const userCode = response.user_code;
+
+          if (!url) {
+            throw new Error('No verification URL returned from server');
+          }
+
+          updateProviderState(providerId, {
+            status: 'polling',
+            url,
+            state,
+            userCode
+          });
+
+          // Open verification URL
+          await openInBrowser(url);
+
+          // Poll for auth status if we have a state
+          if (state) {
+            startPolling(providerId, state);
+          } else {
+            // Fallback: poll for new auth files
+            const initialFiles = files.filter(f =>
+              f.provider?.toLowerCase().includes('github') ||
+              f.filename?.toLowerCase().includes('github')
+            );
+            const initialCount = initialFiles.length;
+
+            const pollTimer = window.setInterval(async () => {
+              try {
+                const listResponse = await authFilesApi.list();
+                const currentFiles = listResponse?.files ?? [];
+                const currentGithubFiles = currentFiles.filter(f =>
+                  f.provider?.toLowerCase().includes('github') ||
+                  f.filename?.toLowerCase().includes('github')
+                );
+
+                if (currentGithubFiles.length > initialCount) {
+                  updateProviderState(providerId, { status: 'success' });
+                  stopPolling(providerId);
+                  setSelectedProvider(null);
+                  setFiles(currentFiles);
+                }
+              } catch {
+                // Ignore polling errors
+              }
+            }, 3000);
+            pollingTimers.current[providerId] = pollTimer;
+          }
+        } catch (err) {
+          updateProviderState(providerId, {
+            status: 'error',
+            error: (err as Error).message
+          });
+        }
+        return;
+      }
+
       const response = await oauthApi.startAuth(providerId, options);
       const url = response.url || response.auth_url;
       const state = response.state;
@@ -506,6 +575,28 @@ export function ProvidersPage() {
                                     {/* Normal Auth Flow (Polling/Waiting) - Only if NOT Error and NOT Idle */}
                                     {state.status !== 'idle' && state.status !== 'error' && (
                                         <>
+                                            {/* Copilot Device Code Display */}
+                                            {provider.id === 'copilot' && state.userCode && (
+                                              <div className="space-y-3">
+                                                <p className="text-sm text-muted-foreground">
+                                                  {t('providers.copilotInstructions', 'Enter the code below at GitHub:')}
+                                                </p>
+                                                <div className="flex items-center justify-center gap-3">
+                                                  <code className="text-3xl font-mono font-bold tracking-widest bg-muted px-4 py-2 rounded-lg">
+                                                    {state.userCode}
+                                                  </code>
+                                                  <Button
+                                                    variant="outline"
+                                                    size="sm"
+                                                    onClick={() => copyToClipboard(state.userCode!)}
+                                                    title={t('common.copy')}
+                                                  >
+                                                    <ClipboardCopy className="h-4 w-4" />
+                                                  </Button>
+                                                </div>
+                                              </div>
+                                            )}
+
                                             <div className="flex items-center gap-2 text-sm text-muted-foreground">
                                                 <Loader2 className="h-4 w-4 animate-spin" />
                                                 <span>{t('providers.completeLogin')}</span>
diff --git a/src/pages/QuotaPage.tsx b/src/pages/QuotaPage.tsx
@@ -50,21 +50,23 @@ interface ProviderSection {
 }
 
 // Identify provider from filename - with null safety
-function getProviderType(file: AuthFile): 'antigravity' | 'codex' | 'gemini-cli' | 'kiro' | 'unknown' {
+function getProviderType(file: AuthFile): 'antigravity' | 'codex' | 'gemini-cli' | 'kiro' | 'copilot' | 'unknown' {
   // Handle missing filename
   const filename = (file?.filename || file?.id || '').toLowerCase();
 
   if (filename.startsWith('antigravity-') || filename.includes('antigravity')) return 'antigravity';
   if (filename.startsWith('codex-') || filename.includes('codex')) return 'codex';
   if (filename.startsWith('gemini-cli-') || filename.includes('gemini')) return 'gemini-cli';
   if (filename.startsWith('kiro-') || filename.includes('kiro')) return 'kiro';
+  if (filename.startsWith('github-copilot-') || filename.includes('copilot')) return 'copilot';
 
   // Fallback to provider field
   const provider = (file?.provider || '').toLowerCase();
   if (provider.includes('antigravity')) return 'antigravity';
   if (provider.includes('codex')) return 'codex';
   if (provider.includes('gemini')) return 'gemini-cli';
   if (provider.includes('kiro')) return 'kiro';
+  if (provider.includes('copilot') || provider.includes('github')) return 'copilot';
 
   return 'unknown';
 }
@@ -101,6 +103,7 @@ export function QuotaPage() {
         codex: [],
         'gemini-cli': [],
         'kiro': [],
+        'copilot': [],
       };
 
       files.forEach((file) => {
@@ -123,6 +126,7 @@ export function QuotaPage() {
         { provider: 'codex', displayName: 'Codex (OpenAI)', files: grouped.codex },
         { provider: 'gemini-cli', displayName: 'Gemini CLI', files: grouped['gemini-cli'] },
         { provider: 'kiro', displayName: 'Kiro (CodeWhisperer)', files: grouped['kiro'] },
+        { provider: 'copilot', displayName: 'GitHub Copilot', files: grouped['copilot'] },
       ]);
 
       // Auto-fetch quota for all files
@@ -256,6 +260,19 @@ export function QuotaPage() {
             error: result.error
           } : f)
         } : s));
+      } else if (targetProvider === 'copilot') {
+        const result = await quotaApi.fetchCopilot(authIndex);
+
+        setSections((prev) => prev.map(s => s.provider === 'copilot' ? {
+          ...s,
+          files: s.files.map(f => f.fileId === fileId ? {
+            ...f,
+            loading: false,
+            plan: result.plan,
+            models: result.models,
+            error: result.error
+          } : f)
+        } : s));
       }
     } catch (err) {
       const msg = (err as Error).message;
@@ -278,6 +295,7 @@ export function QuotaPage() {
         if (key === 'codex') return '/openai/openai.png'; // Assuming Codex uses OpenAI icon
         if (key === 'gemini-cli') return '/gemini/gemini.png';
         if (key === 'kiro') return '/kiro/kiro.png';
+        if (key === 'copilot') return '/copilot/copilot.png';
         return undefined;
     };
 
diff --git a/src/services/api/oauth.ts b/src/services/api/oauth.ts
@@ -9,6 +9,9 @@ const WEBUI_SUPPORTED: OAuthProvider[] = ['codex', 'anthropic', 'antigravity', '
 const CALLBACK_PROVIDER_MAP: Partial<Record<OAuthProvider, string>> = {
   'gemini-cli': 'gemini'
 };
+const AUTH_URL_PROVIDER_MAP: Partial<Record<OAuthProvider, string>> = {
+  'copilot': 'github'
+};
 
 export const oauthApi = {
   startAuth: (provider: OAuthProvider, options?: { projectId?: string }) => {
@@ -19,7 +22,8 @@ export const oauthApi = {
     if (provider === 'gemini-cli' && options?.projectId) {
       params.project_id = options.projectId;
     }
-    return apiClient.get<OAuthStartResponse>(`/${provider}-auth-url`, {
+    const endpointProvider = AUTH_URL_PROVIDER_MAP[provider] ?? provider;
+    return apiClient.get<OAuthStartResponse>(`/${endpointProvider}-auth-url`, {
       params: Object.keys(params).length ? params : undefined
     });
   },
diff --git a/src/services/api/quota.ts b/src/services/api/quota.ts
@@ -15,6 +15,7 @@ const ANTIGRAVITY_QUOTA_URLS = [
 const GEMINI_CLI_QUOTA_URL = 'https://cloudcode-pa.googleapis.com/v1internal:retrieveUserQuota';
 const CODEX_USAGE_URL = 'https://chatgpt.com/backend-api/wham/usage';
 const KIRO_USAGE_URL = 'https://codewhisperer.us-east-1.amazonaws.com/getUsageLimits?isEmailRequired=true&origin=AI_EDITOR&resourceType=AGENTIC_REQUEST';
+const COPILOT_ENTITLEMENT_URL = 'https://api.github.com/copilot_internal/user';
 
 // Headers
 const ANTIGRAVITY_HEADERS = {
@@ -41,6 +42,12 @@ const KIRO_HEADERS = {
   'x-amz-user-agent': 'aws-sdk-js/3.0.0'
 };
 
+const COPILOT_HEADERS = {
+  Authorization: 'Bearer $TOKEN$',
+  Accept: 'application/vnd.github+json',
+  'X-GitHub-Api-Version': '2022-11-28'
+};
+
 import {
   parseCodexUsagePayload,
   formatCodexResetLabel,
@@ -90,6 +97,13 @@ export interface KiroQuotaResult {
   error?: string;
 }
 
+export interface CopilotQuotaResult {
+  models: QuotaModel[];
+  plan?: string;
+  username?: string;
+  error?: string;
+}
+
 // Parse Antigravity models response
 function parseAntigravityModels(body: unknown): QuotaModel[] {
   const models: QuotaModel[] = [];
@@ -398,6 +412,96 @@ function parseKiroQuota(body: unknown): KiroQuotaResult {
   return { models, plan, email };
 }
 
+// Parse Copilot entitlement response
+function parseCopilotQuota(body: unknown): CopilotQuotaResult {
+  const payload = body as Record<string, unknown> | null;
+  if (!payload) return { models: [] };
+
+  const models: QuotaModel[] = [];
+
+  // Extract plan type
+  const accessTypeSku = (payload.access_type_sku as string) || '';
+  const copilotPlan = (payload.copilot_plan as string) || '';
+  let plan = 'Unknown';
+
+  const sku = accessTypeSku.toLowerCase();
+  const planLower = copilotPlan.toLowerCase();
+
+  if (sku.includes('enterprise') || planLower === 'enterprise') {
+    plan = 'Enterprise';
+  } else if (sku.includes('business') || planLower === 'business') {
+    plan = 'Business';
+  } else if (sku.includes('educational') || sku.includes('pro') || planLower.includes('pro')) {
+    plan = 'Pro';
+  } else if (planLower === 'individual' && !sku.includes('free_limited')) {
+    plan = 'Pro';
+  } else if (sku.includes('free_limited') || sku === 'free' || planLower.includes('free')) {
+    plan = 'Free';
+  } else if (copilotPlan) {
+    plan = copilotPlan.charAt(0).toUpperCase() + copilotPlan.slice(1);
+  }
+
+  // Parse reset date
+  const resetDateStr = (payload.quota_reset_date_utc as string) || (payload.quota_reset_date as string) || (payload.limited_user_reset_date as string);
+  let resetTime: string | undefined;
+  if (resetDateStr) {
+    resetTime = formatResetTime(resetDateStr);
+  }
+
+  // Method 1: Parse quota_snapshots (used by paid plans)
+  const quotaSnapshots = payload.quota_snapshots as Record<string, unknown> | undefined;
+  if (quotaSnapshots) {
+    const parseSnapshot = (name: string, snapshot: unknown, defaultTotal: number) => {
+      const snap = snapshot as Record<string, unknown> | undefined;
+      if (!snap || snap.unlimited === true) return;
+
+      let percentage = 100;
+      if (typeof snap.percent_remaining === 'number') {
+        percentage = Math.min(100, Math.max(0, snap.percent_remaining));
+      } else {
+        const remaining = (snap.remaining as number) ?? 0;
+        const total = (snap.entitlement as number) ?? defaultTotal;
+        if (total > 0) {
+          percentage = Math.min(100, Math.max(0, (remaining / total) * 100));
+        }
+      }
+
+      models.push({ name, percentage: Math.round(percentage), resetTime });
+    };
+
+    parseSnapshot('Chat', quotaSnapshots.chat, 50);
+    parseSnapshot('Completions', quotaSnapshots.completions, 2000);
+    parseSnapshot('Premium', quotaSnapshots.premium_interactions, 50);
+  }
+
+  // Method 2: Parse limited_user_quotas + monthly_quotas (for free/individual plans)
+  if (models.length === 0) {
+    const limitedQuotas = payload.limited_user_quotas as Record<string, unknown> | undefined;
+    const monthlyQuotas = payload.monthly_quotas as Record<string, unknown> | undefined;
+
+    if (limitedQuotas && monthlyQuotas) {
+      const parseLimit = (name: string, remainingKey: string, totalKey: string) => {
+        const remaining = (limitedQuotas[remainingKey] as number) ?? 0;
+        const total = (monthlyQuotas[totalKey] as number) ?? 0;
+        if (total > 0) {
+          const percentage = Math.min(100, Math.max(0, (remaining / total) * 100));
+          models.push({ name, percentage: Math.round(percentage), resetTime });
+        }
+      };
+
+      parseLimit('Chat', 'chat', 'chat');
+      parseLimit('Completions', 'completions', 'completions');
+    }
+  }
+
+  // Fallback if no quota info found
+  if (models.length === 0) {
+    models.push({ name: 'Copilot', percentage: 100, resetTime: undefined });
+  }
+
+  return { models, plan };
+}
+
 export const quotaApi = {
   /**
    * Fetch Antigravity quota for an auth file
@@ -508,6 +612,33 @@ export const quotaApi = {
         };
       }
 
+      return { models: [], error: formatQuotaError(result) };
+    } catch (err) {
+      return { models: [], error: (err as Error).message };
+    }
+  },
+
+  /**
+   * Fetch GitHub Copilot quota for an auth file
+   */
+  async fetchCopilot(authIndex: string): Promise<CopilotQuotaResult> {
+    try {
+      const result = await apiCallApi.request({
+        authIndex,
+        method: 'GET',
+        url: COPILOT_ENTITLEMENT_URL,
+        header: { ...COPILOT_HEADERS }
+      });
+
+      if (result.statusCode >= 200 && result.statusCode < 300) {
+        return parseCopilotQuota(result.body);
+      }
+
+      // Handle 401/403 - token expired or no subscription
+      if (result.statusCode === 401 || result.statusCode === 403) {
+        return { models: [], error: 'Token invalid or no Copilot subscription' };
+      }
+
       return { models: [], error: formatQuotaError(result) };
     } catch (err) {
       return { models: [], error: (err as Error).message };
diff --git a/src/types/api.ts b/src/types/api.ts
@@ -9,12 +9,15 @@ export interface Config {
 }
 
 // OAuth types
-export type OAuthProvider = 'codex' | 'anthropic' | 'antigravity' | 'gemini-cli' | 'kiro';
+export type OAuthProvider = 'codex' | 'anthropic' | 'antigravity' | 'gemini-cli' | 'kiro' | 'copilot';
 
 export interface OAuthStartResponse {
   url?: string;
   auth_url?: string;
   state?: string;
+  // Device flow fields (for GitHub Copilot)
+  user_code?: string;
+  verification_uri?: string;
 }
 
 export interface OAuthCallbackResponse {
diff --git a/src/types/index.ts b/src/types/index.ts

Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,6 @@`
`1`	`1`	`{`
`2`	`2`	`"name": "zero-limit",`
`3`		`- "version": "1.0.7",`
	`3`	`+ "version": "1.0.8",`
`4`	`4`	`"private": true,`
`5`	`5`	`"type": "module",`
`6`	`6`	`"scripts": {`
Original file line number	Diff line number	Diff line change
`@@ -9,12 +9,15 @@ export interface Config {`
`9`	`9`	`}`
`10`	`10`
`11`	`11`	`// OAuth types`
`12`		`-export type OAuthProvider = 'codex' \| 'anthropic' \| 'antigravity' \| 'gemini-cli' \| 'kiro';`
	`12`	`+export type OAuthProvider = 'codex' \| 'anthropic' \| 'antigravity' \| 'gemini-cli' \| 'kiro' \| 'copilot';`
`13`	`13`
`14`	`14`	`export interface OAuthStartResponse {`
`15`	`15`	`url?: string;`
`16`	`16`	`auth_url?: string;`
`17`	`17`	`state?: string;`
	`18`	`+ // Device flow fields (for GitHub Copilot)`
	`19`	`+ user_code?: string;`
	`20`	`+ verification_uri?: string;`
`18`	`21`	`}`
`19`	`22`
`20`	`23`	`export interface OAuthCallbackResponse {`