arbi 1.0.7 → 1.0.8

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

@@ -0,0 +1,50 @@
+#--
+# 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 Acpi
+
+  module Utils
+  protected
+
+    def hashize(file)
+      h = {}
+
+      File.readlines(file).each {|line|
+        if (line =~ /^([^:]+):\s*(.+?)\s*$/)
+          key, value = $1, $2
+          key = key.downcase.gsub(/ /, '_')
+          value = Numeric.parse(value) || value
+          h[key] = value
+          h[key.to_sym] = value
+        end
+      } if File.exist?(file)
+
+      h
+    end
+  end
+
+end
+
+end
+
+end

data/lib/arbi/modules/adapter.rb

@@ -0,0 +1,22 @@
+#--
+# 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/adapter'
+
+require 'arbi/modules/sys/adapter' unless Arbi::Modules::Acpi::Adapter.valid?

data/lib/arbi/modules/battery.rb

@@ -0,0 +1,22 @@
+#--
+# 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/battery'
+
+require 'arbi/modules/sys/battery' unless Arbi::Modules::Acpi::Battery.valid?

data/lib/arbi/modules/cpu.rb

@@ -0,0 +1,88 @@
+#--
+# 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
+
+class Cpu < Arbi::Modules::Module
+  def initialize(data = [])
+    super(data)
+
+    @mutex = Mutex.new
+    @prev = {:idle => {}, :total => {}}
+    @datas = []
+
+    every 2, :timeout => 10, &self.method(:update)
+  end
+
+  def valid?
+    File.exist?('/proc/stat')
+  end
+
+  def refresh
+    @mutex.synchronize {
+      @data = @datas.dup
+    }
+  end
+
+  def format
+    tablize([['NAME', 'PERCENT']] + @data.map {|x|
+      [x[:name] || x['name'], x[:percent] || x['percent']]
+    })
+  end
+
+protected
+  def update
+    percents = {}
+    File.readlines('/proc/stat').each {|line|
+      if line =~ /^cpu/
+        name, *datas = line.split(/\s+/)
+        name = (name == 'cpu' ? 'AVERAGE' : name)
+        total = datas.map(&:to_i).inject(:+)
+        idle = datas[3].to_i
+
+        @prev[:idle][name] = @prev[:total][name] = 0 if !@prev[:idle].include?(name) or !@prev[:total].include?(name)
+
+        percents[name] = "#{100 * ((total - @prev[:total][name]) - (idle - @prev[:idle][name])) / (total - @prev[:total][name])}%"
+        @prev[:idle][name] = idle
+        @prev[:total][name] = total
+      end
+    }
+
+    average = {'AVERAGE' => percents['AVERAGE']} and percents.delete('AVERAGE') and percents.merge!(average)
+    self.datas = percents.to_a
+  end
+
+  def datas=(percs)
+    @mutex.synchronize {
+      @datas = []
+      percs.each {|(name, perc)|
+        @datas << {
+          name:     name,
+          percent:  perc
+        }
+      }
+    }
+  end
+end
+
+end
+
+end

data/lib/arbi/modules/diskstat.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/>.
+#++
+
+require 'sys/filesystem'
+
+module Arbi
+
+module Modules
+
+module Sys
+  include ::Sys
+end
+
+class Diskstat < Module
+  def valid?
+    true
+  end
+
+  def refresh
+    @data = []
+    self.devices.each {|device|
+      begin
+        diskstat = Sys::Filesystem.stat(device[:point])
+        device.merge!({
+          usage: "#{100 - (100.0 / diskstat.blocks  * diskstat.blocks_available).round}%",
+          space:  self.unitize(diskstat.blocks * diskstat.fragment_size)
+        })
+
+        @data << device
+      rescue
+      end
+    }
+  end
+
+  def format
+    tablize([['DEVICE', 'POINT', 'USE', 'SPACE']] + @data.map {|dev|
+      [dev[:device] || dev['device'], dev[:point] || dev['point'], dev[:usage] || dev['usage'],
+        dev[:space] || dev['space']]
+    })
+  end
+
+protected
+  def devices
+    Sys::Filesystem.mounts.map {|fs|
+      {device: fs.name, point: fs.mount_point}
+    }
+  end
+
+  def unitize misure
+    u = 'b'
+    %w(Kb Mb Gb Tb).each {|i|
+      if misure >= 1024
+        misure /= 1024.0
+        u = i
+      else
+        return misure.round.to_s + u
+      end
+    }
+    misure.round.to_s + u
+  end
+end
+
+end
+
+end

