2.1-CycleErgometer/ErgometerIPR/ErgometerServer/FileHandler.cs

using ErgometerLibrary;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;

namespace ErgometerServer
{
    public class FileHandler
    {
        public static string DataFolder { get; } = Path.Combine(System.Environment.GetFolderPath(System.Environment.SpecialFolder.MyDocuments), "Ergometer");
        public static string UsersFile { get; } = Path.Combine(DataFolder, "users.ergo");

        public static void CheckStorage()
        {
            if (!Directory.Exists(DataFolder))
            {
                Directory.CreateDirectory(DataFolder);
            }

            if (!File.Exists(UsersFile))
            {
                using (Stream stream = File.Open(UsersFile, FileMode.Create))
                {
                    BinaryWriter writer = new BinaryWriter(stream);
                    writer.Write(1);
                    writer.Write("Doctor0tVfW");
                    writer.Write(Helper.Base64Encode("password"));
                }
            }
        }

        //SESSION 
        public static int GenerateSession()
        {
            string[] existingSessions = Directory.GetDirectories(DataFolder);

            Random rand = new Random();
            int sessionID = rand.Next(0, int.MaxValue);


            while (existingSessions.Contains(sessionID.ToString()))
            {
                sessionID = rand.Next(int.MinValue, int.MaxValue);
            }

            return sessionID;
        }

        public static void CreateSession(int session, string naam)
        {
            Directory.CreateDirectory(GetSessionFolder(session));

            using (File.Create(Path.Combine(GetSessionFile(session)))) ;
            using (File.Create(Path.Combine(GetSessionMetingen(session)))) ;
            using (File.Create(Path.Combine(GetSessionChat(session)))) ;
            using (File.Create(Path.Combine(GetSessionTest(session)))) ;
            using (File.Create(Path.Combine(GetSessionPersonalData(session)))) ;

            File.WriteAllText(GetSessionFile(session), naam + Environment.NewLine + Helper.Now);
            Console.WriteLine("Created session at " + Helper.MillisecondsToTime(Helper.Now));
        }

        public static void WriteMetingen(int session, List<Meting> metingen)
        {
            if (metingen.Count <= 20 && Directory.Exists(GetSessionFolder(session)))
            {
                Directory.Delete(GetSessionFolder(session), true);
            }
            else
            {
                string json = Newtonsoft.Json.JsonConvert.SerializeObject(metingen.ToArray());
                File.WriteAllText(GetSessionMetingen(session), json);
                Console.WriteLine("Writing metingen: " + GetSessionMetingen(session));
                File.WriteAllText(GetSessionFile(session), File.ReadAllText(GetSessionFile(session)) + Environment.NewLine + Helper.Now);
            }
            
        }

        public static List<Meting> ReadMetingen(int session)
        {
            string json = File.ReadAllText(GetSessionMetingen(session));

            List<Meting> metingen = Newtonsoft.Json.JsonConvert.DeserializeObject<List<Meting>>(json);
            if (metingen == null)
                metingen = new List<Meting>();
            Console.WriteLine("Reading metingen: " + GetSessionMetingen(session));
            return metingen;
        }

        public static List<Tuple<int, string, double>> GetAllSessions()
        {
            string[] directories = Directory.GetDirectories(DataFolder);
            List<Tuple<int,string,double>> sessiondata = new List<Tuple<int, string, double>>();

            for (int i = 0; i < directories.Length; i++)
            {
                string directoryname = Path.GetFileName(directories[i]);

                string props = File.ReadAllText(GetSessionFile(int.Parse(directoryname)));

                string[] stringSeparators = new string[] { "\r\n" };
                string[] properties = props.Split(stringSeparators, StringSplitOptions.None);

                int session = int.Parse(directoryname);

                bool active = false;

                foreach(ClientThread t in Server.clients)
                {
                    if (t.session == session)
                        active = true;
                }

                string name = properties[0];
                double date = double.Parse(properties[1]);

                if (!active)
                {
                    Tuple<int, string, double> tup = new Tuple<int, string, double>(session, name, date);
                    sessiondata.Add(tup);
                }
            }

            return sessiondata;
        }

        public static void WriteChat(int session, List<ChatMessage> chat)
        {
            if (Directory.Exists(GetSessionFolder(session)))
            {
                /*
                string write = "";
                foreach (ChatMessage c in chat)
                {
                    write += c.ToString() + "\n";
                }

                File.WriteAllText(GetSessionChat(session), write);
                Console.WriteLine("Writing chat: " + GetSessionChat(session));
                */
                string json = Newtonsoft.Json.JsonConvert.SerializeObject(chat.ToArray());
                File.WriteAllText(GetSessionChat(session), json);
                Console.WriteLine("Writing chat: " + GetSessionChat(session));
            }
        }

