From 4fb98b6f599848c1ea89f7f5cc337555e2946c63 Mon Sep 17 00:00:00 2001
From: Nathan Woodburn <github@nathan.woodburn.au>
Date: Fri, 11 Aug 2023 16:35:36 +1000
Subject: [PATCH] bot: Added commands

---
 bot.py | 54 ++++++++++++++++++++++++++++++++++++++++++++++--------
 1 file changed, 46 insertions(+), 8 deletions(-)

diff --git a/bot.py b/bot.py
index 8748b81..5f7808c 100644
--- a/bot.py
+++ b/bot.py
@@ -2,6 +2,7 @@ import os
 from dotenv import load_dotenv
 import discord
 from discord import app_commands
+import requests
 
 
 load_dotenv()
@@ -10,24 +11,61 @@ TOKEN = os.getenv('DISCORD_TOKEN')
 intents = discord.Intents.default()
 client = discord.Client(intents=intents)
 tree = app_commands.CommandTree(client)
+activityMessage="over the server"
+statusType="watching"
 
 # Commands
 @tree.command(name="ping", description="Check bot connection")
 async def ping(ctx):
     await ctx.response.send_message("Pong!",ephemeral=True)
 
+@tree.command(name="shortlink", description="Shorten a link")
+async def shortlink(ctx, link: str, name: str = None):
+    if (ctx.user.id != 892672018917519370):
+        await ctx.response.send_message("You don't have permission to use this command",ephemeral=True)
+    else:
+        APIKEY=os.getenv('LINK_API_KEY')
+        url="https://l.woodburn.au/api/v2/links"
+        headers = {'X-API-KEY' : APIKEY}
 
+        data = {'target' : link, 'customurl' : name}
+        if (name == None):
+            data = {'target' : link}    
+        x = requests.post(url, data = data, headers = headers)
+        print(x.text)
+        if (x.status_code != 200 and x.status_code != 201):
+            await ctx.response.send_message("ERROR: " + x.text,ephemeral=True)
+        link=x.json()['link']
+        await ctx.response.send_message("Link: " + link,ephemeral=False)
 
-
-
-
-
-
-
-
+@tree.command(name="botstatus", description="Set the bot status")
+async def botstatus(ctx, message: str, statusmethod: str = None):
+    if (ctx.user.id != 892672018917519370):
+        await ctx.response.send_message("You don't have permission to use this command",ephemeral=True)
+    else:
+        global activityMessage
+        activityMessage=message
+        global statusType
+        if (statusmethod == None):
+            statusmethod="watching"
+        else:
+            statusType=statusmethod.lower()
+        updateStatus()
+        await ctx.response.send_message("Status updated",ephemeral=True)
 
 def updateStatus():
-    activity = discord.Activity(name="Nathan try to get set me up", type=discord.ActivityType.watching)
+    global activityMessage
+    global statusType
+    if (statusType == "watching"):
+        activity=discord.Activity(type=discord.ActivityType.watching, name=activityMessage)
+    elif (statusType == "playing"):
+        activity=discord.Activity(type=discord.ActivityType.playing, name=activityMessage)
+    elif (statusType == "listening"):
+        activity=discord.Activity(type=discord.ActivityType.listening, name=activityMessage)
+    elif (statusType == "competing"):
+        activity=discord.Activity(type=discord.ActivityType.competing, name=activityMessage)
+    else:
+        activity=discord.Activity(type=discord.ActivityType.watching, name=activityMessage)
     client.loop.create_task(client.change_presence(activity=activity))
 
 # When the bot is ready