install.sh

#!/bin/bash
set -e

echo "🚀 Starting Between Threads installation"

# Configuration
APP_DIR="$HOME/Desktop/BetweenThreads"
SERVICE_PIGPIO="pigpiod-custom"
SERVICE_BT="between-threads"
SERVICE_DASHBOARD="between-threads-dashboard"
SERVICE_LIVESTREAM="between-threads-livestream"
USER_NAME=$(whoami)

# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

log_info() {
    echo -e "${GREEN}[INFO]${NC} $1"
}

log_warn() {
    echo -e "${YELLOW}[WARN]${NC} $1"
}

log_error() {
    echo -e "${RED}[ERROR]${NC} $1"
}

# ---------------------------
# 1. Create application directory
# ---------------------------
log_info "Creating application directory..."
mkdir -p "$APP_DIR"
cd "$APP_DIR"

# ---------------------------
# 2. Install system dependencies
# ---------------------------
log_info "Installing system dependencies..."

# Update package list
sudo apt-get update -qq

# Install essential tools
sudo apt-get install -y -qq \
    git \
    curl \
    unzip \
    wget \
    build-essential \
    python3 \
    python3-pip \
    nodejs \
    npm \
    ffmpeg

# Install uv if not present
UV_PATH="$HOME/.local/bin/uv"
if [ ! -f "$UV_PATH" ]; then
    log_info "Installing uv..."
    curl -LsSf https://astral.sh/uv/install.sh | sh
    export PATH="$HOME/.local/bin:$PATH"
else
    log_info "uv already installed"
fi

# ---------------------------
# 3. Clone repositories
# ---------------------------
log_info "Cloning repositories..."

# RaspberryPi-Software
if [ ! -d "RaspberryPi-Software" ]; then
    git clone https://github.com/Between-Threads-Project/RaspberryPi-Software
else
    log_warn "RaspberryPi-Software already exists, skipping clone"
fi

# Dashboard
if [ ! -d "Dashboard" ]; then
    git clone https://github.com/Between-Threads-Project/Dashboard
else
    log_warn "Dashboard already exists, skipping clone"
fi

# Live-Streaming
if [ ! -d "Live-Streaming" ]; then
    git clone https://github.com/Between-Threads-Project/Live-Streaming
else
    log_warn "Live-Streaming already exists, skipping clone"
fi

# ---------------------------
# 4. Install pigpio
# ---------------------------
log_info "Installing pigpio..."
cd ~

if [ ! -d "pigpio-master" ]; then
    wget -O pigpio-master.zip https://github.com/joan2937/pigpio/archive/master.zip
    unzip pigpio-master.zip
    cd pigpio-master
    make -j$(nproc)
    sudo make install
else
    log_warn "pigpio already installed, skipping build"
fi

# ---------------------------
# 5. Install Python dependencies
# ---------------------------
log_info "Installing Python dependencies..."

# RaspberryPi-Software dependencies
cd "$APP_DIR/RaspberryPi-Software"
if [ -f "pyproject.toml" ]; then
    "$UV_PATH" sync
else
    log_error "pyproject.toml not found in RaspberryPi-Software"
    exit 1
fi

# Live-Streaming (Node.js application, no build needed)
cd "$APP_DIR/Live-Streaming"
if [ -f "server.js" ]; then
    log_info "Live-Streaming files verified"
else
    log_error "server.js not found in Live-Streaming"
    exit 1
fi

# ---------------------------
# 6. Install Dashboard dependencies
# ---------------------------
log_info "Installing Dashboard dependencies..."
cd "$APP_DIR/Dashboard"

if [ -f "package.json" ]; then
    npm install --silent
    npm run build
else
    log_error "package.json not found in Dashboard"
    exit 1
fi

# ---------------------------
# 7. Create systemd services
# ---------------------------
log_info "Creating systemd services..."

# pigpio service
sudo tee /etc/systemd/system/${SERVICE_PIGPIO}.service > /dev/null <<EOF
[Unit]
Description=Pigpio Daemon
After=network.target

[Service]
ExecStart=/usr/local/bin/pigpiod
Type=forking
PIDFile=/run/pigpio.pid
Restart=always
User=root

[Install]
WantedBy=multi-user.target
EOF

