From 9db2a269a46a7ae3673fa0b4ee7f2dbf8f6306fa Mon Sep 17 00:00:00 2001
From: Nathan Woodburn <github@nathan.woodburn.au>
Date: Fri, 18 Jul 2025 12:26:57 +1000
Subject: [PATCH] test: Initial domain TLD info

---
 shakestation.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/shakestation.py b/shakestation.py
index ea1d32c..346524b 100644
--- a/shakestation.py
+++ b/shakestation.py
@@ -33,7 +33,7 @@ functions = {
         "returns": {
             "info": 
             {
-                "name": "Information on the domain",
+                "name": "Shakestation information",
                 "type": "text"
             }
         }
@@ -48,5 +48,13 @@ def domain_info(params, authentication):
     if not domain:
         return {"info": "No domain provided"}
     
+    response = requests.post(f"https://shakestation.com/api",
+                             json={"action":"getTLD","tld":domain})
+    if response.status_code != 200:
+        return {"info": "Failed to fetch data from Shakestation"}
+    data = response.json()
+    return {"info": data}
+    
+
 
     return {"info": "TODO"}
\ No newline at end of file