from ayon_server.settings import BaseSettingsModel, SettingsField
from .imageio import ImageIOSettings
from .render_settings import (
    RenderSettingsModel, DEFAULT_RENDER_SETTINGS
)
from .create_review_settings import (
    CreateReviewModel, DEFAULT_CREATE_REVIEW_SETTINGS
)
from .publishers import (
    PublishersModel, DEFAULT_PUBLISH_SETTINGS
)


def unit_scale_enum():
    """Return enumerator for scene unit scale."""
    return [
        {"label": "mm", "value": "Millimeters"},
        {"label": "cm", "value": "Centimeters"},
        {"label": "m", "value": "Meters"},
        {"label": "km", "value": "Kilometers"}
    ]


class UnitScaleSettings(BaseSettingsModel):
    enabled: bool = SettingsField(True, title="Enabled")
    scene_unit_scale: str = SettingsField(
        "Centimeters",
        title="Scene Unit Scale",
        enum_resolver=unit_scale_enum
    )


class PRTAttributesModel(BaseSettingsModel):
    _layout = "compact"
    name: str = SettingsField(title="Name")
    value: str = SettingsField(title="Attribute")


class PointCloudSettings(BaseSettingsModel):
    attribute: list[PRTAttributesModel] = SettingsField(
        default_factory=list, title="Channel Attribute")


class MaxSettings(BaseSettingsModel):
    unit_scale_settings: UnitScaleSettings = SettingsField(
        default_factory=UnitScaleSettings,
        title="Set Unit Scale"
    )
    imageio: ImageIOSettings = SettingsField(
        default_factory=ImageIOSettings,
        title="Color Management (ImageIO)"
    )
    RenderSettings: RenderSettingsModel = SettingsField(
        default_factory=RenderSettingsModel,
        title="Render Settings"
    )
    CreateReview: CreateReviewModel = SettingsField(
        default_factory=CreateReviewModel,
        title="Create Review"
    )
    PointCloud: PointCloudSettings = SettingsField(
        default_factory=PointCloudSettings,
        title="Point Cloud"
    )
    publish: PublishersModel = SettingsField(
        default_factory=PublishersModel,
        title="Publish Plugins")


DEFAULT_VALUES = {
    "unit_scale_settings": {
        "enabled": True,
        "scene_unit_scale": "Centimeters"
    },
    "RenderSettings": DEFAULT_RENDER_SETTINGS,
    "CreateReview": DEFAULT_CREATE_REVIEW_SETTINGS,
    "PointCloud": {
        "attribute": [
            {"name": "Age", "value": "age"},
            {"name": "Radius", "value": "radius"},
            {"name": "Position", "value": "position"},
            {"name": "Rotation", "value": "rotation"},
            {"name": "Scale", "value": "scale"},
            {"name": "Velocity", "value": "velocity"},
            {"name": "Color", "value": "color"},
            {"name": "TextureCoordinate", "value": "texcoord"},
            {"name": "MaterialID", "value": "matid"},
            {"name": "custFloats", "value": "custFloats"},
            {"name": "custVecs", "value": "custVecs"},
        ]
    },
    "publish": DEFAULT_PUBLISH_SETTINGS

}