install-bashhub.sh

#!/bin/bash -e
#
# Bashhub.com Installation shell script
#
# Ryan Caloras (ryan@bashhub.com)
#
# http://www.gnu.org/s/hello/manual/autoconf/Portable-Shell.html
#
# The only shell it won't ever work on is cmd.exe.

bash_profile_hook='
### Bashhub.com Installation.
### This Should be at the EOF. https://bashhub.com/docs
if [ -f ~/.bashhub/bashhub.sh ]; then
    source ~/.bashhub/bashhub.sh
fi
'
zsh_profile_hook='
### Bashhub.com Installation
if [ -f ~/.bashhub/bashhub.zsh ]; then
    source ~/.bashhub/bashhub.zsh
fi
'

bash_config_source='
if [ -f ~/.bashrc ]; then
    source ~/.bashrc
fi
'

python_command='
import sys
if (2, 6, 0) < sys.version_info < (3,0):
  sys.exit(0)
else:
  sys.exit(-1)'

bashhub_config=~/.bashhub/config
backup_config=~/.bashhub.config.backup
zshprofile=~/.zshrc

# Optional parameter to specify a github branch
# to pull from.
github_branch=${1:-'1.1.0'}

install_bashhub() {
    check_dependencies
    check_already_installed
    setup_bashhub_files
}

get_and_check_python_version() {
    # Only supporting 2.6 - 2.7 right now. Eventually bump this to include 3.
    # Preface explict versions first to avoid global updates to 3.
    python_version_array=( "python2.7" "python27" "python2.6" "python26" "python2" "python")

    for python_version in "${python_version_array[@]}"; do
        if type "$python_version" &> /dev/null; then
            if "$python_version" -c "$python_command"; then
                echo "$python_version"
                return 0
            fi
        fi

     done
     return 1
}

download_and_install_env() {
    # Select current version of virtualenv:
    VERSION=1.9.1
    # Name your first "bootstrap" environment:
    INITIAL_ENV="env"
    # Options for your first environment:
    ENV_OPTS="--distribute"

    # Only supporting python 2.6 - 2.7 right now.
    python_command=$(get_and_check_python_version)
    if [[ -z "$python_command" ]]; then
        die "\n Sorry you need to have python 2.7 installed. Please install it and rerun this script." 1
    fi

    # Set to whatever python interpreter you want for your first environment
    PYTHON=$(which $python_command)
    URL_BASE=https://pypi.python.org/packages/source/v/virtualenv

    # --- Real work starts here ---
    curl -OL  $URL_BASE/virtualenv-$VERSION.tar.gz
    tar xzf virtualenv-$VERSION.tar.gz

    # Create the first "bootstrap" environment.
    $PYTHON virtualenv-$VERSION/virtualenv.py "$ENV_OPTS" "$INITIAL_ENV"

    # Don't need this anymore.
    rm -rf virtualenv-$VERSION
    # Install the environment.
    $INITIAL_ENV/bin/pip -q install virtualenv-$VERSION.tar.gz
    # Don't need this anymore either.
    rm virtualenv-$VERSION.tar.gz
}

check_dependencies() {
    if [ -z "$(get_and_check_python_version)" ]; then
        die "\n Sorry can't seem to find a version of python 2.7 installed" 1
    fi

    if [ -z "$(detect_shell_type)" ]; then
        die "\n Sorry, couldn't detect your shell type. Bashhub only supports bash or zsh. Your defualt shell is $SHELL." 1
    fi;
}

check_already_installed() {
    if [ -e ~/.bashhub ]; then
        echo -e "\nLooks like the bashhub client is already installed.
        \nLets go ahead and update it.\n"

        # Copy our user credentials so we don't have to ask you for them again.
        if [ -e "$bashhub_config" ]; then
            cp "$bashhub_config" "$backup_config"
        fi

        rm -r ~/.bashhub
    fi
}

install_hooks_for_zsh() {
    # If we're using zsh, install our zsh hooks
    if [ ! -e ~/.zshrc ]; then
        die "No zshfile (.zshrc could be found)" 1
    fi

    # Add our file to our bashprofile if it doesn't exist yet
    if grep -q "source ~/.bashhub/bashhub.zsh" "$zshprofile"
    then
        :
    else
        echo "$zsh_profile_hook" >> "$zshprofile"
    fi

}


# Create two config files .bashrc and .bash_profile since
# OS X and Linux shells use them diffferently. Source .bashrc
# from .bash_profile and everything should work the same now.
generate_bash_config_file() {
    touch ~/.bashrc
    touch ~/.bash_profile
    echo "$bash_config_source" >> ~/.bash_profile
    echo "Created ~/.bash_profile and ~/.bashrc"
}


