TimingsBot/cogs/linking_panel.py


import os

import discord

import requests

import json

import logging

import sys

from discord.ext import commands, tasks

from discord.ext.commands import has_permissions, MissingPermissions

from dotenv import load_dotenv

import aiohttp

import asyncio


class Linking(commands.Cog):


 def __init__(self, bot):

 self.bot = bot

 self.guild_id = int(os.getenv('guild_id'))

 self.crabwings_role_id = int(os.getenv('crabwings_role_id'))

 self.duckfeet_role_id = int(os.getenv('duckfeet_role_id'))

 self.elktail_role_id = int(os.getenv('elktail_role_id'))

 self.client_role_id = int(os.getenv('client_role_id'))

 self.subuser_role_id = int(os.getenv('subuser_role_id'))

 self.verified_role_id = int(os.getenv('verified_role_id'))


 @commands.Cog.listener()

 async def on_message(self, message):

 self.guild = self.bot.get_guild(self.guild_id)

 if message.author != self.bot.user and message.guild is None and self.guild.get_member(message.author.id) is not None:

 if "://" in message.content:

 return

 if self.guild in message.author.guilds:

 channel = message.channel

 await channel.send("Processing, please wait...")

 # Potential API key, so tries it out

 if len(message.content) == 48:

 url = "https://panel.birdflop.com/api/client/account"


 headers = {

 'Accept': 'application/json',

 'Content-Type': 'application/json',

 'Authorization': 'Bearer ' + message.content,

 }


 # response = requests.get(url, headers=headers)


 async with aiohttp.ClientSession() as session:

 async with session.get(url, headers=headers) as response:


 # If API token is verified to be correct:

 if response.status == 200:


 # Formats response of account in JSON format

 json_response = await response.json()


 # Loads contents of users.json

 file = open('users.json', 'r')

 data = json.load(file)

 file.close()


 # Checks if user exists. If so, skips adding them to users.json

 client_id_already_exists = False

 discord_id_already_exists = False

 for user in data['users']:

 if user['client_id'] == json_response['attributes']['id']:

 client_id_already_exists = True

 logging.info("Client ID already exists")

 if user['discord_id'] == message.author.id:

 discord_id_already_exists = True

 logging.info("Discord ID already exists")

 if not client_id_already_exists and not discord_id_already_exists:

 data['users'].append({

 'discord_id': message.author.id,

 'client_id': json_response['attributes']['id'],

 'client_api_key': message.content

 })

 json_dumps = json.dumps(data, indent=2)

 # Adds user to users.json

 file = open('users.json', 'w')

 file.write(json_dumps)

 file.close()


 member = self.guild.get_member(message.author.id)

 if member:


 url = "https://panel.birdflop.com/api/client"


 headers = {

 'Accept': 'application/json',

 'Authorization': 'Bearer ' + message.content,

 }


 async with aiohttp.ClientSession() as session:

 async with session.get(url, headers=headers) as response:


 # If API token is verified to be correct, continues

 if response.status == 200:


 # Formats response for servers in JSON format

 servers_json_response = await response.json()


 user_client = False

 user_subuser = False

 user_crabwings = False

 user_duckfeet = False

 user_elktail = False

 for server in servers_json_response['data']:

 server_owner = server['attributes']['server_owner']

 if server_owner:

 user_client = True

 elif not server_owner:

 user_subuser = True

 server_node = server['attributes']['node']

 if server_node == "Crabwings - NYC":

 user_crabwings = True

 elif server_node == "Duckfeet - EU":

 user_duckfeet = True

 elif server_node == "Elktail - EU":

 user_elktail = True

 if user_client:

 role = discord.utils.get(self.guild.roles, id=self.client_role_id)

 await member.add_roles(role)

 if user_subuser:

 role = discord.utils.get(self.guild.roles, id=self.subuser_role_id)

 await member.add_roles(role)

 if user_crabwings:

 role = discord.utils.get(self.guild.roles, id=self.crabwings_role_id)

 await member.add_roles(role)

 if user_duckfeet:

 role = discord.utils.get(self.guild.roles, id=self.duckfeet_role_id)

 await member.add_roles(role)

 if user_elktail:

 role = discord.utils.get(self.guild.roles, id=self.elktail_role_id)

 await member.add_roles(role)

 role = discord.utils.get(self.guild.roles, id=self.verified_role_id)

 await member.add_roles(role)


 await channel.send(

 'Your Discord account has been linked to your panel account! You may unlink your Discord and panel accounts by reacting in the #verification channel or by deleting your Verification API key.')

 logging.info("Success message sent to " + message.author.name + "#" + str(

 message.author.discriminator) + " (" + str(

 message.author.id) + ")" + ". User linked to API key " + message.content + " and client_id " + str(

 json_response['attributes']['id']))

 elif discord_id_already_exists:

 await channel.send(

 'Sorry, your Discord account is already linked to a panel account. If you would like to link your Discord account to a different panel account, please unlink your Discord account first by reacting in the #verification channel.')

 logging.info("Duplicate Discord message sent to " + message.author.name + "#" + str(

 message.author.discriminator) + " (" + str(

 message.author.id) + ")" + " for using API key " + message.content + " linked to client_id " + str(

 json_response['attributes']['id']))

 elif client_id_already_exists:

 await channel.send(

 'Sorry, your panel account is already linked to a Discord account. If you would like to link your panel account to a different Discord account, please unlink your panel account first by deleting its Verification API key and waiting up to 10 minutes.')

 logging.info("Duplicate panel message sent to " + message.author.name + "#" + str(

 message.author.discriminator) + " (" + str(

 message.author.id) + ")" + " for using API key " + message.content + " linked to client_id " + str(

 json_response['attributes']['id']))

 else:

 # Says if API key is the corect # of characters but invalid

 await channel.send("Sorry, that appears to be an invalid API key.")

 logging.info(

 'invalid sent to ' + message.author.name + "#" + str(

 message.author.discriminator) + " (" + str(

 message.author.id) + ")")

 else:

 # Says this if API key is incorrect # of characters

 await channel.send(

 'Sorry, that doesn\'t appear to be an API token. An API token should be a long string resembling this: ```yQSB12ik6YRcmE4d8tIEj5gkQqDs6jQuZwVOo4ZjSGl28d46```')

 logging.info("obvious incorrect sent to " + message.author.name + "#" + str(

 message.author.discriminator) + " (" + str(message.author.id) + ")")


def setup(bot):

 bot.add_cog(Linking(bot))