entrypoint.sh

#!/bin/bash
set -euo pipefail

INIT_DIR="/docker-entrypoint-init.d"

EMULATOR_TYPE=""
PORT=""
ADDITIONAL_ARGS=()

# Define emulator types that require beta
BETA_EMULATORS=("bigtable" "datastore" "pubsub")

# Define emulator types that have env-init
ENV_INIT_EMULATORS=("bigtable" "datastore" "pubsub")

# check if emulator type requires beta
function is_beta_emulator() {
  local emulator="$1"
  for beta_emulator in "${BETA_EMULATORS[@]}"; do
    if [[ "$emulator" == "$beta_emulator" ]]; then
      return 0
    fi
  done
  return 1
}

# check if emulator type has env-init
function has_env_init() {
  local emulator="$1"
  for env_init_emulator in "${ENV_INIT_EMULATORS[@]}"; do
    if [[ "$emulator" == "$env_init_emulator" ]]; then
      return 0
    fi
  done
  return 1
}

# show_usage displays the usage information
function show_usage() {
  echo "Usage: $0 --emulator=<emulator_type> --port=<port> [additional args...]"
  echo "Example: $0 --emulator=pubsub --port=8085 --project=my-project"
  exit 1
}

# check arguments
if [ "$#" -eq 0 ]; then
  show_usage
fi

# parse command-line arguments
while [[ $# -gt 0 ]]; do
  case $1 in
    --emulator=*)
      EMULATOR_TYPE="${1#*=}"
      shift
      ;;
    --port=*)
      PORT="${1#*=}"
      shift
      ;;
    -h|--help)
      show_usage
      ;;
    *)
      # 残りの引数はすべて追加引数として扱う
      ADDITIONAL_ARGS+=("$1")
      shift
      ;;
  esac
done

# check if required arguments are provided
if [ -z "$EMULATOR_TYPE" ] || [ -z "$PORT" ]; then
  echo "Error: Both --emulator and --port are required"
  show_usage
fi

echo "Emulator type: $EMULATOR_TYPE"
echo "Port: $PORT"

# Construct the gcloud command based on emulator type
if is_beta_emulator "$EMULATOR_TYPE"; then
  COMMAND=("gcloud" "beta" "emulators" "$EMULATOR_TYPE" "start" "--host-port=0.0.0.0:$PORT")
else
  COMMAND=("gcloud" "emulators" "$EMULATOR_TYPE" "start" "--host-port=0.0.0.0:$PORT")
fi

# Add any additional arguments
if [ "${#ADDITIONAL_ARGS[@]}" -gt 0 ]; then
  COMMAND+=("${ADDITIONAL_ARGS[@]}")
fi

echo "Executing command: ${COMMAND[*]}"

# Start emulator in background and redirect output
echo "Starting emulator..."
"${COMMAND[@]}" > /proc/1/fd/1 2>&1 &
EMULATOR_PID=$!

# Wait a bit for the emulator to start properly
sleep 5
echo "Emulator started with PID: $EMULATOR_PID"

# Wait for emulator to be ready
echo "Waiting for emulator to be ready..."
until timeout 1 bash -c "</dev/tcp/localhost/${PORT}"; do
  echo "Waiting for ${EMULATOR_TYPE} emulator to be ready..."
  sleep 5
done
echo "Emulator is ready."

# Set environment variables for connecting to the emulator
echo "Setting up environment variables..."
if has_env_init "$EMULATOR_TYPE"; then
  if is_beta_emulator "$EMULATOR_TYPE"; then
    eval "$(gcloud beta emulators "${EMULATOR_TYPE}" env-init)"
  else
    eval "$(gcloud emulators "${EMULATOR_TYPE}" env-init)"
  fi
  echo "Environment variables set."
fi

# Firestore emulator does not support the env-init command, but requires manual environment variable configuration.
# Therefore, we need to set them manually.
if [ "$EMULATOR_TYPE" == "firestore" ]; then
  export FIRESTORE_EMULATOR_HOST="localhost:$PORT"
  export DATASTORE_EMULATOR_HOST="localhost:$PORT"
  echo "Environment variables set."
fi

# Run ready.d scripts
READY_DIR="${INIT_DIR}/ready.d"
if [ -d "$READY_DIR" ] && [ -n "$(ls -A $READY_DIR 2>/dev/null)" ]; then
  echo "Running ready scripts..."

  for f in "$READY_DIR"/*.sh; do
    if [ -f "$f" ] && [ -x "$f" ]; then
      echo "Running hook: $f"
      "$f"
    elif [ -f "$f" ]; then
      echo "Running hook: $f (with bash)"
      bash "$f"
    fi
  done

  for f in "$READY_DIR"/*.py; do
    if [ -f "$f" ]; then
      echo "Running hook: $f (with python)"
      uv run "$f"
    fi
  done

  echo "Ready scripts completed."
fi

# Create a file to indicate initialization is complete
touch /tmp/init-completed
echo "Initialization completed marker created at /tmp/init-completed"

# Keep the container running by waiting for the emulator process
wait $EMULATOR_PID