@astrojs/db 0.1.16 → 0.1.18

package/dist/config.d.ts

@@ -1,13 +1,12 @@
 import type { SQLiteInsertValue } from 'drizzle-orm/sqlite-core';
 import type { SqliteDB, Table } from './internal.js';
 import type { MaybeArray, collectionSchema } from './types.js';
-import { type BooleanField, type DBFieldInput, type DateFieldInput, type JsonField, type NumberField, type TextField, type MaybePromise } from './types.js';
+import { type BooleanField, type DBFieldInput, type DateFieldInput, type JsonField, type NumberField, type TextField, type indexSchema, type MaybePromise } from './types.js';
 import { z } from 'zod';
-export type DBFieldsConfig = z.input<typeof collectionSchema>['fields'];
 export type DBDataContext = {
     db: SqliteDB;
-    seed<TFields extends DBFieldsConfig>(collection: ResolvedCollectionConfig<TFields>, data: MaybeArray<SQLiteInsertValue<Table<string, 
-    /** TODO: true type inference */ Record<Extract<keyof TFields, string>, DBFieldsConfig[number]>>>>): Promise<any> /** TODO: type output */;
+    seed<TFields extends FieldsConfig>(collection: ResolvedCollectionConfig<TFields>, data: MaybeArray<SQLiteInsertValue<Table<string, 
+    /** TODO: true type inference */ Record<Extract<keyof TFields, string>, FieldsConfig[number]>>>>): Promise<any> /** TODO: type output */;
     mode: 'dev' | 'build';
 };
 export declare const dbConfigSchema: z.ZodObject<{
@@ -113,9 +112,19 @@ export declare const dbConfigSchema: z.ZodObject<{
             unique?: boolean | undefined;
             default?: unknown;
         }>]>>;
-        writable: z.ZodLiteral<false>;
+        indexes: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodObject<{
+            on: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodString, "many">]>;
+            unique: z.ZodOptional<z.ZodBoolean>;
+        }, "strip", z.ZodTypeAny, {
+            on: (string | string[]) & (string | string[] | undefined);
+            unique?: boolean | undefined;
+        }, {
+            on: (string | string[]) & (string | string[] | undefined);
+            unique?: boolean | undefined;
+        }>>>;
         table: z.ZodAny;
         _setMeta: z.ZodOptional<z.ZodFunction<z.ZodTuple<[], z.ZodUnknown>, z.ZodUnknown>>;
+        writable: z.ZodLiteral<false>;
     }, "strip", z.ZodTypeAny, {
         fields: Record<string, {
             type: "boolean";
@@ -152,6 +161,10 @@ export declare const dbConfigSchema: z.ZodObject<{
             default?: unknown;
         }>;
         writable: false;
+        indexes?: Record<string, {
+            on: (string | string[]) & (string | string[] | undefined);
+            unique?: boolean | undefined;
+        }> | undefined;
         table?: any;
         _setMeta?: ((...args: unknown[]) => unknown) | undefined;
     }, {
@@ -190,6 +203,10 @@ export declare const dbConfigSchema: z.ZodObject<{
             default?: unknown;
         }>;
         writable: false;
+        indexes?: Record<string, {
+            on: (string | string[]) & (string | string[] | undefined);
+            unique?: boolean | undefined;
+        }> | undefined;
         table?: any;
         _setMeta?: ((...args: unknown[]) => unknown) | undefined;
     }>, z.ZodObject<{
@@ -293,9 +310,19 @@ export declare const dbConfigSchema: z.ZodObject<{
             unique?: boolean | undefined;
             default?: unknown;
         }>]>>;
-        writable: z.ZodLiteral<true>;
+        indexes: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodObject<{
+            on: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodString, "many">]>;
+            unique: z.ZodOptional<z.ZodBoolean>;
+        }, "strip", z.ZodTypeAny, {
+            on: (string | string[]) & (string | string[] | undefined);
+            unique?: boolean | undefined;
+        }, {
+            on: (string | string[]) & (string | string[] | undefined);
+            unique?: boolean | undefined;
+        }>>>;
         table: z.ZodAny;
         _setMeta: z.ZodOptional<z.ZodFunction<z.ZodTuple<[], z.ZodUnknown>, z.ZodUnknown>>;
+        writable: z.ZodLiteral<true>;
     }, "strip", z.ZodTypeAny, {
         fields: Record<string, {
             type: "boolean";
@@ -332,6 +359,10 @@ export declare const dbConfigSchema: z.ZodObject<{
             default?: unknown;
         }>;
         writable: true;
+        indexes?: Record<string, {
+            on: (string | string[]) & (string | string[] | undefined);
+            unique?: boolean | undefined;
+        }> | undefined;
         table?: any;
         _setMeta?: ((...args: unknown[]) => unknown) | undefined;
     }, {
@@ -370,6 +401,10 @@ export declare const dbConfigSchema: z.ZodObject<{
             default?: unknown;
         }>;
         writable: true;
+        indexes?: Record<string, {
+            on: (string | string[]) & (string | string[] | undefined);
+            unique?: boolean | undefined;
+        }> | undefined;
         table?: any;
         _setMeta?: ((...args: unknown[]) => unknown) | undefined;
     }>]>>>;
