from starlette.applications import Starlette
from starlette.exceptions import HTTPException
from starlette.requests import Request
from starlette.responses import JSONResponse, HTMLResponse
from starlette.staticfiles import StaticFiles
from starlette.routing import Route, Mount
from starlette.templating import Jinja2Templates
import uvicorn
import psutil
from starlette_exporter import PrometheusMiddleware, handle_metrics
from prometheus_client import Summary, Gauge
import background_loop
from typing import List, Optional
import logging
import socket
import sys
if sys.implementation._multiarch == 'darwin':
    import idle_osx as idle
elif 'linux' in sys.implementation._multiarch:
    import idle_linux as idle
else:
    raise NotImplementedError(repr(sys.implementation))

hostname = socket.gethostname().split('.')[0]
logging.basicConfig(level=logging.INFO)
log = logging.getLogger()


def progname(cmdline: List[str]) -> Optional[str]:
    if len(cmdline) < 1:
        return None
    if cmdline[-1].endswith('/steam'):
        return 'steam'
    if cmdline[0].endswith('/minecraft-launcher'):
        return 'minecraft-launcher'
    if cmdline[0].endswith('/java') and '--versionType' in cmdline:
        return 'minecraft'


RACC_RUNNING = Gauge("racc_running", "program is running", ['host', 'prog'])
RACC_IDLE = Gauge("racc_idle", "desktop mouse/kb idle seconds", ['host'])


def update_progs(first_run):
    out = []
    progs = set()
    for proc in psutil.process_iter(['pid', 'name']):
        try:
            prog = progname(proc.cmdline())
            if prog:
                progs.add(prog)
        except (psutil.AccessDenied, psutil.NoSuchProcess):
            pass

    for p in [
            'minecraft',
            'minecraft-launcher',
            'steam',
    ]:
        RACC_RUNNING.labels(host=hostname, prog=p).set(p in progs)
    RACC_IDLE.labels(host=hostname).set(idle.get_idle_seconds())


def root(req: Request) -> HTMLResponse:
    return HTMLResponse(f'''controls for {hostname}''')


def main():
    app = Starlette(debug=True,
                    routes=[
                        Route('/', root),
                    ],
                    on_startup=[
                        lambda: background_loop.loop_forever(update_progs, 3),
                    ])

    app.add_middleware(PrometheusMiddleware, app_name='racc')
    app.add_route("/metrics", handle_metrics)
    return app


if __name__ == "__main__":
    uvicorn.run(main(), host='0.0.0.0', port=5150)