mirror of
https://github.com/TitanEmbeds/Titan.git
synced 2024-11-15 10:22:43 +01:00
96 lines
3.2 KiB
Python
96 lines
3.2 KiB
Python
|
from config import config
|
||
|
|
from requests_oauthlib import OAuth2Session
|
||
|
|
from titanembeds.utils import cache, make_guilds_cache_key
|
||
|
|
from flask import session, abort, url_for
|
||
|
|
|
||
|
|
authorize_url = "https://discordapp.com/api/oauth2/authorize"
|
||
|
|
token_url = "https://discordapp.com/api/oauth2/token"
|
||
|
|
avatar_base_url = "https://cdn.discordapp.com/avatars/"
|
||
|
|
guild_icon_url = "https://cdn.discordapp.com/icons/"
|
||
|
|
|
||
|
|
def update_user_token(discord_token):
|
||
|
|
session['user_keys'] = discord_token
|
||
|
|
|
||
|
|
def make_authenticated_session(token=None, state=None, scope=None):
|
||
|
|
return OAuth2Session(
|
||
|
|
client_id=config['client-id'],
|
||
|
|
token=token,
|
||
|
|
state=state,
|
||
|
|
scope=scope,
|
||
|
|
redirect_uri=url_for("user.callback", _external=True),
|
||
|
|
auto_refresh_kwargs={
|
||
|
|
'client_id': config['client-id'],
|
||
|
|
'client_secret': config['client-secret'],
|
||
|
|
},
|
||
|
|
auto_refresh_url=token_url,
|
||
|
|
token_updater=update_user_token,
|
||
|
|
)
|
||
|
|
|
||
|
|
def discordrest_from_user(endpoint):
|
||
|
|
token = session['user_keys']
|
||
|
|
discord = make_authenticated_session(token=token)
|
||
|
|
req = discord.get("https://discordapp.com/api/v6{}".format(endpoint))
|
||
|
|
return req
|
||
|
|
|
||
|
|
def get_current_authenticated_user():
|
||
|
|
req = discordrest_from_user("/users/@me")
|
||
|
|
if req.status_code != 200:
|
||
|
|
abort(req.status_code)
|
||
|
|
user = req.json()
|
||
|
|
return user
|
||
|
|
|
||
|
|
def user_has_permission(permission, index):
|
||
|
|
return bool((int(permission) >> index) & 1)
|
||
|
|
|
||
|
|
@cache.cached(timeout=120, key_prefix=make_guilds_cache_key)
|
||
|
|
def get_user_guilds():
|
||
|
|
req = discordrest_from_user("/users/@me/guilds")
|
||
|
|
return req
|
||
|
|
|
||
|
|
def get_user_managed_servers():
|
||
|
|
guilds = get_user_guilds()
|
||
|
|
if guilds.status_code != 200:
|
||
|
|
abort(guilds.status_code)
|
||
|
|
guilds = guilds.json()
|
||
|
|
filtered = []
|
||
|
|
for guild in guilds:
|
||
|
|
permission = guild['permissions'] # Manage Server, Ban Members, Kick Members
|
||
|
|
if guild['owner'] or user_has_permission(permission, 5) or user_has_permission(permission, 2) or user_has_permission(permission, 1):
|
||
|
|
filtered.append(guild)
|
||
|
|
filtered = sorted(filtered, key=lambda guild: guild['name'])
|
||
|
|
return filtered
|
||
|
|
|
||
|
|
def get_user_managed_servers_safe():
|
||
|
|
guilds = get_user_managed_servers()
|
||
|
|
if guilds:
|
||
|
|
return guilds
|
||
|
|
return []
|
||
|
|
|
||
|
|
def get_user_managed_servers_id():
|
||
|
|
guilds = get_user_managed_servers_safe()
|
||
|
|
ids=[]
|
||
|
|
for guild in guilds:
|
||
|
|
ids.append(guild['id'])
|
||
|
|
return ids
|
||
|
|
|
||
|
|
def check_user_can_administrate_guild(guild_id):
|
||
|
|
guilds = get_user_managed_servers_id()
|
||
|
|
return guild_id in guilds
|
||
|
|
|
||
|
|
def check_user_permission(guild_id, id):
|
||
|
|
guilds = get_user_managed_servers_safe()
|
||
|
|
for guild in guilds:
|
||
|
|
if guild['id'] == guild_id:
|
||
|
|
return user_has_permission(guild['permissions'], id) or guild['owner']
|
||
|
|
return False
|
||
|
|
|
||
|
|
def generate_avatar_url(id, av):
|
||
|
|
return avatar_base_url + str(id) + '/' + str(av) + '.jpg'
|
||
|
|
|
||
|
|
def generate_guild_icon_url(id, hash):
|
||
|
|
return guild_icon_url + str(id) + "/" + str(hash) + ".jpg"
|
||
|
|
|
||
|
|
def generate_bot_invite_url(guild_id):
|
||
|
|
url = "https://discordapp.com/oauth2/authorize?&client_id={}&scope=bot&permissions={}&guild_id={}&response_type=code&redirect_uri={}".format(config['client-id'], '536083583', guild_id, url_for("user.dashboard", _external=True))
|
||
|
|
return url
|