linuxcmd.sh

#!/bin/bash

## this file is runiing in the $MY_TMP_DIR

if [ -z ${MYSITE+z} ]; then
    read -p "Enter domain.name: " -e MYSITE
    if [ -z ${MYSITE+z} ]; then
        MYSITE=linux.cmd # enter name (DNS) of your system here
    fi
fi

MYSH=/etc/profile.d/my.sh
MYCERT_DIR=/root/certs
#SUDOMC="sudo -H mc"
REPLACE_VIM_WITH_NANO=no
USE_INTERNAL_EDITOR_FOR_MC=no

lowercase(){
    echo "$1" | sed "y/ABCDEFGHIJKLMNOPQRSTUVWXYZ/abcdefghijklmnopqrstuvwxyz/"
}

function get_os_info () {
	OS=`lowercase \`uname\``
	KERNEL=`uname -r`
	BITS=`uname -m`

	if [ "${OS}" = "linux" ] ; then
	  # Figure out which OS we are running on
	  if [ -f /etc/os-release ]; then
		  source /etc/os-release
		  DIST_TYPE=$ID
		  DIST=$NAME
		  REV=$VERSION_ID
		  PSEUDONAME=$PRETTY_NAME
	  elif [ -f /usr/lib/os-release ]; then
		  source /usr/lib/os-release
		  DIST_TYPE=$ID
		  DIST=$NAME
		  REV=$VERSION_ID
	  elif [ -f /etc/redhat-release ]; then
		  DIST_TYPE='RedHat'
		  DIST=`cat /etc/redhat-release |sed s/\ release.*//`
		  PSEUDONAME=`cat /etc/redhat-release | sed s/.*\(// | sed s/\)//`
		  REV=`cat /etc/redhat-release | sed s/.*release\ // | sed s/\ .*//`
	  elif [ -f /etc/system-release ]; then
		  if grep "Amazon Linux AMI" /etc/system-release; then
			DIST_TYPE='amzn'
		  fi
		  DIST=`cat /etc/system-release |sed s/\ release.*//`
		  PSEUDONAME=`cat /etc/system-release | sed s/.*\(// | sed s/\)//`
		  REV=`cat /etc/system-release | sed s/.*release\ // | sed s/\ .*//`
	  elif [ -f /etc/SuSE-release ] ; then
		  DIST_TYPE='SuSe'
		  PSEUDONAME=`cat /etc/SuSE-release | tr "\n" ' '| sed s/VERSION.*//`
		  REV=`cat /etc/SuSE-release | tr "\n" ' ' | sed s/.*=\ //`
	  elif [ -f /etc/mandrake-release ] ; then
		  DIST_TYPE='Mandrake'
		  PSEUDONAME=`cat /etc/mandrake-release | sed s/.*\(// | sed s/\)//`
		  REV=`cat /etc/mandrake-release | sed s/.*release\ // | sed s/\ .*//`
	  elif [ -f /etc/debian_version ] ; then
		  DIST_TYPE='Debian'
		  DIST=`cat /etc/lsb-release | grep '^DISTRIB_ID' | awk -F=  '{ print $2 }'`
		  PSEUDONAME=`cat /etc/lsb-release | grep '^DISTRIB_CODENAME' | awk -F=  '{ print $2 }'`
		  REV=`cat /etc/lsb-release | grep '^DISTRIB_RELEASE' | awk -F=  '{ print $2 }'`
	  fi
	  if [ -f /etc/UnitedLinux-release ] ; then
		  DIST="${DIST}[`cat /etc/UnitedLinux-release | tr "\n" ' ' | sed s/VERSION.*//`]"
	  fi
	fi

	if [ "{$OS}" == "darwin" ]; then
		OS=mac
	fi

	DIST_TYPE=`lowercase $DIST_TYPE`
	UNIQ_OS_ID="${DIST_TYPE}-${KERNEL}-${BITS}"
}

get_os_info
echo "DIST_TYPE = $DIST_TYPE"
echo "sudo hostname $MYSITE" > my.sh # create my.sh
case $DIST_TYPE in
	debian)
        MYUSER=admin
        echo 'echo "Debian `cat /etc/debian_version`"' >> my.sh
		AUTOEXEC_FILE=".bashrc"
        MY_LOG=syslog
        MPM=apt-get
	;;
	ubuntu)
        MYUSER=ubuntu
		AUTOEXEC_FILE=".bashrc" # ".profile"
        MY_LOG=syslog
        MPM=apt-get
	;;
	amzn)
		AUTOEXEC_FILE=".bashrc"
        if [ -d "/home/cloudshell-user" ]; then
            MYUSER=cloudshell-user
        else
            MYUSER=ec2-user
        fi
        MY_LOG=messages
        MPM=yum
	;;
	centos)
        echo 'echo "`cat /home/centos/etc/centos-release`"' >> my.sh
		AUTOEXEC_FILE=".bashrc"
        MYUSER=centos
        MY_LOG=messages
        MPM=yum
	;;
	*)
		whiptail --infobox "Not supported Linux for this script version"
		exit 1
	;;
esac
echo "df -k | awk '\$NF==\"/\"{printf \"Disk Usage: %s\n\", \$5}'" >> my.sh

if [ -d "/home/$MYUSER" ]; then
    MYHOME=/home/$MYUSER
else
    MYUSER=root ## Docker has got only root
    MYHOME=/root
fi


# Update / upgrade system"
function update_system {
    if [[ $DIST_TYPE == "amzn" || $DIST_TYPE == "centos" ]] ; then
        yum update -y
    else
        apt-get update
        ##apt-get -y -o Dpkg::Options::=--force-confdef -o Dpkg::Options::=--force-confold dist-upgrade
    fi
}