        public static List<ChatMessage> ReadChat(int session)
        {
            string json = File.ReadAllText(GetSessionChat(session));

            List<ChatMessage> chat = Newtonsoft.Json.JsonConvert.DeserializeObject<List<ChatMessage>>(json);
            if (chat == null)
                chat = new List<ChatMessage>();
            return chat;
        }

        private static string GetSessionFolder(int session)
        {
            return Path.Combine(DataFolder, session.ToString());
        }
        private static string GetSessionFile(int session)
        {
            return Path.Combine(DataFolder, session.ToString(), "session.prop");
        }
        private static string GetSessionMetingen(int session)
        {
            return Path.Combine(DataFolder, session.ToString(), "metingen.ergo");
        }
        private static string GetSessionChat(int session)
        {
            return Path.Combine(DataFolder, session.ToString(), "chat.ergo");
        }
        private static string GetSessionTest(int session)
        {
            return Path.Combine(DataFolder, session.ToString(), "testresult.ergo");
        }
        private static string GetSessionPersonalData(int session)
        {
            return Path.Combine(DataFolder, session.ToString(), "personal.ergo");
        }

        //USER MANAGEMENT
        public static Dictionary<string, string> LoadUsers()
        {
            Dictionary<string, string> users = new Dictionary<string, string>();

            using (Stream stream = File.Open(UsersFile, FileMode.Open))
            {
                BinaryReader reader = new BinaryReader(stream);
                int count = reader.ReadInt32();
                for (int n = 0; n < count; n++)
                {
                    var key = reader.ReadString();
                    var value = Helper.Base64Decode(reader.ReadString());
                    users.Add(key, value);
                }
            }

            return users;
        }

        public static void SaveUsers(Dictionary<string, string> users)
        {
            using (Stream stream = File.Open(UsersFile, FileMode.Open))
            {
                BinaryWriter writer = new BinaryWriter(stream);
                writer.Write(users.Count);
                foreach (var kvp in users)
                {
                    writer.Write(kvp.Key);
                    writer.Write(Helper.Base64Encode(kvp.Value));
                }
                writer.Flush();
            }
        }



        // TESTRESULTS

        public static void SaveTestResult(NetCommand input)
        {
            using (Stream stream = File.Open(GetSessionTest(input.Session), FileMode.Open))
            {
                BinaryWriter writer = new BinaryWriter(stream);
                writer.Write(input.VO2Max);
                writer.Write(input.MET);
                writer.Write(input.PopulationAvg);
                writer.Write(input.ZScore);
                writer.Write(input.Rating);
                writer.Flush();
            }
        }

        public static NetCommand ReadTestResult(int session)
        {
            NetCommand command = new NetCommand(NetCommand.CommandType.ERROR, session);

            using (Stream stream = File.Open(GetSessionTest(session), FileMode.Open))
            {
                command = new NetCommand(NetCommand.CommandType.TESTRESULT, session);
                BinaryReader reader = new BinaryReader(stream);
                try {
                    command.VO2Max = reader.ReadDouble();
                    command.MET = reader.ReadDouble();
                    command.PopulationAvg = reader.ReadDouble();
                    command.ZScore = reader.ReadDouble();
                    command.Rating = reader.ReadString();
                }
                catch(Exception e)
                {
                    command.VO2Max = 0;
                    command.MET = 0;
                    command.PopulationAvg = 0;
                    command.ZScore = 0;
                    command.Rating = "Niet afgerond";
                }
            }

            return command;
        }



        // PERSONALDATA

        public static void SavePersonalData(NetCommand input)
        {
            using (Stream stream = File.Open(GetSessionPersonalData(input.Session), FileMode.Open))
            {
                BinaryWriter writer = new BinaryWriter(stream);
                writer.Write(input.Geslacht);
                writer.Write(input.Gewicht);
                writer.Write(input.Leeftijd);
                writer.Write(input.Lengte);
                writer.Flush();
            }
        }

        public static NetCommand ReadPersonalData(int session)
        {
            NetCommand command = new NetCommand(NetCommand.CommandType.ERROR, session);

            using (Stream stream = File.Open(GetSessionPersonalData(session), FileMode.Open))
            {
                command = new NetCommand(NetCommand.CommandType.PERSONDATA, session);
                BinaryReader reader = new BinaryReader(stream);
                try {
                    command.Geslacht = reader.ReadChar();
                    command.Gewicht = reader.ReadInt32();
                    command.Leeftijd = reader.ReadInt32();
                    command.Lengte = reader.ReadInt32();
                }
                catch (Exception e)
                {
                    command = new NetCommand(NetCommand.CommandType.ERROR, session);
                }
            }

            return command;
        }
    }
}