repowise 0.1.49 → 0.1.52

package/dist/bin/repowise.js

@@ -73,6 +73,10 @@ import { readFile as readFile2, writeFile as writeFile2, mkdir as mkdir2, chmod,
 import http from "http";
 import { homedir as homedir2 } from "os";
 import { join as join2 } from "path";
+function getCognitoConfigForStorage() {
+  const { domain, clientId, region, customDomain } = getCognitoConfig();
+  return { domain, clientId, region, customDomain };
+}
 function getCognitoConfig() {
   const env = getEnvConfig();
   return {
@@ -259,6 +263,7 @@ async function getValidCredentials() {
   if (Date.now() > creds.expiresAt - 5 * 60 * 1e3) {
     try {
       const refreshed = await refreshTokens(creds.refreshToken);
+      refreshed.cognito = creds.cognito;
       await storeCredentials(refreshed);
       return refreshed;
     } catch {
@@ -288,6 +293,8 @@ Open this URL in your browser to authenticate:
     throw new Error("State mismatch \u2014 possible CSRF attack. Please try again.");
   }
   const credentials = await exchangeCodeForTokens(code, codeVerifier);
+  const { domain, clientId, region, customDomain } = getCognitoConfig();
+  credentials.cognito = { domain, clientId, region, customDomain };
   await storeCredentials(credentials);
   return credentials;
 }
@@ -1810,6 +1817,7 @@ Waiting for authentication...`);
     }
     spinner.start("Exchanging authorization code...");
     const credentials = await exchangeCodeForTokens(code, codeVerifier);
+    credentials.cognito = getCognitoConfigForStorage();
     await storeCredentials(credentials);
     const { email } = decodeIdToken(credentials.idToken);
     spinner.succeed(chalk6.green(`Logged in as ${chalk6.bold(email)}`));
@@ -2182,24 +2190,27 @@ var init_state = __esm({
 import { readFile as readFile8, writeFile as writeFile7, mkdir as mkdir7, chmod as chmod3 } from "fs/promises";
 import { homedir as homedir8 } from "os";
 import { join as join11 } from "path";
-function getCognitoConfig2() {
-  return {
-    domain: process.env["REPOWISE_COGNITO_DOMAIN"] ?? "auth-repowise-dev",
-    clientId: process.env["REPOWISE_COGNITO_CLIENT_ID"] ?? "",
-    region: process.env["REPOWISE_COGNITO_REGION"] ?? "us-east-1"
-  };
-}
-function getTokenUrl2() {
-  const { domain, region } = getCognitoConfig2();
-  return `https://${domain}.auth.${region}.amazoncognito.com/oauth2/token`;
-}
-async function refreshTokens2(refreshToken) {
-  const response = await fetch(getTokenUrl2(), {
+function getTokenUrl2(creds) {
+  const cognito = creds?.cognito;
+  const domain = process.env["REPOWISE_COGNITO_DOMAIN"] ?? cognito?.domain ?? "auth-repowise-dev";
+  const region = process.env["REPOWISE_COGNITO_REGION"] ?? cognito?.region ?? "us-east-1";
+  const customDomain = cognito?.customDomain ?? false;
+  return customDomain ? `https://${domain}/oauth2/token` : `https://${domain}.auth.${region}.amazoncognito.com/oauth2/token`;
+}
+function getClientId(creds) {
+  return process.env["REPOWISE_COGNITO_CLIENT_ID"] ?? creds?.cognito?.clientId ?? "";
+}
+async function refreshTokens2(refreshToken, creds) {
+  const clientId = getClientId(creds);
+  if (!clientId) {
+    throw new Error("No Cognito client ID available. Run `repowise login` first.");
+  }
+  const response = await fetch(getTokenUrl2(creds), {
     method: "POST",
     headers: { "Content-Type": "application/x-www-form-urlencoded" },
     body: new URLSearchParams({
       grant_type: "refresh_token",
-      client_id: getCognitoConfig2().clientId,
+      client_id: clientId,
       refresh_token: refreshToken
     })
   });
@@ -2211,7 +2222,8 @@ async function refreshTokens2(refreshToken) {
     accessToken: data.access_token,
     refreshToken,
     idToken: data.id_token,
-    expiresAt: Date.now() + data.expires_in * 1e3
+    expiresAt: Date.now() + data.expires_in * 1e3,
+    cognito: creds.cognito
   };
 }
 async function getStoredCredentials2() {
@@ -2236,7 +2248,7 @@ async function getValidCredentials2() {
     return null;
   if (Date.now() > creds.expiresAt - 5 * 60 * 1e3) {
     try {
-      const refreshed = await refreshTokens2(creds.refreshToken);
+      const refreshed = await refreshTokens2(creds.refreshToken, creds);
       await storeCredentials2(refreshed);
       return refreshed;
     } catch {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "repowise",
-  "version": "0.1.49",
+  "version": "0.1.52",
   "type": "module",
   "description": "AI-optimized codebase context generator",
   "bin": {