import 'dart:async'; import 'package:easy_localization/easy_localization.dart'; import 'package:hive/hive.dart'; import 'package:pub_semver/pub_semver.dart'; import 'package:selfprivacy/config/get_it_config.dart'; import 'package:selfprivacy/config/hive_config.dart'; import 'package:selfprivacy/logic/api_maps/graphql_maps/server_api/server_api.dart'; import 'package:selfprivacy/logic/models/auto_upgrade_settings.dart'; import 'package:selfprivacy/logic/models/backup.dart'; import 'package:selfprivacy/logic/models/hive/server_details.dart'; import 'package:selfprivacy/logic/models/hive/server_domain.dart'; import 'package:selfprivacy/logic/models/hive/user.dart'; import 'package:selfprivacy/logic/models/json/api_token.dart'; import 'package:selfprivacy/logic/models/json/recovery_token_status.dart'; import 'package:selfprivacy/logic/models/json/server_disk_volume.dart'; import 'package:selfprivacy/logic/models/json/server_job.dart'; import 'package:selfprivacy/logic/models/service.dart'; import 'package:selfprivacy/logic/models/ssh_settings.dart'; import 'package:selfprivacy/logic/models/system_settings.dart'; /// Repository for all API calls /// Stores the current state of all data from API and exposes it to Blocs. class ApiConnectionRepository { Box box = Hive.box(BNames.serverInstallationBox); final ServerApi api = ServerApi(); final ApiData _apiData = ApiData(ServerApi()); ApiData get apiData => _apiData; ConnectionStatus connectionStatus = ConnectionStatus.nonexistent; final _dataStream = StreamController.broadcast(); final _connectionStatusStream = StreamController.broadcast(); Stream get dataStream => _dataStream.stream; Stream get connectionStatusStream => _connectionStatusStream.stream; ConnectionStatus get currentConnectionStatus => connectionStatus; Timer? _timer; Future removeServerJob(final String uid) async { await api.removeApiJob(uid); _apiData.serverJobs.data ?.removeWhere((final ServerJob element) => element.uid == uid); _dataStream.add(_apiData); } Future removeAllFinishedServerJobs() async { final List finishedJobs = _apiData.serverJobs.data ?.where( (final ServerJob element) => element.status == JobStatusEnum.finished || element.status == JobStatusEnum.error, ) .toList() ?? []; // Optimistically remove the jobs from the list _apiData.serverJobs.data?.removeWhere( (final ServerJob element) => element.status == JobStatusEnum.finished || element.status == JobStatusEnum.error, ); _dataStream.add(_apiData); await Future.forEach( finishedJobs, (final ServerJob job) async => removeServerJob(job.uid), ); } Future<(bool, String)> createUser(final User user) async { final List? loadedUsers = _apiData.users.data; if (loadedUsers == null) { return (false, 'basis.network_error'.tr()); } // If user exists on server, do nothing if (loadedUsers .any((final User u) => u.login == user.login && u.isFoundOnServer)) { return (false, 'users.user_already_exists'.tr()); } final String? password = user.password; if (password == null) { return (false, 'users.could_not_create_user'.tr()); } // If API returned error, do nothing final GenericResult result = await api.createUser(user.login, password); if (result.data == null) { return (false, result.message ?? 'users.could_not_create_user'.tr()); } _apiData.users.data?.add(result.data!); _apiData.users.invalidate(); return (true, result.message ?? 'basis.done'.tr()); } Future<(bool, String)> deleteUser(final User user) async { final List? loadedUsers = _apiData.users.data; if (loadedUsers == null) { return (false, 'basis.network_error'.tr()); } // If user is primary or root, don't delete if (user.type != UserType.normal) { return (false, 'users.could_not_delete_user'.tr()); } final GenericResult result = await api.deleteUser(user.login); if (result.success && result.data) { _apiData.users.data?.removeWhere((final User u) => u.login == user.login); _apiData.users.invalidate(); } if (!result.success || !result.data) { return (false, result.message ?? 'jobs.generic_error'.tr()); } return (true, result.message ?? 'basis.done'.tr()); } Future<(bool, String)> changeUserPassword( final User user, final String newPassword, ) async { if (user.type == UserType.root) { return (false, 'users.could_not_change_password'.tr()); } final GenericResult result = await api.updateUser( user.login, newPassword, ); if (result.data == null) { getIt().showSnackBar( result.message ?? 'users.could_not_change_password'.tr(), ); return ( false, result.message ?? 'users.could_not_change_password'.tr(), ); } return (true, result.message ?? 'basis.done'.tr()); } Future<(bool, String)> addSshKey( final User user, final String publicKey, ) async { final List? loadedUsers = _apiData.users.data; if (loadedUsers == null) { return (false, 'basis.network_error'.tr()); } final GenericResult result = await api.addSshKey(user.login, publicKey); if (result.data != null) { final User updatedUser = result.data!; final int index = loadedUsers.indexWhere((final User u) => u.login == user.login); loadedUsers[index] = updatedUser; _apiData.users.invalidate(); } else { return (false, result.message ?? 'users.could_not_add_ssh_key'.tr()); } return (true, result.message ?? 'basis.done'.tr()); } Future<(bool, String)> deleteSshKey( final User user, final String publicKey, ) async { final List? loadedUsers = _apiData.users.data; if (loadedUsers == null) { return (false, 'basis.network_error'.tr()); } final GenericResult result = await api.removeSshKey(user.login, publicKey); if (result.data != null) { final User updatedUser = result.data!; final int index = loadedUsers.indexWhere((final User u) => u.login == user.login); loadedUsers[index] = updatedUser; _apiData.users.invalidate(); } else { return (false, result.message ?? 'jobs.generic_error'.tr()); } return (true, result.message ?? 'basis.done'.tr()); } Future<(bool, String)> setAutoUpgradeSettings( final bool enable, final bool allowReboot, ) async { final GenericResult result = await api.setAutoUpgradeSettings( AutoUpgradeSettings( enable: enable, allowReboot: allowReboot, ), ); _apiData.settings.invalidate(); if (result.data != null) { return (true, result.message ?? 'basis.done'.tr()); } else { return (false, result.message ?? 'jobs.generic_error'.tr()); } } Future<(bool, String)> setServerTimezone( final String timezone, ) async { final GenericResult result = await api.setTimezone(timezone); _apiData.settings.invalidate(); if (result.success) { return (true, result.message ?? 'basis.done'.tr()); } else { return (false, result.message ?? 'jobs.generic_error'.tr()); } } Future<(bool, String)> setSshSettings( final bool enable, final bool passwordAuthentication, ) async { final GenericResult result = await api.setSshSettings( SshSettings( enable: enable, passwordAuthentication: passwordAuthentication, ), ); _apiData.settings.invalidate(); if (result.data != null) { return (true, result.message ?? 'basis.done'.tr()); } else { return (false, result.message ?? 'jobs.generic_error'.tr()); } } void dispose() { _dataStream.close(); _connectionStatusStream.close(); _timer?.cancel(); } ServerHostingDetails? get serverDetails => getIt().serverDetails; ServerDomain? get serverDomain => getIt().serverDomain; void init() async { final serverDetails = getIt().serverDetails; final hasFinalChecked = box.get(BNames.hasFinalChecked, defaultValue: false); if (serverDetails == null || !hasFinalChecked) { return; } connectionStatus = ConnectionStatus.reconnecting; _connectionStatusStream.add(connectionStatus); final String? apiVersion = await api.getApiVersion(); if (apiVersion == null) { connectionStatus = ConnectionStatus.offline; _connectionStatusStream.add(connectionStatus); return; } else { _apiData.apiVersion.data = apiVersion; _dataStream.add(_apiData); } await _refetchEverything(Version.parse(apiVersion)); connectionStatus = ConnectionStatus.connected; _connectionStatusStream.add(connectionStatus); // Use timer to periodically check for new jobs _timer = Timer.periodic( const Duration(seconds: 10), reload, ); } Future _refetchEverything(final Version version) async { await _apiData.serverJobs .refetchData(version, () => _dataStream.add(_apiData)); await _apiData.backups .refetchData(version, () => _dataStream.add(_apiData)); await _apiData.backupConfig .refetchData(version, () => _dataStream.add(_apiData)); await _apiData.services .refetchData(version, () => _dataStream.add(_apiData)); await _apiData.volumes .refetchData(version, () => _dataStream.add(_apiData)); await _apiData.recoveryKeyStatus .refetchData(version, () => _dataStream.add(_apiData)); await _apiData.devices .refetchData(version, () => _dataStream.add(_apiData)); await _apiData.users.refetchData(version, () => _dataStream.add(_apiData)); await _apiData.settings .refetchData(version, () => _dataStream.add(_apiData)); } Future reload(final Timer? timer) async { final serverDetails = getIt().serverDetails; if (serverDetails == null) { return; } final String? apiVersion = await api.getApiVersion(); if (apiVersion == null) { connectionStatus = ConnectionStatus.offline; _connectionStatusStream.add(connectionStatus); return; } else { connectionStatus = ConnectionStatus.connected; _connectionStatusStream.add(connectionStatus); _apiData.apiVersion.data = apiVersion; } final Version version = Version.parse(apiVersion); await _refetchEverything(version); } void emitData() { _dataStream.add(_apiData); } } class ApiData { ApiData(final ServerApi api) : apiVersion = ApiDataElement( fetchData: () async => api.getApiVersion(), ), serverJobs = ApiDataElement>( fetchData: () async => api.getServerJobs(), ttl: 10, ), backupConfig = ApiDataElement( fetchData: () async => api.getBackupsConfiguration(), requiredApiVersion: '>=2.4.2', ttl: 120, ), backups = ApiDataElement>( fetchData: () async => api.getBackups(), requiredApiVersion: '>=2.4.2', ttl: 120, ), services = ApiDataElement>( fetchData: () async => api.getAllServices(), requiredApiVersion: '>=2.4.3', ), volumes = ApiDataElement>( fetchData: () async => api.getServerDiskVolumes(), ), recoveryKeyStatus = ApiDataElement( fetchData: () async => (await api.getRecoveryTokenStatus()).data, ttl: 300, ), devices = ApiDataElement>( fetchData: () async => (await api.getApiTokens()).data, ), users = ApiDataElement>( fetchData: () async => api.getAllUsers(), ), settings = ApiDataElement( fetchData: () async => api.getSystemSettings(), ttl: 600, ); ApiDataElement> serverJobs; ApiDataElement apiVersion; ApiDataElement backupConfig; ApiDataElement> backups; ApiDataElement> services; ApiDataElement> volumes; ApiDataElement recoveryKeyStatus; ApiDataElement> devices; ApiDataElement> users; ApiDataElement settings; } enum ConnectionStatus { nonexistent, connected, reconnecting, offline, unauthorized, } class ApiDataElement { ApiDataElement({ required this.fetchData, final T? data, this.requiredApiVersion = '>=2.3.0', this.ttl = 60, }) : _data = data, _lastUpdated = DateTime.now(); T? _data; final String requiredApiVersion; final Future Function() fetchData; Future refetchData( final Version version, final Function callback, ) async { if (VersionConstraint.parse(requiredApiVersion).allows(version)) { if (isExpired || _data == null) { final newData = await fetchData(); if (T is List) { if (Object.hashAll(newData as Iterable) != Object.hashAll(_data as Iterable)) { _data = [...newData] as T?; } } else { if (newData.hashCode != _data.hashCode) { _data = newData; } } callback(); } } } /// TTL of the data in seconds final int ttl; Type get type => T; void invalidate() { _lastUpdated = DateTime.fromMillisecondsSinceEpoch(0); } /// Timestamp of when the data was last updated DateTime _lastUpdated; bool get isExpired { final now = DateTime.now(); final difference = now.difference(_lastUpdated); return difference.inSeconds > ttl; } T? get data => _data; /// Sets the data and updates the lastUpdated timestamp set data(final T? data) { _data = data; _lastUpdated = DateTime.now(); } /// Returns the last time the data was updated DateTime get lastUpdated => _lastUpdated; }