biddibi_boddibi_boo.sh

#!/bin/bash
# Copyright (C) 2018, Raffaello Bonghi <raffaello@rnext.it>
# All rights reserved
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
#
# 1. Redistributions of source code must retain the above copyright 
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.
# 3. Neither the name of the copyright holder nor the names of its 
#    contributors may be used to endorse or promote products derived 
#    from this software without specific prior written permission.
# 
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND 
# CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, 
# BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 
# FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
# HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; 
# OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 
# WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE 
# OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, 
# EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

# Load environment variables:
# - DISTRIB_ID
# - DISTRIB_RELEASE
# - DISTRIB_CODENAME
# - DISTRIB_DESCRIPTION
source /etc/lsb-release

# Load architecture
OS_ARCHITECTURE=$(uname -m)
OS_KERNEL=$(uname -r)

# Load common script filed
source include/common.sh
# Load user interface
source include/modules.sh
# Load remote sources
source include/remote.sh
        
# To Enable the debug mode
if [ -f DEBUG ]; then
    # Set in debug mode
    DEBUG=1
    # Load source as a file
    source DEBUG
    tput setaf 1
    echo "Load debug variables"
    tput sgr0
fi

# Load USER_PWD -> makeself compliant
if [ -z $USER_PWD ] ; then
    USER_PWD=$(pwd)
fi

# Configuration folder information
config_folder="$USER_PWD/$MODULES_CONFIG_FOLDER"

# Make available the bashrc from first script
# "#!/bin/bash --init-file"
# https://stackoverflow.com/questions/21234025/making-a-bash-script-switch-to-interactive-mode-and-give-a-prompt
# --------------------------------
# MAIN
# --------------------------------

usage()
{
	if [ "$1" != "" ]; then
    	tput setaf 1
		echo "$1"
		tput sgr0
	fi
	
    echo "Bibbibi Boddibi Boo is an automatic install for different type of modules."
    echo "Usage:"
    echo "$0 [options]"
    echo "options,"
    echo "   -h|--help        | This help"
    echo "   --nogui          | Launch the system in silent mode (Without GUI)"
    echo "   -q|--quiet       | If required, force automatically the reboot"
    echo "   -c [file/folder] | Load configuration file/folder from reference"
    echo "   -m [user@host]   | Remote connection with NVIDIA Jetson host"
    echo "   -p [passwd]      | Load password without any other request from the script"
}

loop_gui()
{
    # Load user interface
    source include/gui.sh

	# Load configuration modules
	modules_load_config $config_folder
    # Load all modules
    modules_load
    
    if [ $MODULE_REMOTE == "1" ] ; then
        # Load remote gui interface sources
        source include/remote_gui.sh
        # Load remote menu
        menu_remote
    else
        # Set menu selection
        if [ ! -z $MODULE_IM_HOST ] ; then
            MENU_SELECTION=$1
        fi
        # Load GUI menu loop
	    menu_loop
    fi
}

command_line()
{
    # Load user interface
    source include/command_line.sh
    
    # Load configuration modules
    modules_load_config $config_folder
    case "$?" in
    	1)  # Load default configuration
    		tput setaf 3
    		echo "Load default configuration folder in $MODULES_CONFIG_FILE"
    		tput sgr0
    		;;
    	*)  ;;
    esac
    # Load all modules
    modules_load
    
    echo "no gui"
}

main()
{
	# Decode all information from startup
    while [ -n "$1" ]; do
	    case "$1" in
	        -p) 
	            # Load password without any other request from the script
	            MODULE_PASSWORD="$2"
	            shift 1
	            ;;
	        -c)
	            # Load configuration file from other reference [file]
	            config_folder="$2"
	            shift 1
	            ;;
	        --nogui)
	            # Launch the system in silent mode (Without GUI)
	            NO_GUI=1
	            ;;
            -h|--help)
                # Load help
			    usage
			    exit 0
			    ;;
			-q|--quiet)
			    # If required, force automatically the reboot
			    MODULE_REBOOT=1
			    ;;
			-m) # Load user and HOST
			    remote_get_user_host "$2"
			    # Set anyway in Remote mode
			    MODULE_REMOTE=1
			    shift 1
			    ;;
			-x) # Internal option in remote mode
			    MODULE_IM_HOST=1
			    # Set the page to load
			    MENU_SELECTION="$2"
			    shift 1
			    ;;
		    *)
		        usage "Unknown option: $1"
		        exit 1
			    ;;
	    esac
		shift 1
	done
	
	# LOAD jetson stats folder
	if [ ! -d "jetson" ] ; then
	    echo "Jetson folder does not exist!"
	    echo "Download informations"
	    git clone https://github.com/rbonghi/jetson_stats.git jetson
    fi
    if [ -z $MODULE_IM_HOST ] ; then
        echo "Check updates for jetson stats ..."
        cd jetson
        git pull
        cd ..
	fi
    # Load environment variables:
    # - JETSON_BOARD
    # - JETSON_L4T (JETSON_L4T_RELEASE, JETSON_L4T_REVISION)
    # - JETSON_DESCRIPTION
    # - JETSON_CUDA
    source jetson/scripts/jetson_variables

	# Initialization and run biddibi boddibi boo
	local mytitle="Jetson Easy"
	# Check if the code run in NVIDIA Jetson or in remote
	if [ -z $JETSON_BOARD ] ; then
	    # Set in Remote mode
	    MODULE_REMOTE=1
	    mytitle="Jetson Easy - Remote connection"
		# check if is installed sshpass
		if ! dpkg-query -l sshpass > /dev/null; then
			tput setaf 1
			echo "Install sshpass..."
			tput sgr0
			sudo apt-get install sshpass
		fi
	else
	    mytitle="Jetson Easy on $USER@$HOSTNAME"
	fi
	
    # Set title shellbash
    echo -ne '\033]2;'$mytitle'\007'
	
	if [ ! -z $NO_GUI ] ; then
	    unset NO_GUI
		# Launch the system in silent mode (Without GUI)
		command_line
	else
        # Load GUI menu loop
        loop_gui $MENU_SELECTION
	fi
}

main $@
exit 0