#!/usr/bin/env sh
set -eu

if [ "$#" -ne 5 ]; then
  echo "用法: $0 <环境> <镜像仓库> <镜像标签> <服务> <环境文件>" >&2
  exit 2
fi

DEPLOY_ENV="$1"
IMAGE_REPOSITORY="$2"
IMAGE_TAG="$3"
SERVICE="$4"
ENV_FILE="$5"
COMPOSE_FILE="deploy/docker-compose.yml"
IMAGE="$IMAGE_REPOSITORY/$SERVICE:$IMAGE_TAG"
REGISTRY_HOST="${IMAGE_REPOSITORY%%/*}"
PROJECT_NAME="firstleg-$DEPLOY_ENV"

case "$DEPLOY_ENV" in
  development|test) ;;
  *)
    echo "直接镜像传输仅支持 development/test，当前: $DEPLOY_ENV" >&2
    exit 2
    ;;
esac

case "$SERVICE" in
  *[!A-Za-z0-9_-]*|'')
    echo "非法服务名: $SERVICE" >&2
    exit 2
    ;;
esac

for file in "$ENV_FILE" "$COMPOSE_FILE"; do
  if [ ! -f "$file" ]; then
    echo "必需文件不存在: $file" >&2
    exit 2
  fi
done

env_file_has_key() {
  grep -q "^$1=" "$ENV_FILE"
}

env_file_value() {
  sed -n "s/^$1=//p" "$ENV_FILE" | tail -n 1 | tr -d '\r'
}

if [ "$SERVICE" = "finance" ]; then
  for key in FINANCE_DB_URL FINANCE_DB_USERNAME FINANCE_DB_PASSWORD; do
    if ! env_file_has_key "$key" || [ -z "$(env_file_value "$key")" ]; then
      echo "部署 finance 服务时，环境文件必须包含非空 $key" >&2
      exit 2
    fi
  done
fi

if [ "$SERVICE" = "file" ]; then
  if ! env_file_has_key ROCKETMQ_NAME_SERVER || [ -z "$(env_file_value ROCKETMQ_NAME_SERVER)" ]; then
    echo "部署 file 服务时，环境文件必须包含非空 ROCKETMQ_NAME_SERVER" >&2
    exit 2
  fi
  if ! env_file_has_key MQ_TOPIC_PREFIX; then
    echo "部署 file 服务时，环境文件必须显式包含 MQ_TOPIC_PREFIX" >&2
    exit 2
  fi
  MQ_PREFIX=$(env_file_value MQ_TOPIC_PREFIX)
  if [ "$DEPLOY_ENV" = "test" ] && [ "$MQ_PREFIX" != "test-" ]; then
    echo "测试环境部署 file 服务时 MQ_TOPIC_PREFIX 必须为 test-" >&2
    exit 2
  fi
  if [ "$DEPLOY_ENV" = "development" ] && { [ -z "$MQ_PREFIX" ] || [ "$MQ_PREFIX" = "test-" ]; }; then
    echo "开发环境部署 file 服务时 MQ_TOPIC_PREFIX 必须非空且不能使用 test-" >&2
    exit 2
  fi
fi

: "${REGISTRY_USER:?REGISTRY_USER 未设置}"
: "${REGISTRY_PASSWORD:?REGISTRY_PASSWORD 未设置}"

local_registry_logout() {
  docker logout "$REGISTRY_HOST" >/dev/null 2>&1 || true
}

deploy_local() {
  printf '%s' "$REGISTRY_PASSWORD" | docker login -u "$REGISTRY_USER" --password-stdin "$REGISTRY_HOST"
  trap local_registry_logout EXIT HUP INT TERM
  docker pull "$IMAGE"
  export DEPLOY_ENV IMAGE_REPOSITORY IMAGE_TAG
  docker compose \
    --project-name "$PROJECT_NAME" \
    --env-file "$ENV_FILE" \
    -f "$COMPOSE_FILE" \
    up -d --remove-orphans --pull never "$SERVICE"
  docker compose \
    --project-name "$PROJECT_NAME" \
    --env-file "$ENV_FILE" \
    -f "$COMPOSE_FILE" \
    ps "$SERVICE"
  local_registry_logout
  trap - EXIT HUP INT TERM
}

