from __future__ import annotations

import logging
import time
from datetime import UTC, datetime
from decimal import Decimal
from typing import TYPE_CHECKING, Any

import pytest

from generalresearch.managers.spectrum.survey import (
    SpectrumCriteriaManager,
    SpectrumSurveyManager,
)
from generalresearch.models.spectrum.survey import SpectrumSurvey
from generalresearch.sql_helper import SqlHelper

from .surveys_json import CONDITIONS, SURVEYS_JSON

if TYPE_CHECKING:
    from generalresearch.config import GRLBaseSettings


@pytest.fixture(scope="session")
def spectrum_rw(settings: GRLBaseSettings) -> SqlHelper:
    logging.info(f"{settings.spectrum_rw_db=}")

    assert settings.spectrum_rw_db is not None
    assert "/unittest-" in settings.spectrum_rw_db.path

    return SqlHelper(
        dsn=settings.spectrum_rw_db,
        read_timeout=2,
        write_timeout=1,
        connect_timeout=2,
    )


@pytest.fixture(scope="session")
def spectrum_criteria_manager(spectrum_rw: SqlHelper) -> SpectrumCriteriaManager:
    assert "/unittest-" in spectrum_rw.dsn.path
    return SpectrumCriteriaManager(spectrum_rw)


@pytest.fixture(scope="session")
def spectrum_survey_manager(spectrum_rw: SqlHelper) -> SpectrumSurveyManager:
    assert "/unittest-" in spectrum_rw.dsn.path
    return SpectrumSurveyManager(spectrum_rw)


@pytest.fixture(scope="session")
def setup_spectrum_surveys(
    spectrum_rw: SqlHelper, spectrum_survey_manager, spectrum_criteria_manager
) -> None:
    now = datetime.now(UTC)
    # make sure these example surveys exist in db
    surveys = [SpectrumSurvey.model_validate_json(x) for x in SURVEYS_JSON]
    for s in surveys:
        s.modified_api = datetime.now(tz=UTC)
    spectrum_survey_manager.create_or_update(surveys)
    spectrum_criteria_manager.update(CONDITIONS)

    # and make sure they have allocation for 687
    spectrum_rw.execute_sql_query(
        f"""
    INSERT IGNORE INTO `{spectrum_rw.db}`.spectrum_supplier
    (supplier_id, name, api_key, secret_key, username, password)
    VALUES (%s, %s, %s, %s, %s, %s)""",
        ["687", "GRL", "x", "x", "x", "x"],
        commit=True,
    )
    supplier687_pk = spectrum_rw.execute_sql_query(f"""
    select id from `{spectrum_rw.db}`.spectrum_supplier where supplier_id = '687'""")[
        0
    ]["id"]
    conn = spectrum_rw.make_connection()
    c = conn.cursor()
    c.executemany(
        f"""
    INSERT IGNORE INTO `{spectrum_rw.db}`.spectrum_surveysupplier
    (created, surveySig, supplier_id, survey_id)
    VALUES (%s, %s, %s, %s)
    """,
        [[now, "xxx", supplier687_pk, s.survey_id] for s in surveys],
    )
    conn.commit()
    # Wait a second to make sure the spectrum-grpc pulls these from the db into global-vars
    time.sleep(1)


@pytest.fixture(scope="session")
def spectrum_api_survey_json() -> dict[str, Any]:
    return {
        "survey_id": 29333264,
        "survey_name": "#29333264",
        "survey_status": 22,
        "field_end_date": datetime(2024, 5, 23, 18, 18, 31, tzinfo=UTC),
        "category": "Exciting New",
        "category_code": 232,
        "crtd_on": datetime(2024, 5, 20, 17, 48, 13, tzinfo=UTC),
        "mod_on": datetime(2024, 5, 20, 18, 18, 31, tzinfo=UTC),
        "soft_launch": False,
        "click_balancing": 0,
        "price_type": 1,
        "pii": False,
        "buyer_message": "",
        "buyer_id": 4726,
        "incl_excl": 0,
        "cpi": Decimal("1.20"),
        "last_complete_date": None,
        "project_last_complete_date": None,
        "quotas": [
            {
                "quota_id": "c2bc961e-4f26-4223-b409-ebe9165cfdf5",
                "quantities": {"currently_open": 491, "remaining": 495, "achieved": 0},
                "criteria": [
                    {
                        "qualification_code": 214,
                        "range_sets": [{"units": 311, "to": 64, "from": 18}],
                    }
                ],
            }
        ],
        "qualifications": [
            {
                "range_sets": [{"units": 311, "to": 64, "from": 18}],
                "qualification_code": 212,
            },
            {"condition_codes": ["111", "117", "112"], "qualification_code": 1202},
        ],
        "country_iso": "fr",
        "language_iso": "fre",
        "bid_ir": 0.4,
        "bid_loi": 600,
        "overall_ir": None,
        "overall_loi": None,
        "last_block_ir": None,
        "last_block_loi": None,
        "survey_exclusions": set(),
        "exclusion_period": 0,
    }