import 'dart:async'; 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/peer_tab_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 pullError = "".obs; final pushError = "".obs; final tags = [].obs; final peers = List.empty(growable: true).obs; final sortTags = shouldSortTags().obs; final retrying = false.obs; bool get emtpy => peers.isEmpty && tags.isEmpty; final selectedTags = List.empty(growable: true).obs; var initialized = false; var licensedDevices = 0; var _syncAllFromRecent = true; var _syncFromRecentLock = false; var _timerCounter = 0; var _cacheLoadOnceFlag = false; WeakReference parent; AbModel(this.parent) { if (desktopType == DesktopType.main) { Timer.periodic(Duration(milliseconds: 500), (timer) async { if (_timerCounter++ % 6 == 0) syncFromRecent(); }); } } Future pullAb({force = true, quiet = false}) async { debugPrint("pullAb, force:$force, quiet:$quiet"); if (gFFI.userModel.userName.isEmpty) return; if (abLoading.value) return; if (!force && initialized) return; DateTime startTime = DateTime.now(); if (pushError.isNotEmpty) { try { // push to retry pushAb(toastIfFail: false, toastIfSucc: false); } catch (_) {} } if (!quiet) { abLoading.value = true; pullError.value = ""; } final api = "${await bind.mainGetApiServer()}/api/ab"; int? statusCode; try { var authHeaders = getHttpHeaders(); authHeaders['Content-Type'] = "application/json"; authHeaders['Accept-Encoding'] = "gzip"; final resp = await http.get(Uri.parse(api), headers: authHeaders); statusCode = resp.statusCode; if (resp.body.toLowerCase() == "null") { // normal reply, emtpy ab return null tags.clear(); peers.clear(); } else if (resp.body.isNotEmpty) { Map json = _jsonDecode(utf8.decode(resp.bodyBytes), resp.statusCode); if (json.containsKey('error')) { throw 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)); } } _saveCache(); // save on success } } } } catch (err) { if (!quiet) { pullError.value = '${translate('pull_ab_failed_tip')}: ${translate(err.toString())}'; if (gFFI.peerTabModel.currentTab != PeerTabIndex.ab.index) { BotToast.showText(contentColor: Colors.red, text: pullError.value); } } } finally { var ms = (Duration(milliseconds: 300) - DateTime.now().difference(startTime)) .inMilliseconds; ms = ms > 0 ? ms : 0; Future.delayed(Duration(milliseconds: ms), () { abLoading.value = false; }); initialized = true; _syncAllFromRecent = true; _timerCounter = 0; if (pullError.isNotEmpty) { if (statusCode == 401) { gFFI.userModel.reset(clearAbCache: true); } } } } 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) { final index = peers.indexWhere((e) => e.id == peer.id); if (index >= 0) { peers[index] = merge(peer, peers[index]); } else { peers.add(peer); } } void addPeers(List ps) { for (var p in ps) { addPeer(p); } } 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; } void changeTagForPeers(List ids, List tags) { peers.map((e) { if (ids.contains(e.id)) { e.tags = tags; } }).toList(); } void changeAlias({required String id, required String alias}) { final it = peers.where((element) => element.id == id); if (it.isEmpty) { return; } it.first.alias = alias; } Future pushAb( {bool toastIfFail = true, bool toastIfSucc = true, bool isRetry = false}) async { debugPrint( "pushAb: toastIfFail:$toastIfFail, toastIfSucc:$toastIfSucc, isRetry:$isRetry"); pushError.value = ''; if (isRetry) retrying.value = true; DateTime startTime = DateTime.now(); bool ret = false; try { // avoid double pushes in a row _syncAllFromRecent = true; await syncFromRecent(push: false); //https: //stackoverflow.com/questions/68249333/flutter-getx-updating-item-in-children-list-is-not-reactive peers.refresh(); final api = "${await bind.mainGetApiServer()}/api/ab"; var authHeaders = getHttpHeaders(); authHeaders['Content-Type'] = "application/json"; final peersJsonData = peers.map((e) => e.toAbUploadJson()).toList(); final body = jsonEncode({ "data": jsonEncode({"tags": tags, "peers": peersJsonData}) }); http.Response resp; // support compression if (licensedDevices > 0 && body.length > 1024) { authHeaders['Content-Encoding'] = "gzip"; resp = await http.post(Uri.parse(api), headers: authHeaders, body: GZipCodec().encode(utf8.encode(body))); } else { resp = await http.post(Uri.parse(api), headers: authHeaders, body: body); } if (resp.statusCode == 200 && (resp.body.isEmpty || resp.body.toLowerCase() == 'null')) { ret = true; _saveCache(); } else { Map json = _jsonDecode(resp.body, resp.statusCode); if (json.containsKey('error')) { throw json['error']; } else if (resp.statusCode == 200) { ret = true; _saveCache(); } else { throw 'HTTP ${resp.statusCode}'; } } } catch (e) { pushError.value = '${translate('push_ab_failed_tip')}: ${translate(e.toString())}'; } _syncAllFromRecent = true; if (isRetry) { var ms = (Duration(milliseconds: 200) - DateTime.now().difference(startTime)) .inMilliseconds; ms = ms > 0 ? ms : 0; Future.delayed(Duration(milliseconds: ms), () { retrying.value = false; }); } if (!ret && toastIfFail) { BotToast.showText(contentColor: Colors.red, text: pushError.value); } if (ret && toastIfSucc) { showToast(translate('Successful')); } return ret; } 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 deletePeers(List ids) { peers.removeWhere((e) => ids.contains(e.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 renameTag(String oldTag, String newTag) { if (tags.contains(newTag)) return; tags.value = tags.map((e) { if (e == oldTag) { return newTag; } else { return e; } }).toList(); selectedTags.value = selectedTags.map((e) { if (e == oldTag) { return newTag; } else { return e; } }).toList(); for (var peer in peers) { peer.tags = peer.tags.map((e) { if (e == oldTag) { return newTag; } else { return e; } }).toList(); } } 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; } } Peer merge(Peer r, Peer p) { return Peer( id: p.id, hash: r.hash.isEmpty ? p.hash : r.hash, username: r.username.isEmpty ? p.username : r.username, hostname: r.hostname.isEmpty ? p.hostname : r.hostname, platform: r.platform.isEmpty ? p.platform : r.platform, alias: p.alias.isEmpty ? r.alias : p.alias, tags: p.tags, forceAlwaysRelay: r.forceAlwaysRelay, rdpPort: r.rdpPort, rdpUsername: r.rdpUsername); } Future syncFromRecent({bool push = true}) async { if (!_syncFromRecentLock) { _syncFromRecentLock = true; await _syncFromRecentWithoutLock(push: push); _syncFromRecentLock = false; } } Future _syncFromRecentWithoutLock({bool push = true}) async { bool shouldSync(Peer r, Peer p) { return r.hash != p.hash || r.username != p.username || r.platform != p.platform || r.hostname != p.hostname || (p.alias.isEmpty && r.alias.isNotEmpty); } Future> getRecentPeers() async { try { List filteredPeerIDs; if (_syncAllFromRecent) { _syncAllFromRecent = false; filteredPeerIDs = []; } else { final new_stored_str = await bind.mainGetNewStoredPeers(); if (new_stored_str.isEmpty) return []; filteredPeerIDs = (jsonDecode(new_stored_str) as List) .map((e) => e.toString()) .toList(); if (filteredPeerIDs.isEmpty) return []; } final loadStr = await bind.mainLoadRecentPeersForAb( filter: jsonEncode(filteredPeerIDs)); if (loadStr.isEmpty) { return []; } List mapPeers = jsonDecode(loadStr); List recents = List.empty(growable: true); for (var m in mapPeers) { if (m is Map) { recents.add(Peer.fromJson(m)); } } return recents; } catch (e) { debugPrint('getRecentPeers:$e'); } return []; } try { if (!shouldSyncAb()) return; final recents = await getRecentPeers(); if (recents.isEmpty) return; bool syncChanged = false; bool uiChanged = false; for (var i = 0; i < recents.length; i++) { var r = recents[i]; var index = peers.indexWhere((e) => e.id == r.id); if (index < 0) { peers.add(r); syncChanged = true; uiChanged = true; } else { if (!r.equal(peers[index])) { uiChanged = true; } if (shouldSync(r, peers[index])) { syncChanged = true; } peers[index] = merge(r, peers[index]); } } // Be careful with loop calls if (syncChanged && push) { pushAb(toastIfSucc: false); } else if (uiChanged) { peers.refresh(); } } catch (e) { debugPrint('syncFromRecent:$e'); } } _saveCache() { try { final peersJsonData = peers.map((e) => e.toAbUploadJson()).toList(); final m = { "access_token": bind.mainGetLocalOption(key: 'access_token'), "peers": peersJsonData, "tags": tags.map((e) => e.toString()).toList(), }; bind.mainSaveAb(json: jsonEncode(m)); } catch (e) { debugPrint('ab save:$e'); } } loadCache() async { try { if (_cacheLoadOnceFlag || abLoading.value) return; _cacheLoadOnceFlag = true; final access_token = bind.mainGetLocalOption(key: 'access_token'); if (access_token.isEmpty) return; final cache = await bind.mainLoadAb(); final data = jsonDecode(cache); if (data == null || data['access_token'] != access_token) return; 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 (e) { debugPrint("load ab cache: $e"); } } Map _jsonDecode(String body, int statusCode) { try { Map json = jsonDecode(body); return json; } catch (e) { final err = body.isNotEmpty && body.length < 128 ? body : e.toString(); if (statusCode != 200) { throw 'HTTP $statusCode, $err'; } throw err; } } }