basher.sh

#!/usr/bin/env bash

# if $BSHR_DIR is set use that 
# else use $HOME/.bshr 
options_home_dir="${HOME}/.bshr"
if ! [ "$BSHR_DIR" = "" ]; 
then 
    options_home_dir=$BSHR_DIR
fi


options_add=""
options_delete=""
options_remote=""
options_run=""
options_explain=""
options_edit=""
options_list=0
options_help=0
options_update=0
options_cleanup=0
options_uninstall=0
options_scripts_dir="${options_home_dir}/scripts"
options_aliases_dir="${options_home_dir}/aliases"
options_describ_dir="${options_home_dir}/describe"
options_global_log="${options_home_dir}/log"
options_args=""
options_all_args=$@
options_env_file=""
options_switch_user=""
options_git_pull=""
options_git_push=""


function _log() {
    local msg=$(date +"[ %Y-%m-%d %H:%M:%S ] ")
    local msg="${msg}${1}"

    echo "${msg}" >> "${options_global_log}"
}

function _dieOnFail() {
    if [ ! $1 -eq 0 ];
	then
        _log "_dieOnFail [ $1 ] [ $2 ]"
		if [ ! "$2" = "" ]; then echo "$2"; fi
		exit 1;
	fi
}

function _parseParams() {
    while [[ $# -gt 0 ]];
    do
        local key=$1
        _log "key passed -> ${key}"
        if [ "${key}" = "--args" ];
        then
            shift
            options_args=$@
            break
        fi;

        case $key in
            -a|--add)
               options_add=$2
               shift
               shift
               ;;
            -d|--delete)
               options_delete=$2
               shift
               shift
               ;;
            -e|--edit)
               options_edit=$2
               shift
               shift
               ;;
            -E|--explain)
               options_explain=$2
               shift
               shift
               ;;
            -H|--host)
               options_remote=$2
               shift
               shift
               ;;
            -r|--run)
               options_run=$2
               shift
               shift
               ;;
            -l|--list)
               options_list=1
               shift
               ;;
            -h|--help)
               options_help=1
               shift
               ;;
            -gtp|--git_pull)
               options_git_pull=$2
               shift
               shift
               ;;
            -gps|--git_push)
               if [ "$2" = "" ];
               then
                  options_git_push="-"
               else
                  options_git_push=$2
               fi
               shift
               shift
               ;;
            -su|--switch-user)
               options_switch_user=$2
               shift
               shift
               ;;
            --update)
               options_update=1
               shift
               ;;
            --env|-env)
               options_env_file=$2
               shift
               shift
               ;;
	        --cleanup)
	           options_cleanup=1
	           shift
	           ;;
	        --uninstall)
	           options_uninstall=1
	           shift
	           ;;
            *)
               _cmd=$1
               shift
               _args=$@
               echo "${0} --run ${_cmd} --args ${_args}"
               bash $0 --run $_cmd --args $_args
               exit $?
               ;;
        esac
    done;
}

# check if nececary directories exists
function _checkDirectories() {
    if [ ! -d "$options_home_dir" ]; then mkdir "$options_home_dir"; _dieOnFail $? "Unable to create $options_home_dir"; fi;
    if [ ! -d "$options_scripts_dir" ]; then mkdir "$options_scripts_dir"; _dieOnFail $? "Unable to create $options_scripts_dir"; fi;
    if [ ! -d "$options_aliases_dir" ]; then mkdir "$options_aliases_dir"; _dieOnFail $? " Unable to create $options_aliases_dir"; fi;
    if [ ! -d "$options_describ_dir" ]; then mkdir "$options_describ_dir"; _dieOnFail $? " Unable to create $options_describ_dir"; fi;
}

# adds new script
function _addNewScript() {
    _log "_addNewScript()"

    local file_alias="-"
    if ! [ "${ADD_ALIAS}" = "" ];
    then
        file_alias=${ADD_ALIAS}
    fi

    until [[ ! "$file_alias" =~ [^a-zA-Z0-9_\.] ]];
    do
        echo "Enter alias:"
        read -r file_alias
        if [ -f "${options_aliases_dir}/${file_alias}" ];
        then
                echo "Alias already exists"
                local file_alias="-"
        fi
    done

    local descr="${options_describ_dir}/${file_alias}"

    echo "Added from ${2}" >> "${descr}"
    echo "Added to ${1}" >> "${descr}"

    if ! [ "${ADD_DESCRIPTION}" = "" ];
    then
        local _descr="${ADD_DESCRIPTION}"
    else
        echo "Description: "
        read _descr
    fi

    echo "${_descr}" >> "${descr}"

    ln -s "$1" "${options_aliases_dir}/${file_alias}"

    _dieOnFail $? "Unable to add alias"
}

function _getAlias() {
    _log "_getAlias()"
    if [ -f "${options_aliases_dir}/$1" ]; then echo $(readlink "${options_aliases_dir}/$1"); fi;
}

