pimon 0.1.0

data/config/test.yml

@@ -0,0 +1,27 @@
+basic_auth:
+  enabled: true
+  username: pimon
+  password: pimon
+redis:
+  socket: /thou/shalt/not/use/redis/on/test/environment
+chart:
+  cpu:
+    color: '#D2691E'
+  disk:
+    color: '#CDC673'
+  mem:
+    color: '#87CEFA'
+  temp:
+    color: '#FF9B04'
+  swap:
+    color: '#3CB371'
+queues:
+  time: pimon_time
+  cpu:  pimon_cpu
+  disk: pimon_disk
+  mem:  pimon_mem
+  swap: pimon_swap
+  temp: pimon_temp
+stats_collector:
+  number_of_stats: 6
+  time_period_in_min: 10

data/config/test_broken.yml

@@ -0,0 +1,22 @@
+basic_auth:
+  enabled: true
+redis:
+  socket: /thou/shalt/not/use/redis/on/test/environment
+chart:
+  cpu:
+    color: '#D2691E'
+  disk:
+    color: '#CDC673'
+  mem:
+    color: '#87CEFA'
+  swap:
+    color: '#3CB371'
+queues:
+  time: pimon_time
+  cpu:  pimon_cpu
+  disk: pimon_disk
+  mem:  pimon_mem
+  swap: pimon_swap
+stats_collector:
+  number_of_stats: 6
+  time_period_in_min: 10

data/config/thin/config.yml

@@ -0,0 +1,13 @@
+environment: production
+chdir: /home/pi/app/pimon/current
+address: 127.0.0.1
+user: pi
+group: pi
+port: 4567
+pid: /home/pi/app/pimon/shared/pids/thin.pid
+rackup: /home/pi/app/pimon/current/config/config.ru
+log: /home/pi/app/pimon/shared/log/thin.log
+max_conns: 1024
+timeout: 30
+max_persistent_conns: 512
+daemonize: true

data/lib/pimon.rb

@@ -0,0 +1,63 @@
+$: << File.dirname(__FILE__)
+
+if File.file?("Gemfile")
+  require 'bundler'
+  Bundler.require(:default, ENV['RACK_ENV'])
+end
+require 'eventmachine'
+require 'haml'
+require 'pimon/pimon_config'
+require 'pimon/stats_collector'
+require 'sinatra'
+
+class Pimon < Sinatra::Base
+  set :public_folder, "#{File.dirname(__FILE__)}/pimon/public"
+  set :views,         "#{File.dirname(__FILE__)}/pimon/views"
+  
+  def self.configure_basic_auth
+    if settings.config.is_basic_auth_enabled?
+      use Rack::Auth::Basic, "Restricted Area" do |username, password|
+        [username, password] == config.basic_auth
+      end
+    end
+  end
+  
+  configure :development, :production do
+    require 'redis'
+    
+    config = PimonConfig.create_new(ENV['RACK_ENV'])
+    
+    EventMachine::next_tick do
+      settings.timer = EventMachine::add_periodic_timer(config.stats[:time_period_in_min] * 60) do
+        settings.stats_checker.collect_stats
+      end
+    end
+    
+    set :config, config
+    set :stats_checker, StatsCollector.new(config, Redis.new(:path => config.redis[:socket]))
+    set :timer, nil
+    
+    self.configure_basic_auth
+  end
+  
+  configure :test do
+    require 'mock_redis'
+    
+    config = PimonConfig.create_new('test')
+    
+    set :config, config
+    set :stats_checker, StatsCollector.new(config, MockRedis.new)
+    set :timer, nil
+    
+    self.configure_basic_auth
+  end
+  
+  get '/' do
+    last_update = settings.stats_checker.last_update
+    last_modified(last_update) if ENV['RACK_ENV'] != 'development' && last_update
+    
+    @o = settings.stats_checker.show_stats
+    
+    haml :index
+  end
+end

data/lib/pimon/hash_extensions.rb

