from command import CustomCommand, CustomConfig

from lightning import LightningFlow
from lightning_app.api import Post
from lightning_app.core.app import LightningApp


class ChildFlow(LightningFlow):
    def nested_command(self, name: str):
        print(f"Hello {name}")

    def configure_commands(self):
        return [{"nested_command": self.nested_command}]


class FlowCommands(LightningFlow):
    def __init__(self):
        super().__init__()
        self.names = []
        self.child_flow = ChildFlow()

    def run(self):
        if self.names:
            print(self.names)

    def command_without_client(self, name: str):
        self.names.append(name)

    def command_with_client(self, config: CustomConfig):
        self.names.append(config.name)

    def configure_commands(self):
        commands = [
            {"command_without_client": self.command_without_client},
            {"command_with_client": CustomCommand(self.command_with_client)},
        ]
        return commands + self.child_flow.configure_commands()

    def configure_api(self):
        return [Post("/user/command_without_client", self.command_without_client)]


app = LightningApp(FlowCommands(), debug=True)