# Between Threads main service
sudo tee /etc/systemd/system/${SERVICE_BT}.service > /dev/null <<EOF
[Unit]
Description=Between Threads - Raspberry Pi Software
After=network.target ${SERVICE_PIGPIO}.service
Requires=${SERVICE_PIGPIO}.service

[Service]
WorkingDirectory=$APP_DIR/RaspberryPi-Software
ExecStart=$UV_PATH run web.server
Restart=always
User=$USER_NAME
Environment=PATH=$HOME/.local/bin:$PATH

[Install]
WantedBy=multi-user.target
EOF

# Dashboard service
sudo tee /etc/systemd/system/${SERVICE_DASHBOARD}.service > /dev/null <<EOF
[Unit]
Description=Between Threads - Dashboard
After=network.target

[Service]
WorkingDirectory=$APP_DIR/Dashboard
ExecStart=/usr/bin/npm run start
Restart=always
User=$USER_NAME
Environment=NODE_ENV=production

[Install]
WantedBy=multi-user.target
EOF

# Live-Streaming service
sudo tee /etc/systemd/system/${SERVICE_LIVESTREAM}.service > /dev/null <<EOF
[Unit]
Description=Between Threads - Live Streaming
After=network.target

[Service]
WorkingDirectory=$APP_DIR/Live-Streaming
ExecStart=/usr/bin/node server.js
Restart=always
User=$USER_NAME
Environment=NODE_ENV=production

[Install]
WantedBy=multi-user.target
EOF

# ---------------------------
# 8. Enable and start services
# ---------------------------
log_info "Enabling and starting services..."

sudo systemctl daemon-reload

# Enable services to start on boot
sudo systemctl enable ${SERVICE_PIGPIO}
sudo systemctl enable ${SERVICE_BT}
sudo systemctl enable ${SERVICE_DASHBOARD}
sudo systemctl enable ${SERVICE_LIVESTREAM}

# Start services
sudo systemctl start ${SERVICE_PIGPIO}
sudo systemctl start ${SERVICE_BT}
sudo systemctl start ${SERVICE_DASHBOARD}
sudo systemctl start ${SERVICE_LIVESTREAM}

# ---------------------------
# 9. Verify installation
# ---------------------------
log_info "Verifying installation..."

echo ""
echo "Service status:"
echo "----------------"

# Check pigpio service
if sudo systemctl is-active --quiet ${SERVICE_PIGPIO}; then
    echo -e "${GREEN}✓${NC} ${SERVICE_PIGPIO} is running"
else
    echo -e "${RED}✗${NC} ${SERVICE_PIGPIO} is not running"
fi

# Check between-threads service
if sudo systemctl is-active --quiet ${SERVICE_BT}; then
    echo -e "${GREEN}✓${NC} ${SERVICE_BT} is running"
else
    echo -e "${RED}✗${NC} ${SERVICE_BT} is not running"
fi

# Check dashboard service
if sudo systemctl is-active --quiet ${SERVICE_DASHBOARD}; then
    echo -e "${GREEN}✓${NC} ${SERVICE_DASHBOARD} is running"
else
    echo -e "${RED}✗${NC} ${SERVICE_DASHBOARD} is not running"
fi

# Check live-streaming service
if sudo systemctl is-active --quiet ${SERVICE_LIVESTREAM}; then
    echo -e "${GREEN}✓${NC} ${SERVICE_LIVESTREAM} is running"
else
    echo -e "${RED}✗${NC} ${SERVICE_LIVESTREAM} is not running"
fi

echo ""
echo "Installation paths:"
echo "------------------"
echo "Application directory: $APP_DIR"
echo "RaspberryPi-Software: $APP_DIR/RaspberryPi-Software"
echo "Dashboard: $APP_DIR/Dashboard"
echo "Live-Streaming: $APP_DIR/Live-Streaming"

echo ""
echo "Access information:"
echo "------------------"
echo "Dashboard URL: http://$(hostname -I | awk '{print $1}'):3000"
echo "API URL: http://$(hostname -I | awk '{print $1}'):8000"
echo "Live Stream URL: http://$(hostname -I | awk '{print $1}'):8000"
echo "UDP ports: 5000, 5001 (for external hand recognition)"

echo ""
echo "${GREEN}✅ Installation complete!${NC}"
echo "All services are configured to start automatically on boot."