feat: Add d1 support for kit push

akazwz · Sep 10, 2024 · 8bf8b25 · 8bf8b25
1 parent 239cfa9
commit 8bf8b25
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git 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
@@ -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}`,