@astrojs/db 0.0.0-db-export-bug-20240307130354

package/dist/core/tokens.js

@@ -0,0 +1,131 @@
+import { readFile } from "node:fs/promises";
+import { homedir } from "node:os";
+import { join } from "node:path";
+import { pathToFileURL } from "node:url";
+import { MISSING_PROJECT_ID_ERROR, MISSING_SESSION_ID_ERROR } from "./errors.js";
+import { getAstroStudioEnv, getAstroStudioUrl } from "./utils.js";
+const SESSION_LOGIN_FILE = pathToFileURL(join(homedir(), ".astro", "session-token"));
+const PROJECT_ID_FILE = pathToFileURL(join(process.cwd(), ".astro", "link"));
+class ManagedLocalAppToken {
+  token;
+  constructor(token) {
+    this.token = token;
+  }
+  async renew() {
+  }
+  async destroy() {
+  }
+}
+class ManagedRemoteAppToken {
+  token;
+  session;
+  projectId;
+  ttl;
+  renewTimer;
+  static async create(sessionToken, projectId) {
+    const response = await fetch(new URL(`${getAstroStudioUrl()}/auth/cli/token-create`), {
+      method: "POST",
+      headers: new Headers({
+        Authorization: `Bearer ${sessionToken}`
+      }),
+      body: JSON.stringify({ projectId })
+    });
+    const { token: shortLivedAppToken, ttl } = await response.json();
+    return new ManagedRemoteAppToken({
+      token: shortLivedAppToken,
+      session: sessionToken,
+      projectId,
+      ttl
+    });
+  }
+  constructor(options) {
+    this.token = options.token;
+    this.session = options.session;
+    this.projectId = options.projectId;
+    this.ttl = options.ttl;
+    this.renewTimer = setTimeout(() => this.renew(), 1e3 * 60 * 5 / 2);
+  }
+  async fetch(url, body) {
+    return fetch(`${getAstroStudioUrl()}${url}`, {
+      method: "POST",
+      headers: {
+        Authorization: `Bearer ${this.session}`,
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify(body)
+    });
+  }
+  async renew() {
+    clearTimeout(this.renewTimer);
+    delete this.renewTimer;
+    try {
+      const response = await this.fetch("/auth/cli/token-renew", {
+        token: this.token,
+        projectId: this.projectId
+      });
+      if (response.status === 200) {
+        this.renewTimer = setTimeout(() => this.renew(), 1e3 * 60 * this.ttl / 2);
+      } else {
+        throw new Error(`Unexpected response: ${response.status} ${response.statusText}`);
+      }
+    } catch (error) {
+      const retryIn = 60 * this.ttl / 10;
+      console.error(`Failed to renew token. Retrying in ${retryIn} seconds.`, error?.message);
+      this.renewTimer = setTimeout(() => this.renew(), retryIn * 1e3);
+    }
+  }
+  async destroy() {
+    try {
+      const response = await this.fetch("/auth/cli/token-delete", {
+        token: this.token,
+        projectId: this.projectId
+      });
+      if (response.status !== 200) {
+        throw new Error(`Unexpected response: ${response.status} ${response.statusText}`);
+      }
+    } catch (error) {
+      console.error("Failed to delete token.", error?.message);
+    }
+  }
+}
+async function getProjectIdFromFile() {
+  try {
+    return await readFile(PROJECT_ID_FILE, "utf-8");
+  } catch (error) {
+    return void 0;
+  }
+}
+async function getSessionIdFromFile() {
+  try {
+    return await readFile(SESSION_LOGIN_FILE, "utf-8");
+  } catch (error) {
+    return void 0;
+  }
+}
+async function getManagedAppTokenOrExit(token) {
+  if (token) {
+    return new ManagedLocalAppToken(token);
+  }
+  const { ASTRO_STUDIO_APP_TOKEN } = getAstroStudioEnv();
+  if (ASTRO_STUDIO_APP_TOKEN) {
+    return new ManagedLocalAppToken(ASTRO_STUDIO_APP_TOKEN);
+  }
+  const sessionToken = await getSessionIdFromFile();
+  if (!sessionToken) {
+    console.error(MISSING_SESSION_ID_ERROR);
+    process.exit(1);
+  }
+  const projectId = await getProjectIdFromFile();
+  if (!sessionToken || !projectId) {
+    console.error(MISSING_PROJECT_ID_ERROR);
+    process.exit(1);
+  }
+  return ManagedRemoteAppToken.create(sessionToken, projectId);
+}
+export {
+  PROJECT_ID_FILE,
+  SESSION_LOGIN_FILE,
+  getManagedAppTokenOrExit,
+  getProjectIdFromFile,
+  getSessionIdFromFile
+};