Add extension support for better-sqlite3-driver.

Author: rkistner

packages/better-sqlite3-driver/src/driver.ts

@@ -12,13 +12,13 @@ import {
 import { WorkerDriverConnection } from '@sqlite-js/driver/worker_threads';
 import type * as bsqlite from 'better-sqlite3';
 
-export interface BetterSqliteDriverOptions
-  extends ConnectionPoolOptions,
-    bsqlite.Options {
+export interface BetterSqliteDriverOptions extends ConnectionPoolOptions {
   /**
    * Specify a custom path to a worker script, to customize the loading process.
    */
   workerPath?: string | URL;
+
+  loadExtensions?: string[];
 }
 
 export class BetterSqliteDriver implements SqliteDriverConnectionPool {
@@ -52,7 +52,7 @@ export class BetterSqliteDriver implements SqliteDriverConnectionPool {
       async openConnection(connectionOptions) {
         return new WorkerDriverConnection(workerPath, path, {
           ...options,
-          readonly: (options?.readonly ?? connectionOptions?.readonly) || false
+          readonly: connectionOptions?.readonly || false
         });
       }
     });

packages/better-sqlite3-driver/src/sync-driver.ts

@@ -30,6 +30,7 @@ import {
   SqliteRun,
   SqliteStep
 } from '@sqlite-js/driver/worker_threads';
+import { BetterSqliteDriverOptions } from './driver.js';
 
 interface InternalStatement extends SqliteDriverStatement {
   getColumnsSync(): string[];
@@ -273,10 +274,18 @@ export class BetterSqliteConnection implements SqliteDriverConnection {
   con: bsqlite.Database;
   private statements = new Map<number, InternalStatement>();
 
-  static open(path: string, options?: bsqlite.Options): BetterSqliteConnection {
+  static open(
+    path: string,
+    options?: bsqlite.Options & BetterSqliteDriverOptions
+  ): BetterSqliteConnection {
     const con = new DatabaseConstructor(path, options);
     con.exec('PRAGMA journal_mode = WAL');
     con.exec('PRAGMA synchronous = normal');
+    if (options?.loadExtensions) {
+      for (let extension of options.loadExtensions) {
+        con.loadExtension(extension);
+      }
+    }
     return new BetterSqliteConnection(con);
   }
 

packages/better-sqlite3-driver/src/worker.ts

@@ -1,11 +1,13 @@
+import { BetterSqliteDriverOptions } from './driver.js';
 import { BetterSqliteConnection } from './sync-driver.js';
 import {
   retriedOpen,
-  setupDriverWorker
+  setupDriverWorker,
+  ConnectionOptions
 } from '@sqlite-js/driver/worker_threads/setup';
 
 setupDriverWorker({
-  async openConnection(options) {
+  async openConnection(options: ConnectionOptions & BetterSqliteDriverOptions) {
     return retriedOpen(() => {
       return BetterSqliteConnection.open(options.path, options);
     }, 2_000);