InitUbuntu.sh

#!/bin/bash

# Init option {{{
Color_off='\033[0m' # Text Reset

# Regular Colors
Black='\033[0;30m'  # Black
Red='\033[0;31m'    # Red
Green='\033[0;32m'  # Green
Yellow='\033[0;33m' # Yellow
Blue='\033[0;34m'   # Blue
Purple='\033[0;35m' # Purple
Cyan='\033[0;36m'   # Cyan
White='\033[0;37m'  # White

# verison
VERSION='0.3.2'
# system
SYSTEM="$(uname -s)"
INFO=$(cat /etc/issue)

function welcome() {
	echo -e "\033[36m
   ____     _ __          __  ____             __
  /  _/__  (_) /_        / / / / /  __ _____  / /___ __
 _/ // _ \/ / __/       / /_/ / _ \/ // / _ \/ __/ // /
/___/_//_/_/\__/        \____/_.__/\_,_/_//_/\__/\_,_/

			Version: ${VERSION} 	by: Xu
			System:  ${INFO%\\l}
\033[0m"
}

# success/info/error/warn {{{
msg() {
	printf '%b\n' "$1" >&2
}

info() {
	msg "${Blue}[➭]${Color_off} ${1}${2}"
}

warn() {
	msg "${Red}[►]${Color_off} ${1}${2}"
}

error() {
	msg "${Red}[✘]${Color_off} ${1}${2}"
	exit 1
}
success() {
	msg "${Green}[✔]${Color_off} ${1}${2}"
}

cmdCheck() {
	if ! hash $1 &>/dev/null; then
		error "Command [${1}] not found"
		return 1
	fi
}

aptInstall() {
	# sudo rm -rf /var/lib/dpkg/lock
	# sudo rm -rf /var/cache/apt/archives/lock
	info "Install ${1}"
	warn "Waiting"
	if sudo apt-get install -y $1 >/dev/null; then
		success "Install ${1} Success"
	else
		fail "Install ${1} Failed"
	fi
}

updateANDUpgradeSystem() {
	info "Update system"
	warn "Waiting"
	# sudo rm -rf /var/lib/dpkg/lock
	# sudo rm -rf /var/cache/apt/archives/lock
	sudo apt-get update >/dev/null
	sudo apt-get upgrade -y -q
}

changeSourceForChina() {
	if [ ${1} -eq 1 ]; then
		# etc/sources.list
		info "[USTC] System /etc/apt/sources.list"
		sudo cp /etc/apt/sources.list /etc/apt/sources.list.bak
		sudo sed -i 's/[a-zA-Z]*.archive.ubuntu.com/mirrors.ustc.edu.cn/g' /etc/apt/sources.list
		# sudo sed -i 's/[a-zA-Z]*.archive.ubuntu.com/mirrors.aliyun.com/g' /etc/apt/sources.list
	elif [ ${1} -eq 2 ]; then
		# python pip sources
		info "[tsinghua] python pip ~/.pip/pip.conf"
		mkdir -p ~/.pip
		echo -e "[global]\nindex-url = https://pypi.tuna.tsinghua.edu.cn/simple" >>~/.pip/pip.conf
	elif [ ${1} -eq 3 ]; then
		# docker dameon.json
		info "[USTC] docker /etc/docker/daemon.json"
		sudo echo -e "{\n\"registry-mirrors\": [\"https://docker.mirrors.ustc.edu.cn\"]\n}" >/etc/docker/daemon.json
	fi
}

basicToolsInstall() {
	aptInstall "curl"
	aptInstall "git"
	aptInstall "vim"
	aptInstall "sed"
}

commonTools() {
	case ${1} in
	1)
		info "aria2: a lightweight multi-protocol & multi-source command-line download utility"
		aptInstall aria2
		;;
	2)
		info "tldr: Simplified and community-driven man pages"
		aptInstall tldr
		mkdir -p ~/.local/share
		tldr -u || tldr --update
		;;
	3)
		info "ag: A code-searching tool similar to ack, but faster."
		aptInstall silversearcher-ag
		;;
	4)
		info "fd: A simple, fast and user-friendly alternative to 'find'"
		aptInstall fd-find
		echo 'alias find=fdfind' >> ~/.zshrc
		;;
	5)
		info "A cat(1) clone with syntax highlighting and Git integration."
		aptInstall bat
		echo echo "alias cat='batcat --paging=never'" >> ~/.zshrc
		;;
	6)
		info "exa is a modern replacement for ls."
		aptInstall exa
		echo 'alias ls=exa' >> ~/.zshrc
		;;
	esac
}

