looks good

2025-06-05 16:25:45 -05:00
parent 353284b2c3
commit 2415a8146d
3 changed files with 93 additions and 83 deletions
--- a/garfpy/iputils.py
+++ b/garfpy/iputils.py
@ -1,19 +1,64 @@
+import discord
 import ipaddress
+import subprocess
+from garfpy import logger

-def is_private(target):
-    try:
-        ip_obj = ipaddress.ip_address(target)
-        if ip_obj.is_private:
-            return True
-    except ValueError:
-        if "crate.lan" in target.lower():
-            return True
-        if "crate.zip" in target.lower():
-            return True
-        if "memtec.org" in target.lower():
-            return True
-        if "crateit.net" in target.lower():
-            return True
-        if "garfbot.art" in target.lower():
-            return True
-    return False
+
+class IPUtils:    
+    def is_private(self, target):
+        try:
+            ip_obj = ipaddress.ip_address(target)
+            if ip_obj.is_private:
+                return True
+        except ValueError:
+            if "crate.lan" in target.lower():
+                return True
+            if "crate.zip" in target.lower():
+                return True
+            if "memtec.org" in target.lower():
+                return True
+            if "crateit.net" in target.lower():
+                return True
+            if "garfbot.art" in target.lower():
+                return True
+        return False
+
+    async def scan(self, message, user, guild, query):
+        split = query.split()
+        target = split[-1]
+        if self.is_private(target):
+            return
+
+        if query.startswith("garfping "):
+            try:
+                logger.info(f"Ping Request - User: {user}, Server: {guild}, Target: {target}")
+                await message.channel.send(f"`Pinging {target}...`")
+                result = subprocess.run(['ping', '-c', '4', target], capture_output=True, text=True)
+                embed = discord.Embed(title=f"Ping result:", color=0x4d4d4d)
+                embed.add_field(name=target, value=f"```{result.stdout}```")
+                await message.channel.send(embed=embed)
+            except Exception as e:
+                await message.channel.send(f"`GarfBot Error: {str(e)}`")
+
+        if query.startswith("garfdns "):
+            try:
+                logger.info(f"NSLookup Request - User: {user}, Server: {guild}, Target: {target}")
+                await message.channel.send(f"`Requesting {target}...`")
+                result = subprocess.run(['nslookup', target], capture_output=True, text=True)
+                embed = discord.Embed(title=f"NSLookup result:", color=0x4d4d4d)
+                embed.add_field(name=target, value=f"```{result.stdout}```")
+                await message.channel.send(embed=embed)
+            except Exception as e:
+                await message.channel.send(f"`GarfBot Error: {str(e)}`")
+
+        if query.startswith("garfhack "):
+            try:
+                logger.info(f"Nmap Request - User: {user}, Server: {guild}, Target: {target}")
+                await message.channel.send(f"`Scanning {target}...`")         
+                result = subprocess.run(['nmap', '-Pn', '-O', '-v', target], capture_output=True, text=True)
+                embed = discord.Embed(title=f"Nmap scan result:", color=0x4d4d4d)
+                embed.add_field(name=target, value=f"```{result.stdout}```")
+                embed.set_footer(text="https://nmap.org/")
+                await message.channel.send(embed=embed)
+            except Exception as e:
+                await message.channel.send(f"`GarfBot Error: {str(e)}`")