flutter_opencode_client/lib/data/store/connection_stats.dart

import 'dart:io';

import 'package:fl_lib/fl_lib.dart';
import 'package:hive_ce/hive.dart';
import 'package:server_box/data/model/server/connection_stat.dart';

class ConnectionStatsStore extends HiveStore {
  ConnectionStatsStore._() : super('connection_stats');

  static final instance = ConnectionStatsStore._();

  static const _indexBoxName = 'conn_stats_index';
  static const _maxRecordsPerServer = 100;

  late final Box<dynamic> _indexBox;

  @override
  Future<void> init() async {
    await super.init();
    _indexBox = await Hive.openBox(
      _indexBoxName,
      path: box.path?.substring(0, box.path!.lastIndexOf(Pfs.seperator)),
    );
  }

  Future<void> rebuildIndexAndCompact() async {
    await _cleanAllOldAndRebuildIndex();
    await _compactIfNeeded();
  }

  Future<void> _rebuildIndexCore() async {
    final cutoffTime = DateTime.now().subtract(const Duration(days: 30));
    final serverIdToKeys = <String, List<String>>{};

    for (final key in keys().toList()) {
      final stat = get<ConnectionStat>(key);
      if (stat == null) continue;

      if (stat.timestamp.isBefore(cutoffTime)) {
        remove(key);
        continue;
      }

      final serverId = stat.serverId;
      serverIdToKeys.putIfAbsent(serverId, () => []).add(key);
    }

    final idxKeysToDelete = _indexBox.keys.where((k) => k.toString().startsWith('idx_')).toList();
    for (final k in idxKeysToDelete) {
      await _indexBox.delete(k);
    }

    for (final entry in serverIdToKeys.entries) {
      final keys = entry.value;
      if (keys.length > _maxRecordsPerServer) {
        final keyStatPairs = <(String, ConnectionStat)>[];
        for (final key in keys) {
          final stat = get<ConnectionStat>(key);
          if (stat != null) keyStatPairs.add((key, stat));
        }
        keyStatPairs.sort((a, b) => b.$2.timestamp.compareTo(a.$2.timestamp));
        final toKeep = keyStatPairs.take(_maxRecordsPerServer).map((p) => p.$1).toList().reversed.toList();
        final toRemove = keyStatPairs.skip(_maxRecordsPerServer);
        for (final pair in toRemove) {
          remove(pair.$1);
        }
        await _indexBox.put('idx_${entry.key}', toKeep);
      } else {
        await _indexBox.put('idx_${entry.key}', keys);
      }
    }
  }

  Future<void> _cleanAllOldAndRebuildIndex() async {
    await _rebuildIndexCore();
  }

  Future<void> _compactIfNeeded() async {
    try {
      await box.compact();
      await _indexBox.compact();
    } catch (e, st) {
      Loggers.app.warning('Auto compact failed during init', e, st);
    }
  }

  Future<void> _updateIndex(String serverId, String recordKey) async {
    final indexKey = 'idx_$serverId';
    final keys = (_indexBox.get(indexKey) as List?)?.cast<String>().toList() ?? [];

    if (!keys.contains(recordKey)) {
      keys.add(recordKey);
      if (keys.length > _maxRecordsPerServer) {
        await _pruneExcessRecords(serverId, keys);
      }
      await _indexBox.put(indexKey, keys);
    }
  }

  Future<void> _pruneExcessRecords(String serverId, List<String> keys) async {
    if (keys.length <= _maxRecordsPerServer) return;

    final keyStatPairs = <(String, ConnectionStat)>[];
    for (final key in keys) {
      final stat = get<ConnectionStat>(key);
      if (stat != null) {
        keyStatPairs.add((key, stat));
      }
    }

    keyStatPairs.sort((a, b) => b.$2.timestamp.compareTo(a.$2.timestamp));

    final toRemove = keyStatPairs.skip(_maxRecordsPerServer);
    for (final pair in toRemove) {
      remove(pair.$1);
      keys.remove(pair.$1);
    }
  }

  Future<void> recordConnection(ConnectionStat stat) async {
    final key = '${stat.serverId}_${stat.timestamp.millisecondsSinceEpoch}';
    set(key, stat);
    await _updateIndex(stat.serverId, key);
  }

