Solve the CTAD regression & add up the Shell file for the docker management in testing (#3634)

* Finished the work * Fix the pipeline
2026-05-05 14:11:29 +00:00 · 2026-01-26 10:29:28 -08:00
parent 834642202c
commit 3900e1e7ce
12 changed files with 1150 additions and 6 deletions
--- a/script/tools/ck-stop
+++ b/script/tools/ck-stop
@@ -0,0 +1,141 @@
+#!/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}"