@astrojs/db 0.8.1 → 0.8.2

package/dist/_internal/core/schemas.d.ts

@@ -578,7 +578,7 @@ export declare const jsonColumnSchema: z.ZodObject<{
         default?: unknown;
     };
 }>;
-export declare const columnSchema: z.ZodUnion<[z.ZodObject<{
+export declare const columnSchema: z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
     type: z.ZodLiteral<"boolean">;
     schema: z.ZodObject<{
         name: z.ZodOptional<z.ZodString>;
@@ -1379,7 +1379,7 @@ export declare const referenceableColumnSchema: z.ZodUnion<[z.ZodObject<{
         references?: (() => z.input<typeof numberColumnSchema>) | undefined;
     });
 }>]>;
-export declare const columnsSchema: z.ZodRecord<z.ZodString, z.ZodUnion<[z.ZodObject<{
+export declare const columnsSchema: z.ZodRecord<z.ZodString, z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
     type: z.ZodLiteral<"boolean">;
     schema: z.ZodObject<{
         name: z.ZodOptional<z.ZodString>;
@@ -1871,7 +1871,7 @@ type ForeignKeysOutput = Omit<ForeignKeysInput, 'references'> & {
     references: MaybeArray<Omit<z.output<typeof referenceableColumnSchema>, 'references'>>;
 };
 export declare const tableSchema: z.ZodObject<{
-    columns: z.ZodRecord<z.ZodString, z.ZodUnion<[z.ZodObject<{
+    columns: z.ZodRecord<z.ZodString, z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
         type: z.ZodLiteral<"boolean">;
         schema: z.ZodObject<{
             name: z.ZodOptional<z.ZodString>;
@@ -2441,7 +2441,7 @@ export declare const tableSchema: z.ZodObject<{
     deprecated?: boolean | undefined;
 }>;
 export declare const tablesSchema: z.ZodEffects<z.ZodRecord<z.ZodString, z.ZodObject<{
-    columns: z.ZodRecord<z.ZodString, z.ZodUnion<[z.ZodObject<{
+    columns: z.ZodRecord<z.ZodString, z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
         type: z.ZodLiteral<"boolean">;
         schema: z.ZodObject<{
             name: z.ZodOptional<z.ZodString>;
@@ -3101,7 +3101,7 @@ export declare const tablesSchema: z.ZodEffects<z.ZodRecord<z.ZodString, z.ZodOb
 }>, unknown>;
 export declare const dbConfigSchema: z.ZodObject<{
     tables: z.ZodOptional<z.ZodEffects<z.ZodRecord<z.ZodString, z.ZodObject<{
-        columns: z.ZodRecord<z.ZodString, z.ZodUnion<[z.ZodObject<{
+        columns: z.ZodRecord<z.ZodString, z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
             type: z.ZodLiteral<"boolean">;
             schema: z.ZodObject<{
                 name: z.ZodOptional<z.ZodString>;

package/dist/_internal/core/utils.d.ts

@@ -0,0 +1,19 @@
+import type { AstroConfig, AstroIntegration } from 'astro';
+import type { AstroDbIntegration } from './types.js';
+export type VitePlugin = Required<AstroConfig['vite']>['plugins'][number];
+export declare function getAstroStudioEnv(envMode?: string): Record<`ASTRO_STUDIO_${string}`, string>;
+export declare function getRemoteDatabaseUrl(): string;
+export declare function getAstroStudioUrl(): string;
+export declare function getDbDirectoryUrl(root: URL | string): URL;
+export declare function defineDbIntegration(integration: AstroDbIntegration): AstroIntegration;
+/**
+ * Small wrapper around fetch that throws an error if the response is not OK. Allows for custom error handling as well through the onNotOK callback.
+ */
+export declare function safeFetch(url: Parameters<typeof fetch>[0], options?: Parameters<typeof fetch>[1], onNotOK?: (response: Response) => void | Promise<void>): Promise<Response>;
+export type Result<T> = {
+    success: true;
+    data: T;
+} | {
+    success: false;
+    data: unknown;
+};

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

@@ -7,7 +7,7 @@ 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 { getAstroStudioUrl, safeFetch } from "../../../utils.js";
 async function cmd() {
   const sessionToken = await getSessionIdFromFile();
   if (!sessionToken) {
@@ -45,30 +45,33 @@ async function linkProject(id) {
 }
 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")}
+  const response = await safeFetch(
+    linkUrl,
+    {
+      method: "POST",
+      headers: {
+        Authorization: `Bearer ${await getSessionIdFromFile()}`,
+        "Content-Type": "application/json"
+      }
+    },
+    (res) => {
+      if (res.status === 401) {
+        console.error(
+          `${bgRed("Unauthorized")}
 
   Are you logged in?
   Run ${cyan(
-          "astro db login"
-        )} to authenticate and then try linking again.
+            "astro db login"
+          )} to authenticate and then try linking again.
 
 `
-      );
+        );
+        process.exit(1);
+      }
+      console.error(`Failed to fetch user workspace: ${res.status} ${res.statusText}`);
       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.`);
@@ -82,31 +85,34 @@ async function createNewProject({
   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"
+  const response = await safeFetch(
+    linkUrl,
+    {
+      method: "POST",
+      headers: {
+        Authorization: `Bearer ${await getSessionIdFromFile()}`,
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify({ workspaceId, name, region })
     },
-    body: JSON.stringify({ workspaceId, name, region })
-  });
-  if (!response.ok) {
-    if (response.status === 401) {
-      console.error(
-        `${bgRed("Unauthorized")}
+    (res) => {
+      if (res.status === 401) {
+        console.error(
+          `${bgRed("Unauthorized")}
 
   Are you logged in?
   Run ${cyan(
-          "astro db login"
-        )} to authenticate and then try linking again.
+            "astro db login"
+          )} to authenticate and then try linking again.
 
 `
-      );
+        );
+        process.exit(1);
+      }
+      console.error(`Failed to create project: ${res.status} ${res.statusText}`);
       process.exit(1);
     }
-    console.error(`Failed to create project: ${response.status} ${response.statusText}`);
-    process.exit(1);
-  }
+  );
   const { data, success } = await response.json();
   if (!success) {
     console.error(`Failed to create project.`);
@@ -116,31 +122,34 @@ async function createNewProject({
 }
 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"
+  const response = await safeFetch(
+    linkUrl,
+    {
+      method: "POST",
+      headers: {
+        Authorization: `Bearer ${await getSessionIdFromFile()}`,
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify({ workspaceId })
     },
-    body: JSON.stringify({ workspaceId })
-  });
-  if (!response.ok) {
-    if (response.status === 401) {
-      console.error(
-        `${bgRed("Unauthorized")}
+    (res) => {
+      if (res.status === 401) {
+        console.error(
+          `${bgRed("Unauthorized")}
 
   Are you logged in?
   Run ${cyan(
-          "astro db login"
-        )} to authenticate and then try linking again.
+            "astro db login"
+          )} to authenticate and then try linking again.
 
 `
-      );
+        );
+        process.exit(1);
+      }
+      console.error(`Failed to fetch projects: ${res.status} ${res.statusText}`);
       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.`);

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

@@ -1,7 +1,7 @@
 import { MIGRATION_VERSION } from "../../../consts.js";
 import { getManagedAppTokenOrExit } from "../../../tokens.js";
 import {} from "../../../types.js";
-import { getRemoteDatabaseUrl } from "../../../utils.js";
+import { getRemoteDatabaseUrl, safeFetch } from "../../../utils.js";
 import {
   createCurrentSnapshot,
   createEmptySnapshot,
@@ -64,18 +64,21 @@ async function pushSchema({
     return new Response(null, { status: 200 });
   }
   const url = new URL("/db/push", getRemoteDatabaseUrl());
-  const response = await fetch(url, {
-    method: "POST",
-    headers: new Headers({
-      Authorization: `Bearer ${appToken}`
-    }),
-    body: JSON.stringify(requestBody)
-  });
-  if (!response.ok) {
-    console.error(`${url.toString()} failed: ${response.status} ${response.statusText}`);
-    console.error(await response.text());
-    throw new Error(`/db/push fetch failed: ${response.status} ${response.statusText}`);
-  }
+  const response = await safeFetch(
+    url,
+    {
+      method: "POST",
+      headers: new Headers({
+        Authorization: `Bearer ${appToken}`
+      }),
+      body: JSON.stringify(requestBody)
+    },
+    async (res) => {
+      console.error(`${url.toString()} failed: ${res.status} ${res.statusText}`);
+      console.error(await res.text());
+      throw new Error(`/db/push fetch failed: ${res.status} ${res.statusText}`);
+    }
+  );
   const result = await response.json();
   if (!result.success) {
     console.error(`${url.toString()} unsuccessful`);

package/dist/core/cli/migration-queries.js

@@ -19,7 +19,7 @@ import { RENAME_COLUMN_ERROR, RENAME_TABLE_ERROR } from "../errors.js";
 import { columnSchema } from "../schemas.js";
 import {
 } from "../types.js";
-import { getRemoteDatabaseUrl } from "../utils.js";
+import { getRemoteDatabaseUrl, safeFetch } from "../utils.js";
 const sqlite = new SQLiteAsyncDialect();
 const genTempTableName = customAlphabet("abcdefghijklmnopqrstuvwxyz", 10);
 async function getMigrationQueries({
@@ -324,17 +324,20 @@ async function getProductionCurrentSnapshot({
   appToken
 }) {
   const url = new URL("/db/schema", getRemoteDatabaseUrl());
-  const response = await fetch(url, {
-    method: "POST",
-    headers: new Headers({
-      Authorization: `Bearer ${appToken}`
-    })
-  });
-  if (!response.ok) {
-    console.error(`${url.toString()} failed: ${response.status} ${response.statusText}`);
-    console.error(await response.text());
-    throw new Error(`/db/schema fetch failed: ${response.status} ${response.statusText}`);
-  }
+  const response = await safeFetch(
+    url,
+    {
+      method: "POST",
+      headers: new Headers({
+        Authorization: `Bearer ${appToken}`
+      })
+    },
+    async (res) => {
+      console.error(`${url.toString()} failed: ${res.status} ${res.statusText}`);
+      console.error(await res.text());
+      throw new Error(`/db/schema fetch failed: ${res.status} ${res.statusText}`);
+    }
+  );
   const result = await response.json();
   if (!result.success) {
     console.error(`${url.toString()} unsuccessful`);

package/dist/core/cli/print-help.js

@@ -27,7 +27,7 @@ function printHelp({
     message.push(
       linebreak(),
       `  ${bgGreen(black(` ${commandName} `))} ${green(
-        `v${"0.8.1"}`
+        `v${"0.8.2"}`
       )} ${headline}`
     );
   }

package/dist/core/schemas.d.ts

@@ -578,7 +578,7 @@ export declare const jsonColumnSchema: z.ZodObject<{
         default?: unknown;
     };
 }>;
-export declare const columnSchema: z.ZodUnion<[z.ZodObject<{
+export declare const columnSchema: z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
     type: z.ZodLiteral<"boolean">;
     schema: z.ZodObject<{
         name: z.ZodOptional<z.ZodString>;
@@ -1379,7 +1379,7 @@ export declare const referenceableColumnSchema: z.ZodUnion<[z.ZodObject<{
         references?: (() => z.input<typeof numberColumnSchema>) | undefined;
     });
 }>]>;
-export declare const columnsSchema: z.ZodRecord<z.ZodString, z.ZodUnion<[z.ZodObject<{
+export declare const columnsSchema: z.ZodRecord<z.ZodString, z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
     type: z.ZodLiteral<"boolean">;
     schema: z.ZodObject<{
         name: z.ZodOptional<z.ZodString>;
@@ -1871,7 +1871,7 @@ type ForeignKeysOutput = Omit<ForeignKeysInput, 'references'> & {
     references: MaybeArray<Omit<z.output<typeof referenceableColumnSchema>, 'references'>>;
 };
 export declare const tableSchema: z.ZodObject<{
-    columns: z.ZodRecord<z.ZodString, z.ZodUnion<[z.ZodObject<{
+    columns: z.ZodRecord<z.ZodString, z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
         type: z.ZodLiteral<"boolean">;
         schema: z.ZodObject<{
             name: z.ZodOptional<z.ZodString>;
@@ -2441,7 +2441,7 @@ export declare const tableSchema: z.ZodObject<{
     deprecated?: boolean | undefined;
 }>;
 export declare const tablesSchema: z.ZodEffects<z.ZodRecord<z.ZodString, z.ZodObject<{
-    columns: z.ZodRecord<z.ZodString, z.ZodUnion<[z.ZodObject<{
+    columns: z.ZodRecord<z.ZodString, z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
         type: z.ZodLiteral<"boolean">;
         schema: z.ZodObject<{
             name: z.ZodOptional<z.ZodString>;
@@ -3101,7 +3101,7 @@ export declare const tablesSchema: z.ZodEffects<z.ZodRecord<z.ZodString, z.ZodOb
 }>, unknown>;
 export declare const dbConfigSchema: z.ZodObject<{
     tables: z.ZodOptional<z.ZodEffects<z.ZodRecord<z.ZodString, z.ZodObject<{
-        columns: z.ZodRecord<z.ZodString, z.ZodUnion<[z.ZodObject<{
+        columns: z.ZodRecord<z.ZodString, z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
             type: z.ZodLiteral<"boolean">;
             schema: z.ZodObject<{
                 name: z.ZodOptional<z.ZodString>;

package/dist/core/schemas.js

@@ -96,7 +96,7 @@ const jsonColumnSchema = z.object({
     default: z.unknown().optional()
   })
 });
-const columnSchema = z.union([
+const columnSchema = z.discriminatedUnion("type", [
   booleanColumnSchema,
   numberColumnSchema,
   textColumnSchema,

package/dist/core/tokens.js

@@ -3,7 +3,7 @@ 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";
+import { getAstroStudioEnv, getAstroStudioUrl, getRemoteDatabaseUrl, safeFetch } 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 {
@@ -19,21 +19,35 @@ class ManagedLocalAppToken {
 class ManagedRemoteAppToken {
   token;
   session;
+  region;
   projectId;
   ttl;
   renewTimer;
+  static async getRegionCode() {
+    const pingResponse = await safeFetch(new URL(`${getRemoteDatabaseUrl()}/ping`));
+    const pingResult = await pingResponse.json();
+    return pingResult.data.region;
+  }
   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 region = await ManagedRemoteAppToken.getRegionCode();
+    const response = await safeFetch(
+      new URL(`${getAstroStudioUrl()}/auth/cli/token-create`),
+      {
+        method: "POST",
+        headers: new Headers({
+          Authorization: `Bearer ${sessionToken}`
+        }),
+        body: JSON.stringify({ projectId, region })
+      },
+      (res) => {
+        throw new Error(`Failed to create token: ${res.status} ${res.statusText}`);
+      }
+    );
     const { token: shortLivedAppToken, ttl } = await response.json();
     return new ManagedRemoteAppToken({
       token: shortLivedAppToken,
       session: sessionToken,
+      region,
       projectId,
       ttl
     });
@@ -41,19 +55,26 @@ class ManagedRemoteAppToken {
   constructor(options) {
     this.token = options.token;
     this.session = options.session;
+    this.region = options.region;
     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"
+    return safeFetch(
+      `${getAstroStudioUrl()}${url}`,
+      {
+        method: "POST",
+        headers: {
+          Authorization: `Bearer ${this.session}`,
+          "Content-Type": "application/json"
+        },
+        body: JSON.stringify({ ...body, region: this.region })
       },
-      body: JSON.stringify(body)
-    });
+      () => {
+        throw new Error(`Failed to fetch ${url}.`);
+      }
+    );
   }
   async renew() {
     clearTimeout(this.renewTimer);

package/dist/core/utils.d.ts

@@ -6,3 +6,14 @@ export declare function getRemoteDatabaseUrl(): string;
 export declare function getAstroStudioUrl(): string;
 export declare function getDbDirectoryUrl(root: URL | string): URL;
 export declare function defineDbIntegration(integration: AstroDbIntegration): AstroIntegration;
+/**
+ * Small wrapper around fetch that throws an error if the response is not OK. Allows for custom error handling as well through the onNotOK callback.
+ */
+export declare function safeFetch(url: Parameters<typeof fetch>[0], options?: Parameters<typeof fetch>[1], onNotOK?: (response: Response) => void | Promise<void>): Promise<Response>;
+export type Result<T> = {
+    success: true;
+    data: T;
+} | {
+    success: false;
+    data: unknown;
+};

package/dist/core/utils.js

@@ -17,10 +17,20 @@ function getDbDirectoryUrl(root) {
 function defineDbIntegration(integration) {
   return integration;
 }
+async function safeFetch(url, options = {}, onNotOK = () => {
+  throw new Error(`Request to ${url} returned a non-OK status code.`);
+}) {
+  const response = await fetch(url, options);
+  if (!response.ok) {
+    await onNotOK(response);
+  }
+  return response;
+}
 export {
   defineDbIntegration,
   getAstroStudioEnv,
   getAstroStudioUrl,
   getDbDirectoryUrl,
-  getRemoteDatabaseUrl
+  getRemoteDatabaseUrl,
+  safeFetch
 };

package/dist/runtime/db-client.js

@@ -2,6 +2,7 @@ import { createClient } from "@libsql/client";
 import { drizzle as drizzleLibsql } from "drizzle-orm/libsql";
 import { drizzle as drizzleProxy } from "drizzle-orm/sqlite-proxy";
 import { z } from "zod";
+import { safeFetch } from "../core/utils.js";
 const isWebContainer = !!process.versions?.webcontainer;
 function createLocalDatabaseClient({ dbUrl }) {
   const url = isWebContainer ? "file:content.db" : dbUrl;
@@ -21,21 +22,24 @@ function createRemoteDatabaseClient(appToken, remoteDbURL) {
   const db = drizzleProxy(
     async (sql, parameters, method) => {
       const requestBody = { sql, args: parameters };
-      const res = await fetch(url, {
-        method: "POST",
-        headers: {
-          Authorization: `Bearer ${appToken}`,
-          "Content-Type": "application/json"
+      const res = await safeFetch(
+        url,
+        {
+          method: "POST",
+          headers: {
+            Authorization: `Bearer ${appToken}`,
+            "Content-Type": "application/json"
+          },
+          body: JSON.stringify(requestBody)
         },
-        body: JSON.stringify(requestBody)
-      });
-      if (!res.ok) {
-        throw new Error(
-          `Failed to execute query.
+        (response) => {
+          throw new Error(
+            `Failed to execute query.
 Query: ${sql}
-Full error: ${res.status} ${await res.text()}}`
-        );
-      }
+Full error: ${response.status} ${response.statusText}`
+          );
+        }
+      );
       let remoteResult;
       try {
         const json = await res.json();
@@ -62,20 +66,23 @@ Full error: Unexpected JSON response. ${e instanceof Error ? e.message : String(
     },
     async (queries) => {
       const stmts = queries.map(({ sql, params }) => ({ sql, args: params }));
-      const res = await fetch(url, {
-        method: "POST",
-        headers: {
-          Authorization: `Bearer ${appToken}`,
-          "Content-Type": "application/json"
+      const res = await safeFetch(
+        url,
+        {
+          method: "POST",
+          headers: {
+            Authorization: `Bearer ${appToken}`,
+            "Content-Type": "application/json"
+          },
+          body: JSON.stringify(stmts)
         },
-        body: JSON.stringify(stmts)
-      });
-      if (!res.ok) {
-        throw new Error(
-          `Failed to execute batch queries.
-Full error: ${res.status} ${await res.text()}}`
-        );
-      }
+        (response) => {
+          throw new Error(
+            `Failed to execute batch queries.
+Full error: ${response.status} ${response.statusText}}`
+          );
+        }
+      );
       let remoteResults;
       try {
         const json = await res.json();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@astrojs/db",
-  "version": "0.8.1",
+  "version": "0.8.2",
   "description": "",
   "license": "MIT",
   "type": "module",
@@ -80,7 +80,7 @@
     "mocha": "^10.2.0",
     "typescript": "^5.2.2",
     "vite": "^5.1.4",
-    "astro": "4.5.2",
+    "astro": "4.5.3",
     "astro-scripts": "0.0.14"
   },
   "scripts": {