init.sh

#!/usr/bin/env bash
set -euo pipefail

# Load configuration (USER_NAME, IMAGE_NAME, CONTAINER_NAME)
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
# shellcheck disable=SC1091
source "${SCRIPT_DIR}/config.env"

# Fixed paths inside the container
HOST_ZSHRC="${SCRIPT_DIR}/zshrc"
IN_CONTAINER_HOME="/home/${USER_NAME}"
IN_CONTAINER_ZSHRC="${IN_CONTAINER_HOME}/.zshrc"
IN_CONTAINER_ZSHENV="${IN_CONTAINER_HOME}/.zshenv"

echo "[*] init: container=${CONTAINER_NAME} user=${USER_NAME} image=${IMAGE_NAME}"

# -----------------------------------------------------------------------------
# 0) Ensure the container exists and is running
# -----------------------------------------------------------------------------
if docker inspect "${CONTAINER_NAME}" >/dev/null 2>&1; then
  docker start "${CONTAINER_NAME}" >/dev/null || true
else
  echo "[!] ERROR: container '${CONTAINER_NAME}' not found. Run ./setup.sh first." >&2
  exit 1
fi

# -----------------------------------------------------------------------------
# 1) Set password for USER_NAME (executed as root)
#    Prefer DOCKER_ENV_PASSWORD if provided by setup.sh
# -----------------------------------------------------------------------------
P="${DOCKER_ENV_PASSWORD:-}"
if [ -z "${P}" ]; then
  read -r -s -p "Password (for ${USER_NAME}): " P
  echo
fi
if [ -z "${P}" ]; then
  echo "[!] ERROR: empty password" >&2
  exit 1
fi

printf "%s:%s\n" "${USER_NAME}" "${P}" | docker exec -i -u root "${CONTAINER_NAME}" chpasswd
unset P
echo "[*] init: password set"

# -----------------------------------------------------------------------------
# 2) HOST_LABEL -> write it to ~/.zshenv (prompt uses: ${USER_NAME}@${HOST_LABEL})
#    Prefer DOCKER_ENV_HOST_LABEL if provided by setup.sh
# -----------------------------------------------------------------------------
HOST_LABEL="${DOCKER_ENV_HOST_LABEL:-}"
if [ -z "${HOST_LABEL}" ]; then
  read -r -p "HOST_LABEL (if set to 'dev', prompt becomes '${USER_NAME}@dev'): " HOST_LABEL
fi
if [ -z "${HOST_LABEL}" ]; then
  echo "[!] ERROR: empty HOST_LABEL" >&2
  exit 1
fi

# Minimal escaping for backslashes and double quotes
HOST_LABEL_ESC="${HOST_LABEL//\\/\\\\}"
HOST_LABEL_ESC="${HOST_LABEL_ESC//\"/\\\"}"

docker exec -i -u root "${CONTAINER_NAME}" sh -lc "cat > '${IN_CONTAINER_ZSHENV}' <<EOF
export HOST_LABEL=\"${HOST_LABEL_ESC}\"
EOF
chown 1100:0 '${IN_CONTAINER_ZSHENV}'
chmod 600 '${IN_CONTAINER_ZSHENV}'"

echo "[*] init: wrote ${IN_CONTAINER_ZSHENV} (prompt example: ${USER_NAME}@${HOST_LABEL})"

# -----------------------------------------------------------------------------
# 3) Copy .zshrc template into the container
# -----------------------------------------------------------------------------
if [ ! -f "${HOST_ZSHRC}" ]; then
  echo "[!] ERROR: ${HOST_ZSHRC} not found" >&2
  exit 1
fi

docker cp "${HOST_ZSHRC}" "${CONTAINER_NAME}:${IN_CONTAINER_ZSHRC}"
docker exec -u root "${CONTAINER_NAME}" sh -lc "
  chown 1100:0 '${IN_CONTAINER_ZSHRC}' &&
  chmod 644 '${IN_CONTAINER_ZSHRC}'
"

echo "[*] init: installed ${IN_CONTAINER_ZSHRC}"

# -----------------------------------------------------------------------------
# 4) Install oh-my-zsh if missing (executed as USER_NAME)
# -----------------------------------------------------------------------------
docker exec -u "${USER_NAME}" "${CONTAINER_NAME}" sh -lc '
  if [ ! -f "$HOME/.oh-my-zsh/oh-my-zsh.sh" ]; then
    echo "[*] init: installing oh-my-zsh..."
    RUNZSH=no CHSH=no KEEP_ZSHRC=yes \
      sh -c "$(curl -fsSL https://raw.githubusercontent.com/ohmyzsh/ohmyzsh/master/tools/install.sh)"
  else
    echo "[*] init: oh-my-zsh already installed"
  fi
'

# -----------------------------------------------------------------------------
# 5) Install zsh plugins if missing (executed as USER_NAME)
# -----------------------------------------------------------------------------
docker exec -u "${USER_NAME}" "${CONTAINER_NAME}" sh -lc '
  ZSH_CUSTOM="${ZSH_CUSTOM:-$HOME/.oh-my-zsh/custom}"
  mkdir -p "$ZSH_CUSTOM/plugins"

  if [ ! -d "$ZSH_CUSTOM/plugins/zsh-syntax-highlighting" ]; then
    echo "[*] init: installing plugin: zsh-syntax-highlighting"
    git clone https://github.com/zsh-users/zsh-syntax-highlighting.git \
      "$ZSH_CUSTOM/plugins/zsh-syntax-highlighting"
  fi

  if [ ! -d "$ZSH_CUSTOM/plugins/zsh-autosuggestions" ]; then
    echo "[*] init: installing plugin: zsh-autosuggestions"
    git clone https://github.com/zsh-users/zsh-autosuggestions \
      "$ZSH_CUSTOM/plugins/zsh-autosuggestions"
  fi
'

echo "[*] init: done"