# app.py
import asyncio
import signal
import sys
import time
import uvicorn

from api.health_api import create_health_app
from config.database_config import DatabaseConfig
from db.database import Database
from db.repository import insert_event
from worker.file_watcher import start_observer
from utils.logger import logger

# global flag for shutdown
shutdown_flag = False
observers = []
worker_heartbeat = time.time()


def handle_shutdown(signum, frame):
    global shutdown_flag
    logger.info("🛑 Shutdown signal mottatt, avslutter worker...")
    shutdown_flag = True


def set_heartbeat(ts):
    global worker_heartbeat
    worker_heartbeat = ts


def get_heartbeat():
    return worker_heartbeat


async def run_worker(db: Database, paths, extensions, shutdown_flag_ref):
    global observers
    observers = [start_observer(db, [p], extensions, insert_event) for p in paths]

    try:
        while not shutdown_flag_ref():
            set_heartbeat(time.time())
            await asyncio.sleep(5)
    finally:
        logger.info("🛑 Stopper observer...")
        for obs in observers:
            obs.stop()
            obs.join()
        logger.info("👋 Alle observers stoppet")

    return observers


def main():
    # registrer signal handlers for graceful shutdown
    signal.signal(signal.SIGINT, handle_shutdown)
    signal.signal(signal.SIGTERM, handle_shutdown)

    logger.info("🚀 Starter worker-applikasjon")

    try:
        # DB
        config: DatabaseConfig = DatabaseConfig.from_env()
        db: Database = Database(config)
        db.connect()

        # paths og extensions
        from config.paths_config import PathsConfig
        paths_config = PathsConfig.from_env()
        paths_config.validate()

        # start worker
        loop = asyncio.get_event_loop()
        loop.create_task(run_worker(db, paths_config.watch_paths, paths_config.extensions, lambda: shutdown_flag))

        # health API
        app = create_health_app(
            observers_ref=lambda: observers,
            db_ref=lambda: db,
            heartbeat_ref=get_heartbeat
        )

        uvicorn.run(app, host="0.0.0.0", port=8000)

    except Exception as e:
        logger.error(f"❌ Kritisk feil i app: {e}")
        sys.exit(1)

    logger.info("👋 Worker avsluttet gracefully")


if __name__ == "__main__":
    main()