FileHandler.cs 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284
  1. using ErgometerLibrary;
  2. using System;
  3. using System.Collections.Generic;
  4. using System.IO;
  5. using System.Linq;
  6. namespace ErgometerServer
  7. {
  8. public class FileHandler
  9. {
  10. public static string DataFolder { get; } = Path.Combine(System.Environment.GetFolderPath(System.Environment.SpecialFolder.MyDocuments), "Ergometer");
  11. public static string UsersFile { get; } = Path.Combine(DataFolder, "users.ergo");
  12. public static void CheckStorage()
  13. {
  14. if (!Directory.Exists(DataFolder))
  15. {
  16. Directory.CreateDirectory(DataFolder);
  17. }
  18. if (!File.Exists(UsersFile))
  19. {
  20. using (Stream stream = File.Open(UsersFile, FileMode.Create))
  21. {
  22. BinaryWriter writer = new BinaryWriter(stream);
  23. writer.Write(1);
  24. writer.Write("Doctor0tVfW");
  25. writer.Write(Helper.Base64Encode("password"));
  26. }
  27. }
  28. }
  29. //SESSION
  30. public static int GenerateSession()
  31. {
  32. string[] existingSessions = Directory.GetDirectories(DataFolder);
  33. Random rand = new Random();
  34. int sessionID = rand.Next(0, int.MaxValue);
  35. while (existingSessions.Contains(sessionID.ToString()))
  36. {
  37. sessionID = rand.Next(int.MinValue, int.MaxValue);
  38. }
  39. return sessionID;
  40. }
  41. public static void CreateSession(int session, string naam)
  42. {
  43. Directory.CreateDirectory(GetSessionFolder(session));
  44. using (File.Create(Path.Combine(GetSessionFile(session)))) ;
  45. using (File.Create(Path.Combine(GetSessionMetingen(session)))) ;
  46. using (File.Create(Path.Combine(GetSessionChat(session)))) ;
  47. using (File.Create(Path.Combine(GetSessionTest(session)))) ;
  48. using (File.Create(Path.Combine(GetSessionPersonalData(session)))) ;
  49. File.WriteAllText(GetSessionFile(session), naam + Environment.NewLine + Helper.Now);
  50. Console.WriteLine("Created session at " + Helper.MillisecondsToTime(Helper.Now));
  51. }
  52. public static void WriteMetingen(int session, List<Meting> metingen)
  53. {
  54. if (metingen.Count <= 20 && Directory.Exists(GetSessionFolder(session)))
  55. {
  56. Directory.Delete(GetSessionFolder(session), true);
  57. }
  58. else
  59. {
  60. string json = Newtonsoft.Json.JsonConvert.SerializeObject(metingen.ToArray());
  61. File.WriteAllText(GetSessionMetingen(session), json);
  62. Console.WriteLine("Writing metingen: " + GetSessionMetingen(session));
  63. File.WriteAllText(GetSessionFile(session), File.ReadAllText(GetSessionFile(session)) + Environment.NewLine + Helper.Now);
  64. }
  65. }
  66. public static List<Meting> ReadMetingen(int session)
  67. {
  68. string json = File.ReadAllText(GetSessionMetingen(session));
  69. List<Meting> metingen = Newtonsoft.Json.JsonConvert.DeserializeObject<List<Meting>>(json);
  70. Console.WriteLine("Reading metingen: " + GetSessionMetingen(session));
  71. return metingen;
  72. }
  73. public static List<Tuple<int, string, double>> GetAllSessions()
  74. {
  75. string[] directories = Directory.GetDirectories(DataFolder);
  76. List<Tuple<int,string,double>> sessiondata = new List<Tuple<int, string, double>>();
  77. for (int i = 0; i < directories.Length; i++)
  78. {
  79. string directoryname = Path.GetFileName(directories[i]);
  80. string props = File.ReadAllText(GetSessionFile(int.Parse(directoryname)));
  81. string[] stringSeparators = new string[] { "\r\n" };
  82. string[] properties = props.Split(stringSeparators, StringSplitOptions.None);
  83. int session = int.Parse(directoryname);
  84. bool active = false;
  85. foreach(ClientThread t in Server.clients)
  86. {
  87. if (t.session == session)
  88. active = true;
  89. }
  90. string name = properties[0];
  91. double date = double.Parse(properties[1]);
  92. if (!active)
  93. {
  94. Tuple<int, string, double> tup = new Tuple<int, string, double>(session, name, date);
  95. sessiondata.Add(tup);
  96. }
  97. }
  98. return sessiondata;
  99. }
  100. public static void WriteChat(int session, List<ChatMessage> chat)
  101. {
  102. if (Directory.Exists(GetSessionFolder(session)))
  103. {
  104. /*
  105. string write = "";
  106. foreach (ChatMessage c in chat)
  107. {
  108. write += c.ToString() + "\n";
  109. }
  110. File.WriteAllText(GetSessionChat(session), write);
  111. Console.WriteLine("Writing chat: " + GetSessionChat(session));
  112. */
  113. string json = Newtonsoft.Json.JsonConvert.SerializeObject(chat.ToArray());
  114. File.WriteAllText(GetSessionChat(session), json);
  115. Console.WriteLine("Writing chat: " + GetSessionChat(session));
  116. }
  117. }
  118. public static List<ChatMessage> ReadChat(int session)
  119. {
  120. string json = File.ReadAllText(GetSessionChat(session));
  121. List<ChatMessage> chat = Newtonsoft.Json.JsonConvert.DeserializeObject<List<ChatMessage>>(json);
  122. return chat;
  123. }
  124. private static string GetSessionFolder(int session)
  125. {
  126. return Path.Combine(DataFolder, session.ToString());
  127. }
  128. private static string GetSessionFile(int session)
  129. {
  130. return Path.Combine(DataFolder, session.ToString(), "session.prop");
  131. }
  132. private static string GetSessionMetingen(int session)
  133. {
  134. return Path.Combine(DataFolder, session.ToString(), "metingen.ergo");
  135. }
  136. private static string GetSessionChat(int session)
  137. {
  138. return Path.Combine(DataFolder, session.ToString(), "chat.ergo");
  139. }
  140. private static string GetSessionTest(int session)
  141. {
  142. return Path.Combine(DataFolder, session.ToString(), "testresult.ergo");
  143. }
  144. private static string GetSessionPersonalData(int session)
  145. {
  146. return Path.Combine(DataFolder, session.ToString(), "personal.ergo");
  147. }
  148. //USER MANAGEMENT
  149. public static Dictionary<string, string> LoadUsers()
  150. {
  151. Dictionary<string, string> users = new Dictionary<string, string>();
  152. using (Stream stream = File.Open(UsersFile, FileMode.Open))
  153. {
  154. BinaryReader reader = new BinaryReader(stream);
  155. int count = reader.ReadInt32();
  156. for (int n = 0; n < count; n++)
  157. {
  158. var key = reader.ReadString();
  159. var value = Helper.Base64Decode(reader.ReadString());
  160. users.Add(key, value);
  161. }
  162. }
  163. return users;
  164. }
  165. public static void SaveUsers(Dictionary<string, string> users)
  166. {
  167. using (Stream stream = File.Open(UsersFile, FileMode.Open))
  168. {
  169. BinaryWriter writer = new BinaryWriter(stream);
  170. writer.Write(users.Count);
  171. foreach (var kvp in users)
  172. {
  173. writer.Write(kvp.Key);
  174. writer.Write(Helper.Base64Encode(kvp.Value));
  175. }
  176. writer.Flush();
  177. }
  178. }
  179. // TESTRESULTS
  180. public static void SaveTestResult(NetCommand input)
  181. {
  182. using (Stream stream = File.Open(GetSessionTest(input.Session), FileMode.Open))
  183. {
  184. BinaryWriter writer = new BinaryWriter(stream);
  185. writer.Write(input.VO2Max);
  186. writer.Write(input.MET);
  187. writer.Write(input.PopulationAvg);
  188. writer.Write(input.ZScore);
  189. writer.Write(input.Rating);
  190. writer.Flush();
  191. }
  192. }
  193. public static NetCommand ReadTestResult(int session)
  194. {
  195. NetCommand command = new NetCommand(NetCommand.CommandType.ERROR, session);
  196. using (Stream stream = File.Open(GetSessionTest(session), FileMode.Open))
  197. {
  198. command = new NetCommand(NetCommand.CommandType.TESTRESULT, session);
  199. BinaryReader reader = new BinaryReader(stream);
  200. command.VO2Max = reader.ReadDouble();
  201. command.MET = reader.ReadDouble();
  202. command.PopulationAvg = reader.ReadDouble();
  203. command.ZScore = reader.ReadDouble();
  204. command.Rating = reader.ReadString();
  205. }
  206. return command;
  207. }
  208. // PERSONALDATA
  209. public static void SavePersonalData(NetCommand input)
  210. {
  211. using (Stream stream = File.Open(GetSessionPersonalData(input.Session), FileMode.Open))
  212. {
  213. BinaryWriter writer = new BinaryWriter(stream);
  214. writer.Write(input.Geslacht);
  215. writer.Write(input.Gewicht);
  216. writer.Write(input.Leeftijd);
  217. writer.Write(input.Lengte);
  218. writer.Flush();
  219. }
  220. }
  221. public static NetCommand ReadPersonalData(int session)
  222. {
  223. NetCommand command = new NetCommand(NetCommand.CommandType.ERROR, session);
  224. using (Stream stream = File.Open(GetSessionPersonalData(session), FileMode.Open))
  225. {
  226. command = new NetCommand(NetCommand.CommandType.PERSONDATA, session);
  227. BinaryReader reader = new BinaryReader(stream);
  228. command.Geslacht = reader.ReadChar();
  229. command.Gewicht = reader.ReadInt32();
  230. command.Leeftijd = reader.ReadInt32();
  231. command.Lengte = reader.ReadInt32();
  232. }
  233. return command;
  234. }
  235. }
  236. }