pimon 0.1.11 → 0.2.0

data/lib/pimon/public/piecon.js

@@ -0,0 +1,194 @@
+//
+// piecon.js
+//
+// https://github.com/lipka/piecon
+//
+// Copyright (c) 2012 Lukas Lipka <lukaslipka@gmail.com>. All rights reserved.
+//
+
+(function(){
+    var Piecon = {};
+
+    var currentFavicon = null;
+    var originalFavicon = null;
+    var originalTitle = null;
+    var canvas = null;
+    var options = {};
+    var defaults = {
+        color: '#ff0084',
+        background: '#bbb',
+        shadow: '#fff',
+        fallback: false
+    };
+
+    var isRetina = window.devicePixelRatio > 1;
+
+    var ua = (function () {
+        var agent = navigator.userAgent.toLowerCase();
+        return function (browser) {
+            return agent.indexOf(browser) !== -1;
+        };
+    }());
+
+    var browser = {
+        ie: ua('msie'),
+        chrome: ua('chrome'),
+        webkit: ua('chrome') || ua('safari'),
+        safari: ua('safari') && !ua('chrome'),
+        mozilla: ua('mozilla') && !ua('chrome') && !ua('safari')
+    };
+
+    var getFaviconTag = function() {
+        var links = document.getElementsByTagName('link');
+
+        for (var i = 0, l = links.length; i < l; i++) {
+            if (links[i].getAttribute('rel') === 'icon' || links[i].getAttribute('rel') === 'shortcut icon') {
+                return links[i];
+            }
+        }
+
+        return false;
+    };
+
+    var removeFaviconTag = function() {
+        // Removed link element will not exist in Nodelist
+        var links = Array.prototype.slice.call(document.getElementsByTagName('link'));
+        var head = document.getElementsByTagName('head')[0];
+
+        for (var i = 0, l = links.length; i < l; i++) {
+            if (links[i].getAttribute('rel') === 'icon' || links[i].getAttribute('rel') === 'shortcut icon') {
+                head.removeChild(links[i]);
+            }
+        }
+    };
+
+    var setFaviconTag = function(url) {
+        removeFaviconTag();
+
+        var link = document.createElement('link');
+        link.type = 'image/x-icon';
+        link.rel = 'icon';
+        link.href = url;
+
+        document.getElementsByTagName('head')[0].appendChild(link);
+    };
+
+    var getCanvas = function () {
+        if (!canvas) {
+            canvas = document.createElement("canvas");
+            if (isRetina) {
+                canvas.width = 32;
+                canvas.height = 32;
+            } else {
+                canvas.width = 16;
+                canvas.height = 16;
+            }
+        }
+
+        return canvas;
+    };
+
+    var drawFavicon = function(percentage) {
+        var canvas = getCanvas();
+        var context = canvas.getContext("2d");
+        var percentage = percentage || 0;
+        var src = currentFavicon;
+
+        var faviconImage = new Image();
+        faviconImage.onload = function() {
+            if (context) {
+                context.clearRect(0, 0, canvas.width, canvas.height);
+
+                // Draw shadow
+                context.beginPath();
+                context.moveTo(canvas.width / 2, canvas.height / 2);
+                context.arc(canvas.width / 2, canvas.height / 2, Math.min(canvas.width / 2, canvas.height / 2), 0, Math.PI * 2, false);
+                context.fillStyle = options.shadow;
+                context.fill();
+
+                // Draw background
+                context.beginPath();
+                context.moveTo(canvas.width / 2, canvas.height / 2);
+                context.arc(canvas.width / 2, canvas.height / 2, Math.min(canvas.width / 2, canvas.height / 2) - 2, 0, Math.PI * 2, false);
+                context.fillStyle = options.background;
+                context.fill();
+
+                // Draw pie
+                if (percentage > 0) {
+                    context.beginPath();
+                    context.moveTo(canvas.width / 2, canvas.height / 2);
+                    context.arc(canvas.width / 2, canvas.height / 2, Math.min(canvas.width / 2, canvas.height / 2) - 2, (-0.5) * Math.PI, (-0.5 + 2 * percentage / 100) * Math.PI, false);
+                    context.lineTo(canvas.width / 2, canvas.height / 2);
+                    context.fillStyle = options.color;
+                    context.fill();
+                }
+
+                setFaviconTag(canvas.toDataURL());
+            }
+        };
+
+        // allow cross origin resource requests if the image is not a data:uri
+        // as detailed here: https://github.com/mrdoob/three.js/issues/1305
+        if (!src.match(/^data/)) {
+            faviconImage.crossOrigin = 'anonymous';
+        }
+
+        faviconImage.src = src;
+    };
+
+    var updateTitle = function(percentage) {
+        if (percentage > 0) {
+            document.title = '(' + percentage + '%) ' + originalTitle;
+        } else {
+            document.title = originalTitle;
+        }
+    };
+
+    Piecon.setOptions = function(custom) {
+        options = {};
+
+        for (var key in defaults){
+            options[key] = custom.hasOwnProperty(key) ? custom[key] : defaults[key];
+        }
+
+        return this;
+    };
+
+    Piecon.setProgress = function(percentage) {
+        if (!originalTitle) {
+            originalTitle = document.title;
+        }
+
+        if (!originalFavicon || !currentFavicon) {
+            var tag = getFaviconTag();
+            originalFavicon = currentFavicon = tag ? tag.getAttribute('href') : '/favicon.ico';
+        }
+
+        if (!isNaN(parseFloat(percentage)) && isFinite(percentage)) {
+            if (!getCanvas().getContext || browser.ie || browser.safari || options.fallback === true) {
+                // Fallback to updating the browser title if unsupported
+                return updateTitle(percentage);
+            } else if (options.fallback === 'force') {
+                updateTitle(percentage);
+            }
+
+            return drawFavicon(percentage);
+        }
+
+        return false;
+    };
+
+    Piecon.reset = function() {
+        if (originalTitle) {
+            document.title = originalTitle;
+        }
+
+        if (originalFavicon) {
+            currentFavicon = originalFavicon;
+            setFaviconTag(currentFavicon);
+        }
+    };
+
+    Piecon.setOptions(defaults);
+    window.Piecon = Piecon;
+})();