if [ "${TARGET_IS_LOCAL:-false}" = "true" ]; then
  echo "镜像已在目标服务器本机，直接部署: $IMAGE"
  deploy_local
  exit 0
fi

SSH_TARGET="${TARGET_SSH_TARGET:?TARGET_SSH_TARGET 未设置}"
SSH_KEY="${TARGET_SSH_KEY:?TARGET_SSH_KEY 未设置}"
SSH_KNOWN_HOSTS="${TARGET_SSH_KNOWN_HOSTS:?TARGET_SSH_KNOWN_HOSTS 未设置}"
REMOTE_DIR="${TARGET_DEPLOY_DIR:?TARGET_DEPLOY_DIR 未设置}"

for file in "$SSH_KEY" "$SSH_KNOWN_HOSTS"; do
  if [ ! -f "$file" ]; then
    echo "SSH 文件不存在: $file" >&2
    exit 2
  fi
done

run_ssh() {
  ssh -i "$SSH_KEY" \
    -o BatchMode=yes \
    -o Compression=no \
    -o StrictHostKeyChecking=yes \
    -o UserKnownHostsFile="$SSH_KNOWN_HOSTS" \
    -o ConnectTimeout=10 \
    "$SSH_TARGET" "$@"
}

run_scp() {
  scp -i "$SSH_KEY" \
    -o BatchMode=yes \
    -o Compression=no \
    -o StrictHostKeyChecking=yes \
    -o UserKnownHostsFile="$SSH_KNOWN_HOSTS" \
    -o ConnectTimeout=10 \
    "$@"
}

run_ssh install -d -m 750 "$REMOTE_DIR"

SUFFIX="${IMAGE_TAG}.$$.${SERVICE}"
REMOTE_ENV_TMP="$REMOTE_DIR/.env.$SUFFIX.tmp"
REMOTE_COMPOSE_TMP="$REMOTE_DIR/docker-compose.yml.$SUFFIX.tmp"
run_scp "$ENV_FILE" "$SSH_TARGET:$REMOTE_ENV_TMP"
run_scp "$COMPOSE_FILE" "$SSH_TARGET:$REMOTE_COMPOSE_TMP"
run_ssh sh -s -- "$REMOTE_DIR" "$REMOTE_ENV_TMP" "$REMOTE_COMPOSE_TMP" <<'REMOTE_PREPARE'
set -eu
remote_dir="$1"
env_tmp="$2"
compose_tmp="$3"
chmod 600 "$env_tmp"
chmod 644 "$compose_tmp"
mv -f "$env_tmp" "$remote_dir/.env"
mv -f "$compose_tmp" "$remote_dir/docker-compose.yml"
REMOTE_PREPARE

remote_registry_logout() {
  run_ssh docker logout "$REGISTRY_HOST" >/dev/null 2>&1 || true
}

echo "目标服务器从共享仓库拉取增量镜像: $IMAGE"
printf '%s' "$REGISTRY_PASSWORD" | run_ssh docker login -u "$REGISTRY_USER" --password-stdin "$REGISTRY_HOST"
trap remote_registry_logout EXIT HUP INT TERM
run_ssh docker pull "$IMAGE"

run_ssh sh -s -- "$REMOTE_DIR" "$DEPLOY_ENV" "$IMAGE_REPOSITORY" "$IMAGE_TAG" "$SERVICE" <<'REMOTE_DEPLOY'
set -eu
remote_dir="$1"
deploy_env="$2"
image_repository="$3"
image_tag="$4"
service="$5"
cd "$remote_dir"
export DEPLOY_ENV="$deploy_env"
export IMAGE_REPOSITORY="$image_repository"
export IMAGE_TAG="$image_tag"
docker compose \
  --project-name "firstleg-$deploy_env" \
  --env-file .env \
  -f docker-compose.yml \
  up -d --remove-orphans --pull never "$service"
docker compose \
  --project-name "firstleg-$deploy_env" \
  --env-file .env \
  -f docker-compose.yml \
  ps "$service"
REMOTE_DEPLOY

remote_registry_logout
trap - EXIT HUP INT TERM