@@ -412,6 +447,10 @@ export declare const dbConfigSchema: z.ZodObject<{
             default?: unknown;
         }>;
         writable: false;
+        indexes?: Record<string, {
+            on: (string | string[]) & (string | string[] | undefined);
+            unique?: boolean | undefined;
+        }> | undefined;
         table?: any;
         _setMeta?: ((...args: unknown[]) => unknown) | undefined;
     } | {
@@ -450,6 +489,10 @@ export declare const dbConfigSchema: z.ZodObject<{
             default?: unknown;
         }>;
         writable: true;
+        indexes?: Record<string, {
+            on: (string | string[]) & (string | string[] | undefined);
+            unique?: boolean | undefined;
+        }> | undefined;
         table?: any;
         _setMeta?: ((...args: unknown[]) => unknown) | undefined;
     }> | undefined;
@@ -492,6 +535,10 @@ export declare const dbConfigSchema: z.ZodObject<{
             default?: unknown;
         }>;
         writable: false;
+        indexes?: Record<string, {
+            on: (string | string[]) & (string | string[] | undefined);
+            unique?: boolean | undefined;
+        }> | undefined;
         table?: any;
         _setMeta?: ((...args: unknown[]) => unknown) | undefined;
     } | {
@@ -530,6 +577,10 @@ export declare const dbConfigSchema: z.ZodObject<{
             default?: unknown;
         }>;
         writable: true;
+        indexes?: Record<string, {
+            on: (string | string[]) & (string | string[] | undefined);
+            unique?: boolean | undefined;
+        }> | undefined;
         table?: any;
         _setMeta?: ((...args: unknown[]) => unknown) | undefined;
     }> | undefined;
@@ -642,9 +693,19 @@ export declare const astroConfigWithDbSchema: z.ZodObject<{
                 unique?: boolean | undefined;
                 default?: unknown;
             }>]>>;
-            writable: z.ZodLiteral<false>;
+            indexes: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodObject<{
+                on: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodString, "many">]>;
+                unique: z.ZodOptional<z.ZodBoolean>;
+            }, "strip", z.ZodTypeAny, {
+                on: (string | string[]) & (string | string[] | undefined);
+                unique?: boolean | undefined;
+            }, {
+                on: (string | string[]) & (string | string[] | undefined);
+                unique?: boolean | undefined;
+            }>>>;
             table: z.ZodAny;
             _setMeta: z.ZodOptional<z.ZodFunction<z.ZodTuple<[], z.ZodUnknown>, z.ZodUnknown>>;
+            writable: z.ZodLiteral<false>;
         }, "strip", z.ZodTypeAny, {
             fields: Record<string, {
                 type: "boolean";
@@ -681,6 +742,10 @@ export declare const astroConfigWithDbSchema: z.ZodObject<{
                 default?: unknown;
             }>;
             writable: false;
+            indexes?: Record<string, {
+                on: (string | string[]) & (string | string[] | undefined);
+                unique?: boolean | undefined;
+            }> | undefined;
             table?: any;
             _setMeta?: ((...args: unknown[]) => unknown) | undefined;
         }, {
@@ -719,6 +784,10 @@ export declare const astroConfigWithDbSchema: z.ZodObject<{
                 default?: unknown;
             }>;
             writable: false;
+            indexes?: Record<string, {
+                on: (string | string[]) & (string | string[] | undefined);
+                unique?: boolean | undefined;
+            }> | undefined;
             table?: any;
             _setMeta?: ((...args: unknown[]) => unknown) | undefined;
         }>, z.ZodObject<{
@@ -822,9 +891,19 @@ export declare const astroConfigWithDbSchema: z.ZodObject<{
                 unique?: boolean | undefined;
                 default?: unknown;
             }>]>>;
-            writable: z.ZodLiteral<true>;
+            indexes: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodObject<{
+                on: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodString, "many">]>;
+                unique: z.ZodOptional<z.ZodBoolean>;
+            }, "strip", z.ZodTypeAny, {
+                on: (string | string[]) & (string | string[] | undefined);
+                unique?: boolean | undefined;
+            }, {
+                on: (string | string[]) & (string | string[] | undefined);
+                unique?: boolean | undefined;
+            }>>>;
             table: z.ZodAny;
             _setMeta: z.ZodOptional<z.ZodFunction<z.ZodTuple<[], z.ZodUnknown>, z.ZodUnknown>>;
+            writable: z.ZodLiteral<true>;
         }, "strip", z.ZodTypeAny, {
             fields: Record<string, {
                 type: "boolean";
@@ -861,6 +940,10 @@ export declare const astroConfigWithDbSchema: z.ZodObject<{
                 default?: unknown;
             }>;
             writable: true;
+            indexes?: Record<string, {
+                on: (string | string[]) & (string | string[] | undefined);
+                unique?: boolean | undefined;
+            }> | undefined;
             table?: any;
             _setMeta?: ((...args: unknown[]) => unknown) | undefined;
         }, {
@@ -899,6 +982,10 @@ export declare const astroConfigWithDbSchema: z.ZodObject<{
                 default?: unknown;
             }>;
             writable: true;
+            indexes?: Record<string, {
+                on: (string | string[]) & (string | string[] | undefined);
+                unique?: boolean | undefined;
+            }> | undefined;
             table?: any;
             _setMeta?: ((...args: unknown[]) => unknown) | undefined;
         }>]>>>;
