#!/bin/bash

user=`whoami`
if [ "$user" = "root" ] ; then
  echo -e "\033[0;31m <e> \033[0mroot user support is not yet implemented.\033[0m"
  exit 1
fi

info="\n\033[0;32m<i>\033[0m"
question="\n\033[0;32m<?>\033[0m"
cwd=$(pwd)
source_dir="${source_dir:-`dirname $0 | xargs dirname`}"
if [ ! -d "$source_dir" ] ; then unset source_dir ; fi
source_dir="${source_dir:-$cwd}"
rvm_dir=~/.rvm

echo -e "\nInstalling rvm to ~/.rvm/ ..."
for dir in src scripts bin log archives config gems examples ; do
  mkdir -p $rvm_dir/$dir
done
for dir in config scripts examples lib ; do
  mkdir -p $rvm_dir/$dir
  /bin/cp -Rf $source_dir/$dir $rvm_dir
done
cp $source_dir/scripts/rvm-prompt $rvm_dir/bin/
chmod +x $rvm_dir/bin/rvm-prompt

# State what is required to use rvm
echo -e "\n\033[0;32mrvm\033[0m is a shell script that allows a user to manage multiple ruby versions in their own account."
echo -e "$info In order to use rvm the following line must occur in your shell's loading files:"
echo -e "\n\033[0;32m    (a)\033[0m if [ -s ~/.rvm/scripts/rvm ] ; then source ~/.rvm/scripts/rvm ; fi"

results=$(grep 'scripts/rvm' ~/.bash_profile ~/.bashrc ~/.zshrc > /dev/null)
if [ $? -gt 0 ] ; then
  while : ; do
    echo -n -e "$question Do you want this (s)cript to set this up for you, or do it (m)anually yourself ('s' or 'm')? "
    read response
    if [ "${response:0:1}" = "s" ] ; then
      echo 'w00t!'
      break
    elif [ "${response:0:1}" = "m" ] ; then
      echo -e "$info Please make sure that this line is in place."
      break
    fi
  done

  if [ "${response:0:1}" = "s" ] ; then

    while : ; do
      echo -n -e "$question Do you use b(a)sh or (z)sh or b(o)th for your shell (the default on most *nix systems is bash, press control + C to cancel the installation)\n(a/z/o)? "
      read response
      if [ "${response:0:1}" = "a" ] ; then
        user_shell="bash"
        break
      elif [ "${response:0:1}" = "z" ] ; then
        user_shell="zsh"
        break
      elif [ "${response:0:1}" = "o" ] ; then
        user_shell="both"
        break
      fi
    done

    if [ "$user_shell" = "bash" -o "$user_shell" = "both" ] ; then

      touch ~/.bash_profile
      if [ -z "`grep '\.rvm/scripts/rvm' ~/.bash_profile`" ] ; then
        echo "Adding 'if [ -s ~/.rvm/scripts/rvm ] ; then source ~/.rvm/scripts/rvm ; fi' to your ~/.bash_profile."
        echo -e "\n" >> ~/.bash_profile
        echo "if [ -s ~/.rvm/scripts/rvm ] ; then source ~/.rvm/scripts/rvm ; fi" >> ~/.bash_profile
      fi
      touch ~/.bashrc
      if [ -z "`grep '\.rvm/scripts/rvm' ~/.bashrc`" ] ; then
        echo "Adding 'if [ -s ~/.rvm/scripts/rvm ] ; then source ~/.rvm/scripts/rvm ; fi' to your ~/.bashrc."
        echo -e "\n" >> ~/.bashrc
        echo "if [ -s ~/.rvm/scripts/rvm ] ; then source ~/.rvm/scripts/rvm ; fi" >> ~/.bashrc
      fi

    fi

    if [ "$user_shell" = "zsh" -o "$user_shell" = "both" ] ; then

      touch ~/.zshrc
      if [ -z "`grep '\.rvm/scripts/rvm' ~/.zshrc`" ] ; then
        echo "Adding 'if [ -s ~/.rvm/scripts/rvm ] ; then source ~/.rvm/scripts/rvm ; fi' to your ~/.zshrc."
        echo -e "\n" >> ~/.zshrc
        echo "if [ -s ~/.rvm/scripts/rvm ] ; then source ~/.rvm/scripts/rvm ; fi" >> ~/.zshrc
      fi
    fi

  fi
