import logging
import re
from datetime import date, datetime, time, timezone

from common import dates, endoflife, http
from common.releasedata import ProductData, parse_argv

"""Detect new models and aggregate EOL data for Samsung Mobile devices.

This script works cumulatively: when a model is not listed anymore on https://security.samsungmobile.com/workScope.smsb
it retains the date and use it as the model's EOL date.
"""

TODAY = dates.today_at_midnight()

frontmatter, config = parse_argv()
with ProductData(config.product) as product_data:
    frontmatter_release_names = frontmatter.get_release_names()

    # Copy EOL dates from frontmatter to product data
    for frontmatter_release in frontmatter.get_releases():
        eol = frontmatter_release.get("eol")
        eol = datetime.combine(eol, time.min, tzinfo=timezone.utc) if isinstance(eol, date) else eol

        release = product_data.get_release(frontmatter_release.get("releaseCycle"))
        release.set_eol(eol)


    html = http.fetch_html(config.url)

    sections = config.data.get("sections", {})
    for update_cadence, title in sections.items():
        models_list = html.find(string=lambda text, search=title: search in text if text else False).find_next("ul")

        for item in models_list.find_all("li"):
            models = item.text.replace("Enterprise Models:", "")
            logging.info(f"Found {models} for {update_cadence} security updates")

            for model in re.split(r',\s*', models):
                name = endoflife.to_identifier(model)
                if config.is_excluded(name):
                    logging.debug(f"Ignoring model '{name}', excluded by configuration")
                    continue

                release = product_data.get_release(name)
                release.set_label(model.strip())

                if name in frontmatter_release_names:
                    frontmatter_release_names.remove(name)
                    current_eol = release.get_eol()
                    if current_eol is True or (isinstance(current_eol, datetime) and current_eol <= TODAY):
                        logging.info(f"Known model {name} is incorrectly marked as EOL, updating eol")
                        release.set_eol(False)
                    else:
                        logging.debug(f"Known model {name} is not EOL, keeping eol as {current_eol}")

                else:
                    logging.debug(f"Found new model {name}")
                    release.set_eol(False)

    # the remaining models in frontmatter_release_names are not listed anymore on the Samsung page => they are EOL
    for eol_model_name in frontmatter_release_names:
        release = product_data.get_release(eol_model_name)
        current_eol = release.get_eol()
        if config.is_excluded(eol_model_name):
            logging.debug(f"Skipping model {eol_model_name}, excluded by configuration")
        elif current_eol is False:
            logging.info(f"Model {eol_model_name} is not EOL, setting eol")
            release.set_eol(TODAY)
        elif isinstance(current_eol, datetime):
            if current_eol > TODAY:
                logging.info(f"Model {eol_model_name} is not marked as EOL, setting eol as {TODAY}")
                release.set_eol(TODAY)
            else:
                logging.debug(f"Model {eol_model_name} is already EOL, keeping eol as {current_eol}")