install_hooks_for_bash() {
    local bashprofile=$(find_users_bash_file)


    # If we don't have a bash profile ask if we should generate one.
    if [ -z "$bashprofile" ]; then
        echo "Couldn't find a bash confg file."

        while true; do
            read -p "Would you like to generate one? (y/n): " yn
            case $yn in
                [Yy]* ) generate_bash_config_file; bashprofile="$HOME/.bashrc"; break;;
                [Nn]* ) break;;
                * ) echo "Please answer yes or no.";;
            esac
        done
    fi

    # Have to have this. Error out otherwise.
    if [ -z "$bashprofile" ]; then
        die "No bashfile (e.g. .profile, .bashrc, etc) could be found." 1
    fi

    # Add our file to our bashprofile if it doesn't exist yet
    if grep -q "source ~/.bashhub/bashhub.sh" $bashprofile
    then
        :
    else
        echo "$bash_profile_hook" >> $bashprofile
    fi

}

detect_shell_type() {
    if [ -n "$ZSH_VERSION" ]; then
        echo 'zsh'
    elif [ -n "$BASH_VERSION" ]; then
        echo 'bash'
    else
        :
    fi
}

install_hooks_for_shell() {
    local shell_type
    shell_type=$(detect_shell_type)

    case $shell_type in
        "zsh")
            install_hooks_for_zsh
            ;;
        "bash")
            install_hooks_for_bash
            ;;
        *)
        die "\n Bashhub only supports bash or zsh. Your defualt shell is $SHELL." 1
    esac
}


setup_bashhub_files() {

    mkdir -p ~/.bashhub
    cd ~/.bashhub
    download_and_install_env

    # Grab the code from master off github.
    curl -sL https://github.com/rcaloras/bashhub-client/archive/${github_branch}.tar.gz -o client.tar.gz
    tar -xvf client.tar.gz
    cd bashhub-client*

    # Copy over our dependencies.
    cp -r bashhub/shell/deps ~/.bashhub/

    # Copy over our bashhub sh and zsh files.
    cp bashhub/shell/bashhub.* ~/.bashhub/

    install_hooks_for_shell

    # install our packages. bashhub and dependencies.
    echo "Pulling down a few dependencies...(this may take a moment)"
    ../env/bin/pip -q install .

    # Check if we already have a config. If not run setup.
    if [ -e $backup_config ]; then
        cp "$backup_config" "$bashhub_config"
        rm "$backup_config"

        # Update our system info if we've got an access token
        if grep -Fq "access_token" "$bashhub_config"; then
            ../env/bin/bashhub util update_system_info

            # Possibly an issue with our access token
            # Lets rerun setup just to be safe.
            if [[ "$?" != 0 ]]; then
                ../env/bin/bashhub setup
            fi
        fi
    else
        # Setup our config file
        ../env/bin/bashhub setup
    fi

    # Wire up our bin directory
    mkdir -p ~/.bashhub/bin
    ln -sf ../env/bin/bashhub ~/.bashhub/bin/bashhub
    ln -sf ../env/bin/bh ~/.bashhub/bin/bh

    # Clean up what we downloaded
    cd ~/.bashhub
    rm client.tar.gz
    rm -r bashhub-client*

    # Make sure our config is only readable to us.
    chmod 600 "$bashhub_config"
    chmod 700 ~/.bashhub

    if [ -e "$bashhub_config" ]; then
        echo "Should be good to go! Please close and restart your terminal session."
    else
        echo "Please run 'bashhub setup' after restarting your terminal session."
    fi
}

#
# Find a users active bash file based on
# which looks the largest. The idea being the
# largest is probably the one they actively use.
#
find_users_bash_file() {

    bash_file_array=( ~/.bash_profile ~/.bashrc ~/.profile)

    local largest_file_size=0
    for file in "${bash_file_array[@]}"
    do
        if [ -e $file ]; then
            # Get our file size.
            local file_size=$(wc -c "$file" | awk '{print $1}')

            if [ $file_size -gt $largest_file_size ]; then
                local largest_file_size=$file_size
                local largest_file=$file
            fi
        fi
     done

    # If we found the largest file, return it
    if [ -n "$largest_file" ]; then
        echo $largest_file
        return 0
    fi
}

die () { echo -e $1; exit $2; }

# Run our install so long as we're not in test.
if [[ -z "$bashhub_install_test" ]]; then
    install_bashhub
fi;