import 'dart:convert'; import 'dart:io'; import 'package:dio/dio.dart'; import 'package:easy_localization/easy_localization.dart'; import 'package:selfprivacy/config/get_it_config.dart'; import 'package:selfprivacy/logic/api_maps/rest_maps/server_providers/volume_provider.dart'; import 'package:selfprivacy/logic/api_maps/rest_maps/server_providers/server_provider.dart'; import 'package:selfprivacy/logic/models/disk_size.dart'; import 'package:selfprivacy/logic/models/hive/server_domain.dart'; import 'package:selfprivacy/logic/models/hive/server_details.dart'; import 'package:selfprivacy/logic/models/hive/user.dart'; import 'package:selfprivacy/logic/models/metrics.dart'; import 'package:selfprivacy/logic/models/price.dart'; import 'package:selfprivacy/logic/models/server_basic_info.dart'; import 'package:selfprivacy/logic/models/server_metadata.dart'; import 'package:selfprivacy/logic/models/server_provider_location.dart'; import 'package:selfprivacy/logic/models/server_type.dart'; import 'package:selfprivacy/utils/extensions/string_extensions.dart'; import 'package:selfprivacy/utils/password_generator.dart'; class DigitalOceanApi extends ServerProviderApi with VolumeProviderApi { DigitalOceanApi({ required this.region, this.hasLogger = false, this.isWithToken = true, }); @override bool hasLogger; @override bool isWithToken; final String? region; @override BaseOptions get options { final BaseOptions options = BaseOptions(baseUrl: rootAddress); if (isWithToken) { final String? token = getIt().serverProviderKey; assert(token != null); options.headers = {'Authorization': 'Bearer $token'}; } if (validateStatus != null) { options.validateStatus = validateStatus!; } return options; } @override String get rootAddress => 'https://api.digitalocean.com/v2'; @override String get infectProviderName => 'digitalocean'; @override String get appearanceProviderName => 'Digital Ocean'; @override Future isApiTokenValid(final String token) async { bool isValid = false; Response? response; final Dio client = await getClient(); try { response = await client.get( '/account', options: Options( headers: {'Authorization': 'Bearer $token'}, ), ); } catch (e) { print(e); isValid = false; } finally { close(client); } if (response != null) { if (response.statusCode == HttpStatus.ok) { isValid = true; } else if (response.statusCode == HttpStatus.unauthorized) { isValid = false; } else { throw Exception('code: ${response.statusCode}'); } } return isValid; } /// Hardcoded on their documentation and there is no pricing API at all /// Probably we should scrap the doc page manually @override Future getPricePerGb() async => Price( value: 0.10, currency: 'USD', ); @override Future createVolume() async { ServerVolume? volume; final Response dbCreateResponse; final Dio client = await getClient(); try { final List volumes = await getVolumes(); await Future.delayed(const Duration(seconds: 6)); dbCreateResponse = await client.post( '/volumes', data: { 'size_gigabytes': 10, 'name': 'volume${StringGenerators.dbStorageName()}', 'labels': {'labelkey': 'value'}, 'region': region, 'filesystem_type': 'ext4', }, ); final dbId = dbCreateResponse.data['volume']['id']; final dbSize = dbCreateResponse.data['volume']['size_gigabytes']; final dbName = dbCreateResponse.data['volume']['name']; volume = ServerVolume( id: volumes.length, name: dbName, sizeByte: dbSize, serverId: null, linuxDevice: 'scsi-0DO_Volume_$dbName', uuid: dbId, ); } catch (e) { print(e); } finally { client.close(); } return volume; } @override Future> getVolumes({final String? status}) async { final List volumes = []; final Response dbGetResponse; final Dio client = await getClient(); try { dbGetResponse = await client.get( '/volumes', queryParameters: { 'status': status, }, ); final List rawVolumes = dbGetResponse.data['volumes']; int id = 0; for (final rawVolume in rawVolumes) { final dbId = rawVolume['id']; final int dbSize = rawVolume['size_gigabytes'] * 1024 * 1024 * 1024; final dbDropletIds = rawVolume['droplet_ids']; final String dbName = rawVolume['name']; final volume = ServerVolume( id: id++, name: dbName, sizeByte: dbSize, serverId: dbDropletIds.isNotEmpty ? dbDropletIds[0] : null, linuxDevice: 'scsi-0DO_Volume_$dbName', uuid: dbId, ); volumes.add(volume); } } catch (e) { print(e); } finally { client.close(); } return volumes; } Future getVolume(final String volumeUuid) async { ServerVolume? neededVolume; final List volumes = await getVolumes(); for (final volume in volumes) { if (volume.uuid == volumeUuid) { neededVolume = volume; } } return neededVolume; } @override Future deleteVolume(final ServerVolume volume) async { final Dio client = await getClient(); try { await client.delete('/volumes/$volume.uuid'); } catch (e) { print(e); } finally { client.close(); } } @override Future attachVolume( final ServerVolume volume, final int serverId, ) async { bool success = false; final Response dbPostResponse; final Dio client = await getClient(); try { dbPostResponse = await client.post( '/volumes/actions', data: { 'type': 'attach', 'volume_name': volume.name, 'region': region, 'droplet_id': serverId, }, ); success = dbPostResponse.data['action']['status'].toString() != 'error'; } catch (e) { print(e); } finally { close(client); } return success; } @override Future detachVolume(final ServerVolume volume) async { bool success = false; final Response dbPostResponse; final Dio client = await getClient(); try { dbPostResponse = await client.post( '/volumes/actions', data: { 'type': 'detach', 'volume_name': volume.name, 'droplet_id': volume.serverId, 'region': region, }, ); success = dbPostResponse.data['action']['status'].toString() != 'error'; } catch (e) { print(e); } finally { client.close(); } return success; } @override Future resizeVolume( final ServerVolume volume, final DiskSize size, ) async { bool success = false; final Response dbPostResponse; final Dio client = await getClient(); try { dbPostResponse = await client.post( '/volumes/actions', data: { 'type': 'resize', 'volume_name': volume.name, 'size_gigabytes': size.gibibyte, 'region': region, }, ); success = dbPostResponse.data['action']['status'].toString() != 'error'; } catch (e) { print(e); } finally { client.close(); } return success; } static String getHostnameFromDomain(final String domain) { // Replace all non-alphanumeric characters with an underscore String hostname = domain.split('.')[0].replaceAll(RegExp(r'[^a-zA-Z0-9]'), '-'); if (hostname.endsWith('-')) { hostname = hostname.substring(0, hostname.length - 1); } if (hostname.startsWith('-')) { hostname = hostname.substring(1); } if (hostname.isEmpty) { hostname = 'selfprivacy-server'; } return hostname; } @override Future createServer({ required final String dnsApiToken, required final User rootUser, required final String domainName, required final String serverType, }) async { ServerHostingDetails? serverDetails; final String dbPassword = StringGenerators.dbPassword(); final String apiToken = StringGenerators.apiToken(); final String base64Password = base64.encode(utf8.encode(rootUser.password ?? 'PASS')); final String formattedHostname = getHostnameFromDomain(domainName); const String infectBranch = 'providers/digital-ocean'; final String userdataString = "#cloud-config\nruncmd:\n- curl https://git.selfprivacy.org/SelfPrivacy/selfprivacy-nixos-infect/raw/branch/$infectBranch/nixos-infect | PROVIDER=$infectProviderName DOMAIN='$domainName' LUSER='${rootUser.login}' ENCODED_PASSWORD='$base64Password' CF_TOKEN=$dnsApiToken DB_PASSWORD=$dbPassword API_TOKEN=$apiToken HOSTNAME=$formattedHostname bash 2>&1 | tee /tmp/infect.log"; print(userdataString); final Dio client = await getClient(); try { final Map data = { 'name': formattedHostname, 'size': serverType, 'image': 'ubuntu-20-04-x64', 'user_data': userdataString, 'region': region!, }; print('Decoded data: $data'); final Response serverCreateResponse = await client.post( '/droplets', data: data, ); final int serverId = serverCreateResponse.data['droplet']['id']; final ServerVolume? newVolume = await createVolume(); final bool attachedVolume = await attachVolume(newVolume!, serverId); String? ipv4; int attempts = 0; while (attempts < 5 && ipv4 == null) { await Future.delayed(const Duration(seconds: 20)); final List servers = await getServers(); for (final server in servers) { if (server.name == formattedHostname && server.ip != '0.0.0.0') { ipv4 = server.ip; break; } } ++attempts; } if (attachedVolume && ipv4 != null) { serverDetails = ServerHostingDetails( id: serverId, ip4: ipv4, createTime: DateTime.now(), volume: newVolume, apiToken: apiToken, provider: ServerProvider.digitalOcean, ); } } catch (e) { print(e); } finally { close(client); } return serverDetails; } @override Future deleteServer({ required final String domainName, }) async { final Dio client = await getClient(); final ServerBasicInfo serverToRemove = (await getServers()).firstWhere( (final el) => el.name == domainName, ); final ServerVolume volumeToRemove = (await getVolumes()).firstWhere( (final el) => el.serverId == serverToRemove.id, ); final List laterFutures = []; await detachVolume(volumeToRemove); await Future.delayed(const Duration(seconds: 10)); try { laterFutures.add(deleteVolume(volumeToRemove)); laterFutures.add(client.delete('/droplets/$serverToRemove.id')); await Future.wait(laterFutures); } catch (e) { print(e); } finally { close(client); } } @override Future restart() async { final ServerHostingDetails server = getIt().serverDetails!; final Dio client = await getClient(); try { await client.post( '/droplets/${server.id}/actions', data: { 'type': 'reboot', }, ); } catch (e) { print(e); } finally { close(client); } return server.copyWith(startTime: DateTime.now()); } @override Future powerOn() async { final ServerHostingDetails server = getIt().serverDetails!; final Dio client = await getClient(); try { await client.post( '/droplets/${server.id}/actions', data: { 'type': 'power_on', }, ); } catch (e) { print(e); } finally { close(client); } return server.copyWith(startTime: DateTime.now()); } @override Future getMetrics( final int serverId, final DateTime start, final DateTime end, ) async { ServerMetrics? metrics; return metrics; } @override Future> getMetadata(final int serverId) async { List metadata = []; final Dio client = await getClient(); try { final Response response = await client.get('/droplets/$serverId'); final droplet = response.data!['droplet']; metadata = [ ServerMetadataEntity( type: MetadataType.id, name: 'server.server_id'.tr(), value: droplet['id'].toString(), ), ServerMetadataEntity( type: MetadataType.status, name: 'server.status'.tr(), value: droplet['status'].toString().capitalize(), ), ServerMetadataEntity( type: MetadataType.cpu, name: 'server.cpu'.tr(), value: 'server.core_count'.plural(droplet['vcpus']), ), ServerMetadataEntity( type: MetadataType.ram, name: 'server.ram'.tr(), value: "${droplet['memory'].toString()} MB", ), ServerMetadataEntity( type: MetadataType.cost, name: 'server.monthly_cost'.tr(), value: droplet['size']['price_monthly'].toString(), ), ServerMetadataEntity( type: MetadataType.location, name: 'server.location'.tr(), value: '${droplet['region']['name']} ${getEmojiFlag(droplet['region']['slug'].toString()) ?? ''}', ), ServerMetadataEntity( type: MetadataType.other, name: 'server.provider'.tr(), value: appearanceProviderName, ), ]; } catch (e) { print(e); } finally { close(client); } return metadata; } @override Future> getServers() async { List servers = []; final Dio client = await getClient(); try { final Response response = await client.get('/droplets'); servers = response.data!['droplets'].map( (final server) { String ipv4 = '0.0.0.0'; if (server['networks']['v4'].isNotEmpty) { for (final v4 in server['networks']['v4']) { if (v4['type'].toString() == 'public') { ipv4 = v4['ip_address'].toString(); } } } return ServerBasicInfo( id: server['id'], reverseDns: server['name'], created: DateTime.now(), ip: ipv4, name: server['name'], ); }, ).toList(); } catch (e) { print(e); } finally { close(client); } print(servers); return servers; } String? getEmojiFlag(final String query) { String? emoji; switch (query.toLowerCase().substring(0, 3)) { case 'fra': emoji = '🇩🇪'; break; case 'ams': emoji = '🇳🇱'; break; case 'sgp': emoji = '🇸🇬'; break; case 'lon': emoji = '🇬🇧'; break; case 'tor': emoji = '🇨🇦'; break; case 'blr': emoji = '🇮🇳'; break; case 'nyc': case 'sfo': emoji = '🇺🇸'; break; } return emoji; } @override Future> getAvailableLocations() async { List locations = []; final Dio client = await getClient(); try { final Response response = await client.get( '/regions', ); locations = response.data!['regions'] .map( (final location) => ServerProviderLocation( title: location['slug'], description: location['name'], flag: getEmojiFlag(location['slug']), identifier: location['slug'], ), ) .toList(); } catch (e) { print(e); } finally { close(client); } return locations; } @override Future> getServerTypesByLocation({ required final ServerProviderLocation location, }) async { final List types = []; final Dio client = await getClient(); try { final Response response = await client.get( '/sizes', ); final rawSizes = response.data!['sizes']; for (final rawSize in rawSizes) { for (final rawRegion in rawSize['regions']) { if (rawRegion.toString() == location.identifier) { types.add( ServerType( title: rawSize['description'], identifier: rawSize['slug'], ram: rawSize['memory'].toDouble(), cores: rawSize['vcpus'], disk: DiskSize(byte: rawSize['disk'] * 1024 * 1024 * 1024), price: Price( value: rawSize['price_monthly'], currency: 'USD', ), location: location, ), ); } } } } catch (e) { print(e); } finally { close(client); } return types; } @override Future createReverseDns({ required final ServerHostingDetails serverDetails, required final ServerDomain domain, }) async { /// TODO remove from provider interface } @override ProviderApiTokenValidation getApiTokenValidation() => ProviderApiTokenValidation( regexp: RegExp(r'\s+|[-!$%^&*()@+|~=`{}\[\]:<>?,.\/]'), length: 71, ); }