music-assistant-server

1"""Setup flow for the TeddyCloud provider."""
2
3from __future__ import annotations
4
5from dataclasses import replace
6from typing import TYPE_CHECKING
7
8from music_assistant_models.config_entries import ConfigEntry
9from music_assistant_models.enums import ConfigEntryType
10
11from music_assistant.models.setup_flow import SetupFlowError
12from music_assistant.providers.teddycloud.constants import CONF_URL
13
14if TYPE_CHECKING:
15    from music_assistant.models.setup_flow import SetupSession
16
17_ENTRIES = (
18    ConfigEntry(
19        key="label",
20        type=ConfigEntryType.LABEL,
21    ),
22    ConfigEntry(
23        key=CONF_URL,
24        type=ConfigEntryType.STRING,
25        required=True,
26    ),
27)
28
29
30async def run_setup(session: SetupSession) -> None:
31    """Run the setup flow: collect the connection details and create the provider."""
32    errors: dict[str, str] | None = None
33    setup_data = dict(session.context.setup_data)
34    while True:
35        entries = [
36            replace(entry, value=setup_data.get(entry.key, entry.value)) for entry in _ENTRIES
37        ]
38        submitted = await session.form(entries, step_id="user", errors=errors, last_step=True)
39        setup_data.update(submitted)
40        try:
41            await session.finish(setup_data)
42            return
43        except SetupFlowError as err:
44            errors = {"base": err.translation_key or str(err)}
45