"""
Test bootstrap: stub optional `sientia_do` submodules not shipped in minimal installs.

Must run before importing `model_manager.sientia.models` (pulled in via TrainModelParams).
Stubs Evidently submodules so `model_manager.sientia.reports` imports (via DataManagerRepository).
"""

from __future__ import annotations

import sys
from types import ModuleType


def _make_dummy(name: str) -> type:
    return type(name, (), {})


def _stub_evidently() -> None:
    """Minimal Evidently API surface required to import `model_manager.sientia.reports`."""
    ev = ModuleType('evidently')
    sys.modules['evidently'] = ev

    mp = ModuleType('evidently.metric_preset')
    mp.DataDriftPreset = _make_dummy('DataDriftPreset')  # type: ignore[attr-defined]
    sys.modules['evidently.metric_preset'] = mp

    metrics = ModuleType('evidently.metrics')
    _metric_names = (
        'ColumnSummaryMetric',
        'ConflictTargetMetric',
        'DatasetCorrelationsMetric',
        'DatasetSummaryMetric',
        'RegressionAbsPercentageErrorPlot',
        'RegressionDummyMetric',
        'RegressionErrorDistribution',
        'RegressionErrorPlot',
        'RegressionPerformanceMetrics',
        'RegressionPredictedVsActualPlot',
        'RegressionPredictedVsActualScatter',
    )
    for n in _metric_names:
        setattr(metrics, n, _make_dummy(n))
    sys.modules['evidently.metrics'] = metrics

    base = ModuleType('evidently.metrics.base_metric')

    def generate_column_metrics(*_a, **_k):
        return []

    base.generate_column_metrics = generate_column_metrics  # type: ignore[attr-defined]
    sys.modules['evidently.metrics.base_metric'] = base

    opt = ModuleType('evidently.options')
    opt.ColorOptions = _make_dummy('ColorOptions')  # type: ignore[attr-defined]
    sys.modules['evidently.options'] = opt

    pipeline = ModuleType('evidently.pipeline')
    sys.modules['evidently.pipeline'] = pipeline

    colmap = ModuleType('evidently.pipeline.column_mapping')
    colmap.ColumnMapping = _make_dummy('ColumnMapping')  # type: ignore[attr-defined]
    sys.modules['evidently.pipeline.column_mapping'] = colmap

    rep = ModuleType('evidently.report')
    rep.Report = _make_dummy('Report')  # type: ignore[attr-defined]
    sys.modules['evidently.report'] = rep


def pytest_configure(config) -> None:  # noqa: ARG001
    """Register stub modules so imports used by production code resolve in CI/dev venvs."""
    _stub_evidently()

    if 'sientia_do.operations.df_preprocessor' not in sys.modules:
        df_pre = ModuleType('sientia_do.operations.df_preprocessor')

        def create_features(input_data, *_a, **_k):
            return input_data

        def limit_dataset(input_data, low_lim, upp_lim, *_a, **_k):
            return input_data, low_lim, upp_lim

        def treat_nan(input_data, *_a, **_k):
            return input_data

        df_pre.create_features = create_features  # type: ignore[attr-defined]
        df_pre.limit_dataset = limit_dataset  # type: ignore[attr-defined]
        df_pre.treat_nan = treat_nan  # type: ignore[attr-defined]
        sys.modules['sientia_do.operations.df_preprocessor'] = df_pre

    sys.modules.setdefault('sientia_do.operations', ModuleType('sientia_do.operations'))

    if 'sientia_do.timeseries.analyzer' not in sys.modules:
        ts_an = ModuleType('sientia_do.timeseries.analyzer')

        class TimeSeriesDiscontinuityAnalyzer:  # noqa: D401
            """Stub for tests."""

            pass

        ts_an.TimeSeriesDiscontinuityAnalyzer = TimeSeriesDiscontinuityAnalyzer  # type: ignore[attr-defined]
        sys.modules['sientia_do.timeseries.analyzer'] = ts_an

    sys.modules.setdefault('sientia_do.timeseries', ModuleType('sientia_do.timeseries'))