Add project files.

LeagueAPI/Extensions.cs

@@ -0,0 +1,82 @@
+using System.Diagnostics;
+using System.Management;
+using System.Net.WebSockets;
+using System.Text;
+
+namespace LeagueAPI;
+
+public record WebsocketMessageResult(byte[] Message, WebSocketMessageType MessageType);
+
+public static class Extensions
+{
+    extension(Process process)
+    {
+        public string GetCommandLine()
+        {
+            if (!OperatingSystem.IsWindows())
+            {
+                throw new PlatformNotSupportedException("Only supported on Windows.");
+            }
+            using ManagementObjectSearcher searcher = new("SELECT CommandLine FROM Win32_Process WHERE ProcessId = " + process.Id);
+            using ManagementObjectCollection objects = searcher.Get();
+            return objects.Cast<ManagementBaseObject>().SingleOrDefault()?["CommandLine"]?.ToString() ?? string.Empty;
+        }
+    }
+
+    extension(string? s)
+    {
+        public int ParseInt(int defaultValue = default)
+        {
+            if (s is null || !int.TryParse(s, out int i))
+            {
+                return defaultValue;
+            }
+            return i;
+        }
+
+        public string AsFormatWith(params object?[] args)
+        {
+            if (s is null)
+            {
+                return string.Empty;
+            }
+            return string.Format(s, args);
+        }
+    }
+
+    extension<T>(Task<T> task)
+    {
+        public T WaitForResult()
+        {
+            return task.GetAwaiter().GetResult();
+        }
+    }
+
+    extension(WebSocket socket)
+    {
+        public async Task<WebsocketMessageResult> ReadFullMessage(CancellationToken cancellationToken)
+        {
+            byte[] buffer = new byte[1024 * 8];
+            using MemoryStream memoryStream = new();
+            ValueWebSocketReceiveResult receiveResult;
+
+            do
+            {
+                Memory<byte> memoryBuffer = new(buffer);
+                receiveResult = await socket.ReceiveAsync(memoryBuffer, cancellationToken);
+
+                memoryStream.Write(buffer, 0, receiveResult.Count);
+            }
+            while (!receiveResult.EndOfMessage);
+
+            if (receiveResult.MessageType is WebSocketMessageType.Close)
+            {
+                return new([], receiveResult.MessageType);
+            }
+
+            byte[] fullMessageBytes = memoryStream.ToArray();
+            return new(fullMessageBytes, receiveResult.MessageType);
+        }
+    }
+
+}