LinuxGSM/functions/command_fastdl.sh

#!/bin/bash
# LGSM command_fastdl function
# Author: Daniel Gibbs
# Contributor: UltimateByte
# Website: http://gameservermanagers.com
lgsm_version="190216"

# Description: Creates a FastDL folder

local modulename="FastDL"
function_selfname="$(basename $(readlink -f "${BASH_SOURCE[0]}"))"

check.sh

# Directories
webdir="${rootdir}/www"
fastdldir="${webdir}/fastdl"
addonsdir="${systemdir}/addons"
# Server lua autorun dir, used to autorun lua on client connect to the server
luasvautorundir="${systemdir}/lua/audoturn/server"
luafastdlfile="lgsm_cl_force_fastdl.lua"
luafastdlfullpath="${luasvautorundir}/${luafastdlfile}"

fn_check_bzip2(){
# Returns true if not installed
if [ -z "$(command -v bzip2)" ]; then
	bzip2installed="0"
	fn_printinfo "bzip2 is not installed !"
	fn_scriptlog "bzip2 is not installed"
	echo -en "\n"
	sleep 1
	echo "We advise using it"
	echo "For more information, see https://github.com/dgibbs64/linuxgsm/wiki/Fastdl#bzip2-compression"
	sleep 2
else
	bzip2installed="1"
fi
}

fn_fastdl_init(){
# User confirmation
fn_printok "Welcome to LGSM's FastDL generator"
echo -en "\n"
fn_scriptlog "Started FastDL creation"
sleep 1
while true; do
	read -p "Continue? [y/N]" yn
	case $yn in
	[Yy]* ) break;;
	[Nn]* ) exit;;
	* ) echo "Please answer yes or no.";;
	esac
done
fn_scriptlog "Initiating FastDL creation"

# Check and create folders
if [ ! -d "${webdir}" ]; then
	fn_printdots  "Creating www directory..."
	sleep 0.5
	mkdir "${webdir}"
	sleep 1
	fn_scriptlog "FastDL created ${webdir}"
fi
if [ ! -d "${fastdldir}" ]; then
	newfastdl=1
	fn_printdots "Creating FastDL directory..."
	sleep 0.5
	mkdir "${fastdldir}"
	sleep 1
	fn_scriptlog "FastDL created ${fastdldir}"
	fn_printok "Folders created"
	echo -en "\n"
else
	newfastdl=0
fi
}

fn_fastdl_config(){
fn_printinfo "Entering configuration"
fn_scriptlog "Configuration"
echo -en "\n"
sleep 2
if [ ${newfastdl} == 0 ]; then
	fn_printdots
	while true; do
		read -p "Clear old FastDL files? [y/n]" yn
		case $yn in
		[Yy]* ) clearoldfastdl="on"; fn_scriptlog "clearoldfastdl enabled"; fn_printok "Clearing Enabled"; break;;
		[Nn]* ) clearoldfastdl="off"; fn_scriptlog "clearoldfastdl disabled"; fn_printok "Clearing Disabled"; break;;
		* ) echo "Please answer yes or no.";;
		esac
	done
	sleep 1
	echo -en "\n"
fi
if [ ${bzip2installed} == 1 ]; then
	fn_printdots
	sleep 1
	while true; do
		read -p "Enable file compression using bzip2? [Y/n]" Yn
		case $yn in
		[Yy]* ) bzip2enable="on"; fn_scriptlog "bzip2 enabled"; fn_printok "bzip2 Enabled"; break;;
		[Nn]* ) bzip2enable="off"; fn_scriptlog "bzip2 disabled"; fn_printok "bzip2 Disabled;" break;;
		* ) echo "Please answer yes or no.";;
		esac
	done
	sleep 1
	echo -en "\n"
fi
}

fn_fastdl_gmod_config(){
# Prompt for download enforcer, that is using a .lua addfile resource generator
fn_printdots
while true; do
	read -p "Use client download enforcer? [y/n]" yn
	case $yn in
	[Yy]* ) luaressource="on"; fn_scriptlog "DL enforcer Enabled"; fn_printok "Enforcer Enabled"; break;;
	[Nn]* ) luaressource="off"; "DL enforcer Disabled"; fn_printok "Enforcer Disabled"; break;;
	* ) echo "Please answer yes or no.";;
	esac
	sleep1
	echo -en "\n"
done
}

