import os from dotenv import load_dotenv import discord from discord import app_commands import requests import dns.resolver load_dotenv() 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" logChannel=1139449826287824996 # Commands @tree.command(name="ping", description="Check bot connection") async def ping(ctx): await log("User: " + str(ctx.user.name) + " used the ping command") 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 log("User: " + str(ctx.user.name) + " tried to use the shortlink command") 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 log("User: " + str(ctx.user.name) + " tried to use the botstatus command") 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) @tree.command(name="dig", description="Dig a dns record") async def dig(ctx, domain: str, record_type: str = "A"): record_type = record_type.upper() # Format dig @100.74.29.146 -p 5350 resolver = dns.resolver.Resolver() resolver.nameservers = ["100.74.29.146"] resolver.port = 5350 try: # Query the DNS record response = resolver.resolve(domain, record_type) print(response) records = "" for record in response: print(record) records = records + "\n" + str(record) # Send the result to the Discord channel await ctx.response.send_message(f"DNS records for {domain} ({record_type}):{records}") except dns.resolver.NXDOMAIN: await ctx.response.send_message(f"Domain {domain} not found.") except dns.exception.DNSException as e: await ctx.response.send_message(f"An error occurred: {e}") @tree.command(name="curl", description="HTTP request") async def curl(ctx, url: str): try: proxyURL = "https://proxy.hnsproxy.au" response = requests.get(url, proxies={"http": proxyURL, "https": proxyURL},verify=False) output = response.text if (len(output) > 1900): output = output[:1900] await ctx.response.send_message(f"Response for {url}:\n{output}") except requests.exceptions.RequestException as e: await ctx.response.send_message(f"An error occurred: {e}") @tree.command(name="invite", description="Invite me to your server") async def invite(ctx): await ctx.response.send_message("https://discord.com/api/oauth2/authorize?client_id=1006128164218621972&permissions=0&scope=bot",ephemeral=True) async def log(message): channel=client.get_channel(logChannel) await channel.send(message) def updateStatus(): 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 @client.event async def on_ready(): await tree.sync() print("Ready!") updateStatus() client.run(TOKEN)