# -*- coding: utf-8 -*- """ Yuuki_Libs (c) 2019 Star Inc. This Source Code Form is subject to the terms of the Mozilla Public License, v. 2.0. If a copy of the MPL was not distributed with this file, You can obtain one at http://mozilla.org/MPL/2.0/. """ import hashlib import json import random import time from flask import Flask, render_template, Response, request, redirect from flask_bootstrap import Bootstrap from gevent.pywsgi import WSGIServer from .reader import Yuuki_WebDataReader wa_app = Flask(__name__) Yuuki_Handle = None Yuuki_DataHandle = None passports = [] password = str(hash(random.random())) class Yuuki_WebAdmin: def __init__(self, Yuuki): global Yuuki_Handle, Yuuki_DataHandle Yuuki_Handle = Yuuki Yuuki_DataHandle = Yuuki_WebDataReader(Yuuki_Handle.data) self.app = wa_app Bootstrap(self.app) @staticmethod @wa_app.route("/") def index(): if "yuuki_admin" in request.cookies: if request.cookies["yuuki_admin"] in passports: return render_template( 'manage/index.html', version=Yuuki_Handle.YuukiConfigs["version"], LINE_Media_server=Yuuki_Handle.LINE_Media_server, profileName=Yuuki_Handle.profile.displayName, pictureStatus=Yuuki_Handle.profile.pictureStatus ) else: response = redirect("/") response.set_cookie( key='yuuki_admin', value='', expires=0 ) return response else: return render_template( 'index.html', name=Yuuki_Handle.YuukiConfigs["name"], version=Yuuki_Handle.YuukiConfigs["version"] ) @staticmethod @wa_app.route("/groups") def groups(): if "yuuki_admin" in request.cookies: if request.cookies["yuuki_admin"] in passports: return render_template( 'manage/groups.html' ) response = redirect("/") response.set_cookie( key='yuuki_admin', value='', expires=0 ) return response @staticmethod @wa_app.route("/helpers") def helpers(): if "yuuki_admin" in request.cookies: if request.cookies["yuuki_admin"] in passports: return render_template( 'manage/groups.html' ) response = redirect("/") response.set_cookie( key='yuuki_admin', value='', expires=0 ) return response @staticmethod @wa_app.route("/settings") def settings(): if "yuuki_admin" in request.cookies: if request.cookies["yuuki_admin"] in passports: return render_template( 'manage/groups.html' ) response = redirect("/") response.set_cookie( key='yuuki_admin', value='', expires=0 ) return response @staticmethod @wa_app.route("/verify", methods=['GET', 'POST']) def verify(): result = {"status": 403} if request.method == "POST" and "code" in request.values: if request.values["code"] == password: seed = hash(random.random() + time.time()) seed = str(seed).encode('utf-8') session_key = hashlib.sha256(seed).hexdigest() passports.append(session_key) result = {"status": 200, "session": session_key} else: result = {"status": 401} return Response(json.dumps(result), mimetype='application/json') @staticmethod @wa_app.route("/logout") def logout(): response = redirect("/") if "yuuki_admin" in request.cookies: if request.cookies.get("yuuki_admin") in passports: passports.remove(request.cookies.get("yuuki_admin")) response.set_cookie( key='yuuki_admin', value='', expires=0 ) return response @staticmethod @wa_app.route("/api/logs") @wa_app.route("/api/logs/") def logs(name=""): result = {"status": 403} if "yuuki_admin" in request.cookies: if request.cookies.get("yuuki_admin") in passports: if name: result = Yuuki_DataHandle.get_log(name) else: result = Yuuki_DataHandle.get_all_logs() return Response(json.dumps(result), mimetype='application/json') @staticmethod @wa_app.route("/api/i") def i(): return Yuuki_Handle.YuukiConfigs["name"] @staticmethod def set_password(code): global password password = code def start(self, admin_password): self.set_password(admin_password) http_server = WSGIServer(('', 2020), self.app) http_server.serve_forever()