# This script gets a Raspberry Pi setup with essential applications.
#
# For explanations of commands, visit raspi.vintagecoding.net or use Google.

# GLOBAL VARIABLES
MANUAL_PORT=5090
JELLYFIN_PORT=5091
OLLAMA_PORT=5092

help() {
	echo "This script gets a Raspberry Pi setup with essential applications.";
	echo "By default, everything will be installed. But, there are some flags";
	echo "available for the end user. Multiple flags can be used together.";
	echo "";
	echo "";
	echo "";
	echo "If you're new to the command line, the structure of these help";
	echo "pages is often like this:";
	echo "    command { --full-name-of-flag | -f }";
	echo "Where, the curly braces indicates a grouping of options, typically";
	echo "the full name of a flag, | (or) the short flag.";
	echo "Therefore, either of the following commands are valid:";
	echo "    command --full-name-of-flag";
	echo "or";
	echo "    command -f";
	echo "Furthermore, different flags can be chained together. e.g.,";
	echo "    command --flag-1 --flag-2";
	echo "";
	echo "";
	echo "";
	echo "If you're new, checkout the Raspberry Pi manual available at";
	echo "raspi.vintagecoding.net.";
	echo "";
	echo "";
	echo "";
	echo "NOTE: This script CANNOT be copied to /usr/local/bin, or anywhere on";
	echo "the PATH environment variable. That interferes with the system-level";
	echo "install command.";
	echo "";
	echo "";
	echo "";
	echo "Usage:";
	echo "    ./install.sh { --help | -h }";
	echo "";
	echo "";
	echo "";
	echo "The user will be prompted for each feature beyond the the barebones";
	echo "system utilities installation.";
}

# This function initializes core services and packages for getting started.
init() {
	# Update repositories and upgrade installed packages.
	sudo apt update && sudo apt upgrade;

	# Install requisite packages for cloudflared.
	sudo apt install curl lsb-release;

	# Downloads the GPG key (a package integrity technology).
	curl -L https://pkg.cloudflare.com/cloudflare-main.gpg | sudo tee \
		/usr/share/keyrings/cloudflare-archive-keyring.gpg > /dev/null;

	# Add the GPG key to the allowed repositories to download packages from.
	echo "deb [signed-by=/usr/share/keyrings/cloudflare-archive.keyring.gpg] \
		https://pkg.cloudflare.com/cloudflared $(lsb_release -cs) main" \
		| sudo tee /etc/apt/sources.list.d/cloudflared.list;

	# Install cloudflared.
	# Install podman, a container technology for easily deploying projects.
	# Install golang, a server-side programming language.
	# Install tools for the user
	sudo apt update && sudo apt install cloudflared podman golang tldr neovim;
}

# This function initializes the manual mirror of raspi.vintagecoding.net
init_manual() {
	cd ~/ras-pi/man/pages;
	sudo echo "
		[Unit]
		Description=Raspberry Pi Manual Mirror
		After=network.target
		StartLimitIntervalSec=0

		[Service]
		Type=simple
		Restart=always
		RestartSec=1
		User=$(whoami)
		WorkingDirectory=$(pwd)
		ExecStartPre=/usr/bin/git pull
		ExecStart=/usr/bin/python3 -m http.server $MANUAL_PORT

		[Install]
		WantedBy=multi-user.target
	" > /etc/systemd/system/manual.service;

	sudo systemctl enable manual.service;
	sudo systemctl start manual.service;
	sudo systemctl daemon-reload;
}

# This function initializes Jellyfin.
init_jellyfin() {
	# Get the official image of Jellyfin software
	podman pull jellyfin/jellyfin;

	echo "Making media directories where Jellyfin will retrieve content.";
	echo "It is up to you to provide supported files and build a library.";

	cd;
	mkdir media;

	podman run \
		--detach \
		--label "io.containers.autoupdate=registry" \
		--name myjellyfin \
		--publish $JELLYFIN_PORT:8096/tcp \
		--user $(id -u):$(id -g) \
		--userns keep-id \
		--volume jellyfin-cache:/cache:Z \
		--volume jellyfin-config:/config:Z \
		--mount type=bind,source=/$(pwd)/media,destination=/media,ro=true,relabel=private \
		--restart always \
	jellyfin

	echo "Finished creating the Jellyfin container! It is now live on"
	echo "http://localhost:$JELLYFIN_PORT. If you enabled it, it'll also be"
	echo "available at media.yourdomain.com."
}

# This function initializes Ollama.
init_ollama() {
	echo "This will take a while..."
}

# This function initializes cloudflared.
init_cloudflared() {
	echo "There will be some manual intervention you'll need to make to \
		to setup cloudflared. Please see "
	cloudflared tunnel login
	cloudflared tunnel create raspi
	read "?What is your domain name? e.g., vintagecoding.net" DOMAIN

	# TODO: Implement tunnel creation

	echo "
		[Unit]
		Description=Cloudflare Tunnel Daemon
		After=network.target

		[Service]
		Type=simple
		User=$(whoami)
		ExecStart=/usr/bin/cloudflared tunnel run raspi
		Restart=always

		[Install]
		WantedBy=multi-user.target
	" > /etc/systemd/system/cloudflared.service;

	sudo systemctl enable cloudflared.service;
	sudo systemctl start cloudflared.service;
}

# Display help if the user requests it.
if [[ $# == 1 && ($1 == "--help" || $1 == "-h") ]]; then
	help
	exit
fi

# Get user input for some default applications to get started.
read "?Do you have a Cloudflare account and own a domain? (y/N)" CLOUDFLARE
read "?Do you want a mirror of the manual from raspi.vintagecoding.net\
	on this device? (y/N)" MANUAL
read "?Do you want to setup Jellyfin, a personal Netflix alternative? (y/N)" JELLYFIN
read "?Do you want to setup Ollama, a personal ChatGPT alternative? (y/N)" OLLAMA

init

if [[ $MANUAL == "y" ]]; then
	init_manual
fi

if [[ $JELLYFIN == "y" ]]; then
	init_jellyfin
fi

if [[ $CLOUDFLARE == "y" ]]; then
	init_cloudflare
fi

if [[ $OLLAMA == "y" ]]; then
	init_ollama
fi