import 'dart:async'; import 'dart:io'; import 'package:device_info_plus/device_info_plus.dart'; import 'package:easy_localization/easy_localization.dart'; import 'package:flutter/foundation.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/models/json/dns_records.dart'; import 'package:selfprivacy/logic/providers/provider_settings.dart'; import 'package:selfprivacy/logic/api_maps/graphql_maps/server_api/server_api.dart'; import 'package:selfprivacy/logic/api_maps/tls_options.dart'; import 'package:selfprivacy/logic/cubit/server_installation/server_installation_cubit.dart'; import 'package:selfprivacy/logic/models/hive/backups_credential.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/device_token.dart'; import 'package:selfprivacy/logic/models/server_basic_info.dart'; import 'package:selfprivacy/logic/models/server_type.dart'; import 'package:selfprivacy/logic/providers/providers_controller.dart'; import 'package:selfprivacy/utils/network_utils.dart'; class IpNotFoundException implements Exception { IpNotFoundException(this.message); final String message; } class ServerAuthorizationException implements Exception { ServerAuthorizationException(this.message); final String message; } class ServerInstallationRepository { Box box = Hive.box(BNames.serverInstallationBox); Box usersBox = Hive.box(BNames.usersBox); Future load() async { final String? providerApiToken = getIt().serverProviderKey; final String? location = getIt().serverLocation; final String? dnsApiToken = getIt().dnsProviderKey; final String? serverTypeIdentificator = getIt().serverType; final ServerDomain? serverDomain = getIt().serverDomain; final DnsProviderType? dnsProvider = getIt().dnsProvider; final ServerProviderType? serverProvider = getIt().serverProvider; final BackupsCredential? backblazeCredential = getIt().backblazeCredential; final ServerHostingDetails? serverDetails = getIt().serverDetails; if (serverProvider != null || (serverDetails != null && serverDetails.provider != ServerProviderType.unknown)) { ProvidersController.initServerProvider( ServerProviderSettings( provider: serverProvider ?? serverDetails!.provider, isAuthorized: providerApiToken != null, location: location, ), ); } if (dnsProvider != null || (serverDomain != null && serverDomain.provider != DnsProviderType.unknown)) { ProvidersController.initDnsProvider( DnsProviderSettings( isAuthorized: dnsApiToken != null, provider: dnsProvider ?? serverDomain!.provider, ), ); } if (box.get(BNames.hasFinalChecked, defaultValue: false)) { TlsOptions.verifyCertificate = true; if (serverTypeIdentificator == null && serverDetails != null) { final finalServerType = await ProvidersController.currentServerProvider! .getServerType(serverDetails.id); await saveServerType(finalServerType.data!); await ProvidersController.currentServerProvider! .trySetServerLocation(finalServerType.data!.location.identifier); return ServerInstallationFinished( installationDialoguePopUp: null, providerApiToken: providerApiToken!, serverTypeIdentificator: finalServerType.data!.identifier, dnsApiToken: dnsApiToken!, serverDomain: serverDomain!, backblazeCredential: backblazeCredential!, serverDetails: serverDetails, rootUser: box.get(BNames.rootUser), isServerStarted: box.get(BNames.isServerStarted, defaultValue: false), isServerResetedFirstTime: box.get(BNames.isServerResetedFirstTime, defaultValue: false), isServerResetedSecondTime: box.get(BNames.isServerResetedSecondTime, defaultValue: false), ); } else { return ServerInstallationFinished( installationDialoguePopUp: null, providerApiToken: providerApiToken!, serverTypeIdentificator: serverTypeIdentificator!, dnsApiToken: dnsApiToken!, serverDomain: serverDomain!, backblazeCredential: backblazeCredential!, serverDetails: serverDetails!, rootUser: box.get(BNames.rootUser), isServerStarted: box.get(BNames.isServerStarted, defaultValue: false), isServerResetedFirstTime: box.get(BNames.isServerResetedFirstTime, defaultValue: false), isServerResetedSecondTime: box.get(BNames.isServerResetedSecondTime, defaultValue: false), ); } } if (box.get(BNames.isRecoveringServer, defaultValue: false) && serverDomain != null) { return ServerInstallationRecovery( providerApiToken: providerApiToken, dnsApiToken: dnsApiToken, serverDomain: serverDomain, serverTypeIdentificator: serverTypeIdentificator, backblazeCredential: backblazeCredential, serverDetails: serverDetails, rootUser: box.get(BNames.rootUser), currentStep: _getCurrentRecoveryStep( providerApiToken, dnsApiToken, serverDomain, serverDetails, ), recoveryCapabilities: await getRecoveryCapabilities(serverDomain), ); } return ServerInstallationNotFinished( providerApiToken: providerApiToken, dnsApiToken: dnsApiToken, serverDomain: serverDomain, serverTypeIdentificator: serverTypeIdentificator, backblazeCredential: backblazeCredential, serverDetails: serverDetails, rootUser: box.get(BNames.rootUser), isServerStarted: box.get(BNames.isServerStarted, defaultValue: false), isServerResetedFirstTime: box.get(BNames.isServerResetedFirstTime, defaultValue: false), isServerResetedSecondTime: box.get(BNames.isServerResetedSecondTime, defaultValue: false), isLoading: box.get(BNames.isLoading, defaultValue: false), dnsMatches: null, ); } RecoveryStep _getCurrentRecoveryStep( final String? serverProviderToken, final String? cloudflareToken, final ServerDomain serverDomain, final ServerHostingDetails? serverDetails, ) { if (serverDetails != null) { if (serverProviderToken != null) { if (serverDetails.provider != ServerProviderType.unknown) { if (serverDomain.provider != DnsProviderType.unknown) { return RecoveryStep.backblazeToken; } return RecoveryStep.dnsProviderToken; } return RecoveryStep.serverSelection; } return RecoveryStep.serverProviderToken; } return RecoveryStep.selecting; } void clearAppConfig() { box.clear(); usersBox.clear(); } Future startServer( final ServerHostingDetails server, ) async { final result = await ProvidersController.currentServerProvider!.powerOn( server.id, ); if (result.success && result.data != null) { server.copyWith(startTime: result.data); } return server; } Future getDomainId(final String token, final String domain) async { final result = await ProvidersController.currentDnsProvider!.tryInitApiByToken(token); if (!result.success) { return null; } await setDnsApiToken(token); return (await ProvidersController.currentDnsProvider!.getZoneId( domain, )) .data; } Future> isDnsAddressesMatch( final String? domainName, final String? ip4, final Map skippedMatches, ) async { final Map matches = {}; await InternetAddress.lookup(domainName!).then( (final records) { for (final record in records) { if (skippedMatches[record.host] ?? false) { matches[record.host] = true; continue; } if (record.address == ip4!) { matches[record.host] = true; } } }, ); return matches; } Future createDkimRecord(final ServerDomain cloudFlareDomain) async { final ServerApi api = ServerApi(); late DnsRecord record; try { record = extractDkimRecord(await api.getDnsRecords())!; } catch (e) { print(e); rethrow; } await ProvidersController.currentDnsProvider!.setDnsRecord( record, cloudFlareDomain, ); } Future isHttpServerWorking() async { final ServerApi api = ServerApi(); return api.isHttpServerWorking(); } Future restart() async { final server = getIt().serverDetails!; final result = await ProvidersController.currentServerProvider!.restart( server.id, ); if (result.success && result.data != null) { server.copyWith(startTime: result.data); } return server; } Future powerOn() async { final server = getIt().serverDetails!; return startServer(server); } Future getRecoveryCapabilities( final ServerDomain serverDomain, ) async { final ServerApi serverApi = ServerApi( isWithToken: false, overrideDomain: serverDomain.domainName, ); final String? serverApiVersion = await serverApi.getApiVersion(); if (serverApiVersion == null) { return ServerRecoveryCapabilities.none; } try { final Version parsedVersion = Version.parse(serverApiVersion); if (!VersionConstraint.parse('>=1.2.0').allows(parsedVersion)) { return ServerRecoveryCapabilities.legacy; } return ServerRecoveryCapabilities.loginTokens; } on FormatException { return ServerRecoveryCapabilities.none; } } Future getServerIpFromDomain(final ServerDomain serverDomain) async { String? domain; await InternetAddress.lookup(serverDomain.domainName).then( (final records) { for (final record in records) { domain = record.address; } }, ); if (domain == null || domain!.isEmpty) { throw IpNotFoundException('No IP found for domain $serverDomain'); } return domain!; } Future getDeviceName() async { final DeviceInfoPlugin deviceInfo = DeviceInfoPlugin(); if (kIsWeb) { return deviceInfo.webBrowserInfo.then( (final WebBrowserInfo value) => '${value.browserName} ${value.platform}', ); } else { if (Platform.isAndroid) { return deviceInfo.androidInfo.then( (final AndroidDeviceInfo value) => '${value.model} ${value.version.release}', ); } else if (Platform.isIOS) { return deviceInfo.iosInfo.then( (final IosDeviceInfo value) => '${value.utsname.machine} ${value.systemName} ${value.systemVersion}', ); } else if (Platform.isLinux) { return deviceInfo.linuxInfo .then((final LinuxDeviceInfo value) => value.prettyName); } else if (Platform.isMacOS) { return deviceInfo.macOsInfo.then( (final MacOsDeviceInfo value) => '${value.hostName} ${value.computerName}', ); } else if (Platform.isWindows) { return deviceInfo.windowsInfo .then((final WindowsDeviceInfo value) => value.computerName); } } return 'Unidentified'; } Future authorizeByNewDeviceKey( final ServerDomain serverDomain, final String newDeviceKey, final ServerRecoveryCapabilities recoveryCapabilities, ) async { final ServerApi serverApi = ServerApi( isWithToken: false, overrideDomain: serverDomain.domainName, ); final String serverIp = await getServerIpFromDomain(serverDomain); final GenericResult result = await serverApi.authorizeDevice( DeviceToken(device: await getDeviceName(), token: newDeviceKey), ); if (result.success) { return ServerHostingDetails( apiToken: result.data, volume: ServerVolume( id: 0, name: '', sizeByte: 0, serverId: 0, linuxDevice: '', ), provider: ServerProviderType.unknown, id: 0, ip4: serverIp, startTime: null, createTime: null, ); } throw ServerAuthorizationException( result.message ?? result.data, ); } Future authorizeByRecoveryKey( final ServerDomain serverDomain, final String recoveryKey, final ServerRecoveryCapabilities recoveryCapabilities, ) async { final ServerApi serverApi = ServerApi( isWithToken: false, overrideDomain: serverDomain.domainName, ); final String serverIp = await getServerIpFromDomain(serverDomain); final GenericResult result = await serverApi.useRecoveryToken( DeviceToken(device: await getDeviceName(), token: recoveryKey), ); if (result.success) { return ServerHostingDetails( apiToken: result.data, volume: ServerVolume( id: 0, name: '', sizeByte: 0, serverId: 0, linuxDevice: '', ), provider: ServerProviderType.unknown, id: 0, ip4: serverIp, startTime: null, createTime: null, ); } throw ServerAuthorizationException( result.message ?? result.data, ); } Future authorizeByApiToken( final ServerDomain serverDomain, final String apiToken, final ServerRecoveryCapabilities recoveryCapabilities, ) async { final ServerApi serverApi = ServerApi( isWithToken: true, overrideDomain: serverDomain.domainName, customToken: apiToken, ); final String serverIp = await getServerIpFromDomain(serverDomain); if (recoveryCapabilities == ServerRecoveryCapabilities.legacy) { if (await serverApi.isHttpServerWorking()) { return ServerHostingDetails( apiToken: apiToken, volume: ServerVolume( id: 0, name: '', serverId: 0, sizeByte: 0, linuxDevice: '', ), provider: ServerProviderType.unknown, id: 0, ip4: serverIp, startTime: null, createTime: null, ); } else { throw ServerAuthorizationException( "Couldn't connect to server with this token", ); } } final GenericResult deviceAuthKey = await serverApi.createDeviceToken(); final GenericResult result = await serverApi.authorizeDevice( DeviceToken(device: await getDeviceName(), token: deviceAuthKey.data), ); if (result.success) { return ServerHostingDetails( apiToken: result.data, volume: ServerVolume( id: 0, name: '', sizeByte: 0, serverId: 0, linuxDevice: '', ), provider: ServerProviderType.unknown, id: 0, ip4: serverIp, startTime: null, createTime: null, ); } throw ServerAuthorizationException( result.message ?? result.data, ); } Future getMainUser() async { final ServerApi serverApi = ServerApi(); const User fallbackUser = User( isFoundOnServer: false, type: UserType.primary, note: "Couldn't find main user on server, API is outdated", login: 'UNKNOWN', sshKeys: [], ); final String? serverApiVersion = await serverApi.getApiVersion(); final users = await serverApi.getAllUsers(); if (serverApiVersion == null || users.isEmpty) { return fallbackUser; } try { final Version parsedVersion = Version.parse(serverApiVersion); if (!VersionConstraint.parse('>=1.2.5').allows(parsedVersion)) { return fallbackUser; } return users.firstWhere( (final User user) => user.type == UserType.primary, ); } on FormatException { return fallbackUser; } } Future> getServersOnProviderAccount() async => (await ProvidersController.currentServerProvider!.getServers()).data; Future saveServerDetails( final ServerHostingDetails serverDetails, ) async { await getIt().storeServerDetails(serverDetails); } Future deleteServerDetails() async { await box.delete(BNames.serverDetails); getIt().init(); } Future saveServerProviderType(final ServerProviderType type) async { await getIt().storeServerProviderType(type); } Future saveDnsProviderType(final DnsProviderType type) async { await getIt().storeDnsProviderType(type); } Future saveServerProviderKey(final String key) async { await getIt().storeServerProviderKey(key); } Future saveServerType(final ServerType serverType) async { await getIt().storeServerTypeIdentifier( serverType.identifier, ); await getIt().storeServerLocation( serverType.location.identifier, ); } Future deleteServerProviderKey() async { await box.delete(BNames.hetznerKey); getIt().init(); } Future saveBackblazeKey( final BackupsCredential backblazeCredential, ) async { await getIt().storeBackblazeCredential(backblazeCredential); } Future deleteBackblazeKey() async { await box.delete(BNames.backblazeCredential); getIt().init(); } Future setDnsApiToken(final String key) async { await getIt().storeDnsProviderKey(key); } Future deleteDnsProviderKey() async { await box.delete(BNames.cloudFlareKey); getIt().init(); } Future saveDomain(final ServerDomain serverDomain) async { await getIt().storeServerDomain(serverDomain); } Future deleteDomain() async { await box.delete(BNames.serverDomain); getIt().init(); } Future saveIsServerStarted(final bool value) async { await box.put(BNames.isServerStarted, value); } Future saveIsServerResetedFirstTime(final bool value) async { await box.put(BNames.isServerResetedFirstTime, value); } Future saveIsServerResetedSecondTime(final bool value) async { await box.put(BNames.isServerResetedSecondTime, value); } Future saveRootUser(final User rootUser) async { await box.put(BNames.rootUser, rootUser); } Future saveIsRecoveringServer(final bool value) async { await box.put(BNames.isRecoveringServer, value); } Future saveHasFinalChecked(final bool value) async { await box.put(BNames.hasFinalChecked, value); } Future deleteServer(final ServerDomain serverDomain) async { final deletionResult = await ProvidersController.currentServerProvider!.deleteServer( serverDomain.domainName, ); if (!deletionResult.success) { getIt() .showSnackBar('modals.server_validators_error'.tr()); return false; } await box.put(BNames.hasFinalChecked, false); await box.put(BNames.isServerStarted, false); await box.put(BNames.isServerResetedFirstTime, false); await box.put(BNames.isServerResetedSecondTime, false); await box.put(BNames.isLoading, false); await box.put(BNames.serverDetails, null); final GenericResult removalResult = await ProvidersController .currentDnsProvider! .removeDomainRecords(domain: serverDomain); if (!removalResult.success) { getIt().showSnackBar('modals.dns_removal_error'.tr()); } return true; } Future deleteServerRelatedRecords() async { await box.deleteAll([ BNames.serverDetails, BNames.isServerStarted, BNames.isServerResetedFirstTime, BNames.isServerResetedSecondTime, BNames.hasFinalChecked, BNames.isLoading, ]); getIt().init(); } }