replicas-engine 0.1.17 → 0.1.18

package/dist/src/index.js

@@ -384,7 +384,7 @@ function convertCodexEvent(event, linearSessionId) {
 
 // src/utils/git.ts
 import { execSync } from "child_process";
-var cachedPrUrl = null;
+var cachedPr = null;
 function runGitCommand(command, cwd) {
   return execSync(command, {
     cwd,
@@ -443,16 +443,17 @@ function getGitDiff(cwd) {
   }
 }
 function getPullRequestUrl(cwd) {
-  if (cachedPrUrl) {
-    return cachedPrUrl;
-  }
   try {
-    const branch = getCurrentBranch(cwd);
-    if (!branch) {
+    const currentBranch = getCurrentBranch(cwd);
+    if (!currentBranch) {
       return null;
     }
+    if (cachedPr && cachedPr.branch === currentBranch) {
+      return cachedPr.prUrl;
+    }
+    cachedPr = null;
     try {
-      const remoteRef = execSync(`git ls-remote --heads origin ${branch}`, {
+      const remoteRef = execSync(`git ls-remote --heads origin ${currentBranch}`, {
         cwd,
         encoding: "utf-8",
         stdio: ["pipe", "pipe", "pipe"]
@@ -470,8 +471,11 @@ function getPullRequestUrl(cwd) {
         stdio: ["pipe", "pipe", "pipe"]
       }).trim();
       if (prInfo) {
-        cachedPrUrl = prInfo;
-        return cachedPrUrl;
+        cachedPr = {
+          prUrl: prInfo,
+          branch: currentBranch
+        };
+        return cachedPr.prUrl;
       }
     } catch {
       return null;
@@ -1300,9 +1304,186 @@ var GitHubTokenManager = class {
 };
 var githubTokenManager = new GitHubTokenManager();
 
+// src/services/claude-token-manager.ts
+import { promises as fs2 } from "fs";
+import path2 from "path";
+var ClaudeTokenManager = class {
+  refreshInterval = null;
+  REFRESH_INTERVAL_MS = 45 * 60 * 1e3;
+  // 45 minutes
+  async start() {
+    const monolithUrl = process.env.MONOLITH_URL;
+    const workspaceId = process.env.WORKSPACE_ID;
+    const engineSecret = process.env.REPLICAS_ENGINE_SECRET;
+    if (!monolithUrl || !workspaceId || !engineSecret) {
+      console.log("[ClaudeTokenManager] Skipping: missing MONOLITH_URL, WORKSPACE_ID, or REPLICAS_ENGINE_SECRET");
+      return;
+    }
+    console.log("[ClaudeTokenManager] Starting token refresh service");
+    await this.refreshCredentials();
+    this.refreshInterval = setInterval(() => {
+      this.refreshCredentials().catch((error) => {
+        console.error("[ClaudeTokenManager] Scheduled refresh failed:", error);
+      });
+    }, this.REFRESH_INTERVAL_MS);
+    console.log("[ClaudeTokenManager] Token refresh scheduled every 45 minutes");
+  }
+  stop() {
+    if (this.refreshInterval) {
+      clearInterval(this.refreshInterval);
+      this.refreshInterval = null;
+      console.log("[ClaudeTokenManager] Stopped");
+    }
+  }
+  async refreshCredentials() {
+    const monolithUrl = process.env.MONOLITH_URL;
+    const workspaceId = process.env.WORKSPACE_ID;
+    const engineSecret = process.env.REPLICAS_ENGINE_SECRET;
+    if (!monolithUrl || !workspaceId || !engineSecret) {
+      console.log("[ClaudeTokenManager] Refresh skipped: missing configuration");
+      return;
+    }
+    console.log("[ClaudeTokenManager] Refreshing Claude credentials...");
+    try {
+      const response = await fetch(`${monolithUrl}/v1/engine/claude/refresh-credentials`, {
+        method: "POST",
+        headers: {
+          "Authorization": `Bearer ${engineSecret}`,
+          "X-Workspace-Id": workspaceId,
+          "Content-Type": "application/json"
+        }
+      });
+      if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(`Credentials refresh failed: ${response.status} ${errorText}`);
+      }
+      const data = await response.json();
+      await this.updateClaudeCredentials(data);
+      console.log(`[ClaudeTokenManager] Credentials refreshed successfully, expires at ${data.expiresAt}`);
+    } catch (error) {
+      console.error("[ClaudeTokenManager] Failed to refresh credentials:", error);
+    }
+  }
+  async updateClaudeCredentials(credentials) {
+    const workspaceHome = process.env.WORKSPACE_HOME || process.env.HOME;
+    if (!workspaceHome) {
+      console.warn("[ClaudeTokenManager] No WORKSPACE_HOME or HOME set, skipping credentials update");
+      return;
+    }
+    const claudeDir = path2.join(workspaceHome, ".claude");
+    const credentialsPath = path2.join(claudeDir, ".credentials.json");
+    const claudeCliConfig = {
+      claudeAiOauth: {
+        accessToken: credentials.accessToken,
+        refreshToken: credentials.refreshToken,
+        expiresAt: new Date(credentials.expiresAt).getTime(),
+        scopes: credentials.scopes,
+        subscriptionType: credentials.subscriptionType
+      }
+    };
+    try {
+      await fs2.mkdir(claudeDir, { recursive: true, mode: 448 });
+      await fs2.writeFile(credentialsPath, JSON.stringify(claudeCliConfig, null, 2), { mode: 384 });
+      console.log(`[ClaudeTokenManager] Updated ${credentialsPath}`);
+    } catch (error) {
+      console.error("[ClaudeTokenManager] Failed to update credentials file:", error);
+    }
+  }
+};
+var claudeTokenManager = new ClaudeTokenManager();
+
+// src/services/codex-token-manager.ts
+import { promises as fs3 } from "fs";
+import path3 from "path";
+var CodexTokenManager = class {
+  refreshInterval = null;
+  REFRESH_INTERVAL_MS = 45 * 60 * 1e3;
+  // 45 minutes
+  async start() {
+    const monolithUrl = process.env.MONOLITH_URL;
+    const workspaceId = process.env.WORKSPACE_ID;
+    const engineSecret = process.env.REPLICAS_ENGINE_SECRET;
+    if (!monolithUrl || !workspaceId || !engineSecret) {
+      console.log("[CodexTokenManager] Skipping: missing MONOLITH_URL, WORKSPACE_ID, or REPLICAS_ENGINE_SECRET");
+      return;
+    }
+    console.log("[CodexTokenManager] Starting token refresh service");
+    await this.refreshCredentials();
+    this.refreshInterval = setInterval(() => {
+      this.refreshCredentials().catch((error) => {
+        console.error("[CodexTokenManager] Scheduled refresh failed:", error);
+      });
+    }, this.REFRESH_INTERVAL_MS);
+    console.log("[CodexTokenManager] Token refresh scheduled every 45 minutes");
+  }
+  stop() {
+    if (this.refreshInterval) {
+      clearInterval(this.refreshInterval);
+      this.refreshInterval = null;
+      console.log("[CodexTokenManager] Stopped");
+    }
+  }
+  async refreshCredentials() {
+    const monolithUrl = process.env.MONOLITH_URL;
+    const workspaceId = process.env.WORKSPACE_ID;
+    const engineSecret = process.env.REPLICAS_ENGINE_SECRET;
+    if (!monolithUrl || !workspaceId || !engineSecret) {
+      console.log("[CodexTokenManager] Refresh skipped: missing configuration");
+      return;
+    }
+    console.log("[CodexTokenManager] Refreshing Codex credentials...");
+    try {
+      const response = await fetch(`${monolithUrl}/v1/engine/codex/refresh-credentials`, {
+        method: "POST",
+        headers: {
+          "Authorization": `Bearer ${engineSecret}`,
+          "X-Workspace-Id": workspaceId,
+          "Content-Type": "application/json"
+        }
+      });
+      if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(`Credentials refresh failed: ${response.status} ${errorText}`);
+      }
+      const data = await response.json();
+      await this.updateCodexCredentials(data);
+      console.log(`[CodexTokenManager] Credentials refreshed successfully, expires at ${data.expiresAt}`);
+    } catch (error) {
+      console.error("[CodexTokenManager] Failed to refresh credentials:", error);
+    }
+  }
+  async updateCodexCredentials(credentials) {
+    const workspaceHome = process.env.WORKSPACE_HOME || process.env.HOME;
+    if (!workspaceHome) {
+      console.warn("[CodexTokenManager] No WORKSPACE_HOME or HOME set, skipping credentials update");
+      return;
+    }
+    const codexDir = path3.join(workspaceHome, ".codex");
+    const authPath = path3.join(codexDir, "auth.json");
+    const codexAuthConfig = {
+      OPENAI_API_KEY: null,
+      tokens: {
+        id_token: credentials.idToken,
+        access_token: credentials.accessToken,
+        refresh_token: credentials.refreshToken,
+        account_id: credentials.accountId
+      },
+      last_refresh: (/* @__PURE__ */ new Date()).toISOString()
+    };
+    try {
+      await fs3.mkdir(codexDir, { recursive: true, mode: 448 });
+      await fs3.writeFile(authPath, JSON.stringify(codexAuthConfig, null, 2), { mode: 384 });
+      console.log(`[CodexTokenManager] Updated ${authPath}`);
+    } catch (error) {
+      console.error("[CodexTokenManager] Failed to update credentials file:", error);
+    }
+  }
+};
+var codexTokenManager = new CodexTokenManager();
+
 // src/services/git-init.ts
 import { existsSync } from "fs";
-import path2 from "path";
+import path4 from "path";
 var initializedBranch = null;
 function findAvailableBranchName(baseName, cwd) {
   if (!branchExists(baseName, cwd)) {
@@ -1336,7 +1517,7 @@ async function initializeGitRepository() {
       error: "No WORKSPACE_NAME environment variable set"
     };
   }
-  const repoPath = path2.join(workspaceHome, "workspaces", repoName);
+  const repoPath = path4.join(workspaceHome, "workspaces", repoName);
   if (!existsSync(repoPath)) {
     console.log(`[GitInit] Repository directory does not exist: ${repoPath}`);
     console.log("[GitInit] Waiting for initializer to clone the repository...");
@@ -1345,7 +1526,7 @@ async function initializeGitRepository() {
       branch: null
     };
   }
-  if (!existsSync(path2.join(repoPath, ".git"))) {
+  if (!existsSync(path4.join(repoPath, ".git"))) {
     return {
       success: false,
       branch: null,
@@ -1468,6 +1649,8 @@ serve(
       console.warn(`Git initialization warning: ${gitResult.error}`);
     }
     await githubTokenManager.start();
+    await claudeTokenManager.start();
+    await codexTokenManager.start();
     const monolithService2 = new MonolithService();
     await monolithService2.sendEvent({ type: "workspace_ready", payload: {} });
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "replicas-engine",
-  "version": "0.1.17",
+  "version": "0.1.18",
   "description": "Lightweight API server for Replicas workspaces",
   "type": "module",
   "main": "dist/src/index.js",