76 lines
2.2 KiB
Python
76 lines
2.2 KiB
Python
import argparse
|
|
from typing import List, Optional
|
|
|
|
from textual.app import App
|
|
|
|
from cobot.commands.doc_setup import run as _doc_setup
|
|
from cobot.commands.docker_setup import run as _docker_setup
|
|
from cobot.commands.local_setup import run as _local_setup
|
|
from cobot.commands.robot_setup import run as _robot_setup
|
|
from cobot.tui import SCREEN_CSS, PickScreen
|
|
|
|
|
|
class _Ask(App[Optional[str]]):
|
|
"""Single-question picker that exits immediately with the chosen value."""
|
|
CSS = SCREEN_CSS
|
|
|
|
def __init__(self, step: str, question: str, options: List[str], default: str):
|
|
super().__init__()
|
|
self._step = step
|
|
self._question = question
|
|
self._options = options
|
|
self._default = default
|
|
|
|
def on_mount(self) -> None:
|
|
self.push_screen(
|
|
PickScreen(self._step, self._question, self._options, self._default),
|
|
self.exit,
|
|
)
|
|
|
|
|
|
def _ask(step: str, question: str, options: List[str], default: str) -> Optional[str]:
|
|
return _Ask(step, question, options, default).run()
|
|
|
|
|
|
def register(subparsers):
|
|
p = subparsers.add_parser("setup", help="First-time project setup")
|
|
p.set_defaults(func=run)
|
|
|
|
|
|
def run(args: argparse.Namespace) -> None:
|
|
# Step 1 — documentation
|
|
v = _ask("Step 1 of 3", "Set up the documentation server?", ["Yes", "No"], "Yes")
|
|
if v is None:
|
|
return
|
|
if v == "Yes":
|
|
_doc_setup(args)
|
|
|
|
# Step 2 — build environment
|
|
v = _ask(
|
|
"Step 2 of 3",
|
|
"How do you want to set up the build environment?",
|
|
[
|
|
"local-setup — install ROS2 Jazzy on this machine and build with colcon",
|
|
"docker-setup — build a Docker image with ROS2 Jazzy pre-installed",
|
|
],
|
|
"local-setup — install ROS2 Jazzy on this machine and build with colcon",
|
|
)
|
|
if v is None:
|
|
return
|
|
if v.startswith("local"):
|
|
_local_setup(args)
|
|
else:
|
|
_docker_setup(args)
|
|
|
|
# Step 3 — robot parameters
|
|
v = _ask(
|
|
"Step 3 of 3",
|
|
"Configure robot parameters (cobot-setting.yaml)?",
|
|
["Yes", "No"],
|
|
"Yes",
|
|
)
|
|
if v is None:
|
|
return
|
|
if v == "Yes":
|
|
_robot_setup(args)
|