arbi 1.0.7 → 1.0.8

data/lib/arbi/modules/sys/adapter.rb

@@ -0,0 +1,60 @@
+#--
+# Copyleft shura. [ shura1991@gmail.com ]
+#
+# This file is part of arbi.
+#
+# arbi is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# arbi is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with arbi. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+module Arbi
+
+module Modules
+
+module Sys
+
+class Adapter < Arbi::Modules::Module
+  def initialize(data = [])
+    super(data)
+    @adapters = Dir.glob('/sys/class/power_supply/*').select {|ad|
+      File.read("#{ad}/type").strip == 'Mains'
+    }
+  end
+
+  def valid?
+    !@adapters.empty?
+  end
+
+  def refresh
+    @data = []
+
+    @adapters.each {|adapter|
+      @data << {
+        name:   File.basename(adapter),
+        state:  (File.read("#{adapter}/online").to_i.zero? ? false : true)
+      }
+    }
+  end
+
+  def format
+    tablize([['NAME', 'STATE']] + @data.map {|adapter|
+      [adapter[:name] || adapter['name'], (adapter[:state] || adapter['state']) ? 'on' : 'off']
+    })
+  end
+end
+
+end
+
+end
+
+end

data/lib/arbi/modules/sys/battery.rb

@@ -0,0 +1,101 @@
+#--
+# Copyleft shura. [ shura1991@gmail.com ]
+#
+# This file is part of arbi.
+#
+# arbi is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# arbi is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with arbi. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+module Arbi
+
+module Modules
+
+module Sys
+
+class Battery < Arbi::Modules::Module
+  def initialize(data = [])
+    super(data)
+    @batteries = Dir.glob('/sys/class/power_supply/*').select {|dir| File.read("#{dir}/type").strip == 'Battery' }
+  end
+
+  def valid?
+    !@batteries.empty?
+  end
+
+  def refresh
+    @data = []
+
+    @batteries.each {|battery|
+      @data << self.battery_info(battery)
+    }
+    @data << self.average(@data)
+  end
+
+  def format
+    tablize([['NAME', 'PERCENT', 'STATE', 'SANITY']] + @data.map {|x|
+      [x[:name] || x['name'], x[:percent] || x['percent'], x[:state] || x['state'], x[:sanity] || x['sanity']]
+    })
+  end
+
+protected
+  def battery_info(dir)
+    raw = {
+      design_capacity:    File.read("#{dir}/energy_full_design").strip.to_i,
+      last_full_capacity: File.read("#{dir}/energy_full").strip.to_i,
+      remaining_capacity: File.read("#{dir}/energy_now").strip.to_i,
+      present:            (File.read("#{dir}/present").strip.to_i.zero? ? false : true),
+      charging_state:     File.read("#{dir}/status").strip.downcase.gsub(/^full$/, 'charged')
+    }
+
+    return not_present(File.basename(dir)) unless raw[:present]
+
+    {
+      name:     File.basename(dir),
+      sanity:   ("%.1f%%" % [100.0 / raw[:design_capacity] * raw[:last_full_capacity]]),
+      state:    raw[:charging_state],
+      percent:  ("%.1f%%" % [100.0 / raw[:last_full_capacity] * raw[:remaining_capacity]])
+    }
+  end
+
+  def average(batteries)
+    avg, n = 0, 0
+
+    batteries.each {|battery|
+      avg += battery[:percent].gsub(/%$/, '').to_f and n += 1 if battery[:percent]
+    }
+    avg = "%.1f%%" % [avg / n] rescue false
+
+    {
+      name:     'AVERAGE',
+      sanity:   false,
+      state:    false,
+      percent:  avg
+    }
+  end
+
+  def not_present(name)
+    {
+      name:     name,
+      sanity:   false,
+      state:    false,
+      percent:  false
+    }
+  end
+end
+
+end
+
+end
+
+end

data/lib/arbi/modules/sys/thermal.rb

@@ -0,0 +1,71 @@
+#--
+# Copyleft shura. [ shura1991@gmail.com ]
+#
+# This file is part of arbi.
+#
+# arbi is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# arbi is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with arbi. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+module Arbi
+
+module Modules
+
+module Sys
+
+class Thermal < Arbi::Modules::Module
+  def initialize(data = [])
+    super(data)
+    @trips = {}
+    @tempdirs = Dir.glob('/sys/class/thermal/thermal_zone*')
+
+    @tempdirs.each {|dir|
+      name = File.basename(dir)
+
+      Dir.glob("#{dir}/trip_point_*").select {|x| x =~ /_temp$/ }.each {|trip|
+        if File.read(trip.gsub(/emp$/, 'ype')).strip == 'critical'
+          @trips[name] = "#{File.read(trip).to_i / 1000} C"
+        end
+      }
+    }
+  end
+
+  def valid?
+    !@tempdirs.empty?
+  end
+
+  def refresh
+    @data = []
+
+    @tempdirs.each {|dir|
+      name = File.basename(dir)
+      @data << {
+        name:         name,
+        temperature:  "#{File.read("#{dir}/temp").to_i / 1000} C",
+        trip_point:   @trips[name]
+      }
+    }
+  end
+
+  def format
+    tablize([['NAME', 'TEMPERATURE', 'TRIP POINT']] + @data.map {|therm|
+      [therm[:name] || therm['name'], therm[:temperature] || therm['temperature'], therm[:trip_point] || therm['trip_point']]
+    })
+  end
+end
+
+end
+
+end
+
+end

data/lib/arbi/modules/version.rb

