stack_wallet/lib/db/sqlite/firo_cache_worker.dart

part of 'firo_cache.dart';

enum FCFuncName {
  _updateSparkAnonSetCoinsWith,
  _updateSparkUsedTagsWith,
}

class FCTask {
  final id = const Uuid().v4();
  final FCFuncName func;
  final dynamic data;

  FCTask({required this.func, required this.data});
}

class _FiroCacheWorker {
  final SendPort _commands;
  final ReceivePort _responses;
  final Map<String, Completer<Object?>> _activeRequests = {};

  Future<Object?> runTask(FCTask task) async {
    final completer = Completer<Object?>.sync();
    _activeRequests[task.id] = completer;
    _commands.send(task);
    return await completer.future;
  }

  static Future<_FiroCacheWorker> spawn() async {
    final sqliteDir = await StackFileSystem.applicationSQLiteDirectory();
    final dbFilePath = "${sqliteDir.path}/${_FiroCache.sqliteDbFileName}";

    final initPort = RawReceivePort();
    final connection = Completer<(ReceivePort, SendPort)>.sync();

    initPort.handler = (dynamic initialMessage) {
      final commandPort = initialMessage as SendPort;
      connection.complete(
        (
          ReceivePort.fromRawReceivePort(initPort),
          commandPort,
        ),
      );
    };

    try {
      await Isolate.spawn(
        _startWorkerIsolate,
        (initPort.sendPort, dbFilePath),
      );
    } catch (_) {
      initPort.close();
      rethrow;
    }

    final (receivePort, sendPort) = await connection.future;

    return _FiroCacheWorker._(receivePort, sendPort);
  }

  _FiroCacheWorker._(this._responses, this._commands) {
    _responses.listen(_handleResponsesFromIsolate);
  }

  void _handleResponsesFromIsolate(dynamic message) {
    final (id, error) = message as (String, Object?);
    final completer = _activeRequests.remove(id)!;

    if (error != null) {
      completer.completeError(error);
    } else {
      completer.complete(id);
    }
  }

  static void _handleCommandsToIsolate(
    ReceivePort receivePort,
    SendPort sendPort,
    Database db,
    Mutex mutex,
  ) {
    receivePort.listen((message) {
      final task = message as FCTask;

      mutex.protect(() async {
        try {
          final FCResult result;
          switch (task.func) {
            case FCFuncName._updateSparkAnonSetCoinsWith:
              final data = task.data as (int, Map<String, dynamic>);
              result = _updateSparkAnonSetCoinsWith(db, data.$2, data.$1);
              break;

            case FCFuncName._updateSparkUsedTagsWith:
              result = _updateSparkUsedTagsWith(db, task.data as List<String>);
              break;
          }

          if (result.success) {
            sendPort.send((task.id, null));
          } else {
            sendPort.send((task.id, result.error!));
          }
        } catch (e) {
          sendPort.send((task.id, e));
        }
      });
    });
  }

  static void _startWorkerIsolate((SendPort, String) args) {
    final receivePort = ReceivePort();
    args.$1.send(receivePort.sendPort);
    final mutex = Mutex();
    final db = sqlite3.open(
      args.$2,
      mode: OpenMode.readWrite,
    );
    _handleCommandsToIsolate(receivePort, args.$1, db, mutex);
  }
}