install.sh

#!/usr/bin/env bash
# =============================================================================
# Traveller World Generator — Quick Install (macOS / Linux)
# =============================================================================
# Creates a Python virtual environment, installs the GUI library (PySide6),
# and generates launcher scripts for the desktop app and command-line tools.
#
# Usage:
#   bash install.sh
#
# Requirements:
#   Python 3.9 or later  — https://www.python.org/downloads/
# =============================================================================

set -euo pipefail

# ── Colours ───────────────────────────────────────────────────────────────────
if [ -t 1 ]; then
    GREEN='\033[0;32m'; YELLOW='\033[1;33m'; RED='\033[0;31m'; CYAN='\033[0;36m'; NC='\033[0m'
else
    GREEN=''; YELLOW=''; RED=''; CYAN=''; NC=''
fi

info()  { echo -e "${GREEN}[install]${NC} $*"; }
warn()  { echo -e "${YELLOW}[warning]${NC} $*"; }
fail()  { echo -e "${RED}[error]${NC} $*" >&2; exit 1; }

SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
cd "$SCRIPT_DIR"

echo ""
echo -e "${CYAN}Traveller World Generator — Installation${NC}"
echo "========================================="
echo ""

# ── 1. Locate Python 3.9+ ─────────────────────────────────────────────────────
info "Checking Python..."

PYTHON=""
for cmd in python3 python python3.11 python3.10 python3.9; do
    if command -v "$cmd" &>/dev/null; then
        major=$("$cmd" -c "import sys; print(sys.version_info.major)" 2>/dev/null || echo 0)
        minor=$("$cmd" -c "import sys; print(sys.version_info.minor)" 2>/dev/null || echo 0)
        if [ "$major" -gt 3 ] || { [ "$major" -eq 3 ] && [ "$minor" -ge 9 ]; }; then
            PYTHON="$cmd"
            break
        fi
    fi
done

