feat: update robot setup to use ruamel.yaml for preserving YAML comments and formatting

feat: enhance run command with Docker volume for Webots asset caching and GPU detection

feat: improve setup command to sequentially run sub-commands for documentation, environment, and robot parameters

fix: update update command to provide clearer logging during git operations and installation

refactor: enhance TUI components with better threading and logging for long-running tasks

chore: improve install script with better error handling, logging, and interactive setup wizard
This commit is contained in:
Даниил Грабарь
2026-05-21 12:46:54 +10:00
parent e18ff53532
commit 6be6032838
11 changed files with 647 additions and 124 deletions
+70 -14
View File
@@ -1,9 +1,15 @@
#!/bin/bash
# Stop the script immediately if any command exits with an error.
# Останавливаем скрипт сразу, если какая-либо команда завершилась с ошибкой.
set -e
# Disable uv user/project config files so the environment is always clean.
# Отключаем пользовательские и проектные конфиги uv, чтобы среда всегда была чистой.
export UV_NO_CONFIG=1
# Terminal color codes for nicer output.
# Коды цветов для красивого вывода в терминал.
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[0;33m'
@@ -11,19 +17,30 @@ CYAN='\033[0;36m'
NC='\033[0m'
BOLD='\033[1m'
# Python version that the cobot CLI requires.
# Версия Python, которая нужна для работы cobot CLI.
PYTHON_VERSION="3.11"
REPO_URL="https://gitverse.ru/daniel-robotics/lightweight-cobot.git"
# Where to clone the project. Can be overridden by the user with COBOT_INSTALL_DIR.
# Куда клонировать проект. Пользователь может переопределить через COBOT_INSTALL_DIR.
INSTALL_DIR="${COBOT_INSTALL_DIR:-$HOME/.lwc}"
# Определяем интерактивный режим: при запуске через curl | bash stdin не является терминалом
# Detect interactive mode - when run via curl | bash, stdin is not a terminal.
# Определяем интерактивный режим - при запуске через curl | bash stdin не является терминалом.
if [ -t 0 ]; then IS_INTERACTIVE=true; else IS_INTERACTIVE=false; fi
# Logging helpers - one line per severity level.
# Вспомогательные функции логирования - одна строка на уровень важности.
log_info() { echo -e "${CYAN}[*]${NC} $1"; }
log_success() { echo -e "${GREEN}[ok]${NC} $1"; }
log_warn() { echo -e "${YELLOW}[!]${NC} $1"; }
log_error() { echo -e "${RED}[err]${NC} $1"; exit 1; }
# Запускает команду тихо, показывает вывод только при ошибке
# Run a command silently and only print its output if it fails.
# This makes the normal install look clean while still showing errors when something breaks.
# Запускает команду тихо и показывает вывод только если она завершилась с ошибкой.
# Это делает обычную установку аккуратной, но при ошибке мы всё равно видим детали.
run_quiet() {
local _log
_log="$(mktemp /tmp/lwc-cmd.XXXXXX.log)"
@@ -45,11 +62,14 @@ print_banner() {
echo -e "${NC}"
}
# Detect the current OS and package manager so later steps know how to install things.
# Определяем текущую ОС и пакетный менеджер, чтобы следующие шаги знали как устанавливать пакеты.
detect_os() {
case "$(uname -s)" in
Linux*)
OS="linux"
# Определяем пакетный менеджер для установки зависимостей
# Pick the first package manager we can find on this system.
# Выбираем первый найденный пакетный менеджер.
if command -v apt-get &>/dev/null; then PKG_MANAGER="apt"
elif command -v dnf &>/dev/null; then PKG_MANAGER="dnf"
elif command -v pacman &>/dev/null; then PKG_MANAGER="pacman"
@@ -62,7 +82,8 @@ detect_os() {
log_info "OS: $OS"
}
# Устанавливает системные пакеты через найденный пакетный менеджер
# Install system packages using whatever package manager was detected above.
# Устанавливает системные пакеты через найденный пакетный менеджер.
pkg_install() {
case "$PKG_MANAGER" in
apt) run_quiet sudo apt-get update -qq && run_quiet sudo apt-get install -y --no-install-recommends "$@" ;;
@@ -72,6 +93,8 @@ pkg_install() {
esac
}
# Check if git is installed and install it if not.
# Проверяем наличие git и устанавливаем его если он отсутствует.
check_git() {
log_info "Checking git..."
if command -v git &>/dev/null; then
@@ -87,6 +110,8 @@ check_git() {
log_success "git $(git --version | awk '{print $3}') installed"
}
# Check if Docker is installed and install it if not.
# Проверяем наличие Docker и устанавливаем его если он отсутствует.
check_docker() {
log_info "Checking Docker..."
if command -v docker &>/dev/null; then
@@ -94,8 +119,10 @@ check_docker() {
return
fi
log_info "Installing Docker..."
# Скачиваем установщик во временный файл, а не запускаем через pipe —
# так видны ошибки сети отдельно от ошибок самого установщика
# Download the installer to a temp file instead of piping directly through bash.
# This way network errors and installer errors are shown separately.
# Скачиваем установщик во временный файл, а не запускаем через pipe.
# Так ошибки сети и ошибки самого установщика видны по отдельности.
local _installer
_installer="$(mktemp /tmp/lwc-docker.XXXXXX.sh)"
if ! curl -fsSL https://get.docker.com -o "$_installer"; then
@@ -106,16 +133,20 @@ check_docker() {
rm -f "$_installer"
command -v docker &>/dev/null || log_error "Docker not found after installation"
log_success "Docker $(docker --version | awk '{print $3}' | tr -d ',') installed"
# Добавляем пользователя в группу docker, чтобы не требовался sudo
# Add the current user to the docker group so sudo is not needed every time.
# Добавляем текущего пользователя в группу docker, чтобы не требовался sudo каждый раз.
if [ "$(id -u)" -ne 0 ] && command -v usermod &>/dev/null; then
sudo usermod -aG docker "$USER"
log_warn "Added $USER to docker group — re-login to apply"
fi
}
# Install the uv package manager. We need it to create isolated Python environments.
# Устанавливаем пакетный менеджер uv. Он нужен для создания изолированных Python-окружений.
install_uv() {
log_info "Checking uv..."
# uv может быть установлен в ~/.local/bin или ~/.cargo/bin, проверяем оба
# uv can end up in ~/.local/bin or ~/.cargo/bin depending on how it was installed.
# uv может оказаться в ~/.local/bin или ~/.cargo/bin в зависимости от способа установки.
UV_CMD=""
for candidate in "uv" "$HOME/.local/bin/uv" "$HOME/.cargo/bin/uv"; do
if command -v "$candidate" &>/dev/null 2>&1; then
@@ -128,7 +159,10 @@ install_uv() {
return
fi
log_info "Installing uv..."
# Два отдельных файла: лог и установщик — чтобы различать ошибки скачивания и установки
# Two separate temp files - one for the installer script and one for its log.
# This lets us tell apart "download failed" from "installer failed".
# Два отдельных временных файла - для установщика и для его лога.
# Это позволяет различить ошибку скачивания и ошибку самого установщика.
local _log _installer
_log="$(mktemp /tmp/lwc-uv.XXXXXX.log)"
_installer="$(mktemp /tmp/lwc-uv-installer.XXXXXX.sh)"
@@ -153,6 +187,8 @@ install_uv() {
fi
}
# Check that the required Python version is available via uv and install it if not.
# Проверяем наличие нужной версии Python через uv и устанавливаем её если она отсутствует.
check_python() {
log_info "Checking Python $PYTHON_VERSION..."
local py_path
@@ -168,6 +204,14 @@ check_python() {
log_success "Python $PYTHON_VERSION installed"
}
# Figure out where the project lives. Three cases are handled:
# 1. We are already inside the cloned repo - use it directly.
# 2. The repo was cloned before - just pull the latest changes.
# 3. First time - clone the repo fresh.
# Определяем где находится проект. Обрабатываем три случая:
# 1. Мы уже внутри клонированного репозитория - используем его напрямую.
# 2. Репозиторий уже был клонирован ранее - просто тянем последние изменения.
# 3. Первый запуск - клонируем репозиторий заново.
resolve_install_dir() {
local script_dir
script_dir="$(cd "$(dirname "${BASH_SOURCE[0]:-$0}")" 2>/dev/null && pwd || pwd)"
@@ -196,8 +240,11 @@ resolve_install_dir() {
log_info "Cloning repo into $INSTALL_DIR..."
mkdir -p "$(dirname "$INSTALL_DIR")"
# Retry up to 5 times because the git server can be unreliable on slow connections.
# Повторяем до 5 раз, потому что git-сервер может быть нестабильным на медленных соединениях.
local attempt=1
while [ $attempt -le 5 ]; do
# TODO: Изменить на --depth 1 --branch main после слияния dev в main.
if git clone --depth 1 --branch dev "$REPO_URL" "$INSTALL_DIR" </dev/null; then
log_success "Repo cloned to $INSTALL_DIR"
return
@@ -209,15 +256,20 @@ resolve_install_dir() {
log_error "Failed to clone repo after 5 attempts"
}
# Install the cobot CLI as an editable uv tool so changes in the source are reflected immediately.
# Устанавливаем cobot CLI как редактируемый uv-инструмент, чтобы изменения в исходниках применялись сразу.
install_cobot() {
log_info "Installing cobot CLI..."
cd "$INSTALL_DIR"
# uv tool install создаёт изолированное окружение и кладёт бинарник cobot в ~/.local/bin
# uv tool install creates an isolated environment and puts the cobot binary into ~/.local/bin.
# uv tool install создаёт изолированное окружение и кладёт бинарник cobot в ~/.local/bin.
run_quiet "$UV_CMD" tool install --python "$PYTHON_VERSION" --editable . \
|| log_error "Failed to install cobot"
log_success "cobot installed"
}
# Add ~/.local/bin to PATH in the user's shell config if it is not there yet.
# Добавляем ~/.local/bin в PATH в конфиге оболочки пользователя, если его там ещё нет.
setup_path() {
local bin_dir="$HOME/.local/bin"
local shell_rc
@@ -226,11 +278,11 @@ setup_path() {
*/fish) shell_rc="$HOME/.config/fish/config.fish" ;;
*) shell_rc="$HOME/.bashrc" ;;
esac
# Добавляем ~/.local/bin в PATH, если его там ещё нет
# Also export into the current session right now so cobot setup works below without a re-login.
# Также экспортируем прямо сейчас, чтобы cobot setup заработал ниже без перезахода.
if [[ ":$PATH:" != *":$bin_dir:"* ]]; then
echo "" >> "$shell_rc"
echo "export PATH=\"$bin_dir:\$PATH\"" >> "$shell_rc"
# Обновляем PATH внутри скрипта — нужно чтобы cobot setup сработал ниже
export PATH="$bin_dir:$PATH"
fi
command -v cobot &>/dev/null && log_success "cobot -> $(command -v cobot)"
@@ -247,8 +299,11 @@ print_success() {
echo ""
}
# Launch the interactive setup wizard right after installation.
# Запускаем интерактивный мастер настройки сразу после установки.
run_setup() {
# Явная проверка — PATH мог не подхватиться если uv положил бинарник в нестандартное место
# Explicit check because PATH might not include ~/.local/bin yet in this shell session.
# Явная проверка, потому что PATH может ещё не включать ~/.local/bin в этой сессии.
if ! command -v cobot &>/dev/null; then
log_warn "cobot not found on PATH, trying full path..."
local cobot_bin="$HOME/.local/bin/cobot"
@@ -262,7 +317,8 @@ run_setup() {
log_info "Running cobot setup..."
cobot setup
# curl | bash: дочерний процесс не может обновить терминал родителя
# When run via curl | bash the child process cannot update the parent terminal's environment.
# При запуске через curl | bash дочерний процесс не может обновить окружение родительского терминала.
if [ "$IS_INTERACTIVE" = false ]; then
echo ""
echo " To apply PATH changes in this terminal, run:"