#!/usr/bin/env bash

# CONSTANTS

CONFIG_FILE=/opt/miaou-proxmox/config/default.conf

# FUNCTIONS

function halt_on_error {
	>&2 echo "ERROR: $1"
	exit ${2:-1}
}

function assert_root {
    [ "$(id -u)" -ne 0 ] && halt_on_error 'root privilege required!' 1
}

function assert_proxmox {
	grep -q ^ID=debian /etc/os-release || halt_on_error 'distro debian required!' 2
	command -v pct >/dev/null || halt_on_error 'command `pct` not found!' 3
	command -v curl >/dev/null || halt_on_error 'command `curl` not found!' 4
}

function install_miaou_bash {
	if [[ ! -d /opt/miaou-bash ]]; then
		echo 'installing miaou-bash...'
		curl https://git.artcode.re/miaou/miaou-bash/raw/branch/main/install.sh | sudo bash -s
	fi
}

function install_miaou_proxmox {
	if [[ ! -d /opt/miaou-proxmox ]]; then
		echo 'installing miaou-proxmox...'
		git clone https://git.artcode.re/miaou/miaou-proxmox.git /opt/miaou-proxmox
	else
		cd /opt/miaou-proxmox
		git pull
		cd - >/dev/null
	fi
	status=$(append_or_replace  '^PATH=\$PATH:/opt/miaou-proxmox/bin' 'PATH=$PATH:/opt/miaou-proxmox/bin' $HOME/.bashrc)
	[[ $status == 'appended' ]] && echo 'successfully installed! please run `source $HOME/.bashrc`'
}

function compute_cpu_cores {
	cores=$(nproc)
	if [[ $cores -ge 16 ]]; then 
		cores=$(( cores / 4 ))
	else	
		if [[ $cores -ge 4 ]]; then
			cores=$(( cores / 2 ))
		else
			cores=1
		fi
	fi
	echo $cores
}

function choose_default {
	storage_size='20G'
	memory='2G'
	swap='2G'
	cpu=$(compute_cpu_cores)

	mapfile -t storages <<< $(pvesm status | grep zfspool | cut -d' ' -f1)
	PS3="Choose default storage number: "
	echo "TYPE=ZFSPOOL STORAGE LIST:"
	echo "--------------------------"
	select storage_disk in "${storages[@]}"; do
		[[ -n $storage_disk ]] && break
	done

	cat > $CONFIG_FILE <<EOF 
STORAGE_DISK=\${STORAGE_DISK:-$storage_disk}
STORAGE_SIZE=\${STORAGE_SIZE:-$storage_size}
MEMORY=\${MEMORY:-$memory}
SWAP=\${SWAP:-$swap}
CPU=\${CPU:-$cpu}
EOF
}

function initialize {
	if [[ ! -d /opt/miaou-proxmox/config ]]; then
		echo 'initializing default values...'
		mkdir /opt/miaou-proxmox/config
		choose_default
	else
		echo '------------------------'
		pct-default
	fi

	echo '------------------------'
	echo 'successfully configured!'
	echo '------------------------'
	echo 'you can display or edit default values with:'
	echo '  `pct-default`'
	echo '  `pct-default --edit`'
	echo '------------------------'
	echo 'try out any of the `pct-*` commands (--help might be useful)'
}

# MAIN

set Eue
assert_root
assert_proxmox
install_miaou_bash
install_miaou_proxmox
initialize