Maximilian G.
f1cdfba603
- Implemented `remove_key` function in verify.py to allow key removal by user ID. - Updated run.sh to include additional dependencies for QR code and TOTP support. - Introduced ios_fixes.js for iOS-specific enhancements including file input handling, form submission fixes, and image handling optimizations. - Added mobile_compatibility.js for mobile device detection and utility functions to improve compatibility, especially for iOS. - Created base.html template for consistent layout and styling across the application. - Developed login.html template with a password toggle feature for better user experience.
133 lines
3.7 KiB
Python
133 lines
3.7 KiB
Python
from pathlib import Path
|
|
import sys
|
|
from flask import Flask, render_template, request, jsonify, flash, redirect, url_for, get_flashed_messages, session
|
|
import verify
|
|
import pyotp
|
|
import qrcode
|
|
|
|
def _resolve_template_dir() -> Path:
|
|
candidates = [
|
|
Path(__file__).resolve().parent / "templates",
|
|
Path(sys.argv[0]).resolve().parent / "templates",
|
|
Path.cwd() / "templates",
|
|
]
|
|
|
|
for path in candidates:
|
|
if path.is_dir():
|
|
return path
|
|
|
|
return candidates[0]
|
|
|
|
totp_key = "Hsdfisdf4n34234dfiseLoasjfj3asnnvhxbbfgrzzuewwndcodrweokyn"
|
|
|
|
app = Flask(__name__, template_folder=str(_resolve_template_dir()))
|
|
app.secret_key = "Test123"
|
|
|
|
def _read_app_version() -> str:
|
|
candidates = [
|
|
Path(__file__).resolve().parent / "version.txt",
|
|
Path(sys.argv[0]).resolve().parent / "version.txt",
|
|
Path.cwd() / "version.txt",
|
|
]
|
|
|
|
for path in candidates:
|
|
if path.is_file():
|
|
return path.read_text(encoding="utf-8").strip() or "unknown"
|
|
|
|
return "unknown"
|
|
|
|
def generate_totp_qrcode():
|
|
uri = pyotp.totp.TOTP(totp_key).provisioning_uri(name='',issuer_name='Key Verification Server')
|
|
qrcode.make(uri).save("qr.png")
|
|
|
|
def check_totp(key):
|
|
totp = pyotp.TOTP(totp_key)
|
|
return totp.verify(key)
|
|
|
|
APP_VERSION = _read_app_version()
|
|
|
|
@app.route("/validate__information", methods=['POST'])
|
|
def validate__information():
|
|
data = request.get_json(silent=True)
|
|
if not data:
|
|
return jsonify({"error": "No JSON data provided"}), 400
|
|
license_key = data.get("license")
|
|
hwid_uuid = data.get("hwid")
|
|
if not license_key or not hwid_uuid:
|
|
return jsonify({"error": "Missing 'license' or 'hwid' in JSON data"}), 400
|
|
if verify.check(license_key, hwid_uuid):
|
|
return jsonify({"status": "ok"}), 200
|
|
else:
|
|
return jsonify({"status": "invalid"}), 402
|
|
|
|
@app.route('/login', methods=['GET', 'POST'])
|
|
def login():
|
|
"""
|
|
User login route.
|
|
Authenticates users and redirects to appropriate homepage based on role.
|
|
|
|
Returns:
|
|
flask.Response: Rendered template or redirect
|
|
"""
|
|
if 'username' in session:
|
|
return redirect(url_for('default'))
|
|
if request.method == 'POST':
|
|
totp = request.form['password']
|
|
if not totp:
|
|
flash('Please fill all fields', 'error')
|
|
return redirect(url_for('login'))
|
|
|
|
user_log = check_totp(totp)
|
|
|
|
if user_log:
|
|
session['username'] = "Whatareyoulookingfor"
|
|
return redirect(url_for('default'))
|
|
else:
|
|
flash('Invalid credentials', 'error')
|
|
get_flashed_messages()
|
|
return render_template('login.html')
|
|
|
|
@app.route("/")
|
|
def default():
|
|
if 'username' not in session:
|
|
return redirect(url_for('login'))
|
|
|
|
keys = verify.load_file()
|
|
return render_template("main.html", keys=keys)
|
|
|
|
|
|
@app.route("/generate_new", methods=["POST"])
|
|
def generate_new():
|
|
if 'username' not in session:
|
|
return redirect(url_for('login'))
|
|
|
|
new_license = verify.new_key()
|
|
flash(f"New key generated: {new_license}", "success")
|
|
return redirect(url_for('default'))
|
|
|
|
|
|
@app.route("/remove_key/<user_id>", methods=["POST"])
|
|
def remove_key(user_id):
|
|
if 'username' not in session:
|
|
return redirect(url_for('login'))
|
|
|
|
if verify.remove_key(user_id):
|
|
flash(f"Key for user {user_id} removed", "success")
|
|
else:
|
|
flash(f"No key found for user {user_id}", "error")
|
|
|
|
return redirect(url_for('default'))
|
|
|
|
|
|
@app.route('/logout')
|
|
def logout():
|
|
session.pop('username', None)
|
|
flash('Logged out successfully', 'info')
|
|
return redirect(url_for('login'))
|
|
|
|
|
|
def main():
|
|
app.run(host="0.0.0.0", port=5000, debug=False)
|
|
|
|
if __name__ == "__main__":
|
|
main() |