  ServerConnectionStats getServerStats(String serverId, String serverName) {
    final allStats = getConnectionHistory(serverId);

    if (allStats.isEmpty) {
      return ServerConnectionStats(
        serverId: serverId,
        serverName: serverName,
        totalAttempts: 0,
        successCount: 0,
        failureCount: 0,
        recentConnections: [],
        successRate: 0.0,
      );
    }

    final totalAttempts = allStats.length;
    final successCount = allStats.where((s) => s.result.isSuccess).length;
    final failureCount = totalAttempts - successCount;
    final successRate = totalAttempts > 0 ? (successCount / totalAttempts) : 0.0;

    final successTimes = allStats
        .where((s) => s.result.isSuccess)
        .map((s) => s.timestamp)
        .toList();
    final failureTimes = allStats
        .where((s) => !s.result.isSuccess)
        .map((s) => s.timestamp)
        .toList();

    DateTime? lastSuccessTime;
    DateTime? lastFailureTime;

    if (successTimes.isNotEmpty) {
      successTimes.sort((a, b) => b.compareTo(a));
      lastSuccessTime = successTimes.first;
    }

    if (failureTimes.isNotEmpty) {
      failureTimes.sort((a, b) => b.compareTo(a));
      lastFailureTime = failureTimes.first;
    }

    final recentConnections = allStats.take(20).toList();

    return ServerConnectionStats(
      serverId: serverId,
      serverName: serverName,
      totalAttempts: totalAttempts,
      successCount: successCount,
      failureCount: failureCount,
      lastSuccessTime: lastSuccessTime,
      lastFailureTime: lastFailureTime,
      recentConnections: recentConnections,
      successRate: successRate,
    );
  }

  List<ConnectionStat> getConnectionHistory(String serverId) {
    final indexKey = 'idx_$serverId';
    final keys = (_indexBox.get(indexKey) as List?)?.cast<String>() ?? [];

    final stats = <ConnectionStat>[];
    for (final key in keys) {
      final stat = get<ConnectionStat>(key);
      if (stat != null) {
        stats.add(stat);
      }
    }

    stats.sort((a, b) => b.timestamp.compareTo(a.timestamp));
    return stats;
  }

  List<ServerConnectionStats> getAllServerStats() {
    final indexKeys = _indexBox.keys
        .where((k) => k is String && k.startsWith('idx_'))
        .cast<String>()
        .toList();

    final allStats = <ServerConnectionStats>[];
    for (final indexKey in indexKeys) {
      final serverId = indexKey.substring(4);
      final keys = (_indexBox.get(indexKey) as List?)?.cast<String>() ?? [];

      if (keys.isEmpty) continue;

      String? serverName;
      for (final key in keys.reversed) {
        final stat = get<ConnectionStat>(key);
        if (stat != null) {
          serverName = stat.serverName;
          break;
        }
      }

      if (serverName == null) continue;

      final stats = getServerStats(serverId, serverName);
      allStats.add(stats);
    }

    return allStats;
  }

  Future<void> clearAll() async {
    await box.clear();
    await _indexBox.clear();
  }

  Future<void> clearServerStats(String serverId) async {
    final indexKey = 'idx_$serverId';
    final keys = (_indexBox.get(indexKey) as List?)?.cast<String>() ?? [];

    for (final key in keys) {
      remove(key);
    }
    await _indexBox.delete(indexKey);
  }

  Future<void> compact() async {
    Loggers.app.info('Start compacting connection_stats database...');
    try {
      await box.compact();
      await _indexBox.compact();
      Loggers.app.info('Finished compacting connection_stats database');
    } catch (e, st) {
      Loggers.app.warning('Failed compacting connection_stats database', e, st);
      rethrow;
    }
  }

  String? get dbPath => box.path;

  String? get indexDbPath => _indexBox.path;

  Iterable<dynamic> get indexDbKeys => _indexBox.keys.where((k) => k.toString().startsWith('idx_'));

  Future<int> dbSizeAsync() async {
    final path = dbPath;
    if (path == null) return 0;
    final file = File(path);
    return await file.exists() ? await file.length() : 0;
  }

  Future<int> indexDbSizeAsync() async {
    final path = indexDbPath;
    if (path == null) return 0;
    final file = File(path);
    return await file.exists() ? await file.length() : 0;
  }
}