# import requests as r
from flask import jsonify, render_template, send_file, redirect
from poll_services import start_async_loop
from mem import services, app, db
import threading
from flask_migrate import upgrade, stamp
from pathlib import Path
from models import service, log
from typing import Any, Optional, cast
import json
from datetime import timedelta


def split_graph(logs: list[log]) -> tuple[list[str], list[Optional[int]]]:
    if len(logs) <= 0:
        return ([], [])

    x = [logs[0].dateCreated.isoformat()]
    y = [logs[0].ping]

    for i in range(1, len(logs)):
        log1 = logs[i]
        log2 = logs[i - 1]

        if (log1.dateCreated - log2.dateCreated) > timedelta(seconds=6):
            x.append(log2.dateCreated.isoformat())
            y.append(None)

        x.append(log1.dateCreated.isoformat())
        y.append(log1.ping)
    return (x, y)


# Init and upgrade
with app.app_context():
    # Check if DB file is missing
    if not (Path("./instance/app.db").is_file()):
        with app.app_context():
            db.create_all()
        stamp()
    # Upgrade db if any new migrations exist
    upgrade()

with app.app_context():
    if not db.session.query(service).first():
        for s in services:
            db.session.add(
                service(
                    url=s.url,
                    label=s.label,
                    public_access=s.public,
                    ping_method=s.ping_type,
                )
            )
        db.session.commit()


@app.route("/")
def homepage():
    return render_template("home.html")


@app.route("/chart")
def chart():
    with app.app_context():
        logs = []
        s = db.session.query(service).first()
        if s:
            logs = cast(
                list[log],
                s.logs.order_by(log.dateCreated.desc())  # type: ignore
                .limit(300)
                .all(),
            )
        else:
            return redirect("/")
    x, y = split_graph(logs=logs)

    return render_template(
        "chart.html",
        dates=x,
        values=json.dumps(y),
    )


@app.route("/api/status")
def status():
    results: list[dict[str, Any]] = []
    with app.app_context():
        a = db.session.query(service).all()
        for s in a:
            b = cast(
                Optional[log],
                s.logs.order_by(
                    log.dateCreated.desc()  # type: ignore
                ).first(),
            )
            if b:
                results.append(s.to_dict() | b.to_dict())

    return jsonify(results)


@app.route("/favicon.svg")
def favicon():
    return send_file("/static/favicon.svg")


# Only run if directly running file
if __name__ == "__main__":
    t = threading.Thread(target=start_async_loop, daemon=True)
    t.start()

    # Run flask app
    app.run(host="0.0.0.0", port=80, debug=True, use_reloader=False)