@@ -941,6 +1028,10 @@ export declare const astroConfigWithDbSchema: z.ZodObject<{
                 default?: unknown;
             }>;
             writable: false;
+            indexes?: Record<string, {
+                on: (string | string[]) & (string | string[] | undefined);
+                unique?: boolean | undefined;
+            }> | undefined;
             table?: any;
             _setMeta?: ((...args: unknown[]) => unknown) | undefined;
         } | {
@@ -979,6 +1070,10 @@ export declare const astroConfigWithDbSchema: z.ZodObject<{
                 default?: unknown;
             }>;
             writable: true;
+            indexes?: Record<string, {
+                on: (string | string[]) & (string | string[] | undefined);
+                unique?: boolean | undefined;
+            }> | undefined;
             table?: any;
             _setMeta?: ((...args: unknown[]) => unknown) | undefined;
         }> | undefined;
@@ -1021,6 +1116,10 @@ export declare const astroConfigWithDbSchema: z.ZodObject<{
                 default?: unknown;
             }>;
             writable: false;
+            indexes?: Record<string, {
+                on: (string | string[]) & (string | string[] | undefined);
+                unique?: boolean | undefined;
+            }> | undefined;
             table?: any;
             _setMeta?: ((...args: unknown[]) => unknown) | undefined;
         } | {
@@ -1059,6 +1158,10 @@ export declare const astroConfigWithDbSchema: z.ZodObject<{
                 default?: unknown;
             }>;
             writable: true;
+            indexes?: Record<string, {
+                on: (string | string[]) & (string | string[] | undefined);
+                unique?: boolean | undefined;
+            }> | undefined;
             table?: any;
             _setMeta?: ((...args: unknown[]) => unknown) | undefined;
         }> | undefined;
@@ -1103,6 +1206,10 @@ export declare const astroConfigWithDbSchema: z.ZodObject<{
                 default?: unknown;
             }>;
             writable: false;
+            indexes?: Record<string, {
+                on: (string | string[]) & (string | string[] | undefined);
+                unique?: boolean | undefined;
+            }> | undefined;
             table?: any;
             _setMeta?: ((...args: unknown[]) => unknown) | undefined;
         } | {
@@ -1141,6 +1248,10 @@ export declare const astroConfigWithDbSchema: z.ZodObject<{
                 default?: unknown;
             }>;
             writable: true;
+            indexes?: Record<string, {
+                on: (string | string[]) & (string | string[] | undefined);
+                unique?: boolean | undefined;
+            }> | undefined;
             table?: any;
             _setMeta?: ((...args: unknown[]) => unknown) | undefined;
         }> | undefined;
@@ -1185,6 +1296,10 @@ export declare const astroConfigWithDbSchema: z.ZodObject<{
                 default?: unknown;
             }>;
             writable: false;
+            indexes?: Record<string, {
+                on: (string | string[]) & (string | string[] | undefined);
+                unique?: boolean | undefined;
+            }> | undefined;
             table?: any;
             _setMeta?: ((...args: unknown[]) => unknown) | undefined;
         } | {
@@ -1223,21 +1338,30 @@ export declare const astroConfigWithDbSchema: z.ZodObject<{
                 default?: unknown;
             }>;
             writable: true;
+            indexes?: Record<string, {
+                on: (string | string[]) & (string | string[] | undefined);
+                unique?: boolean | undefined;
+            }> | undefined;
             table?: any;
             _setMeta?: ((...args: unknown[]) => unknown) | undefined;
         }> | undefined;
         data?: ((...args: unknown[]) => void | Promise<void>) | undefined;
     } | undefined;
 }>;
