import json
from os.path import exists

from sqlmodel import Session

from app.db import engine
from app.writer.common import load_jira_issue, jira_issue_to_support_ticket
from app.models import JiraSite
from app.models.jira_site import JiraSiteConfig
from app.models.support_ticket import SupportTicket


def get_support_ticket_for_test(
    jira_site_id: int, jira_issue_key: str
) -> SupportTicket:
    with Session(engine) as session:
        jira_site = session.get(JiraSite, jira_site_id)
        assert jira_site is not None

        filepath = f"../data/support_tickets/{jira_issue_key}.json"
        if exists(filepath):
            with open(filepath, "r") as file:
                d = json.load(file)
                return SupportTicket.model_validate(d)

        jira_site_config = JiraSiteConfig.model_validate(jira_site.config)
        jira_issue = load_jira_issue(
            session, jira_site_id, jira_issue_key, jira_site_config
        )

        support_ticket = jira_issue_to_support_ticket(jira_issue, jira_site_config)
        with open(filepath, "w") as f:
            j = support_ticket.model_dump_json(indent=4)
            print(j, file=f)
            return support_ticket