#!/bin/sh
#
#    sens_update_rrd -
#	Update a sensors rrd database.
#	Sample usage:
#		sens_update_rrd /var/lib/database.rrd w83781d-isa-0290
#	Sample cron entry:
#		*/5 * * * * /usr/local/bin/sens_update_rrd /var/lib/sensors-rrd/sensors.rrd w83781d-isa-0290
#
#################################################################
#
#    Copyright 2001,2005 Mark D. Studebaker <mdsxyz123@yahoo.com>
#
#    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
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with this program; if not, write to the Free Software
#    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
#
#################################################################
#
if [ $# -ne 2 ]
then
	echo "usage: $0 database.rrd sensor"
	echo "       sensor example: w83781d-isa-0290 (2.4 kernel)"
	echo "                       0-0290 (early 2.6 kernel)"
	echo "                       hwmon0 (kernel 2.6.14 or later)"
	exit 1
fi
#
RRDPATH=/usr/local/rrdtool/bin
RRDB=$1

SENSDIR=/proc/sys/dev/sensors
SDIR=/sys/bus/i2c/devices
HWMONDIR=/sys/class/hwmon
SENSDEV=$2
if [ -d $HWMONDIR ]
then
	SYSFS=1
	SENSDIR=$HWMONDIR
	SENSDEV=$SENSDEV/device
elif [ -d $SDIR ]
then
	SYSFS=1
	SENSDIR=$SDIR
elif [ ! -d $SENSDIR]
then
	echo $0: 'No sensors found! (modprobe sensor modules?)'
	exit 1
fi

SENS=$SENSDIR/$SENSDEV
if [ ! -r $SENS ]
then
	echo $0: 'No sensors found! (modprobe sensor modules?)'
	exit 1
fi

STRING=N
if [ "$SYSFS" = "1" ]
then
	#
	# Get the value from these sensor files (/sys)
	#
	SENSORS="fan1 fan2 fan3"
	for i in $SENSORS
	do
		V="`cat $SENSDIR/$SENSDEV/${i}_input 2> /dev/null`"
		if [ $? -ne 0 ]
		then
			STRING="${STRING}:U"
		else
			STRING="${STRING}:${V}"
		fi
	done
	#
	# Get the value from these sensor files (/sys) and divide by 1000
	#
	SENSORS="temp1 temp2 temp3 in0 in1 in2 in3 in4 in5 in6"
	for i in $SENSORS
	do
		V="`cat $SENSDIR/$SENSDEV/${i}_input 2> /dev/null`"
		if [ $? -ne 0 ]
		then
			STRING="${STRING}:U"
		else
			V=`echo "3k0 ${V/-/_} 1000/p"|dc`
			STRING="${STRING}:${V}"
		fi
	done
else
	#
	# Get the second value from these sensor files (/proc)
	#
	SENSORS="fan1 fan2 fan3"
	for i in $SENSORS
	do
		V="`cat $SENSDIR/$SENSDEV/$i 2> /dev/null`"
		if [ $? -ne 0 ]
		then
			STRING="${STRING}:U"
		else
			V="`echo $V | cut -d ' ' -f 2`"
			STRING="${STRING}:${V}"
		fi
	done
	#
	# Get the third value from these sensor files (/proc)
	#
	SENSORS="temp1 temp2 temp3 in0 in1 in2 in3 in4 in5 in6"
	for i in $SENSORS
	do
		V="`cat $SENSDIR/$SENSDEV/$i 2> /dev/null`"
		if [ $? -ne 0 ]
		then
			STRING="${STRING}:U"
		else
			V="`echo $V | cut -d ' ' -f 3`"
			STRING="${STRING}:${V}"
		fi
	done
fi

#
# Get the first value from these /proc files
#
SENSORS="loadavg"
for i in $SENSORS
do
	V="`cat /proc/$i 2> /dev/null`"
	if [ $? -ne 0 ]
	then
		STRING="${STRING}:U"
	else
		V="`echo $V | cut -d ' ' -f 1`"
		STRING="${STRING}:${V}"
	fi
done

$RRDPATH/rrdtool update $RRDB $STRING
