From b1645bc9685e4803db3ee412279f2a1ea3af4a45 Mon Sep 17 00:00:00 2001
From: phixxy <bottlecap@protonmail.com>
Date: Tue, 16 Apr 2024 09:07:05 -0700
Subject: [PATCH] add WOL to allow waking stable diffusion

---
 cogs/wake_on_lan.py | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 cogs/wake_on_lan.py

diff --git a/cogs/wake_on_lan.py b/cogs/wake_on_lan.py
new file mode 100644
index 0000000..0390578
--- /dev/null
+++ b/cogs/wake_on_lan.py
@@ -0,0 +1,33 @@
+import logging
+import os
+import discord
+from discord.ext import commands
+import wakeonlan
+
+
+class WakeOnLan(commands.Cog):
+
+    def __init__(self, bot):
+        self.bot = bot
+        self.admin_id = 242018983241318410
+        self.logger = logging.getLogger("bot")
+        self.mac_address = "9C-6B-00-38-08-8D"
+
+    def wake_on_lan(self, mac_address):
+        self.logger.info(f"WakeOnLan: Waking up {mac_address}")
+        wakeonlan.send_magic_packet(mac_address)
+
+    def sleep_on_lan(self, mac_address):
+        self.logger.info(f"WakeOnLan: Sleeping {mac_address}")
+        wakeonlan.send_magic_packet(mac_address, magic_packet="000000000000")
+
+    @commands.command()
+    async def wake(self, ctx):
+        if ctx.author.id != self.admin_id:
+            return
+        self.wake_on_lan(self.mac_address)
+        await ctx.send(f"Waking up !imagine server")
+
+
+async def setup(bot):
+    await bot.add_cog(WakeOnLan(bot))
\ No newline at end of file