@@ -0,0 +1,12 @@
+# As seen in https://gist.github.com/151324
+module HashExtensions
+  def symbolize_keys
+    inject({}) do |acc, (k,v)|
+      key = String === k ? k.to_sym : k
+      value = Hash === v ? v.symbolize_keys : v
+      acc[key] = value
+      acc
+    end
+  end
+end
+Hash.send(:include, HashExtensions)

data/lib/pimon/pimon_config.rb

@@ -0,0 +1,75 @@
+require 'pimon/hash_extensions'
+require 'yaml'
+
+class PimonConfig
+  def self.create_new(environment = nil)
+    config = self.new(environment)
+    
+    return config if config.valid?
+  end
+  
+  def basic_auth
+    if is_basic_auth_enabled? && has_authentication_details?
+      [@config[:basic_auth][:username], @config[:basic_auth][:password]]
+    else
+      nil
+    end
+  end
+  
+  def chart
+    @config[:chart]
+  end
+  
+  def environment
+    @config[:environment]
+  end
+  
+  def is_basic_auth_enabled?
+    @config[:basic_auth][:enabled]
+  end
+  
+  def queues
+    @config[:queues]
+  end
+  
+  def redis
+    @config[:redis]
+  end
+  
+  def stats
+    @config[:stats_collector]
+  end
+  
+  def valid?
+    raise "Basic auth enabled with no authentication details" if is_basic_auth_enabled? && !has_authentication_details?
+    raise "Redis has no socket details" unless has_redis_details?
+    true
+  end
+  
+  private
+  
+  def initialize(environment)
+    begin
+      filename = "#{File.dirname(__FILE__)}/../../config/#{ environment || 'development' }.yml"
+      @config = YAML.load_file(filename).symbolize_keys
+      @config.merge!({ :environment => "#{ environment || 'development'}"})
+      @config.freeze
+    rescue Exception => e
+      puts "Error while loading config file: #{filename}"
+      raise e
+    end
+  end
+  
+  def has_authentication_details?
+    @config[:basic_auth].has_key?(:username) && @config[:basic_auth].has_key?(:password)
+  end
+  
+  def has_redis_details?
+    # Thou shalt not use a live redis in test environment
+    is_test_environment? || (@config.has_key?(:redis) && @config[:redis].has_key?(:socket))
+  end
+  
+  def is_test_environment?
+    @config[:environment] == 'test'
+  end
+end

data/lib/pimon/probe/cpu_usage.rb

@@ -0,0 +1,11 @@
+require 'pimon/probe/probe'
+
+class Probe::CpuUsage < Probe
+  def self.check
+    100 - `vmstat 1 2`.split(/\n/)[3].split(" ")[14].to_i
+  end
+  
+  def self.symbol
+    :cpu
+  end
+end

data/lib/pimon/probe/disk_usage.rb

@@ -0,0 +1,11 @@
+require 'pimon/probe/probe'
+
+class Probe::DiskUsage < Probe
+  def self.check
+    `df`.split(/\n/)[1].split(" ")[4].to_i
+  end
+  
+  def self.symbol
+    :disk
+  end
+end

data/lib/pimon/probe/memory_usage.rb

@@ -0,0 +1,13 @@
+require 'pimon/probe/probe'
+require 'pimon/probe/system_memory'
+
+class Probe::MemoryUsage < Probe
+  
+  def self.check
+    SystemMemory.check(:mem)
+  end
+  
+  def self.symbol
+    :mem
+  end
+end

data/lib/pimon/probe/probe.rb

@@ -0,0 +1,13 @@
+class Probe
+  def self.check
+    raise "Not implemented"
+  end
+
+  def self.symbol
+    raise "Not implemented"
+  end
+  
+  def self.unit
+    '%'
+  end
+end

data/lib/pimon/probe/swap_usage.rb

@@ -0,0 +1,13 @@
+require 'pimon/probe/probe'
+require 'pimon/probe/system_memory'
+
+class Probe::SwapUsage < Probe
+  
+  def self.check
+    SystemMemory.check(:swap)
+  end
+  
+  def self.symbol
+    :swap
+  end
+end