pythonDevelopEnv() {
	case ${1} in
	1)
		info "pip3"
		aptInstall "python3-dev python3-pip"
		;;
	2)
		info "pyenv: Simple Python version management"
		aptInstall "make build-essential libssl-dev zlib1g-dev libbz2-dev
						libreadline-dev libsqlite3-dev wget --no-check-certificate curl llvm libncurses5-dev libncursesw5-dev
						xz-utils tk-dev libffi-dev liblzma-dev"
		curl -L https://github.com/pyenv/pyenv-installer/raw/master/bin/pyenv-installer | bash
		echo '# pyenv' >>~/.zshrc
		echo 'export PATH="$HOME/.pyenv/bin:$PATH"' >>~/.zshrc
		echo 'eval "$(pyenv init -)"' >>~/.zshrc
		echo 'eval "$(pyenv virtualenv-init -)"' >>~/.zshrc
		;;
	3)
		info "pipenv: Python Development Workflow for Humans"
		if cmdCheck pip3 -eq 0; then
		# why not use pip3: https://stackoverflow.com/questions/49836676/error-after-upgrading-pip-cannot-import-name-main
			python3 -m pip install --user pipenv
		fi
		echo '# pipenv' >>~/.zshrc
		echo 'alias pipenv="$HOME/.local/bin/pipenv"' >>~/.zshrc
		;;
	esac
}

javaDevelopEnv() {
	case ${1} in
	1)
		info "openjdk"
		aptInstall openjdk-11-jdk
		;;
	2)
		info "maven: A software project management and comprehension tool"
		aptInstall maven
		;;
	esac
}

javaScriptDevelopEnv() {
	case ${1} in
	1)
		info "nvm: Node Version Manager - Simple bash script to manage multiple active node.js versions"
		curl -o- https://raw.githubusercontent.com/creationix/nvm/v0.39.5/install.sh | bash
		;;
	esac
}

dockerDevelopEnv() {
	case ${1} in
	1)
		info "docker-ce"
		curl https://get.docker.com/ | bash
		;;
	2)
		info "docker-compose: a tool for defining and running multi-container Docker applications"
		if cmdCheck pip3 -eq 0; then
			python3 -m pip install --user docker-compose
		fi
		;;
	esac
}

humansTerminal() {
	case ${1} in
	1)
		aptInstall "zsh"
		git clone https://github.com/ohmyzsh/ohmyzsh.git ~/.oh-my-zsh
		cp ~/.oh-my-zsh/templates/zshrc.zsh-template ~/.zshrc
		chsh -s $(which zsh)
		;;
	2)
		info "Configure ~/.zshrc"

		info "Set theme to "Powerlevel10k""
		git clone --depth=1 https://github.com/romkatv/powerlevel10k.git ${ZSH_CUSTOM:-$HOME/.oh-my-zsh/custom}/themes/powerlevel10k
		sed -i 's/ZSH_THEME="robbyrussell"/ZSH_THEME="powerlevel10k\/powerlevel10k"/' ~/.zshrc

		info "Enable plugin: extract"
		sed -i 's/plugins=(/plugins=(extract /' ~/.zshrc

		info "Enable plugin: sudo"
		sed -i 's/plugins=(/plugins=(sudo /' ~/.zshrc

		info "Enable zsh plugin:zsh-syntax-highlighting"
		git clone https://github.com/zsh-users/zsh-syntax-highlighting.git ${ZSH_CUSTOM:-~/.oh-my-zsh/custom}/plugins/zsh-syntax-highlighting
		sed -i 's/plugins=(/plugins=(zsh-syntax-highlighting /' ~/.zshrc
		
		info "Enable zsh plugin:z"
		sed -i 's/plugins=(/plugins=(z /' ~/.zshrc
		;;
	5)
		info "tmux"
		aptInstall "tmux"
		;;
	6)
		info "Install Oh my tmux theme"
		git clone https://github.com/gpakosz/.tmux.git ~/.tmux
		ln -sf ~/.tmux/.tmux.conf ~/.tmux.conf
		cp ~/.tmux/.tmux.conf.local ~/.tmux.conf.local
		;;
	esac
}