fn_clear_old_fastdl(){
if [ clearoldfastdl == "on" ]; then
	fn_printinfo "Clearing existing FastDL folder"
	fn_scriptlog "Clearing existing FastDL folder"
	sleep 1
	rm -R "${fastdldir}"/*
	fn_printok "Old FastDL folder cleared"
	fn_scriptlog "Old FastDL folder cleared"
	echo -en "\n"
fi
}

fn_gmod_fastdl(){
# Copy all needed files for fastDL
fn_printdots "Gathering all needed FastDL files..."
echo -en "\n"
sleep 1

# Map Files
fn_printdots "Copying map files..."
fn_scriptlog "Copying map files"
sleep 1
find "${addonsdir}" -name '*.bsp' | cpio -updm "${fastdldir}" > /dev/null 2>&1
fn_printok
echo -en "\n"
sleep 1

# Materials
fn_printdots "Copying materials"
fn_scriptlog "Copying materials"
sleep 1
find "${addonsdir}" -name '*.vtf' | cpio -updm "${fastdldir}" > /dev/null 2>&1
find "${addonsdir}" -name '*.vmt' | cpio -updm "${fastdldir}" > /dev/null 2>&1
fn_printok
echo -en "\n"
sleep 1

# Models
fn_printdots "Copying models"
fn_scriptlog "Copying models"
sleep 1
find "${addonsdir}" -name '*.vtx' | cpio -updm "${fastdldir}" > /dev/null 2>&1
find "${addonsdir}" -name '*.vvd' | cpio -updm "${fastdldir}" > /dev/null 2>&1
find "${addonsdir}" -name '*.mdl' | cpio -updm "${fastdldir}" > /dev/null 2>&1
find "${addonsdir}" -name '*.phy' | cpio -updm "${fastdldir}" > /dev/null 2>&1
fn_printok
echo -en "\n"
sleep 1

# Particles
fn_printdots "Copying particles"
fn_scriptlog "Copying particles"
sleep 1
find "${addonsdir}" -name '*.pcf' | cpio -updm "${fastdldir}" > /dev/null 2>&1
fn_printok
echo -en "\n"
sleep 1

# Sounds
fn_printdots "Copying sounds"
fn_scriptlog "Copying sounds"
sleep 1
find "${addonsdir}" -name '*.wav' | cpio -updm "${fastdldir}" > /dev/null 2>&1
find "${addonsdir}" -name '*.mp3' | cpio -updm "${fastdldir}" > /dev/null 2>&1
find "${addonsdir}" -name '*.ogg' | cpio -updm "${fastdldir}" > /dev/null 2>&1
fn_printok
echo -en "\n"
sleep 1

# Resources (mostly fonts)
fn_printdots "Copying fonts and png"
fn_scriptlog "Copying fonts and png"
sleep 1
find "${addonsdir}" -name '*.otf' | cpio -updm "${fastdldir}" > /dev/null 2>&1
find "${addonsdir}" -name '*.ttf' | cpio -updm "${fastdldir}" > /dev/null 2>&1
find "${addonsdir}" -name '*.png' | cpio -updm "${fastdldir}" > /dev/null 2>&1
fn_printok
echo -en "\n"
sleep 1

# Correct addons folder structure
if [ -d "${fastdldir}/addons" ]; then
	echo "Possible FastDL files found into addons"
	echo "Copying those files to their correct folder"
	sleep 2
	cp -Rf "${fastdldir}"/addons/*/* "${fastdldir}"
# As we're not sure about the correct file structure, duplicate instead of remove
#	rm -R "${fastdldir}/addons"
	fn_printok
	echo -en "\n"
	sleep 1
fi

# Correct content that may be into a lua folder by mistake like some darkrpmodification addons
if [ -d "${fastdldir}/lua" ]; then
	fn_printdots "Stupid filestructure fix"
	sleep 1
	cp -Rf "${fastdldir}/lua/"* "${fastdldir}"
	fn_printok
	echo -en "\n"
	sleep 1
fi
}

fn_fastdl_bzip2(){

if [ ${bzip2enable} == "on" ]; then
	fn_printinfo "Compressing files using bzip2..."
	fn_scriptlog "Compressing files using bzip2..."
	sleep 2
	# bzip2 all files that are not already compressed (keeping original files)
	find "${fastdldir}" -not -name \*.bz2 -exec bzip2 -k \{\} \; > /dev/null 2>&1
	fn_printinfo "bzip2 compression done"
	fn_scriptlog "bzip2 compression done"
	sleep 1
fi
}

# Generate lua file that will force download any file into the FastDL folder
fn_lua_fastdl(){
# Remove lua file if luaressource is turned off
if [ "${luaressource}" == "off" ]; then
	if [ -f "${luafastdlfullpath}" ]; then
		echo "Removing download enforcer"
		sleep 1
		rm -R "${luafastdlfullpath}"
	fi
fi
if [ "${luaressource}" == "on" ]; then
	if [ -f "${luafastdlfullpath}" ]; then
		echo "Removing old download enforcer"
		sleep 1
		rm "${luafastdlfullpath}"
	fi
	echo "Generating new download enforcer"
	sleep 1
	find "${fastdldir}" \( -name "." ! -name "*.bz2" \) -printf '%P\n' | while read line; do
		echo "resource.AddFile("\""${line}"\"")" >> "${luafastdlfullpath}"
	done
	echo "Download enforcer generated"
	sleep 1
fi
}

fn_fastdl_completed(){
echo "----------------------------------"
echo "Congratulations, it's done"
echo "Now you should configure your HTTP server to target the fastdl folder that was created in ${fastdldir}"
echo "Or copy files to an external server"
echo "Don't forget to change your sv_downloadurl accordingly in ${servercfgfullpath}"
echo "You may want to use the www folder to host a loadingurl too,"
echo "for that purpose, just make a loadingurl folder next to the fastdl folder and put your loadingurl in it"
if [ "$bzip2installed" == "0" ]; then
echo "By the way, you'd better install bzip2 an re-run this command"
fi
echo "----------------------------------"
echo "For more information, see https://github.com/dgibbs64/linuxgsm/wiki/Fastdl"
}

# Game checking
# Garry's Mod
if [ "${gamename}" == "Garry's Mod" ]; then
	fn_check_bzip2
	fn_fastdl_init
	fn_fastdl_config
	fn_fastdl_gmod_config
	fn_clear_old_fastdl
	fn_gmod_fastdl
	if [ "${bzip2installed}" == "1" ]; then
		fn_fastdl_bzip2
	fi
	fn_lua_fastdl
	fn_fastdl_completed
	exit
fi