-type CollectionConfig<TFields extends DBFieldsConfig> = {
+export type FieldsConfig = z.input<typeof collectionSchema>['fields'];
+interface CollectionConfig<TFields extends FieldsConfig> extends Pick<z.input<typeof collectionSchema>, 'fields' | 'indexes'> {
     fields: TFields;
-};
-export type ResolvedCollectionConfig<TFields extends DBFieldsConfig = DBFieldsConfig, Writable extends boolean = boolean> = CollectionConfig<TFields> & {
+    indexes?: Record<string, IndexConfig<TFields>>;
+}
+interface IndexConfig<TFields extends FieldsConfig> extends z.input<typeof indexSchema> {
+    on: MaybeArray<Extract<keyof TFields, string>>;
+}
+export type ResolvedCollectionConfig<TFields extends FieldsConfig = FieldsConfig, Writable extends boolean = boolean> = CollectionConfig<TFields> & {
     writable: Writable;
     table: Table<string, TFields>;
 };
-export declare function defineCollection<TFields extends DBFieldsConfig>(userConfig: CollectionConfig<TFields>): ResolvedCollectionConfig<TFields, false>;
-export declare function defineWritableCollection<TFields extends DBFieldsConfig>(userConfig: CollectionConfig<TFields>): ResolvedCollectionConfig<TFields, true>;
+export declare function defineCollection<TFields extends FieldsConfig>(userConfig: CollectionConfig<TFields>): ResolvedCollectionConfig<TFields, false>;
+export declare function defineWritableCollection<TFields extends FieldsConfig>(userConfig: CollectionConfig<TFields>): ResolvedCollectionConfig<TFields, true>;
 export type AstroConfigWithDB = z.infer<typeof astroConfigWithDbSchema>;
 type FieldOpts<T extends DBFieldInput> = Omit<T, 'type'>;
 export declare const field: {

package/dist/file-url-integration.d.ts

@@ -0,0 +1,2 @@
+import type { AstroIntegration } from 'astro';
+export declare function fileURLIntegration(): AstroIntegration;

package/dist/file-url-integration.js

@@ -0,0 +1,71 @@
+import fs from "node:fs";
+import { pathToFileURL } from "node:url";
+import path from "node:path";
+async function copyFile(toDir, fromUrl, toUrl) {
+  await fs.promises.mkdir(toDir, { recursive: true });
+  await fs.promises.rename(fromUrl, toUrl);
+}
+function fileURLIntegration() {
+  const fileNames = [];
+  function createVitePlugin(command) {
+    const referenceIds = [];
+    return {
+      name: "@astrojs/db/file-url",
+      enforce: "pre",
+      async load(id) {
+        if (id.endsWith("?fileurl")) {
+          const filePath = id.slice(0, id.indexOf("?"));
+          if (command === "build") {
+            const data = await fs.promises.readFile(filePath);
+            const name = path.basename(filePath);
+            const referenceId = this.emitFile({
+              name,
+              source: data,
+              type: "asset"
+            });
+            referenceIds.push(referenceId);
+            return `export default import.meta.ROLLUP_FILE_URL_${referenceId};`;
+          } else {
+            return `export default new URL(${JSON.stringify(pathToFileURL(filePath).toString())})`;
+          }
+        }
+      },
+      generateBundle() {
+        for (const referenceId of referenceIds) {
+          fileNames.push(this.getFileName(referenceId));
+        }
+      }
+    };
+  }
+  let config;
+  return {
+    name: "@astrojs/db/file-url",
+    hooks: {
+      "astro:config:setup"({ updateConfig, command }) {
+        updateConfig({
+          vite: {
+            plugins: [createVitePlugin(command)]
+          }
+        });
+      },
+      "astro:config:done": ({ config: _config }) => {
+        config = _config;
+      },
+      async "astro:build:done"() {
+        if (config.output !== "static") {
+          const moves = [];
+          for (const fileName of fileNames) {
+            const fromUrl = new URL(fileName, config.build.client);
+            const toUrl = new URL(fileName, config.build.server);
+            const toDir = new URL("./", toUrl);
+            moves.push(copyFile(toDir, fromUrl, toUrl));
+          }
+          await Promise.all(moves);
+        }
+      }
+    }
+  };
+}
+export {
+  fileURLIntegration
+};

package/dist/integration.d.ts

@@ -1,2 +1,2 @@
 import type { AstroIntegration } from 'astro';
-export declare function integration(): AstroIntegration;
+export declare function integration(): AstroIntegration[];

package/dist/integration.js

@@ -12,7 +12,8 @@ import { errorMap } from "./error-map.js";
 import { dirname } from "path";
 import { fileURLToPath } from "url";
 import { bold } from "kleur/colors";
-function integration() {
+import { fileURLIntegration } from "./file-url-integration.js";
+function astroDBIntegration() {
   return {
     name: "astro:db",
     hooks: {
@@ -39,7 +40,8 @@ function integration() {
           dbPlugin = vitePluginDb({
             connectToStudio: true,
             collections,
-            appToken
+            appToken,
+            output: config.output
           });
         } else {
           const dbUrl = getLocalDbUrl(config.root);
@@ -61,12 +63,25 @@ function integration() {
             mode: command === "dev" ? "dev" : "build"
           });
           logger.info("Collections set up \u{1F680}");
-          dbPlugin = vitePluginDb({ connectToStudio: false, collections, dbUrl: dbUrl.href });
+          dbPlugin = vitePluginDb({
+            connectToStudio: false,
+            collections
+          });
         }
         updateConfig({
           vite: {
             assetsInclude: [DB_PATH],
-            plugins: [dbPlugin, vitePluginInjectEnvTs(config)]
+            plugins: [dbPlugin, vitePluginInjectEnvTs(config), {
+              name: "my-plugin",
+              resolveId(id) {
+                if (id.endsWith("?server-path")) {
+                }
+              },
+              load(id) {
+                if (id.endsWith("?server-path")) {
+                }
+              }
+            }]
           }
         });
         await typegen({ collections, root: config.root });
@@ -74,6 +89,9 @@ function integration() {
     }
   };
 }
+function integration() {
+  return [astroDBIntegration(), fileURLIntegration()];
+}
 export {
   integration
 };

package/dist/internal.d.ts

@@ -1,18 +1,17 @@
 import type { SqliteRemoteDatabase } from 'drizzle-orm/sqlite-proxy';
 import { type BooleanField, type DBCollection, type DBCollections, type DBField, type DateField, type FieldType, type JsonField, type NumberField, type TextField } from './types.js';
-import { type LibSQLDatabase } from 'drizzle-orm/libsql';
-import { type ColumnDataType } from 'drizzle-orm';
+import { type SQL, type ColumnDataType } from 'drizzle-orm';
 import type { AstroIntegrationLogger } from 'astro';
 import type { DBUserConfig } from './config.js';
 export type SqliteDB = SqliteRemoteDatabase;
 export type { Table } from './types.js';
-export { createRemoteDatabaseClient, findLocalDatabase } from './utils-runtime.js';
+export { createRemoteDatabaseClient } from './utils-runtime.js';
 export declare function hasPrimaryKey(field: DBField): boolean;
 export declare function createLocalDatabaseClient({ collections, dbUrl, seeding, }: {
     dbUrl: string;
     collections: DBCollections;
     seeding: boolean;
-}): Promise<LibSQLDatabase<Record<string, never>>>;
+}): Promise<import("drizzle-orm/libsql").LibSQLDatabase<Record<string, never>>>;
 export declare function setupDbTables({ db, data, collections, logger, mode, }: {
     db: SqliteRemoteDatabase;
     data?: DBUserConfig['data'];
@@ -21,6 +20,7 @@ export declare function setupDbTables({ db, data, collections, logger, mode, }:
     mode: 'dev' | 'build';
 }): Promise<void>;
 export declare function getCreateTableQuery(collectionName: string, collection: DBCollection): string;
+export declare function getTableIndexQueries(collectionName: string, collection: DBCollection): SQL<unknown>[];
 export declare function schemaTypeToSqlType(type: FieldType): 'text' | 'integer';
 export declare function getModifiers(fieldName: string, field: DBField): string;
 type WithDefaultDefined<T extends DBField> = T & Required<Pick<T, 'default'>>;

package/dist/internal.js

@@ -12,10 +12,11 @@ import {
   customType,
   integer,
   sqliteTable,
-  text
+  text,
+  index
 } from "drizzle-orm/sqlite-core";
 import { z } from "zod";
-import { createRemoteDatabaseClient, findLocalDatabase } from "./utils-runtime.js";
+import { createRemoteDatabaseClient } from "./utils-runtime.js";
 const sqlite = new SQLiteAsyncDialect();
 function hasPrimaryKey(field) {
   return "primaryKey" in field && !!field.primaryKey;
@@ -63,7 +64,8 @@ async function setupDbTables({
   for (const [name, collection] of Object.entries(collections)) {
     const dropQuery = sql.raw(`DROP TABLE IF EXISTS ${name}`);
     const createQuery = sql.raw(getCreateTableQuery(name, collection));
-    setupQueries.push(dropQuery, createQuery);
+    const indexQueries = getTableIndexQueries(name, collection);
+    setupQueries.push(dropQuery, createQuery, ...indexQueries);
   }
   for (const q of setupQueries) {
     await db.run(q);
@@ -112,6 +114,21 @@ function getCreateTableQuery(collectionName, collection) {
   query += colQueries.join(", ") + ")";
   return query;
 }
+function getTableIndexQueries(collectionName, collection) {
+  let queries = [];
+  for (const [indexName, indexProps] of Object.entries(collection.indexes ?? {})) {
+    const onColNames = Array.isArray(indexProps.on) ? indexProps.on : [indexProps.on];
+    const onCols = onColNames.map((colName) => sqlite.escapeName(colName));
+    const unique = indexProps.unique ? "UNIQUE " : "";
+    const indexQuery = sql.raw(
+      `CREATE ${unique}INDEX ${sqlite.escapeName(indexName)} ON ${sqlite.escapeName(
+        collectionName
+      )} (${onCols.join(", ")})`
+    );
+    queries.push(indexQuery);
+  }
+  return queries;
+}
 function schemaTypeToSqlType(type) {
   switch (type) {
     case "date":
@@ -204,9 +221,22 @@ function collectionToTable(name, collection, isJsonSerializable = true) {
   for (const [fieldName, field] of Object.entries(collection.fields)) {
     columns[fieldName] = columnMapper(fieldName, field, isJsonSerializable);
   }
-  const table = sqliteTable(name, columns);
+  const table = sqliteTable(name, columns, (ormTable) => {
+    const indexes = {};
+    for (const [indexName, indexProps] of Object.entries(collection.indexes ?? {})) {
+      const onColNames = Array.isArray(indexProps.on) ? indexProps.on : [indexProps.on];
+      const onCols = onColNames.map((colName) => ormTable[colName]);
+      if (!atLeastOne(onCols))
+        continue;
+      indexes[indexName] = index(indexName).on(...onCols);
+    }
+    return indexes;
+  });
   return table;
 }
+function atLeastOne(arr) {
+  return arr.length > 0;
+}
 function columnMapper(fieldName, field, isJsonSerializable) {
   let c;
   switch (field.type) {
@@ -268,9 +298,9 @@ export {
   collectionToTable,
   createLocalDatabaseClient,
   createRemoteDatabaseClient,
-  findLocalDatabase,
   getCreateTableQuery,
   getModifiers,
+  getTableIndexQueries,
   hasDefault,
   hasPrimaryKey,
   schemaTypeToSqlType,

package/dist/root.d.ts

@@ -0,0 +1,3 @@
+import type { AstroConfig } from 'astro';
+export declare function findRoot(): URL;
+export declare function findLocalDatabase(localDbURL: string, output: AstroConfig['output']): string;

package/dist/types.d.ts

@@ -305,6 +305,16 @@ declare const fieldsSchema: z.ZodRecord<z.ZodString, z.ZodUnion<[z.ZodObject<{
     unique?: boolean | undefined;
     default?: unknown;
 }>]>>;
+export declare const indexSchema: z.ZodObject<{
+    on: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodString, "many">]>;
+    unique: z.ZodOptional<z.ZodBoolean>;
+}, "strip", z.ZodTypeAny, {
+    on: (string | string[]) & (string | string[] | undefined);
+    unique?: boolean | undefined;
+}, {
+    on: (string | string[]) & (string | string[] | undefined);
+    unique?: boolean | undefined;
+}>;
 export declare const readableCollectionSchema: z.ZodObject<{
     fields: z.ZodRecord<z.ZodString, z.ZodUnion<[z.ZodObject<{
         label: z.ZodOptional<z.ZodString>;
@@ -406,9 +416,19 @@ export declare const readableCollectionSchema: z.ZodObject<{
         unique?: boolean | undefined;
         default?: unknown;
     }>]>>;
-    writable: z.ZodLiteral<false>;
+    indexes: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodObject<{
+        on: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodString, "many">]>;
+        unique: z.ZodOptional<z.ZodBoolean>;
+    }, "strip", z.ZodTypeAny, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }>>>;
     table: z.ZodAny;
     _setMeta: z.ZodOptional<z.ZodFunction<z.ZodTuple<[], z.ZodUnknown>, z.ZodUnknown>>;
+    writable: z.ZodLiteral<false>;
 }, "strip", z.ZodTypeAny, {
     fields: Record<string, {
         type: "boolean";
@@ -445,6 +465,10 @@ export declare const readableCollectionSchema: z.ZodObject<{
         default?: unknown;
     }>;
     writable: false;
+    indexes?: Record<string, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }> | undefined;
     table?: any;
     _setMeta?: ((...args: unknown[]) => unknown) | undefined;
 }, {
@@ -483,6 +507,10 @@ export declare const readableCollectionSchema: z.ZodObject<{
         default?: unknown;
     }>;
     writable: false;
+    indexes?: Record<string, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }> | undefined;
     table?: any;
     _setMeta?: ((...args: unknown[]) => unknown) | undefined;
 }>;
@@ -587,9 +615,19 @@ export declare const writableCollectionSchema: z.ZodObject<{
         unique?: boolean | undefined;
         default?: unknown;
     }>]>>;
-    writable: z.ZodLiteral<true>;
+    indexes: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodObject<{
+        on: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodString, "many">]>;
+        unique: z.ZodOptional<z.ZodBoolean>;
+    }, "strip", z.ZodTypeAny, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }>>>;
     table: z.ZodAny;
     _setMeta: z.ZodOptional<z.ZodFunction<z.ZodTuple<[], z.ZodUnknown>, z.ZodUnknown>>;
+    writable: z.ZodLiteral<true>;
 }, "strip", z.ZodTypeAny, {
     fields: Record<string, {
         type: "boolean";
@@ -626,6 +664,10 @@ export declare const writableCollectionSchema: z.ZodObject<{
         default?: unknown;
     }>;
     writable: true;
+    indexes?: Record<string, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }> | undefined;
     table?: any;
     _setMeta?: ((...args: unknown[]) => unknown) | undefined;
 }, {
@@ -664,6 +706,10 @@ export declare const writableCollectionSchema: z.ZodObject<{
         default?: unknown;
     }>;
     writable: true;
+    indexes?: Record<string, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }> | undefined;
     table?: any;
     _setMeta?: ((...args: unknown[]) => unknown) | undefined;
 }>;
@@ -768,9 +814,19 @@ export declare const collectionSchema: z.ZodUnion<[z.ZodObject<{
         unique?: boolean | undefined;
         default?: unknown;
     }>]>>;
-    writable: z.ZodLiteral<false>;
+    indexes: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodObject<{
+        on: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodString, "many">]>;
+        unique: z.ZodOptional<z.ZodBoolean>;
+    }, "strip", z.ZodTypeAny, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }>>>;
     table: z.ZodAny;
     _setMeta: z.ZodOptional<z.ZodFunction<z.ZodTuple<[], z.ZodUnknown>, z.ZodUnknown>>;
+    writable: z.ZodLiteral<false>;
 }, "strip", z.ZodTypeAny, {
     fields: Record<string, {
         type: "boolean";
@@ -807,6 +863,10 @@ export declare const collectionSchema: z.ZodUnion<[z.ZodObject<{
         default?: unknown;
     }>;
     writable: false;
+    indexes?: Record<string, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }> | undefined;
     table?: any;
     _setMeta?: ((...args: unknown[]) => unknown) | undefined;
 }, {
@@ -845,6 +905,10 @@ export declare const collectionSchema: z.ZodUnion<[z.ZodObject<{
         default?: unknown;
     }>;
     writable: false;
+    indexes?: Record<string, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }> | undefined;
     table?: any;
     _setMeta?: ((...args: unknown[]) => unknown) | undefined;
 }>, z.ZodObject<{
@@ -948,9 +1012,19 @@ export declare const collectionSchema: z.ZodUnion<[z.ZodObject<{
         unique?: boolean | undefined;
         default?: unknown;
     }>]>>;
-    writable: z.ZodLiteral<true>;
+    indexes: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodObject<{
+        on: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodString, "many">]>;
+        unique: z.ZodOptional<z.ZodBoolean>;
+    }, "strip", z.ZodTypeAny, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }>>>;
     table: z.ZodAny;
     _setMeta: z.ZodOptional<z.ZodFunction<z.ZodTuple<[], z.ZodUnknown>, z.ZodUnknown>>;
