You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

Program.cs 4.2 KiB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135
  1. using System;
  2. using System.Collections.Generic;
  3. using System.IO;
  4. using System.Linq;
  5. using System.Reflection;
  6. using System.Threading.Tasks;
  7. using Microsoft.CodeAnalysis.CSharp.Scripting;
  8. using Microsoft.CodeAnalysis.Scripting;
  9. using Discord;
  10. using Discord.WebSocket;
  11. using System.Collections.Concurrent;
  12. using System.Threading;
  13. using System.Text;
  14. using System.Diagnostics;
  15. namespace idn
  16. {
  17. public class Program
  18. {
  19. public static readonly string[] Imports =
  20. {
  21. "System",
  22. "System.Collections.Generic",
  23. "System.Linq",
  24. "System.Threading.Tasks",
  25. "System.Diagnostics",
  26. "System.IO",
  27. "Discord",
  28. "Discord.Rest",
  29. "Discord.WebSocket",
  30. "idn"
  31. };
  32. static async Task Main(string[] args)
  33. {
  34. var token = File.ReadAllText("token.ignore");
  35. var client = new DiscordSocketClient(new DiscordSocketConfig { LogLevel = LogSeverity.Debug });
  36. var logQueue = new ConcurrentQueue<LogMessage>();
  37. var logCancelToken = new CancellationTokenSource();
  38. client.Log += msg =>
  39. {
  40. logQueue.Enqueue(msg);
  41. return Task.CompletedTask;
  42. };
  43. var logTask = Task.Run(async () =>
  44. {
  45. var fs = new FileStream("idn.log", FileMode.Append);
  46. //var f = File.Open("idn.log", FileMode.Append);
  47. StringBuilder logStringBuilder = new StringBuilder(200);
  48. string logString = "";
  49. byte[] helloBytes = Encoding.UTF8.GetBytes($"### new log session: {DateTime.Now} ###\n\n");
  50. await fs.WriteAsync(helloBytes);
  51. while (!logCancelToken.IsCancellationRequested)
  52. {
  53. if (logQueue.TryDequeue(out var msg))
  54. {
  55. _ = msg.ToString(builder: logStringBuilder);
  56. logStringBuilder.AppendLine();
  57. logString = logStringBuilder.ToString();
  58. Debug.Write(logString, "DNET");
  59. await fs.WriteAsync(Encoding.UTF8.GetBytes(logString), logCancelToken.Token);
  60. }
  61. await fs.FlushAsync();
  62. await Task.Delay(100, logCancelToken.Token);
  63. }
  64. byte[] goodbyeBytes = Encoding.UTF8.GetBytes($"#!! end log session: {DateTime.Now} !!#\n\n\n");
  65. await fs.WriteAsync(goodbyeBytes);
  66. await fs.DisposeAsync();
  67. });
  68. await client.LoginAsync(TokenType.Bot, token);
  69. await client.StartAsync();
  70. var options = ScriptOptions.Default
  71. .AddReferences(GetAssemblies().ToArray())
  72. .AddImports(Imports);
  73. var globals = new ScriptGlobals
  74. {
  75. Client = client,
  76. };
  77. while (true)
  78. {
  79. Console.Write("> ");
  80. string input = Console.ReadLine();
  81. if (input == "quit!")
  82. {
  83. break;
  84. }
  85. object eval;
  86. try
  87. {
  88. eval = await CSharpScript.EvaluateAsync(input, options, globals);
  89. }
  90. catch (Exception e)
  91. {
  92. eval = e;
  93. }
  94. Console.WriteLine(Inspector.Inspect(eval));
  95. }
  96. await client.StopAsync();
  97. client.Dispose();
  98. logCancelToken.Cancel();
  99. try
  100. { await logTask; }
  101. finally { Console.WriteLine("goodbye!"); }
  102. }
  103. static IEnumerable<Assembly> GetAssemblies()
  104. {
  105. var Assemblies = Assembly.GetEntryAssembly().GetReferencedAssemblies();
  106. foreach (var a in Assemblies)
  107. {
  108. var asm = Assembly.Load(a);
  109. yield return asm;
  110. }
  111. yield return Assembly.GetEntryAssembly();
  112. }
  113. public class ScriptGlobals
  114. {
  115. public DiscordSocketClient Client { get; set; }
  116. }
  117. }
  118. }