if [ -z "$PYTHON" ]; then
    warn "Python 3.9 or later was not found."

    OS_TYPE="$(uname -s)"

    if [ "$OS_TYPE" = "Darwin" ]; then
        # macOS
        if command -v brew &>/dev/null; then
            info "Installing Python 3.11 via Homebrew..."
            brew install python@3.11 \
                || fail "Homebrew failed to install Python. Download from: https://www.python.org/downloads/"
        else
            fail "Python 3.9 or later is required but was not found.
       Install Homebrew (https://brew.sh) first, then run: brew install python@3.11
       Or download Python directly from: https://www.python.org/downloads/"
        fi
    elif [ "$OS_TYPE" = "Linux" ]; then
        # Linux — detect package manager
        if command -v apt-get &>/dev/null; then
            info "Installing Python 3 via apt..."
            sudo apt-get update -qq \
                && sudo apt-get install -y python3 python3-venv \
                || fail "apt failed to install Python. Try: sudo apt-get install python3 python3-venv"
        elif command -v dnf &>/dev/null; then
            info "Installing Python 3 via dnf..."
            sudo dnf install -y python3 \
                || fail "dnf failed to install Python. Try: sudo dnf install python3"
        elif command -v pacman &>/dev/null; then
            info "Installing Python 3 via pacman..."
            sudo pacman -Sy --noconfirm python \
                || fail "pacman failed to install Python. Try: sudo pacman -S python"
        else
            fail "No supported package manager found (apt, dnf, pacman).
       Download Python 3.11 from: https://www.python.org/downloads/
       After installing, re-run this script."
        fi
    else
        fail "Unsupported OS: $OS_TYPE.
       Download Python 3.11 from: https://www.python.org/downloads/"
    fi

    # Re-check after auto-install
    for cmd in python3.11 python3 python; do
        if command -v "$cmd" &>/dev/null; then
            major=$("$cmd" -c "import sys; print(sys.version_info.major)" 2>/dev/null || echo 0)
            minor=$("$cmd" -c "import sys; print(sys.version_info.minor)" 2>/dev/null || echo 0)
            if [ "$major" -gt 3 ] || { [ "$major" -eq 3 ] && [ "$minor" -ge 9 ]; }; then
                PYTHON="$cmd"
                break
            fi
        fi
    done

    if [ -z "$PYTHON" ]; then
        fail "Python was installed but cannot be found. Open a new terminal and re-run install.sh"
    fi
fi

PY_VER=$("$PYTHON" -c "import sys; print('%d.%d' % sys.version_info[:2])")
info "Found Python $PY_VER  ($PYTHON)"

# ── 2. Create virtual environment ─────────────────────────────────────────────
VENV_DIR="$SCRIPT_DIR/.venv"

if [ -d "$VENV_DIR" ]; then
    info "Virtual environment already exists — will update packages."
else
    info "Creating virtual environment in .venv/ ..."
    "$PYTHON" -m venv "$VENV_DIR" || fail "Could not create virtual environment."
fi

VENV_PYTHON="$VENV_DIR/bin/python3"

# ── 3. Install dependencies ───────────────────────────────────────────────────
info "Upgrading pip..."
"$VENV_PYTHON" -m pip install --quiet --upgrade pip

info "Installing traveller-gen package (generation modules + CLI entry points)..."
"$VENV_PYTHON" -m pip install --quiet -e "$SCRIPT_DIR" \
    || fail "Failed to install traveller-gen package."

info "Installing PySide6 and PySide6-Addons (desktop GUI library) — this may take a few minutes..."
"$VENV_PYTHON" -m pip install --quiet -r "$SCRIPT_DIR/gen-ui/requirements.txt" \
    || fail "Failed to install PySide6. Check your internet connection and try again."

info "Installing dev tools (pytest, pylint)..."
"$VENV_PYTHON" -m pip install --quiet -r "$SCRIPT_DIR/requirements-dev.txt" \
    || fail "Failed to install dev tools."
info "All dependencies installed."

# ── 4. Create VS Code settings ───────────────────────────────────────────────
VSCODE_DIR="$SCRIPT_DIR/.vscode"
VSCODE_SETTINGS="$VSCODE_DIR/settings.json"
if [ ! -f "$VSCODE_SETTINGS" ]; then
    info "Creating VS Code settings (.vscode/settings.json)..."
    mkdir -p "$VSCODE_DIR"
    cat > "$VSCODE_SETTINGS" << 'VSCODE'
{
  "python.defaultInterpreterPath": "${workspaceFolder}/.venv/bin/python3",
  "python.terminal.activateEnvironment": true,
  "python.terminal.activateEnvInCurrentTerminal": true,
  "python.testing.pytestEnabled": true,
  "python.testing.pytestArgs": [
    "tests"
  ],
  "python.testing.unittestEnabled": false,
  "python.analysis.venvPath": "${workspaceFolder}",
  "python.analysis.venv": ".venv",
  "python.analysis.extraPaths": [
    "${workspaceFolder}"
  ],
  "python.analysis.typeCheckingMode": "basic",
  "editor.formatOnSave": true,
  "editor.rulers": [
    88
  ],
  "files.exclude": {
    "**/__pycache__": true,
    "**/*.pyc": true,
    ".pytest_cache": true
  },
  "[python]": {
    "editor.defaultFormatter": "ms-python.python"
  },
  "azureFunctions.deploySubpath": ".",
  "azureFunctions.scmDoBuildDuringDeployment": true,
  "azureFunctions.pythonVenv": ".venv",
  "azureFunctions.projectLanguage": "Python",
  "azureFunctions.projectRuntime": "~4",
  "debug.internalConsoleOptions": "neverOpen",
  "azureFunctions.projectLanguageModel": 2,
  "pylint.path": [
    "${workspaceFolder}/.venv/bin/pylint"
  ],
  "pylint.interpreter": [
    "${workspaceFolder}/.venv/bin/python3"
  ],
  "pylint.args": [
    "--disable=duplicate-code,too-many-lines"
  ]
}
VSCODE
else
    info "VS Code settings already exist - skipping."
fi

# ── 5. Create launcher scripts ────────────────────────────────────────────────
info "Creating launcher scripts..."

# -- Desktop GUI (.command = double-clickable in Finder on macOS) -------------
cat > "$SCRIPT_DIR/run-gui.command" << 'LAUNCHER'
#!/usr/bin/env bash
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
cd "$SCRIPT_DIR"
exec "$SCRIPT_DIR/.venv/bin/python3" "$SCRIPT_DIR/gen-ui/app.py"
LAUNCHER
chmod +x "$SCRIPT_DIR/run-gui.command"

# -- World generator CLI -------------------------------------------------------
cat > "$SCRIPT_DIR/run-world.sh" << 'LAUNCHER'
#!/usr/bin/env bash
# Generate one or more Traveller mainworlds.
#
# Options:
#   --name NAME       World name (default: auto-numbered)
#   --count N         Number of worlds to generate (default: 1)
#   --seed N          RNG seed for reproducible results
#   --json            Output as JSON instead of text
#   --html            Output as an HTML card
#
# Examples:
#   bash run-world.sh
#   bash run-world.sh --name "New Terra" --count 3
#   bash run-world.sh --name Zhodane --seed 42 --json
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
cd "$SCRIPT_DIR"
exec "$SCRIPT_DIR/.venv/bin/traveller-world" "$@"
LAUNCHER
chmod +x "$SCRIPT_DIR/run-world.sh"

# -- TravellerMap lookup CLI ---------------------------------------------------
cat > "$SCRIPT_DIR/run-mapfetch.sh" << 'LAUNCHER'
#!/usr/bin/env bash
# Fetch a world from TravellerMap and generate a full star system.
# Requires an internet connection.
#
# Options (--sector is always required):
#   --sector SECTOR   Sector name, e.g. "Spinward Marches"
#   --name NAME       World name within the sector
#   --hex NNNN        4-digit hex position (alternative to --name)
#   --seed N          RNG seed for reproducible results
#   --detail          Include secondary world and moon profiles
#   --json            Output as JSON
#   --html            Output as an HTML card
#
# Examples:
#   bash run-mapfetch.sh --sector "Spinward Marches" --name Regina
#   bash run-mapfetch.sh --sector "Spinward Marches" --hex 1910 --detail
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
cd "$SCRIPT_DIR"
exec "$SCRIPT_DIR/.venv/bin/traveller-mapfetch" "$@"
LAUNCHER
chmod +x "$SCRIPT_DIR/run-mapfetch.sh"

# -- FastAPI server ------------------------------------------------------------
cat > "$SCRIPT_DIR/run-fastapi.sh" << 'LAUNCHER'
#!/usr/bin/env bash
# Start the Traveller World Generator FastAPI server.
#
# The server listens on http://localhost:8000 by default.
#
# Options (passed through to uvicorn):
#   --host HOST   Bind address (default: 127.0.0.1)
#   --port PORT   Port number   (default: 8000)
#   --reload      Auto-reload on source changes (development only)
#
# Examples:
#   bash run-fastapi.sh
#   bash run-fastapi.sh --host 0.0.0.0 --port 8080
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
cd "$SCRIPT_DIR/fastapi"
exec "$SCRIPT_DIR/.venv/bin/uvicorn" app:app --host 127.0.0.1 --port 8000 "$@"
LAUNCHER
chmod +x "$SCRIPT_DIR/run-fastapi.sh"

# ── 6. Activate virtual environment ──────────────────────────────────────────
if [ "${VIRTUAL_ENV:-}" != "$VENV_DIR" ]; then
    info "Activating virtual environment..."
    # shellcheck disable=SC1091
    source "$VENV_DIR/bin/activate"
else
    info "Virtual environment already active."
fi

# ── 7. Done ───────────────────────────────────────────────────────────────────
echo ""
info "Installation complete!"
echo ""
echo -e "${CYAN}  ┌──────────────────────────────────────────────────────────────┐${NC}"
echo -e "${CYAN}  │  Desktop app                                                 │${NC}"
echo -e "${CYAN}  │    macOS:  double-click  run-gui.command  in Finder          │${NC}"
echo -e "${CYAN}  │    Linux:  bash run-gui.command                              │${NC}"
echo -e "${CYAN}  │                                                              │${NC}"
echo -e "${CYAN}  │  World generator (command line)                              │${NC}"
echo -e "${CYAN}  │    bash run-world.sh [--name NAME] [--count N]              │${NC}"
echo -e "${CYAN}  │                                                              │${NC}"
echo -e "${CYAN}  │  TravellerMap lookup (command line, needs internet)          │${NC}"
echo -e "${CYAN}  │    bash run-mapfetch.sh --sector SECTOR --name NAME         │${NC}"
echo -e "${CYAN}  └──────────────────────────────────────────────────────────────┘${NC}"
echo ""
echo "  Examples:"
echo "    bash run-world.sh"
echo "    bash run-world.sh --name \"New Terra\" --count 5"
echo "    bash run-mapfetch.sh --sector \"Spinward Marches\" --name Regina"
echo ""
echo -e "  ${YELLOW}macOS note:${NC} If Finder shows a security warning when opening"
echo "  run-gui.command, right-click it and choose Open, then click Open."
echo ""
echo -e "  ${YELLOW}To activate the venv in a new terminal:${NC}"
echo "    source .venv/bin/activate"
echo "    # or dot-source this script:  . ./install.sh"
echo ""