+    writable: z.ZodLiteral<true>;
 }, "strip", z.ZodTypeAny, {
     fields: Record<string, {
         type: "boolean";
@@ -987,6 +1061,10 @@ export declare const collectionSchema: z.ZodUnion<[z.ZodObject<{
         default?: unknown;
     }>;
     writable: true;
+    indexes?: Record<string, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }> | undefined;
     table?: any;
     _setMeta?: ((...args: unknown[]) => unknown) | undefined;
 }, {
@@ -1025,6 +1103,10 @@ export declare const collectionSchema: z.ZodUnion<[z.ZodObject<{
         default?: unknown;
     }>;
     writable: true;
+    indexes?: Record<string, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }> | undefined;
     table?: any;
     _setMeta?: ((...args: unknown[]) => unknown) | undefined;
 }>]>;
@@ -1129,9 +1211,19 @@ export declare const collectionsSchema: z.ZodRecord<z.ZodString, z.ZodUnion<[z.Z
         unique?: boolean | undefined;
         default?: unknown;
     }>]>>;
-    writable: z.ZodLiteral<false>;
+    indexes: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodObject<{
+        on: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodString, "many">]>;
+        unique: z.ZodOptional<z.ZodBoolean>;
+    }, "strip", z.ZodTypeAny, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }>>>;
     table: z.ZodAny;
     _setMeta: z.ZodOptional<z.ZodFunction<z.ZodTuple<[], z.ZodUnknown>, z.ZodUnknown>>;
