mirror of
https://github.com/TitanEmbeds/Titan.git
synced 2024-11-15 10:22:43 +01:00
272 lines
11 KiB
Python
272 lines
11 KiB
Python
from titanembeds.database import db, Guilds, UnauthenticatedUsers, UnauthenticatedBans, AuthenticatedUsers
|
|
from titanembeds.decorators import valid_session_required, discord_users_only
|
|
from titanembeds.utils import get_client_ipaddr, discord_api, rate_limiter, channel_ratelimit_key, guild_ratelimit_key, cache, make_guildchannels_cache_key
|
|
from titanembeds.oauth import user_has_permission, generate_avatar_url
|
|
from flask import Blueprint, abort, jsonify, session, request
|
|
from sqlalchemy import and_
|
|
import random
|
|
import requests
|
|
import json
|
|
import datetime
|
|
from config import config
|
|
|
|
api = Blueprint("api", __name__)
|
|
|
|
def user_unauthenticated():
|
|
if 'unauthenticated' in session:
|
|
return session['unauthenticated']
|
|
return True
|
|
|
|
def checkUserRevoke(guild_id, user_key=None):
|
|
revoked = True #guilty until proven not revoked
|
|
if user_unauthenticated():
|
|
dbUser = UnauthenticatedUsers.query.filter(and_(UnauthenticatedUsers.guild_id == guild_id, UnauthenticatedUsers.user_key == user_key)).first()
|
|
revoked = dbUser.isRevoked()
|
|
else:
|
|
banned = checkUserBanned(guild_id)
|
|
if banned:
|
|
return revoked
|
|
member = discord_api.get_guild_member(guild_id, session['user_id'])
|
|
if member['code'] == 200:
|
|
revoked = False
|
|
return revoked
|
|
|
|
def checkUserBanned(guild_id, ip_address=None):
|
|
banned = True
|
|
if user_unauthenticated():
|
|
dbUser = UnauthenticatedBans.query.filter(and_(UnauthenticatedBans.guild_id == guild_id, UnauthenticatedBans.ip_address == ip_address)).all()
|
|
if not dbUser:
|
|
banned = False
|
|
else:
|
|
for usr in dbUser:
|
|
if usr.lifter_id is not None:
|
|
banned = False
|
|
else:
|
|
banned = False
|
|
bans = discord_api.get_guild_bans(guild_id)['content']
|
|
for user in bans:
|
|
if session['user_id'] == user['user']['id']:
|
|
return True
|
|
return banned
|
|
|
|
def check_guild_existance(guild_id):
|
|
dbGuild = Guilds.query.filter_by(guild_id=guild_id).first()
|
|
if not dbGuild:
|
|
return False
|
|
guilds = discord_api.get_all_guilds()
|
|
for guild in guilds:
|
|
if guild_id == guild['id']:
|
|
return True
|
|
return False
|
|
|
|
def update_user_status(guild_id, username, user_key=None):
|
|
if user_unauthenticated():
|
|
ip_address = get_client_ipaddr()
|
|
status = {
|
|
'ip_address': ip_address,
|
|
'username': username,
|
|
'user_key': user_key,
|
|
'guild_id': guild_id,
|
|
'banned': checkUserBanned(guild_id, ip_address),
|
|
'revoked': checkUserRevoke(guild_id, user_key),
|
|
}
|
|
if status['banned'] or status['revoked']:
|
|
return status
|
|
dbUser = UnauthenticatedUsers.query.filter(and_(UnauthenticatedUsers.guild_id == guild_id, UnauthenticatedUsers.user_key == user_key)).first()
|
|
dbUser.bumpTimestamp()
|
|
if dbUser.username != username or dbUser.ip_address != ip_address:
|
|
dbUser.username = username
|
|
dbUser.ip_address = ip_address
|
|
db.session.commit()
|
|
else:
|
|
status = {
|
|
'username': username,
|
|
'guild_id': guild_id,
|
|
'user_id': session['user_id'],
|
|
'banned': checkUserBanned(guild_id),
|
|
'revoked': checkUserRevoke(guild_id)
|
|
}
|
|
if status['banned'] or status['revoked']:
|
|
return status
|
|
dbUser = db.session.query(AuthenticatedUsers).filter(AuthenticatedUsers.guild_id == guild_id, AuthenticatedUsers.client_id == status['user_id']).first()
|
|
dbUser.bumpTimestamp()
|
|
return status
|
|
|
|
def check_user_in_guild(guild_id):
|
|
if user_unauthenticated():
|
|
return guild_id in session['user_keys']
|
|
else:
|
|
return 200 == discord_api.get_guild_member_nocache(guild_id, session['user_id'])['code']
|
|
|
|
@cache.cached(timeout=300, key_prefix=make_guildchannels_cache_key)
|
|
def get_guild_channels(guild_id):
|
|
if user_unauthenticated():
|
|
roles = [guild_id] #equivilant to @everyone role
|
|
else:
|
|
member = discord_api.get_guild_member(guild_id, session['user_id'])['content']
|
|
roles = member['roles']
|
|
guild_channels = discord_api.get_guild_channels(guild_id)['content']
|
|
guild_owner = discord_api.get_guild(guild_id)['content']['owner_id']
|
|
result_channels = []
|
|
for channel in guild_channels:
|
|
if channel['type'] == 0:
|
|
if guild_owner == session['user_id']:
|
|
result_channels.append(channel)
|
|
continue
|
|
if len(channel['permission_overwrites']) == 0:
|
|
result_channels.append(channel)
|
|
else:
|
|
for overwrite in channel['permission_overwrites']:
|
|
if overwrite['type'] == "role" and overwrite['id'] == roles[-1] and not user_has_permission(overwrite['deny'], 10):
|
|
result_channels.append(channel)
|
|
break
|
|
elif overwrite['type'] == "member" and not user_unauthenticated and overwrite['id'] == session['user_id'] and not user_has_permission(overwrite['deny'], 10):
|
|
result_channels.append(channel)
|
|
break
|
|
return result_channels
|
|
|
|
def get_online_discord_users(guild_id):
|
|
embed = discord_api.get_widget(guild_id)
|
|
return embed['members']
|
|
|
|
def get_online_embed_users(guild_id):
|
|
time_past = (datetime.datetime.now() - datetime.timedelta(seconds = 120)).strftime('%Y-%m-%d %H:%M:%S')
|
|
unauths = db.session.query(UnauthenticatedUsers).filter(UnauthenticatedUsers.last_timestamp > time_past, UnauthenticatedUsers.revoked == False).all()
|
|
auths = db.session.query(AuthenticatedUsers).filter(AuthenticatedUsers.last_timestamp > time_past).all()
|
|
users = {'unauthenticated':[], 'authenticated':[]}
|
|
for user in unauths:
|
|
meta = {
|
|
'username': user.username,
|
|
'discriminator': user.discriminator,
|
|
}
|
|
users['unauthenticated'].append(meta)
|
|
for user in auths:
|
|
client_id = user.client_id
|
|
u = discord_api.get_guild_member(guild_id, client_id)['content']['user']
|
|
meta = {
|
|
'username': u['username'],
|
|
'discriminator': u['discriminator'],
|
|
'avatar': generate_avatar_url(u['id'], u['avatar']),
|
|
}
|
|
users['authenticated'].append(meta)
|
|
return users
|
|
|
|
def guild_query_unauth_users_bool(guild_id):
|
|
dbGuild = Guilds.query.filter_by(guild_id=guild_id).first()
|
|
return dbGuild.unauth_users
|
|
|
|
@api.route("/fetch", methods=["GET"])
|
|
@valid_session_required(api=True)
|
|
@rate_limiter.limit("2500/hour")
|
|
@rate_limiter.limit("12/minute", key_func = channel_ratelimit_key)
|
|
def fetch():
|
|
channel_id = request.args.get('channel_id')
|
|
after_snowflake = request.args.get('after', None, type=int)
|
|
if user_unauthenticated():
|
|
key = session['user_keys'][channel_id]
|
|
else:
|
|
key = None
|
|
status = update_user_status(channel_id, session['username'], key)
|
|
if status['banned'] or status['revoked']:
|
|
messages = {}
|
|
status_code = 401
|
|
else:
|
|
messages = discord_api.get_channel_messages(channel_id, after_snowflake)
|
|
status_code = messages['code']
|
|
response = jsonify(messages=messages.get('content', messages), status=status)
|
|
resonse.status_code = status_code
|
|
return response
|
|
|
|
@api.route("/post", methods=["POST"])
|
|
@valid_session_required(api=True)
|
|
@rate_limiter.limit("1200/hour")
|
|
@rate_limiter.limit("6/minute", key_func = channel_ratelimit_key)
|
|
def post():
|
|
channel_id = request.form.get('channel_id')
|
|
content = request.form.get('content')
|
|
if user_unauthenticated():
|
|
key = session['user_keys'][channel_id]
|
|
else:
|
|
key = None
|
|
status = update_user_status(channel_id, session['username'], key)
|
|
if status['banned'] or status['revoked']:
|
|
message = {}
|
|
status_code = 401
|
|
else:
|
|
message = discord_api.create_message(channel_id, content)
|
|
status_code = messages['code']
|
|
response = jsonify(message=message.get('content', message), status=status)
|
|
response.status_code = status_code
|
|
return response
|
|
|
|
@api.route("/create_unauthenticated_user", methods=["POST"])
|
|
@rate_limiter.limit("4/hour", key_func=guild_ratelimit_key)
|
|
def create_unauthenticated_user():
|
|
session['unauthenticated'] = True
|
|
username = request.form['username']
|
|
guild_id = request.form['guild_id']
|
|
ip_address = get_client_ipaddr()
|
|
if not check_guild_existance(guild_id):
|
|
abort(404)
|
|
if not guild_query_unauth_users_bool(guild_id):
|
|
abort(401)
|
|
if not checkUserBanned(guild_id, ip_address):
|
|
session['username'] = username
|
|
if 'user_id' not in session:
|
|
session['user_id'] = random.randint(0,9999)
|
|
user = UnauthenticatedUsers(guild_id, username, session['user_id'], ip_address)
|
|
db.session.add(user)
|
|
db.session.commit()
|
|
key = user.user_key
|
|
if 'user_keys' not in session:
|
|
session['user_keys'] = {guild_id: key}
|
|
else:
|
|
session['user_keys'][guild_id] = key
|
|
status = update_user_status(guild_id, username, key)
|
|
return jsonify(status=status)
|
|
else:
|
|
status = {'banned': True}
|
|
response = jsonify(status=status)
|
|
response.status_code = 403
|
|
return response
|
|
|
|
@api.route("/query_guild", methods=["GET"])
|
|
@valid_session_required(api=True)
|
|
def query_guild():
|
|
guild_id = request.args.get('guild_id')
|
|
if check_guild_existance(guild_id):
|
|
if check_user_in_guild(guild_id):
|
|
channels = get_guild_channels(guild_id)
|
|
discordmembers = get_online_discord_users(guild_id)
|
|
embedmembers = get_online_embed_users(guild_id)
|
|
return jsonify(channels=channels, discordmembers=discordmembers, embedmembers=embedmembers)
|
|
abort(403)
|
|
abort(404)
|
|
|
|
@api.route("/create_authenticated_user", methods=["POST"])
|
|
@discord_users_only(api=True)
|
|
def create_authenticated_user():
|
|
guild_id = request.form.get('guild_id')
|
|
if session['unauthenticated']:
|
|
response = jsonify(error=True)
|
|
response.status_code = 401
|
|
return response
|
|
else:
|
|
if not check_guild_existance(guild_id):
|
|
abort(404)
|
|
if not checkUserBanned(guild_id):
|
|
db_user = db.session.query(AuthenticatedUsers).filter(AuthenticatedUsers.guild_id == guild_id, AuthenticatedUsers.client_id == session['user_id']).first()
|
|
if not db_user:
|
|
db_user = AuthenticatedUsers(guild_id, session['user_id'])
|
|
db.session.add(db_user)
|
|
db.session.commit()
|
|
if not check_user_in_guild(guild_id):
|
|
discord_api.add_guild_member(guild_id, session['user_id'], session['user_keys']['access_token'])
|
|
status = update_user_status(guild_id, session['username'])
|
|
return jsonify(error=False)
|
|
else:
|
|
status = {'banned': True}
|
|
response = jsonify(status=status)
|
|
response.status_code = 403
|
|
return response
|