mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-03-17 05:37:38 +00:00
142 lines
3.3 KiB
Bash
Executable File
142 lines
3.3 KiB
Bash
Executable File
#!/bin/bash
|
|
# Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
# CK Stop - Stop and remove Docker container
|
|
|
|
set -e
|
|
set -o pipefail
|
|
|
|
# Find script directory and load common utilities
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
source "${SCRIPT_DIR}/common.sh"
|
|
|
|
# Initialize configuration
|
|
PROJECT_ROOT=$(get_project_root "${SCRIPT_DIR}")
|
|
CONTAINER_NAME=$(get_container_name "${PROJECT_ROOT}")
|
|
|
|
# Help message
|
|
show_help() {
|
|
cat << EOF
|
|
CK Stop - Stop and remove Docker container
|
|
|
|
Usage: ck-stop [options] [container_name]
|
|
|
|
Options:
|
|
-h, --help Show this help message
|
|
-f, --force Force stop without confirmation
|
|
--all Stop all CK containers for this user
|
|
|
|
Arguments:
|
|
container_name Optional container name (default: ck_<username>_<branch>)
|
|
|
|
Environment:
|
|
CK_CONTAINER_NAME - Override default container name
|
|
|
|
Examples:
|
|
ck-stop # Stop default container
|
|
ck-stop my_ck_container # Stop specific container
|
|
ck-stop --all # Stop all user's CK containers
|
|
ck-stop --force # Stop without confirmation
|
|
|
|
EOF
|
|
}
|
|
|
|
# Parse arguments
|
|
force=false
|
|
stop_all=false
|
|
|
|
while [[ $# -gt 0 ]]; do
|
|
case $1 in
|
|
-h|--help)
|
|
show_help
|
|
exit 0
|
|
;;
|
|
-f|--force)
|
|
force=true
|
|
shift
|
|
;;
|
|
--all)
|
|
stop_all=true
|
|
shift
|
|
;;
|
|
*)
|
|
CONTAINER_NAME="$1"
|
|
shift
|
|
;;
|
|
esac
|
|
done
|
|
|
|
# Function to stop a single container
|
|
stop_container() {
|
|
local name="$1"
|
|
|
|
if ! container_exists "${name}"; then
|
|
echo "Container '${name}' does not exist"
|
|
return 1
|
|
fi
|
|
|
|
echo "Stopping and removing container '${name}'..."
|
|
docker stop "${name}" 2>/dev/null || true
|
|
docker rm "${name}" 2>/dev/null || true
|
|
echo "Container '${name}' stopped and removed"
|
|
}
|
|
|
|
# Stop all user containers
|
|
if [ "$stop_all" = true ]; then
|
|
username=$(get_username)
|
|
containers=$(docker ps -a --filter "name=ck_${username}_" --format '{{.Names}}')
|
|
|
|
if [ -z "$containers" ]; then
|
|
echo "No CK containers found for user '${username}'"
|
|
exit 0
|
|
fi
|
|
|
|
echo "Found CK containers for user '${username}':"
|
|
echo "$containers"
|
|
echo ""
|
|
|
|
if [ "$force" = false ]; then
|
|
read -p "Stop and remove all these containers? (y/N) " -n 1 -r
|
|
echo ""
|
|
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
|
|
echo "Cancelled"
|
|
exit 0
|
|
fi
|
|
fi
|
|
|
|
echo ""
|
|
while IFS= read -r container; do
|
|
stop_container "$container"
|
|
done <<< "$containers"
|
|
|
|
echo ""
|
|
echo "All containers stopped and removed"
|
|
exit 0
|
|
fi
|
|
|
|
# Stop single container
|
|
if ! container_exists "${CONTAINER_NAME}"; then
|
|
echo "Container '${CONTAINER_NAME}' does not exist"
|
|
exit 0
|
|
fi
|
|
|
|
# Show container info
|
|
if container_is_running "${CONTAINER_NAME}"; then
|
|
echo "Container '${CONTAINER_NAME}' is currently running"
|
|
else
|
|
echo "Container '${CONTAINER_NAME}' exists but is stopped"
|
|
fi
|
|
|
|
# Confirm if not forced
|
|
if [ "$force" = false ]; then
|
|
read -p "Stop and remove container '${CONTAINER_NAME}'? (y/N) " -n 1 -r
|
|
echo ""
|
|
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
|
|
echo "Cancelled"
|
|
exit 0
|
|
fi
|
|
fi
|
|
|
|
stop_container "${CONTAINER_NAME}"
|