function _process() {

    function _action_help() {
        _log "Action help()"
        echo "BSHR is small script that enables you running and managing shell scripts with one command
bshr
     -a   | --add     <script>     - add script to basher, local or remote
     -e   | --edit    <alias>      - edit script that you saved by some alias
     -r   | --run     <alias>      - run script that you saved by some alias
     -d   | --delete  <alias>      - deletes script
     -H   | --host    <user@host>  - destination where to run script
     -E   | --explain <alias>      - print description message
     -l   | --list                 - list all scripts
     -h   | --help                 - this message
     -su  | --switch-user <user>   - run as this user (requires su permissions)
     -gtp | --git_pull <repo>      - pull repository with scrits
     -gps | --git_push [<repo>]    - push to repository

     -env                          - load env variables from files
     --args                        - pass arguments to script
     --update                      - update basher
     --cleanup                     - clear all


Script can be added from remote url or locally. If alias and description not provided with env variables,
program expect them from stdin.

Example:
    ADD_ALIAS=test ADD_DESCRIPTION=\"some test description\" bshr -a /tmp/test.sh
    bshr -a /tmp/test.sh
    bshr -e test # edits test script
    DEFAULT_EDITOR=vim bshr -e test
    bshr -r test
    bshr -r test -H root@192.168.0.100
    bshr -r test --args 'pass to script' 'also this'
    bsht -r test -H root@192.168.0.100 --args 'additional' 'arg'
";
    }

    # adds script from url or from file
    # $1 should be path to file
    function _action_add() {
        _log "Action add()"
        if [ "$1" = "" ]; then echo "Nothing to add"; exit 2; fi;

        local script_path=$1
        local file_name="`date +"%Y_%m_%d_%H_%M_%S_%N"`_${RANDOM}.sh"

        if [[ "$script_path" =~ ^https?: ]];
        then
            local _curl=$(which curl)
            local _wget=$(which wget)

            if [ "${_curl}${_wget}" = "" ]; then echo "Curl or wget are needed for download"; exit 3; fi;

            if [ ! "${_curl}" = "" ];
            then
                _log "Add via curl to ${file_name}"
                curl -s $1 > "${options_scripts_dir}/${file_name}"; _dieOnFail $? "Unable to download from $2";
                _addNewScript "${options_scripts_dir}/${file_name}" "$1"
            elif [ ! "$_wget" = "" ];
            then
                _log "Add via wget to ${file_name}"
                wget -O "${options_scripts_dir}/${file_name}" $1; _dieOnFail $? "Unable to download from $2"
                _addNewScript "${options_scripts_dir}/${file_name}" "$1"
            fi;
        else
            if [ ! -f "$1" ]; then "File $1 does not exissts"; exit 5; fi;

            cp "$1" "${options_scripts_dir}/${file_name}"
            _dieOnFail $? "Unable to copy $1"
            _addNewScript "${options_scripts_dir}/${file_name}" "$1"
        fi;
    }

    function _action_list() {
        _log "Action list()"
        echo "Scripts found: "
        for item in $(ls "${options_aliases_dir}"); do echo " - ${item}"; done;
    }

    function _action_remove() {
        _log "Action remove()"
        local file_alias=$1
        local script=$(_getAlias $file_alias)
        _log "Remove ${script}"
        if [ "$script" = "" ]; then echo "Unable to find alias"; exit 6; fi
        _log "${options_aliases_dir}/${file_alias}"
        rm "${options_aliases_dir}/${file_alias}"
        _log "rm $script"
        rm "$script"
        if [ -f "${options_describ_dir}/${file_alias}" ];
        then
            _log "${options_describ_dir}/${file_alias}"
            rm "${options_describ_dir}/${file_alias}"
        fi
    }

    function _action_explain() {
        _log "Action explain()"
        local file_alias=$1
        if [ -f "${options_describ_dir}/${file_alias}" ];
        then
            _log "${options_describ_dir}/${file_alias}"
            cat "${options_describ_dir}/${file_alias}"
        fi
    }

    function _action_remote() {
        _log "Action remote(${1} ${2})"
        local script=$(_getAlias $1)
        if [ "${script}" = "" ]; then echo "Alias not found"; exit 7; fi;
        if [ "$2" = "" ]; then echo "Remote server not set"; exit 9; fi;
        if [ "`which ssh`" = "" ]; then echo "ssh command not found"; exit 8; fi;

        ssh "$2" "bash -s" -- < "${script}" "${options_args}"
    }

    function _action_run() {
        local script=$(_getAlias $1)
        if [ "${script}" = "" ]; then echo "Alias not found"; exit 10; fi;
        local bsh=$(which bash)

        if [ "${options_switch_user}" = "" ];
        then
            _log "Action run(${1}) with arguments ${options_args}"
            eval "${bsh} ${script} ${options_args}"
        else
            _log "Action run(${1}) as user ${options_switch_user} with arguments ${options_args}"
            # copy script to tmp so other user can read it
            _copy_script="/tmp/${options_switch_user}_`basename $script`"
            cp ${script} ${_copy_script}
            # make script readable
            chmod o+x ${_copy_script}
            su -s $bsh -c "${_copy_script} $options" $options_switch_user
            # remove script at the end
            rm ${_copy_script}
        fi
    }

    function _load_env_vars() {
        if ! [ "${options_env_file}" = "" ] && [ -f "${options_env_file}" ];
        then
            _log "Load env data from file ${options_env_file}"
            while IFS= read -r line
            do
                eval "export $line"
            done < "${options_env_file}"
        fi
    }

    function _action_update() {
    	_log "Action update()"
        local basher=$(which bshr)
        if [ ! "$basher" = "" ];
        then
            echo "Updating..."
            local abspath=$(readlink "$basher")
            curl -o "$abspath" "https://raw.githubusercontent.com/dalibor91/basher/master/basher.sh?timestamp=$(date +"%s")"
            if [ $? -eq 0 ];
            then
                echo "Updated."
            else
                echo "Update failed"
            fi
        fi
    }

    function _action_edit() {
    	_log "Action edit()"
    	local script=$(_getAlias $1)
        if [ "${script}" = "" ]; then echo "Alias not found"; exit 10; fi;

        if [ "$DEFAULT_EDITOR" ];
        then
            $DEFAULT_EDITOR "$script"
        else
            nano $script
        fi
    }

    function _action_cleanup() {
    	_log "Action cleanup()"
	    echo "Are you sure? Y/n"
    	if [[ "`read u; echo $u`" = [yY] ]]; then
	        rm -rf $options_home_dir
	    fi;
    }

    function _action_uninstall() {
    	_log "Action uninstall()"
	    echo "Are you sure? Y/n"
    	if [[ "`read u; echo $u`" = [yY] ]]; then
            local bshr=$(which bshr)
            local bshr_loc=$(readlink $bshr)
            rm $bshr
            rm -rf $(dirname $bshr_loc)
	    fi;
    }

    function _action_git_pull() {
        which git > /dev/null 2>&1
        _dieOnFail $? "Git doesnt exists"

	    echo "Are you sure? All saved alliases will be lost. [Y/n]"
    	if [[ "`read u; echo $u`" = [yY] ]]; then
	        rm -rf $options_home_dir
	    fi;

        git clone $options_git_pull $options_home_dir

        exit
    }

    function _action_git_push() {
        which git > /dev/null 2>&1
        _dieOnFail $? "Git doesnt exists"

        cd ${options_home_dir}

        if ! [ -d "${options_home_dir}/.git" ];
        then
            _log "initialize git..."
            git init
            echo "log" >> "${options_home_dir}/.gitignore"
        fi

        git add -A .
        git commit -m "Update `date +%F_%H:%M:%S`"
        git push

        if ! [ $options_git_push = "-" ];
        then
            git remote add origin $options_git_push
            git push -u origin master
        else
            git push
        fi

        exit
    }

    if [ $options_help -eq 1 ]; then _action_help; fi;
    if [ $options_list -eq 1 ]; then _action_list; fi;
    if [ $options_cleanup -eq 1 ]; then _action_cleanup; fi;
    if [ $options_uninstall -eq 1 ]; then _action_uninstall; fi;
    if [ ! "$options_edit" = "" ]; then _action_edit $options_edit; fi;
    if [ ! "$options_delete" = "" ]; then _action_remove $options_delete ; fi;
    if [ ! "$options_add" = "" ]; then _action_add $options_add ; fi;
    if [ ! "$options_explain" = "" ]; then _action_explain $options_explain ; fi;
    if [ ! "$options_git_push" = "" ]; then _action_git_push ; fi;
    if [ ! "$options_git_pull" = "" ]; then _action_git_pull ; fi;
    if [ $options_update -eq 1 ];
    then
        echo "Updating..."
        _action_update
	    echo "Done updating"
	    exit
    fi;
    if [ ! "$options_remote" = "" ];
    then
        _action_remote $options_run $options_remote;
    elif [ ! "$options_run" = "" ];
    then
        _load_env_vars
        _action_run $options_run
    fi

    if [ "${options_all_args}" = "" ]; then _action_list; fi;

}

#create directories
_checkDirectories

#parse input params
_parseParams $@

#process it
_process

if [ ! "${DEBUG_BSHR}" = "" ];
then
echo "
 options_add         = ${options_add}
 options_delete      = ${options_delete}
 options_remote      = ${options_remote}
 options_run         = ${options_run}
 options_explain     = ${options_explain}
 options_edit        = ${options_edit}
 options_list        = ${options_list}
 options_help        = ${options_help}
 options_update      = ${options_update}
 options_cleanup     = ${options_cleanup}
 options_uninstall   = ${options_uninstall}
 options_home_dir    = ${options_home_dir}
 options_scripts_dir = ${options_scripts_dir}
 options_aliases_dir = ${options_aliases_dir}
 options_describ_dir = ${options_describ_dir}
 options_global_log  = ${options_global_log}
 options_args        = ${options_args}
 options_all_args    = ${options_all_args}
 options_env_file    = ${options_env_file}
 options_switch_user = ${options_switch_user}
 options_git_pull    = ${options_git_pull}
 options_git_push    = ${options_git_push}
";
fi

exit;