typesql-cli 0.11.1 → 0.12.0-next2

package/sqlite-query-analyzer/code-generator.js

@@ -113,6 +113,17 @@ function createTsDescriptor(queryInfo, client) {
     return tsDescriptor;
 }
 function mapColumns(client, queryType, columns, returning = false) {
+    const resultColumns = getInsertUpdateResult(client);
+    if (queryType === 'Insert' && !returning) {
+        return resultColumns;
+    }
+    if (queryType === 'Update' || queryType === 'Delete') {
+        return [resultColumns[0]];
+    }
+    const escapedColumnsNames = (0, code_generator_1.renameInvalidNames)(columns.map((col) => col.columnName));
+    return columns.map((col, index) => mapColumnToTsFieldDescriptor(Object.assign(Object.assign({}, col), { columnName: escapedColumnsNames[index] }), client));
+}
+function getInsertUpdateResult(client) {
     const sqliteInsertColumns = [
         {
             name: 'changes',
@@ -137,14 +148,27 @@ function mapColumns(client, queryType, columns, returning = false) {
             notNull: true
         }
     ];
-    if (queryType === 'Insert' && !returning) {
-        return client === 'better-sqlite3' || client == 'bun:sqlite' ? sqliteInsertColumns : libSqlInsertColumns;
-    }
-    if (queryType === 'Update' || queryType === 'Delete') {
-        return client === 'better-sqlite3' || client == 'bun:sqlite' ? [sqliteInsertColumns[0]] : [libSqlInsertColumns[0]];
+    const d1InsertColumns = [
+        {
+            name: 'changes',
+            tsType: 'number',
+            notNull: true
+        },
+        {
+            name: 'last_row_id',
+            tsType: 'number',
+            notNull: true
+        }
+    ];
+    switch (client) {
+        case 'better-sqlite3':
+        case 'bun:sqlite':
+            return sqliteInsertColumns;
+        case 'libsql':
+            return libSqlInsertColumns;
+        case 'd1:sqlite':
+            return d1InsertColumns;
     }
-    const escapedColumnsNames = (0, code_generator_1.renameInvalidNames)(columns.map((col) => col.columnName));
-    return columns.map((col, index) => mapColumnToTsFieldDescriptor(Object.assign(Object.assign({}, col), { columnName: escapedColumnsNames[index] }), client));
 }
 function mapFieldToTsField(columns, field, client) {
     const tsField = {
@@ -227,8 +251,6 @@ function mapColumnType(sqliteType, client) {
             return client === 'better-sqlite3' ? 'Uint8Array' : 'ArrayBuffer';
         case 'BOOLEAN':
             return 'boolean';
-        case 'BOOLEAN':
-            return 'boolean';
     }
     if (sqliteType.startsWith('ENUM')) {
         const enumValues = sqliteType.substring(sqliteType.indexOf('(') + 1, sqliteType.indexOf(')'));
@@ -237,7 +259,7 @@ function mapColumnType(sqliteType, client) {
     return 'any';
 }
 function generateCodeFromTsDescriptor(client, queryName, tsDescriptor, isCrud = false, tableName = '') {
-    var _a;
+    var _a, _b;
     const writer = new code_block_writer_1.default({
         useTabs: true
     });
@@ -257,7 +279,11 @@ function generateCodeFromTsDescriptor(client, queryName, tsDescriptor, isCrud =
     const uniqueUpdateParams = (0, code_generator_1.removeDuplicatedParameters2)(tsDescriptor.data || []);
     const orderByField = generateOrderBy ? `orderBy: [${orderByTypeName}, 'asc' | 'desc'][]` : undefined;
     const paramsTypes = (0, code_generator_1.removeDuplicatedParameters2)(tsDescriptor.dynamicQuery2 == null ? tsDescriptor.parameters : (0, ts_dynamic_query_descriptor_1.mapToDynamicParams)(tsDescriptor.parameters));
-    let functionArguments = client === 'better-sqlite3' || client == 'bun:sqlite' ? 'db: Database' : 'client: Client | Transaction';
+    let functionArguments = client === 'better-sqlite3' || client === 'bun:sqlite'
+        ? 'db: Database'
+        : client === 'd1:sqlite'
+            ? 'db: D1Database'
+            : 'client: Client | Transaction';
     functionArguments += queryType === 'Update' ? `, data: ${dataTypeName}` : '';
     if (tsDescriptor.dynamicQuery2 == null) {
         functionArguments += tsDescriptor.parameters.length > 0 || generateOrderBy ? `, params: ${paramsTypeName}` : '';
@@ -270,17 +296,9 @@ function generateCodeFromTsDescriptor(client, queryName, tsDescriptor, isCrud =
     const allParameters = (((_a = tsDescriptor.data) === null || _a === void 0 ? void 0 : _a.map((param) => fromDriver('data', param))) || []).concat(tsDescriptor.parameters.map((param) => fromDriver('params', param)));
     const queryParamsWithoutBrackets = allParameters.length > 0 ? `${allParameters.join(', ')}` : '';
     const queryParams = queryParamsWithoutBrackets !== '' ? `[${queryParamsWithoutBrackets}]` : '';
-    if (client === 'better-sqlite3') {
-        writer.writeLine(`import type { Database } from 'better-sqlite3';`);
-    }
-    if (client === 'bun:sqlite') {
-        writer.writeLine(`import type { Database } from 'bun:sqlite';`);
-    }
-    if (client === 'libsql') {
-        writer.writeLine(`import type { Client, Transaction } from '@libsql/client';`);
-    }
+    const isDynamicQuery = tsDescriptor.dynamicQuery2 != null;
+    writeImports(writer, client, isDynamicQuery);
     if (tsDescriptor.dynamicQuery2 != null) {
-        writer.writeLine(`import { EOL } from 'os';`);
         writer.blankLine();
         writer.write(`export type ${dynamicParamsTypeName} = `).block(() => {
             writer.writeLine(`select?: ${selectColumnsTypeName};`);
@@ -323,8 +341,8 @@ function generateCodeFromTsDescriptor(client, queryName, tsDescriptor, isCrud =
             });
         });
         writer.blankLine();
-        const asyncModified = client === 'libsql' ? 'async ' : '';
-        const returnTypeModifier = client === 'libsql' ? `Promise<${returnType}>` : returnType;
+        const asyncModified = client === 'libsql' || client === 'd1:sqlite' ? 'async ' : '';
+        const returnTypeModifier = client === 'libsql' || client === 'd1:sqlite' ? `Promise<${returnType}>` : returnType;
         writer.write(`export ${asyncModified}function ${camelCaseName}(${functionArguments}): ${returnTypeModifier}`).block(() => {
             var _a, _b, _c, _d, _e, _f;
             writer.write('const where = whereConditionsToObject(params?.where);').newLine();
@@ -334,7 +352,7 @@ function generateCodeFromTsDescriptor(client, queryName, tsDescriptor, isCrud =
             writer.write('const paramsValues: any = [];').newLine();
             const hasCte = (((_a = tsDescriptor.dynamicQuery2) === null || _a === void 0 ? void 0 : _a.with.length) || 0) > 0;
             if (hasCte) {
-                writer.writeLine(`const withClause = [];`);
+                writer.writeLine('const withClause = [];');
                 (_b = tsDescriptor.dynamicQuery2) === null || _b === void 0 ? void 0 : _b.with.forEach((withFragment, index, array) => {
                     var _a;
                     const selectConditions = withFragment.dependOnFields.map((fieldIndex) => `params.select.${tsDescriptor.columns[fieldIndex].name}`);
@@ -463,6 +481,14 @@ function generateCodeFromTsDescriptor(client, queryName, tsDescriptor, isCrud =
                     .indent()
                     .write(`.map(data => mapArrayTo${resultTypeName}(data, params?.select))${tsDescriptor.multipleRowsResult ? '' : '[0]'};`);
             }
+            if (client === 'd1:sqlite') {
+                writer.write('return db.prepare(sql)').newLine();
+                writer.indent().write('.bind(...paramsValues)').newLine();
+                writer.indent().write('.raw()').newLine();
+                writer
+                    .indent()
+                    .write(`.then(rows => rows.map(row => mapArrayTo${resultTypeName}(row, params?.select)))${tsDescriptor.multipleRowsResult ? '' : '[0]'};`);
+            }
             if (client === 'libsql') {
                 writer.write('return client.execute({ sql, args: paramsValues })').newLine();
                 writer.indent().write('.then(res => res.rows)').newLine();
@@ -631,120 +657,38 @@ function generateCodeFromTsDescriptor(client, queryName, tsDescriptor, isCrud =
         writer.blankLine();
     }
     if (isCrud) {
-        const crudFunction = client === 'libsql'
+        const crudFunction = client === 'libsql' || client === 'd1:sqlite'
             ? `async function ${camelCaseName}(${functionArguments}): Promise<${returnType}>`
             : `function ${camelCaseName}(${functionArguments}): ${returnType}`;
         writer.write(`export ${crudFunction}`).block(() => {
             const idColumn = tsDescriptor.parameters[0].name;
-            writeExecuteCrudBlock(client, queryType, tableName, tsDescriptor.columns, idColumn, client == 'bun:sqlite' ? queryParamsWithoutBrackets : queryParams, paramsTypeName, dataTypeName, resultTypeName, writer);
+            writeExecuteCrudBlock(client, queryType, tableName, tsDescriptor.columns, idColumn, client === 'bun:sqlite' || client === 'd1:sqlite' ? queryParamsWithoutBrackets : queryParams, paramsTypeName, dataTypeName, resultTypeName, writer);
         });
-    }
-    if (tsDescriptor.dynamicQuery2 == null && !isCrud && (queryType === 'Select' || (queryType === 'Insert' && tsDescriptor.returning))) {
-        if (client === 'better-sqlite3') {
-            writer.write(`export function ${camelCaseName}(${functionArguments}): ${returnType}`).block(() => {
-                const processedSql = tsDescriptor.orderByColumns ? (0, code_generator_1.replaceOrderByParam)(sql) : sql;
-                const sqlSplit = processedSql.split('\n');
-                writer.write('const sql = `').newLine();
-                sqlSplit.forEach((sqlLine) => {
-                    writer.indent().write(sqlLine).newLine();
-                });
-                writer.indent().write('`').newLine();
-                writer.write('return db.prepare(sql)').newLine();
-                writer.indent().write('.raw(true)').newLine();
-                writer.indent().write(`.all(${queryParams})`).newLine();
-                writer.indent().write(`.map(data => mapArrayTo${resultTypeName}(data))${tsDescriptor.multipleRowsResult ? '' : '[0]'};`);
-            });
-        }
-        if (client === 'bun:sqlite') {
-            writer.write(`export function ${camelCaseName}(${functionArguments}): ${returnType}`).block(() => {
-                const processedSql = tsDescriptor.orderByColumns ? (0, code_generator_1.replaceOrderByParam)(sql) : sql;
-                const sqlSplit = processedSql.split('\n');
-                writer.write('const sql = `').newLine();
-                sqlSplit.forEach((sqlLine) => {
-                    writer.indent().write(sqlLine).newLine();
-                });
-                writer.indent().write('`').newLine();
-                writer.write('return db.prepare(sql)').newLine();
-                writer.indent().write(`.values(${queryParamsWithoutBrackets})`).newLine();
-                writer.indent().write(`.map(data => mapArrayTo${resultTypeName}(data))${tsDescriptor.multipleRowsResult ? '' : '[0]'};`);
-            });
-        }
-        if (!isCrud && client === 'libsql') {
-            writer.write(`export async function ${camelCaseName}(${functionArguments}): Promise<${returnType}>`).block(() => {
-                const processedSql = tsDescriptor.orderByColumns ? (0, code_generator_1.replaceOrderByParam)(sql) : sql;
-                const sqlSplit = processedSql.split('\n');
-                writer.write('const sql = `').newLine();
-                sqlSplit.forEach((sqlLine) => {
-                    writer.indent().write(sqlLine).newLine();
-                });
-                writer.indent().write('`').newLine();
-                const executeParams = queryParams !== '' ? `{ sql, args: ${queryParams} }` : 'sql';
-                writer.write(`return client.execute(${executeParams})`).newLine();
-                if (queryType === 'Select') {
-                    writer.indent().write('.then(res => res.rows)').newLine();
-                    if (tsDescriptor.multipleRowsResult) {
-                        writer.indent().write(`.then(rows => rows.map(row => mapArrayTo${resultTypeName}(row)));`);
-                    }
-                    else {
-                        writer.indent().write(`.then(rows => mapArrayTo${resultTypeName}(rows[0]));`);
-                    }
-                }
-                if (queryType === 'Insert') {
-                    if (tsDescriptor.returning) {
-                        writer.indent().write('.then(res => res.rows)').newLine();
-                        writer.indent().write(`.then(rows => mapArrayTo${resultTypeName}(rows[0]));`);
-                    }
-                    else {
-                        writer.indent().write(`.then(res => mapArrayTo${resultTypeName}(res));`);
-                    }
-                }
-            });
-        }
-    }
-    if (!isCrud && (queryType === 'Update' || queryType === 'Delete' || (queryType === 'Insert' && !tsDescriptor.returning))) {
-        if (client === 'better-sqlite3' || client === 'bun:sqlite') {
-            writer.write(`export function ${camelCaseName}(${functionArguments}): ${resultTypeName}`).block(() => {
-                const params = client === 'better-sqlite3' ? queryParams : queryParamsWithoutBrackets;
-                writeExecuteBlock(sql, params, resultTypeName, writer);
-            });
+        if (client !== 'd1:sqlite' && (queryType === 'Select' || tsDescriptor.returning)) {
+            writer.blankLine();
+            writeMapFunction(writer, { resultTypeName, columns: tsDescriptor.columns });
         }
-        if (client === 'libsql') {
-            writer.write(`export async function ${camelCaseName}(${functionArguments}): Promise<${resultTypeName}>`).block(() => {
-                const sqlSplit = sql.split('\n');
-                writer.write('const sql = `').newLine();
-                sqlSplit.forEach((sqlLine) => {
-                    writer.indent().write(sqlLine).newLine();
-                });
-                writer.indent().write('`').newLine();
-                const executeParams = queryParams !== '' ? `{ sql, args: ${queryParams} }` : 'sql';
-                writer.write(`return client.execute(${executeParams})`).newLine();
-                writer.indent().write(`.then(res => mapArrayTo${resultTypeName}(res));`);
-            });
+        if (client === 'libsql' && queryType !== 'Select' && !tsDescriptor.returning) {
+            writer.blankLine();
+            writeMapFunctionByName(writer, { resultTypeName, columns: tsDescriptor.columns });
         }
     }
-    if ((queryType === 'Select' || tsDescriptor.returning) && tsDescriptor.dynamicQuery2 == null) {
-        writer.blankLine();
-        writer.write(`function mapArrayTo${resultTypeName}(data: any) `).block(() => {
-            writer.write(`const result: ${resultTypeName} = `).block(() => {
-                tsDescriptor.columns.forEach((col, index) => {
-                    const separator = index < tsDescriptor.columns.length - 1 ? ',' : '';
-                    writer.writeLine(`${col.name}: ${toDriver(`data[${index}]`, col)}${separator}`);
-                });
-            });
-            writer.writeLine('return result;');
-        });
-    }
-    else if (client === 'libsql' && !tsDescriptor.returning && tsDescriptor.dynamicQuery2 == null) {
-        writer.blankLine();
-        writer.write(`function mapArrayTo${resultTypeName}(data: any) `).block(() => {
-            writer.write(`const result: ${resultTypeName} = `).block(() => {
-                tsDescriptor.columns.forEach((col, index) => {
-                    const separator = index < tsDescriptor.columns.length - 1 ? ',' : '';
-                    writer.writeLine(`${col.name}: data.${col.name}${separator}`);
-                });
-            });
-            writer.writeLine('return result;');
-        });
+    const executeFunctionParams = {
+        functionName: camelCaseName,
+        returnType,
+        resultTypeName,
+        dataTypeName,
+        sql: (0, code_generator_1.replaceOrderByParam)(sql),
+        multipleRowsResult: tsDescriptor.multipleRowsResult,
+        parameters: allParameters,
+        columns: tsDescriptor.columns,
+        queryType,
+        paramsTypeName,
+        returning: tsDescriptor.returning || false,
+        orderBy: (((_b = tsDescriptor.orderByColumns) === null || _b === void 0 ? void 0 : _b.length) || 0) > 0
+    };
+    if (tsDescriptor.dynamicQuery2 == null && !isCrud) {
+        writeExecFunction(writer, client, executeFunctionParams);
     }
     if (tsDescriptor.orderByColumns) {
         const orderByType = tsDescriptor.dynamicQuery2 == null ? paramsTypeName : dynamicParamsTypeName;
@@ -786,7 +730,7 @@ function generateCodeFromTsDescriptor(client, queryName, tsDescriptor, isCrud =
         relations.forEach((relation, index) => {
             const relationType = (0, code_generator_1.generateRelationType)(capitalizedName, relation.name);
             if (index === 0) {
-                if (client === 'better-sqlite3' || client == 'bun:sqlite') {
+                if (client === 'better-sqlite3' || client === 'bun:sqlite') {
                     writer.write(`export function ${camelCaseName}Nested(${functionArguments}): ${relationType}[]`).block(() => {
                         const params = tsDescriptor.parameters.length > 0 ? ', params' : '';
                         writer.writeLine(`const selectResult = ${camelCaseName}(db${params});`);
@@ -796,10 +740,11 @@ function generateCodeFromTsDescriptor(client, queryName, tsDescriptor, isCrud =
                         writer.writeLine(`return collect${relationType}(selectResult);`);
                     });
                 }
-                else if (client === 'libsql') {
+                else if (client === 'libsql' || client === 'd1:sqlite') {
                     writer.write(`export async function ${camelCaseName}Nested(${functionArguments}): Promise<${relationType}[]>`).block(() => {
                         const params = tsDescriptor.parameters.length > 0 ? ', params' : '';
-                        writer.writeLine(`const selectResult = await ${camelCaseName}(client${params});`);
+                        const functionParam = client === 'libsql' ? `client${params}` : `db${params}`;
+                        writer.writeLine(`const selectResult = await ${camelCaseName}(${functionParam});`);
                         writer.write('if (selectResult.length == 0)').block(() => {
                             writer.writeLine('return [];');
                         });
@@ -823,16 +768,6 @@ function generateCodeFromTsDescriptor(client, queryName, tsDescriptor, isCrud =
     }
     return writer.toString();
 }
-function writeExecuteBlock(sql, queryParams, resultTypeName, writer) {
-    const sqlSplit = sql.split('\n');
-    writer.write('const sql = `').newLine();
-    sqlSplit.forEach((sqlLine) => {
-        writer.indent().write(sqlLine).newLine();
-    });
-    writer.indent().write('`').newLine();
-    writer.write('return db.prepare(sql)').newLine();
-    writer.indent().write(`.run(${queryParams}) as ${resultTypeName};`);
-}
 function writeExecuteCrudBlock(client, queryType, tableName, columns, idColumn, queryParams, paramTypeName, dataTypeName, resultTypeName, writer) {
     switch (queryType) {
         case 'Select':
@@ -866,6 +801,11 @@ function writeExecutSelectCrudBlock(client, tableName, idColumn, columns, queryP
         writer.indent().write(`.values(${queryParams})`).newLine();
         writer.indent().write(`.map(data => mapArrayTo${resultTypeName}(data))[0];`);
     }
+    else if (client === 'd1:sqlite') {
+        writer.write('return db.prepare(sql)').newLine();
+        writer.indent().write(`.bind(${queryParams})`).newLine();
+        writer.indent().write('.first();').newLine();
+    }
     else {
         writer.write(`return client.execute({ sql, args: ${queryParams} })`).newLine();
         writer.indent().write('.then(res => res.rows)').newLine();
@@ -886,10 +826,16 @@ function writeExecuteInsertCrudBlock(client, tableName, paramTypeName, resultTyp
         writer.write('return db.prepare(sql)').newLine();
         writer.indent().write(`.run(values) as ${resultTypeName};`);
     }
-    else if (client == 'bun:sqlite') {
+    else if (client === 'bun:sqlite') {
         writer.write('return db.prepare(sql)').newLine();
         writer.indent().write(`.run(...values) as ${resultTypeName};`);
     }
+    else if (client === 'd1:sqlite') {
+        writer.write('return db.prepare(sql)').newLine();
+        writer.indent().write('.bind(...values)').newLine();
+        writer.indent().write('.run()').newLine();
+        writer.indent().write('.then(res => res.meta);');
+    }
     else {
         writer.write('return client.execute({ sql, args: values })').newLine();
         writer.indent().write(`.then(res => mapArrayTo${resultTypeName}(res));`).newLine();
@@ -914,6 +860,12 @@ function writeExecuteUpdateCrudBlock(client, tableName, idColumn, paramTypeName,
         writer.write('return db.prepare(sql)').newLine();
         writer.indent().write(`.run(...values) as ${resultTypeName};`);
     }
+    else if (client === 'd1:sqlite') {
+        writer.write('return db.prepare(sql)').newLine();
+        writer.indent().write(`.bind(params.${idColumn})`).newLine();
+        writer.indent().write('.run()').newLine();
+        writer.indent().write('.then(res => res.meta);');
+    }
     else {
         writer.write('return client.execute({ sql, args: values })').newLine();
         writer.indent().write(`.then(res => mapArrayTo${resultTypeName}(res));`).newLine();
@@ -933,6 +885,12 @@ function writeExecutDeleteCrudBlock(client, tableName, idColumn, queryParams, re
         writer.write('return db.prepare(sql)').newLine();
         writer.indent().write(`.run(${queryParams}) as ${resultTypeName};`).newLine();
     }
+    else if (client === 'd1:sqlite') {
+        writer.write('return db.prepare(sql)').newLine();
+        writer.indent().write(`.bind(${queryParams})`).newLine();
+        writer.indent().write('.run()').newLine();
+        writer.indent().write('.then(res => res.meta);');
+    }
     else {
         writer.write(`return client.execute({ sql, args: ${queryParams} })`).newLine();
         writer.indent().write(`.then(res => mapArrayTo${resultTypeName}(res));`).newLine();
@@ -989,4 +947,191 @@ function writeCollectFunction(writer, relation, columns, capitalizedName, result
             .write('))');
     });
 }
+function writeImports(writer, client, isDynamicQuery) {
+    switch (client) {
+        case 'better-sqlite3':
+            writer.writeLine(`import type { Database } from 'better-sqlite3';`);
+            if (isDynamicQuery) {
+                writer.writeLine(`import { EOL } from 'os';`);
+            }
+            return;
+        case 'libsql':
+            writer.writeLine(`import type { Client, Transaction } from '@libsql/client';`);
+            if (isDynamicQuery) {
+                writer.writeLine(`import { EOL } from 'os';`);
+            }
+            return;
+        case 'bun:sqlite':
+            writer.writeLine(`import type { Database } from 'bun:sqlite';`);
+            if (isDynamicQuery) {
+                writer.writeLine(`import { EOL } from 'os';`);
+            }
+            return;
+        case 'd1:sqlite':
+            writer.writeLine(`import type { D1Database } from '@cloudflare/workers-types';`);
+            if (isDynamicQuery) {
+                writer.writeLine(`const EOL = '\\n';`);
+            }
+            return;
+        default:
+            return client;
+    }
+}
+function writeExecFunction(writer, client, params) {
+    const { functionName, returnType, sql, multipleRowsResult, parameters, columns, queryType, returning, paramsTypeName, resultTypeName, dataTypeName, orderBy } = params;
+    let restParameters = queryType === 'Update' ? `, data: ${dataTypeName}` : '';
+    const dynamicQuery = false;
+    if (!dynamicQuery) {
+        restParameters += parameters.length > 0 || orderBy ? `, params: ${paramsTypeName}` : '';
+    }
+    const queryParametersWithoutBrackes = parameters.join(', ');
+    const queryParams = queryParametersWithoutBrackes != '' ? `[${queryParametersWithoutBrackes}]` : '';
+    const mapFunctionParams = {
+        resultTypeName,
+        columns
+    };
+    switch (client) {
+        case 'better-sqlite3':
+            const betterSqliteArgs = 'db: Database' + restParameters;
+            if (queryType === 'Select') {
+                writer.write(`export function ${functionName}(${betterSqliteArgs}): ${returnType}`).block(() => {
+                    writeSql(writer, sql);
+                    writer.write('return db.prepare(sql)').newLine();
+                    writer.indent().write('.raw(true)').newLine();
+                    writer.indent().write(`.all(${queryParams})`).newLine();
+                    writer.indent().write(`.map(data => mapArrayTo${resultTypeName}(data))${multipleRowsResult ? '' : '[0]'};`);
+                });
+                writer.blankLine();
+                writeMapFunction(writer, mapFunctionParams);
+            }
+            if (queryType === 'Update' || queryType === 'Delete' || (queryType === 'Insert' && !returning)) {
+                writer.write(`export function ${functionName}(${betterSqliteArgs}): ${resultTypeName}`).block(() => {
+                    writeSql(writer, sql);
+                    writer.write('return db.prepare(sql)').newLine();
+                    writer.indent().write(`.run(${queryParams}) as ${resultTypeName};`);
+                });
+            }
+            return;
+        case 'libsql':
+            const libSqlArgs = 'client: Client | Transaction' + restParameters;
+            writer.write(`export async function ${functionName}(${libSqlArgs}): Promise<${returnType}>`).block(() => {
+                writeSql(writer, sql);
+                const executeParams = queryParametersWithoutBrackes !== '' ? `{ sql, args: [${queryParametersWithoutBrackes}] }` : 'sql';
+                writer.write(`return client.execute(${executeParams})`).newLine();
+                if (queryType === 'Select') {
+                    writer.indent().write('.then(res => res.rows)').newLine();
+                    if (multipleRowsResult) {
+                        writer.indent().write(`.then(rows => rows.map(row => mapArrayTo${resultTypeName}(row)));`);
+                    }
+                    else {
+                        writer.indent().write(`.then(rows => mapArrayTo${resultTypeName}(rows[0]));`);
+                    }
+                }
+                if (queryType === 'Insert') {
+                    if (returning) {
+                        writer.indent().write('.then(res => res.rows)').newLine();
+                        writer.indent().write(`.then(rows => mapArrayTo${resultTypeName}(rows[0]));`);
+                    }
+                }
+                if (queryType === 'Update' || queryType === 'Delete' || (queryType === 'Insert' && !returning)) {
+                    writer.indent().write(`.then(res => mapArrayTo${resultTypeName}(res));`);
+                }
+            });
+            writer.blankLine();
+            if (queryType === 'Select' || returning) {
+                writeMapFunction(writer, mapFunctionParams);
+            }
+            else {
+                writeMapFunctionByName(writer, mapFunctionParams);
+            }
+            return;
+        case 'bun:sqlite':
+            const bunArgs = 'db: Database' + restParameters;
+            if (queryType === 'Select') {
+                writer.write(`export function ${functionName}(${bunArgs}): ${returnType}`).block(() => {
+                    writeSql(writer, sql);
+                    writer.write('return db.prepare(sql)').newLine();
+                    writer.indent().write(`.values(${queryParametersWithoutBrackes})`).newLine();
+                    writer.indent().write(`.map(data => mapArrayTo${resultTypeName}(data))${multipleRowsResult ? '' : '[0]'};`);
+                });
+                writer.blankLine();
+                writeMapFunction(writer, mapFunctionParams);
+            }
+            if (queryType === 'Update' || queryType === 'Delete' || (queryType === 'Insert' && !returning)) {
+                writer.write(`export function ${functionName}(${bunArgs}): ${resultTypeName}`).block(() => {
+                    writeSql(writer, sql);
+                    writer.write('return db.prepare(sql)').newLine();
+                    writer.indent().write(`.run(${queryParametersWithoutBrackes}) as ${resultTypeName};`);
+                });
+            }
+            return;
+        case 'd1:sqlite':
+            const d1Args = 'db: D1Database' + restParameters;
+            writer.write(`export async function ${functionName}(${d1Args}): Promise<${returnType}>`).block(() => {
+                writeSql(writer, sql);
+                writer.write('return db.prepare(sql)').newLine();
+                if (queryParametersWithoutBrackes !== '') {
+                    writer.indent().write(`.bind(${queryParametersWithoutBrackes})`).newLine();
+                }
+                if (queryType === 'Select') {
+                    writer.indent().write('.raw({ columnNames: false })').newLine();
+                    if (multipleRowsResult) {
+                        writer.indent().write(`.then(rows => rows.map(row => mapArrayTo${resultTypeName}(row)));`);
+                    }
+                    else {
+                        writer.indent().write(`.then(rows => rows.map(row => mapArrayTo${resultTypeName}(row))[0]);`);
+                    }
+                }
+                if (queryType === 'Insert' || queryType === 'Update' || queryType === 'Delete') {
+                    if (returning) {
+                        writer.indent().write('.raw({ columnNames: false })').newLine();
+                        writer.indent().write(`.then(rows => rows.map(row => mapArrayTo${resultTypeName}(row))[0]);`);
+                    }
+                    else {
+                        writer.indent().write('.run()').newLine();
+                        writer.indent().write(`.then(res => res.meta);`);
+                    }
+                }
+            });
+            if (queryType === 'Select' || returning) {
+                writer.blankLine();
+                writeMapFunction(writer, mapFunctionParams);
+            }
+            return;
+        default:
+            return client;
+    }
+}
+function writeSql(writer, sql) {
+    const sqlSplit = sql.split('\n');
+    writer.write('const sql = `').newLine();
+    sqlSplit.forEach((sqlLine) => {
+        writer.indent().write(sqlLine).newLine();
+    });
+    writer.indent().write('`').newLine();
+}
+function writeMapFunction(writer, params) {
+    const { resultTypeName, columns } = params;
+    writer.write(`function mapArrayTo${resultTypeName}(data: any) `).block(() => {
+        writer.write(`const result: ${resultTypeName} = `).block(() => {
+            columns.forEach((col, index) => {
+                const separator = index < columns.length - 1 ? ',' : '';
+                writer.writeLine(`${col.name}: ${toDriver(`data[${index}]`, col)}${separator}`);
+            });
+        });
+        writer.writeLine('return result;');
+    });
+}
+function writeMapFunctionByName(writer, params) {
+    const { resultTypeName, columns } = params;
+    writer.write(`function mapArrayTo${resultTypeName}(data: any) `).block(() => {
+        writer.write(`const result: ${resultTypeName} = `).block(() => {
+            columns.forEach((col, index) => {
+                const separator = index < columns.length - 1 ? ',' : '';
+                writer.writeLine(`${col.name}: data.${col.name}${separator}`);
+            });
+        });
+        writer.writeLine('return result;');
+    });
+}
 //# sourceMappingURL=code-generator.js.map