+    writable: z.ZodLiteral<false>;
 }, "strip", z.ZodTypeAny, {
     fields: Record<string, {
         type: "boolean";
@@ -1168,6 +1260,10 @@ export declare const collectionsSchema: z.ZodRecord<z.ZodString, z.ZodUnion<[z.Z
         default?: unknown;
     }>;
     writable: false;
+    indexes?: Record<string, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }> | undefined;
     table?: any;
     _setMeta?: ((...args: unknown[]) => unknown) | undefined;
 }, {
@@ -1206,6 +1302,10 @@ export declare const collectionsSchema: z.ZodRecord<z.ZodString, z.ZodUnion<[z.Z
         default?: unknown;
     }>;
     writable: false;
+    indexes?: Record<string, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }> | undefined;
     table?: any;
     _setMeta?: ((...args: unknown[]) => unknown) | undefined;
 }>, z.ZodObject<{
@@ -1309,9 +1409,19 @@ export declare const collectionsSchema: z.ZodRecord<z.ZodString, z.ZodUnion<[z.Z
         unique?: boolean | undefined;
         default?: unknown;
     }>]>>;
-    writable: z.ZodLiteral<true>;
+    indexes: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodObject<{
+        on: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodString, "many">]>;
+        unique: z.ZodOptional<z.ZodBoolean>;
+    }, "strip", z.ZodTypeAny, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }>>>;
     table: z.ZodAny;
     _setMeta: z.ZodOptional<z.ZodFunction<z.ZodTuple<[], z.ZodUnknown>, z.ZodUnknown>>;
