Changed default ZFS pool name, created prompt-based conditional with a case allowing for custom datasets, added cases for custom encrypted dataset or symbolic link creation
This commit is contained in:
		@@ -3,7 +3,7 @@ set -euo pipefail
 | 
			
		||||
 | 
			
		||||
SCRIPT_ROOT=$(dirname "$0")
 | 
			
		||||
# @TODO: Implement more soft-coding and interactivity to this script
 | 
			
		||||
ZFS_POOL_NAME="media"
 | 
			
		||||
ZFS_POOL_NAME="medialib"
 | 
			
		||||
ZFS_COMPAT="openzfs-2.1-linux"
 | 
			
		||||
ZFS_KEY_LOC="file:///etc/zfs/keys/zroot.key"
 | 
			
		||||
ZFS_KEY_FORMAT="passphrase"
 | 
			
		||||
@@ -31,40 +31,72 @@ create_zpool () {
 | 
			
		||||
 | 
			
		||||
mkdataset () {
 | 
			
		||||
	mkdir -p "$1"
 | 
			
		||||
	sudo zfs create -o mountpoint="$1" "$2"
 | 
			
		||||
	sudo zfs create -v -o mountpoint="$1" "$2"
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
create_datasets () {
 | 
			
		||||
	DATASET=("${HOME}/Documents/.calibre" "${1}/LITERATURE")
 | 
			
		||||
	mkdataset "${DATASET[0]}" "${DATASET[1]}"
 | 
			
		||||
	read -r -p "Provide custom datasets? (y/n): " CUSTOMIZE_DATASETS
 | 
			
		||||
	
 | 
			
		||||
	if [[ "$CUSTOMIZE_DATASETS" =~ [Nn]* ]]; then
 | 
			
		||||
		DATASET=("${HOME}/Documents/.calibre" "${1}/LITERATURE")
 | 
			
		||||
		mkdataset "${DATASET[0]}" "${DATASET[1]}"
 | 
			
		||||
 | 
			
		||||
	DATASET=("${HOME}/Videos" "${1}/FILM")
 | 
			
		||||
	mkdataset "${DATASET[0]}" "${DATASET[1]}"
 | 
			
		||||
		DATASET=("${HOME}/Videos" "${1}/FILM")
 | 
			
		||||
		mkdataset "${DATASET[0]}" "${DATASET[1]}"
 | 
			
		||||
 | 
			
		||||
	DATASET=("${HOME}/.var/app/com.valvesoftware.Steam" "${1}/STEAMGAMES")
 | 
			
		||||
	mkdataset "${DATASET[0]}" "${DATASET[1]}"
 | 
			
		||||
		DATASET=("${HOME}/.var/app/com.valvesoftware.Steam" "${1}/STEAMGAMES")
 | 
			
		||||
		mkdataset "${DATASET[0]}" "${DATASET[1]}"
 | 
			
		||||
 | 
			
		||||
	DATASET=("${HOME}/.var/app/org.libretro.RetroArch" "${1}/RARCHGAMES")
 | 
			
		||||
	mkdataset "${DATASET[0]}" "${DATASET[1]}"
 | 
			
		||||
		DATASET=("${HOME}/.var/app/org.libretro.RetroArch" "${1}/RARCHGAMES")
 | 
			
		||||
		mkdataset "${DATASET[0]}" "${DATASET[1]}"
 | 
			
		||||
 | 
			
		||||
	DATASET=("${HOME}/Games" "${1}/GAMES")
 | 
			
		||||
	mkdataset "${DATASET[0]}" "${DATASET[1]}"
 | 
			
		||||
	ln -s "${HOME}/.var/app/com.valvesoftware.Steam" "${DATASET[0]}/Steam"
 | 
			
		||||
	ln -s "${HOME}/.var/app/org.libretro.RetroArch" "${DATASET[0]}/libRetro"
 | 
			
		||||
		DATASET=("${HOME}/Games" "${1}/GAMES")
 | 
			
		||||
		mkdataset "${DATASET[0]}" "${DATASET[1]}"
 | 
			
		||||
		ln -s "${HOME}/.var/app/com.valvesoftware.Steam" "${DATASET[0]}/Steam"
 | 
			
		||||
		ln -s "${HOME}/.var/app/org.libretro.RetroArch" "${DATASET[0]}/libRetro"
 | 
			
		||||
 | 
			
		||||
	DATASET=("${HOME}/.var/app/org.libretro.RetroArch" "${1}/RARCHGAMES")
 | 
			
		||||
	mkdataset "${DATASET[0]}" "${DATASET[1]}"
 | 
			
		||||
		DATASET=("${HOME}/.var/app/org.libretro.RetroArch" "${1}/RARCHGAMES")
 | 
			
		||||
		mkdataset "${DATASET[0]}" "${DATASET[1]}"
 | 
			
		||||
 | 
			
		||||
	DATASET=("${HOME}/.XXX" "${1}/XXX")
 | 
			
		||||
	mkdir -p "${DATASET[0]}"
 | 
			
		||||
	sudo zfs create -o keylocation=prompt -o keyformat=passphrase -o mountpoint="${DATASET[0]}" "${DATASET[1]}"
 | 
			
		||||
		DATASET=("${HOME}/.XXX" "${1}/XXX")
 | 
			
		||||
		mkdir -p "${DATASET[0]}"
 | 
			
		||||
		sudo zfs create -o keylocation=prompt -o keyformat=passphrase -o mountpoint="${DATASET[0]}" "${DATASET[1]}"
 | 
			
		||||
	elif [[ "$CUSTOMIZE_DATASETS" =~ [Yy]* ]]; then
 | 
			
		||||
		while true; do
 | 
			
		||||
			read -r -a DATASET -p "Define a mount path and its corresponding dataset name or route, in that order: "
 | 
			
		||||
 | 
			
		||||
			if [ ${#DATASET[@]} -eq 2 ]; then
 | 
			
		||||
				mkdataset "${DATASET[0]}" "${1}"/"${DATASET[1]}"
 | 
			
		||||
				
 | 
			
		||||
				read -r -p "Create another ZFS dataset? (y/n): " ANOTHER_DATASET
 | 
			
		||||
 | 
			
		||||
				if [[ "$ANOTHER_DATASET" =~ [Yy]* ]]; then
 | 
			
		||||
					continue
 | 
			
		||||
				elif [[ "$ANOTHER_DATASET" =~ [Nn]* ]]; then
 | 
			
		||||
					break
 | 
			
		||||
				fi
 | 
			
		||||
			else
 | 
			
		||||
				printf "Error: must provide 2 arguments (i.e., mount path and dataset name or route)."
 | 
			
		||||
				read -r -p "Try again? (y/n): " RETRY_PROMPT
 | 
			
		||||
 | 
			
		||||
				if [[ "$RETRY_PROMPT" =~ [Yy]* ]]; then
 | 
			
		||||
					continue
 | 
			
		||||
				elif [[ "$RETRY_PROMPT" =~ [Nn]* ]]; then
 | 
			
		||||
					break
 | 
			
		||||
				fi
 | 
			
		||||
			fi
 | 
			
		||||
		done
 | 
			
		||||
	else
 | 
			
		||||
		exit 1
 | 
			
		||||
	fi
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
print_menu () {
 | 
			
		||||
	echo "$(<"${SCRIPT_ROOT}"/zfs-media-drive.help)"
 | 
			
		||||
	printf "%s" "$(<"${SCRIPT_ROOT}"/zfs-media-drive.help)"
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
if [ -z "$1" ]; then
 | 
			
		||||
	printf "Subcommand must be provided"
 | 
			
		||||
	exit 1
 | 
			
		||||
fi
 | 
			
		||||
 | 
			
		||||
@@ -84,12 +116,71 @@ if [ -n "$1" ] && [ "$1" == "init" ]; then
 | 
			
		||||
	done
 | 
			
		||||
 | 
			
		||||
	if [ -z "$ZFS_TARGET_DRIVE" ]; then
 | 
			
		||||
		echo "Error: It is necessary to argue a -d option."
 | 
			
		||||
		printf "Error: It is necessary to argue a -d option."
 | 
			
		||||
		exit 1
 | 
			
		||||
	fi
 | 
			
		||||
 | 
			
		||||
	export ZFS_POOL_NAME
 | 
			
		||||
	create_zpool "$ZFS_TARGET_DRIVE" "$ZFS_POOL_NAME" "$ZFS_COMPAT" "$ZFS_KEY_LOC" "$ZFS_KEY_FORMAT" "$ZFS_COMPRESSION"
 | 
			
		||||
	printf "Following pool has been created: \n"
 | 
			
		||||
	zpool list -v | grep "$ZFS_POOL_NAME"
 | 
			
		||||
	create_datasets "$ZFS_POOL_NAME"
 | 
			
		||||
	printf "Following datasets have been created: \n"
 | 
			
		||||
	zfs list | grep "$ZFS_POOL_NAME"
 | 
			
		||||
elif [ -n "$1" ] && [ "$1" == "enc" ]; then
 | 
			
		||||
	while true; do
 | 
			
		||||
		read -r -a DATASET -p "Define a mount path and its corresponding dataset name or route, in that order: "
 | 
			
		||||
 | 
			
		||||
		if [ ${#DATASET[@]} -eq 2 ]; then
 | 
			
		||||
			mkdir -p "${DATASET[0]}"
 | 
			
		||||
			sudo zfs create -v -o keylocation=prompt -o keyformat=passphrase -o mountpoint="${DATASET[0]}" "${ZFS_POOL_NAME}"/"${DATASET[1]}"
 | 
			
		||||
				
 | 
			
		||||
			read -r -p "Create another ZFS dataset? (y/n): " ANOTHER_DATASET
 | 
			
		||||
 | 
			
		||||
			if [[ "$ANOTHER_DATASET" =~ [Yy]* ]]; then
 | 
			
		||||
				continue
 | 
			
		||||
			elif [[ "$ANOTHER_DATASET" =~ [Nn]* ]]; then
 | 
			
		||||
				break
 | 
			
		||||
			fi
 | 
			
		||||
		else
 | 
			
		||||
			printf "Error: must provide 2 arguments (i.e., mount path and dataset name or route)."
 | 
			
		||||
			read -r -p "Try again? (y/n): " RETRY_PROMPT
 | 
			
		||||
 | 
			
		||||
			if [[ "$RETRY_PROMPT" =~ [Yy]* ]]; then
 | 
			
		||||
				continue
 | 
			
		||||
			elif [[ "$RETRY_PROMPT" =~ [Nn]* ]]; then
 | 
			
		||||
				break
 | 
			
		||||
			fi
 | 
			
		||||
		fi
 | 
			
		||||
	done
 | 
			
		||||
elif [ -n "$1" ] && [ "$1" == "link" ]; then
 | 
			
		||||
	while true; do
 | 
			
		||||
		read -r -a LINKAGE -p "Define a symbolic link source path and destination dataset: "
 | 
			
		||||
 | 
			
		||||
		if [ ${#LINKAGE[@]} -eq 2 ]; then
 | 
			
		||||
			if ! [ -d "${LINKAGE[0]}" ]; then
 | 
			
		||||
				mkdir -p "${LINKAGE[0]}"
 | 
			
		||||
			fi
 | 
			
		||||
			ln -s "${LINKAGE[0]}" "${ZFS_POOL_NAME}"/"${LINKAGE[1]}"
 | 
			
		||||
 | 
			
		||||
			read -r -p "Create another symbolic link? (y/n): " ANOTHER_LINKAGE
 | 
			
		||||
 | 
			
		||||
			if [[ "$ANOTHER_LINKAGE" =~ [Yy]* ]]; then
 | 
			
		||||
				continue
 | 
			
		||||
			elif [[ "$ANOTHER_LINKAGE" =~ [Nn]* ]]; then
 | 
			
		||||
				break
 | 
			
		||||
			fi
 | 
			
		||||
		else
 | 
			
		||||
			printf "Error: must provide 2 arguments (i.e., source path and destination path)."
 | 
			
		||||
			read -r -p "Try again? (y/n): " RETRY_PROMPT
 | 
			
		||||
 | 
			
		||||
			if [[ "$RETRY_PROMPT" =~ [Yy]* ]]; then
 | 
			
		||||
				continue
 | 
			
		||||
			elif [[ "$RETRY_PROMPT" =~ [Nn]* ]]; then
 | 
			
		||||
				break
 | 
			
		||||
			fi
 | 
			
		||||
		fi
 | 
			
		||||
	done
 | 
			
		||||
else
 | 
			
		||||
	while getopts "h" flag; do
 | 
			
		||||
		case "$flag" in
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user