FileHandler.cs 7.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209
  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. File.WriteAllText(GetSessionFile(session), naam + Environment.NewLine + Helper.Now);
  48. Console.WriteLine("Created session at " + Helper.MillisecondsToTime(Helper.Now));
  49. }
  50. public static void WriteMetingen(int session, List<Meting> metingen)
  51. {
  52. if (metingen.Count <= 20 && Directory.Exists(GetSessionFolder(session)))
  53. {
  54. Directory.Delete(GetSessionFolder(session), true);
  55. }
  56. else
  57. {
  58. string json = Newtonsoft.Json.JsonConvert.SerializeObject(metingen.ToArray());
  59. File.WriteAllText(GetSessionMetingen(session), json);
  60. Console.WriteLine("Writing metingen: " + GetSessionMetingen(session));
  61. File.WriteAllText(GetSessionFile(session), File.ReadAllText(GetSessionFile(session)) + Environment.NewLine + Helper.Now);
  62. }
  63. }
  64. public static List<Meting> ReadMetingen(int session)
  65. {
  66. string json = File.ReadAllText(GetSessionMetingen(session));
  67. List<Meting> metingen = Newtonsoft.Json.JsonConvert.DeserializeObject<List<Meting>>(json);
  68. Console.WriteLine("Reading metingen: " + GetSessionMetingen(session));
  69. return metingen;
  70. }
  71. internal static void SaveTestResult(NetCommand input, int session)
  72. {
  73. throw new NotImplementedException();
  74. }
  75. public static List<Tuple<int, string, double>> GetAllSessions()
  76. {
  77. string[] directories = Directory.GetDirectories(DataFolder);
  78. List<Tuple<int,string,double>> sessiondata = new List<Tuple<int, string, double>>();
  79. for (int i = 0; i < directories.Length; i++)
  80. {
  81. string directoryname = Path.GetFileName(directories[i]);
  82. string props = File.ReadAllText(GetSessionFile(int.Parse(directoryname)));
  83. string[] stringSeparators = new string[] { "\r\n" };
  84. string[] properties = props.Split(stringSeparators, StringSplitOptions.None);
  85. int session = int.Parse(directoryname);
  86. bool active = false;
  87. foreach(ClientThread t in Server.clients)
  88. {
  89. if (t.session == session)
  90. active = true;
  91. }
  92. string name = properties[0];
  93. double date = double.Parse(properties[1]);
  94. if (!active)
  95. {
  96. Tuple<int, string, double> tup = new Tuple<int, string, double>(session, name, date);
  97. sessiondata.Add(tup);
  98. }
  99. }
  100. return sessiondata;
  101. }
  102. public static void WriteChat(int session, List<ChatMessage> chat)
  103. {
  104. if (Directory.Exists(GetSessionFolder(session)))
  105. {
  106. /*
  107. string write = "";
  108. foreach (ChatMessage c in chat)
  109. {
  110. write += c.ToString() + "\n";
  111. }
  112. File.WriteAllText(GetSessionChat(session), write);
  113. Console.WriteLine("Writing chat: " + GetSessionChat(session));
  114. */
  115. string json = Newtonsoft.Json.JsonConvert.SerializeObject(chat.ToArray());
  116. File.WriteAllText(GetSessionChat(session), json);
  117. Console.WriteLine("Writing chat: " + GetSessionChat(session));
  118. }
  119. }
  120. public static List<ChatMessage> ReadChat(int session)
  121. {
  122. string json = File.ReadAllText(GetSessionChat(session));
  123. List<ChatMessage> chat = Newtonsoft.Json.JsonConvert.DeserializeObject<List<ChatMessage>>(json);
  124. return chat;
  125. }
  126. private static string GetSessionFolder(int session)
  127. {
  128. return Path.Combine(DataFolder, session.ToString());
  129. }
  130. private static string GetSessionFile(int session)
  131. {
  132. return Path.Combine(DataFolder, session.ToString(), "session.prop");
  133. }
  134. private static string GetSessionMetingen(int session)
  135. {
  136. return Path.Combine(DataFolder, session.ToString(), "metingen.ergo");
  137. }
  138. private static string GetSessionChat(int session)
  139. {
  140. return Path.Combine(DataFolder, session.ToString(), "chat.ergo");
  141. }
  142. //USER MANAGEMENT
  143. public static Dictionary<string, string> LoadUsers()
  144. {
  145. Dictionary<string, string> users = new Dictionary<string, string>();
  146. using (Stream stream = File.Open(UsersFile, FileMode.Open))
  147. {
  148. BinaryReader reader = new BinaryReader(stream);
  149. int count = reader.ReadInt32();
  150. for (int n = 0; n < count; n++)
  151. {
  152. var key = reader.ReadString();
  153. var value = Helper.Base64Decode(reader.ReadString());
  154. users.Add(key, value);
  155. }
  156. }
  157. return users;
  158. }
  159. public static void SaveUsers(Dictionary<string, string> users)
  160. {
  161. using (Stream stream = File.Open(UsersFile, FileMode.Open))
  162. {
  163. BinaryWriter writer = new BinaryWriter(stream);
  164. writer.Write(users.Count);
  165. foreach (var kvp in users)
  166. {
  167. writer.Write(kvp.Key);
  168. writer.Write(Helper.Base64Encode(kvp.Value));
  169. }
  170. writer.Flush();
  171. }
  172. }
  173. }
  174. }