bloopletech-webstats 0.1.0

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2009 Brenton Fletcher (http://i.bloople.net i@bloople.net)
+
+Permission is hereby granted, free of charge, to any person
+obtaining a copy of this software and associated documentation
+files (the "Software"), to deal in the Software without
+restriction, including without limitation the rights to use,
+copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the
+Software is furnished to do so, subject to the following
+conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
+OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
+HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
+WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+OTHER DEALINGS IN THE SOFTWARE.

data/README

@@ -0,0 +1,71 @@
+Webstats
+========
+
+Webstats is a server and clients that monitors your servers performance (CPU usage, memory usage, disk usage) and allows you to get notifications on your computer when the server is having problems, as well as showing (http://kimag.es/share/91090734.png) the current performance stats for the server on a web page.
+
+Webstats has 2 components: the _server_ and one or more _clients_.
+
+Server
+------
+The server runs on the computer whose performance you want to monitor; the server application's job is to (1) monitor the computer's performance and (2) report these performance statistics through a webserver that the server app runs.
+
+See Installation below for install instructions.
+
+The server app executes in the background and starts a tiny webserver on port 9970. You can view the performance stats for your server by going to http://<server's hostname>:9970/, for example if my server was under the domain name bloople.net, I would go to http://bloople.net:9970/ to see my performance stats. The statistics on this page automatically update every 5 seconds. You may need to open port 9970 if you have a firewall on your server. Note that the various clients get their data via the webserver the server app runs, just like the built-in stats page.
+
+You can prevent public access to your webstats by invoking the server application with a single argument (e.g. ruby webstats.rb <password>). Then, when you visit the webstats, you'll be prompted for a username and password; the username is always 'webstats', and the password is the password specified just above.
+
+The server application requires Linux kernel 2.6+, will not work in *BSD or OS X; the server application requires only Ruby; it does not rely on rubygems or any external libraries that aren't included with Ruby; the server application uses very little RAM, approximately 7MB.
+
+All the client applications depend on the server application being run to work.
+
+Clients
+-------
+If all you want to do is be able to view the performance stats for your server in a web browser, then you don't need to use any of these clients; going to http://<server's hostname>:9970/ will do just fine. But if you want to have Growl or email notification when something goes wrong, without having to look at a web page all the time, then you'll want one of the client applications that come with Webstats.
+
+Right now there's only one client application available, which is a Growl notifier. The Growl notifier only works on OS X, with RubyCocoa installed. This application runs in the background and monitors the server; if the server goes under high CPU load for more than a few seconds, or if the server is nearly out of memory, you'll get a Growl notification saying what the problem is. This way, you can just set and forget, knowing that Webstats will report any problems.
+
+To run the Growl notifier, open a terminal on your computer, then run (the first line is only needed the first time you run the notifier):
+~$ git clone git://github.com/bloopletech/webstats.git
+~$ cd webstats/clients/growl_notifier
+~/webstats/clients/growl_notifier$ ruby growl_notifier.rb http://<server's hostname>:9970/
+
+Where server's hostname is the host name of the server you want to monitor (e.g. bloople.net). The Growl notifier will then run in the background.
+
+Installation
+------------
+
+You can now install webstats using rubygems; this saves you a bit of hassle. To install webstats on the server via rubygems, ensure github gems are in your gems sources; then run:
+~$ sudo gem install bloopletech-webstats
+
+To run the server application, run:
+~$ webstats
+
+The server application will start in the background.
+
+Coming soon: growl notifier client via rubygems.
+
+If you don't want to use rubygems, then follow the below instructions:
+To install the server app, ssh into the computer you want to monitor. then run:
+~$ git clone git://github.com/bloopletech/webstats.git
+~$ cd webstats
+~/webstats$ cd server/data_providers
+~/webstats/server/data_providers$ ruby extconf.rb
+~/webstats/server/data_providers$ make
+
+To run the server app, ssh into the computer you want to monitor. then run:
+~$ cd webstats
+~/webstats$ ruby server/webstats.rb
+
+Todo
+----
+* (DONE) (Coming soon) Add disk space remaining, with warnings / danger signals.
+* (DONE) (Coming soon) Add optional authentication to server application.
+* (Coming soon) Make webstats configurable.
+* Add email notifier to client applications.
+* More client applications.
+* Extend server to work on *BSD (including OS X).
+* Make webstats server optionally install itself to run at boot.
+* Make client applications optionally install themselves at boot, where suitable.
+* (DONE) Make into a gem?
+* (Maybe) Move all server code into it's own module and make code work without running the web server, so you can use it to just grab stats in your own code.

data/Rakefile

@@ -0,0 +1,23 @@
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |s|
+      s.name = %q{webstats}
+      s.version = "0.1.0"
+
+      s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+      s.authors = ["Brenton Fletcher"]
+      s.date = %q{2009-04-22}
+      s.description = s.summary = %q{Display server CPU/Memory/Disk Usage on a web page, suitable for remote performance monitoring.}
+      s.email = %q{i@bloople.net}
+      s.files = Dir['**/*']
+      s.executables = ['webstats']
+      s.extensions = ["server/data_providers/extconf.rb"]
+      s.has_rdoc = false
+      s.homepage = %q{http://github.com/bloopletech/webstats}
+      s.require_paths = [""]
+  end
+rescue LoadError
+  puts "Jeweler not available. Install it with: sudo gem install technicalpickles-jeweler -s http://gems.github.com"
+end

data/VERSION.yml

@@ -0,0 +1,4 @@
+--- 
+:major: 0
+:minor: 1
+:patch: 0

data/bin/webstats

@@ -0,0 +1 @@
+require File.dirname(__FILE__) + '/../server/webstats'

data/clients/email_notifier/email_notifier.rb

@@ -0,0 +1,29 @@
+require 'rubygems'
+require 'net/smtp'
+require 'net/dns/resolver'
+require 'net/dns/rr'
+require 'time'
+
+def send_mail_hardcore(recipient, recipient_name, from, from_name, subject, message)
+  username, domain = recipient.split('@', 2) #yeah yeah, username can contain @ sign, will fix later
+
+  mxrs = Net::DNS::Resolver.new.mx(domain)
+  if mxrs.empty?
+    puts "No MX records on domain; bad domain name"
+    return
+  end
+
+  msg = <<END_OF_MESSAGE
+From: #{from_name || recipient_name} <#{from || recipient}>
+To: #{recipient_name} <#{recipient}>
+Subject: #{subject}
+Date: #{Time.now.rfc2822}
+Message-Id: <#{Time.now.to_i}.#{rand(10000000)}@#{domain}>
+
+#{message}
+END_OF_MESSAGE
+
+  Net::SMTP.start(mxrs.first.exchange, 25, domain) { |smtp| smtp.send_message msg, recipient, recipient }
+end
+
+send_mail_hardcore(*ARGV)

data/clients/growl_notifier/Growl.rb

@@ -0,0 +1,125 @@
+### Copyright
+#
+# Copyright 2004 Thomas Kollbach <dev@bitfever.de>
+# 
+# Released under the BSD license.
+#
+### Description
+#
+# A ruby class that enables posting notifications to the Growl daemon.
+# See <http://growl.info> for more information.
+# 
+# Requires RubyCocoa (http://www.fobj.com/rubycocoa/) and Ruby 1.8
+# (http://ruby-lang.org).
+#
+### Versions
+# 
+# v0.1- 25.11.2004 - Initial version, this is less more then a ruby translation of 
+#                    the python bindings
+#
+# TODO: transform this into a ruby-module, so it is usable as a mixin
+#       for ruby-scripts
+#
+### Usage
+#
+# Here is a short example how to use this in a script
+#
+#    n = GrowlNotifier.new('bla',['Foo'],nil,OSX::NSWorkspace.sharedWorkspace().iconForFileType_('unknown'))
+#    n.register()
+#
+#    n.notify('Foo', 'Test Notification', 'Blah blah blah')   
+#
+###
+
+require 'osx/cocoa'
+
+$priority = {"Very Low" => -2,
+             "Moderate" => -1,
+             "Normal"   =>  0,
+             "High"     =>  1,
+             "Emergency"=>  2
+             }
+             
+
+class GrowlNotifier
+#    A class that abstracts the process of registering and posting
+#    notifications to the Growl daemon.
+#
+#    `appName': The name of the application
+#    `notifications': an array of notifications - default is an empty array
+#
+#    `defaultNotifications': optional - defaults to the value of
+#    `notifications'  
+#    `appIcon' is also optional but defaults to a senseless icon so 
+#    so you are higly encouraged to pass it along.
+#
+
+    def initialize(appName='GrowlNotifier', notifications=[], defaultNotifications=nil, appIcon=nil)
+        @appName = appName
+        @notifications = notifications
+        @defaultNotifications = defaultNotifications
+        @appIcon = appIcon       
+    end #initialize
+    
+    def register
+        if @appIcon == nil  then
+            @appIcon = OSX::NSWorkspace.sharedWorkspace().iconForFileType_("txt") 
+        end
+        if @defaultNotifications == nil then
+            @defaultNotifications = @notifications
+        end
+        
+        regData = {
+            'ApplicationName'=>@appName,
+            'AllNotifications'=> OSX::NSArray.arrayWithArray(@notifications),
+            'DefaultNotifications'=> OSX::NSArray.arrayWithArray(@defaultNotifications),
+            'ApplicationIcon'=> @appIcon.TIFFRepresentation
+                  }
+                  
+        dict = OSX::NSDictionary.dictionaryWithDictionary(regData)
+        notifyCenter = OSX::NSDistributedNotificationCenter.defaultCenter
+        
+        notifyCenter.postNotificationName_object_userInfo_deliverImmediately_("GrowlApplicationRegistrationNotification", nil, dict, true)
+    end #register
+
+    def notify(noteType, title, description, icon=nil, appIcon=nil, sticky=false, priority=nil)
+#        Post a notification to the Growl daemon.
+#
+#        `noteType' is the name of the notification that is being posted.
+#        `title' is the user-visible title for this notification.
+#        `description' is the user-visible description of this notification.
+#        `icon' is an optional icon for this notification.  It defaults to
+#            `@applicationIcon'.
+#        `appIcon' is an optional icon for the sending application.
+#        `sticky' is a boolean controlling whether the notification is sticky.
+
+ 
+       @notifications << noteType
+       if icon == nil then icon = @appIcon end
+       
+       notification = {'NotificationName'=> noteType,
+             'ApplicationName'=> @appName,
+             'NotificationTitle'=> title,
+             'NotificationDescription'=> description,
+             'NotificationIcon'=> icon.TIFFRepresentation()}
+       
+       unless appIcon == nil 
+            notification['NotificationAppIcon'] = appicon.TIFFRepresentation
+       end     
+       
+       if sticky 
+            notification['NotificationSticky'] = OSX::NSNumber.numberWithBool_(true) 
+       end
+            
+       unless priority == nil 
+            notification['NotificationPriority'] = OSX::NSNumber.numberWithInt_(priority)
+       end
+            
+       d = OSX::NSDictionary.dictionaryWithDictionary_(notification)
+       
+       notCenter = OSX::NSDistributedNotificationCenter.defaultCenter()
+       notCenter.postNotificationName_object_userInfo_deliverImmediately_('GrowlNotification', nil, d, true)
+       
+    end #notify
+end #class growlnotifier
+

data/clients/growl_notifier/growl_notifier.rb

@@ -0,0 +1,45 @@
+exit if fork
+
+require 'rubygems'
+require 'json'
+require 'net/http'
+require 'uri'
+require 'Growl'
+
+url = ARGV[0]
+
+g = GrowlNotifier.new("Webstats for #{url}",['Webstats Notification'], nil, OSX::NSWorkspace.sharedWorkspace().iconForFileType_('unknown'))
+g.register
+
+meta_info = JSON.parse(Net::HTTP.get(URI.join(url, "information")))
+last_warnings_text = last_danger_text = nil
+last_time = 0
+
+while(true)
+  data = JSON.parse(Net::HTTP.get(URI.join(url, "update")))
+
+  bad = data.sort { |a, b| b[1]['importance'].to_f <=> a[1]['importance'].to_f }.select { |(k, v)| !v['status'].nil? && v['status'] != '' }
+
+  has_warnings = bad.detect { |(k, v)| v['status'] == 'warning' }
+  has_dangers = bad.detect { |(k, v)| v['status'] == 'danger' }
+  
+  title = []
+  title << "Danger" if has_dangers
+  title << "Warnings" if has_warnings
+  title = title.join(" & ") + " for host #{URI.parse(url).host}"
+  
+  warnings_text = has_warnings ? "Warnings for #{bad.select { |(k, v)| v['status'] == 'warning' }.map { |(k, v)| meta_info[k]['in_sentence'] }.join(", ")}." : nil
+  danger_text = has_dangers ? "Dangerous situation for #{bad.select { |(k, v)| v['status'] == 'danger' }.map { |(k, v)| meta_info[k]['in_sentence'] }.join(", ")}." : nil
+
+  if last_warnings_text != warnings_text or danger_text != last_danger_text or (Time.now - last_time) > 60
+    last_warnings_text = warnings_text
+    last_danger_text = danger_text
+    last_time = Time.now
+
+    unless bad.empty?
+      g.notify "Webstats Notification", title, [danger_text, warnings_text].compact.join(" "), nil, nil, true, (has_dangers ? 2 : 1)
+    end
+  end
+
+  sleep(10)
+end

data/server/data_providers/cpu_info.rb

@@ -0,0 +1,72 @@
+class DataProviders::CpuInfo
+  def initialize
+    @readings = []
+    @mutex = Mutex.new
+
+    @thread = Thread.new do
+      last_time = last_user = last_nice = last_system = last_idle = last_iowait = 0
+      first_time = true
+      while(true)
+        time = (Time.new.to_f * 1000).to_i
+
+        user, nice, system, idle, iowait, crap = IO.readlines("/proc/stat").first.split(' ', 6).map { |i| i.to_i }
+
+        if first_time
+          first_time = false
+        else
+          temp_user = (user - last_user)
+          temp_nice = (nice - last_nice)
+          temp_system = (system - last_system)
+          temp_idle = (idle - last_idle)
+          temp_iowait = (iowait - last_iowait)
+
+          @mutex.synchronize do
+            @readings.unshift((temp_user + temp_nice + temp_system) / (temp_idle.to_f < 1 ? 1 : temp_idle.to_f))
+            @readings.pop while @readings.length > 5
+          end
+        end
+        last_user = user
+        last_nice = nice
+        last_system = system
+        last_idle = idle
+        last_iowait = iowait
+        last_time = time
+        sleep(2.5)
+      end
+    end
+  end  
+    
+  def get
+    out = { :usage => 0 }
+    @mutex.synchronize do
+      unless @readings.empty?
+        out[:usage] = @readings.first
+        out[:status] = 'warning' unless @readings.detect { |r| out[:usage] < 95 }
+        out[:status] = 'danger' unless @readings.detect { |r| out[:usage] < 99.5 }
+      end
+    end
+    out[:loadavg_1], out[:loadavg_5], out[:loadavg_15] = IO.readlines("/proc/loadavg").first.split(' ', 4).map { |v| v.to_f }
+    out
+  end
+
+  def renderer
+    information.merge({ :contents => %{
+sc.innerHTML = "<div class='major_figure'><span class='title'>Usage</span><span class='figure'>" + data_source['usage'] + "</span><span class='unit'>%</span></div>" + 
+"<div class='major_figure'><span class='title'>Load average</span><span class='figure'>" + data_source['loadavg_1'] +
+"</span><span class='unit'>1m</span><span class='divider'>/</span><span class='figure'>" + data_source['loadavg_5'] +
+"</span><span class='unit'>5m</span><span class='divider'>/</span><span class='figure'>" + data_source['loadavg_15'] + "</span><span class='unit'>15m</span></div>";
+} })
+  end
+
+  def information
+    { :name => "CPU Info", :in_sentence => 'CPU load', :importance => importance }
+  end
+
+  def importance
+    100
+  end
+
+  def kill
+    @thread.kill
+  end
+end

data/server/data_providers/disk_activity.rb

@@ -0,0 +1,50 @@
+class DataProviders::DiskActivity
+  def initialize
+    @reads_sec = 0
+    @writes_sec = 0
+
+    @thread = Thread.new do
+      last_time = last_reads = last_writes = 0
+      first_time = true
+      while(true)
+        time = (Time.new.to_f * 1000).to_i
+
+        reads, writes = IO.readlines("/proc/diskstats").map { |l| parts = l.split; [parts[5].to_i, parts[7].to_i] }.inject([0, 0]) { |sum, vals| [sum[0] + vals[0], sum[1] + vals[1]] }
+
+        if first_time
+          first_time = false
+        else
+          @reads_sec = ((reads - last_reads) / ((time - last_time).to_f / 1000.0)) * 512
+          @writes_sec = ((writes - last_writes) / ((time - last_time).to_f / 1000.0)) * 512
+        end
+        last_reads = reads
+        last_writes = writes
+        last_time = time
+        sleep(2.5)
+      end
+    end
+  end
+
+  def get
+    { :reads => @reads_sec / 1024.0, :writes => @writes_sec / 1024.0 }
+  end
+
+  def renderer
+    information.merge({ :name => "Disk Activity", :in_sentence => "Disk Activity", :importance => importance, :contents => %{
+sc.innerHTML = "<div class='major_figure'><span class='title'>Reads</span><span class='figure'>" + data_source['reads'] + "</span><span class='unit'>mb/s</span></div>" +
+"<div class='major_figure'><span class='title'>Writes</span><span class='figure'>" + data_source['writes'] + "</span><span class='unit'>mb/s</span></div>";
+} })
+  end
+
+  def information
+    { :name => "Disk Activity", :in_sentence => "Disk Activity", :importance => importance }
+  end
+
+  def importance
+    70
+  end
+
+  def kill
+    @thread.kill
+  end
+end

data/server/data_providers/disk_usage.c

@@ -0,0 +1,29 @@
+#include <ruby.h>
+#include <sys/vfs.h>
+
+static VALUE get_disk_usage(VALUE self, VALUE mount_point)
+{
+   char* mount_point_str = RSTRING(mount_point)->ptr;
+
+   VALUE out_hash = rb_hash_new();
+   rb_hash_aset(out_hash, rb_str_new2("free"), Qnil);
+   rb_hash_aset(out_hash, rb_str_new2("total"), Qnil);
+
+   struct statfs result;
+
+   if(statfs(mount_point_str, &result) == 0)
+   {
+      rb_hash_aset(out_hash, rb_str_new2("free"), LL2NUM((long long)result.f_bfree * (long long)result.f_bsize));
+      rb_hash_aset(out_hash, rb_str_new2("total"), LL2NUM((long long)result.f_blocks * (long long)result.f_bsize));
+   }
+
+   return out_hash;
+}
+
+void Init_disk_usage()
+{
+   VALUE mDataProviders = rb_const_get(rb_cObject, rb_intern("DataProviders"));
+   VALUE cDiskUsage = rb_const_get(mDataProviders, rb_intern("DiskUsage"));
+   
+   rb_define_method(cDiskUsage, "get_disk_usage", get_disk_usage, 1);
+}

data/server/data_providers/disk_usage.rb

@@ -0,0 +1,53 @@
+class DataProviders::DiskUsage
+  def initialize
+  end
+
+  def get
+    out = { :mounts => [] }
+
+    mtab = IO.readlines("/etc/mtab").sort_by { |l| l.split[1] }
+    mtab.map do |mp|
+      parts = mp.split
+      next unless parts[3].split(",").detect { |p| p == "rw" }
+      du = get_disk_usage(parts[1])
+      out[:mounts] << [parts[1], du] unless du['total'] == 0 or (du['total'] > 5242880 && ((du['total'] - du['free']) <= 1048576))
+    end
+
+    out[:mounts].map do |mp|
+      mp[1]['free'] /= (1024.0 * 1024)
+      mp[1]['total'] /= (1024.0 * 1024)
+      out[:status] = "warning" if mp[1]['free'] < 50 and mp[1]['total'] > 100 and out[:status] != 'danger'
+      out[:status] = "danger" if mp[1]['free'] < 10 and mp[1]['total'] > 20
+    end
+
+    out
+  end
+
+  def renderer
+    information.merge({ :name => "Disk Usage by Mount Point", :in_sentence => "Disk Usage", :importance => importance, :contents => %{
+var temp = "";
+for(var i = 0; i < data_source['mounts'].length; i++)
+{
+   var mpd = data_source['mounts'][i][1];
+   temp += "<div class='major_figure'><span class='title'>" + data_source['mounts'][i][0] + "</span><span class='figure'>" + mpd['free'] +
+    "</span><span class='unit'>mb free</span><span class='divider'>/</span><span class='figure'>" + mpd['total'] +
+     "</span><span class='unit'>mb total</span></div>";
+}
+
+sc.innerHTML = temp;
+} })
+  end
+
+  def information
+    { :name => "Disk Usage by Mount Point", :in_sentence => "Disk Usage", :importance => importance }
+  end
+
+  def importance
+    80
+  end
+
+  def kill
+  end
+end
+
+require File.dirname(__FILE__) + '/disk_usage.so'

data/server/data_providers/extconf.rb

@@ -0,0 +1,5 @@
+require 'mkmf'
+
+dir_config("c_extensions")
+
+create_makefile("disk_usage")

data/server/data_providers/mem_info.rb

@@ -0,0 +1,52 @@
+class DataProviders::MemInfo
+  def initialize
+    @readings = []
+    @mutex = Mutex.new
+
+    @thread = Thread.new do
+      while(true)
+        out = {}
+        out[:total], out[:free], out[:buffers], out[:cached] = IO.readlines("/proc/meminfo")[0..4].map { |l| l =~ /^.*?\: +(.*?) kB$/; $1.to_i / 1024.0 }
+        out[:free_total] = out[:free] + out[:buffers] + out[:cached]
+
+        @mutex.synchronize do
+          @readings.unshift(out)
+          @readings.pop while @readings.length > 5
+        end
+        sleep(2.5)
+      end
+    end
+  end
+
+  def get
+    out = { :total => 0, :free => 0, :buffers => 0, :cached => 0, :free_total => 0 }
+    @mutex.synchronize do
+      unless @readings.empty?
+        out = @readings.first.dup
+        out[:status] = 'warning' unless @readings.detect { |r| r[:free] > 5 }
+        out[:status] = 'danger' unless @readings.detect { |r| r[:free_total] > 1 }
+      end
+    end
+    out
+  end
+
+  def renderer
+    information.merge({ :contents => %{
+sc.innerHTML = "<div class='major_figure'><span class='title'>Free</span><span class='figure'>" + data_source['free'] + "</span><span class='unit'>mb</span></div>" +
+"<div class='major_figure'><span class='title'>Free -buffers/cache</span><span class='figure'>" + data_source['free_total'] + "</span><span class='unit'>mb</span></div>" +
+"<div class='major_figure'><span class='title'>Total</span><span class='figure'>" + data_source['total'] + "</span><span class='unit'>mb</span></div>";
+} })
+  end
+
+  def information
+    { :name => "Memory Info", :in_sentence => 'Memory Usage', :importance => importance }
+  end
+
+  def importance
+    90
+  end
+
+  def kill
+    @thread.kill
+  end
+end

data/server/webstats.rb

@@ -0,0 +1,212 @@
+require 'webrick'
+
+if $DEBUG
+  Thread.abort_on_exception
+else
+  exit if fork
+  $stdout = File.new('/dev/null', 'w')
+  $stderr = File.new('/dev/null', 'w')
+end
+
+Thread.new do
+  while(true)
+    sleep(300)
+    GC.start
+  end
+end
+
+class String
+  def underscore
+    self.gsub(/::/, '/').
+     gsub(/([A-Z]+)([A-Z][a-z])/,'\1_\2').
+     gsub(/([a-z\d])([A-Z])/,'\1_\2').
+     tr("-", "_").
+     downcase
+  end
+  alias_method :to_json, :inspect
+end
+
+class Numeric
+  def formatted(precision = 1)
+    rounded_number = (Float(self) * (10 ** precision)).round.to_f / 10 ** precision
+    parts = ("%01.#{precision}f" % rounded_number).to_s.split('.')
+    parts[0].gsub!(/(\d)(?=(\d\d\d)+(?!\d))/, "\\1,")
+    parts.join(".")
+  end
+  alias_method :to_json, :inspect
+end
+
+class Array
+  def to_json
+    "[#{map { |e| e.to_json }.join(',')}]"
+  end
+end
+
+class Hash
+  def formatted
+    out = self.dup
+    out.each_pair { |k, v| out[k] = v.formatted }
+    out
+  end
+  def to_json
+    arr = []
+    each_pair { |k, v| arr << "#{k.to_json}:#{v.to_json}" }
+    "{#{arr.join(',')}}"
+  end
+end
+
+class Symbol
+  def to_json
+    to_s.inspect
+  end
+end
+
+module DataProviders
+  DATA_SOURCES = {}
+  def self.setup
+    Dir.glob("#{File.dirname(__FILE__)}/data_providers/*.rb").each { |file| load file }
+    DataProviders.constants.each do |c|
+      c = DataProviders.const_get(c)
+      DATA_SOURCES[c.to_s.gsub(/^DataProviders::/, '').underscore] = c.new if c.is_a? Class
+    end
+  end
+end
+
+DataProviders.setup
+
+class Webstats < WEBrick::HTTPServlet::AbstractServlet
+  def do_GET(req, res)
+    WEBrick::HTTPAuth.basic_auth(req, res, "Webstats") { |user, pass| user == 'webstats' and pass == ARGV[0] } unless ARGV.empty?
+
+    body = ""
+    if req.path_info == '/'
+      body << <<-EOF
+<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01//EN" "http://www.w3.org/TR/html4/strict.dtd">
+<html>
+  <head>
+    <title>Webstats</title>
+    <style type="text/css">
+      * { margin: 0; padding: 0; font-family: "Lucida Grande", Helvetica, Arial, sans-serif; font-size: 100%; }
+      body { font-size: 95%; }
+      p { margin: 0 0 1em 0; }
+
+      h1 { margin: 1em; }
+      h1 span { font-size: 160%; font-weight: bold; }
+      .source .danger { background-color: #FF0D33; }
+      .source .warning { background-color: #F1FF28; }
+
+      .source { width: 500px; border: 1px solid #000000; margin: 1em; }
+      .source h2 { padding: 0 0.8em 0 0.8em; background-color: #C98300; }
+      .source h2 span { font-size: 130%; font-weight: bold; padding: 0.2em 0; display: block; }
+      .source .source_contents { padding: 0.8em; }
+      .source .title { padding-right: 0.5em; }
+      .source .major_figure { font-size: 130%; margin: 0.3em 0; }
+      .source .major_figure .figure { font-size: 120%; font-weight: bold; font-family: Georgia, serif; }
+      .source .major_figure .unit { font-family: Georgia, serif; font-size: 70%; }
+      .source .minor_figure { font-family: Georgia, serif; }
+      .source .divider { margin-left: 0.2em; margin-right: 0.2em; font-weight: normal; }
+    </style>
+    <script type="text/javascript">
+       var http = null;
+
+       function getLatest()
+       {
+          http.open("get", "/update", true);
+          http.send(null);
+       }
+
+       window.onload = function()
+       {
+          http = !!(window.attachEvent && !window.opera) ? new ActiveXObject("Microsoft.XMLHTTP") : new XMLHttpRequest();
+
+          http.onreadystatechange = function()
+          {
+            if(http.readyState == 4)
+            {
+               var results = eval("(" + http.responseText + ")");
+               if(!results) return;
+EOF
+
+      DataProviders::DATA_SOURCES.each_pair do |k, v|
+        body << %{var data_source = results['#{k}']; var sc = document.getElementById('source_contents_#{k}'); sc.className = "source_contents " + (data_source['status'] ? data_source['status'] : ''); #{v.renderer[:contents]}\n}
+      end
+
+body << <<-EOF
+            }
+         };
+
+         window.setInterval("getLatest()", 5000);
+         getLatest();
+      }
+    </script>
+  </head>
+  <body id="body">
+    <div id="main">
+      <h1><span>Stats for #{req.host}</span></h1>
+EOF
+      DataProviders::DATA_SOURCES.sort { |a, b| b[1].importance <=> a[1].importance }.each do |(k, v)|
+        r = v.renderer
+        body << %{<div class="source" id="source_#{k}"><h2><span>#{r[:name]}</span></h2><div class="source_contents" id="source_contents_#{k}">Loading...</div></div>}
+      end
+
+      body << <<-EOF
+    </div>
+  </body>
+</html>
+EOF
+    elsif req.path_info == '/update'
+      out = {}
+      DataProviders::DATA_SOURCES.each_pair do |k, v|
+        out[k] = v.get
+      end
+
+      fix_leaves_hash(out)
+
+      body << out.to_json
+    elsif req.path_info == '/information'
+      out = {}
+      DataProviders::DATA_SOURCES.each_pair { |k, v| out[k] = v.information }
+      body << out.to_json
+    end
+
+    res.body = body
+    res['Content-Type'] = "text/html"
+  end
+
+  private
+  def fix_leaves_array(array)
+    array.each_with_index do |v, i|
+      if v.is_a? Numeric
+        array[i] = v.formatted
+      elsif v.is_a? Hash
+        array[i] = fix_leaves_hash(array[i])
+      elsif v.is_a? Array
+        array[i] = fix_leaves_array(array[i])
+      end
+    end
+  end
+
+  def fix_leaves_hash(hash)
+    hash.each_pair do |k, v|
+      if v.is_a? Numeric
+        hash[k] = v.formatted
+      elsif v.is_a? Hash
+        hash[k] = fix_leaves_hash(hash[k])
+      elsif v.is_a? Array
+        hash[k] = fix_leaves_array(hash[k])
+      end
+    end
+  end
+end
+
+s = WEBrick::HTTPServer.new(:Port => 9970)
+
+death = proc do
+  s.shutdown
+  DataProviders::DATA_SOURCES.each_pair { |k, v| v.kill }
+end
+trap("INT", death)
+trap("TERM", death)
+
+s.mount("/", Webstats)
+s.start

data/webstats.gemspec

@@ -0,0 +1,52 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{webstats}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Brenton Fletcher"]
+  s.date = %q{2009-04-22}
+  s.default_executable = %q{webstats}
+  s.description = %q{Display server CPU/Memory/Disk Usage on a web page, suitable for remote performance monitoring.}
+  s.email = %q{i@bloople.net}
+  s.executables = ["webstats"]
+  s.extensions = ["server/data_providers/extconf.rb"]
+  s.extra_rdoc_files = [
+    "LICENSE",
+    "README"
+  ]
+  s.files = [
+    "LICENSE",
+    "README",
+    "Rakefile",
+    "VERSION.yml",
+    "bin/webstats",
+    "clients/email_notifier/email_notifier.rb",
+    "clients/growl_notifier/Growl.rb",
+    "clients/growl_notifier/growl_notifier.rb",
+    "server/data_providers/cpu_info.rb",
+    "server/data_providers/disk_activity.rb",
+    "server/data_providers/disk_usage.c",
+    "server/data_providers/disk_usage.rb",
+    "server/data_providers/extconf.rb",
+    "server/data_providers/mem_info.rb",
+    "server/webstats.rb",
+    "webstats.gemspec"
+  ]
+  s.homepage = %q{http://github.com/bloopletech/webstats}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = [""]
+  s.rubygems_version = %q{1.3.1}
+  s.summary = %q{Display server CPU/Memory/Disk Usage on a web page, suitable for remote performance monitoring.}
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 2
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+    else
+    end
+  else
+  end
+end

metadata

@@ -0,0 +1,69 @@
+--- !ruby/object:Gem::Specification 
+name: bloopletech-webstats
+version: !ruby/object:Gem::Version 
+  version: 0.1.0
+platform: ruby
+authors: 
+- Brenton Fletcher
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-04-22 00:00:00 -07:00
+default_executable: webstats
+dependencies: []
+
+description: Display server CPU/Memory/Disk Usage on a web page, suitable for remote performance monitoring.
+email: i@bloople.net
+executables: 
+- webstats
+extensions: 
+- server/data_providers/extconf.rb
+extra_rdoc_files: 
+- LICENSE
+- README
+files: 
+- LICENSE
+- README
+- Rakefile
+- VERSION.yml
+- bin/webstats
+- clients/email_notifier/email_notifier.rb
+- clients/growl_notifier/Growl.rb
+- clients/growl_notifier/growl_notifier.rb
+- server/data_providers/cpu_info.rb
+- server/data_providers/disk_activity.rb
+- server/data_providers/disk_usage.c
+- server/data_providers/disk_usage.rb
+- server/data_providers/extconf.rb
+- server/data_providers/mem_info.rb
+- server/webstats.rb
+- webstats.gemspec
+has_rdoc: false
+homepage: http://github.com/bloopletech/webstats
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- ""
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 2
+summary: Display server CPU/Memory/Disk Usage on a web page, suitable for remote performance monitoring.
+test_files: []
+