add more apis

Author: marandaneto

apps/array/package.json

@@ -94,6 +94,7 @@
     "@codemirror/view": "^6.38.8",
     "@dnd-kit/react": "^0.1.21",
     "@floating-ui/dom": "^1.7.4",
+    "@octokit/rest": "^22.0.1",
     "@parcel/watcher": "^2.5.1",
     "@phosphor-icons/react": "^2.1.10",
     "@posthog/agent": "workspace:*",
@@ -122,12 +123,12 @@
     "cmdk": "^1.1.1",
     "date-fns": "^3.3.1",
     "detect-libc": "^1.0.3",
-    "is-glob": "^4.0.3",
-    "micromatch": "^4.0.5",
     "electron-log": "^5.4.3",
     "electron-store": "^11.0.0",
     "file-icon": "^6.0.0",
     "idb-keyval": "^6.2.2",
+    "is-glob": "^4.0.3",
+    "micromatch": "^4.0.5",
     "node-addon-api": "^8.5.0",
     "node-machine-id": "^1.1.12",
     "node-pty": "1.1.0-beta39",

apps/array/src/api/githubClient.ts

@@ -0,0 +1,51 @@
+import { Octokit } from "@octokit/rest";
+
+export class GitHubAPIClient {
+  private octokit: Octokit;
+  private owner: string;
+  private repo: string;
+
+  constructor(accessToken: string, owner: string, repo: string) {
+    this.octokit = new Octokit({
+      auth: accessToken,
+    });
+    this.owner = owner;
+    this.repo = repo;
+  }
+
+  async createComment(issueNumber: number, body: string) {
+    try {
+      const response = await this.octokit.rest.issues.createComment({
+        owner: this.owner,
+        repo: this.repo,
+        issue_number: issueNumber,
+        body,
+      });
+      return response.data;
+    } catch (error) {
+      throw new Error(`Failed to create comment: ${error}`);
+    }
+  }
+
+  async getCurrentUser() {
+    try {
+      const response = await this.octokit.rest.users.getAuthenticated();
+      return response.data;
+    } catch (error) {
+      throw new Error(`Failed to get current user: ${error}`);
+    }
+  }
+
+  async getPullRequest(pullNumber: number) {
+    try {
+      const response = await this.octokit.rest.pulls.get({
+        owner: this.owner,
+        repo: this.repo,
+        pull_number: pullNumber,
+      });
+      return response.data;
+    } catch (error) {
+      throw new Error(`Failed to get pull request: ${error}`);
+    }
+  }
+}

apps/array/src/api/posthogClient.ts

@@ -370,4 +370,25 @@ export class PostHogAPIClient {
     });
     return data.results ?? [];
   }
+
+  async getGitHubAccessToken(
+    integrationId: string | number,
+  ): Promise<string | null> {
+    const teamId = await this.getTeamId();
+    const url = new URL(
+      `${this.api.baseUrl}/api/environments/${teamId}/integrations/${integrationId}/`,
+    );
+    const response = await this.api.fetcher.fetch({
+      method: "get",
+      url,
+      path: `/api/environments/${teamId}/integrations/${integrationId}/`,
+    });
+
+    if (!response.ok) {
+      throw new Error(`Failed to fetch integration: ${response.statusText}`);
+    }
+
+    const data = await response.json();
+    return data.access_token || null;
+  }
 }

apps/array/src/renderer/features/auth/stores/authStore.ts

@@ -1,10 +1,13 @@
+import { GitHubAPIClient } from "@api/githubClient";
 import { PostHogAPIClient } from "@api/posthogClient";
 import { identifyUser, resetUser, track } from "@renderer/lib/analytics";
 import { electronStorage } from "@renderer/lib/electronStorage";
 import { logger } from "@renderer/lib/logger";
 import { queryClient } from "@renderer/lib/queryClient";
+import { parseRepository } from "@renderer/utils/repository";
 import type { CloudRegion } from "@shared/types/oauth";
 import { useNavigationStore } from "@stores/navigationStore";
