@astrojs/db 0.4.1 → 0.6.0

package/dist/core/cli/commands/execute/index.d.ts

@@ -0,0 +1,8 @@
+import type { AstroConfig } from 'astro';
+import type { Arguments } from 'yargs-parser';
+import { type DBConfig } from '../../../types.js';
+export declare function cmd({ astroConfig, dbConfig, flags, }: {
+    astroConfig: AstroConfig;
+    dbConfig: DBConfig;
+    flags: Arguments;
+}): Promise<void>;

package/dist/core/cli/commands/execute/index.js

@@ -0,0 +1,32 @@
+import { existsSync } from "node:fs";
+import { FILE_NOT_FOUND_ERROR, MISSING_EXECUTE_PATH_ERROR } from "../../../errors.js";
+import { getStudioVirtualModContents } from "../../../integration/vite-plugin-db.js";
+import { bundleFile, importBundledFile } from "../../../load-file.js";
+import { getManagedAppTokenOrExit } from "../../../tokens.js";
+import {} from "../../../types.js";
+async function cmd({
+  astroConfig,
+  dbConfig,
+  flags
+}) {
+  const filePath = flags._[4];
+  if (typeof filePath !== "string") {
+    console.error(MISSING_EXECUTE_PATH_ERROR);
+    process.exit(1);
+  }
+  const fileUrl = new URL(filePath, astroConfig.root);
+  if (!existsSync(fileUrl)) {
+    console.error(FILE_NOT_FOUND_ERROR(filePath));
+    process.exit(1);
+  }
+  const appToken = await getManagedAppTokenOrExit(flags.token);
+  const virtualModContents = getStudioVirtualModContents({
+    tables: dbConfig.tables ?? {},
+    appToken: appToken.token
+  });
+  const { code } = await bundleFile({ virtualModContents, root: astroConfig.root, fileUrl });
+  await importBundledFile({ code, root: astroConfig.root });
+}
+export {
+  cmd
+};

package/dist/core/cli/commands/link/index.d.ts

@@ -1,8 +1,20 @@
-import type { AstroConfig } from 'astro';
-import type { Arguments } from 'yargs-parser';
-export declare function cmd({ flags }: {
-    config: AstroConfig;
-    flags: Arguments;
-}): Promise<void>;
-export declare function promptProjectName(defaultName?: string): Promise<string>;
-export declare function promptWorkspaceName(defaultName?: string): Promise<string>;
+export declare function cmd(): Promise<void>;
+export declare function createNewProject({ workspaceId, name, region, }: {
+    workspaceId: string;
+    name: string;
+    region: string;
+}): Promise<{
+    id: string;
+    idName: string;
+}>;
+export declare function promptExistingProjectName({ workspaceId }: {
+    workspaceId: string;
+}): Promise<{
+    id: string;
+    idName: string;
+}>;
+export declare function promptBegin(): Promise<void>;
+export declare function promptLinkExisting(): Promise<boolean>;
+export declare function promptLinkNew(): Promise<boolean>;
+export declare function promptNewProjectName(): Promise<string>;
+export declare function promptNewProjectRegion(): Promise<string>;

package/dist/core/cli/commands/link/index.js

@@ -1,29 +1,94 @@
 import { mkdir, writeFile } from "node:fs/promises";
+import { homedir } from "node:os";
+import { basename } from "node:path";
+import { slug } from "github-slugger";
 import { bgRed, cyan } from "kleur/colors";
+import ora from "ora";
 import prompts from "prompts";
+import { MISSING_SESSION_ID_ERROR } from "../../../errors.js";
 import { PROJECT_ID_FILE, getSessionIdFromFile } from "../../../tokens.js";
 import { getAstroStudioUrl } from "../../../utils.js";
