umbrix/lib/features/proxy/active/active_proxy_notifier.dart

import 'package:dio/dio.dart';
import 'package:hiddify/features/connection/notifier/connection_notifier.dart';
import 'package:hiddify/features/proxy/data/proxy_data_providers.dart';
import 'package:hiddify/features/proxy/model/ip_info_entity.dart';
import 'package:hiddify/features/proxy/model/proxy_entity.dart';
import 'package:hiddify/features/proxy/model/proxy_failure.dart';
import 'package:hiddify/utils/riverpod_utils.dart';
import 'package:hiddify/utils/utils.dart';
import 'package:loggy/loggy.dart';
import 'package:riverpod_annotation/riverpod_annotation.dart';

part 'active_proxy_notifier.g.dart';

typedef ActiveProxyInfo = ({
  ProxyItemEntity proxy,
  AsyncValue<IpInfo?> ipInfo,
});

@riverpod
Stream<ProxyItemEntity?> activeProxyGroup(ActiveProxyGroupRef ref) async* {
  final serviceRunning = await ref.watch(serviceRunningProvider.future);
  if (!serviceRunning) {
    throw const ServiceNotRunning();
  }
  yield* ref
      .watch(proxyRepositoryProvider)
      .watchActiveProxies()
      .map((event) => event.getOrElse((l) => throw l))
      .map((event) {
    final Map<String, ProxyGroupEntity> itemMap =
        event.fold({}, (Map<String, ProxyGroupEntity> map, item) {
      map[item.tag] = item;
      return map;
    });

    // Start from the item with the tag "Select"
    var current = itemMap["select"];
    ProxyItemEntity? selected;

    // Traverse through the map based on the selected tag until reaching the end
    while (current != null) {
      selected = current.items.firstOrNull;
      current = itemMap[selected?.tag];
    }

    return selected;
  });
}

@riverpod
Future<IpInfo?> proxyIpInfo(ProxyIpInfoRef ref) async {
  final serviceRunning = await ref.watch(serviceRunningProvider.future);
  if (!serviceRunning) {
    return null;
  }
  final cancelToken = CancelToken();
  ref.onDispose(() {
    Loggy("ProxyIpInfo").debug("canceling");
    cancelToken.cancel();
  });
  return ref
      .watch(proxyRepositoryProvider)
      .getCurrentIpInfo(cancelToken)
      .getOrElse(
    (err) {
      Loggy("ProxyIpInfo").error("error getting proxy ip info", err);
      throw err;
    },
  ).run();
}

@riverpod
class ActiveProxyNotifier extends _$ActiveProxyNotifier with AppLogger {
  @override
  AsyncValue<ActiveProxyInfo> build() {
    ref.disposeDelay(const Duration(seconds: 20));
    ref.onDispose(() {
      _urlTestDebouncer.dispose();
    });
    final ipInfo = ref.watch(proxyIpInfoProvider);
    final activeProxies = ref.watch(activeProxyGroupProvider);
    return switch (activeProxies) {
      AsyncData(value: final activeGroup?) =>
        AsyncData((proxy: activeGroup, ipInfo: ipInfo)),
      AsyncError(:final error, :final stackTrace) =>
        AsyncError(error, stackTrace),
      _ => const AsyncLoading(),
    };
  }

  final _urlTestDebouncer = CallbackDebouncer(const Duration(seconds: 1));

  Future<void> refreshIpInfo() async {
    if (state case AsyncData(:final value) when !value.ipInfo.isLoading) {
      ref.invalidate(proxyIpInfoProvider);
    }
  }

  Future<void> urlTest(String groupTag) async {
    _urlTestDebouncer(
      () async {
        loggy.debug("testing group: [$groupTag]");
        if (state case AsyncData()) {
          await ref
              .read(proxyRepositoryProvider)
              .urlTest(groupTag)
              .getOrElse((err) {
            loggy.error("error testing group", err);
            throw err;
          }).run();
        }
      },
    );
  }
}