#!/bin/bash

# Install dependencies for Unitree GO2 Custom Controller
# Supports Ubuntu/Debian systems

set -e

echo "Installing dependencies for Unitree GO2 Custom Controller..."

# Update package list
sudo apt update

# Install build essentials
echo "Installing build tools..."
sudo apt install -y \
    build-essential \
    cmake \
    pkg-config \
    git

# Install MQTT libraries
echo "Installing MQTT libraries..."
sudo apt install -y \
    libpaho-mqtt-dev \
    libpaho-mqttpp-dev

# Install JSON library
echo "Installing JSON library..."
sudo apt install -y nlohmann-json3-dev

# Install SSL libraries (required by MQTT)
echo "Installing SSL libraries..."
sudo apt install -y libssl-dev

# Install additional utilities
echo "Installing utilities..."
sudo apt install -y \
    mosquitto-clients \
    net-tools \
    iproute2

echo "Dependencies installed successfully!"

# Check if Unitree SDK2 exists
if [ ! -d "../unitree_sdk2" ]; then
    echo "WARNING: unitree_sdk2 not found at ../unitree_sdk2"
    echo "Please ensure the Unitree SDK2 is available at the correct location"
fi

echo "Setup complete. You can now build the project with:"
echo "  mkdir build && cd build"
echo "  cmake .."
echo "  make -j\$(nproc)"