import logging as log
from zabbix import get_active_problems
from config import conf, icon_dict
from time import sleep
from redis_db import (
    get_all_keys,
    get_value,
    set_value,
    del_value,
)
import asyncio
from telegram import del_message, send_message


async def main_loop():
    active_alerts = get_active_problems()
    telegram_alerts = await get_all_keys()

    new_alerts_id = []
    closed_alerts_id = []

    for active_alert in active_alerts["event_ids"]:
        if active_alert not in telegram_alerts:
            new_alerts_id.append(active_alert)

    for telegram_alert in telegram_alerts:
        if telegram_alert not in active_alerts["event_ids"]:
            closed_alerts_id.append(telegram_alert)

    log.info("new " + str(new_alerts_id))
    log.info("closed " + str(closed_alerts_id))

    for new_alert in new_alerts_id:
        message = (
            icon_dict[active_alerts[new_alert]["severity"]]
            + f"{active_alerts[new_alert]['host']}\n"
            + f"{active_alerts[new_alert]['name']}"
        )
        msg_id = await send_message(message)
        if msg_id["status"] == 200:
            await set_value(key=new_alert, value=msg_id["msg_id"])

    for closed_alert in closed_alerts_id:
        msg_id = await get_value(closed_alert)
        if msg_id:
            resp = await del_message(int(msg_id))
            if resp["status"] == 200:
                await del_value(closed_alert)
            if resp["status"] == 400:
                log.warning(f"remove olg message {msg_id} from reddis")
                await del_value(closed_alert)


if __name__ == "__main__":
    log.info("Starting app")
    try:
        while True:
            asyncio.run(main_loop())
            sleep(conf.zabbix.upd_interval)
    except KeyboardInterrupt:
        log.info("Manual app stopped")
    log.info("App stopped")