data/lib/pimon/probe/system_memory.rb

@@ -0,0 +1,19 @@
+class SystemMemory
+  
+  def self.check(type)
+    case type
+    when :mem
+      index = 1
+    when :swap
+      index = 2
+    else
+      raise "Undefined memory type: #{type}"
+    end
+    
+    memory = `free -o -m`.split(/\n/)[index].split(" ")
+    
+    # memory[1] holds total memory
+    # memory[2] holds used memory
+    ((memory[2].to_f / memory[1].to_f) * 100).to_i
+  end
+end

data/lib/pimon/probe/temperature.rb

@@ -0,0 +1,16 @@
+# encoding: UTF-8
+require 'pimon/probe/probe'
+
+class Probe::Temperature < Probe
+  def self.check
+    `cat /sys/class/thermal/thermal_zone0/temp`[0..1].to_i
+  end
+  
+  def self.symbol
+    :temp
+  end
+  
+  def self.unit
+    'ºC'
+  end
+end

data/lib/pimon/stats_collector.rb

@@ -0,0 +1,65 @@
+require 'date'
+
+require "pimon/probe/cpu_usage"
+require "pimon/probe/disk_usage"
+require "pimon/probe/memory_usage"
+require "pimon/probe/swap_usage"
+require "pimon/probe/temperature"
+
+class StatsCollector
+  def initialize(config, redis)
+    @config = config
+    @redis = redis
+    @probes = [Probe::CpuUsage, Probe::MemoryUsage, Probe::SwapUsage, Probe::DiskUsage, Probe::Temperature]
+  end
+  
+  def collect_stats
+    pop_old_stats
+    
+    @redis.rpush(@config.queues[:time], Time.now.strftime("%Y-%m-%d %H:%M:%S"))
+    @probes.each do |probe|
+      @redis.rpush(@config.queues[probe.symbol], probe.check)
+    end
+  end
+  
+  def last_update
+    time = @redis.lindex(@config.queues[:time], @config.stats[:number_of_stats] - 1)
+    
+    DateTime.parse(time) if time
+  end
+  
+  def show_stats
+    time = @redis.lrange(@config.queues[:time], 0, -1)
+    
+    stats = {
+              :time => { :stats => time.map { |t| (/\d\d:\d\d:\d\d/.match(t))[0] } },
+              :refresh_interval_in_millis => @config.stats[:time_period_in_min] * 60 * 1000
+            }
+    
+    @probes.each do |probe|
+      stats[probe.symbol] =
+       { 
+         :stats => @redis.lrange(@config.queues[probe.symbol],  0, -1).map(&:to_i),
+         :color => @config.chart[probe.symbol][:color],
+         :unit => probe.unit
+       }
+    end
+    
+    stats
+  end
+  
+  private
+   
+  def pop_all_queues
+    @redis.lpop(@config.queues[:time])
+    @probes.each { |probe| @redis.lpop(@config.queues[probe.symbol]) }
+  end
+  
+  def pop_old_stats
+    queue_size = @redis.llen(@config.queues[:time])
+    
+    if queue_size >= @config.stats[:number_of_stats]
+      (queue_size - @config.stats[:number_of_stats] + 1).times { pop_all_queues }
+    end
+  end
+end

data/lib/pimon/version.rb

@@ -0,0 +1,3 @@
+module Pimon
+  VERSION = '0.1.0'
+end

data/lib/pimon/views/index.haml

@@ -0,0 +1,10 @@
+%html
+  %header
+    %title Pimon
+    %script{ :src => '//cdnjs.cloudflare.com/ajax/libs/jquery/1.8.0/jquery-1.8.0.min.js' }
+    %script{ :src => '//cdnjs.cloudflare.com/ajax/libs/highcharts/2.3.1/highcharts.js' }
+    %script{ :src => '//lipka.github.com/piecon/piecon.min.js' }
+  %body
+    #chart
+    
+    = render(:haml,:'index.js',:locals => { :o => @o })

