Start looking at sqlite3_web integration

Author: simolus3

future/drift_sqlite/lib/src/connection/web.dart

@@ -0,0 +1,84 @@
+import 'dart:js_interop';
+
+import 'package:drift3/drift.dart';
+import 'package:sqlite3/common.dart';
+import 'package:sqlite3_web/protocol_utils.dart' as utils;
+import 'package:sqlite3_web/sqlite3_web.dart';
+
+import 'web/protocol.dart';
+
+abstract base class _BaseWebSession implements DriftSession {
+  final Database database;
+
+  _BaseWebSession(this.database);
+
+  LockToken? get lockToken => null;
+  bool get requireInTransaction => false;
+
+  @override
+  Future<QueryResult> execute(StatementInfo statement) async {
+    final (params, paramTypes) = utils.serializeParameters(
+      statement.variables.map((p) => p.rawValue).toList(),
+    );
+
+    final response = await database.customRequest(
+      ExecuteRequest(
+        sql: statement.sql,
+        needsResultSet: statement.needsResultSet,
+        parameters: params,
+        parameterTypes: paramTypes,
+        requireInTransaction: requireInTransaction,
+      ),
+      token: lockToken,
+    );
+
+    return (response as SerializedQueryResult).asQueryResult();
+  }
+
+  @override
+  Future<List<QueryResult>> executeBatch(StatementBatch batch) {
+    // TODO: implement executeBatch
+    throw UnimplementedError();
+  }
+}
+
+final class WebSession extends _BaseWebSession
+    implements DriftSessionWithInternalLocks {
+  WebSession(super.database);
+
+  @override
+  Future<void> close() {
+    // TODO: implement close
+    throw UnimplementedError();
+  }
+
+  @override
+  // TODO: implement closed
+  Future<void> get closed => throw UnimplementedError();
+
+  @override
+  // TODO: implement isClosed
+  bool get isClosed => throw UnimplementedError();
+
+  @override
+  DriftSessionWithInternalLocks? get locks => this;
+
+  @override
+  PersistentSchemaVersion? get persistentSchemaVersion =>
+      throw UnimplementedError();
+
+  @override
+  Object? get tag => null;
+
+  @override
+  DriftTransactionSession? get transaction => null;
+
+  @override
+  DriftTransactionParent? get transactionParent => throw UnimplementedError();
+
+  @override
+  Future<DriftSession> exclusive() {
+    // TODO: implement exclusive
+    throw UnimplementedError();
+  }
+}

future/drift_sqlite/lib/src/connection/web/protocol.dart

@@ -0,0 +1,67 @@
+import 'dart:js_interop';
+
+import 'package:drift3/drift.dart';
+import 'package:meta/meta.dart';
+import 'package:sqlite3_web/protocol_utils.dart' as utils;
+
+@JS()
+@anonymous
+@internal
+extension type ExecuteRequest._(JSObject _) implements JSObject {
+  @JS('a')
+  external String get sql;
+  @JS('b')
+  external JSArray get parameters;
+  @JS('c')
+  external JSArrayBuffer get parameterTypes;
+  @JS('d')
+  external bool get needsResultSet;
+  @JS('e')
+  external bool requireInTransaction;
+
+  external factory ExecuteRequest({
+    @JS('a') required String sql,
+    @JS('b') required JSArray parameters,
+    @JS('c') required JSArrayBuffer parameterTypes,
+    @JS('d') required bool needsResultSet,
+    @JS('e') required bool requireInTransaction,
+  });
+
+  List<Object?> get decodedParameters =>
+      utils.deserializeParameters(parameters, parameterTypes);
+}
+
+@JS()
+@anonymous
+@internal
+extension type SerializedQueryResult._(JSObject _) implements JSObject {
+  @JS('a')
+  external JSNumber? get affectedRows;
+  @JS('b')
+  external JSNumber? get lastInsertRowId;
+  @JS('c')
+  external JSObject? get serializedResultSet;
+
+  external factory SerializedQueryResult({
+    @JS('a') required JSNumber? affectedRows,
+    @JS('b') required JSNumber? lastInsertRowId,
+    @JS('c') required JSObject? serializedResultSet,
+  });
+
+  QueryResult asQueryResult() {
+    RawResultSet? resultSet;
+    if (serializedResultSet case final serialized?) {
+      final rawResultSet = utils.deserializeResultSet(serialized);
+      RawResultSet.fromRows(
+        columnNames: rawResultSet.columnNames,
+        rows: rawResultSet.rows,
+      );
+    }
+
+    return QueryResult(
+      resultSet: resultSet,
+      affectedRows: affectedRows?.toDartInt,
+      lastInsertRowId: lastInsertRowId?.toDartInt,
+    );
+  }
+}

future/drift_sqlite/lib/src/connection/web/worker.dart

@@ -0,0 +1,46 @@
+import 'dart:js_interop';
+
+import 'package:sqlite3/common.dart';
+import 'package:sqlite3_web/sqlite3_web.dart';
+import 'package:sqlite3_web/protocol_utils.dart' as utils;
+
+import '../shared.dart';
+import 'protocol.dart';
+
+final class DriftDatabase extends WorkerDatabase {
+  @override
+  final CommonDatabase database;
+
+  DriftDatabase(this.database);
+
+  @override
+  Future<JSAny?> handleCustomRequest(
+    ClientConnection connection,
+    CustomClientDatabaseRequest request,
+  ) {
+    final data = request.request as ExecuteRequest;
+
+    return request.useLock(() {
+      if (data.needsResultSet && data.parameters.length == 0) {
+        database.execute(data.sql);
+      }
+
+      final stmt = database.prepare(data.sql, checkNoTail: true);
+      final params = data.decodedParameters;
+      ResultSet? resultSet;
+      if (data.needsResultSet) {
+        resultSet = stmt.select(params);
+      } else {
+        stmt.execute(params);
+      }
+
+      return SerializedQueryResult(
+        affectedRows: database.updatedRows.toJS,
+        lastInsertRowId: database.lastInsertRowId.toJS,
+        serializedResultSet: resultSet == null
+            ? null
+            : utils.serializeResultSet(resultSet),
+      );
+    });
+  }
+}

future/drift_sqlite/pubspec.yaml

@@ -15,6 +15,7 @@ dependencies:
   collection: ^1.18.0
   sqlite3: ^3.1.6
   sqlite3_connection_pool: ^0.2.3
+  sqlite3_web: ^0.6.0
 
 dev_dependencies:
   lints: ^6.0.0