yuyr
7430896fab
Co-authored-by: sundapeng <sundp@mail.zgclab.edu.cn> Co-authored-by: xuxt <xuxt@zgclab.edu.cn> Reviewed-on: #52
121 lines
3.2 KiB
Bash
121 lines
3.2 KiB
Bash
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
# Shared helper to load Argus build user/group configuration.
|
|
# Usage:
|
|
# source "${PROJECT_ROOT}/scripts/common/build_user.sh"
|
|
# load_build_user
|
|
# echo "$ARGUS_BUILD_UID:$ARGUS_BUILD_GID"
|
|
|
|
ARGUS_BUILD_UID_DEFAULT=2133
|
|
ARGUS_BUILD_GID_DEFAULT=2015
|
|
|
|
shopt -s extglob
|
|
|
|
_ARGUS_BUILD_USER_LOADED="${_ARGUS_BUILD_USER_LOADED:-0}"
|
|
|
|
_argus_build_user_script_dir() {
|
|
local dir
|
|
dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
echo "$dir"
|
|
}
|
|
|
|
argus_project_root() {
|
|
local script_dir
|
|
script_dir="$(_argus_build_user_script_dir)"
|
|
(cd "$script_dir/../.." >/dev/null && pwd)
|
|
}
|
|
|
|
_argus_trim() {
|
|
local value="$1"
|
|
value="${value##+([[:space:]])}"
|
|
value="${value%%+([[:space:]])}"
|
|
printf '%s' "$value"
|
|
}
|
|
|
|
_argus_is_number() {
|
|
[[ "$1" =~ ^[0-9]+$ ]]
|
|
}
|
|
|
|
_argus_read_user_from_files() {
|
|
local uid_out_var="$1" gid_out_var="$2"; shift 2
|
|
local uid_val="$ARGUS_BUILD_UID_DEFAULT" gid_val="$ARGUS_BUILD_GID_DEFAULT"
|
|
local config
|
|
for config in "$@"; do
|
|
if [[ -f "$config" ]]; then
|
|
while IFS= read -r raw_line || [[ -n "$raw_line" ]]; do
|
|
local line key value
|
|
line="${raw_line%%#*}"
|
|
line="$(_argus_trim "${line}")"
|
|
[[ -z "$line" ]] && continue
|
|
if [[ "$line" != *=* ]]; then
|
|
echo "[ARGUS build_user] Ignoring malformed line in $config: $raw_line" >&2
|
|
continue
|
|
fi
|
|
key="${line%%=*}"
|
|
value="${line#*=}"
|
|
key="$(_argus_trim "$key")"
|
|
value="$(_argus_trim "$value")"
|
|
case "$key" in
|
|
UID) uid_val="$value" ;;
|
|
GID) gid_val="$value" ;;
|
|
*) echo "[ARGUS build_user] Unknown key '$key' in $config" >&2 ;;
|
|
esac
|
|
done < "$config"
|
|
break
|
|
fi
|
|
done
|
|
printf -v "$uid_out_var" '%s' "$uid_val"
|
|
printf -v "$gid_out_var" '%s' "$gid_val"
|
|
}
|
|
|
|
load_build_user_profile() {
|
|
local profile="${1:-default}"
|
|
if [[ "$_ARGUS_BUILD_USER_LOADED" == "1" ]]; then
|
|
return 0
|
|
fi
|
|
local project_root uid gid
|
|
project_root="$(argus_project_root)"
|
|
case "$profile" in
|
|
pkg)
|
|
_argus_read_user_from_files uid gid \
|
|
"$project_root/configs/build_user.pkg.conf" \
|
|
"$project_root/configs/build_user.local.conf" \
|
|
"$project_root/configs/build_user.conf"
|
|
;;
|
|
default|*)
|
|
_argus_read_user_from_files uid gid \
|
|
"$project_root/configs/build_user.local.conf" \
|
|
"$project_root/configs/build_user.conf"
|
|
;;
|
|
esac
|
|
|
|
if [[ -n "${ARGUS_BUILD_UID:-}" ]]; then uid="$ARGUS_BUILD_UID"; fi
|
|
if [[ -n "${ARGUS_BUILD_GID:-}" ]]; then gid="$ARGUS_BUILD_GID"; fi
|
|
|
|
if ! _argus_is_number "$uid"; then
|
|
echo "[ARGUS build_user] Invalid UID '$uid'" >&2; return 1
|
|
fi
|
|
if ! _argus_is_number "$gid"; then
|
|
echo "[ARGUS build_user] Invalid GID '$gid'" >&2; return 1
|
|
fi
|
|
export ARGUS_BUILD_UID="$uid"
|
|
export ARGUS_BUILD_GID="$gid"
|
|
_ARGUS_BUILD_USER_LOADED=1
|
|
}
|
|
|
|
load_build_user() {
|
|
local profile="${ARGUS_BUILD_PROFILE:-default}"
|
|
load_build_user_profile "$profile"
|
|
}
|
|
|
|
argus_build_user_args() {
|
|
load_build_user
|
|
printf '%s' "--build-arg ARGUS_BUILD_UID=${ARGUS_BUILD_UID} --build-arg ARGUS_BUILD_GID=${ARGUS_BUILD_GID}"
|
|
}
|
|
|
|
print_build_user() {
|
|
load_build_user
|
|
echo "ARGUS build user: UID=${ARGUS_BUILD_UID} GID=${ARGUS_BUILD_GID}"
|
|
}
|