build_image.sh

#!/bin/bash

# build_image.sh
IMG_NAME="emb_server" #


# 获取架构
ARCH=$(uname -m)

case "$ARCH" in
  aarch64)
    if [[ -f "/etc/nv_tegra_release" ]] || grep -qi "nvidia" /proc/device-tree/model 2>/dev/null; then
      ARCH_TAG="jet"
    else
      ARCH_TAG="arm"
    fi
    ;;
  x86_64)
    ARCH_TAG="amd"
    ;;
  *)
    ARCH_TAG="unknown"
    ;;
esac


while [[ $# -gt 0 ]]; do
  case $1 in
    --profile)
      PROFILE="$2"
      shift 2
      ;;
    *)
      echo "Unknown option: $1"
      exit 1
      ;;
  esac
done

# 根据 PROFILE 生成 PROFILE_TAG
case "$PROFILE" in
  Dockerfile)
    PROFILE_TAG="${ARCH_TAG}"
    ;;
  Dockerfile_cu124)
    PROFILE_TAG="${ARCH_TAG}_cu124"
    ;;
  Dockerfile_cu128)
    PROFILE_TAG="${ARCH_TAG}_cu128"
    ;;
  Dockerfile_l4t)
    PROFILE_TAG="${ARCH_TAG}_l4t"
    ;;
  *)
    echo "Unsupported profile: $PROFILE"
    exit 1
    ;;
esac

BUILD_ARGS=()
if [[ -n "${PROXY:-}" ]]; then
  echo "Using PROXY=${PROXY}"
  BUILD_ARGS+=(--build-arg "PROXY=${PROXY}")
fi

if [[ -n "${USE_OLD_TRANSFORMERS:-}" ]]; then
  echo "Using USE_OLD_TRANSFORMERS=${USE_OLD_TRANSFORMERS}"
  BUILD_ARGS+=(--build-arg "USE_OLD_TRANSFORMERS=${USE_OLD_TRANSFORMERS}")
fi

# 获取日期
DATE=$(date +%Y%m%d)

# 检查 version.txt
if [[ -f "VERSION" ]]; then
    VERSION=$(cat VERSION | tr -d ' \t\n\r')
    TAG="${PROFILE_TAG}_${VERSION}_${DATE}"
    echo "Using version from VERSION: ${VERSION}"
else
    TAG="${PROFILE_TAG}_${DATE}"
    echo "No VERSION file found, using default tag format."
fi


# 构建并推送镜像
docker build \
    "${BUILD_ARGS[@]}" \
    -t emb_server \
    -f $PROFILE .
docker tag emb_server swr.cn-southwest-2.myhuaweicloud.com/ictrek/${IMG_NAME}:${TAG}

docker push swr.cn-southwest-2.myhuaweicloud.com/ictrek/${IMG_NAME}:${TAG}