+import { repositoryWorkspaceStore } from "@stores/repositoryWorkspaceStore";
 import { create } from "zustand";
 import { persist } from "zustand/middleware";
 import {
@@ -36,6 +39,9 @@ interface AuthState {
   client: PostHogAPIClient | null;
   projectId: number | null; // Current team/project ID
 
+  // GitHub client
+  githubClient: GitHubAPIClient | null;
+
   // OpenAI API key (separate concern, kept for now)
   openaiApiKey: string | null;
   encryptedOpenaiKey: string | null;
@@ -50,6 +56,9 @@ interface AuthState {
   // Other methods
   setOpenAIKey: (apiKey: string) => Promise<void>;
   setDefaultWorkspace: (workspace: string) => void;
+  createGitHubClient: (accessToken: string, repository: string) => boolean;
+  initializeGitHubClient: (accessToken: string) => boolean;
+  initializeGitHubClientFromIntegration: () => Promise<boolean>;
   logout: () => void;
 }
 
@@ -70,6 +79,9 @@ export const useAuthStore = create<AuthState>()(
       client: null,
       projectId: null,
 
+      // GitHub client
+      githubClient: null,
+
       // OpenAI key
       openaiApiKey: null,
       encryptedOpenaiKey: null,
@@ -143,6 +155,13 @@ export const useAuthStore = create<AuthState>()(
             project_id: projectId.toString(),
             region,
           });
+
+          // Initialize GitHub client if available
+          try {
+            await get().initializeGitHubClientFromIntegration();
+          } catch (error) {
+            log.warn("Failed to initialize GitHub client during login:", error);
+          }
         } catch {
           throw new Error("Failed to authenticate with PostHog");
         }
@@ -316,6 +335,16 @@ export const useAuthStore = create<AuthState>()(
               region: tokens.cloudRegion,
             });
 
+            // Initialize GitHub client if available
+            try {
+              await get().initializeGitHubClientFromIntegration();
+            } catch (error) {
+              log.warn(
+                "Failed to initialize GitHub client during OAuth init:",
+                error,
+              );
+            }
+
             if (state.encryptedOpenaiKey) {
               const decryptedOpenaiKey =
                 await window.electronAPI.retrieveApiKey(
@@ -359,6 +388,73 @@ export const useAuthStore = create<AuthState>()(
       setDefaultWorkspace: (workspace: string) => {
         set({ defaultWorkspace: workspace });
       },
+
+      createGitHubClient: (accessToken: string, repository: string) => {
+        const parsed = parseRepository(repository);
+        if (!parsed) {
+          log.error("Invalid repository format:", repository);
+          return false;
+        }
+
+        const githubClient = new GitHubAPIClient(
+          accessToken,
+          parsed.organization,
+          parsed.repoName,
+        );
+        set({ githubClient });
+        return true;
+      },
+
+      initializeGitHubClient: (accessToken: string) => {
+        const { selectedRepository } = repositoryWorkspaceStore.getState();
+
+        if (!selectedRepository) {
+          log.warn("No repository selected, cannot create GitHub client");
+          return false;
+        }
+
+        return get().createGitHubClient(accessToken, selectedRepository);
+      },
+
+      initializeGitHubClientFromIntegration: async () => {
+        const { client } = get();
+
+        if (!client) {
+          log.warn(
+            "No PostHog client available, cannot fetch GitHub integration",
+          );
+          return false;
+        }
+
+        try {
+          const integrations = await client.getIntegrations();
+          const githubIntegration = integrations.find(
+            (i: any) => i.kind === "github",
+          );
+
+          if (!githubIntegration) {
+            log.warn("No GitHub integration found");
+            return false;
+          }
+
+          const accessToken = await client.getGitHubAccessToken(
+            githubIntegration.id,
+          );
+
+          if (!accessToken) {
+            log.warn("No access token found for GitHub integration");
+            return false;
+          }
+
+          return get().initializeGitHubClient(accessToken);
+        } catch (error) {
+          log.error(
+            "Failed to initialize GitHub client from integration:",
+            error,
+          );
+          return false;
+        }
+      },
       logout: () => {
         track(ANALYTICS_EVENTS.USER_LOGGED_OUT);
         resetUser();
@@ -381,6 +477,7 @@ export const useAuthStore = create<AuthState>()(
           isAuthenticated: false,
           client: null,
           projectId: null,
+          githubClient: null,
           openaiApiKey: null,
           encryptedOpenaiKey: null,
         });