program.sh

#!/bin/bash

# Define colors
RED='\033[0;31m'
GREEN='\033[0;32m'
NC='\033[0m'
YELLOW='\033[1;33m'

# Define directories
MQTT_DIR="mqtt"
SERVER_DIR="server"
CLIENT_DIR="client"
DATABASE_DIR="database"
TEST_DIR="test"

echo -e "${YELLOW}Weather Station Control Script${NC}"

# Function to show help message
function show_help() {
    echo "Usage: $0 [command] [options]"
    echo ""
    echo "Commands:"
    echo "  install       Install Node.js dependencies for server and client."
    echo "  start         Start MQTT broker, Node.js server, and Vite server."
    echo "  test          Run tests sensors and send data to MQTT broker."
    echo "  stop          Stop MQTT broker, Node.js server, and Vite server."
    echo "  status        Show the status of the Mosquitto MQTT broker."
    echo "  logs          Fetch and display the logs of the Mosquitto MQTT broker."
    echo ""
    echo "Options:"
    echo "  -h, help        Show this help message and exit."
}

# Function to stop processes listening on specific ports
function stop_ports() {
    echo "Attempting to free up required ports..."
    for PORT in "$@"; do
        echo "- Checking port $PORT..."
        PROCESSES=$(lsof -ti:$PORT)
        if [ ! -z "$PROCESSES" ]; then
            echo "  > Found processes on port $PORT. Stopping them..."
            echo $PROCESSES | xargs kill -9
            echo -e "  > Port $PORT has been ${GREEN}freed${NC}."
        else
            echo -e "  > No processes found on port $PORT. It's already ${GREEN}available${NC}."
        fi
    done
}

# Function to run tests
function run_tests() {
    echo "Setting up test environment..."
    (cd $TEST_DIR && python3 -m venv venv && source venv/bin/activate && pip install -r requirements.txt) && echo -e "  > Test environment ${GREEN}setup successfully${NC}."

    echo "Running app.py in the test directory..."
    # Activate the virtual environment and run app.py
    (
        cd $TEST_DIR && source venv/bin/activate && python app.py
    ) && echo -e "  > app.py ${GREEN}completed successfully${NC}." || echo -e "  > Running app.py ${RED}failed${NC}."

}

# Function to install Node.js dependencies
function install_dependencies() {
    echo "Installing Node.js dependencies..."
    echo "- Server dependencies..."
    (cd $SERVER_DIR && npm install) && echo -e "  > Server dependencies ${GREEN}installed successfully${NC}." || echo -e "  > ${RED}Failed to install server dependencies${NC}."

    echo "- Client dependencies..."
    (cd $CLIENT_DIR && npm install) && echo -e "  > Client dependencies ${GREEN}installed successfully${NC}." || echo -e "  > ${RED}Failed to install client dependencies${NC}."
}

# Function to start the application components
function start_app() {
    echo "Starting application components..."
    stop_ports 3000 5173 27017

    echo "- Starting MQTT broker..."
    (cd $MQTT_DIR && docker-compose up -d) && echo -e "  > MQTT broker ${GREEN}started${NC}." || echo -e "  > ${RED}Failed to start MQTT broker${NC}."

    echo "- Starting MongoDB database..."
    (cd $DATABASE_DIR && docker-compose up -d) && echo -e "  > MongoDB database ${GREEN}started${NC}." || echo -e "  > ${RED}Failed to start MongoDB database${NC}."

    echo "- Starting Node.js server..."
    (
        cd $SERVER_DIR && nohup node server.js >server.log 2>&1 &
        echo $! >"$SERVER_DIR/server.PID"
    ) && echo -e "  > Node.js server ${GREEN}started${NC}." || echo -e "  > ${RED}Failed to start Node.js server${NC}."

    echo "- Starting Vite server..."
    (
        cd $CLIENT_DIR && nohup npm run dev >client.log 2>&1 &
        echo $! >"$CLIENT_DIR/vite.PID"
    ) && echo -e "  > Vite server ${GREEN}started${NC} at http://localhost:5173/." || echo -e "  > ${RED}Failed to start Vite server${NC}."

    echo -e "All components ${GREEN}started successfully${NC}."
}

# Function to stop the application components
function stop_app() {
    echo "Stopping application components..."

    echo "- Stopping MQTT broker..."
    (cd $MQTT_DIR && docker-compose down) && echo -e "  > MQTT broker ${GREEN}stopped${NC}." || echo -e "  > ${RED}Failed to stop MQTT broker${NC}."

    echo "- Stopping MongoDB database..."
    (cd $DATABASE_DIR && docker-compose down) && echo -e "  > MongoDB database ${GREEN}stopped${NC}." || echo -e "  > ${RED}Failed to stop MongoDB database${NC}."

    stop_servers_and_clear_logs
    stop_ports 3000 5173 27017

    echo -e "All components ${GREEN}stopped successfully${NC}."
}

function stop_servers_and_clear_logs() {
    if [ -f "$SERVER_DIR/server.PID" ]; then
        echo "- Stopping Node.js server..."
        kill -9 $(cat "$SERVER_DIR/server.PID") && echo -e "  > Node.js server ${GREEN}stopped${NC}." || echo -e "  > ${RED}Failed to stop Node.js server${NC}."
        rm "$SERVER_DIR/server.PID"
    else
        echo "- No Node.js server PID file found. Skipping."
    fi

    if [ -f "$CLIENT_DIR/vite.PID" ]; then
        echo "- Stopping Vite server..."
        kill -9 $(cat "$CLIENT_DIR/vite.PID") && echo -e "  > Vite server ${GREEN}stopped${NC}." || echo -e "  > ${RED}Failed to stop Vite server${NC}."
        rm "$CLIENT_DIR/vite.PID"
    else
        echo "- No Vite server PID file found. Skipping."
    fi

    echo "- Clearing server logs..."
    >"$SERVER_DIR/server.log"
    >"$CLIENT_DIR/client.log"
    echo -e "  > Server logs ${GREEN}cleared${NC}."
}

# Function to show the status of the Mosquitto MQTT broker
function status_app() {
    echo "Checking application status..."
    docker ps -a | grep mosquitto && echo -e "MQTT broker is ${GREEN}running${NC}." || echo -e "MQTT broker is ${RED}not running${NC}."
}

# Function to show logs of the Mosquitto MQTT broker
function log_app() {
    echo "Fetching application logs..."
    docker logs mosquitto
}

# Initial values for flags
verbose=0
simulate=0

# Parse global options before the command
while [[ "$#" -gt 0 ]]; do
    case $1 in
    -h | help)
        show_help
        exit 0
        ;;
    *) break ;;
    esac
done

# Now, $1 should be the command
COMMAND="$1"
shift     # Remove the command from the arguments list
ARGS="$@" # Remaining arguments

# Parse the command
case "$COMMAND" in
install)
    install_dependencies $ARGS
    ;;
start)
    start_app $ARGS
    ;;
stop)
    stop_app $ARGS
    ;;
status)
    status_app $ARGS
    ;;
logs)
    log_app $ARGS
    ;;
test)
    run_tests $ARGS
    ;;
*)
    echo -e "${RED}Invalid command.${NC}\n"
    show_help
    exit 1
    ;;
esac