Checkmk-Development/plugins/edirectory_monitor/server_side_calls/special_agent.py

#!/usr/bin/env python3

from collections.abc import Iterable, Sequence

from pydantic import BaseModel

from cmk.server_side_calls.v1 import HostConfig, SpecialAgentCommand, SpecialAgentConfig
from cmk.server_side_calls.v1._utils import Secret

class _Params(BaseModel):
    ldap_uri: str
    user: str
    password: Secret

def _commands_function(params: _Params, host_config: HostConfig) -> Iterable[SpecialAgentCommand]:
    command_arguments: list[str | Secret] = [
        params.ldap_uri,
        params.user,
        params.password.unsafe(),
    ]

    yield SpecialAgentCommand(command_arguments=command_arguments)

special_agent_edirectory_monitor = SpecialAgentConfig(
    name="edirectory_monitor",
    parameter_parser=_Params.model_validate,
    commands_function=_commands_function,
)