FileHandler.cs 7.2 KB

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