#!/usr/bin/env bash
# Install managed easyzsh shell layout into a home directory.
# Overwrites: ~/.zshenv ~/.zshrc ~/.zprofile ~/.config/zsh/{env,interactive,login}
# Never overwrites: ~/.config/zsh/local.zsh  ~/.config/zshrc/**  ~/.local/bin/**
#
# Usage:
#   ./deploy.sh
#   ./deploy.sh --home /path/to/home
#   ./deploy.sh --migrate-zprofile   # fold pre-easyzsh ~/.zprofile into local.zsh

set -euo pipefail

ROOT=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
TARGET_HOME="${HOME}"
MIGRATE_ZPROFILE=0

while [[ $# -gt 0 ]]; do
  case "$1" in
    --home)
      TARGET_HOME="$2"
      shift 2
      ;;
    --migrate-zprofile)
      MIGRATE_ZPROFILE=1
      shift
      ;;
    -h|--help)
      sed -n '2,12p' "$0"
      exit 0
      ;;
    *)
      echo "Unknown argument: $1" >&2
      exit 2
      ;;
  esac
done

if [[ ! -d "$TARGET_HOME" ]]; then
  echo "Home does not exist: $TARGET_HOME" >&2
  exit 1
fi

ZSH_CFG="${TARGET_HOME}/.config/zsh"
LOCAL_ZSH="${ZSH_CFG}/local.zsh"
ts=$(date +%Y%m%d-%H%M%S)

backup_file() {
  local f="$1"
  if [[ -e "$f" || -L "$f" ]]; then
    cp -Pp "$f" "${f}.backup.${ts}"
    echo "Backed up $f -> ${f}.backup.${ts}"
  fi
}

file_has_content() {
  [[ -f "$1" ]] && grep -q '[^[:space:]]' "$1"
}

echo "Deploying easyzsh managed shell layout into $TARGET_HOME"

mkdir -p \
  "${ZSH_CFG}/env" \
  "${ZSH_CFG}/interactive" \
  "${ZSH_CFG}/login" \
  "${TARGET_HOME}/.config/zshrc" \
  "${TARGET_HOME}/.cache/zsh" \
  "${TARGET_HOME}/.local/bin"

# Migrate split local/* and legacy paths into ~/.config/zsh/local.zsh once.
if [[ ! -f "$LOCAL_ZSH" ]]; then
  tmp=$(mktemp)
  : >"$tmp"
  if file_has_content "${TARGET_HOME}/.config/zshenv.local"; then
    cat "${TARGET_HOME}/.config/zshenv.local" >>"$tmp"
    printf '\n' >>"$tmp"
  fi
  if [[ -d "${ZSH_CFG}/local" ]]; then
    for part in env.local.zsh login.local.zsh interactive.local.zsh; do
      if file_has_content "${ZSH_CFG}/local/${part}"; then
        cat "${ZSH_CFG}/local/${part}" >>"$tmp"
        printf '\n' >>"$tmp"
      fi
    done
  fi
  if file_has_content "$tmp"; then
    # De-dup consecutive identical non-empty lines lightly by collapsing exact file repeats later if needed.
    cp "$tmp" "$LOCAL_ZSH"
    echo "Migrated machine-local fragments -> ~/.config/zsh/local.zsh"
  fi
  rm -f "$tmp"
fi

# After migration (or if local.zsh already exists), drop obsolete split files.
if [[ -d "${ZSH_CFG}/local" ]]; then
  rm -f \
    "${ZSH_CFG}/local/env.local.zsh" \
    "${ZSH_CFG}/local/login.local.zsh" \
    "${ZSH_CFG}/local/interactive.local.zsh" \
    "${ZSH_CFG}/local/env.local.zsh.example" \
    "${ZSH_CFG}/local/login.local.zsh.example" \
    "${ZSH_CFG}/local/interactive.local.zsh.example"
  rmdir "${ZSH_CFG}/local" 2>/dev/null || true
fi

# Fold a pre-easyzsh ~/.zprofile into local.zsh. Skip if already an easyzsh stub.
if [[ "$MIGRATE_ZPROFILE" -eq 1 && -f "${TARGET_HOME}/.zprofile" ]]; then
  if grep -Fq 'easyzsh stub' "${TARGET_HOME}/.zprofile"; then
    echo "Skipping zprofile migration: already an easyzsh stub"
  elif [[ ! -f "$LOCAL_ZSH" ]]; then
    zprofile_extract=$(mktemp)
    awk '
      /Created by `pipx`/ { skip_pipx=1; next }
      skip_pipx && /local\/bin/ { skip_pipx=0; next }
      skip_pipx { next }
      { print }
    ' "${TARGET_HOME}/.zprofile" >"$zprofile_extract"
    if file_has_content "$zprofile_extract"; then
      cp "$zprofile_extract" "$LOCAL_ZSH"
      echo "Migrated ~/.zprofile body -> ~/.config/zsh/local.zsh"
    fi
    rm -f "$zprofile_extract"
  fi
fi

# Remove a broken local.zsh produced by an earlier stub-body migration.
if [[ -f "$LOCAL_ZSH" ]] && grep -Fq 'local/login.local.zsh' "$LOCAL_ZSH" && ! grep -Eq 'cargo|OrbStack|MacPorts|Kiro|fnm' "$LOCAL_ZSH"; then
  backup_file "$LOCAL_ZSH"
  rm -f "$LOCAL_ZSH"
  echo "Removed broken stub residue from local.zsh"
fi

backup_file "${TARGET_HOME}/.zshenv"
backup_file "${TARGET_HOME}/.zshrc"
backup_file "${TARGET_HOME}/.zprofile"

install -m 644 "${ROOT}/.zshenv" "${TARGET_HOME}/.zshenv"
install -m 644 "${ROOT}/.zshrc" "${TARGET_HOME}/.zshrc"
install -m 644 "${ROOT}/.zprofile" "${TARGET_HOME}/.zprofile"

# Replace managed modules only (never touch local.zsh).
rm -f "${ZSH_CFG}/env"/*.zsh
rm -f "${ZSH_CFG}/interactive"/*.zsh
rm -f "${ZSH_CFG}/login"/*.zsh
install -m 644 "${ROOT}/config/env/"*.zsh "${ZSH_CFG}/env/"
install -m 644 "${ROOT}/config/interactive/"*.zsh "${ZSH_CFG}/interactive/"
if compgen -G "${ROOT}/config/login/"*.zsh > /dev/null; then
  install -m 644 "${ROOT}/config/login/"*.zsh "${ZSH_CFG}/login/"
fi

if [[ -f "${ROOT}/local.zsh.example" ]]; then
  install -m 644 "${ROOT}/local.zsh.example" "${ZSH_CFG}/local.zsh.example"
fi

# Ensure intentional grok shim when the binary tree exists.
if [[ -e "${TARGET_HOME}/.grok/bin/grok" ]]; then
  ln -sfn "${TARGET_HOME}/.grok/bin/grok" "${TARGET_HOME}/.local/bin/grok"
  echo "Ensured ~/.local/bin/grok -> ~/.grok/bin/grok"
fi

echo "Deploy complete."
echo "Verify: env -i HOME=\"$TARGET_HOME\" PATH=/usr/bin:/bin zsh -c '. ~/.zshenv; command -v grok; echo PATH=\$PATH'"