test: Initial domain TLD info

2025-07-18 12:26:57 +10:00
parent 26e77913fb
commit 9db2a269a4
1 changed files with 9 additions and 1 deletions
--- 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"}