dto.py

"""ldap schema DTO.

Copyright (c) 2025 MultiFactor
License: https://github.com/MultiDirectoryLab/MultiDirectory/blob/main/LICENSE
"""

from dataclasses import dataclass, field
from typing import Generic, TypeVar

from enums import EntityTypeNames, KindType

_IdT = TypeVar("_IdT", int, None)


@dataclass
class AttributeTypeDTO(Generic[_IdT]):
    """Attribute Type DTO."""

    oid: str
    name: str
    ldap_display_name: str
    syntax: str
    single_value: bool
    no_user_modification: bool
    is_system: bool
    system_flags: int
    is_included_anr: bool
    id: _IdT = None  # type: ignore
    object_class_names: set[str] = field(default_factory=set)


_LinkT = TypeVar("_LinkT", AttributeTypeDTO, str)


@dataclass
class ObjectClassDTO(Generic[_IdT, _LinkT]):
    """Object Class DTO."""

    oid: str
    name: str
    superior_name: str | None
    kind: KindType
    is_system: bool
    attribute_types_must: list[_LinkT]
    attribute_types_may: list[_LinkT]
    id: _IdT = None  # type: ignore
    entity_type_names: set[str] = field(default_factory=set)


@dataclass
class EntityTypeDTO(Generic[_IdT]):
    """Entity Type DTO."""

    name: EntityTypeNames | str
    is_system: bool
    object_class_names: list[str]
    id: _IdT = None  # type: ignore


@dataclass
class AttributeDTO:
    name: str
    values: list[str]


@dataclass
class CreateDirDTO:
    name: str
    entity_type_name: EntityTypeNames
    attributes: tuple[AttributeDTO, ...]
    is_system: bool