"""Units domain registration"""
from ...common.registry import DomainRegistry
from schemas.project_extensions.generator_unit import GeneratorUnitProject
from .sync_node import UnitsSyncNode
from .sync_strategy import UnitsSyncStrategy
from .jsonl_handler import UnitsJsonlHandler
from .api_handler import UnitsApiHandler

# 注册 units domain
DomainRegistry.register(
    node_type="units",
    schema=GeneratorUnitProject,
    node_class=UnitsSyncNode,
    strategy_class=UnitsSyncStrategy,
    jsonl_handler_class=UnitsJsonlHandler,
    api_handler_class=UnitsApiHandler,
)