#!/bin/bash

# Capture all arguments as array
Z_CMD=("$@")
Z_FLAG=false
case "${Z_CMD[0]}" in
    docker | d)
        case "${Z_CMD[1]}" in
            restart | r)
                echo "Restarting Docker containers..."
                docker compose down && docker compose up -d
                ;;
            up | u)
                echo "Starting Docker containers..."
                docker compose up -d
                ;;
            down | d)
                echo "Stopping Docker containers..."
                docker compose down
                ;;
            *)
                Z_FLAG=true
                ;;
        esac
        ;;
    caddy | c)
        case "${Z_CMD[1]}" in
            reload | r)
                echo "Reloading Caddy..."
                docker exec -w /etc/caddy caddy sh -c 'caddy fmt --overwrite && caddy reload'
                ;;
            restart)
                echo "Restarting Caddy container..."
                docker compose restart caddy
                ;;
            *)
                Z_FLAG=true
                ;;
        esac
        ;;
    *)
        Z_FLAG=true
        ;;
esac

case $Z_FLAG in
    true)
        echo "Usage: $0 caddy [reload|restart|watch]"
        exit 1
        ;;
esac