From bacfa9d09b1c91e16d8875fc655c3421490094ac Mon Sep 17 00:00:00 2001
From: Reorx <novoreorx@gmail.com>
Date: Sat, 20 Dec 2025 11:06:59 +0800
Subject: [PATCH] support passing `driver` to SetupEnv

and facilitate this feature to add `driver: 'pglite'` to pglite-new doc
---
 src/content/docs/get-started/pglite-new.mdx | 4 ++--
 src/mdx/get-started/SetupConfig.mdx         | 6 +++---
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/content/docs/get-started/pglite-new.mdx b/src/content/docs/get-started/pglite-new.mdx
index 355a1496e..bf6020292 100644
--- a/src/content/docs/get-started/pglite-new.mdx
+++ b/src/content/docs/get-started/pglite-new.mdx
@@ -47,7 +47,7 @@ import SetupEnv from '@mdx/get-started/SetupEnv.mdx';
 
 #### Step 5 - Setup Drizzle config file
 
-<SetupConfig dialect='postgresql' env_variable='DATABASE_URL'/>
+<SetupConfig dialect='postgresql' env_variable='DATABASE_URL' driver='pglite'/>
 
 #### Step 6 - Applying changes to the database
 
@@ -59,4 +59,4 @@ import SetupEnv from '@mdx/get-started/SetupEnv.mdx';
 
 #### Step 8 - Run index.ts file
 
-<RunFile/>
\ No newline at end of file
+<RunFile/>
diff --git a/src/mdx/get-started/SetupConfig.mdx b/src/mdx/get-started/SetupConfig.mdx
index 5e5bbbcfc..fa0a4a804 100644
--- a/src/mdx/get-started/SetupConfig.mdx
+++ b/src/mdx/get-started/SetupConfig.mdx
@@ -4,7 +4,7 @@ import CodeWithProps from "@mdx/CodeWithProps.astro";
 
 Create a `drizzle.config.ts` file in the root of your project and add the following content:
 
-<CodeWithProps dialect={props.dialect} env_variable={props.env_variable}>
+<CodeWithProps dialect={props.dialect} env_variable={props.env_variable} driverLine={props.driver ? `\n  driver: '${props.driver}',\n` : ''}>
 ```typescript copy filename="drizzle.config.ts"
 import 'dotenv/config';
 import { defineConfig } from 'drizzle-kit';
@@ -12,10 +12,10 @@ import { defineConfig } from 'drizzle-kit';
 export default defineConfig({
   out: './drizzle',
   schema: './src/db/schema.ts',
-  dialect: '$dialect$',
+  dialect: '$dialect$',$driverLine$
   dbCredentials: {
     url: process.env.$env_variable$!,
   },
 });
 ```
-</CodeWithProps>
\ No newline at end of file
+</CodeWithProps>