ConsoleComponent.cs 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120
  1. using System;
  2. using System.Threading;
  3. using System.Threading.Tasks;
  4. namespace ETModel
  5. {
  6. [ObjectSystem]
  7. public class ConsoleComponentAwakeSystem : StartSystem<ConsoleComponent>
  8. {
  9. public override void Start(ConsoleComponent self)
  10. {
  11. self.Start().NoAwait();
  12. }
  13. }
  14. public static class ConsoleMode
  15. {
  16. public const string None = "";
  17. public const string Repl = "repl";
  18. }
  19. public class ConsoleComponent: Entity
  20. {
  21. public CancellationTokenSource CancellationTokenSource;
  22. public string Mode = "";
  23. public async ETVoid Start()
  24. {
  25. this.CancellationTokenSource = new CancellationTokenSource();
  26. while (true)
  27. {
  28. try
  29. {
  30. string line = await Task.Factory.StartNew(() =>
  31. {
  32. Console.Write($"{this.Mode}> ");
  33. return Console.In.ReadLine();
  34. }, this.CancellationTokenSource.Token);
  35. line = line.Trim();
  36. switch (line)
  37. {
  38. case "reload":
  39. try
  40. {
  41. Game.EventSystem.Add(DLLType.Hotfix, DllHelper.GetHotfixAssembly());
  42. }
  43. catch (Exception e)
  44. {
  45. Console.WriteLine(e);
  46. }
  47. break;
  48. case "repl":
  49. try
  50. {
  51. this.Mode = ConsoleMode.Repl;
  52. this.AddComponent<ReplComponent>();
  53. }
  54. catch (Exception e)
  55. {
  56. Console.WriteLine(e);
  57. }
  58. break;
  59. case "exit":
  60. switch (this.Mode)
  61. {
  62. case ConsoleMode.Repl:
  63. this.RemoveComponent<ReplComponent>();
  64. break;
  65. }
  66. this.Mode = ConsoleMode.None;
  67. break;
  68. case "reset":
  69. switch (this.Mode)
  70. {
  71. case ConsoleMode.Repl:
  72. this.RemoveComponent<ReplComponent>();
  73. this.AddComponent<ReplComponent>();
  74. break;
  75. }
  76. break;
  77. default:
  78. switch (this.Mode)
  79. {
  80. case ConsoleMode.Repl:
  81. {
  82. ReplComponent replComponent = this.GetComponent<ReplComponent>();
  83. if (replComponent == null)
  84. {
  85. Console.WriteLine($"no command: {line}!");
  86. break;
  87. }
  88. try
  89. {
  90. await replComponent.Run(line, this.CancellationTokenSource.Token);
  91. }
  92. catch (Exception e)
  93. {
  94. Console.WriteLine(e);
  95. }
  96. break;
  97. }
  98. }
  99. break;
  100. }
  101. }
  102. catch (Exception e)
  103. {
  104. Console.WriteLine(e);
  105. }
  106. }
  107. }
  108. }
  109. }