data/lib/pimon/views/index.js.haml

@@ -0,0 +1,111 @@
+:javascript
+  $(function () {
+    var chart,
+        cpu = #{o[:cpu][:stats]},
+        disk = #{o[:disk][:stats]},
+        mem = #{o[:mem][:stats]},
+        swap = #{o[:swap][:stats]},
+        time = #{o[:time][:stats]},
+        temp = #{o[:temp][:stats]},
+        titleStats = [ { 'legend' : 'CPU', 'stat' : cpu[cpu.length - 1], 'color' : '#{o[:cpu][:color]}', 'unit' : '#{o[:cpu][:unit]}' } ,
+                       { 'legend' : 'MEM', 'stat' : mem[mem.length - 1], 'color' : '#{o[:mem][:color]}', 'unit' : '#{o[:mem][:unit]}' } ,
+                       { 'legend' : 'SWAP', 'stat' : swap[swap.length - 1], 'color' : '#{o[:swap][:color]}', 'unit' : '#{o[:swap][:unit]}'},
+                       { 'legend' : 'DISK', 'stat' : disk[disk.length - 1], 'color' : '#{o[:disk][:color]}', 'unit' : '#{o[:disk][:unit]}'},
+                       { 'legend' : 'TEMP', 'stat' : temp[temp.length - 1], 'color' : '#{o[:temp][:color]}', 'unit' : '#{o[:temp][:unit]}'} ],
+        i = 0;
+    
+    $(document).ready(function() {
+        chart = new Highcharts.Chart({
+            chart: {
+                renderTo: 'chart',
+                type: 'line',
+                marginRight: 130,
+                marginBottom: 25
+            },
+            title: {
+                text: 'Pimon',
+                x: -20
+            },
+            subtitle: {
+                text: 'My raspberry PI',
+                x: -20
+            },
+            xAxis: {
+                categories: time
+            },
+            yAxis: {
+                title: {
+                    text: 'Usage'
+                },
+                plotLines: [{
+                    value: 0,
+                    width: 1,
+                    color: '#808080'
+                }],
+                max: 100,
+                min: 0
+            },
+            tooltip: {
+                formatter: function() {
+                        return '<b>'+ this.series.name +'</b><br/>'+
+                        this.x +': '+ this.y + (this.series.name === 'temp' ? '#{o[:temp][:unit]}' : '%');
+                }
+            },
+            legend: {
+                layout: 'vertical',
+                align: 'right',
+                verticalAlign: 'top',
+                x: -10,
+                y: 100,
+                borderWidth: 0
+            },
+            series: [{
+                name: 'cpu',
+                color: '#{o[:cpu][:color]}',
+                data: cpu
+            },
+            {
+                name: 'mem',
+                color: '#{o[:mem][:color]}',
+                data: mem
+            },
+            {
+                name: 'swap',
+                color: '#{o[:swap][:color]}',
+                data: swap
+            },
+            {
+                name: 'disk',
+                color: '#{o[:disk][:color]}',
+                data: disk
+            },
+            {
+                name: 'temp',
+                color: '#{o[:temp][:color]}',
+                data: temp
+            }]
+        });
+    });
+    
+    function changeFavicon() {
+      Piecon.setOptions({
+        color: titleStats[i]['color'], // Pie chart color
+        background: '#bbb', // Empty pie chart color
+        shadow: '#fff', // Outer ring color
+        fallback: 'force' // Toggles displaying percentage in the title bar (possible values - true, false, 'force')
+      });
+      Piecon.setProgress(titleStats[i]['stat']);
+      
+      $('title').text(titleStats[i]['stat'] + titleStats[i]['unit'] + ' ' + titleStats[i]['legend'])
+      i++;
+      if (i >= titleStats.length) {
+        i = 0;
+      } 
+    }
+    
+    setInterval(changeFavicon, 3000);
+    changeFavicon();
+    setTimeout(function() {
+      window.location.reload();
+    }, #{@o[:refresh_interval_in_millis]});
+  });