from scanner.checks.base import BaseCheck
from scanner.checks.cookie_banner import CookieBannerCheck
from scanner.checks.privacy_link import PrivacyLinkCheck
from scanner.checks.form_checkbox import FormCheckboxCheck
from scanner.checks.server_location import ServerLocationCheck
from scanner.checks.external_scripts import ExternalScriptsCheck
from scanner.checks.consent_doc import ConsentDocCheck
from scanner.checks.operator_info import OperatorInfoCheck
from scanner.checks.cookie_audit import CookieAuditCheck
from scanner.checks.llm_policy import LLMPolicyCheck

# Реестр всех проверок
ALL_CHECKS: dict[str, type[BaseCheck]] = {
    "cookie_banner": CookieBannerCheck,
    "cookie_audit": CookieAuditCheck,
    "privacy_link": PrivacyLinkCheck,
    "form_checkbox": FormCheckboxCheck,
    "server_location": ServerLocationCheck,
    "external_scripts": ExternalScriptsCheck,
    "consent_doc": ConsentDocCheck,
    "operator_info": OperatorInfoCheck,
    "llm_policy": LLMPolicyCheck,  # Опциональный, запускается отдельно
}

QUICK_CHECKS = [
    "cookie_audit", "privacy_link", "form_checkbox",
    "server_location", "external_scripts",
    "consent_doc", "operator_info",
]
"""Быстрые проверки без LLM — запускаются по умолчанию."""

__all__ = [
    "BaseCheck",
    "CookieBannerCheck", "PrivacyLinkCheck", "FormCheckboxCheck",
    "ServerLocationCheck", "ExternalScriptsCheck",
    "ConsentDocCheck", "OperatorInfoCheck", "LLMPolicyCheck",
    "ALL_CHECKS", "QUICK_CHECKS",
]