data/lib/pimon/public/style.css

@@ -0,0 +1,13 @@
+.hostname {
+  text-align: center;
+}
+
+.single_stat {
+  padding-left: 20px;
+}
+
+.chart {
+  display: inline-block;
+  min-width: 25%;
+  padding: 0 20px;
+}

data/lib/pimon/stats_collector.rb

@@ -1,75 +1,75 @@
 require 'date'
 require 'json'
-require 'pimon/probe/cpu_usage'
-require 'pimon/probe/disk_usage'
-require 'pimon/probe/memory_usage'
-require 'pimon/probe/swap_usage'
-require 'pimon/probe/time_check'
-require 'pimon/probe/temperature'
-require 'pimon/probe/uptime'
-require 'pimon/stats'
 
-class StatsCollector
-  def initialize(config)
-    @config = config
-    @probes = [
-      Probe::TimeCheck,
-      Probe::CpuUsage,
-      Probe::MemoryUsage,
-      Probe::SwapUsage,
-      Probe::DiskUsage,
-      Probe::Temperature,
-      Probe::Uptime
+require 'pimon/probe'
+
+module Pimon
+  class StatsCollector
+    attr_accessor :config
+    attr_accessor :last_update
+    attr_accessor :probes
+    attr_accessor :probe_threads
+    attr_reader :stats
+
+    CHART_PROBES = [
+      Pimon::Probe::CpuUsage,
+      Pimon::Probe::MemoryUsage,
+      Pimon::Probe::SwapUsage,
+      Pimon::Probe::DiskUsage,
+      Pimon::Probe::Temperature
     ]
-    @stats = Stats.new(@probes.map(&:symbol).concat([:time]))
-  end
-  
-  def collect_stats
-    pop_old_stats
-    
-    @probes.each do |probe|
-      @stats.push(probe.symbol, probe.check)
+
+    SINGLE_PROBES = [
+      Pimon::Probe::Uptime,
+      Pimon::Probe::CpuFreq
+    ]
+
+    def initialize(config)
+      @config = config
+      @probes = {
+        charts: CHART_PROBES,
+        single: SINGLE_PROBES
+      }
+      @stats = { colors: config.colors, charts: {}, single: {}, hostname: config.hostname }
+      @probe_threads = []
+      @collection_mutex = Mutex.new
     end
-  end
-  
-  def last_update
-    time = @stats.index(:time, @config.stats[:number_of_stats] - 1)
-    
-    DateTime.parse(time) if time
-  end
-  
-  def show_stats
-    time = @stats.range(:time)
-    uptime = @stats.range(:uptime)
-    
-    stats = {
-      :time => { :stats => time.map { |t| (/\d\d:\d\d:\d\d/.match(t))[0] } },
-      :hostname => @config.hostname,
-      :uptime => uptime
-    }
-    
-    @probes.each do |probe|
-      stats[probe.symbol] = {
-        :stats => @stats.range(probe.symbol),
-        :color => @config.chart[probe.symbol][:color],
-        :unit => probe.unit
-      } unless [:time, :uptime].include?(probe.symbol)
+
+    def collect_stats
+      @collection_mutex.synchronize do
+        self.last_update = Time.now
+
+        [:charts, :single].each do |stats_type|
+          probes[stats_type].each do |probe|
+            self.probe_threads = []
+            probe_threads << Thread.new do
+              sample = probe.check(last_update)
+              stats[stats_type][sample.probe_name] = sample
+            end
+          end
+        end
+      end
     end
-    
-    stats.to_json
-  end
-  
-  private
-   
-  def pop_all_queues
-    @probes.each { |probe| @stats.shift(probe.symbol) }
-  end
-  
-  def pop_old_stats
-    queue_size = @stats.range(:time).length
-    
-    if queue_size >= @config.stats[:number_of_stats]
-      (queue_size - @config.stats[:number_of_stats] + 1).times { pop_all_queues }
+
+    def show_stats
+      wait_for_stats_collection { stats }
+    end
+
+    def charts
+      wait_for_stats_collection { stats[:charts].keys }
+    end
+
+    def single_stats
+      wait_for_stats_collection { stats[:single].keys }
+    end
+
+    private
+
+    def wait_for_stats_collection
+      @collection_mutex.synchronize do
+        probe_threads.each(&:join)
+        yield if block_given?
+      end
     end
   end
 end

data/lib/pimon/version.rb

@@ -1,3 +1,3 @@
 module Pimon
-  VERSION = '0.1.11'.freeze
+  VERSION = '0.2.0'.freeze
 end

data/lib/pimon/views/index.haml

@@ -1,12 +1,22 @@
 %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 => '/piecon.min.js' }
