"""Contract Lifecycle Schedule marshmallow schema."""

from abacus_common_logic.marshalling.custom_fields import ma
from marshmallow import validates_schema, ValidationError
from marshmallow.validate import Range

from abacus_contract.constants.constants import \
    CONTRACT_LIFECYCLE_SCHEDULE_DETAIL_PERIOD_TYPES
from abacus_contract.constants.constants import \
    CONTRACT_LIFECYCLE_SCHEDULE_RENEWAL_TYPES
from abacus_contract.constants.error import ERROR_RENEWAL_OFFSET_NOT_REQUIRED
from abacus_contract.constants.error import ERROR_RENEWAL_OFFSET_REQUIRED
from abacus_contract.constants.error import ERROR_SCHEDULE_END_NOT_REQUIRED
from abacus_contract.schemas.contract_lifecycle import ContractLifecyclePostSchema


class ContractLifecycleScheduleSchema(ma.Schema):
    """Schema for contract_lifecycle_schedule details."""

    contract_lifecycle_schedule_id = ma.IntegerId(required=True)
    contract_id = ma.IntegerId(required=True)
    termination_notice_detail_id = ma.IntegerId()
    renewal_offset_detail_id = ma.IntegerId()
    collection_period_detail_id = ma.IntegerId()
    renewal_type = ma.Enum(
        options=CONTRACT_LIFECYCLE_SCHEDULE_RENEWAL_TYPES,
        required=True,
        data_key='renewal_type'
    )
    schedule_end = ma.FormattedDate(allow_none=True)
    contract_lifecycle = ma.Nested(ContractLifecyclePostSchema, allow_none=True)


class ContractLifecycleSchedulePostSchema(ma.Schema):
    """Schema for contract_lifecycle_schedule POST request."""  # noqa: E501

    renewal_type = ma.Enum(
        options=CONTRACT_LIFECYCLE_SCHEDULE_RENEWAL_TYPES,
        required=True,
        data_key='renewal_type'
    )
    termination_notice_detail_interval = ma.IntegerId(
        required=True,
        validate=[Range(min=1)]
    )
    termination_notice_detail_type = ma.Enum(
        options=CONTRACT_LIFECYCLE_SCHEDULE_DETAIL_PERIOD_TYPES,
        required=True,
        data_key='termination_notice_detail_type'
    )
    renewal_offset_detail_interval = ma.IntegerId(
        allow_none=True,
        validate=[Range(min=1)]
    )
    renewal_offset_detail_type = ma.Enum(
        options=CONTRACT_LIFECYCLE_SCHEDULE_DETAIL_PERIOD_TYPES,
        data_key='renewal_offset_detail_type',
        allow_none=True
    )
    collection_period_detail_interval = ma.IntegerId(
        allow_none=True,
        validate=[Range(min=1)]
    )
    collection_period_detail_type = ma.Enum(
        options=CONTRACT_LIFECYCLE_SCHEDULE_DETAIL_PERIOD_TYPES,
        data_key='collection_period_detail_type',
        allow_none=True
    )
    schedule_end = ma.FormattedDate(allow_none=True)
    contract_lifecycle = ma.Nested(ContractLifecyclePostSchema, allow_none=True)

    @validates_schema
    def validate_schedule_detail_fields(self, data, **kwargs):
        """Validate schedule detail fields."""
        renew_periodically_renewal_type = \
            CONTRACT_LIFECYCLE_SCHEDULE_RENEWAL_TYPES.RENEW_PERIODICALLY
        renew_after_certain_date_renewal_type = \
            CONTRACT_LIFECYCLE_SCHEDULE_RENEWAL_TYPES.RENEW_AFTER_CERTAIN_DATE

        renewal_type = data.get('renewal_type')
        schedule_end = data.get('schedule_end')
        renewal_offset_detail_interval = data.get('renewal_offset_detail_interval')
        renewal_offset_detail_type = data.get('renewal_offset_detail_type')

        if (renewal_type == renew_periodically_renewal_type):
            if (
                renewal_offset_detail_interval is None
                and renewal_offset_detail_type is None
            ) or (
                renewal_offset_detail_interval is not None
                and renewal_offset_detail_type is None
            ) or (
                renewal_offset_detail_interval is None
                and renewal_offset_detail_type is not None
            ):
                raise ValidationError(
                    ERROR_RENEWAL_OFFSET_REQUIRED.format(
                        renew_periodically_renewal_type
                    )
                )
        else:
            if (
                renewal_offset_detail_interval is not None
                or renewal_offset_detail_type is not None
            ):
                raise ValidationError(
                    ERROR_RENEWAL_OFFSET_NOT_REQUIRED.format(
                        renew_periodically_renewal_type
                    )
                )

        # check schedule_end
        if (
            schedule_end is not None
            and renewal_type != renew_after_certain_date_renewal_type
        ):
            raise ValidationError(
                ERROR_SCHEDULE_END_NOT_REQUIRED.format(
                    renew_after_certain_date_renewal_type
                )
            )


class ContractLifecycleSchedulePutSchema(ContractLifecycleSchedulePostSchema):
    """Schema for contract_lifecycle_schedule PUT request."""

    contract_lifecycle_schedule_id = ma.IntegerId(allow_none=True)