Files
umbrix/lib/features/common/connectivity/connectivity_controller.dart
problematicconsumer 577bb3ded3 Fix bootstrap bug
2023-11-12 22:23:09 +03:30

114 lines
3.7 KiB
Dart

import 'package:hiddify/core/prefs/prefs.dart';
import 'package:hiddify/core/prefs/service_prefs.dart';
import 'package:hiddify/data/data_providers.dart';
import 'package:hiddify/domain/connectivity/connectivity.dart';
import 'package:hiddify/domain/core_facade.dart';
import 'package:hiddify/features/common/active_profile/active_profile_notifier.dart';
import 'package:hiddify/utils/utils.dart';
import 'package:riverpod_annotation/riverpod_annotation.dart';
import 'package:rxdart/rxdart.dart';
part 'connectivity_controller.g.dart';
@Riverpod(keepAlive: true)
class ConnectivityController extends _$ConnectivityController with AppLogger {
@override
Stream<ConnectionStatus> build() {
ref.listen(
activeProfileProvider.select((value) => value.asData?.value),
(previous, next) async {
if (previous == null) return;
final shouldReconnect = next == null || previous.id != next.id;
if (shouldReconnect) {
await reconnect(next?.id);
}
},
);
return _core.watchConnectionStatus().doOnData((event) {
if (event case Disconnected(:final connectionFailure?)
when PlatformUtils.isDesktop) {
ref.read(startedByUserProvider.notifier).update(false);
}
loggy.info("connection status: ${event.format()}");
});
}
CoreFacade get _core => ref.watch(coreFacadeProvider);
Future<void> mayConnect() async {
if (state case AsyncData(:final value)) {
if (value case Disconnected()) return _connect();
}
}
Future<void> toggleConnection() async {
if (state case AsyncError()) {
await _connect();
} else if (state case AsyncData(:final value)) {
switch (value) {
case Disconnected():
await ref.read(startedByUserProvider.notifier).update(true);
await _connect();
case Connected():
await ref.read(startedByUserProvider.notifier).update(false);
await _disconnect();
default:
loggy.warning("switching status, debounce");
}
}
}
Future<void> reconnect(String? profileId) async {
if (state case AsyncData(:final value) when value == const Connected()) {
if (profileId == null) {
loggy.info("no active profile, disconnecting");
return _disconnect();
}
loggy.info("active profile changed, reconnecting");
await ref.read(startedByUserProvider.notifier).update(true);
await _core
.restart(profileId, ref.read(disableMemoryLimitProvider))
.mapLeft((err) {
loggy.warning("error reconnecting", err);
state = AsyncError(err, StackTrace.current);
}).run();
}
}
Future<void> abortConnection() async {
if (state case AsyncData(:final value)) {
switch (value) {
case Connected() || Connecting():
loggy.debug("aborting connection");
await _disconnect();
default:
}
}
}
Future<void> _connect() async {
final activeProfile = await ref.read(activeProfileProvider.future);
await _core
.start(activeProfile!.id, ref.read(disableMemoryLimitProvider))
.mapLeft((err) async {
loggy.warning("error connecting", err);
await ref.read(startedByUserProvider.notifier).update(false);
state = AsyncError(err, StackTrace.current);
}).run();
}
Future<void> _disconnect() async {
await _core.stop().mapLeft((err) {
loggy.warning("error disconnecting", err);
state = AsyncError(err, StackTrace.current);
}).run();
}
}
@Riverpod(keepAlive: true)
Future<bool> serviceRunning(ServiceRunningRef ref) => ref
.watch(
connectivityControllerProvider.selectAsync((data) => data.isConnected),
)
.onError((error, stackTrace) => false);