#!/bin/bash

###############################################################
# MOTD/Login Quick Information script                         #
# This script pulls information from a variety of sources     #
# in the a linux system, and aggregates it into a small table #
# that is displayed upon terminal login to the system.        #
#                                                             #
#           Written by Alan "pyr0ball" Weinstock              #
###############################################################

quickinfo_version=2.1.1
prbl_functons_req_ver=1.6.0

# TODO: Need to re-test after v2.2.x

### TODO: This implementation is still broken. context for location breaks during login
# if ! [[ $scriptname =~ "-bash" ]] ; then
#   #rundir_absolute=$(cd $rundir && pwd)
#   rundir_absolute=$(pushd $rundir && pwd && popd)
#   settingsfile=$(echo "$rundir_absolute/$scriptname" | sed -E 's/(.*)bashrc/\1settings/')
# else
#   if ! [ -f "$HOME/.bashrc.d/11-quickinfo.settings"] ; then
#     settingsfile="$HOME/.bashrc.d/11-quickinfo.settings"
#   else

    ########################################
    # Default quickinfo bashrc Preferences #
    ########################################

    # Disable run on non-interactive sessions (set true to disable)
    interactive_only=false

    # Network Adapter Preferences

    # set false to hide network adapters without valid IP's
    show_disconnected=true

    # Ignored network adapter names (regex match)
    # separate adapter names with '\|' ex. "lo\|tun0"
    filtered_adapters="lo"

    # Disks

    allowed_disk_prefixes=(
      sd
      md
      mapper
      nvme
      mmcblk
      root
    )

    disallowed_disk_prefixes=(
      boot
    )

    #######################################################
    # Do not change these unless you know what you're doing
    # convert array to grep string
    for prefix in "${allowed_disk_prefixes[@]}" ; do
      allowed_disk_prefixes_string+="\\|$prefix"
    done
    # Remove the leading '\|' from the grep string
    allowed_disk_prefixes_string="${allowed_disk_prefixes_string:2}"

    # Manual regex string
    # To use, uncomment and separate disk types with '\|' ex. "sd\|nvme"
    # or use any other grep-compatible filtering desired
    #allowed_disk_prefixes_string="sd\|md\|mapper\|nvme\|mmcblk\|root"


    for prefix in "${disallowed_disk_prefixes[@]}" ; do
      disallowed_disk_prefixes_string+="\\|$prefix"
    done
    disallowed_disk_prefixes_string="${allowed_disk_prefixes_string:2}"
#   fi
# fi
# source $settingsfile

# source PRbL functions
# Source PRbL Functions locally or retrieve from online
if [ ! -z $prbl_functions ] ; then
    source $prbl_functions
else
    if [ -f ${rundir}/functions ] ; then
        source ${rundir}/functions
    else
        # Iterate through get commands and fall back on next if unavailable
        if command -v curl >/dev/null 2>&1; then
            source <(curl -ks 'https://raw.githubusercontent.com/pyr0ball/PRbL/main/functions')
        elif command -v wget >/dev/null 2>&1; then
            source <(wget -qO- 'https://raw.githubusercontent.com/pyr0ball/PRbL/main/functions')
        elif command -v fetch >/dev/null 2>&1; then
            source <(fetch -qo- 'https://raw.githubusercontent.com/pyr0ball/PRbL/main/functions')
        else
            echo "Error: curl, wget, and fetch commands are not available. Please install one to retrieve PRbL functions."
            exit 1
        fi
    fi
fi

# if [ ! -z $prbl_functions ] ; then
#   source $prbl_functions
# else
#   # Failover if global variable is not defined. Checks for 'functions' in same location
#   echo -e "PRbL functions not defined. Check ~/.bashrc"
#   if [ -f "${BASH_SOURCE[0]%/*}/functions" ] ; then
#     source $rundir/functions
#   else
#     echo -e "local functions file also missing. Some visual elements will be missing"
#   fi
# fi
# Locate and import settings (shares same name with script apart from file extension)
scriptname="${BASH_SOURCE[0]##*/}"
rundir="${BASH_SOURCE[0]%/*}"

# check PRbL functions version
if [[ $(vercomp $functionsrev $prbl_functons_req_ver) == 2 ]] ; then
  warn "PRbL functions installed are lower than recommended ($prbl_functons_req_ver)"
  warn "Some features may not work as expected"
else
  if ! vercomp 1 1 ; then
    warn "PRbL functions library is older than 1.1.3 (detection function missing), please update!"
    warn "cd $rundir/PRbL ; git stash ; git pull ; cp ./functions $prbl_functions"
    warn "Some features may not work as expected"
  fi
fi

################################

# Help - displayed when user inputs the -h option, or upon
# running this script with an invalid argument

read -r -d '' usage << EOF
$scriptname [-h] -- Displays information relevent to the system upon login

Usage:
    -h  show this help text
EOF

################################

# Options parser

