rustdesk/flutter/lib/models/user_model.dart

import 'dart:async';
import 'dart:convert';

import 'package:get/get.dart';
import 'package:http/http.dart' as http;

import '../common.dart';
import 'model.dart';
import 'platform_model.dart';

class UserModel {
  var userName = "".obs;
  WeakReference<FFI> parent;

  UserModel(this.parent) {
    refreshCurrentUser();
  }

  void refreshCurrentUser() async {
    await getUserName();
    final token = await bind.mainGetLocalOption(key: "access_token");
    if (token == '') return;
    final url = await bind.mainGetApiServer();
    final body = {
      'id': await bind.mainGetMyId(),
      'uuid': await bind.mainGetUuid()
    };
    try {
      final response = await http.post(Uri.parse('$url/api/currentUser'),
          headers: {
            "Content-Type": "application/json",
            "Authorization": "Bearer $token"
          },
          body: json.encode(body));
      final status = response.statusCode;
      if (status == 401 || status == 400) {
        resetToken();
        return;
      }
      await _parseResp(response.body);
    } catch (e) {
      print('Failed to refreshCurrentUser: $e');
    }
  }

  void resetToken() async {
    await bind.mainSetLocalOption(key: "access_token", value: "");
    await bind.mainSetLocalOption(key: "user_info", value: "");
    userName.value = "";
  }

  Future<String> _parseResp(String body) async {
    final data = json.decode(body);
    final error = data['error'];
    if (error != null) {
      return error!;
    }
    final token = data['access_token'];
    if (token != null) {
      await bind.mainSetLocalOption(key: "access_token", value: token);
    }
    final info = data['user'];
    if (info != null) {
      final value = json.encode(info);
      await bind.mainSetOption(key: "user_info", value: value);
      userName.value = info["name"];
    }
    return '';
  }

  Future<String> getUserName() async {
    if (userName.isNotEmpty) {
      return userName.value;
    }
    final userInfo = await bind.mainGetLocalOption(key: 'user_info');
    if (userInfo.trim().isEmpty) {
      return "";
    }
    final m = jsonDecode(userInfo);
    userName.value = m['name'] ?? '';
    return userName.value;
  }

  Future<void> logOut() async {
    final tag = gFFI.dialogManager.showLoading(translate('Waiting'));
    final url = await bind.mainGetApiServer();
    final _ = await http.post(Uri.parse("$url/api/logout"),
        body: {
          "id": await bind.mainGetMyId(),
          "uuid": await bind.mainGetUuid(),
        },
        headers: await getHttpHeaders());
    await Future.wait([
      bind.mainSetLocalOption(key: 'access_token', value: ''),
      bind.mainSetLocalOption(key: 'user_info', value: ''),
      bind.mainSetLocalOption(key: 'selected-tags', value: ''),
    ]);
    parent.target?.abModel.clear();
    userName.value = "";
    gFFI.dialogManager.dismissByTag(tag);
  }

  Future<Map<String, dynamic>> login(String userName, String pass) async {
    final url = await bind.mainGetApiServer();
    try {
      final resp = await http.post(Uri.parse("$url/api/login"),
          headers: {"Content-Type": "application/json"},
          body: jsonEncode({
            "username": userName,
            "password": pass,
            "id": await bind.mainGetMyId(),
            "uuid": await bind.mainGetUuid()
          }));
      final body = jsonDecode(resp.body);
      bind.mainSetLocalOption(
          key: "access_token", value: body['access_token'] ?? "");
      bind.mainSetLocalOption(
          key: "user_info", value: jsonEncode(body['user']));
      this.userName.value = body['user']?['name'] ?? "";
      return body;
    } catch (err) {
      return {"error": "$err"};
    }
  }
}