+    -#%link{ :href => '/favicon.ico', :rel => :icon, :type => :"image/x-icon" }
+    %link{ :href => '/normalize.css', :rel => :stylesheet, :type => :"text/css" }
+    %link{ :href => '/style.css', :rel => :stylesheet, :type => :"text/css" }
+    %script{ :src => 'https://cdnjs.cloudflare.com/ajax/libs/jquery/1.8.0/jquery-1.8.0.min.js' }
+    %script{ :src => '/piecon.js' }
+    %script{ :src => 'https://cdnjs.cloudflare.com/ajax/libs/smoothie/1.27.0/smoothie.min.js' }
     %script{ :src => '/index.js' }
+
   %body
-    #chart
-    #uptime
-      Uptime: 
-      %strong.up
+    %h1.hostname= hostname
+    - single_stats.each do |stat_name|
+      .single_stat{ :id => stat_name }
+        %strong= "#{stat_name.upcase}: "
+        %span.stat
+
+    - charts.each do |chart_name|
+      .chart
+        %h2= chart_name.upcase
+        %canvas{ :id => "#{chart_name}_chart" }

data/pimon.gemspec

@@ -1,35 +1,36 @@
-# encoding: UTF-8
-$: << File.expand_path('./lib')
-require 'pimon/version'
+require_relative 'lib/pimon/version'
 
 Gem::Specification.new do |s|
-  s.name        = "pimon"
-  s.version     = Pimon::VERSION.dup
+  s.name        = 'pimon'
+  s.version     = Pimon::VERSION
+  s.licenses    = ['WTFPL']
   s.platform    = Gem::Platform::RUBY
-  s.authors     = [ 'Pedro Carriço' ]
-  s.email       = [ 'pedro.carrico@gmail.com' ]
+  s.authors     = ['Pedro Carriço']
+  s.email       = ['pedro.carrico@gmail.com']
   s.homepage    = 'http://pimon.pedrocarrico.net/'
   s.summary     = 'Pimon - Raspberry Pi server monitor'
   s.description = 'Pimon is a simple server monitor designed for the Raspberry Pi.'
   s.post_install_message = 'Thank you for using Pimon :-)'
-  
+
   s.required_ruby_version = '>= 1.9'
-  
-  s.files         = `git ls-files`.split("\n")
-  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
-  s.executables   = `git ls-files -- bin/pimon`.split("\n").map{ |f| File.basename(f) }
+
+  s.files         = Dir.glob('{lib,config}/**/*') << 'pimon.gemspec'
+  s.test_files    = Dir.glob('{test,spec,features}/**/*')
+  s.executables   = 'pimon'
   s.require_paths = ['lib']
-  
-  s.add_runtime_dependency     'haml',              '~> 4.0.3'
-  s.add_runtime_dependency     'sinatra',           '~> 1.4.3'
-  s.add_runtime_dependency     'em-websocket',      '= 0.3.8'
-  s.add_runtime_dependency     'sinatra-websocket', '~> 0.2.1'
-  s.add_runtime_dependency     'sys-uptime',        '~> 0.6.1'
-  s.add_runtime_dependency     'thin',              '~> 1.5.1'
-  
-  s.add_development_dependency 'rack-test',       '~> 0.6.2'
-  s.add_development_dependency 'rspec',           '~> 2.14.1'
-  s.add_development_dependency 'simplecov',       '~> 0.7.1'
-  s.add_development_dependency 'simplecov-rcov',  '~> 0.2.3'
-  s.add_development_dependency 'timecop',         '~> 0.6.1'
+
+  s.add_runtime_dependency 'haml',           '= 4.0.7'
+  s.add_runtime_dependency 'sinatra',        '= 1.4.6'
+  s.add_runtime_dependency 'faye-websocket', '= 0.10.0'
+  s.add_runtime_dependency 'sys-uptime',     '= 0.6.2'
+  s.add_runtime_dependency 'thin',           '= 1.6.3'
+
+  s.add_development_dependency 'rack-test',      '= 0.6.2'
+  s.add_development_dependency 'rspec',          '= 3.0.0'
+  s.add_development_dependency 'rubocop',        '=0.33.0'
+  s.add_development_dependency 'simplecov',      '= 0.10.0'
+  s.add_development_dependency 'simplecov-rcov', '= 0.2.3'
+  s.add_development_dependency 'timecop',        '= 0.7.1'
+  s.add_development_dependency 'coveralls',      '= 0.8.2'
+  s.add_development_dependency 'pry',            '= 0.10.0'
 end