arbi 1.0.7 → 1.0.8

data/lib/arbi/client.rb

@@ -0,0 +1,52 @@
+#--
+# 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 'socket'
+require 'arbi/modules'
+
+module Arbi
+
+class Client
+  attr_reader :sock
+
+  def initialize(address = '127.0.0.1', port = 6969)
+    @address  = address || '127.0.0.1'
+    @port     = port || 6969
+
+    self.connect
+  end
+
+  def connect
+    @sock = TCPSocket.new(@address, @port)
+    @sock.gets
+  end
+
+  def get(what = 'help')
+    @sock.print "#{what.strip}\r\n"
+    @sock.flush
+    JSON.parse(@sock.gets.strip)
+  rescue Errno::EPIPE
+    self.connect
+    retry
+  rescue NoMethodError
+    nil
+  end
+end
+
+end

data/lib/arbi/config.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 'yaml'
+require 'singleton'
+
+module Arbi
+
+class Config < Hash
+  include Singleton
+
+  def initialize
+    super()
+    self.parse
+  end
+
+  def parse(file = '/etc/arbi.conf')
+    config = YAML.load_file(file)
+  rescue
+    Arbi.debug "Config: Error in parsing"
+  ensure
+    config ||= {}
+    config[:server] ||= {}
+    config[:server][:address] ||= '127.0.0.1'
+    config[:server][:port] ||= 6969
+    config[:client] ||= {}
+    config[:client][:address] ||= '127.0.0.1'
+    config[:client][:port] ||= 6969
+    config[:client][:default_cmd] ||= ['help', 'quit']
+    config[:modules] ||= {}
+    config[:modules][:path] ||= []
+    config[:modules][:modules] ||= []
+
+    config[:client][:default_cmd].map!(&:to_s).uniq!
+    config[:client][:default_cmd].delete('quit')
+    config[:client][:default_cmd] << 'quit'
+    config[:modules][:path] << File.join(File.dirname(__FILE__), 'modules')
+    config[:modules][:path].uniq!
+
+    self.replace(config)
+  end
+
+  class << self
+    def [](*args)
+      self.instance.[](*args)
+    end
+
+    def parse(*args)
+      self.instance.parse(*args)
+    end
+  end
+end
+
+end

data/lib/arbi/modules.rb

@@ -0,0 +1,183 @@
+#--
+# 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 'json'
+require 'singleton'
+require 'arbi/config'
+require 'arbi/timeline'
+require 'arbi/utils/table'
+require 'arbi/utils/numeric'
+
+module Arbi
+
+module Modules
+  PATH = [File.join(File.dirname(__FILE__), 'modules')]
+
+  @@logger  = STDERR
+
+  class Error
+    def initialize(data = "")
+      @data = data
+    end
+
+    def to_json(*a)
+      ({
+        'json_class'  => self.class.name,
+        'data'        => @data
+      }.to_json(*a)) + "\n"
+    end
+
+    def format
+      @data
+    end
+
+    def self.json_create(o)
+      self.new(o['data'])
+    end
+  end
+
+  class Module
+    attr_reader :data
+
+    def initialize(data = [])
+      @data = data
+    end
+
+    def every(ev = 2, args = {}, &blk)
+      args[:timeout] ||= 5
+      TimeLine::Job.new(ev, args[:timeout], &blk)
+    end
+
+    def to_json(*a)
+      self.refresh if self.respond_to?(:refresh)
+
+      {
+        'json_class'  => self.class.to_s,
+        'data'        => @data
+      }.to_json(*a) + "\n"
+    end
+
+    def format
+      "#{self.class.name} has no format specified, writing raw:\n#{data.to_json}"
+    end
+
+    def valid?
+      false
+    end
+
+    class << self
+      def json_create(o)
+        self.new(o['data'])
+      end
+
+      def inherited(obj)
+        @@modules ||= {}
+        Singleton.__init__(obj)
+        Arbi::Modules.uninit
+        @@modules[obj.name] = obj
+      end
+
+      def modules
+        @@modules
+      end
+
+      def name
+        self.to_s.downcase.split('::').last
+      end
+
+      alias __method_missing__ method_missing
+      def method_missing(sym, *args, &blk)
+        if self.instance.respond_to?(sym)
+          self.instance.send(sym, *args, &blk)
+        else
+          self.__method_missing__(sym, *args, &blk)
+        end
+      end
+
+      undef_method :to_json
+    end
+  end
+
+  class << self
+    @@init = false
+
+    def modules
+      Arbi::Modules::Module.modules.dup
+    end
+
+    def init
+      return if self.initialized?
+      Arbi::Modules::Module.modules.replace(
+        Hash[Arbi::Modules::Module.modules.to_a.select {|(key, value)|
+          value.valid?
+      }])
+      @@init = true
+    end
+
+    def uninit
+      @@init = false
+    end
+
+    def initialized?
+      @@init
+    end
+  end
+end
+
+def self.debug(str)
+  @@logger.puts(str.to_s)
+rescue
+  STDERR.puts(str.to_s)
+end
+
+def self.init
+  Arbi::Modules.init
+end
+
+def self.modules
+  Arbi.init
+  Arbi::Modules.modules
+end
+
+end
+
+def load_module(file)
+  lp = $:.dup
+  $:.replace(Arbi::Config[:modules][:path] + $:)
+  require file
+rescue Exception => e
+  Arbi.debug("Error loading module: #{e}")
+ensure
+  $:.replace(lp)
+end
+
+load_module('help')
+load_module('version')
+load_module('cpu')
+load_module('ram')
+load_module('net')
+load_module('adapter')
+load_module('battery')
+load_module('diskstat')
+
+Arbi::Config[:modules][:modules].each {|mod|
+  load_module(mod)
+}
+
+Arbi.init

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

@@ -0,0 +1,62 @@
+#--
+# 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/modules/acpi/utils'
+
+module Arbi
+
+module Modules
+
+module Acpi
+
+class Adapter < Arbi::Modules::Module
+  include Arbi::Modules::Acpi::Utils
+
+  def initialize(data = [])
+    super(data)
+    @adapters = Dir.glob('/proc/acpi/ac_adapter/*')
+  end
+
+  def valid?
+    !@adapters.empty?
+  end
+
+  def refresh
+    @data = []
+
+    @adapters.each {|adapter|
+      @data << {
+        name:   File.basename(adapter),
+        state:  (hashize("#{adapter}/state")[:state] == 'on-line' ? true : false)
+      }
+    }
+  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/acpi/battery.rb

@@ -0,0 +1,99 @@
+#--
+# 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/modules/acpi/utils'
+
+module Arbi
+
+module Modules
+
+module Acpi
+
+class Battery < Arbi::Modules::Module
+  include Arbi::Modules::Acpi::Utils
+
+  def initialize(data = [])
+    super(data)
+    @batteries = Dir.glob("/proc/acpi/battery/*")
+  end
+
+  def valid?
+    File.exist?('/proc/acpi/battery') and !@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 = self.hashize(dir + '/info').merge(self.hashize(dir + '/state'))
+    # Return if battery isn't present
+    return not_present(File.basename(dir)) unless {yes: true, no: false}[(raw[:present].to_sym rescue nil)]
+
+    {
+      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