docker_update_py.sh

#!/bin/bash
# utility script to build the ardupilot_wiki with docker

BOLD='\e[1m'
RED='\e[31m'
GREEN='\e[32m'
YELLOW='\e[33m'
MAGENTA='\e[35m'
NC='\e[0m' # no color

# check for docker image and offer to build if not present
check_docker_image() {
    if ! docker image inspect ardupilot_wiki > /dev/null 2>&1; then
        echo -e "\n${BOLD}${RED}Docker image ${NC}${YELLOW}'ardupilot_wiki'${NC}${BOLD}${RED} not found.${NC}"
        echo -en "\nBuild image? (Y/n): "
        read build_choice
        if [[ -z "$build_choice" || "$build_choice" =~ ^([yY][eE][sS]|[yY])$ ]]; then
            echo # newline
            docker build . -t ardupilot_wiki --build-arg USER_UID=$(id -u) --build-arg USER_GID=$(id -g)
        else
            exit 1
        fi
    fi
}

show_menu() {
    echo -e "\n${BOLD}${YELLOW}Select ArduPilot Wiki site to build:${NC}\n"
    echo -e "  ${MAGENTA}1) Copter${NC}"
    echo -e "  ${MAGENTA}2) Plane${NC}"
    echo -e "  ${MAGENTA}3) Rover${NC}"
    echo -e "  ${MAGENTA}4) Dev${NC}"
    echo -e "  ${MAGENTA}5) All${NC}"
    echo -en "\nEnter choice [1-5]: "
    read choice

    case $choice in
        1) site_option="--site copter" ;;
        2) site_option="--site plane" ;;
        3) site_option="--site rover" ;;
        4) site_option="--site dev" ;;
        5) site_option="" ;;
        *) echo -e "\n${BOLD}${RED}Invalid choice.${NC}" && show_menu ;;
    esac

    echo -en "\nBuild fast? (Y/n): "
    read fast_choice
    if [[ -z "$fast_choice" || "$fast_choice" =~ ^([yY][eE][sS]|[yY])$ ]]; then
        fast_option="--fast"
    else
        fast_option=""
    fi

    echo -e "\nRunning: ${BOLD}${GREEN}update.py $fast_option $site_option${NC}\n"
    run_command $fast_option $site_option
}

run_command() {
    docker run --rm -it -v "${PWD}:/ardupilot_wiki" -u "$(id -u):$(id -g)" ardupilot_wiki python3 update.py "$@"
}

check_docker_image

if [ "$#" -eq 0 ]; then
    # if no arguments given, show menu of options for building
    show_menu
else
    # build with docker using arguments passed via CLI
    run_command "$@"
fi