help() {
	echo "Usage: bash InitUbuntu.sh [type] [target] [options]"
	echo
	echo "TYPE and TARGET"
	echo
	echo "[update]"
	echo "[source]"
	echo "	apt:    http://mirrors.ustc.edu.cn/ubuntu/"
	echo "	pip:    https://pypi.tuna.tsinghua.edu.cn/simple"
	echo "	docker: https://docker.mirrors.ustc.edu.cn"
	echo "[common]"
	echo "	aira2: A lightweight multi-protocol & multi-source command-line download utility"
	echo "	tldr: Simplified and community-driven man pages"
	echo "	ag: A code-searching tool similar to ack, but faster."
	echo "	fd: A simple, fast and user-friendly alternative to 'find'"
	echo "	bat: A cat(1) clone with syntax highlighting and Git integration."
	echo "	exa: exa is a modern replacement for ls."
	echo "[python]"
	echo "	pip: pip3"
	echo "	pyenv: Simple Python version management"
	echo "	pipenv: Python Development Workflow for Humans"
	echo "[java]"
	echo "	jdk: openjdk-11"
	echo "	maven: A software project management and comprehension tool"
	echo "[javascript]"
	echo "	nvm: Node Version Manager - Simple bash script to manage multiple active node.js versions"
	echo "[docker]"
	echo "	docker-ce: "
	echo "	docker-compose: A tool for defining and running multi-container Docker applications"
	echo "[shell]"
	echo "	zsh: a delightful, open source, community-driven framework for managing your Zsh configuration."
	echo "	zshrc: Configure ~/.zshrc: Powerlevel10k;Plugins:extract/sudo/zsh-syntax-highlighting/z"
	echo "	tmux: terminal multiplexer"
	echo "	.tmux: My self-contained, pretty & versatile tmux configuration"
	echo
	echo "OPTIONS"
	echo
	echo " -b,--basic 	Basic Tools Install: curl,git,vim"
	echo " -v,--Version 	Show version"
	echo " -h,--help 	Show this help message and exit"
	echo
	echo "Example:"
	echo
	echo "	Update System"
	echo "		bash InitUbuntu.sh update"
	echo "	Install java maven"
	echo "		bash InitUbuntu.sh java meven"
	echo "	Install all shell tools"
	echo "		bash InitUbuntu.sh shell"
}

main() {
	if [ $# -eq 0 ]; then
		welcome
		echo "Usage: bash InitUbuntu.sh [type] [target] [options]"
		echo
		echo "InitUbuntu.sh [--help|-h] [--version|-v] [--basic|-b]"
		echo "	{update,source,python,java,javascript,docker,shell} [target]"
		echo
		cmdCheck curl
		cmdCheck git
	else
		case $@ in
		"update")
			updateANDUpgradeSystem
			;;
		"source")
			changeSourceForChina 1
			changeSourceForChina 2
			changeSourceForChina 3
			;;
		"source apt")
			changeSourceForChina 1
			;;
		"source pip")
			changeSourceForChina 2
			;;
		"source docker")
			changeSourceForChina 3
			;;
		"common")
			commonTools 1
			commonTools 2
			commonTools 3
			commonTools 4
			commonTools 5
			commonTools 6
			;;
		"common aira2")
			commonTools 1
			;;
		"common tldr")
			commonTools 2
			;;
		"common ag")
			commonTools 3
			;;
		"common fd")
			commonTools 4
			;;
		"common bat")
			commonTools 5
			;;
		"common exa")
			commonTools 6
			;;
		"python")
			pythonDevelopEnv 1
			pythonDevelopEnv 2
			pythonDevelopEnv 3
			;;
		"python pip")
			pythonDevelopEnv 1
			;;
		"python pyenv")
			pythonDevelopEnv 2
			;;
		"python pipenv")
			pythonDevelopEnv 3
			;;
		"java")
			javaDevelopEnv 1
			javaDevelopEnv 2
			;;
		"java jdk")
			javaDevelopEnv 1
			;;
		"java maven")
			javaDevelopEnv 2
			;;
		"javascript")
			javaScriptDevelopEnv 1
			;;
		"javascript nvm")
			javaScriptDevelopEnv 1
			;;
		"docker")
			dockerDevelopEnv 1
			dockerDevelopEnv 2
			;;
		"docker docker-ce")
			dockerDevelopEnv 1
			;;
		"docker docker-compose")
			dockerDevelopEnv 2
			;;
		"shell")
			humansTerminal 1
			humansTerminal 2
			humansTerminal 5
			humansTerminal 6
			;;
		"shell zsh")
			humansTerminal 1
			;;
		"shell zshrc")
			humansTerminal 2
			;;
		"shell tmux")
			humansTerminal 5
			;;
		"shell .tmux")
			humansTerminal 6
			;;
		--basic | -b)
			basicToolsInstall
			;;
		--version | -v)
			echo "Version: ${VERSION}"
			;;
		--help | -h)
			help
			;;
		esac
	fi
}

main $@