prepare.sh

#!/bin/bash

#
# Prepare function files to be "pasted" on mikrotik console or executed as script
#

ALL_FUNCTIONS=0
DEFS_OVERRIDE=${DEFS_OVERRIDE:-0}
FUNCTIONS_FILE="/tmp/ifm.mk.functions.$$"
RM_FUNCTIONS_FILE=1
VERSION=${MK_VERSION:-6}
while [ "$1" != "" ]
do
	if [ "$1" = "-all" ]
	then
		ALL_FUNCTIONS=1
	elif [ "$1" = "-fn" ] && [ "$2" != "" ]
	then
		FUNCTIONS_FILE="$(realpath $2)"
		RM_FUNCTIONS_FILE=0
		shift
	fi
	shift
done

FUNCTIONS_DIR="functions"
TMP_FUNCTIONS_DIR="/tmp/ifm.mk.functions.d.$$"

rm -Rf $TMP_FUNCTIONS_DIR
mkdir -p $TMP_FUNCTIONS_DIR
for dir in ${FUNCTIONS_DIR} ${FUNCTIONS_DIR}_V${VERSION} ${EXTRA_FUNCTIONS_DIRS}
do
	if [ -d $dir ]
	then
		cp $dir/* $TMP_FUNCTIONS_DIR
	fi
done
FUNCTIONS_DIR=${TMP_FUNCTIONS_DIR}

FUNCTION_LIST="/tmp/ifm.mk.functions_list.$$"
rm -f $FUNCTION_LIST $FUNCTIONS_FILE
touch $FUNCTION_LIST

function mk_add_script() {
	local PERMS="$1"
	local SCRIPT="$2"
	local FILE="$3"
	local EXEC_FUNCTIONS="$4"
	local OVERRIDE="$5"

	if [ -f "${FILE}" ]
	then
		echo ""
		echo ":put \"adding script $SCRIPT from ${FILE}\""
		local FIND_OPTS=""
		local EXISTS_MSG=""
		if [ "$OVERRIDE" = "0" ]
		then
			FIND_OPTS="source=\"X\""
			EXISTS_MSG=":put \"script $SCRIPT already exists and override disabled\""
		fi
		echo ":do {/system script add name=\"$SCRIPT\" source=\"X\" } on-error={ $EXISTS_MSG }"
		echo -n "/system script set [ find name=\"$SCRIPT\" $FIND_OPTS ] $PERMS source=\""
		if [ "$EXEC_FUNCTIONS" = "1" ]
		then
			echo -n "/system script run IFMMkFunctions\n" 
		fi
		if [ "$FUNCTION_LIST" != "" ] && [ -f "$FUNCTION_LIST" ]
		then
			grep "^##USE_FUNCTION" "${FILE}" | cut -f 2 -d " " >> $FUNCTION_LIST
		fi
		cat "${FILE}" | sed -e 's/\(["\$?]\)/\\\1/g' | sed -e 's/##USE_FUNCTION/:global/' | sed -e 's/$/\\n/' | tr -d '\n'
		echo "\""	
	fi
}

function mk_add_scripts() {
	local PERMS="$1"
	shift
	for fn in $*
	do
		if [ -f $fn ]
		then
			mk_add_script "$PERMS" "$fn" "$fn" 0 1
		else
			echo "# $fn not found"
		fi
	done
}

function check_dependencies() {
	local CNT_START=$(cat $FUNCTION_LIST | wc -l)
	sort $FUNCTION_LIST | uniq | while read func
	do
		grep "^##USE_FUNCTION" ${FUNCTIONS_DIR}/$func | cut -f 2 -d " " | while read depen
		do
			grep -q "^$depen\$" $FUNCTION_LIST
			if [ $? -ne 0 ]
			then
				echo "$depen" >> $FUNCTION_LIST
			fi
		done
	done
	local CNT_END=$(cat $FUNCTION_LIST | wc -l)
	if [ $CNT_START -ne $CNT_END ]
	then
		check_dependencies
	fi
	
}


if [ "$1" != "" ]
then
	mk_add_scripts "" $*
else
	# ftp permission is required for export but not for backup
	mk_add_script "policy=read,write,sensitive,test,password,policy,ftp dont-require-permissions=yes" IFMMkBackup scripts/IFMMkBackup 1 1 
	mk_add_script "policy=read,write,sensitive,test,password,policy dont-require-permissions=yes" IFMMkStats scripts/IFMMkStats 1 1 
	mk_add_script "policy=read,write,sensitive,test,password,policy dont-require-permissions=yes" IFMCheckNetwatch scripts/IFMCheckNetwatch 0 1
	mk_add_script "policy=read,write,sensitive,test,password,policy dont-require-permissions=yes" IFMCheckLoginFailures scripts/IFMCheckLoginFailures 0 1
	# IFMMkDefs will have definitions which will persist on reboots
	mk_add_script "policy=read,write,sensitive,test,password,policy dont-require-permissions=yes" IFMMkDefs scripts/IFMMkDefs 0 $DEFS_OVERRIDE
	if [ $MK_VERSION -eq 7 ]
	then
            mk_add_script "policy=read,write,sensitive,test,password,policy dont-require-permissions=yes" CheckGateways scripts_V7/CheckGateways 0 1
	fi
fi


if [ $ALL_FUNCTIONS -ne 1 ]
then
	check_dependencies
fi

cd $FUNCTIONS_DIR
if [ $ALL_FUNCTIONS -eq 1 ]
then
	ls -1  | grep -v -E ".*(~|.ignore|.bak)\$" > $FUNCTION_LIST
fi
HASH=$(sort $FUNCTION_LIST | grep -E -i "^[a-z0-9]+\$" | uniq | xargs cat | sha1sum | awk '{print $1}')
cd ..

cat >> $FUNCTIONS_FILE  <<__EOF__ 
# https://github.com/ivanfmartinez/Mikrotik
# generated : $(date) 
# functions included = $(sort $FUNCTION_LIST | uniq | tr "\n" " ")

:global IFMMkFunctionsHash
# The functions will be defined again only if changed

if ("\$IFMMkFunctionsHash" != "$HASH") do={
        :global IFMMkFunctionsHash "$HASH"
        :log info "loading IFMMkFunctions definitions"
__EOF__

sort $FUNCTION_LIST | grep -E -i "^[a-z0-9]+\$" | uniq | while read func 
do
	echo "# $func " >> $FUNCTIONS_FILE
	cat ${FUNCTIONS_DIR}/$func >> $FUNCTIONS_FILE
	echo "" >> $FUNCTIONS_FILE
done

cat >> $FUNCTIONS_FILE <<__EOF__ 

	if ([:len [/system script find name=IFMMkDefs]] = 1) do={
		/system script run IFMMkDefs
	}

}; 

__EOF__

mk_add_script 'policy=read,write,test,policy dont-require-permissions=yes' IFMMkFunctions $FUNCTIONS_FILE 0

cat <<__EOF__

:put "Adding schedulers"
# Automaticaly load functions on startup
/system scheduler add name=IFMMkFunctions    start-time=startup 		on-event="/system script run IFMMkFunctions" policy=read,write,test,policy
/system scheduler add name=IFMMkBackup       start-time=02:00:00 interval=1w 	on-event="/system script run IFMMkBackup" policy=read,write,password,sensitive,test,policy
/system scheduler add name=IFMCheckNetwatch  start-time=02:00:00 interval=5m 	on-event="/system script run IFMCheckNetwatch" policy=read,write,password,sensitive,test,policy

:put "Executing IFMMkFunctions"
/system script run IFMMkFunctions

__EOF__

if [ "$RM_FUNCTIONS_FILE" = "1" ]
then
    rm -f $FUNCTIONS_FILE 
fi
rm -f $FUNCTION_LIST 
rm -Rf $TMP_FUNCTIONS_DIR