mirror of
https://github.com/csunny/DB-GPT.git
synced 2026-07-17 01:58:47 +00:00
91 lines
2.5 KiB
Bash
Executable File
91 lines
2.5 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# common.sh - Shared utility functions for DB-GPT installer
|
|
# shellcheck disable=SC2034
|
|
|
|
# ── Colors ────────────────────────────────────────────────────────────────────
|
|
readonly COLOR_RESET='\033[0m'
|
|
readonly COLOR_RED='\033[1;31m'
|
|
readonly COLOR_GREEN='\033[1;32m'
|
|
readonly COLOR_YELLOW='\033[1;33m'
|
|
readonly COLOR_BLUE='\033[1;34m'
|
|
readonly COLOR_CYAN='\033[1;36m'
|
|
|
|
# ── Logging ───────────────────────────────────────────────────────────────────
|
|
info() {
|
|
printf "${COLOR_BLUE}[INFO]${COLOR_RESET} %s\n" "$*"
|
|
}
|
|
|
|
warn() {
|
|
printf "${COLOR_YELLOW}[WARN]${COLOR_RESET} %s\n" "$*"
|
|
}
|
|
|
|
error() {
|
|
printf "${COLOR_RED}[ERROR]${COLOR_RESET} %s\n" "$*" >&2
|
|
}
|
|
|
|
success() {
|
|
printf "${COLOR_GREEN}[OK]${COLOR_RESET} %s\n" "$*"
|
|
}
|
|
|
|
prompt_input() {
|
|
local prompt="$1"
|
|
local __resultvar="$2"
|
|
local answer=""
|
|
|
|
if [[ -r /dev/tty ]]; then
|
|
if read -r -p "${prompt}" answer < /dev/tty 2>/dev/null; then
|
|
printf -v "${__resultvar}" '%s' "${answer}"
|
|
return
|
|
fi
|
|
fi
|
|
|
|
if [[ -t 0 ]]; then
|
|
read -r -p "${prompt}" answer || die "Failed to read input from stdin."
|
|
else
|
|
die "Interactive input requires a terminal. Re-run with explicit flags such as --profile <name> or --yes."
|
|
fi
|
|
|
|
printf -v "${__resultvar}" '%s' "${answer}"
|
|
}
|
|
|
|
die() {
|
|
error "$*"
|
|
exit 1
|
|
}
|
|
|
|
# ── Command helpers ───────────────────────────────────────────────────────────
|
|
|
|
# Check that a command exists, die if not.
|
|
require_cmd() {
|
|
local cmd="$1"
|
|
if ! command -v "${cmd}" >/dev/null 2>&1; then
|
|
die "Required command not found: ${cmd}. Please install it first."
|
|
fi
|
|
}
|
|
|
|
# Run a command with a description printed before it.
|
|
run() {
|
|
local desc="$1"
|
|
shift
|
|
info "${desc}"
|
|
if ! "$@"; then
|
|
die "Command failed: $*"
|
|
fi
|
|
}
|
|
|
|
# ── Interactive helpers ───────────────────────────────────────────────────────
|
|
|
|
# Ask a yes/no question. Respects the global YES flag for non-interactive use.
|
|
confirm() {
|
|
local prompt="$1"
|
|
|
|
if [[ "${YES:-false}" == "true" ]]; then
|
|
return 0
|
|
fi
|
|
|
|
local answer
|
|
prompt_input "${prompt} [y/N]: " answer
|
|
[[ "${answer}" =~ ^[Yy]$ ]]
|
|
}
|
|
|