data/lib/arbi/modules/help.rb

@@ -0,0 +1,40 @@
+#--
+# 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
+
+class Help < Module
+  def valid?
+    true
+  end
+
+  def refresh
+    @data = Arbi::Modules.modules.keys
+  end
+
+  def format
+    @data.join("\n")
+  end
+end
+
+end
+
+end

data/lib/arbi/modules/net.rb

@@ -0,0 +1,120 @@
+#--
+# 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'
+
+module Arbi
+
+module Modules
+
+class Net < Arbi::Modules::Module
+  def initialize(data = [])
+    super(data)
+
+    @mutex = Mutex.new
+    @prev = {:up => {}, :down => {}}
+    @datas = []
+
+    every 2, :timeout => 10, &self.method(:update)
+  end
+
+  def valid?
+    File.exist?('/proc/net/dev') and File.exist?('/proc/net/route') and
+      File.exist?('/proc/net/wireless')
+  end
+
+  def refresh
+    @mutex.synchronize {
+      @data = @datas.dup
+    }
+  end
+
+  def format
+    tablize([['IFACE', 'UP', 'DOWN', 'STATE', 'QUALITY', 'ESSID']] + @data.map {|x|
+      [x[:name] || x['name'], x[:up] || x['up'], x[:down] || x['down'],
+        ((x[:state] || x['state']) ? 'on' : 'off'), x[:quality] || x['quality'],
+        x[:essid] || x['essid']]
+    })
+  end
+
+  def update
+    stats = {}
+    File.read('/proc/net/dev').gsub(/^.*\|.*?\n/m, '').each_line {|line|
+      name, *datas = line.strip.split(/:?\s+/)
+      down = datas[0].to_i
+       up  = datas[8].to_i
+
+      @prev[:up][name] = @prev[:down][name] = 0 if !@prev[:up].include?(name) and !@prev[:down].include?(name)
+
+      stats[name] = {
+        up:       "%.1fkB" % [( up  - @prev[ :up ][name]).to_f / 1024],
+        down:     "%.1fkB" % [(down - @prev[:down][name]).to_f / 1024],
+        state:    (File.read('/proc/net/route') =~ /#{Regexp.escape(name)}/ ? true : false),
+        quality:  nil,
+        essid:    nil
+      }
+
+      if stats[name][:state]
+        stats[name][:quality] = quality(name)
+        begin
+          stats[name][ :essid ] = essid(name) if stats[name][:quality]
+        rescue Exception => e
+          Arbi.debug("ESSID: #{e}")
+        end
+      end
+
+      @prev[ :up ][name] = up
+      @prev[:down][name] = down
+    }
+
+    self.datas = stats
+  rescue Exception => e
+    Arbi.debug(e.to_s)
+  end
+
+protected
+  def quality(interface)
+    File.read('/proc/net/wireless').match(/^\s*#{Regexp.escape(interface)}:.*$/)[0].strip.split(/:?\s+/)[2].gsub('.', '') + "%"
+  rescue
+    nil
+  end
+
+  def essid(interface)
+    iwreq = [interface, " " * 32, 32, 0].pack("a16pII")
+    sock = Socket.new(Socket::AF_INET, Socket::SOCK_DGRAM, 0)
+    sock.ioctl(0x8B1B, iwreq)
+    return iwreq.unpack("a16pII")[1].strip
+  rescue
+    nil
+  end
+
+  def datas=(stats)
+    @mutex.synchronize {
+      @datas = []
+      stats.each {|(key, value)|
+        value[:name] = key
+        @datas << value
+      }
+    }
+  end
+end
+
+end
+
+end

data/lib/arbi/modules/ram.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 'arbi/modules/acpi/utils'
+
+module Arbi
+
+module Modules
+
+class Ram < Arbi::Modules::Module
+  include Arbi::Modules::Acpi::Utils
+
+  def valid?
+    File.exist?('/proc/meminfo')
+  end
+
+  def refresh
+    @data = []
+
+    mem = hashize('/proc/meminfo')
+    perc = 100.0 / mem[:memtotal] * (mem[:memfree] + mem[:buffers] + mem[:cached])
+
+    @data = [
+      ("%.1f%%" % [100.0 - perc]),
+      ("%.1f%%" % [perc])
+    ]
+  end
+
+  def format
+    tablize([['USED', 'FREE'], @data])
+  end
+end
+
+end
+
+end