#!/bin/bash
# manage.sh - Manage discord-stock-bot on Unraid

UNRAID_HOST="${UNRAID_HOST:-root@192.168.2.61}"
REMOTE_PATH="/mnt/user/docker/custom-dockers/discord-stock-bot"

CMD="$1"

run_remote() {
    ssh ${UNRAID_HOST} "cd ${REMOTE_PATH} && $1"
}

case "$CMD" in
    logs)
        run_remote "docker compose logs -f"
        ;;
    status)
        run_remote "docker compose ps"
        ;;
    restart)
        run_remote "docker compose restart"
        echo "Bot restarted"
        ;;
    stop)
        run_remote "docker compose down"
        echo "Bot stopped"
        ;;
    start)
        run_remote "docker compose up -d"
        echo "Bot started"
        ;;
    rebuild)
        run_remote "docker compose up -d --build"
        echo "Bot rebuilt and started"
        ;;
    shell)
        ssh -t ${UNRAID_HOST} "cd ${REMOTE_PATH} && bash"
        ;;
    *)
        echo "Usage: $0 {logs|status|restart|stop|start|rebuild|shell}"
        echo ""
        echo "Commands:"
        echo "  logs     - View live logs"
        echo "  status   - Show container status"
        echo "  restart  - Restart the bot"
        echo "  stop     - Stop the bot"
        echo "  start    - Start the bot"
        echo "  rebuild  - Rebuild and restart the bot"
        echo "  shell    - SSH into Unraid at bot directory"
        exit 1
        ;;
esac