import json
import sys
from typing import List

ALERT_PREFIX = "Elementary: "


def get_elementary_log_lines(dbt_log: str):
    return [line for line in dbt_log.splitlines() if ALERT_PREFIX in line]


def get_json_logs(log_lines: List[str]):
    json_logs = []
    for log in log_lines:
        try:
            json_logs.append(json.loads(log))
        except json.JSONDecodeError:
            pass
    return json_logs


def get_elementary_alerts(elementary_json_logs: List[dict]):
    elementary_alerts = []
    for json_log in elementary_json_logs:
        elementary_alerts.extend(
            json.loads("".join(json_log["data"]["msg"].split(ALERT_PREFIX, 1)[1:]))
        )
    return elementary_alerts


def main():
    with open(sys.argv[1], encoding="utf-8") as f:
        dbt_log = f.read()
    elementary_log_lines = get_elementary_log_lines(dbt_log)
    elementary_json_logs = get_json_logs(elementary_log_lines)
    elementary_alerts = get_elementary_alerts(elementary_json_logs)
    with open("elementary_alerts.json", "w", encoding="utf-8") as f:
        f.write(json.dumps(elementary_alerts))


if __name__ == "__main__":
    main()