feat: add scripts/manage.sh

This commit is contained in:
cătălin 2024-07-14 19:37:37 +02:00
commit a4b6a8c186
No known key found for this signature in database
3 changed files with 165 additions and 30 deletions

134
scripts/manage.sh Executable file
View file

@ -0,0 +1,134 @@
#!/bin/sh
print_usage() {
echo "Usage:"
echo " $0 <up|down> <service_name|all> [-d|--daemon]"
echo " $0 sh <service_name> <subservice_name> [--shell <shell_name>]"
echo " $0 logs <service_name> [<subservice_name>] [-f|--follow]"
exit 1
}
# Check if correct number of arguments is provided
if [ $# -lt 2 ]; then
print_usage
fi
action=$1
service=$2
daemon_flag=""
shell="sh"
follow_flag=""
case "$action" in
up|down)
if [ $# -gt 3 ]; then
print_usage
fi
if [ $# -eq 3 ]; then
if [ "$3" = "-d" ] || [ "$3" = "--daemon" ]; then
daemon_flag="-d"
else
echo "Invalid flag. Use '-d' or '--daemon' for detached mode."
exit 1
fi
fi
;;
sh)
if [ $# -lt 3 ]; then
print_usage
fi
subservice=$3
shift 3
while [ $# -gt 0 ]; do
case "$1" in
--shell)
if [ -z "$2" ]; then
echo "Shell name is missing after --shell flag."
exit 1
fi
shell="$2"
shift 2
;;
*)
echo "Unknown option: $1"
print_usage
;;
esac
done
;;
logs)
subservice=""
shift 2
while [ $# -gt 0 ]; do
case "$1" in
-f|--follow)
follow_flag="--follow"
shift
;;
*)
if [ -z "$subservice" ]; then
subservice="$1"
else
echo "Unknown option: $1"
print_usage
fi
shift
;;
esac
done
;;
*)
echo "Invalid action. Use 'up', 'down', 'sh', or 'logs'."
exit 1
;;
esac
execute_docker_compose() {
dir=$1
cmd=$2
echo "Executing in $dir: docker compose $cmd $daemon_flag"
cd "$dir" && docker compose "$cmd" $daemon_flag
cd - > /dev/null || exit
}
execute_docker_shell() {
dir=$1
subservice=$2
echo "Executing in $dir: docker compose exec $subservice $shell"
cd "$dir" && docker compose exec "$subservice" "$shell"
cd - > /dev/null || exit
}
execute_docker_logs() {
dir=$1
subservice=$2
echo "Executing in $dir: docker compose logs $follow_flag $subservice"
cd "$dir" && docker compose logs $follow_flag "$subservice"
cd - > /dev/null || exit
}
if [ "$action" = "sh" ]; then
if [ ! -d "$service" ]; then
echo "Service directory '$service' not found."
exit 1
fi
execute_docker_shell "$service" "$subservice"
elif [ "$action" = "logs" ]; then
if [ ! -d "$service" ]; then
echo "Service directory '$service' not found."
exit 1
fi
execute_docker_logs "$service" "$subservice"
elif [ "$service" = "all" ]; then
for dir in */; do
if [ -f "${dir}docker-compose.yml" ] || [ -f "${dir}docker-compose.yaml" ]; then
execute_docker_compose "$dir" "$action"
fi
done
else
if [ ! -d "$service" ]; then
echo "Service directory '$service' not found."
exit 1
fi
execute_docker_compose "$service" "$action"
fi