@@ -0,0 +1,42 @@
+#--
+# Copyleft shura. [ shura1991@gmail.com ]
+#
+# This file is part of arbi.
+#
+# arbi is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# arbi is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with arbi. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+require 'arbi/version'
+
+module Arbi
+
+module Modules
+
+class Version < Module
+  def valid?
+    true
+  end
+
+  def refresh
+    @data = Arbi::VERSION
+  end
+
+  def format
+    @data
+  end
+end
+
+end
+
+end

data/lib/arbi/server.rb

@@ -0,0 +1,67 @@
+#--
+# Copyleft shura. [ shura1991@gmail.com ]
+#
+# This file is part of arbi.
+#
+# arbi is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# arbi is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with arbi. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+require 'eventmachine'
+require 'arbi/modules'
+
+module Arbi
+
+class Server
+  module Server
+    def post_init
+      send_data("OH HAI! ^_^\n")
+    end
+
+    def receive_data(data)
+      if data.strip.downcase =~ /^(quit|exit)$/
+        close_connection
+        return
+      end
+
+      obj = Arbi.modules[data.strip.downcase]
+
+      if obj
+        send_data(obj.to_json)
+      else
+        send_data(Arbi::Modules::Error.new('Command doesn\'t exist').to_json)
+      end
+    rescue Exception => e
+      Arbi.debug(e.to_s)
+    end
+  end
+
+  def initialize(address = '127.0.0.1', port = 6969)
+    @address  = address || '127.0.0.1'
+    @port     = port    ||  6969
+  end
+
+  def start
+    EventMachine.run {
+      TimeLine.run
+      EventMachine.start_server(@address, @port, Arbi::Server::Server)
+      Arbi.debug("Starting server on #{@address}:#{@port}")
+    }
+  end
+
+  def self.start(address = '127.0.0.1', port = 6969)
+    self.new(address, port).start
+  end
+end
+
+end

data/lib/arbi/timeline.rb

@@ -0,0 +1,70 @@
+#--
+# Copyleft shura. [ shura1991@gmail.com ]
+#
+# This file is part of arbi.
+#
+# arbi is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# arbi is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with arbi. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+require 'eventmachine'
+require 'timeout'
+
+module TimeLine
+  class Job
+    attr_reader :every, :timeout, :proc
+
+    def initialize(ev = 2, to = 5, &blk)
+      self.every = ev
+      self.timeout = to
+      self.proc = blk
+
+      TimeLine.register(self)
+    end
+
+    def every=(ev)
+      raise ArgumentError, "every must be an Integer > 0" unless ev.is_a?(Integer) and ev > 0
+      @every = ev
+    end
+
+    def timeout=(to)
+      raise ArgumentError, "timeout must be an Integer > 0" unless to.is_a?(Integer) and to > 0
+      @timeout = to
+    end
+
+    def proc=(blk)
+      raise ArgumentError, "proc must be a Proc" unless blk.is_a?(Proc)
+      @proc = blk
+    end
+  end
+
+  class << self
+    def register(job)
+      raise ArgumentError, "job must be a TimeLine::Job" unless job.is_a?(TimeLine::Job)
+      @@jobs ||= []
+      @@jobs << job
+    end
+
+    def run
+      @@jobs ||= []
+      @@jobs.each {|job|
+        EventMachine::PeriodicTimer.new(job.every) do
+          timeout(job.timeout) {
+            job.proc.call
+          }
+        end
+      }
+      self
+    end
+  end
+end

data/lib/arbi/utils/numeric.rb

@@ -0,0 +1,81 @@
+#--
+# Copyleft shura. [ shura1991@gmail.com ]
+#
+# This file is part of arbi.
+#
+# arbi is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# arbi is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with arbi. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+class Numeric
+
+module Scalar
+  attr_accessor :unit
+
+  MULTIPLIERS = {
+    ?Y => 10 ** 24, ?Z => 10 ** 21, ?E => 10 ** 18,
+    ?P => 10 ** 15, ?T => 10 ** 12, ?G => 10 ** 9,
+    ?M => 10 ** 6, ?k => 10 ** 3, ?h => 10 ** 2, 'da' => 10,
+
+    ?y => 10 ** -24, ?z => 10 ** -21, ?a => 10 ** -18,
+    ?f => 10 ** -15, ?p => 10 ** -12, ?n => 10 ** -9,
+    ?u => 10 ** -9, ?m => 10 ** -3, ?c => 10 ** -2, ?d => 10 ** -1
+  }
+
+  alias __method_missing__ method_missing
+  def method_missing(sym, *args, &blk)
+    if MULTIPLIERS.include?(sym.to_s)
+      self / MULTIPLIERS[sym.to_s]
+    else
+      self.__method_missing__(sym, *args, &blk) if sym != :to_str
+    end
+  end
+end
+
+  def self.parse(str)
+    if str.strip.match(/^([^\s]+)\s+(#{Numeric::Scalar::MULTIPLIERS.keys.map {|x|
+          Regexp.escape(x)
+          }.join(?|)})?(.*?)$/)
+      num, mul, unit = $1, $2, $3
+      v = begin
+        Integer(num)
+      rescue ArgumentError
+        Float(num) rescue nil
+      end
+
+      return nil unless v
+
+      if Numeric::Scalar::MULTIPLIERS[mul]
+        v = (v * Numeric::Scalar::MULTIPLIERS[mul]).to_f
+      end
+
+      if !unit.empty?
+        v.unit = unit.dup
+      end
+
+      return v
+    end
+  end
+end
+
+class Integer
+  include Numeric::Scalar
+end
+
+class Float
+  include Numeric::Scalar
+end
+
+class Rational
+  include Numeric::Scalar
+end