simple-chat-server/bin/tcp_server.dart

/*
 * @Author       : Linloir
 * @Date         : 2022-10-06 15:44:16
 * @LastEditTime : 2022-10-14 10:26:00
 * @Description  : 
 */

import 'dart:io';

import 'package:tcp_server/database.dart';
import 'package:tcp_server/requesthandler.dart';
import 'package:tcp_server/tcpcontroller/controller.dart';
import 'package:tcp_server/tcpcontroller/payload/message.dart';
import 'package:tcp_server/tcpcontroller/request.dart';
import 'package:tcp_server/tcpcontroller/response.dart';

void main(List<String> arguments) async {
  //Set port
  var port = arguments.isEmpty ? 20706 : int.tryParse(arguments[0]) ?? 20706;
  
  //Create nessesary working directories
  await Directory('${Directory.current.path}/.tmp').create();
  await Directory('${Directory.current.path}/.data').create();
  await Directory('${Directory.current.path}/.data/files').create();

  await DataBaseHelper().initialize();
  var tokenMap = <int, TCPController>{};
  var controllerMap = <TCPController, Future<int>>{};
  var listenSocket = await ServerSocket.bind('127.0.0.1', port);
  listenSocket.listen(
    (socket) {
      var controller = TCPController(socket: socket);
      controller.requestStreamBroadcast.listen((request) async {
        print('[L] ${request.toJSON}');
        if(!(await DataBaseHelper().isTokenValid(tokenid: request.tokenID))) {
          if(controllerMap[controller] == null) {
            controllerMap[controller] = (() async => (await DataBaseHelper().createToken()))();
          }
          request.tokenID = await controllerMap[controller];
          var tokenResponse = TCPResponse(
            type: ResponseType.token,
            status: ResponseStatus.ok,
            body: {
              "tokenid": request.tokenID
            }
          );
          controller.outStream.add(tokenResponse);
        }
        tokenMap[request.tokenID!] = tokenMap[request.tokenID!] ?? controller;
        switch(request.requestType) {
          case RequestType.checkState: {
            var response = await onCheckState(request, socket);
            controller.outStream.add(response);
            break;
          }
          case RequestType.register: {
            var response = await onRegister(request, socket);
            controller.outStream.add(response);
            break;
          }
          case RequestType.login: {
            var response = await onLogin(request, socket);
            controller.outStream.add(response);
            break;
          }
          case RequestType.logout: {
            var response = await onLogout(request, socket);
            controller.outStream.add(response);
            break;
          }
          case RequestType.profile: {
            var response = await onFetchProfile(request, socket);
            controller.outStream.add(response);
            break;
          }
          case RequestType.modifyProfile: {
            var response = await onModifyProfile(request, socket);
            controller.outStream.add(response);
            break;
          }
          case RequestType.modifyPassword: {
            var response = await onModifyPassword(request, socket);
            controller.outStream.add(response);
            break;
          }
          case RequestType.sendMessage: {
            //Forword Message
            var message = Message.fromJSONObject(request.body);
            await DataBaseHelper().setFetchHistoryFor(
              tokenID: request.tokenID, 
              newTimeStamp: message.timestamp
            );
            var originUserID = message.senderID;
            var onlineDevices = await DataBaseHelper().fetchTokenIDsViaUserID(userID: originUserID);
            for(var device in onlineDevices) {
              if(device == request.tokenID) {
                continue;
              }
              var targetController = tokenMap[device];
              var forwardResponse = TCPResponse(
                type: ResponseType.forwardMessage,
                status: ResponseStatus.ok,
                body: message.jsonObject
              );
              targetController?.outStream.add(forwardResponse);
              //Update Fetch Histories
              await DataBaseHelper().setFetchHistoryFor(
                tokenID: device, 
                newTimeStamp: message.timestamp
              );
            }
            var targetUserID = message.receiverID;
            var targetDevices = await DataBaseHelper().fetchTokenIDsViaUserID(userID: targetUserID);
            for(var device in targetDevices) {
              //Forward to socket
              var targetController = tokenMap[device];
              var forwardResponse = TCPResponse(
                type: ResponseType.forwardMessage,
                status: ResponseStatus.ok,
                body: message.jsonObject
              );
              targetController?.outStream.add(forwardResponse);
              //Update Fetch Histories
              await DataBaseHelper().setFetchHistoryFor(
                tokenID: device, 
                newTimeStamp: message.timestamp
              );
            }
            var response = await onSendMessage(request, socket);
            controller.outStream.add(response);
            break;
          }
          case RequestType.fetchMessage: {
            var response = await onFetchMessage(request, socket);
            controller.outStream.add(response);
            break;
          }
          case RequestType.findFile: {
            var response = await onFindFile(request, socket);
            controller.outStream.add(response);
            break;
          }
          case RequestType.fetchFile: {
            var response = await onFetchFile(request, socket);
            controller.outStream.add(response);
            break;
          }
          case RequestType.searchUser: {
            var response = await onSearchUser(request, socket);
            controller.outStream.add(response);
            break;
          }
          case RequestType.addContact: {
            var response = await onAddContact(request, socket);
            controller.outStream.add(response);
            break;
          }
          case RequestType.fetchContact: {
            var response = await onFetchContact(request, socket);
            controller.outStream.add(response);
            break;
          }
          case RequestType.unknown: {
            var response = await onUnknownRequest(request, socket);
            controller.outStream.add(response);
            break;
          }
          default: {
            print('[E] Drop out of switch case');
          }
        }
        //Clear temp file
        request.payload?.delete();
      });
    },
  );
}