From 8bf8b255d1b489fd38d986ebecdfcc9f9ab15c92 Mon Sep 17 00:00:00 2001
From: akazwz <50396286+akazwz@users.noreply.github.com>
Date: Tue, 10 Sep 2024 23:29:35 +0800
Subject: [PATCH] feat: Add d1 support for kit push

---
 drizzle-kit/src/cli/commands/push.ts | 2 ++
 drizzle-kit/src/cli/connections.ts   | 8 +++++++-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/drizzle-kit/src/cli/commands/push.ts b/drizzle-kit/src/cli/commands/push.ts
index 404000608..c6feb82dc 100644
--- a/drizzle-kit/src/cli/commands/push.ts
+++ b/drizzle-kit/src/cli/commands/push.ts
@@ -373,6 +373,8 @@ export const sqlitePush = async (
 					await db.query('rollback');
 					process.exit(1);
 				}
+			} else if (credentials.driver === "d1-http") {
+				await db.run(statementsToExecute.join(''));
 			}
 			render(`[${chalk.green('✓')}] Changes applied`);
 		}
diff --git a/drizzle-kit/src/cli/connections.ts b/drizzle-kit/src/cli/connections.ts
index 6c4b44634..87bdca895 100644
--- a/drizzle-kit/src/cli/connections.ts
+++ b/drizzle-kit/src/cli/connections.ts
@@ -500,13 +500,19 @@ export const connectToSQLite = async (
 				params,
 				method,
 			) => {
+				let body:string;
+				if (params.length > 0) {
+					body = JSON.stringify({ sql, params });
+				} else {
+					body = JSON.stringify({ sql });
+				}
 				const res = await fetch(
 					`https://api.cloudflare.com/client/v4/accounts/${credentials.accountId}/d1/database/${credentials.databaseId}/${
 						method === 'values' ? 'raw' : 'query'
 					}`,
 					{
 						method: 'POST',
-						body: JSON.stringify({ sql, params }),
+						body,
 						headers: {
 							'Content-Type': 'application/json',
 							Authorization: `Bearer ${credentials.token}`,