Revert "see if tests work"

This reverts commit 3910a48.
sveltejs · Feb 28, 2025 · f923703 · f923703
1 parent 3910a48
commit f923703
Showing 1 changed file with 6 additions and 11 deletions.
diff --git a/packages/addons/drizzle/index.ts b/packages/addons/drizzle/index.ts
@@ -261,6 +261,7 @@ export default defineAddon({
 			const { ast, generateCode } = parseScript(content);
 
 			imports.addNamed(ast, '$env/dynamic/private', { env: 'env' });
+			imports.addNamespace(ast, './schema', 'schema');
 
 			// env var checks
 			const dbURLCheck = common.statementFromString(
@@ -269,7 +270,6 @@ export default defineAddon({
 			common.addStatement(ast, dbURLCheck);
 
 			let clientExpression;
-			let addSchema = true;
 			// SQLite
 			if (options.sqlite === 'better-sqlite3') {
 				imports.addDefault(ast, 'better-sqlite3', 'Database');
@@ -304,14 +304,12 @@ export default defineAddon({
 				clientExpression = common.expressionFromString(
 					'await mysql.createConnection(env.DATABASE_URL)'
 				);
-				addSchema = false;
 			}
 			if (options.mysql === 'planetscale') {
 				imports.addNamed(ast, '@planetscale/database', { Client: 'Client' });
 				imports.addNamed(ast, 'drizzle-orm/planetscale-serverless', { drizzle: 'drizzle' });
 
 				clientExpression = common.expressionFromString('new Client({ url: env.DATABASE_URL })');
-				addSchema = false;
 			}
 			// PostgreSQL
 			if (options.postgresql === 'neon') {
@@ -332,14 +330,11 @@ export default defineAddon({
 			common.addStatement(ast, clientIdentifier);
 
 			const drizzleCall = functions.callByIdentifier('drizzle', ['client']);
-			if (addSchema) {
-				imports.addNamespace(ast, './schema', 'schema');
-				drizzleCall.arguments.push(
-					object.create({
-						schema: variables.identifier('schema')
-					})
-				);
-			}
+			drizzleCall.arguments.push(
+				object.create({
+					schema: variables.identifier('schema')
+				})
+			);
 			const db = variables.declaration(ast, 'const', 'db', drizzleCall);
 			exports.namedExport(ast, 'db', db);