#!/bin/bash
# Pavel Ruzicka <ruza@ruza.eu>
# based on http://pivo.glogow.net.pl/linux/docs/lm-25032004.pdf

RRDFILE="${PWD}/torstats/torstats.rrd"
GRAPH="${RRDFILE/.rrd/.png}"

TOR=$(/sbin/iptables -L -n -x -v |egrep 'owner UID match (118|debian-tor)'|\
head -n 1| cut -d ' ' -f 2 )


rrdtool update $RRDFILE N:$TOR

rrdtool graph $GRAPH -a PNG --title="BrmTor - DAILY" --vertical-label "bytes" \
DEF:tor=$RRDFILE:tor:AVERAGE AREA:tor#0000ff:tor 1>/dev/null

rrdtool graph ${GRAPH/.png/-weekly.png} -a PNG --start -1w --title="BrmTor - \
WEEKLY" --vertical-label "bytes" DEF:tor=$RRDFILE:tor:AVERAGE AREA:tor#0000ff:tor 1>/dev/null

rrdtool graph ${GRAPH/.png/-monthly.png} -a PNG --start -1m --title="BrmTor - \
MONTHLY" --vertical-label "bytes" DEF:tor=$RRDFILE:tor:AVERAGE AREA:tor#0000ff:tor 1>/dev/null

rrdtool graph ${GRAPH/.png/-yearly.png} -a PNG --start -1y --title="BrmTor - \
YEARLY" --vertical-label "bytes" DEF:tor=$RRDFILE:tor:AVERAGE AREA:tor#0000ff:tor 1>/dev/null