fi

echo -e "\n\033[0;32m<i>\033[0m Ensuring that rvm script location in $file is scripts/rvm not bin/rvm for: ~/.bash_profile, ~/.bashrc, ~/.zshrc"
for file in ~/.bash_profile ~/.bashrc ~/.zshrc ; do
  if [ -s $file ] ; then
    if [ -L $file ] ; then         # If the file is a symlink,
      actual_file=`readlink $file` # read the link target so we can preserve it.
    else
      actual_file="$file"
    fi

    grep 'rvm\/bin\/rvm' $actual_file > /dev/null
    if [ $? -eq 0 ] ; then
      sed -i.orig 's/rvm\/bin\/rvm/rvm\/scripts\/rvm/g' $actual_file
      rm -f $actual_file.orig
    fi

    if [ -f ~/.profile ] ; then
      grep '.profile' $actual_file > /dev/null
      if [ $? -gt 0 ] ; then
        profile=~/.profile
        echo -e "\n" >> $actual_file
        echo "source $profile" >> $actual_file
      fi
    fi
  fi
done

if [ -f ~/.rvm/bin/rvm ] ; then
  echo -e "\n\033[0;32m<i>\033[0m Removing old rvm file from ~/.rvm/bin/rvm"
  rm -f ~/.rvm/bin/rvm
fi

system=`uname`
if [ "$system" = "Linux" ] ; then
  rvm_apt_get_binary=`which apt-get 2> /dev/null`
  rvm_emerge_binary=`which emerge 2> /dev/null`
  rvm_pacman_binary=`which pacman 2> /dev/null`
  rvm_yum_binary=`which yum 2> /dev/null`

  if [ ! -z "$rvm_apt_get_binary" ] ; then
    echo -e "\033[0;33m <w> \033[0mFor jRuby (if you wish to use it) you will need:"
    echo -e "\033[0;32m <i> \033[0msudo apt-get install sun-java6-bin sun-java6-jre sun-java6-jdk"
    echo -e "\033[0;33m <w> \033[0mFor ree (if you wish to use it) you will need:"
    echo -e "\033[0;32m <i> \033[0msudo apt-get install libreadline5-dev libssl-dev bison"

  elif [ ! -z "$rvm_emerge_binary" ] ; then
    echo -e "\033[0;33m <w> \033[0mFor jRuby (if you wish to use it) you will need:"
    echo -e "\033[0;32m <i> \033[0msudo emerge dev-java/sun-jdk dev-java/sun-jre-bin"

  elif [ ! -z "$rvm_pacman_binary" ] ; then
    echo -e "\033[0;33m <w> \033[0mFor jRuby (if you wish to use it) you will need:"
    echo -e "\033[0;32m <i> \033[0msudo pacman -Sy jdk jre"

  elif [ ! -z "$rvm_yum_binary" ] ; then
    echo -e "\033[0;33m <w> \033[0mFor ree (if you wish to use it) you will need:"
    echo -e "\033[0;32m <i> \033[0myum install -y rpm-build gcc gcc-c++ redhat-rpm-config ; then download and rpmbuild and install the sdk, Have fun..."

  else
    echo -e "\033[0;33m <w> \033[0mFor jRuby (if you wish to use it) you will need:"
    echo -e "\033[0;32m <i> \033[0mThe SUN java runtime environment and development kit."
  fi
elif [ "$system" = "Darwin" ] ; then
  echo -e "\033[0;32m <i> \033[0mBe sure that you have XCode Tools installed in order to use rvm."
  echo -e "\033[0;32m <i> \033[0mIf you intend on installing MacRuby you must install LLVM first."
fi

echo -e "$info There have been a great many updates since previous releases, please:"
echo -e "$info   * remove sourcing of ~/.rvm/current from your shell profiles."
echo -e "$info   * CLOSE THIS SHELL AND OPEN A NEW ONE FOR rvm."

echo -e "\n$info  Please visit the website for much more information: http://rvm.beginrescueend.com/\n"
echo '    w⦿‿⦿t! '
echo -e "\n        ~ Wayne"

source $rvm_dir/scripts/rvm

rvm symlinks
echo
rvm -v
echo