install.sh

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

usage() {
  cat << 'USAGE'
Usage: install.sh [options]

Options:
  --shell <shell>  Shell to configure (zsh, bash, or none). Default: auto-detect
  --prefix <dir>   Install wt into the specified directory (default: $HOME/.local/bin)
  -h, --help       Show this help message

Examples:
  # Auto-detect shell and configure
  ./install.sh

  # Install for specific shell
  ./install.sh --shell zsh
  ./install.sh --shell bash

  # Install without shell configuration
  ./install.sh --shell none

  # Install via curl for zsh
  curl -fsSL https://raw.githubusercontent.com/notdp/worktree.sh/main/install.sh | bash -s -- --shell zsh

The script installs the wt CLI. If a shell is specified (or auto-detected),
it will also configure the shell integration for automatic directory switching.
USAGE
}

path_has() {
  case ":$PATH:" in
  *":$1:"*) return 0 ;;
  *) return 1 ;;
  esac
}

die() {
  printf 'install.sh: %s\n' "$*" >&2
  exit 1
}

detect_shell() {
  local shell_name="${SHELL##*/}"
  case "$shell_name" in
  bash) echo "bash" ;;
  zsh) echo "zsh" ;;
  *) echo "none" ;;
  esac
}

SHELL_HOOK_STATUS=""
SHELL_SOURCE_HINT=""

readonly STYLE_BLUE_BOLD=$'\033[1;94m'
readonly COLOR_RESET=$'\033[0m'

install_shell_hook() {
  local shell_type="$1"
  local wt_bin="$2"

  SHELL_HOOK_STATUS=""
  SHELL_SOURCE_HINT=""

  if [ "$shell_type" = "none" ]; then
    return 0
  fi

  local hook_file=""
  local hook_marker="# wt shell integration: auto-cd after wt add/path/main/remove/clean"

  case "$shell_type" in
  zsh)
    hook_file="$HOME/.zshrc"
    ;;
  bash)
    hook_file="$HOME/.bashrc"
    ;;
  *)
    printf 'Warning: Unknown shell type %s, skipping shell integration\n' "$shell_type" >&2
    return 0
    ;;
  esac

  # Create rc file if it doesn't exist
  mkdir -p "$(dirname "$hook_file")"
  if [ ! -f "$hook_file" ]; then
    touch "$hook_file"
  fi

  # Check if hook already exists
  if grep -Fq "$hook_marker" "$hook_file"; then
    SHELL_HOOK_STATUS="$(printf 'wt shell hook already present in %s (skipped append).' "$hook_file")"
  else
    printf '\n%s\n' "$("$wt_bin" shell-hook "$shell_type")" >> "$hook_file"
    SHELL_HOOK_STATUS="$(printf 'Appended wt shell hook to %s.' "$hook_file")"
  fi

  SHELL_SOURCE_HINT="source $hook_file"
}

