#!/usr/bin/ksh 

# This script organizes config files for MRTG
# and restarts the MRTG processes cleanly.

# Author: Scott Ripley, U.S. House of Representatives
# Version: 1.0 (and all that a 1.0 release entails)
# Contact: scott.ripley@mail.house.gov
# Copyright: (C) 2001 Scott Ripley 

# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# See the GNU General Public License for more details.


# Change these directores to match your configuration for MRTG
MRTG=/usr/local/mrtg/bin
# TARHOME is the location of the back-up files. It should be another volume
TARHOME=/usr/local/mrtg/bin
# Location of log files
WEBHOME=/usr/local/apache/htdocs/mrtg_images

cd $MRTG 

# Initialize the variables and temp files

typeset -i CFGLINE
typeset -i MAXLINE
CFGLINE=1
MAXLINE=1
CFG="$MRTG"/mrtgcfgs.run
CFGNUM="$MRTG"/cfgnum.run
MCNAME=`uname -n`
EXIT=N
DATE=`date '+%m%d%y%H%M' `
# TARHOME is the location of the back-up files. It should be another volume
TARHOME=/usr/local/mrtg/bin
WEBHOME=/usr/local/apache/htdocs/mrtg_images

# Set tabs for display columns
stty tabs
tabs 10,30,40,45 
 
# Main loop
while [ "$EXIT" = N ]
do
	# reads the pid files and inputs them into a file 
	ls -go *pid | sed 's/[ ][ ]/ /g' | cut -d"." -f1 > "$CFG"

	nl -bt -nrz -w2 "$CFG" > "$CFGNUM" # numbers the lines in the file

	# sets the number of configs loaded, for error testing
	MAXLINE=`wc -l "$CFG" | cut -d" " -f7 `

	clear
	printf "\n\nThe following MRTG configs are running on "$MCNAME":\n"

	printf "\n\tConfig name\t\tRunning Since\n"
	printf "\t-----------\t\t-------------\n"
	

		# prints out the appropriate fields on-screen
		nawk ' { print "\t" $1,$8".cfg \t\t" $5,$6,$7 } ' "$CFGNUM"
		printf "\n"	
		printf " Which file would you like to reload? (b to back-up, q to quit) \c"
		read X

	case "$X" in
	[qQ] )
		EXIT=Y # I quit!
	;;

	[bB] )
		printf "\n Backup *.cfg & *.log files? \c"
		read V
		case "$V" in
		'y' | 'Y' )
			
			CNAME=MRTG_"$DATE".cfgs.tar
			tar cvf "$TARHOME"/"$CNAME" "$MRTG"/*.cfg
			chmod 0400 "$TARHOME"/"$CNAME"
			LNAME=MRTG_"$DATE".logs.tar
			find "$WEBHOME" | egrep  -e "\.log$|\.old$" > mrtgtar.run
			tar cvf "$LNAME" -I mrtgtar.run
			chmod 0600 "$TARHOME"/"$CNAME"
			chmod 0600 "$TARHOME"/"$LNAME"
		#	printf "\nGzipping files..."
		#	gzip "$TARHOME"/"$CNAME" 
		#	gzip "$TARHOME"/"$LNAME" 
			printf "\nConfig files backed up to `ls "$TARHOME"/MRTG_"$DATE".cfg*` "
			printf "\nLog files backed up to `ls "$TARHOME"/MRTG_"$DATE".log*` "
			printf "\nPress return to continue..."
			read U
			;;
		* )
			printf "\nBackup aborted."
					sleep 3
			;;
		esac
	;;

	[0-9][0-9] ) # If the number entered is between 0 and 99...

		# and if the number is <= the no. of configs loaded
		if [ "$X" -le "$MAXLINE" ] # 
		then
				
		FILE=`sed -n ''$X'p' "$CFGNUM" | cut -d" " -f11 ` # put filename into $FILE

		echo " Okay to restart "$FILE".cfg? \c"	
		read Y

		case "$Y" in 
		'y' | 'Y' )
			FKILL=`cat "$MRTG""/""$FILE"".pid"` # Set the variable with the PID
			kill -9 $FKILL	 			# Kill it 
			rm $MRTG/$FILE".pid"			# Remove the file
			$MRTG"/mrtg" $FILE".cfg"		# Restart MRTG with the config file
			sleep 8					# Wait for new PID file to be created
			echo "New PID for "$FILE".cfg is `cat "$MRTG"/"$FILE".pid` "
		;;
		*)
			echo "\nRestart aborted!"
		;;
		esac
		else
			echo "Entry Invalid! Press return to continue..."
			read ZZ
		fi
	;;
	*)
		echo "Entry Invalid! Press return to continue..."
                read ZZ
	;;
	esac

done
	# For security/housekeeping reasons, we clean out the temp files here.
	cat /dev/null >	 "$MRTG"/mrtgcfgs.run
	cat /dev/null >	 "$MRTG"/cfgnum.run
	cat /dev/null >	 "$MRTG"/cfgsed.run
	cat /dev/null >	 "$MRTG"/mrtgtar.run
exit 0