"""Schema for source of streams endpoint."""

from marshmallow import fields

from sound_recordings.schemas.common import BaseSchema, SourceSchema


class SourceOfStreamsBreakdownItemSchema(BaseSchema):
    """Marshmallow schema for SOS breakdown items section."""

    growth_percentage = fields.Number(data_key="growth_percentage")
    total = fields.Integer()
    value = fields.Number()


class SourceOfStreamsSchema(BaseSchema):
    """Marshmallow for Source of Streams breakdown."""

    isrc = fields.String()
    active = fields.Nested(SourceOfStreamsBreakdownItemSchema)
    passive = fields.Nested(SourceOfStreamsBreakdownItemSchema)
    collection = fields.Nested(SourceOfStreamsBreakdownItemSchema)
    sources = fields.Nested(SourceSchema, many=True)