rustdesk/flutter/lib/models/chat_model.dart

import 'dart:convert';

import 'package:dash_chat/dash_chat.dart';
import 'package:flutter/material.dart';

import '../../mobile/widgets/overlay.dart';
import 'model.dart';

class MessageBody {
  ChatUser chatUser;
  List<ChatMessage> chatMessages;
  MessageBody(this.chatUser, this.chatMessages);

  void add(ChatMessage cm) {
    this.chatMessages.add(cm);
  }

  void clear() {
    this.chatMessages.clear();
  }
}

class ChatModel with ChangeNotifier {
  static final clientModeID = -1;

  final ChatUser me = ChatUser(
    uid: "",
    name: "Me",
  );

  late final Map<int, MessageBody> _messages = Map()
    ..[clientModeID] = MessageBody(me, []);

  final _scroller = ScrollController();

  var _currentID = clientModeID;

  ScrollController get scroller => _scroller;

  Map<int, MessageBody> get messages => _messages;

  int get currentID => _currentID;

  WeakReference<FFI> _ffi;

  /// Constructor
  ChatModel(this._ffi);

  ChatUser get currentUser {
    final user = messages[currentID]?.chatUser;
    if (user == null) {
      _currentID = clientModeID;
      return me;
    } else {
      return user;
    }
  }

  changeCurrentID(int id) {
    if (_messages.containsKey(id)) {
      _currentID = id;
      notifyListeners();
    } else {
      final client = _ffi.target?.serverModel.clients[id];
      if (client == null) {
        return debugPrint(
            "Failed to changeCurrentID,remote user doesn't exist");
      }
      final chatUser = ChatUser(
        uid: client.peerId,
        name: client.name,
      );
      _messages[id] = MessageBody(chatUser, []);
      _currentID = id;
      notifyListeners();
    }
  }

  receive(int id, String text) {
    if (text.isEmpty) return;
    // first message show overlay icon
    if (chatIconOverlayEntry == null) {
      showChatIconOverlay();
    }
    late final chatUser;
    if (id == clientModeID) {
      chatUser = ChatUser(
        name: _ffi.target?.ffiModel.pi.username,
        uid: _ffi.target?.getId(),
      );
    } else {
      final client = _ffi.target?.serverModel.clients[id];
      if (client == null) {
        return debugPrint("Failed to receive msg,user doesn't exist");
      }
      chatUser = ChatUser(uid: client.peerId, name: client.name);
    }

    if (!_messages.containsKey(id)) {
      _messages[id] = MessageBody(chatUser, []);
    }
    _messages[id]!.add(ChatMessage(text: text, user: chatUser));
    _currentID = id;
    notifyListeners();
    scrollToBottom();
  }

  scrollToBottom() {
    Future.delayed(Duration(milliseconds: 500), () {
      _scroller.animateTo(_scroller.position.maxScrollExtent,
          duration: Duration(milliseconds: 200),
          curve: Curves.fastLinearToSlowEaseIn);
    });
  }

  send(ChatMessage message) {
    if (message.text != null && message.text!.isNotEmpty) {
      _messages[_currentID]?.add(message);
      if (_currentID == clientModeID) {
        _ffi.target?.setByName("chat_client_mode", message.text!);
      } else {
        final msg = Map()
          ..["id"] = _currentID
          ..["text"] = message.text!;
        _ffi.target?.setByName("chat_server_mode", jsonEncode(msg));
      }
    }
    notifyListeners();
    scrollToBottom();
  }

  close() {
    hideChatIconOverlay();
    hideChatWindowOverlay();
    notifyListeners();
  }

  resetClientMode() {
    _messages[clientModeID]?.clear();
  }
}