# generated by datamodel-codegen:
#   filename:  update-media-buy-response.json
#   timestamp: 2025-11-15T22:03:55+00:00

from __future__ import annotations

from typing import Annotated, Any

from adcp.types.base import AdCPBaseModel
from pydantic import AwareDatetime, ConfigDict, Field, RootModel

from . import error


class AffectedPackage(AdCPBaseModel):
    model_config = ConfigDict(
        extra="forbid",
    )
    buyer_ref: Annotated[str, Field(description="Buyer's reference for the package")]
    package_id: Annotated[str, Field(description="Publisher's package identifier")]


class UpdateMediaBuyResponse1(AdCPBaseModel):
    model_config = ConfigDict(
        extra="forbid",
    )
    affected_packages: Annotated[
        list[AffectedPackage] | None, Field(description="Array of packages that were modified")
    ] = None
    buyer_ref: Annotated[str, Field(description="Buyer's reference identifier for the media buy")]
    context: Annotated[
        dict[str, Any] | None,
        Field(
            description="Initiator-provided context echoed inside the task payload. Opaque metadata such as UI/session hints, correlation tokens, or tracking identifiers."
        ),
    ] = None
    implementation_date: Annotated[
        AwareDatetime | None,
        Field(description="ISO 8601 timestamp when changes take effect (null if pending approval)"),
    ] = None
    media_buy_id: Annotated[str, Field(description="Publisher's identifier for the media buy")]


class UpdateMediaBuyResponse2(AdCPBaseModel):
    model_config = ConfigDict(
        extra="forbid",
    )
    context: Annotated[
        dict[str, Any] | None,
        Field(
            description="Initiator-provided context echoed inside the task payload. Opaque metadata such as UI/session hints, correlation tokens, or tracking identifiers."
        ),
    ] = None
    errors: Annotated[
        list[error.Error],
        Field(description="Array of errors explaining why the operation failed", min_length=1),
    ]


class UpdateMediaBuyResponse(RootModel[UpdateMediaBuyResponse1 | UpdateMediaBuyResponse2]):
    root: Annotated[
        UpdateMediaBuyResponse1 | UpdateMediaBuyResponse2,
        Field(
            description="Response payload for update_media_buy task. Returns either complete success data OR error information, never both. This enforces atomic operation semantics - updates are either fully applied or not applied at all.",
            title="Update Media Buy Response",
        ),
    ]
