twonly-app/lib/src/database/signal/connect_session_store.dart

import 'package:drift/drift.dart';
import 'package:libsignal_protocol_dart/libsignal_protocol_dart.dart';
import 'package:twonly/globals.dart';
import 'package:twonly/src/database/twonly_database.dart';

class ConnectSessionStore extends SessionStore {
  @override
  Future<bool> containsSession(SignalProtocolAddress address) async {
    final sessions = await (twonlyDB.select(twonlyDB.signalSessionStores)
          ..where(
            (tbl) =>
                tbl.deviceId.equals(address.getDeviceId()) &
                tbl.name.equals(address.getName()),
          ))
        .get();
    return sessions.isNotEmpty;
  }

  @override
  Future<void> deleteAllSessions(String name) async {
    await (twonlyDB.delete(twonlyDB.signalSessionStores)
          ..where((tbl) => tbl.name.equals(name)))
        .go();
  }

  @override
  Future<void> deleteSession(SignalProtocolAddress address) async {
    await (twonlyDB.delete(twonlyDB.signalSessionStores)
          ..where(
            (tbl) =>
                tbl.deviceId.equals(address.getDeviceId()) &
                tbl.name.equals(address.getName()),
          ))
        .go();
  }

  @override
  Future<List<int>> getSubDeviceSessions(String name) async {
    final deviceIds = await (twonlyDB.select(twonlyDB.signalSessionStores)
          ..where(
            (tbl) => tbl.deviceId.equals(1).not() & tbl.name.equals(name),
          ))
        .get();
    return deviceIds.map((row) => row.deviceId).toList();
  }

  @override
  Future<SessionRecord> loadSession(SignalProtocolAddress address) async {
    final dbSession = await (twonlyDB.select(twonlyDB.signalSessionStores)
          ..where(
            (tbl) =>
                tbl.deviceId.equals(address.getDeviceId()) &
                tbl.name.equals(address.getName()),
          ))
        .get();

    if (dbSession.isEmpty) {
      return SessionRecord();
    }
    final session = dbSession.first.sessionRecord;
    return SessionRecord.fromSerialized(session);
  }

  @override
  Future<void> storeSession(
    SignalProtocolAddress address,
    SessionRecord record,
  ) async {
    final sessionCompanion = SignalSessionStoresCompanion(
      deviceId: Value(address.getDeviceId()),
      name: Value(address.getName()),
      sessionRecord: Value(record.serialize()),
    );

    if (!await containsSession(address)) {
      await twonlyDB
          .into(twonlyDB.signalSessionStores)
          .insert(sessionCompanion);
    } else {
      await (twonlyDB.update(twonlyDB.signalSessionStores)
            ..where(
              (tbl) =>
                  tbl.deviceId.equals(address.getDeviceId()) &
                  tbl.name.equals(address.getName()),
            ))
          .write(sessionCompanion);
    }
  }
}