"""Validation utils."""
from typing import List

from ddex_ingester_common.helpers.catalog_ingestion import (
    CatalogIngestionSession, CatalogIngestionValidationResult,
)
from ddex_ingester_common.validation.rules import RuleResult


def save_validation_result(
        rule_results: List[RuleResult],
        state_machine_name: str,
        execution_name: str,
        catalog_ingestion_session: CatalogIngestionSession
) -> List[CatalogIngestionValidationResult]:
    """Save validation results to DB."""
    results = []

    for rule_result in rule_results:
        results.append(
            CatalogIngestionValidationResult(
                state_machine_name=state_machine_name,
                state_machine_execution_name=execution_name,
                validation_rule_id=rule_result.validation_rule_id,
                response=rule_result.response,
                message=rule_result.message
            )
        )

    catalog_ingestion_session.add(results)
    catalog_ingestion_session.save()
    return results