#!/bin/bash

set -e

usage() {
    echo "Usage: restart.sh [sdm|switch]"
    echo "  sdm     - Restart 'sdm' service"
    echo "  switch  - Restart 'switch' service"
    echo "  (none)  - Restart both 'sdm' and 'switch' service"
    echo "  -h      - Show this help message"
}

if [[ $# -eq 0 ]]; then
    read -p "Do you want to restart both 'sdm' and 'switch' services? (y/n) [default: y]: " choice
    choice="${choice:-y}"

    if [[ $choice == "y" ]]; then
        echo "Restarting 'sdm' service..."
        docker restart sdm
        echo "Restarting 'switch' service..."
        docker restart switch
    else
        echo "No services restarted."
    fi

elif [[ $# -eq 1 ]]; then
    if [[ $1 == "sdm" ]]; then
        echo "Restarting 'sdm' service..."
        docker restart sdm
    elif [[ $1 == "switch" ]]; then
        echo "Restarting 'switch' service..."
        docker restart switch
    elif [[ $1 == "-h" ]]; then
        usage
    else
        echo "Invalid argument: $1"
        usage
    fi

else
    echo "Invalid number of arguments"
    usage
fi