-import { MISSING_SESSION_ID_ERROR } from "../../../errors.js";
-async function cmd({ flags }) {
-  const linkUrl = new URL(getAstroStudioUrl() + "/auth/cli/link");
+async function cmd() {
   const sessionToken = await getSessionIdFromFile();
   if (!sessionToken) {
     console.error(MISSING_SESSION_ID_ERROR);
     process.exit(1);
   }
-  let body = { id: flags._[4] };
-  if (!body.id) {
-    const workspaceIdName = await promptWorkspaceName();
-    const projectIdName = await promptProjectName();
-    body = { projectIdName, workspaceIdName };
+  const getWorkspaceIdAsync = getWorkspaceId();
+  await promptBegin();
+  const isLinkExisting = await promptLinkExisting();
+  if (isLinkExisting) {
+    const workspaceId = await getWorkspaceIdAsync;
+    const existingProjectData = await promptExistingProjectName({ workspaceId });
+    return await linkProject(existingProjectData.id);
+  }
+  const isLinkNew = await promptLinkNew();
+  if (isLinkNew) {
+    const workspaceId = await getWorkspaceIdAsync;
+    const newProjectName = await promptNewProjectName();
+    const newProjectRegion = await promptNewProjectRegion();
+    const spinner = ora("Creating new project...").start();
+    const newProjectData = await createNewProject({
+      workspaceId,
+      name: newProjectName,
+      region: newProjectRegion
+    });
+    await new Promise((r) => setTimeout(r, 4e3));
+    spinner.succeed("Project created!");
+    return await linkProject(newProjectData.id);
   }
+}
+async function linkProject(id) {
+  await mkdir(new URL(".", PROJECT_ID_FILE), { recursive: true });
+  await writeFile(PROJECT_ID_FILE, `${id}`);
+  console.info("Project linked.");
+}
+async function getWorkspaceId() {
+  const linkUrl = new URL(getAstroStudioUrl() + "/api/cli/workspaces.list");
+  const response = await fetch(linkUrl, {
+    method: "POST",
+    headers: {
+      Authorization: `Bearer ${await getSessionIdFromFile()}`,
+      "Content-Type": "application/json"
+    }
+  });
+  if (!response.ok) {
+    if (response.status === 401) {
+      console.error(
+        `${bgRed("Unauthorized")}
+
+  Are you logged in?
+  Run ${cyan(
+          "astro db login"
+        )} to authenticate and then try linking again.
+
+`
+      );
+      process.exit(1);
+    }
+    console.error(`Failed to fetch user workspace: ${response.status} ${response.statusText}`);
+    process.exit(1);
+  }
+  const { data, success } = await response.json();
+  if (!success) {
+    console.error(`Failed to fetch user's workspace.`);
+    process.exit(1);
+  }
+  return data[0].id;
+}
+async function createNewProject({
+  workspaceId,
+  name,
+  region
+}) {
+  const linkUrl = new URL(getAstroStudioUrl() + "/api/cli/projects.create");
   const response = await fetch(linkUrl, {
     method: "POST",
     headers: {
       Authorization: `Bearer ${await getSessionIdFromFile()}`,
       "Content-Type": "application/json"
     },
-    body: JSON.stringify(body)
+    body: JSON.stringify({ workspaceId, name, region })
   });
   if (!response.ok) {
     if (response.status === 401) {
@@ -39,40 +104,135 @@ async function cmd({ flags }) {
       );
       process.exit(1);
     }
-    console.error(`Failed to link project: ${response.status} ${response.statusText}`);
+    console.error(`Failed to create project: ${response.status} ${response.statusText}`);
     process.exit(1);
   }
-  const { data } = await response.json();
-  await mkdir(new URL(".", PROJECT_ID_FILE), { recursive: true });
-  await writeFile(PROJECT_ID_FILE, `${data.id}`);
-  console.info("Project linked.");
+  const { data, success } = await response.json();
+  if (!success) {
+    console.error(`Failed to create project.`);
+    process.exit(1);
+  }
+  return { id: data.id, idName: data.idName };
 }
-async function promptProjectName(defaultName) {
-  const { projectName } = await prompts({
-    type: "text",
-    name: "projectName",
-    message: "Project ID",
-    initial: defaultName
+async function promptExistingProjectName({ workspaceId }) {
+  const linkUrl = new URL(getAstroStudioUrl() + "/api/cli/projects.list");
+  const response = await fetch(linkUrl, {
+    method: "POST",
+    headers: {
+      Authorization: `Bearer ${await getSessionIdFromFile()}`,
+      "Content-Type": "application/json"
+    },
+    body: JSON.stringify({ workspaceId })
+  });
+  if (!response.ok) {
+    if (response.status === 401) {
+      console.error(
+        `${bgRed("Unauthorized")}
+
+  Are you logged in?
+  Run ${cyan(
+          "astro db login"
+        )} to authenticate and then try linking again.
+
+`
+      );
+      process.exit(1);
+    }
+    console.error(`Failed to fetch projects: ${response.status} ${response.statusText}`);
+    process.exit(1);
+  }
+  const { data, success } = await response.json();
+  if (!success) {
+    console.error(`Failed to fetch projects.`);
+    process.exit(1);
+  }
+  const { projectId } = await prompts({
+    type: "autocomplete",
+    name: "projectId",
+    message: "What is your project name?",
+    limit: 5,
+    choices: data.map((p) => ({ title: p.name, value: p.id }))
   });
-  if (typeof projectName !== "string") {
+  if (typeof projectId !== "string") {
+    console.log("Canceled.");
     process.exit(0);
   }
-  return projectName;
+  const selectedProjectData = data.find((p) => p.id === projectId);
+  return selectedProjectData;
 }
-async function promptWorkspaceName(defaultName) {
-  const { workspaceName } = await prompts({
+async function promptBegin() {
+  const prettyCwd = process.cwd().replace(homedir(), "~");
+  const { begin } = await prompts({
+    type: "confirm",
+    name: "begin",
+    message: `Link "${prettyCwd}" with Astro Studio?`,
+    initial: true
+  });
+  if (!begin) {
+    console.log("Canceled.");
+    process.exit(0);
+  }
+}
+async function promptLinkExisting() {
+  const { linkExisting } = await prompts({
+    type: "confirm",
+    name: "linkExisting",
+    message: `Link with an existing project in Astro Studio?`,
+    initial: true
+  });
+  return !!linkExisting;
+}
+async function promptLinkNew() {
+  const { linkNew } = await prompts({
+    type: "confirm",
+    name: "linkNew",
+    message: `Create a new project in Astro Studio?`,
+    initial: true
+  });
+  if (!linkNew) {
+    console.log("Canceled.");
+    process.exit(0);
+  }
+  return true;
+}
+async function promptNewProjectName() {
+  const { newProjectName } = await prompts({
     type: "text",
-    name: "workspaceName",
-    message: "Workspace ID",
-    initial: defaultName
+    name: "newProjectName",
+    message: `What is your new project's name?`,
+    initial: basename(process.cwd()),
+    format: (val) => slug(val)
+  });
+  if (!newProjectName) {
+    console.log("Canceled.");
+    process.exit(0);
+  }
+  return newProjectName;
+}
+async function promptNewProjectRegion() {
+  const { newProjectRegion } = await prompts({
+    type: "select",
+    name: "newProjectRegion",
+    message: `Where should your new database live?`,
+    choices: [
+      { title: "North America (East)", value: "NorthAmericaEast" },
+      { title: "North America (West)", value: "NorthAmericaWest" }
+    ],
+    initial: 0
   });
-  if (typeof workspaceName !== "string") {
+  if (!newProjectRegion) {
+    console.log("Canceled.");
     process.exit(0);
   }
-  return workspaceName;
+  return newProjectRegion;
 }
 export {
   cmd,
-  promptProjectName,
-  promptWorkspaceName
+  createNewProject,
+  promptBegin,
+  promptExistingProjectName,
+  promptLinkExisting,
+  promptLinkNew,
+  promptNewProjectName,
+  promptNewProjectRegion
 };

package/dist/core/cli/commands/login/index.d.ts

@@ -1,6 +1,8 @@
 import type { AstroConfig } from 'astro';
 import type { Arguments } from 'yargs-parser';
-export declare function cmd({ flags }: {
-    config: AstroConfig;
+import type { DBConfig } from '../../../types.js';
+export declare function cmd({ flags, }: {
+    astroConfig: AstroConfig;
+    dbConfig: DBConfig;
     flags: Arguments;
 }): Promise<void>;

package/dist/core/cli/commands/login/index.js

@@ -1,41 +1,50 @@
-import { cyan } from "kleur/colors";
 import { mkdir, writeFile } from "node:fs/promises";
-import { createServer } from "node:http";
-import ora from "ora";
-import { getAstroStudioUrl } from "../../../utils.js";
+import { createServer as _createServer } from "node:http";
+import { listen } from "async-listen";
+import { cyan } from "kleur/colors";
 import open from "open";
+import ora from "ora";
 import { SESSION_LOGIN_FILE } from "../../../tokens.js";
-function serveAndResolveSession() {
+import { getAstroStudioUrl } from "../../../utils.js";
+async function createServer() {
   let resolve, reject;
-  const sessionPromise = new Promise((_resolve, _reject) => {
-    resolve = _resolve;
-    reject = _reject;
-  });
-  const server = createServer((req, res) => {
-    res.writeHead(200);
-    res.end();
+  const server = _createServer((req, res) => {
     const url = new URL(req.url ?? "/", `http://${req.headers.host}`);
-    const session = url.searchParams.get("session");
-    if (!session) {
-      reject();
+    const sessionParam = url.searchParams.get("session");
+    res.statusCode = 302;
+    if (!sessionParam) {
+      res.setHeader("location", getAstroStudioUrl() + "/auth/cli/error");
+      reject(new Error("Failed to log in"));
     } else {
-      resolve(session);
+      res.setHeader("location", getAstroStudioUrl() + "/auth/cli/success");
+      resolve(sessionParam);
     }
-  }).listen(5710, "localhost");
-  return sessionPromise.finally(() => {
+    res.end();
+  });
+  const { port } = await listen(server, 0, "127.0.0.1");
+  const serverUrl = `http://localhost:${port}`;
+  const sessionPromise = new Promise((_resolve, _reject) => {
+    resolve = _resolve;
+    reject = _reject;
+  }).finally(() => {
     server.closeAllConnections();
     server.close();
   });
+  return { url: serverUrl, promise: sessionPromise };
 }
-async function cmd({ flags }) {
+async function cmd({
+  flags
+}) {
   let session = flags.session;
-  const loginUrl = getAstroStudioUrl() + "/auth/cli";
   if (!session) {
-    console.log(`Opening ${cyan(loginUrl)} in your browser...`);
+    const { url, promise } = await createServer();
+    const loginUrl = getAstroStudioUrl() + "/auth/cli/login?returnTo=" + encodeURIComponent(url);
+    console.log(`Opening the following URL in your browser...`);
+    console.log(cyan(loginUrl));
     console.log(`If something goes wrong, copy-and-paste the URL into your browser.`);
     open(loginUrl);
     const spinner = ora("Waiting for confirmation...");
-    session = await serveAndResolveSession();
+    session = await promise;
     spinner.succeed("Successfully logged in!");
   }
   await mkdir(new URL(".", SESSION_LOGIN_FILE), { recursive: true });

package/dist/core/cli/commands/logout/index.d.ts

@@ -1,6 +1 @@
-import type { AstroConfig } from 'astro';
-import type { Arguments } from 'yargs-parser';
-export declare function cmd({}: {
-    config: AstroConfig;
-    flags: Arguments;
-}): Promise<void>;
+export declare function cmd(): Promise<void>;

package/dist/core/cli/commands/logout/index.js

@@ -1,6 +1,6 @@
 import { unlink } from "node:fs/promises";
 import { SESSION_LOGIN_FILE } from "../../../tokens.js";
-async function cmd({}) {
+async function cmd() {
   await unlink(SESSION_LOGIN_FILE);
   console.log("Successfully logged out of Astro Studio.");
 }

package/dist/core/cli/commands/push/index.d.ts

@@ -1,6 +1,8 @@
 import type { AstroConfig } from 'astro';
 import type { Arguments } from 'yargs-parser';
-export declare function cmd({ config, flags }: {
-    config: AstroConfig;
+import { type DBConfig } from '../../../types.js';
+export declare function cmd({ dbConfig, flags, }: {
+    astroConfig: AstroConfig;
+    dbConfig: DBConfig;
     flags: Arguments;
 }): Promise<void>;