+    writable: z.ZodLiteral<true>;
 }, "strip", z.ZodTypeAny, {
     fields: Record<string, {
         type: "boolean";
@@ -1348,6 +1458,10 @@ export declare const collectionsSchema: z.ZodRecord<z.ZodString, z.ZodUnion<[z.Z
         default?: unknown;
     }>;
     writable: true;
+    indexes?: Record<string, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }> | undefined;
     table?: any;
     _setMeta?: ((...args: unknown[]) => unknown) | undefined;
 }, {
@@ -1386,6 +1500,10 @@ export declare const collectionsSchema: z.ZodRecord<z.ZodString, z.ZodUnion<[z.Z
         default?: unknown;
     }>;
     writable: true;
+    indexes?: Record<string, {
+        on: (string | string[]) & (string | string[] | undefined);
+        unique?: boolean | undefined;
+    }> | undefined;
     table?: any;
     _setMeta?: ((...args: unknown[]) => unknown) | undefined;
 }>]>>;

package/dist/types.js

@@ -40,23 +40,29 @@ const fieldSchema = z.union([
   jsonFieldSchema
 ]);
 const fieldsSchema = z.record(fieldSchema);
-const readableCollectionSchema = z.object({
-  fields: fieldsSchema,
-  writable: z.literal(false),
-  table: z.any(),
-  _setMeta: z.function().optional()
+const indexSchema = z.object({
+  on: z.string().or(z.array(z.string())),
+  unique: z.boolean().optional()
 });
-const writableCollectionSchema = z.object({
+const indexesSchema = z.record(indexSchema);
+const baseCollectionSchema = z.object({
   fields: fieldsSchema,
-  writable: z.literal(true),
+  indexes: indexesSchema.optional(),
   table: z.any(),
   _setMeta: z.function().optional()
 });
+const readableCollectionSchema = baseCollectionSchema.extend({
+  writable: z.literal(false)
+});
+const writableCollectionSchema = baseCollectionSchema.extend({
+  writable: z.literal(true)
+});
 const collectionSchema = z.union([readableCollectionSchema, writableCollectionSchema]);
 const collectionsSchema = z.record(collectionSchema);
 export {
   collectionSchema,
   collectionsSchema,
+  indexSchema,
   readableCollectionSchema,
   writableCollectionSchema
 };

package/dist/utils-runtime.d.ts

@@ -1,2 +1 @@
-export declare function findLocalDatabase(localDbURL: string): string;
 export declare function createRemoteDatabaseClient(appToken: string, remoteDbURL: string): import("drizzle-orm/sqlite-proxy").SqliteRemoteDatabase<Record<string, never>>;

package/dist/utils-runtime.js

@@ -1,15 +1,5 @@
 import { drizzle } from "drizzle-orm/sqlite-proxy";
 import { z } from "zod";
-import { DB_PATH } from "./consts.js";
-function findLocalDatabase(localDbURL) {
-  let dbURL = void 0;
-  if (process.env.VERCEL) {
-    dbURL = new URL(DB_PATH, "file://" + process.cwd() + "/vercel/path0/");
-  } else {
-    dbURL = new URL(localDbURL);
-  }
-  return dbURL.toString();
-}
 function createRemoteDatabaseClient(appToken, remoteDbURL) {
   const url = new URL("./db/query/", remoteDbURL);
   const db = drizzle(async (sql, parameters, method) => {
@@ -58,6 +48,5 @@ Full error: Unexpected JSON response. ${e instanceof Error ? e.message : String(
   return db;
 }
 export {
-  createRemoteDatabaseClient,
-  findLocalDatabase
+  createRemoteDatabaseClient
 };

package/dist/vite-plugin-db.d.ts

@@ -1,17 +1,17 @@
+import type { AstroConfig } from 'astro';
 import type { DBCollections } from './types.js';
 import type { VitePlugin } from './utils.js';
 export declare function vitePluginDb(params: {
     connectToStudio: false;
     collections: DBCollections;
-    dbUrl: string;
 } | {
+    output: AstroConfig['output'];
     connectToStudio: true;
     collections: DBCollections;
     appToken: string;
 }): VitePlugin;
-export declare function getVirtualModContents({ collections, dbUrl, }: {
+export declare function getVirtualModContents({ collections, }: {
     collections: DBCollections;
-    dbUrl: string;
 }): string;
 export declare function getStudioVirtualModContents({ collections, appToken, }: {
     collections: DBCollections;

package/dist/vite-plugin-db.js

@@ -1,5 +1,4 @@
 import { DRIZZLE_MOD_IMPORT, INTERNAL_MOD_IMPORT, VIRTUAL_MODULE_ID, DB_PATH } from "./consts.js";
-import fs from "node:fs";
 const resolvedVirtualModuleId = "\0" + VIRTUAL_MODULE_ID;
 function vitePluginDb(params) {
   return {
@@ -17,27 +16,15 @@ function vitePluginDb(params) {
         return getStudioVirtualModContents(params);
       }
       return getVirtualModContents(params);
-    },
-    async buildEnd() {
-      if ("dbUrl" in params) {
-        const data = await fs.promises.readFile(new URL(params.dbUrl));
-        this.emitFile({
-          fileName: "content.db",
-          source: data,
-          type: "asset"
-        });
-      }
     }
   };
 }
 function getVirtualModContents({
-  collections,
-  dbUrl
+  collections
 }) {
   return `
-import { collectionToTable, createLocalDatabaseClient, findLocalDatabase } from ${INTERNAL_MOD_IMPORT};
-
-export const dbUrl = findLocalDatabase(${JSON.stringify(dbUrl)});
+import { collectionToTable, createLocalDatabaseClient } from ${INTERNAL_MOD_IMPORT};
+import dbUrl from './.astro/content.db?fileurl';
 
 const params = ${JSON.stringify({
     collections,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@astrojs/db",
-  "version": "0.1.16",
+  "version": "0.1.18",
   "description": "",
   "license": "MIT",
   "type": "module",