polychat/PolyChat/Controller.cs

using System.Collections.Generic;
using System.Diagnostics;
using Newtonsoft.Json.Linq;
using System.Net;
using SocketIOSharp.Server;
using SocketIOSharp.Server.Client;
using PolyChat.Models;
using System.IO;
using System.Threading;
using System;
using System.Text;
using System.Security.Cryptography;
using Windows.Security.Cryptography.Core;
using Windows.Security.Cryptography;
using Windows.Storage.Streams;

namespace PolyChat
{

    // 10.1.211.26 Marc
    // 10.1.218.90 Felix
    // 10.4.141.77 Pat
    class Controller
    {
        // Constants
        private const ushort PORT = 8050;
        // Controller
        private readonly MainPage UIController;
        private readonly FileManager fileManager;
        // Props
        private Dictionary<string, Connection> Connections = new Dictionary<string, Connection>();
        private string OwnName = "";
        private string OwnIP;

        /// <summary>
        /// Initializes Controller with UI access
        /// </summary>
        /// <param name="uiController">UWP UI Controller</param>
        public Controller(MainPage uiController)
        {
            UIController = uiController;
            fileManager = new FileManager(uiController);
            OwnIP = getIP();
            fileManager.loadChats();
            Serve();

            // test
            /*
            UIController.OnIncomingConnection("1.1.1.1");
            UIController.OnIncomingConnection("1.2.3.4");
            UIController.OnIncomingConnection("1.2.4.8");
            */
        }

        public void Connect(string ip)
        {
            Debug.WriteLine("--- Controller.Connect ---");
            if (isInConnections(ip))
            {
                Debug.WriteLine("---- We have an active connection to this client. ABORT! ----");
                CloseChatUI(ip);
                //Todo show error!
            }
            else
            {
                Connections.Add(ip, new Connection(ip, PORT, Data => OnMessage(ip, Data), CloseChat));
            }

        }

        private void Serve()
        {
            Debug.WriteLine("--- Controller.Serve ---");
            SocketIOServer server = new SocketIOServer(new SocketIOServerOption(PORT));
            server.Start();
            Debug.WriteLine("Port " + server.Option.Port);
            Debug.WriteLine("Path " + server.Option.Path);
            // listen for connection
            server.OnConnection((SocketIOSocket socket) =>
            {
                Debug.WriteLine("--- Client connected! ---");
                // setup event listeners
                socket.On("initial", (JToken[] data) =>
                {
                    Debug.WriteLine("--- initial packet received ---");
                    string ForeignIp = data[0].ToString();

                    Debug.WriteLine($"--- this ip was in the inital packet: {ForeignIp} ---");
                    if (isInConnections(ForeignIp))
                    {
                        Debug.WriteLine("---- We have an active connection to this client. ABORT! ----");//Todo show error!
                        CloseChatUI(ForeignIp);
                    }
                    else
                    {
                        Debug.WriteLine("---- Added new Connection ----");
                        Connections.Add(ForeignIp, new Connection(socket, Data => OnMessage(ForeignIp, Data), CloseChat));
                        UIController.OnIncomingConnection(ForeignIp);
                        fileManager.loadChat(ForeignIp);
                    }
                });
            });
        }

        public void SendBroadcastMessage(string type, string content)
        {
            Debug.WriteLine("--- Controller.Broadcast ---");
            foreach (KeyValuePair<string, Connection> entry in Connections)
            {
                SendMessage(entry.Key, type, content);
            }
        }

        public void SendMessage(string ip, string type, string content)
        {
            Debug.WriteLine("--- Controller.SendMessage ---");
            Debug.WriteLine($"{type} -> {ip} content: {content}");
            JObject json = new JObject(
                new JProperty("type", type),
                new JProperty("content", content)
            );
            Debug.WriteLine($"json: {json.ToString()}");
            // send as json
            Connections[ip].SendMessage(json.ToString());
            // save to logs
            fileManager.saveChats(ip, json.ToString(), DateTime.Now);
        }

        private void OnMessage(string ip, JToken[] data)
        {
            Debug.WriteLine("--- Controller.OnMessage ---");
            if (data != null && data.Length > 0 && data[0] != null)
            {
                DateTime now = DateTime.Now;
                Debug.WriteLine("RAW: " + data[0]);
                UIController.OnIncomingMessage(ip, data[0].ToString(), now);
                fileManager.saveChats(ip, data[0].ToString(), now);
            }
            else Debug.WriteLine("Undefined: " + data);
        }

        public void CloseChat(string IP, bool wasConnected = true)
        {
            Debug.WriteLine($"Deleting connection with IP:{IP}");
            if (IP != null && Connections.ContainsKey(IP))
            {
                Connections[IP].Close();
                Connections.Remove(IP);
            }
            CloseChatUI(IP, wasConnected);
        }

        private void CloseChatUI(string IP, bool wasConnected = true)
        {
            UIController.OnChatPartnerDeleted(IP);
            string heading = wasConnected ? "Connection Closed" : "Connection Failed";
            UIController.ShowConnectionError(IP, heading, $"Connecting to {IP} failed...");
        }

        private bool isInConnections(string IP)
        {
            return Connections.ContainsKey(IP);
        }
        public bool IsConnected(string ip)
        {
            return Connections.ContainsKey(ip) && Connections[ip].IsConnected();
        }

        public static string getIP()
        {
            IPHostEntry ipEntry = Dns.GetHostEntry(Dns.GetHostName());
            IPAddress[] addrList = ipEntry.AddressList;
            for (short i = 0; i < addrList.Length; i++)
            {
                if (addrList[i].ToString().Substring(0, 3).Equals("10."))
                {
                    return addrList[i].ToString();
                }
            }
            return null;
        }

        private void encode(string json)
        {
            String ecryptetText = FileManager.encrypt(json);
            Debug.WriteLine(ecryptetText);
            Debug.WriteLine(FileManager.decrypt(ecryptetText));
        }
    }
}