#!/usr/bin/env bash

#set -u # Error on uninitialized variables.

if [[ -z "$rvm_selfcontained" ]]; then
  if [[ "root" = "$(whoami)" ]] || [[ -n "$rvm_prefix" && "$rvm_prefix" != "$HOME"/* ]]; then
    export rvm_selfcontained=0
  else
    export rvm_selfcontained=1
  fi
fi

unset rvm_auto_flag

install_source_path="$(dirname "$0" | sed 's#\/scripts$##')"
if [[ -d "$install_source_path/scripts" ]] && [[ -s "$install_source_path/scripts/utility" ]] ; then
  builtin cd "$install_source_path"
fi

sleep=0.010

source scripts/version
source scripts/utility

usage() {
  printf "
  Usage:
    ${0} [options]

  options:
    --auto    : Automatically update shell profile files.
    --prefix  : Installation prefix directory.
    --help    : Display help/usage (this) message
    --version : display rvm package version
  "
}

check_rubyopt_conditions() {
  if [[ -n "$RUBYOPT" ]]; then
    printf "\n\nWARNING: You have RUBYOPT set in your current environment. This may interfere with normal rvm"
    printf "\n operation. If errors show up, please try unsetting RUBYOPT first.\n"
  fi
}

spinner_counter=0
spinner() {
  array=('/' '-' '\\' '|' '/' '-' '\\' '|')
  index=$((($spinner_counter % 8)))
  printf "\r${array[$index]}"
  let "spinner_counter=spinner_counter+1"
  sleep $sleep
}

if echo "$*" | \grep -q 'trace' ; then echo "$*" ; env | \grep '^rvm_' ; set -x ; fi

while [[ $# -gt 0 ]] ; do
  token="$1" ; shift
  case "$token" in
    --auto)    rvm_auto_flag=1 ;;
    --prefix)  rvm_prefix="$1" ; shift ;;
    --version) rvm_path="$PWD" ; __rvm_version ; unset rvm_path ; exit ;;
    --help|*)  usage ;;
  esac
done

if [[ -z "$rvm_prefix" ]] ; then
  __rvm_load_rvmrc
  if [[ "$rvm_selfcontained" = "0" ]] ; then
    rvm_prefix="${rvm_prefix:-"/usr/local/"}"
  else
    rvm_prefix="${rvm_prefix:-"$HOME/."}"
  fi
fi

if [[ -z "$rvm_path" ]] ; then rvm_path="${rvm_prefix}rvm" ; fi

source scripts/initialize

if \grep -q 'scripts/rvm' "$HOME"/.bash* 2>/dev/null || \grep -q 'scripts/rvm' "$HOME"/.zsh* 2>/dev/null; then
  if [[ -d "$rvm_path" ]] && [[ -s "${rvm_path}/scripts/rvm" ]] ; then
    export upgrade_flag=1
  else
    export upgrade_flag=0
  fi
else
  export upgrade_flag=0
fi

__rvm_initialize

#
# Setup & Configuration
#
item="$(tput setaf 2)* $(tput sgr0)"
question="\n$(tput setaf 2)<?>$(tput sgr0)"
cwd="$PWD"
source_path="${source_path:-$cwd}"
rvm_archives_path="${rvm_archives_path:-"$rvm_path/archives"}"
rvm_src_path="${rvm_src_path:-"$rvm_path/src"}"
rvm_log_path="${rvm_log_path:-"$rvm_path/log"}"

if [[ "$rvm_selfcontained" = "0" ]] ; then
  rvm_bin_path="${rvm_bin_path:-"${rvm_prefix}bin"}"
  rvm_man_path="${rvm_man_path:-"${rvm_prefix}share/man"}"
  rvm_rc_files="${rvm_rc_files:-"/etc/profile /etc/zshenv"}"
else
  rvm_man_path="${rvm_man_path:-"$rvm_path/man"}"
  rvm_bin_path="${rvm_bin_path:-"$rvm_path/bin"}"
  rvm_rc_files="${rvm_rc_files:-"$HOME/.bash_profile $HOME/.bashrc $HOME/.zshenv"}"
fi

rvm_gems_path="${rvm_gems_path:-"$rvm_path/gems"}"
rvm_rubies_path="${rvm_rubies_path:-"$rvm_path/rubies"}"
rvm_scripts_path="${rvm_scripts_path:-"$rvm_path/scripts"}"
rvm_config_path="${rvm_config_path:-"$rvm_path/config"}"
rvm_hooks_path="${rvm_hooks_path:-"$rvm_path/hooks"}"
rvm_tmp_path="${rvm_tmp_path:-"$rvm_path/tmp"}"

export rvm_gemset_separator="@" # TODO: Remove this after a while.

printf "\n$(tput setaf 2)RVM:$(tput sgr0)   shell scripts which allow management of multiple ruby interpreters and environments."
printf "\n$(tput setaf 2)RTFM: $(tput sgr0) http://rvm.beginrescueend.com/"
printf "\n$(tput setaf 2)HELP: $(tput sgr0) http://webchat.freenode.net/?channels=rvm (#rvm on irc.freenode.net)"

if [[ "$upgrade_flag" -eq 1 ]] ;then
  printf "\n\n  Upgrading the RVM installation in $rvm_path/"
else
  printf "\n\n  Installing rvm to $rvm_path/"
fi

spinner

\mkdir -p "$rvm_archives_path" "$rvm_src_path" "$rvm_log_path" "$rvm_bin_path" "$rvm_gems_path" "$rvm_rubies_path" "$rvm_config_path" "$rvm_hooks_path" "$rvm_tmp_path"

for file in README LICENCE ; do
  spinner
  \cp -f "$source_path/$file" "$rvm_path/"
done

#
# Scripts
#
for dir_name in config scripts examples lib hooks help patches; do
  spinner
  \mkdir -p "$rvm_path/$dir_name"
  if [[ -d "$source_path/$dir_name" ]] ; then
    \cp -Rf "$source_path/$dir_name" "$rvm_path"
  fi
done ; unset dir_name

spinner
if [[ ! -s "$rvm_config_path/user" ]] ; then
  \mkdir -p "$rvm_config_path/"
  echo '# Users settings file, overrides db file settings and persists across installs.' >> "$rvm_config_path/user"
fi

scripts=("monitor" "match" "log" "install" "color" "db" "fetch" "log" "set" "package")
for script_name in "${scripts[@]}" ; do
  spinner
  chmod +x "$rvm_scripts_path/$script_name" 2>/dev/null # Hide errors.
done

#
# Bin Scripts
#
# Cleanse and purge...
for file in rvm-prompt rvm rvmsudo rvm-shell rvm-auto-ruby ; do
  spinner && sleep $sleep
  \rm -f "$rvm_bin_path/$file"
  \cp -f "$source_path/binscripts/$file" $rvm_bin_path/
done
spinner
chmod +x "$rvm_bin_path"/*

#
# RC Files
#
spinner
if [[ -n "$rvm_auto_flag" ]] ; then
  printf "Checking rc files... ($rvm_rc_files)"
  if [[ "$rvm_loaded_flag" != "1" ]] ; then
    for rcfile in $(printf $rvm_rc_files) ; do
      if [[ ! -f $rcfile ]] ; then \touch $rcfile ; fi
      if [[ -s "$HOME/.profile" ]] ; then
        if ! \grep -q '.profile' "$rcfile" ; then
          echo "    Adding 'if [[ -s \$HOME/.profile ]] ; then source \$HOME ; fi' to $rcfile."
          printf "\n# rvm-install added line:\nif [[ -s \$HOME/.profile ]] ; then source \$HOME/.profile ; fi\n" >> "$rcfile"
        fi
      fi
      if ! \grep -q "scripts\/rvm" "$rcfile" ; then
        echo "    Adding 'if [[ -s $rvm_scripts_path/rvm ]] ; then source $rvm_scripts_path/rvm ; fi' to $rcfile."
        printf "\n# rvm-install added:\nif [[ -s $rvm_scripts_path/rvm ]] ; then source $rvm_scripts_path/rvm ; fi\n" >> "$rcfile"
      fi
    done
  fi
fi

#
# Initial Interpreter Gemsets.
#
spinner
if [[ -d gemsets/ ]] ; then
  \mkdir -p "$rvm_gemsets_path"
  for gemset_file in $(builtin cd gemsets ; find \. -iname '*.gems' | sed 's/^\.\///') ; do
    destination="$rvm_gemsets_path/$gemset_file"
    destination_path="$(dirname "$destination")"
    if [[ ! -s "$destination" ]] ; then
      \mkdir -p "$destination_path"
      \cp "gemsets/$gemset_file" "$destination"
    fi
  done ; unset destination destination_path gemset_file
fi

spinner
if [[ -d patchsets/ ]] ; then
  \mkdir -p "$rvm_patchsets_path"
  for patchset_file in $(builtin cd patchsets ; find \. -iname '*' | sed 's/^\.\///') ; do
    destination="$rvm_patchsets_path/$patchset_file"
    destination_path="$(dirname "$destination")"
    if [[ ! -s "$destination" ]] ; then
      \mkdir -p "$destination_path"
      \cp "patchsets/$patchset_file" "$destination"
    fi
  done ; unset destination destination_path patchset_file
fi

#
# Migrate ~/.rvm/ruby-X,jruby-X,... to ~/.rvm/rubies/ for 0.1.0.
#
rvm_rubies_path="${rvm_rubies_path:-"$rvm_path/rubies"}"
\mkdir -p "$rvm_rubies_path/"
spinner
for ruby in $rvm_path/ruby-* $rvm_path/jruby-* $rvm_path/mput-* $rvm_path/rbx-* $rvm_path/maglev-* $rvm_path/ree-* ; do
  if [[ -d "$ruby" ]] ; then
    mv "$ruby" "$rvm_rubies_path/"
    new_path="$rvm_rubies_path/$(basename $ruby)"
    for file in gem rake ; do
      spinner
      if [[ -s "$new_path/bin/$file" ]] ; then
        sed -e '1,1s=.*=#!'"${new_path}/bin/ruby=" "${new_path}/bin/${file}" > "${new_path}/bin/${file}.new"
        mv -f "$new_path/bin/$file.new" "$new_path/bin/$file"
        chmod +x "$new_path/bin/$file"
      fi
    done
  fi
done
#
# End of rubies migration.
#

#
# Migrate old gemset directories to new gemset pattern.
#
spinner
printf "\r*" # Stop spinner.

for gemset in "$rvm_path"/gems/*\%* ; do
  new_path=${gemset/\%/${rvm_gemset_separator}}
  if [[ -d "$gemset" ]] && [[ ! -d "$new_path" ]] ; then
    printf "\n    Renaming $(basename "$gemset") to $(basename "$new_path") for new gemset separator."
    mv "$gemset" "$new_path"
  fi
done

for gemset in "$rvm_path"/gems/*\+* ; do
  new_path=${gemset/\+/${rvm_gemset_separator}}
  if [[ -d "$gemset" ]] && [[ ! -d "$new_path" ]] ; then
    printf "\n    Renaming $(basename "$gemset") to $(basename "$new_path") for new gemset separator."
    mv $gemset $new_path
  fi
done

for gemset in "$rvm_path"/gems/*\@ ; do
  new_path=$(echo $gemset | sed -e 's#\@$##')
  if [[ -d "$gemset" ]] && [[ ! -d "$new_path" ]] ; then
    printf "\n    Fixing: $(basename "$gemset") to $(basename "$new_path") for new gemset separator."
    mv "$gemset" "$new_path"
  fi
done

# Move from legacy defaults to the new, alias based system.
if [[ -s "$rvm_config_path/default" ]]; then
  original_version="$(basename "$(\grep GEM_HOME "$rvm_config_path/default" | awk -F"'" '{print $2}' | sed "s#\%#${rvm_gemset_separator}#")")"
  if [[ -n "$original_version" ]]; then
    "$rvm_scripts_path/alias" create default "$original_version" &> /dev/null
  fi ; unset original_version
  \rm -rf "$rvm_config_path/default"
fi

#
# End of gemset migration.
#
printf "\n    Correct permissions for base binaries in $rvm_bin_path..."
\mkdir -p "$rvm_bin_path"
for file in rvm rvmsudo rvm-shell rvm-auto-ruby; do
  [[ -s "$rvm_bin_path/$file" ]] && chmod +x "$rvm_bin_path/$file"
done; unset file
printf "\n    Copying manpages into place."
for man_file in $(\ls "$install_source_path/man"); do
  \rm -rf "$rvm_man_path/$man_file"
  \cp -R "$install_source_path/man/$man_file" "$rvm_man_path/"
done

[[ "$upgrade_flag" = 0 ]] && ./scripts/notes

name="$(awk -F= '/^[[:space:]]*name/{print $2}' ~/.gitconfig 2>/dev/null)"
printf "\n\n${name:-"$(whoami)"},\n"
printf "\nThank you for using rvm. I hope that it makes your work easier and more enjoyable."
printf "\nIf you have any questions, issues and/or ideas for improvement please hop in #rvm on irc.freenode.net and let me know."
printf "\nMy irc nickname is 'wayneeseguin' and I hang out from ~09:00-17:00EST and again from ~21:00EST-~00:00EST."
printf "\nIf I do not respond right away, please hang around after asking your question, I will respond as soon as I am back."
printf "\nBe sure to get head often as rvm development happens fast, you can do this by running 'rvm update --head'."
printf "\n  w$(tput setaf 2)⦿‿⦿$(tput sgr0)t!"
printf "\n    ~ Wayne\n"

if [[ "$upgrade_flag" -eq 1 ]] ; then
  printf "\n$(tput setaf 6)Upgrade Notes\n"
  printf "\n  * 'rvm notes' tells you OS dependency packages for installing rubies."
  printf "\n  * If you encounter any issues with a ruby your best bet is to 'rvm remove X ; rvm install X' "
  printf "\n  * Gemset separator is '@' and will remain unless any rubies error using it."
  printf "\n  * If you wish to have the 'pretty colors' again, set 'export rvm_pretty_print=1' in ~/.rvmrc.\n"
  printf "\n$(tput sgr0)\n"
  check_rubyopt_conditions
  printf "\nUpgrade of RVM in $rvm_path/ is complete.\n\n"
else
  if [[ "$rvm_selfcontained" = "1" ]] ; then
    printf "\n$(tput setaf 1)You must now finish the install manually:$(tput sgr0)"
    printf "\n1) Place the folowing line at the end of your shell's loading files(.bashrc or .bash_profile for bash and .zshrc for zsh), after all path/variable settings:"
    printf "\n     [[ -s \$HOME/.rvm/scripts/rvm ]] && source \$HOME/.rvm/scripts/rvm"
    printf "\n   Please note that this must only occur once - so, you only need to add it the first time you install rvm."
    printf "\n2) Ensure that there is no 'return' from inside the .bashrc file. (otherwise rvm will be prevented from working properly)."
    printf "\n   This means that if you see '[ -z \"\$PS1\" ] && return' then you must change this line to:"
    printf "\n   if [[ -n \"\$PS1\" ]] ; then"
    printf "\n     ... original content that was below the && return line ..."
    printf "\n   fi # <= be sure to close the if."
    printf "\n   #EOF .bashrc"
    printf "\n   Be absolutely *sure* to REMOVE the '&& return'."
    printf "\n   If you wish to DRY up your config you can 'source ~/.bashrc' at the bottom of your .bash_profile."
    printf "\n   placing all non-interactive items in the .bashrc, including the 'source' line above"
    printf "\n3) Then $(tput setaf 1)CLOSE THIS SHELL$(tput sgr0) and open a new one in order to use rvm.\n"
  fi
  if [[ -s "$HOME/.bashrc" ]] && \grep -q '&& return' "$HOME/.bashrc" ; then
    printf "\n\nWARNING:  you have a 'return' statement in your .bashrc, likely this will cause untold havoc."
    printf "\n   This means that if you see '[ -z \"\$PS1\" ] && return' then you must change this line to:"
    printf "\n   if [[ -n \"\$PS1\" ]] ; then"
    printf "\n     ... original content that was below the && return line ..."
    printf "\n   fi # <= be sure to close the if."
    printf "\n   #EOF .bashrc"
    printf "\nEven if you use zsh you should still adjust the .bashrc as above."
    printf "\nIf you have any questions about this please visit #rvm on irc.freenode.net.\n"
  fi
  check_rubyopt_conditions
  printf "\nInstallation of RVM to $rvm_path/ is complete.\n\n"
fi

exit 0