sharedinbox/lib/data/repositories/mailbox_repository_impl.dart

import 'package:drift/drift.dart';
import 'package:enough_mail/enough_mail.dart' as imap;

import '../../core/models/mailbox.dart';
import '../../core/repositories/account_repository.dart';
import '../../core/repositories/mailbox_repository.dart';
import '../db/database.dart';
import '../db/database.dart' as db show Mailbox;
import '../imap/imap_client_factory.dart';

class MailboxRepositoryImpl implements MailboxRepository {
  MailboxRepositoryImpl(this._db, this._accounts);

  final AppDatabase _db;
  final AccountRepository _accounts;

  @override
  Stream<List<Mailbox>> observeMailboxes(String accountId) {
    return (_db.select(_db.mailboxes)
          ..where((t) => t.accountId.equals(accountId))
          ..orderBy([(t) => OrderingTerm.asc(t.path)]))
        .watch()
        .map((rows) => rows.map(_toModel).toList());
  }

  @override
  Future<void> syncMailboxes(String accountId) async {
    final account = (await _accounts.getAccount(accountId))!;
    final password = await _accounts.getPassword(accountId);
    final client = await connectImap(account, password);
    try {
      final response = await client.listMailboxes();
      for (final mb in response.mailboxes ?? const <imap.Mailbox>[]) {
        final path = mb.path ?? mb.name;
        final id = '${accountId}:$path';
        await _db.into(_db.mailboxes).insertOnConflictUpdate(
              MailboxesCompanion.insert(
                id: id,
                accountId: accountId,
                path: path,
                name: mb.name,
                unreadCount: Value(mb.messagesUnseen ?? 0),
                totalCount: Value(mb.messagesExists ?? 0),
              ),
            );
      }
    } finally {
      await client.logout();
    }
  }

  Mailbox _toModel(db.Mailbox row) => Mailbox(
        id: row.id,
        accountId: row.accountId,
        path: row.path,
        name: row.name,
        unreadCount: row.unreadCount,
        totalCount: row.totalCount,
      );
}