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

INSTALL_DIR="${GIT_VIBE_HOME:-$HOME/.git-vibe}"
BIN_DIR="${INSTALL_DIR}/bin"
HOOK_DIR="${INSTALL_DIR}/hooks"
REF="${GIT_VIBE_REF:-main}"
REPO_SLUG="${GIT_VIBE_REPO:-sailscastshq/git-vibe}"
RAW_BASE="https://raw.githubusercontent.com/${REPO_SLUG}/${REF}"
SCRIPT_PATH="${BASH_SOURCE[0]:-$0}"
SCRIPT_DIR="$(cd "$(dirname "${SCRIPT_PATH}")" && pwd 2>/dev/null || pwd)"
PATH_LINE="export PATH=\"${BIN_DIR}:\$PATH\""
PATH_MARKER_BEGIN="# >>> git-vibe >>>"
PATH_MARKER_END="# <<< git-vibe <<<"

download() {
  local remote_path target
  remote_path="$1"
  target="$2"
  curl -fsSL "${RAW_BASE}/${remote_path}" -o "${target}"
}

shell_integration_block() {
  cat <<EOF
${PATH_MARKER_BEGIN}
${PATH_LINE}

git() {
  if [ "\$1" = "vc" ]; then
    __git_vibe_output="\$(command git vibe code "\${@:2}" --shell-output 2>&1)"
    __git_vibe_status=\$?
    __git_vibe_path="\$(printf '%s\n' "\$__git_vibe_output" | sed -n 's/^__GIT_VIBE_CHDIR__=//p' | tail -n 1)"
    printf '%s\n' "\$__git_vibe_output" | sed '/^__GIT_VIBE_CHDIR__=/d'

    if [ "\$__git_vibe_status" -eq 0 ] && [ -n "\$__git_vibe_path" ]; then
      builtin cd "\$__git_vibe_path" || return "\$__git_vibe_status"
    fi

    return "\$__git_vibe_status"
  fi

  if [ "\$1" = "vibe" ] && { [ "\$2" = "code" ] || [ "\$2" = "start" ] || [ "\$2" = "enter" ] || [ "\$2" = "finish" ]; }; then
    __git_vibe_output="\$(command git "\$@" --shell-output 2>&1)"
    __git_vibe_status=\$?
    __git_vibe_path="\$(printf '%s\n' "\$__git_vibe_output" | sed -n 's/^__GIT_VIBE_CHDIR__=//p' | tail -n 1)"
    printf '%s\n' "\$__git_vibe_output" | sed '/^__GIT_VIBE_CHDIR__=/d'

    if [ "\$__git_vibe_status" -eq 0 ] && [ -n "\$__git_vibe_path" ]; then
      builtin cd "\$__git_vibe_path" || return "\$__git_vibe_status"
    fi

    return "\$__git_vibe_status"
  fi

  command git "\$@"
}
${PATH_MARKER_END}
EOF
}

detect_profile() {
  local shell_name
  shell_name="$(basename "${SHELL:-}")"

  case "${shell_name}" in
    zsh)
      printf '%s\n' "${HOME}/.zshrc"
      ;;
    bash)
      if [[ -f "${HOME}/.bashrc" || ! -f "${HOME}/.bash_profile" ]]; then
        printf '%s\n' "${HOME}/.bashrc"
      else
        printf '%s\n' "${HOME}/.bash_profile"
      fi
      ;;
    *)
      printf '%s\n' "${HOME}/.profile"
      ;;
  esac
}

ensure_shell_integration() {
  local profile temp
  profile="$(detect_profile)"
  touch "${profile}"
  temp="$(mktemp)"

  awk -v begin="${PATH_MARKER_BEGIN}" -v end="${PATH_MARKER_END}" '
    $0 == begin { skipping = 1; next }
    $0 == end { skipping = 0; next }
    !skipping { print }
  ' "${profile}" > "${temp}"

  printf '\n' >> "${temp}"
  shell_integration_block >> "${temp}"
  mv "${temp}" "${profile}"

  printf '%s\n' "${profile}"
}

mkdir -p "${BIN_DIR}" "${HOOK_DIR}"

if [[ -f "${SCRIPT_DIR}/bin/git-vibe" && -f "${SCRIPT_DIR}/VERSION" ]]; then
  cp "${SCRIPT_DIR}/bin/git-vibe" "${BIN_DIR}/git-vibe"
  cp "${SCRIPT_DIR}/VERSION" "${INSTALL_DIR}/VERSION"
else
  download "bin/git-vibe" "${BIN_DIR}/git-vibe"
  download "VERSION" "${INSTALL_DIR}/VERSION"
fi

chmod +x "${BIN_DIR}/git-vibe"

for hook in pre-commit commit-msg pre-push; do
  cat > "${HOOK_DIR}/${hook}" <<EOF
#!/usr/bin/env bash
exec "${BIN_DIR}/git-vibe" hook ${hook} "\$@"
EOF
  chmod +x "${HOOK_DIR}/${hook}"
done

git config --global core.hooksPath "${HOOK_DIR}"
git config --global vibe.baseBranch main
git config --global vibe.branchPrefix feat/
git config --global vibe.worktreeRoot ../.vibe
git config --global alias.vibe "!${BIN_DIR}/git-vibe"
git config --global alias.vc "!${BIN_DIR}/git-vibe code"
git config --global alias.vr "!${BIN_DIR}/git-vibe release"

PROFILE_FILE="$(ensure_shell_integration)"

cat <<EOF
Git Vibe $(<"${INSTALL_DIR}/VERSION") installed to ${INSTALL_DIR}

Global hooks path:
  ${HOOK_DIR}

Git aliases:
  git vibe -> ${BIN_DIR}/git-vibe
  git vc -> git vibe code
  git vr -> git vibe release

Git config defaults:
  vibe.baseBranch=main
  vibe.branchPrefix=feat/
  vibe.worktreeRoot=../.vibe

Shell profile updated:
  ${PROFILE_FILE}

Current terminal note:
  The installer cannot change the current shell session that launched it.
  Open a new terminal, run: source "${PROFILE_FILE}", or run:
  ${PATH_LINE}
  After the profile is loaded, git vibe code/enter/finish and git vc will auto-jump between worktrees.

After reloading your shell, run:
  git vibe version
EOF