Fixed lack of sql imports on uuid defaults

drizzle-team · Aug 8, 2024 · 1f3d4f8 · 1f3d4f8
1 parent 42fa8a9
commit 1f3d4f8
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 1 deletion.
diff --git a/package.json b/package.json
@@ -1,7 +1,7 @@
 {
 	"name": "drizzle-prisma-generator",
 	"author": "Drizzle Team",
-	"version": "0.1.5",
+	"version": "0.1.6",
 	"description": "Generate Drizzle schema from Prisma schema",
 	"scripts": {
 		"run": "pnpm run build && cd ../prisma-testbench && npx prisma generate dev && code ./src/db/schema.ts",

diff --git a/src/util/generators/mysql.ts b/src/util/generators/mysql.ts
@@ -90,6 +90,8 @@ const addColumnModifiers = (field: DMMF.Field, column: string) => {
 
 				if (/^uuid\([0-9]*\)$/.test(value.name)) {
 					column = column + `.default(sql\`uuid()\`)`;
+
+					drizzleImports.add('sql');
 					break;
 				}
 

diff --git a/src/util/generators/pg.ts b/src/util/generators/pg.ts
@@ -91,6 +91,8 @@ const addColumnModifiers = (field: DMMF.Field, column: string) => {
 
 				if (/^uuid\([0-9]*\)$/.test(value.name)) {
 					column = column + `.default(sql\`uuid()\`)`;
+
+					drizzleImports.add('sql');
 					break;
 				}
 

diff --git a/src/util/generators/sqlite.ts b/src/util/generators/sqlite.ts
@@ -84,6 +84,8 @@ const addColumnModifiers = (field: DMMF.Field, column: string) => {
 
 				if (/^uuid\([0-9]*\)$/.test(value.name)) {
 					column = column + `.default(sql\`uuid()\`)`;
+
+					drizzleImports.add('sql');
 					break;
 				}