Fix db seeding when srcDir is root

Author: matthewp

.changeset/spotty-hounds-ring.md

@@ -0,0 +1,5 @@
+---
+"@astrojs/db": patch
+---
+
+Fix db seeding when srcDir is root

packages/db/src/core/integration/vite-plugin-db.ts

@@ -40,6 +40,7 @@ type VitePluginDBParams =
 
 export function vitePluginDb(params: VitePluginDBParams): VitePlugin {
 	const srcDirPath = normalizePath(fileURLToPath(params.srcDir));
+	const dbDirPath = normalizePath(fileURLToPath(getDbDirectoryUrl(params.root)));
 	let command: 'build' | 'serve' = 'build';
 	return {
 		name: 'astro:db',
@@ -54,7 +55,7 @@ export function vitePluginDb(params: VitePluginDBParams): VitePlugin {
 			const importer = rawImporter ? await this.resolve(rawImporter) : null;
 			if (!importer) return resolved.virtual;
 
-			if (importer.id.startsWith(srcDirPath)) {
+			if (importer.id.startsWith(srcDirPath) && !importer.id.startsWith(dbDirPath)) {
 				// Seed only if the importer is in the src directory.
 				// Otherwise, we may get recursive seed calls (ex. import from db/seed.ts).
 				return resolved.seedVirtual;

packages/db/test/db-in-src.test.js

@@ -0,0 +1,37 @@
+import { expect } from 'chai';
+import { load as cheerioLoad } from 'cheerio';
+import testAdapter from '../../astro/test/test-adapter.js';
+import { loadFixture } from '../../astro/test/test-utils.js';
+
+describe('astro:db', () => {
+	let fixture;
+	before(async () => {
+		fixture = await loadFixture({
+			root: new URL('./fixtures/db-in-src/', import.meta.url),
+			output: 'server',
+			srcDir: '.',
+			adapter: testAdapter(),
+		});
+	});
+
+	describe('development: db/ folder inside srcDir', () => {
+		let devServer;
+
+		before(async () => {
+			devServer = await fixture.startDevServer();
+		});
+
+		after(async () => {
+			await devServer.stop();
+		});
+
+		it('Prints the list of authors', async () => {
+			const html = await fixture.fetch('/').then((res) => res.text());
+			const $ = cheerioLoad(html);
+
+			const ul = $('.users-list');
+			expect(ul.children()).to.have.a.lengthOf(1);
+			expect($('.users-list li').text()).to.equal('Mario');
+		});
+	});
+});

packages/db/test/fixtures/db-in-src/astro.config.ts

@@ -0,0 +1,10 @@
+import db from '@astrojs/db';
+import { defineConfig } from 'astro/config';
+
+// https://astro.build/config
+export default defineConfig({
+	integrations: [db()],
+	devToolbar: {
+		enabled: false,
+	},
+});

packages/db/test/fixtures/db-in-src/db/config.ts

@@ -0,0 +1,13 @@
+import { column, defineDb, defineTable } from 'astro:db';
+
+const User = defineTable({
+	columns: {
+		id: column.text({ primaryKey: true, optional: false }),
+		username: column.text({ optional: false, unique: true }),
+		password: column.text({ optional: false }),
+	},
+});
+
+export default defineDb({
+	tables: { User },
+});

packages/db/test/fixtures/db-in-src/db/seed.ts

@@ -0,0 +1,8 @@
+import { asDrizzleTable } from '@astrojs/db/utils';
+import { User, db } from 'astro:db';
+
+export default async function () {
+	await db.batch([
+		db.insert(User).values([{ id: 'mario', username: 'Mario', password: 'itsame' }]),
+	]);
+}

packages/db/test/fixtures/db-in-src/package.json

@@ -0,0 +1,14 @@
+{
+  "name": "@test/db-db-in-src",
+  "version": "0.0.0",
+  "private": true,
+  "scripts": {
+    "dev": "astro dev",
+    "build": "astro build",
+    "preview": "astro preview"
+  },
+  "dependencies": {
+    "@astrojs/db": "workspace:*",
+    "astro": "workspace:*"
+  }
+}

packages/db/test/fixtures/db-in-src/pages/index.astro

@@ -0,0 +1,11 @@
+---
+/// <reference path="../.astro/db-types.d.ts" />
+import { db, User } from 'astro:db';
+
+const users = await db.select().from(User);
+---
+
+<h2>Users</h2>
+<ul class="users-list">
+	{users.map((user) => <li>{user.username}</li>)}
+</ul>