89 lines
3.0 KiB
Python
89 lines
3.0 KiB
Python
# Copyright (c) 2021 iliana etaoin
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify it under
|
|
# the terms of the GNU Affero General Public License as published by the Free
|
|
# Software Foundation, either version 3 of the License, or (at your option) any
|
|
# later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful, but WITHOUT
|
|
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
|
# FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
|
|
# details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
import botocore.session
|
|
import operator
|
|
import requests
|
|
import serverless_wsgi
|
|
import urllib.parse
|
|
|
|
from flask import Flask, redirect, render_template, request, url_for
|
|
|
|
app = Flask(__name__)
|
|
|
|
class ForbiddenEndpointError(Exception):
|
|
pass
|
|
|
|
@app.route('/<domain>')
|
|
def emojo(domain):
|
|
if request.args.get('show_all', '') == 'on':
|
|
show_all = True
|
|
else:
|
|
show_all = False
|
|
if request.args.get('show_animated', '') == 'on':
|
|
show_animated = True
|
|
else:
|
|
show_animated = False
|
|
|
|
try:
|
|
url = urllib.parse.urlunsplit(('https', domain, '/api/v1/custom_emojis', '', ''))
|
|
response = requests.get(url)
|
|
if response.status_code == 401:
|
|
raise ForbiddenEndpointError
|
|
|
|
if show_all:
|
|
emojo = sorted(response.json(), key=operator.itemgetter('shortcode'))
|
|
else:
|
|
emojo = sorted(filter(lambda x: x.get('visible_in_picker', True), response.json()),
|
|
key=operator.itemgetter('shortcode'))
|
|
return render_template('emojo.html', domain=domain, emojo=emojo, show_animated=show_animated)
|
|
except requests.exceptions.RequestException as e:
|
|
return render_template('oh_no.html', domain=domain)
|
|
except ForbiddenEndpointError:
|
|
return render_template('forbidden.html', domain=domain)
|
|
|
|
|
|
@app.route('/favicon.ico')
|
|
@app.route('/robots.txt')
|
|
def no_content():
|
|
return ('', 204)
|
|
|
|
|
|
@app.route('/code')
|
|
def code():
|
|
context = request.environ.get('context')
|
|
session = botocore.session.get_session()
|
|
# region name is detected from lambda environment
|
|
client = session.create_client('lambda')
|
|
code = client.get_function(FunctionName=context.function_name, Qualifier=context.function_version)
|
|
return redirect(code['Code']['Location'], code=303)
|
|
|
|
|
|
@app.route('/', methods=('GET', 'POST'))
|
|
def index():
|
|
if request.method == 'POST':
|
|
if 'instance' in request.form:
|
|
show_all = request.form.get('show_all')
|
|
show_animated = request.form.get('show_animated')
|
|
return redirect(
|
|
url_for('emojo', domain=request.form['instance'], show_all=show_all, show_animated=show_animated))
|
|
else:
|
|
return redirect(url_for('index'))
|
|
else:
|
|
return render_template('index.html')
|
|
|
|
def handle_request(event, context):
|
|
return serverless_wsgi.handle_request(app, event, context)
|