#!/usr/bin/python3
import flask
import threading
import os
import secrets
import subprocess
import hashlib
import base64
import time

http = flask.Flask(__name__)
onemogočeni = set(["onemogočen.py"])
skrivnost = secrets.token_bytes(8)
čas_vzorca = 5
proces = None
naslednji = None

def trenuten():
	if proces:
		return proces.args[1]
	return None

def http_event_stream():
	while sprememba_stanja.wait():
		print("geslo je: " + kode()[0])
		sprememba_stanja.clear()
		yield "data: " + http.json.dumps({"čas": čas_vzorca, "vzorci": os.listdir("vzorci"), "trenuten": trenuten(), "onemogočeni": [x for x in onemogočeni]}) + "\n\n"

@http.route("/vzorci/<vzorec>", methods=["GET"])
def vzorec(vzorec):
	if "/" in vzorec:
		return flask.Response(http.json.dumps({"napaka": {"koda": 3, "besedilo": "vzorec ne sme vsebovati poševnice"}}), status=403)
	return flask.send_file("vzorci/" + vzorec, mimetype="text/x-python")

@http.route("/", methods=["GET"])
def index():
	return flask.render_template("index.html");

@http.route("/stream/", methods=["GET"])
def stream():
	return flask.Response(http_event_stream(), headers={"Access-Control-Allow-Origin": "*", "Content-Type": "text/event-stream"})

@http.route("/<koda>", methods=["POST"])
def update(koda):
	global onemogočeni
	global čas_vzorca
	global naslednji
	if koda not in kode():
		return flask.Response(http.json.dumps({"napaka": {"koda": 1, "besedilo": "napačna koda"}}), status=200)
	data = flask.json.loads(flask.request.data)
	if "omogoči" in data.keys():
		onemogočeni -= set(data["omogoči"])
	if "onemogoči" in data.keys():
		onemogočeni |= set(data["onemogoči"])
	if "čas" in data.keys():
		if data["čas"] < 1 or data["čas"] > 300:
			return flask.Response(http.json.dumps({"napaka": {"koda": 2, "besedilo": "čas ni v [1,300]"}}), status=200)
		čas_vzorca = data["čas"]
	sprememba_stanja.set()
	if "začni" in data.keys() and type(data["začni"]) == str:
		naslednji = data["začni"]
		začni_vzorec.set()
	return "true"

@http.route("/<koda>", methods=["OPTIONS"])
def update_preflight(koda):
	return flask.Response("", status=204, headers={"Access-Control-Allow-Origin": "*", "Access-Control-Allow-Methods": "GET, POST, OPTIONS", "Access-Control-Allow-Headers": "Content-Type", "Access-Control-Max-Age": "86400"})

def kode():
    return [base64.b64encode(hashlib.sha256(skrivnost+bytes(str(int(time.time()/1000)), encoding="utf-8")).digest()).replace(b"/", b"").replace(b"+", b"")[:4].lower().decode(), base64.b64encode(hashlib.sha256(skrivnost+bytes(str(int(time.time()/1000)-1), encoding="utf-8")).digest()).replace(b"/", b"").replace(b"+", b"")[:4].lower().decode(), os.getenv("KODA")]

def zaznaj_smrt():
	proces.wait()
	začni_vzorec.set()

def subprocess_manager():
	global naslednji
	global proces
	global začni_vzorec
	while True:
		if not naslednji:
			vzorci = set(os.listdir("vzorci"))-set(["začasen.py"])-onemogočeni
			naslednji = sorted(vzorci)[0]
		print("daemon: starting " + naslednji)
		proces = subprocess.Popen(["./wrapper.py", naslednji])
		sprememba_stanja.set()
		vzorci = set(os.listdir("vzorci"))-set(["začasen.py"])-onemogočeni
		naslednji = sorted(vzorci)[0]
		if trenuten() in vzorci:
			naslednji = sorted(vzorci)[(sorted(vzorci).index(trenuten())+1)%len(vzorci)]
		začni_vzorec.clear()
		threading.Thread(target=zaznaj_smrt).start()
		razlog = začni_vzorec.wait(timeout=čas_vzorca)
		proces.send_signal(subprocess.signal.SIGINT)
		proces.wait()
	
začni_vzorec = threading.Event()
sprememba_stanja = threading.Event()

if __name__ == "__main__":
	port = 6969
	if os.getenv("PORT"):
		port = int(os.getenv("PORT"))
	threading.Thread(target=lambda: http.run(host="::", port=port, debug=True, use_reloader=False)).start()
	threading.Thread(target=subprocess_manager).start()