from flask import Flask, request, make_response, render_template, send_from_directory, url_for
from flask_httpauth import HTTPBasicAuth
from configparser import ConfigParser
import os
import sys
import time

class AdminEntry:
    def __init__(self, name, sdserial, nandserial, twlnserial, secinfoserial, date):
        self.name = name
        self.retrieveurl = url_for('retrieve', username=name)
        self.deleteurl = url_for('delete', username=name)
        self.sdserial = sdserial
        self.nandserial = nandserial
        self.twlnserial = twlnserial
        self.secinfoserial = secinfoserial
        self.date = date

def get_essential_list():
    essential_entries = []
    for file in os.listdir("essentials"):
        if not file.endswith(".txt"):
            try:
                with open(f"essentials/{file}.serials.txt", "r") as f:
                    arr = f.readlines()
                    essential_entries.append(AdminEntry(file[10:].split('.exefs')[0], arr[0], arr[1], arr[2], arr[3], arr[4]))
            except FileNotFoundError:
                essential_entries.append(AdminEntry(file[10:].split('.exefs')[0], "", "", "", "", ""))
    return essential_entries

if not "config.ini" in os.listdir():
    print("Configuration doesn't exist")
    sys.exit(4) 

config = ConfigParser()
config.read("config.ini")
config_username = config["Access_Credentials"]["username"]
config_password = config["Access_Credentials"]["password"]
if config_password == "<password here>":
    print("Password not set (too dangerous)")
    sys.exit(4)

if not "essentials" in os.listdir():
    print("Create an \"essentials\" folder")
    sys.exit(4)

app = Flask(__name__)
auth = HTTPBasicAuth()

@app.route('/favicon.ico')
def favicon():
    return send_from_directory(os.path.join(app.root_path, 'static'),
                               'favicon.ico', mimetype='image/vnd.microsoft.icon')

@auth.verify_password
def verify_password(username, password):
    if config_password == password and config_username == username:
        return username

@app.route('/submit', methods=['POST'])
def get_submission():
    curtime = int(time.time())
    print(f"essential.exefs submitted by {request.form['discordhandle']}")
    with open(f"essentials/essential_{request.form['discordhandle']}.exefs", "wb") as f:
        f.write(request.files['file'].read())

    with open(f"essentials/essential_{request.form['discordhandle']}.exefs.serials.txt", "w") as f:
        f.write(f"{request.form['sd']}\n{request.form['nand']}\n{request.form['twln']}\n{request.form['secinfo']}\n{curtime}")

    return "<p>Hi</p>"

@app.route('/')
def index():
    return render_template('index.html')

@app.route('/admin')
@auth.login_required
def admin():
    return render_template('admin.html', entries=get_essential_list())

@app.route('/retrieve', methods=['GET'])
@auth.login_required
def retrieve():
    name = request.args['username']
    print(f"{name} retrieved by {request.remote_addr}")
    with open(f"essentials/essential_{name}.exefs", "rb") as f:
        essential = f.read()
    response = make_response(essential)
    response.headers.set('Content-Type', 'application/octet-stream')
    response.headers.set('Content-Disposition', 'attachment', filename=f'essential_{name}.exefs')
    return response

@app.route('/delete')
@auth.login_required
def delete():
    name = request.args['username']
    print(f"{name} deleted by {request.remote_addr}")
    os.remove(f"essentials/essential_{name}.exefs")
    try: 
        os.remove(f"essentials/essential_{name}.exefs.serials.txt")
    except FileNotFoundError:
        pass
    return "Deleted successfully. <a href=\"/admin\">Return to Admin Panel</a>"