update

#!/usr/bin/env zsh
# 
# update
#     Part of: malte70/scripts
# 
# System update function
# 

SCRIPT_NAME="$(basename $0)"
SCRIPT_VERSION="0.20230126"

source $(dirname $0)/_base.inc.sh


#APP_ACTION_FILLER="."
APP_ACTION_FILLER=" "
APP_ACTION_LENGTH=21





########################################################################
# Usage help / parse command line options
# 
function usage {
	echo "Usage:"
	echo "  $SCRIPT_NAME"
	echo "  $SCRIPT_NAME <options>"
	echo
	echo "Update the system:"
	echo "  → Git: dotfiles/scripts"
	echo "  → pass(1) Git repo"
	echo "  → Installed packages"
	echo "     → ArchLinux: pacman + AUR"
	echo "     → Ubuntu/Debian: APT"
	echo "     → macOS: Homebrew, MacPorts and App Store[1]"
	echo "  → "
	echo
	echo "[1]: Requires <https://github.com/mas-cli/mas> (available via Homebrew)"
	echo
	echo "Options:"
	echo "  --version -V  Show the version and exit"
	echo "  --help    -h  Show this help and exit"
	echo "  --dry-run -d  Dry run, don't do any changes"
	echo
}

DO_DRY_RUN="false"
while [[ $# -gt 0 ]]; do
	if [[ "$1" == "--help" || "$1" == "-h" ]]; then
		version
		echo
		usage
		exit 0

	elif [[ "$1" == "--version" || "$1" == "-V" ]]; then
		version
		exit 0

	elif [[ "$1" == "--dry-run" || "$1" == "-d" ]]; then
		DO_DRY_RUN="true"

	elif [[ "${1:0:1}" == "-" ]]; then
		message_error "Unknown option: \"$1\""

	else

	fi
	shift

done





########################################################################
# Utility functions
# 
# Usage:
#     message_status "$(message_spaces_fill "<Action>" <ActionMaxLength>)" <message ...>

# Usage: message_spaces_fill <message> <max_lenth>
function message_spaces_fill {
	#python ~malte70/code/rahmenwerk.py/rahmenwerk/term.py "text_fill_spaces" "$1" "$2" "."
	#	env msg_text="$1" msg_max_length="$2" msg_space=" " python -c 'import os,sys
	#msg_text=os.environ.get("msg_text",sys.argv[0]);msg_max_length=os.environ.get("msg_max_length");msg_space=os.environ.get("msg_space"," ")
	#print(msg_text + (msg_max_length - min(len(msg_text),msg_max_length)))'
	python3 -c '# python -c […] "<msg_text>" <msg_max_length> "<msg_space>"
def text_fill_spaces(msg_text: str, msg_max_length: int = 0, msg_space: str = " ") -> str:
        msg_space_count = msg_max_length - min(len(msg_text), msg_max_length)
        return msg_text + (msg_space_count*msg_space)
import sys
print(text_fill_spaces(sys.argv[1], int(sys.argv[2]), sys.argv[3]))' "$1" "$2" "$3"
}

# Usage: message_status <AppAction> <message ...>
function message_status {
	if [ -t 1 ]; then
		#echo "${_ANSI_COLOR_CYAN}::${_ANSI_COLOR_LIGHT_BLUE} $1 ${_ANSI_COLOR_CYAN}::${_ANSI_COLOR_WHITE} $2 ${_ANSI_RESET}"
		_print_term "${_ANSI_COLOR_CYAN}"
		_print ":: "
		_print_term "${_ANSI_COLOR_LIGHT_BLUE}"
		_print "$1 "
		shift
		_print_term "${_ANSI_COLOR_CYAN}"
		_print ":: "
		_print_term "${_ANSI_COLOR_WHITE}"
		_print "$@"
		_print_term "${_ANSI_RESET}"
		_println ""
	else
		echo "[$1]" "$@"
	fi
}

function message2 {
	echo "${_ANSI_COLOR_DARK_CYAN}==[${_ANSI_COLOR_LIGHT_GREEN} $@ ${_ANSI_COLOR_DARK_CYAN}]==${_ANSI_RESET}"
}





########################################################################
# Sync Git repositories
# 

##################################################
# dotfiles (~/)
# 
if [[ -d "$HOME/.git" ]]; then
	set-xterm-title "[update] git: ~"
	message_status "$(message_spaces_fill 'git: malte70/dotfiles' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "pull / push / submodule update"

	pushd "$HOME" >/dev/null
	if [[ $DO_DRY_RUN == "true" ]]; then
		message2 "Dry-run: git fetch"
		git fetch
	else
		git pull
		git submodule init >/dev/null
		git submodule sync >/dev/null
		git submodule update >/dev/null
		git push
	fi
	popd
	echo
fi


##################################################
# scripts (~/bin)
# 
if [[ -d "$HOME/bin/.git" ]]; then
	set-xterm-title "[update] git: ~/bin"
	message_status "$(message_spaces_fill 'git:malte70/scripts' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "pull / push"
	pushd $HOME/bin >/dev/null
	if [[ $DO_DRY_RUN == "true" ]]; then
		message2 "Dry-run: git fetch"
		git fetch
	else
		git pull
		git push
	fi
	popd
	echo
fi


##################################################
# pass (~/.password-store, via “pass git”)
# 
if which pass &>/dev/null; then
	set-xterm-title "[update] pass"
	message_status "$(message_spaces_fill 'pass' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "sync"
	if [[ $DO_DRY_RUN == "true" ]]; then
		message2 "Dry-run: Skipped."

	else
		if [ ! -f ~/.update-skip-pass ]; then
			pass git pull
			pass git push
			
		fi

	fi
	echo
fi





########################################################################
# Package managers
# 

##########################################
# Arch Linux: pacman/AUR
# 
if [[ $OSVARIANT == "Arch" ]]; then
	set-xterm-title "[update] pacman upgrade"

	message_status "$(message_spaces_fill 'ArchLinux' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "pacman -Syuw --noconfirm"
	sudo pacman -Syuw --noconfirm

	if [[ $DO_DRY_RUN == "true" ]]; then
		message2 "Dry-run: Only downloading upgrades"

	else
		message_status "$(message_spaces_fill 'ArchLinux' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "pacman -Su"
		sudo pacman -Su
		echo

		# AUR Upgrade
		message_status "$(message_spaces_fill 'ArchLinux' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "trizen -Sua"
		trizen -Sua
		echo
		
		# Clean up unneccessary (left over) packages
		message_status "$(message_spaces_fill 'ArchLinux' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "pacman -Rcs \$(pacman -Qqdtt)"
		pacman -Qqdt &>/dev/null \
			&& sudo pacman -Rcs $(pacman -Qqdt)

	fi
	echo
fi


##########################################
# Debian/Ubuntu: APT
# 
if [[ $OSVARIANT == "Debian" || $OSVARIANT == "Raspbian" || $OSVARIANT == "Ubuntu" ]]; then
	set-xterm-title "[update] APT upgrade"

	message_status "$(message_spaces_fill "$OSVARIANT" $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "Update"
	sudo apt-get update
	echo

	if [[ $DO_DRY_RUN == "true" ]]; then
		message2 "Dry-run: Only downloading upgrades"
		sudo apt-get upgrade -y -d
		sudo apt-get dist-upgrade -y -d
		echo

	else
		message_status "$(message_spaces_fill "$OSVARIANT" $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "Upgrade"
		sudo apt-get upgrade
		echo

		message_status "$(message_spaces_fill "$OSVARIANT" $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "Dist-Upgrade"
		sudo apt-get dist-upgrade
		echo

		message_status "$(message_spaces_fill "$OSVARIANT" $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "Cleanup"
		sudo apt-get autoremove
		echo
		
	fi
fi


##########################################
# Rocky Linux: YUM
# 
if [[ $OSVARIANT == "Rocky Linux" ]]; then
	set-xterm-title "[update] YUM update"

	if [[ $DO_DRY_RUN == "true" ]]; then
		message2 "Dry-run not supported by YUM"

	else
		message_status "$(message_spaces_fill "$OSVARIANT" $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "Update"
		sudo yum update
		
	fi
	echo
fi


##################################################
# Flatpak
#
if which flatpak &>/dev/null; then
	if [[ $DO_DRY_RUN == "true" ]]; then

		set-xterm-title "[update] Flatpak (System)"
		#message_status"$(message_spaces_fill "Flatpak (User)" 15)" "Updating Flatpak (System) packages"
		message_status "$(message_spaces_fill 'Flatpak' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "Updating Flatpak (System) packages"
		message2 "Dry-run: Skipping package deployment"
		flatpak update --no-deploy --no-deps --noninteractive --system
		echo

		set-xterm-title "[update] Flatpak (User)"
		#m_essage_status "$(message_spaces_fill "Flatpak User" 15)" "Updating Flatpak (User) packages"
		message_status "$(message_spaces_fill 'Flatpak' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "Updating Flatpak (User) packages"
		message2 "Dry-run: Skipping package deployment"
		flatpak update --no-deploy --no-deps --noninteractive --user

	else
		set-xterm-title "[update] Flatpak (System)"
		message_status "$(message_spaces_fill 'Flatpak' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "Updating Flatpak (System) packages"
		flatpak update --system
		echo

		set-xterm-title "[update] Flatpak (User)"
		message_status "$(message_spaces_fill 'Flatpak' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "Updating Flatpak (User) packages"
		flatpak update --user

	fi
	echo
fi


##################################################
# macOS: Homebrew/mas
# 
if [[ $OS == "Mac OS X" ]]; then
	set-xterm-title "[update] Homebrew upgrade"

	message_status "$(message_spaces_fill 'macOS :: Homebrew' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "Update"
	brew update
	echo

	if [[ $DO_DRY_RUN == "true" ]]; then
		# 
		# Homebrew
		# 
		message_status "$(message_spaces_fill 'macOS :: Homebrew' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "Upgrade"
		message2 "Dry-run: Passing option \"--dry-run\" to brew"
		brew upgrade --dry-run

		# 
		# MacPorts
		# 
		if which port &>/dev/null; then
			echo
			set-xterm-title "[update] MacPorts upgrade"
			message_status "$(message_spaces_fill 'macOS :: MacPorts' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "Self-Update"
			sudo port selfupdate
			
			echo
			message_status "$(message_spaces_fill 'macOS :: MacPorts' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "Upgrade outdated"
			message2 "Dry-run: Passing option \"-y\" to brew"
			sudo port -y upgrade outdated
		fi

	else
		message_status "$(message_spaces_fill 'macOS :: Homebrew' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "Upgrade"
		brew upgrade
		echo

		message_status "$(message_spaces_fill 'macOS :: Homebrew' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "Cleanup"
		brew autoremove
		# brew cleanup
		brew cleanup --prune=7
		brew doctor

		# 
		# MacPorts
		# 
		if which port &>/dev/null; then
			echo
			set-xterm-title "[update] MacPorts upgrade"
			message_status "$(message_spaces_fill 'macOS :: MacPorts' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "Self-Update"
			sudo port selfupdate
			
			echo
			message_status "$(message_spaces_fill 'macOS :: MacPorts' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "Upgrade outdated"
			sudo port upgrade outdated
		fi
		
		# 
		# App Store
		# 
		if which mas &>/dev/null; then
			echo
			set-xterm-title "[update] App Store update"
			message_status "$(message_spaces_fill 'macOS' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "App Store update via mas"
			mas upgrade
		fi
	fi
	echo
fi





########################################################################
# tldr
# 
if which tldr &>/dev/null; then
	set-xterm-title "[update] tldr"
	message_status "$(message_spaces_fill 'tldr' $APP_ACTION_LENGTH "$APP_ACTION_FILLER")" "update"
	if [[ $DO_DRY_RUN == "true" ]]; then
		message2 "Dry-run: Skipped."
	else
		tldr --update
	fi
	echo
fi



set-xterm-title ""