initial commit

1 files changed, 82 insertions, 0 deletions

diff --git a/diddlycord.py b/diddlycord.py
new file mode 100644
index 0000000..63e9062
--- /dev/null
+++ b/diddlycord.py
@@ -0,0 +1,82 @@
+import requests
+from dotenv import load_dotenv
+import os
+
+load_dotenv()
+
+WEB_QUERY_API = os.environ.get("WEB_QUERY_URL", "")
+API_KEY = os.environ.get("API_KEY", "")
+
+DISCORD_BOT_TOKEN = os.environ.get("DISCORD_BOT_TOKEN")
+DISCORD_CHANNEL_ID = os.environ.get("DISCORD_CHANNEL_ID")
+
+def fetch_clients():
+    headers = {"x-api-key": API_KEY}
+    try:
+        response = requests.get(f"{WEB_QUERY_API}/1/clientlist", headers=headers, timeout=5)
+        response.raise_for_status()
+        clients = response.json().get('body', [])
+        return [c for c in clients if c.get("client_nickname") != "serveradmin"]
+    except requests.exceptions.RequestException as e:
+        print(f"Couldn't fetch client list: {e}")
+        return []
+
+def get_last_message():
+    url = f"https://discord.com/api/v10/channels/{DISCORD_CHANNEL_ID}/messages?limit=1"
+    headers = {"Authorization": f"Bot {DISCORD_BOT_TOKEN}"}
+    resp = requests.get(url, headers=headers)
+    if resp.ok:
+        messages = resp.json()
+        return messages[0] if messages else None
+    else:
+        print(f"Failed to fetch messages: {resp.status_code} {resp.text}")
+        return None
+
+def edit_or_send_message(clients):
+    client_names = "\n".join(f"- {c.get('client_nickname')}" for c in clients)
+    content = f"\n**Users currently in Teamspeak:**\n{client_names}" if clients else "\nNo users online"
+
+    last_msg = get_last_message()
+    headers = {"Authorization": f"Bot {DISCORD_BOT_TOKEN}", "Content-Type": "application/json"}
+
+    if last_msg:
+        # Edit the last message
+        url = f"https://discord.com/api/v10/channels/{DISCORD_CHANNEL_ID}/messages/{last_msg['id']}"
+        resp = requests.patch(url, headers=headers, json={"content": content})
+        if resp.ok:
+            print("Discord message edited")
+        else:
+            print(f"Failed to edit message: {resp.status_code} {resp.text}")
+    else:
+        # Send a new message
+        url = f"https://discord.com/api/v10/channels/{DISCORD_CHANNEL_ID}/messages"
+        resp = requests.post(url, headers=headers, json={"content": content})
+        if resp.ok:
+            print("Discord message sent")
+        else:
+            print(f"Failed to send message: {resp.status_code} {resp.text}")
+
+def update_discord_channel_name(clients):
+    count = len(clients)
+    new_name = f"teamspeak-{count}"
+
+    url = f"https://discord.com/api/v10/channels/{DISCORD_CHANNEL_ID}"
+    headers = {
+        "Authorization": f"Bot {DISCORD_BOT_TOKEN}",
+        "Content-Type": "application/json"
+    }
+    payload = {"name": new_name}
+
+    resp = requests.patch(url, headers=headers, json=payload)
+    if resp.ok:
+        print(f"Channel renamed to {new_name}")
+    else:
+        print(f"Failed to rename channel: {resp.status_code} {resp.text}")
+
+def main():
+    clients = fetch_clients()
+    edit_or_send_message(clients)
+    update_discord_channel_name(clients)
+
+if __name__ == "__main__":
+    main()