#!/bin/sh
#
#
. /etc/scripts/globals

# imported from HFSC script
#set class numbers

N_CLASS_D_1=70
N_CLASS_D_2=100
N_CLASS_U_1=130
N_CLASS_U_2=160
SH_TMP="/etc/scripts/tmp1.txt"

if [ "$1" = "config" ]; then

        echo "graph_title HFSC Traffic Shaping Stats - UP/DOWN"
        echo 'graph_vlabel bytes DOWN(-)/UP(+) per ${graph_period}'
        echo 'graph_width 450'
	echo 'graph_category network'

	j=1

	while [ $j -le $L_USERS ]
	do
	        echo "${USERNAMES[${j}]}_down.label ${USERNAMES[${j}]}"
	        echo "${USERNAMES[${j}]}_down.type COUNTER"
	        echo "${USERNAMES[${j}]}_down.graph no"

		if [ $j == 1 ]; then
		        echo "${USERNAMES[${j}]}_down.draw AREA"
		else
		        echo "${USERNAMES[${j}]}_down.draw STACK"
		fi;

	        echo "${USERNAMES[${j}]}_down.info Stats for ${USERNAMES[${j}]} - ${USER_IP[${j}]}"
	        echo "${USERNAMES[${j}]}_down.min 0"
	        echo "${USERNAMES[${j}]}_down.max 130000"

	        echo "${USERNAMES[${j}]}_up.label ${USERNAMES[${j}]}"
	        echo "${USERNAMES[${j}]}_up.type COUNTER"
	        echo "${USERNAMES[${j}]}_up.negative ${USERNAMES[${j}]}_down"

		if [ $j == 1 ]; then
		        echo "${USERNAMES[${j}]}_up.draw AREA"
		else
		        echo "${USERNAMES[${j}]}_up.draw STACK"
		fi;

	        echo "${USERNAMES[${j}]}_up.info Stats for ${USERNAMES[${j}]} - ${USER_IP[${j}]}"
	        echo "${USERNAMES[${j}]}_up.min 0"
	        echo "${USERNAMES[${j}]}_up.max 30000"


		j=$[$j+1]

	done;

#customized colours
	echo 'Serwer_down.colour 000000'
	echo 'Serwer_up.colour 000000'

        exit 0

fi;

#### DOWNLOAD

temp1=`/sbin/tc -s class show dev imq0 > $SH_TMP`

#echo "$tttt" | while read line
while read line
do
	test_hfsc=`echo $line | grep "hfsc"`

	j=1

        while [ $j -le $L_USERS ]
        do
		case $test_hfsc in

		*hfsc[\ ]1:$[$N_CLASS_D_1+$j]*)
		# check  N_CLASS_D_1 stats for every user
			read line
			temp1=`echo $line | awk '{ print $2; }'`
			STAT_USER[$j]=$[${STAT_USER[$j]}+$temp1]
#			echo "N_CLASS_D_1="$temp1
#			echo "N_CLASS_D_1_SUM "$j" ="${STAT_USER[$j]}
		;;
		*hfsc[\ ]1:$[$N_CLASS_D_2+$j]*)
		# check  N_CLASS_D_2 stats for every user
			read line
			temp1=`echo $line | awk '{ print $2; }'`
			STAT_USER[$j]=$[${STAT_USER[$j]}+$temp1]
#			echo "N_CLASS_D_2="$temp1
#			echo "N_CLASS_D_2_SUM "$j" ="${STAT_USER[$j]}
		;;
		esac

                j=$[$j+1]

        done;

done < $SH_TMP


####

j=1
while [ $j -le $L_USERS ]
do
	echo ${USERNAMES[${j}]}"_down.value "${STAT_USER[$j]}
	j=$[$j+1]
done;

#reset values

j=1
while [ $j -le $L_USERS ]
do
	STAT_USER[$j]=0
	j=$[$j+1]
done;

#### UPLOAD

temp1=`/sbin/tc -s class show dev imq1 > $SH_TMP`

#echo "$tttt" | while read line
while read line
do
	test_hfsc=`echo $line | grep "hfsc"`

	j=1

        while [ $j -le $L_USERS ]
        do
		case $test_hfsc in

		*hfsc[\ ]1:$[$N_CLASS_U_1+$j]*)
		# check  N_CLASS_U_1 stats for every user
			read line
			temp1=`echo $line | awk '{ print $2; }'`
			STAT_USER[$j]=$[${STAT_USER[$j]}+$temp1]
#			echo "N_CLASS_U_1="$temp1
#			echo "N_CLASS_U_1_SUM "$j" ="${STAT_USER[$j]}
		;;
		*hfsc[\ ]1:$[$N_CLASS_U_2+$j]*)
		# check  N_CLASS_U_2 stats for every user
			read line
			temp1=`echo $line | awk '{ print $2; }'`
			STAT_USER[$j]=$[${STAT_USER[$j]}+$temp1]
#			echo "N_CLASS_U_2="$temp1
#			echo "N_CLASS_U_2_SUM "$j" ="${STAT_USER[$j]}
		;;
		esac

                j=$[$j+1]

        done;

done < $SH_TMP

j=1
while [ $j -le $L_USERS ]
do
	echo ${USERNAMES[${j}]}"_up.value "${STAT_USER[$j]}
	j=$[$j+1]
done;


## clean temp file
temp1=`echo "" > $SH_TMP`