import 'dart:convert'; import 'dart:io'; import 'package:flutter/material.dart'; import 'package:flutter_hbb/models/model.dart'; import 'package:flutter_hbb/models/peer_model.dart'; import 'package:flutter_hbb/models/platform_model.dart'; import 'package:get/get.dart'; import 'package:bot_toast/bot_toast.dart'; import 'package:http/http.dart' as http; import '../common.dart'; final syncAbOption = 'sync-ab-with-recent-sessions'; bool shouldSyncAb() { return bind.mainGetLocalOption(key: syncAbOption).isNotEmpty; } final sortAbTagsOption = 'sync-ab-tags'; bool shouldSortTags() { return bind.mainGetLocalOption(key: sortAbTagsOption).isNotEmpty; } class AbModel { final abLoading = false.obs; final abError = "".obs; final tags = [].obs; final peers = List.empty(growable: true).obs; final sortTags = shouldSortTags().obs; final selectedTags = List.empty(growable: true).obs; var initialized = false; var licensedDevices = 0; WeakReference parent; AbModel(this.parent); Future pullAb({force = true, quiet = false}) async { if (gFFI.userModel.userName.isEmpty) return; if (abLoading.value) return; if (!force && initialized) return; if (!quiet) { abLoading.value = true; abError.value = ""; } final api = "${await bind.mainGetApiServer()}/api/ab/get"; try { var authHeaders = getHttpHeaders(); authHeaders['Content-Type'] = "application/json"; authHeaders['Accept-Encoding'] = "gzip"; final resp = await http.get(Uri.parse(api), headers: authHeaders); if (resp.body.isNotEmpty && resp.body.toLowerCase() != "null") { Map json = jsonDecode(resp.body); if (json.containsKey('error')) { abError.value = json['error']; } else if (json.containsKey('data')) { try { gFFI.abModel.licensedDevices = json['licensed_devices']; // ignore: empty_catches } catch (e) {} final data = jsonDecode(json['data']); if (data != null) { tags.clear(); peers.clear(); if (data['tags'] is List) { tags.value = data['tags']; } if (data['peers'] is List) { for (final peer in data['peers']) { peers.add(Peer.fromJson(peer)); } } } } } } catch (err) { abError.value = err.toString(); } finally { abLoading.value = false; initialized = true; } } Future reset() async { await bind.mainSetLocalOption(key: "selected-tags", value: ''); tags.clear(); peers.clear(); initialized = false; } void addId(String id, String alias, List tags) { if (idContainBy(id)) { return; } final peer = Peer.fromJson({ 'id': id, 'alias': alias, 'tags': tags, }); peers.add(peer); } bool isFull(bool warn) { final res = licensedDevices > 0 && peers.length >= licensedDevices; if (res && warn) { BotToast.showText( contentColor: Colors.red, text: translate("exceed_max_devices")); } return res; } void addPeer(Peer peer) { peers.removeWhere((e) => e.id == peer.id); peers.add(peer); } void addTag(String tag) async { if (tagContainBy(tag)) { return; } tags.add(tag); } void changeTagForPeer(String id, List tags) { final it = peers.where((element) => element.id == id); if (it.isEmpty) { return; } it.first.tags = tags; } Future pushAb() async { final api = "${await bind.mainGetApiServer()}/api/ab"; var authHeaders = getHttpHeaders(); authHeaders['Content-Type'] = "application/json"; final peersJsonData = peers.map((e) => e.toJson()).toList(); final body = jsonEncode({ "data": jsonEncode({"tags": tags, "peers": peersJsonData}) }); var request = http.Request('POST', Uri.parse(api)); // support compression if (licensedDevices > 0 && body.length > 1024) { authHeaders['Content-Encoding'] = "gzip"; request.bodyBytes = GZipCodec().encode(utf8.encode(body)); } else { request.body = body; } request.headers.addAll(authHeaders); try { await http.Client().send(request); await pullAb(quiet: true); } catch (e) { BotToast.showText(contentColor: Colors.red, text: e.toString()); } finally {} } Peer? find(String id) { return peers.firstWhereOrNull((e) => e.id == id); } bool idContainBy(String id) { return peers.where((element) => element.id == id).isNotEmpty; } bool tagContainBy(String tag) { return tags.where((element) => element == tag).isNotEmpty; } void deletePeer(String id) { peers.removeWhere((element) => element.id == id); } void deleteTag(String tag) { gFFI.abModel.selectedTags.remove(tag); tags.removeWhere((element) => element == tag); for (var peer in peers) { if (peer.tags.isEmpty) { continue; } if (peer.tags.contains(tag)) { ((peer.tags)).remove(tag); } } } void unsetSelectedTags() { selectedTags.clear(); } List getPeerTags(String id) { final it = peers.where((p0) => p0.id == id); if (it.isEmpty) { return []; } else { return it.first.tags; } } Future setPeerAlias(String id, String value) async { final it = peers.where((p0) => p0.id == id); if (it.isNotEmpty) { it.first.alias = value; await pushAb(); } } Future setPeerForceAlwaysRelay(String id, bool value) async { final it = peers.where((p0) => p0.id == id); if (it.isNotEmpty) { it.first.forceAlwaysRelay = value; await pushAb(); } } Future setRdp(String id, String port, String username) async { final it = peers.where((p0) => p0.id == id); if (it.isNotEmpty) { it.first.rdpPort = port; it.first.rdpUsername = username; await pushAb(); } } }