import os

import anyio
from anyio import create_task_group
from pyslobs import ConnectionConfig, ScenesService, SlobsConnection


async def setup(conn: SlobsConnection):
    ss = ScenesService(conn)
    await ss.create_scene("slobs-test-scene-1")
    await ss.create_scene("slobs-test-scene-2")
    await ss.create_scene("slobs-test-scene-3")

    conn.close()


async def main():
    conn = SlobsConnection(
        ConnectionConfig(
            domain=os.environ["SLOBS_DOMAIN"],
            port=59650,
            token=os.environ["SLOBS_TOKEN"],
        )
    )

    async with create_task_group() as tg:
        tg.start_soon(conn.background_processing)
        tg.start_soon(setup, conn)


if __name__ == "__main__":
    anyio.run(main)