rsrcmeter for TxD (See related posts)

#!/usr/local/bin/ruby
Dir.chdir(ENV['HOME'])
email_result = false

if ARGV.include?('-h')
  puts <<EOM
TextDrive (Unofficial) Disk Usage and Bandwidth Reporter
To have results e-mailed, just provide an e-mail address
(i.e. rsrcmeter [email protected])
EOM
  Kernel.exit(0)
end
if ARGV[0] != nil and ARGV[0] =~ /^[-A-Za-z0-9!\#\$%&'*+\/\=?_`{|}~][-A-Za-z0-9!\#\$%&'*+\/\=?_`{|}~.]*[-A-Za-z0-9!\#\$%&'*+\/\=?_`{|}~]@[-A-Za-z0-9][-A-Za-z0-9.]*[-A-Za-z0-9]$/
  send_to = ARGV[0]
  email_result = true
else
  puts "--> WARNING: Invalid e-mail address specified. Cannot e-mail results."
end

# Disk usage
quotaline = `quota -g | tail -n 1`
usage = `echo -n "#{quotaline}" | awk '{print $2}'`.to_f
quota = `echo -n "#{quotaline}" | awk '{print $3}'`.to_f
percent_used = (usage / quota) * 100
du_result = "Disk usage: " + sprintf("%.4f", usage/1024) + " MiB (Quota: " + sprintf("%.4f", quota/1024/1024) +" GiB; " + sprintf("%.1f", percent_used) + "% used)"
du_result += "\nWARNING: Using over 90% of your disk quota!" if percent_used > 90.0
puts du_result unless email_result

# HTTP Bandwidth
print "Calculating Bandwidth Usage..." unless email_result
month = `date +"%B %Y"`.chomp
access_logs="access_log." + `date +%Y%m`.chomp + "??"
system("cat logs/access_log 2>/dev/null > temp-bandwidthcount")
system("cat domains/*/logs/access_log 2>/dev/null >> temp-bandwidthcount")
system("cat logs/#{access_logs} 2>/dev/null >> temp-bandwidthcount")
system("cat domains/*/logs/#{access_logs} 2>/dev/null >> temp-bandwidthcount")
system("zcat logs/#{access_logs}.gz 2>/dev/null >> temp-bandwidthcount")
system("zcat domains/*/logs/#{access_logs}.gz 2>/dev/null >> temp-bandwidthcount")


usage = `cat temp-bandwidthcount | awk '{sum += $10} END {print sum}'`.chomp.to_f / 1024 / 1024
File.delete("temp-bandwidthcount")

bw_result = "Bandwidth used for #{month}: " + sprintf("%.4f", usage) + " MiB"
unless email_result
  30.times {print "\b"}
  puts bw_result
end

if email_result
  message = <<EOM
From: TxD Resource Meter <#{ENV['USER']}-noreply@#{`/bin/hostname`.chomp}>
To: #{send_to}
Subject: [TxD Resource Meter] #{Time.now} report for #{ENV['USER']}
X-Mailer: rsrcmeter / http://textsnippets.com/posts/show/632

Resource Report - #{Time.now}
#{du_result} 
#{bw_result}
EOM
File.open("temp-emailresult", "w") { |file| file.print message }
`cat temp-emailresult | sendmail -t`
File.delete("temp-emailresult")
end