Web remote power menu

Levminer · Jun 14, 2024 · af313ff · af313ff
1 parent 1a1bb6d
commit af313ff
Show file tree

Hide file tree

Showing 3 changed files with 50 additions and 35 deletions.
diff --git a/packages/ui/components/powerDropdown.svelte b/packages/ui/components/powerDropdown.svelte
@@ -12,7 +12,21 @@
 	>
 		<DropdownMenu.Item
 			on:click={() => {
-				connect("sleep")
+				action("disconnect")
+			}}
+			class="flex cursor-pointer select-none items-center gap-3 rounded-xl p-2 duration-200 ease-in-out data-[highlighted]:bg-gray-600"
+		>
+			<div class="flex items-center">
+				<ScreenShareOff />
+			</div>
+			<div class="flex flex-col items-start">
+				<h4>Disconnect</h4>
+			</div>
+		</DropdownMenu.Item>
+		<DropdownMenu.Separator class="my-1 -ml-1 -mr-1 block h-px bg-gray-600" />
+		<DropdownMenu.Item
+			on:click={() => {
+				action("sleep")
 			}}
 			class="flex cursor-pointer select-none items-center gap-3 rounded-xl p-2 duration-200 ease-in-out data-[highlighted]:bg-gray-600"
 		>
@@ -26,7 +40,7 @@
 		<DropdownMenu.Separator class="my-1 -ml-1 -mr-1 block h-px bg-gray-600" />
 		<DropdownMenu.Item
 			on:click={() => {
-				connect("shutdown")
+				action("shutdown")
 			}}
 			class="flex cursor-pointer select-none items-center gap-3 rounded-xl p-2 duration-200 ease-in-out data-[highlighted]:bg-gray-600"
 		>
@@ -40,7 +54,7 @@
 		<DropdownMenu.Separator class="my-1 -ml-1 -mr-1 block h-px bg-gray-600" />
 		<DropdownMenu.Item
 			on:click={() => {
-				connect("restart")
+				action("restart")
 			}}
 			class="flex cursor-pointer select-none items-center gap-3 rounded-xl p-2 duration-200 ease-in-out data-[highlighted]:bg-gray-600"
 		>
@@ -55,36 +69,9 @@
 </DropdownMenu.Root>
 
 <script lang="ts">
-	import { Avatar, DropdownMenu } from "bits-ui"
+	import { DropdownMenu } from "bits-ui"
 	import { flyAndScale } from "../utils/transitions.ts"
-	import { Moon, Power, RotateCcw, UserCircle } from "lucide-svelte"
-	import { settings } from "ui/stores/settings.ts"
-	import { EzRTCClient } from "ezrtc"
-
-	let connect = (action: string) => {
-		let client: EzRTCClient
-		let sent = false
-
-		client = new EzRTCClient("wss://rtc-usw.levminer.com/one-to-many", $settings.connectionCode, [
-			{
-				urls: "stun:stun.relay.metered.ca:80",
-			},
-			{
-				urls: "turn:standard.relay.metered.ca:80",
-				username: "56feef2e09dcd8d33c5f67eb",
-				credential: "ynk5rIg6gGh4lEAk",
-			},
-		])
-
-		client.onMessage((message) => {
-			if (!sent) {
-				client.dataChannel.send(action)
-				sent = true
-			}
-		})
+	import { Moon, Power, RotateCcw, ScreenShareOff } from "lucide-svelte"
 
-		/* client.peerConnection.onconnectionstatechange = (event) => {
-			console.log("EEEEEEEEVVVVVEEEEENNNNttt")
-		} */
-	}
+	export let action = (type: string) => {}
 </script>
diff --git a/platforms/windows/lib/Interfaces.cs b/platforms/windows/lib/Interfaces.cs
@@ -30,11 +30,11 @@ public API Data {
 }
 
 public class Message {
-	public string Name {
+	public string Type {
 		get; set;
 	}
 
-	public string Content {
+	public string Data {
 		get; set;
 	}
 }

diff --git a/platforms/windows/service/WSServer.cs b/platforms/windows/service/WSServer.cs
@@ -65,6 +65,9 @@ static async Task ProcessWebSocketRequestAsync(WebSocket socket, HardwareInfo ha
 				await socket.CloseAsync(WebSocketCloseStatus.NormalClosure, string.Empty, CancellationToken.None);
 				connectedClients.TryRemove(socket, out _);
 				break;
+			} else if (result.MessageType == WebSocketMessageType.Text) {
+				string receivedText = Encoding.UTF8.GetString(receiveBuffer.Array, receiveBuffer.Offset, result.Count);
+				HandleMessage(receivedText);
 			}
 		}
 
@@ -75,4 +78,29 @@ static async Task ProcessWebSocketRequestAsync(WebSocket socket, HardwareInfo ha
 	public void Stop() {
 		listener.Stop();
 	}
+
+	static void HandleMessage(string message) {
+		try {
+			var netMessage = JsonSerializer.Deserialize<Message>(message, Program.SerializerOptions);
+
+			switch (netMessage?.Data) {
+				case "shutdown":
+					Commands.ExecuteCommand(@"shutdown /s /t 30");
+					break;
+
+				case "sleep":
+					Commands.ExecuteCommand("Start-Process rundll32.exe -ArgumentList 'powrprof.dll,SetSuspendState 0,1,0'");
+					break;
+
+				case "restart":
+					Commands.ExecuteCommand("shutdown /r /t 30");
+					break;
+			}
+
+			return;
+		}
+		catch (Exception) {
+			Console.WriteLine("Invalid message");
+		}
+	}
 }