# Install useful packets
function useful_packets {
    $MPM install -y --skip-broken mc ftp bzip2 zip nano lynx wget curl telnet hostname
}

# Set nice prompt (float wigth)
function set_myprompt {
    if ! grep -q "myprompt" $MYHOME/$AUTOEXEC_FILE ; then	# protect from repeated running
        cat prompt.sh >> $MYHOME/$AUTOEXEC_FILE
        cat alias.sh >> $MYHOME/$AUTOEXEC_FILE
        cat prompt.sh >> /root/$AUTOEXEC_FILE  # .bashrc
        echo "export TERM=xterm-256color" >> /root/$AUTOEXEC_FILE  # fix missing TERM in some dockers or kubernetes 
        if [ -z ${SUDOMC+z} ]; then						# add autostart mc if it was added in config
            echo $SUDOMC >> $MYHOME/$AUTOEXEC_FILE
        fi
    fi
}

# Create www dir and useful links"
function useful_links {
    mkdir -p /var/www
    for MYPATH in  /var/www  /etc  /usr/local/src  /usr  /var/log  /var/log/$MY_LOG
    do
        ln -s $MYPATH $MYHOME > /dev/null 2> /dev/null
    done
}

# Create certs dir"
function certs_dir {
    mkdir -p $MYCERT_DIR
    ln -s $MYCERT_DIR $MYHOME		> /dev/null 2> /dev/null
}

# Nano tuning"
function nano_tuning {
    sed -i 's|color green|color brightgreen|' /usr/share/nano/xml.nanorc
    sed -i 's~(cat|cd|chmod|chown|cp|echo|env|export|grep|install|let|ln|make|mkdir|mv|rm|sed|set|tar|touch|umask|unset)~(apt-get|awk|cat|cd|chmod|chown|cp|cut|echo|env|export|grep|install|let|ln|make|mkdir|mv|rm|sed|set|tar|touch|umask|unset)~' /usr/share/nano/sh.nanorc
    wget https://raw.githubusercontent.com/serialhex/nano-highlight/master/yaml.nanorc -O /usr/share/nano/yaml.nanorc

    if ! grep -q "SELECTED_EDITOR" $MYHOME/.selected_editor > /dev/null 2> /dev/null ; then # protect from repeated running
        echo "SELECTED_EDITOR=/bin/nano" >> $MYHOME/.selected_editor
    fi
    if ! grep -q "SELECTED_EDITOR" /root/.selected_editor > /dev/null 2> /dev/null ; then # protect from repeated running
        echo "SELECTED_EDITOR=/bin/nano" >> /root/.selected_editor
    fi
}

function vim_nano {
    if [[ $REPLACE_VIM_WITH_NANO == "yes" ]] ; then
        if [ -f /bin/vi_orig ] ; then # protect from repeated running
            rm /bin/vi                          > /dev/null 2> /dev/null
            ln -s /usr/bin/nano /bin/vi	        > /dev/null 2> /dev/null
            rm /usr/bin/vim	                    > /dev/null 2> /dev/null
            ln -s /usr/bin/nano /usr/bin/vim	> /dev/null 2> /dev/null
        else
            mv /bin/vi /bin/vi_orig             > /dev/null 2> /dev/null
            ln -s /usr/bin/nano /bin/vi         > /dev/null 2> /dev/null
            mv /usr/bin/vim /usr/bin/vim_orig	> /dev/null 2> /dev/null
            ln -s /usr/bin/nano /usr/bin/vim	> /dev/null 2> /dev/null
        fi
    fi
}

function internal_mcedit {
    if [[ $USE_INTERNAL_EDITOR_FOR_MC == "no" ]] ; then
        if mc -V | grep "Midnight Commander 4.7" ; then # directory for old mc version (to ex. in AMILinux)
            MC_XDG=""
        else
            MC_XDG="config/" # 4.8+ use XDG-support path for config files
        fi

        if [ -f $MYHOME/.${MC_XDG}mc/ini ] ; then
            sed -i "s|^use_internal_edit=.*|use_internal_edit=0|" $MYHOME/.${MC_XDG}mc/ini
        else
            mkdir -p $MYHOME/.${MC_XDG}mc > /dev/null 2> /dev/null
            echo "[Midnight-Commander]" > $MYHOME/.${MC_XDG}mc/ini
            echo "use_internal_edit=0" >> $MYHOME/.${MC_XDG}mc/ini
        fi
        if [ -f /root/.${MC_XDG}mc/ini ] ; then
            sed -i "s|^use_internal_edit=.*|use_internal_edit=0|" /root/.${MC_XDG}mc/ini
        else
            mkdir -p /root/.${MC_XDG}mc > /dev/null 2> /dev/null
            echo "[Midnight-Commander]" > /root/.${MC_XDG}mc/ini
            echo "use_internal_edit=0" >> /root/.${MC_XDG}mc/ini
        fi
    fi
}

# Add /bin/false in to /etc/shells (for correct proftpd working)
function false_shells {
    if ! grep -q "/bin/false" /etc/shells ; then # protect from repeated running
        echo "/bin/false" >> /etc/shells
    fi
}

# Install custom script for startup"
function custom_script {
    if [ ! -f $MYSH ] ; then # protect from repeated running
        cat my.sh >> $MYSH
    else
        sed -i "s|^sudo hostname.*|sudo hostname $MYSITE|" $MYSH
    fi
}

function finish_actions {
    # Chown content of user's home directory
    chown -R $MYUSER:$MYUSER $MYHOME
}

update_system
useful_packets
set_myprompt
useful_links
certs_dir
nano_tuning
vim_nano
internal_mcedit
false_shells
custom_script
finish_actions

exit 0

## EOF