forked from elliot/TimingsBot
DrBot7
4 years ago
5 changed files with 45 additions and 375 deletions
@ -1,174 +0,0 @@ |
|||||
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)) |
|
||||
@ -1,115 +0,0 @@ |
|||||
import os |
|
||||
import discord |
|
||||
import requests |
|
||||
import json |
|
||||
import logging |
|
||||
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_updater(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')) |
|
||||
|
|
||||
@tasks.loop(minutes=10) |
|
||||
async def linking_updater(self): |
|
||||
logging.info("Synchronizing roles") |
|
||||
file = open('users.json', 'r') |
|
||||
data = json.load(file) |
|
||||
file.close() |
|
||||
guild = self.bot.get_guild(self.guild_id) |
|
||||
i = -1 |
|
||||
for client in data['users']: |
|
||||
i += 1 |
|
||||
member = guild.get_member(client['discord_id']) |
|
||||
if member: |
|
||||
api_key = client['client_api_key'] |
|
||||
url = "https://panel.birdflop.com/api/client" |
|
||||
headers = { |
|
||||
'Accept': 'application/json', |
|
||||
'Authorization': 'Bearer ' + api_key, |
|
||||
} |
|
||||
async with aiohttp.ClientSession() as session: |
|
||||
async with session.get(url, headers=headers) as response: |
|
||||
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 == True: |
|
||||
user_client = True |
|
||||
elif server_owner == False: |
|
||||
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 |
|
||||
role = discord.utils.get(guild.roles, id=self.client_role_id) |
|
||||
if user_client == True: |
|
||||
await member.add_roles(role) |
|
||||
else: |
|
||||
await member.remove_roles(role) |
|
||||
role = discord.utils.get(guild.roles, id=self.subuser_role_id) |
|
||||
if user_subuser == True: |
|
||||
await member.add_roles(role) |
|
||||
else: |
|
||||
await member.remove_roles(role) |
|
||||
role = discord.utils.get(guild.roles, id=self.crabwings_role_id) |
|
||||
if user_crabwings == True: |
|
||||
await member.add_roles(role) |
|
||||
else: |
|
||||
await member.remove_roles(role) |
|
||||
role = discord.utils.get(guild.roles, id=self.duckfeet_role_id) |
|
||||
if user_duckfeet == True: |
|
||||
await member.add_roles(role) |
|
||||
else: |
|
||||
await member.remove_roles(role) |
|
||||
role = discord.utils.get(guild.roles, id=self.elktail_role_id) |
|
||||
if user_elktail == True: |
|
||||
await member.add_roles(role) |
|
||||
else: |
|
||||
await member.remove_roles(role) |
|
||||
else: |
|
||||
data['users'].pop(i) |
|
||||
json_dumps = json.dumps(data, indent=2) |
|
||||
file = open('users.json', 'w') |
|
||||
file.write(json_dumps) |
|
||||
file.close() |
|
||||
await member.edit(roles=[]) |
|
||||
logging.info("removed discord_id " + str(client['discord_id']) + " with client_id " + str( |
|
||||
client['client_id']) + " and INVALID client_api_key " + client['client_api_key']) |
|
||||
else: |
|
||||
data['users'].pop(i) |
|
||||
json_dumps = json.dumps(data, indent=2) |
|
||||
file = open('users.json', 'w') |
|
||||
file.write(json_dumps) |
|
||||
file.close() |
|
||||
logging.info("removed discord_id " + str(client['discord_id']) + " with client_id " + str( |
|
||||
client['client_id']) + " and client_api_key " + client['client_api_key']) |
|
||||
|
|
||||
@linking_updater.before_loop |
|
||||
async def before_linking_updater(self): |
|
||||
logging.info('waiting to enter loop') |
|
||||
await self.bot.wait_until_ready() |
|
||||
|
|
||||
def setup(bot): |
|
||||
bot.add_cog(Linking_updater(bot)) |
|
||||
Loading…
Reference in new issue