main() {
  local prefix="${WT_INSTALL_PREFIX:-$HOME/.local/bin}"
  local shell_type=""

  # Parse arguments
  while [ $# -gt 0 ]; do
    case "$1" in
    --shell)
      shift || die "--shell requires a value"
      shell_type="$1"
      ;;
    --prefix)
      shift || die "--prefix requires a value"
      prefix="$1"
      ;;
    -h | --help)
      usage
      exit 0
      ;;
    --)
      shift
      break
      ;;
    -*)
      die "unknown option: $1"
      ;;
    *)
      break
      ;;
    esac
    shift || true
  done

  # Auto-detect shell if not specified
  if [ -z "$shell_type" ]; then
    shell_type=$(detect_shell)
    printf 'Auto-detected shell: %s\n' "$shell_type"
  fi

  # Validate shell type
  case "$shell_type" in
  zsh | bash | none) ;;
  *) die "Invalid shell type: $shell_type (use zsh, bash, or none)" ;;
  esac

  # Find the wt script
  local script_path
  script_path=$(dirname "${BASH_SOURCE[0]:-${0}}")
  local script_dir
  script_dir=$(cd "$script_path" 2> /dev/null && pwd || true)

  local source_file=""
  local messages_source=""
  local lib_source_dir=""
  local -a lib_modules=("runtime.sh" "commands.sh")
  local tmpdir=""

  if [ -n "$script_dir" ] && [ -f "$script_dir/bin/wt" ]; then
    source_file="$script_dir/bin/wt"
    if [ -f "$script_dir/bin/messages.sh" ]; then
      messages_source="$script_dir/bin/messages.sh"
    fi
    if [ -d "$script_dir/bin/lib" ]; then
      lib_source_dir="$script_dir/bin/lib"
    else
      die "missing wt modules in $script_dir/bin/lib"
    fi
  else
    command -v curl > /dev/null 2>&1 || die "curl is required"
    tmpdir=$(mktemp -d)
    trap 'rm -rf "${tmpdir:-}"' EXIT

    local source_url="https://raw.githubusercontent.com/notdp/worktree.sh/main/bin/wt"
    printf 'Downloading wt from %s...\n' "$source_url"
    curl -fsSL "$source_url" -o "$tmpdir/wt" || die "failed to download wt from $source_url"
    source_file="$tmpdir/wt"

    local messages_url="https://raw.githubusercontent.com/notdp/worktree.sh/main/bin/messages.sh"
    printf 'Downloading wt messages from %s...\n' "$messages_url"
    if curl -fsSL "$messages_url" -o "$tmpdir/messages.sh"; then
      messages_source="$tmpdir/messages.sh"
    else
      printf 'Warning: failed to download messages file from %s\n' "$messages_url" >&2
    fi

    local modules_url_base="https://raw.githubusercontent.com/notdp/worktree.sh/main/bin/lib"
    lib_source_dir="$tmpdir/lib"
    mkdir -p "$lib_source_dir"
    local module_name module_url
    for module_name in "${lib_modules[@]}"; do
      module_url="$modules_url_base/$module_name"
      printf 'Downloading wt module from %s...\n' "$module_url"
      curl -fsSL "$module_url" -o "$lib_source_dir/$module_name" || die "failed to download wt module from $module_url"
    done
  fi

  # Install wt binary
  install -d "$prefix"
  install -m 0755 "$source_file" "$prefix/wt"

  local messages_installed=0
  if [ -n "$messages_source" ] && [ -f "$messages_source" ]; then
    install -m 0644 "$messages_source" "$prefix/messages.sh"
    messages_installed=1
  fi

  local modules_installed=0
  if [ -n "$lib_source_dir" ]; then
    install -d "$prefix/lib"
    local module_name
    for module_name in "${lib_modules[@]}"; do
      if [ ! -f "$lib_source_dir/$module_name" ]; then
        die "missing wt module source: $lib_source_dir/$module_name"
      fi
      install -m 0644 "$lib_source_dir/$module_name" "$prefix/lib/$module_name"
    done
    modules_installed=1
  fi

  # Check PATH
  if ! path_has "$prefix"; then
    printf 'Warning: %s is not on your PATH. Add it manually to use wt.\n' "$prefix" >&2
  fi

  # Install shell hook if requested
  if [ "$shell_type" != "none" ]; then
    install_shell_hook "$shell_type" "$prefix/wt"
  else
    SHELL_HOOK_STATUS=""
    SHELL_SOURCE_HINT=""
  fi

  # Print installation summary
  printf '\nInstallation summary:\n'
  local summary_index=1
  printf '  %d. Installed wt to %s\n' "$summary_index" "$prefix/wt"
  summary_index=$((summary_index + 1))

  if [ "$messages_installed" -eq 1 ]; then
    printf '  %d. Installed messages to %s\n' "$summary_index" "$prefix/messages.sh"
    summary_index=$((summary_index + 1))
  fi

  if [ "$modules_installed" -eq 1 ]; then
    printf '  %d. Installed support modules to %s/lib\n' "$summary_index" "$prefix"
    summary_index=$((summary_index + 1))
  fi

  if [ "$shell_type" != "none" ]; then
    local hook_line="${SHELL_HOOK_STATUS:-}"
    if [ -z "$hook_line" ]; then
      local shell_label="$shell_type"
      case "$shell_type" in
      zsh) shell_label="Zsh" ;;
      bash) shell_label="Bash" ;;
      esac
      hook_line="Configured shell integration for ${shell_label}."
    fi
    printf '  %d. %s\n' "$summary_index" "$hook_line"
    summary_index=$((summary_index + 1))
  else
    printf '  %d. Skipped shell hook setup (use --shell zsh or --shell bash to enable).\n' "$summary_index"
    summary_index=$((summary_index + 1))
  fi

  printf '  %d. Configuration is stored in %s (key=value).\n' "$summary_index" "${WT_CONFIG_FILE:-$HOME/.worktree.sh/config.kv}"

  # Print next steps
  printf '\nNext steps:\n'
  local next_index=1
  if [ "$shell_type" != "none" ]; then
    local source_hint="$SHELL_SOURCE_HINT"
    if [ -z "$source_hint" ]; then
      case "$shell_type" in
      zsh) source_hint="source $HOME/.zshrc" ;;
      bash) source_hint="source $HOME/.bashrc" ;;
      esac
    fi
    # shellcheck disable=SC2016
    printf '  %d. Run `%s%s%s` (or start a new shell) to enable wt auto-cd now.\n' "$next_index" "$STYLE_BLUE_BOLD" "$source_hint" "$COLOR_RESET"
    next_index=$((next_index + 1))
  else
    # shellcheck disable=SC2016
    printf '  %d. Add the shell hook later with `wt shell-hook zsh >> ~/.zshrc` or `wt shell-hook bash >> ~/.bashrc`.\n' "$next_index"
    next_index=$((next_index + 1))
  fi

  # shellcheck disable=SC2016
  printf '  %d. Run `%swt init%s` in each project so wt tracks your default branch and path (set up wt command tracking).\n' "$next_index" "$STYLE_BLUE_BOLD" "$COLOR_RESET"
  next_index=$((next_index + 1))

  # shellcheck disable=SC2016
  printf '  %d. To switch CLI language anytime, run `wt config set language zh` (or `en`).\n' "$next_index"
}

main "$@"