while getopts ":h" opt
	do
		case ${opt} in
			h)	echo "$usage"
				exit
				;;
			:)	
				;;
			\?)	printf "illegal option: -%s\n" "$OPTARG" >&2
				echo "$usage" >&2
				exit 1
				;;
			esac
		done
		shift $((OPTIND - 1))

################################

# Uncomment this section to disable running this in non-interactive sessions
# For example, on automated logins for a git process, the entire output of
# this script will be written to logs. Disabling non-interactive sessions
# will prevent this script from clogging the logging, but it also can cause
# unexpected exits if improperly deployed (immediate exit on ssh login)
#if [[ $interactive_only == true ]] ; then
#  [[ "$-" == *i* ]] || fail "non-interactive session"
#fi
################################

# Global parameters pulled from system environment
  location=$(uname -a | awk '{print $2}')
  image_version=$(uname -r)
  software_version="$OS $VER"
# Checks if variables are empty and fills the variables
# with generic warning

empty_var="MISSING"

if [ -z "$location" ]
	then
		location=$empty_var
		any_missing=true
fi
if [ "$any_missing" == "true" ]
	then
		vars_missing=$(echo "Something went wrong gathering information")
fi

# Gets public IP address using opendns
# TODO: optimize this to run after time delay using timestamp in settings
set_spinner spinner19
#spin "eval $(wan_ip=$(wget -qO- http://ipecho.net/plain \| xargs echo ))"
#spin read -r wan_ip < <(wget -qO- http://ipecho.net/plain \| xargs echo)
wan_ip=$(wget -qO- http://ipecho.net/plain \| xargs echo )
#spin read -r wan_ip < <(wget -qO- https://ident.me/)

# Checks memory usage
mem_usage=$(free -m | grep Mem | awk '{print $3"M/"$2"M"}')

# Grab number of CPU's for calculating total system load
num_cpus=$(lscpu | grep -v node | grep CPU\(s\)\: | awk '{print $2}')

# Checks load averages using uptime, cuts out the load
# average numbers and prints them as percentages after
# accounting for CPU threads
load_check=$(uptime | sed -r 's|.*load average: ([\.0-9]+), ([\.0-9]+), ([\.0-9]+)|\1 \2 \3|g')
load_averages=$(echo "$load_check $num_cpus" | awk '{printf "5min: %.0f%% ", $1/$4*100} {printf "10min: %.0f%% ", $2/$4*100} {printf "15min: %.0f%%", $3/$4*100}' ORS=' ')

# Alternate CPU Utilization Calculation
# Read /proc/stat file (for first datapoint)
read cpu user nice system idle iowait irq softirq steal guest< /proc/stat

# compute active and total utilizations
cpu_active_prev=$((user+system+nice+softirq+steal))
cpu_total_prev=$((user+system+nice+softirq+steal+idle+iowait))

sleep 0.3

# Read /proc/stat file (for second datapoint)
read cpu user nice system idle iowait irq softirq steal guest< /proc/stat

# compute active and total utilizations
cpu_active_cur=$((user+system+nice+softirq+steal))
cpu_total_cur=$((user+system+nice+softirq+steal+idle+iowait))

# compute CPU utilization (%)
cpu_util=$((100*( cpu_active_cur-cpu_active_prev ) / (cpu_total_cur-cpu_total_prev) ))

# Checks system uptime
if [[ "$(uptime | grep -iq day ; echo $?)" == "0" ]] ; then
    # first logic gate, if the system has been online for at least a day
    if [[ "$(uptime | grep -iq min ; echo $?)" == "0" ]] ; then
        # nested logic gate. If the system has been online for more than a day
        # but less than a day and an hour, the tokens shift and requires alternate parsing
        s_uptime=$(uptime | awk '{print $3 " " $4 " " $5 " Minutes"}')
    else # gate for if the system has been online longer than one day and one hour
        s_uptime=$(uptime | awk '{print $3 " " $4 " " $5 " Hours"}')
    fi
else # logic gate for if the system has been online less than a day but more than an hour
    if [[ "$(uptime | grep -iq min ; echo $?)" == "1" ]] ; then
        s_uptime=$(uptime | awk '{print $3 " Hours  "}')
    else
        s_uptime=$(uptime | awk '{print $3 " Minutes    "}')
    fi
fi


# Checks CPU temperature
cputemp=$(printf "%d" $(sensors | grep -m 1 Core\ 0 | awk '{print $3}') 2> /dev/null)
# Checks CPU and CXPS temperatures,changes output text to warn user if temperatures
# are too high (greater than 65°C is considered hot)
if [[ "$cputemp" -gt "65" ]] ; then
    cputemp="${ong}${blk}${cputemp}°C${dfl}"
else
    cputemp="${cputemp}°C"
fi

################################

# Check for release upgrade on Debian
if [ -x /usr/lib/ubuntu-release-upgrader/release-upgrade-motd ]; then
    if [ -f /var/lib/ubuntu-release-upgrader/release-upgrade-available ] ; then
        release_upgrade="$(cat /var/lib/ubuntu-release-upgrader/release-upgrade-available)"
    fi
fi
if [ "$(lsb_release -sd | cut -d ' ' -f4)" = "(development" ]; then
    unset release_upgrade
fi

################################

# Check for fsck message
if [ -x /usr/lib/update-notifier/update-motd-fsck-at-reboot ]; then
    fsck_needed="$(exec /usr/lib/update-notifier/update-motd-fsck-at-reboot)"
fi

################################

# Check if reboot required by updates
if [ -f /var/run/reboot-required ]; then
    reboot_required=$(cat /var/run/reboot-required)
fi

################################

# Network display and filtering

declare -a adapters=()
declare -a ips=()
declare -a macs=()
declare -a ifups=()
for device in $(ls /sys/class/net/ | grep -v "$filtered_adapters") ; do
  adapters+=($device)
  _ip=$(ip -o -f inet addr show $device | awk '{print $4}' | cut -d/ -f 1 | head -n 1)
  if valid-ip $_ip ; then
    ips+=("$_ip")
    ifups+=("up")
  else
    ips+=("${red}disconnected${dfl}")
    ifups+=("down")
  fi
  macs+=("$(cat /sys/class/net/${device}/address)")
done

################################

# Disk array setup
declare -a logicals=()
declare -a mounts=()
declare -a usages=()
declare -a freespaces=()
diskinfo=$(/bin/df -h | grep "$allowed_disk_prefixes_string" | grep -v "$disallowed_disk_prefixes_string")
logicals=($(cut -d ' ' -f1 <<< "${diskinfo}"))
mounts=($(awk '{print $6}' <<< "${diskinfo}"))
usages=($(awk '{print $5}' <<< "${diskinfo}"))
freespaces=($(awk '{print $4}' <<< "${diskinfo}"))

################################
#check for updates
if [ -f /usr/lib/update-notifier/apt-check ] ; then
  packages=$(/usr/lib/update-notifier/apt-check --human-readable | grep "can be")
  supdates=$(/usr/lib/update-notifier/apt-check --human-readable | grep "security updates" | cut -d '.' -f1)
  packages=${packages%%\.*}
  supdates=${supdates%%\.*}
fi
# Check for updates
if [[ $(echo ${packages} | grep -c updates) != 1 ]] || [ -z "${supdates}" ] || [ -z "${release_upgrade}" ] ; then
#if [[ $(echo ${packages} | grep -c ^0\ updates) == 1 ]] ; then
  need_updates=false
else
  need_updates=true
fi

# Begin echo out of formatted table with aggregated information 

boxtop
boxline ""
boxline "${bld}${unl}Location:${dfl}  ${grn}${unl}$location${dfl}"
boxline ""
# Echo out network arrays
for ((i=0; i<"${#adapters[@]}"; i++ )) ; do
  if [[ $show_disconnected != true ]] ; then
    if [[ ${ifups[$i]} == up ]] ; then
	    boxline "	${adapters[$i]}: ${cyn}${ips[$i]}${dfl} |  ${blu}${macs[$i]}${dfl}"
    fi
  # else
  #   boxline "	${adapters[$i]}: ${cyn}${ips[$i]}${dfl} |  ${blu}${macs[$i]}${dfl}"
  fi
done
boxline "	WAN IP:	${ylw}${wan_ip}${dfl}"
boxline ""
boxline "${bld}${unl}Linux Info${dfl}"
boxline "	Kernel Ver: ${pur}${image_version}${dfl}	|  OS Ver: ${ong}${software_version}${dfl}"
boxline ""
boxline "${bld}${unl}System Status${dfl}"
boxline "	System Load: ${load_averages}"
boxline "	CPU Temp: ${lbl}${cputemp}${dfl}	|  Utilization: ${lrd}${cpu_util}%${dfl}"
boxline "	Memory used/total: ${mem_usage}"
boxline "	${unl}Disk Info:${dfl}"
boxline "${unl}$(printf '\t|%-4s\t%-4s\t%-4s\t%-4s\n' Usage Free Mount Volumes)${dfl}"
for ((i=0; i<"${#logicals[@]}" ; i++ )) ; do
  boxline "       $(printf '|%-4s\t%-4s\t%-4s\t%-4s\n' ${usages[$i]} ${freespaces[$i]} ${mounts[$i]} ${logicals[$i]})"
done
boxline ""
if [[ "$need_updates" == "true" ]] ; then
  boxline "${bld}${unl}Updates${dfl}"
  boxline "	${packages}"
  boxline "	${supdates}"
  boxline "	${release_upgrade}"
fi
if [ -z "${fsck_needed}" ] || [ -z "${reboot_required}" ] ; then
  boxline "	${fsck_needed} ${reboot_required} "
fi
boxbottom