dag-amazing 0.1

data/lib/amazing/widgets/clock.rb

@@ -0,0 +1,31 @@
+# Copyright 2008 Dag Odenhall <dag.odenhall@gmail.com>
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'amazing/widget'
+
+module Amazing
+  module Widgets
+    class Clock < Widget
+      description "Displays date and time"
+      option :format, "Time format as described in DATE(1)", "%R"
+      option :offset, "UTC offset in hours", Time.now.utc_offset / 3600
+      field :time, "Formatted time"
+      default { @time }
+
+      init do
+        @time = (Time.now.utc + @offset.to_f * 3600).strftime(@format)
+      end
+    end
+  end
+end

data/lib/amazing/widgets/cpu_info.rb

@@ -0,0 +1,35 @@
+# Copyright 2008 Dag Odenhall <dag.odenhall@gmail.com>
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'amazing/widget'
+require 'amazing/proc_file'
+
+module Amazing
+  module Widgets
+
+    # TODO: There might be more yummy stuff in /proc/cpuinfo
+    class CpuInfo < Widget
+      description "CPU Information"
+      option :cpu, "CPU number for default format (0 based)", 0
+      field :speed, "CPU Speed in MHz", []
+      default { @speed[@cpu].to_i  }
+
+      init do
+        ProcFile.parse_file("cpuinfo").each_with_index do |info, cpu|
+          @speed[cpu] = info["cpu MHz"].to_f
+        end
+      end
+    end
+  end
+end

data/lib/amazing/widgets/cpu_usage.rb

@@ -0,0 +1,50 @@
+# Copyright 2008 Dag Odenhall <dag.odenhall@gmail.com>
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'amazing/widget'
+
+module Amazing
+  module Widgets
+    class CpuUsage < Widget
+      description "CPU usage"
+      option :cpu, "CPU number, 0 is all", 0
+      field :usage, "Percent of CPU in use", []
+      default { @usage[@cpu].to_i }
+
+      init do
+        first = gather_data()
+        sleep 1
+        second = gather_data()
+        first.each_index do |cpunum|
+          idle = second[cpunum][3].to_i - first[cpunum][3].to_i
+          sum = second[cpunum][0..5].inject {|a,b| a.to_i + b.to_i } -
+            first[cpunum][0..5].inject {|a,b| a.to_i + b.to_i }
+          @usage[cpunum] = 100 - (idle * 100) / sum.to_f
+        end
+      end
+
+      private
+
+      def gather_data
+        cpus = []
+        ::File.readlines("/proc/stat").select {|l| l =~ /^cpu/ }.each do |l|
+          l = l.split
+          cpunum = l[0] == "cpu" ? 0 : l[0].scan(/\d+/)[0].to_i + 1
+          cpus[cpunum] = l[1..-1]
+        end
+        cpus
+      end
+    end
+  end
+end

data/lib/amazing/widgets/file.rb

@@ -0,0 +1,44 @@
+# Copyright 2008 Dag Odenhall <dag.odenhall@gmail.com>
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'amazing/widget'
+
+module Amazing
+  module Widgets
+    class File < Widget
+      description "Information for a file"
+      option :file, "The file to work with", ""
+      option :time_format, "Time format for timestamps as described in DATE(1)", "%c"
+      field :lines, "All the lines in the file", []
+      field :first, "The first line in the file", ""
+      field :last, "The last line in the file", ""
+      field :count, "Number of lines in the file", 0
+      field :atime, "Access time", Time.now.strftime("%c")
+      field :ctime, "Change time", Time.now.strftime("%c")
+      field :mtime, "Modification time", Time.now.strftime("%c")
+      default { @last }
+
+      init do
+        @file = ::File.expand_path(@file, "~")
+        @lines = ::File.readlines(@file).map {|line| line.chomp }
+        @first = @lines.first || ""
+        @last = @lines.last || ""
+        @count = @lines.size
+        @atime = ::File.atime(@file).strftime(@time_format)
+        @ctime = ::File.ctime(@file).strftime(@time_format)
+        @mtime = ::File.mtime(@file).strftime(@time_format)
+      end
+    end
+  end
+end

data/lib/amazing/widgets/file_system.rb

@@ -0,0 +1,42 @@
+# Copyright 2008 Dag Odenhall <dag.odenhall@gmail.com>
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'amazing/widget'
+
+module Amazing
+  module Widgets
+    class FileSystem < Widget
+      description "Various filesystem information"
+      option :mountpoint, "Mountpoint for default format", "/"
+      field :size, "Total size of volume in MB", {}
+      field :used, "Size of used data in MB", {}
+      field :free, "Size of free data in MB", {}
+      field :percentage, "Percentage of used space", {}
+      field :device, "Device name of mount point", {}
+      default { @percentage[@mountpoint] }
+
+      init do
+        IO.popen("df") do |io|
+          io.readlines[1..-1].map {|l| l.split }.each do |fs|
+            @device[fs[5]] = fs[0].to_s
+            @size[fs[5]] = fs[1].to_f / 1024
+            @used[fs[5]] = fs[2].to_f / 1024
+            @free[fs[5]] = fs[3].to_f / 1024
+            @percentage[fs[5]] = fs[4].to_i
+          end
+        end
+      end
+    end
+  end
+end

data/lib/amazing/widgets/gmail.rb

@@ -0,0 +1,61 @@
+# Copyright 2008 Dag Odenhall <dag.odenhall@gmail.com>
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'amazing/widget'
+
+module Amazing
+  module Widgets
+    class Gmail < Widget
+      description "GMail checker"
+      dependency "net/https", "Ruby standard library"
+      dependency "rexml/document", "Ruby standard library"
+      dependency "time", "Ruby standard library"
+      option :username, "Username"
+      option :password, "Password"
+      option :certificates, "Path to SSL certificates", "/etc/ssl/certs"
+      option :verify, "Verify certificates", false
+      field :messages, "List of new messages"
+      field :count, "Number of new messages"
+      default { @count }
+
+      init do
+        http = Net::HTTP.new("mail.google.com", 443)
+        http.use_ssl = true
+        if @verify
+          http.verify_mode = OpenSSL::SSL::VERIFY_PEER
+          http.ca_path = @certificates
+        else
+          http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+        end
+        request = Net::HTTP::Get.new("/mail/feed/atom")
+        request.basic_auth(@username, @password)
+        doc = REXML::Document.new(http.request(request).body)
+        begin
+          @messages = doc.elements.to_a("//entry").map do |e|
+            {:subject => e.elements["title"].text,
+             :summary => e.elements["summary"].text,
+             :from => e.elements.to_a("author").map do |a|
+               {:name => a.elements["name"].text,
+                :email => a.elements["email"].text}
+             end[0],
+             :date => Time.xmlschema(e.elements["issued"].text),
+             :link => e.elements["link"].attributes["href"]}
+          end
+        rescue
+        end
+        @count = doc.root.elements["fullcount"].text.to_i
+      end
+    end
+  end
+end

data/lib/amazing/widgets/maildir.rb

@@ -0,0 +1,34 @@
+# Copyright 2008 Dag Odenhall <dag.odenhall@gmail.com>
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'amazing/widget'
+
+module Amazing
+  module Widgets
+    class Maildir < Widget
+      description "Mail count in maildirs"
+      option :directories, "Globs of maildirs" # TODO: does a default make sense?
+      field :count, "Ammount of mail in searched directories", 0
+      default { @count }
+
+      init do
+        raise WidgetError, "No directories configured" unless @directories
+        @directories.to_a.each do |glob|
+          glob = ::File.expand_path(glob, "~")
+          @count += Dir["#{glob}/*"].size
+        end
+      end
+    end
+  end
+end

data/lib/amazing/widgets/memory.rb

@@ -0,0 +1,47 @@
+# Copyright 2008 Dag Odenhall <dag.odenhall@gmail.com>
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'amazing/widget'
+require 'amazing/proc_file'
+
+module Amazing
+  module Widgets
+    class Memory < Widget
+      description "Various memory related data"
+      field :total, "Total kilobytes of memory"
+      field :free, "Free kilobytes of memory"
+      field :buffers, "Buffered kilobytes of memory"
+      field :cached, "Cached kilobytes of memory"
+      field :usage, "Percentage of used memory"
+      field :swap_total, "Total kilobytes of swap"
+      field :swap_free, "Free kilobytes of swap"
+      field :swap_cached, "Cached kilobytes of swap"
+      field :swap_usage, "Percentage of used swap"
+      default { @usage.to_i }
+
+      init do
+        meminfo = ProcFile.parse_file("meminfo")[0]
+        @total = meminfo["MemTotal"].to_i
+        @free = meminfo["MemFree"].to_i
+        @buffers = meminfo["Buffers"].to_i
+        @cached = meminfo["Cached"].to_i
+        @usage = ((@total - @free - @cached - @buffers) * 100) / @total.to_f
+        @swap_total = meminfo["SwapTotal"].to_i
+        @swap_free = meminfo["SwapFree"].to_i
+        @swap_cached = meminfo["SwapCached"].to_i
+        @swap_usage = ((@swap_total - @swap_free - @swap_cached) * 100) / @swap_total.to_f
+      end
+    end
+  end
+end

data/lib/amazing/widgets/moc.rb

@@ -0,0 +1,80 @@
+# Copyright 2008 Dag Odenhall <dag.odenhall@gmail.com>
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'amazing/widget'
+require 'amazing/proc_file'
+
+module Amazing
+  module Widgets
+    class Moc < Widget
+      description "Music On Console status"
+      option :mocp, "Path to mocp program", "mocp"
+      field :state, "Play state, :playing, :paused or :stopped"
+      field :file, "Playing file"
+      field :title, "Title as seen by MOC"
+      field :artist, "Artist name"
+      field :song_title, "Song title"
+      field :album, "Album of song"
+      field :total_time, "Total length of song"
+      field :time_left, "Time left of playing song"
+      field :total_sec, "Total length of song in seconds"
+      field :current_time, "Current position in playing song"
+      field :current_sec, "Current position in playing song in seconds"
+      field :bitrate, "Song bitrate"
+      field :rate, "Song sample rate"
+
+      default do
+        case @state
+        when :playing
+          "#@artist - #@song_title"
+        when :paused
+          "#@artist - #@song_title [paused]"
+        when :stopped
+          "[moc not playing]"
+        end
+      end
+
+      init do
+        moc = ProcFile.new(IO.popen("#@mocp --info"))[0]
+        @state = {:play => :playing, :pause => :paused, :stop => :stopped}[moc["State"].downcase.to_sym]
+        @file = moc["File"]
+        @title = moc["Title"]
+        @artist = moc["Artist"]
+        @song_title = moc["SongTitle"]
+        @album = moc["Album"]
+        @total_time = moc["TotalTime"]
+        @time_left = moc["TimeLeft"]
+        @total_sec = moc["TotalSec"].to_i
+        @current_time = moc["CurrentTime"]
+        @current_sec = moc["CurrentSec"].to_i
+        @bitrate = moc["Bitrate"]
+        @rate = moc["Rate"]
+      end
+
+      private
+
+      def playing?
+        @state == :playing
+      end
+
+      def paused?
+        @state == :paused
+      end
+
+      def stopped?
+        @state == :stopped
+      end
+    end
+  end
+end

data/lib/amazing/widgets/mpd.rb

@@ -0,0 +1,146 @@
+# Copyright 2008 Dag Odenhall <dag.odenhall@gmail.com>
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'amazing/widget'
+require 'amazing/proc_file'
+
+module Amazing
+  module Widgets
+    class Mpd < Widget
+      description "MPD Information"
+      dependency "socket", "Ruby standard library"
+      option :hostname, "MPD server hostname", "localhost"
+      option :port, "MPD server port number", 6600
+      option :password, "Authentication password"
+      field :state, "Play state, :playing, :paused or :stopped"
+      field :artist, "Song artist"
+      field :title, "Song title"
+      field :length, "Song length"
+      field :position, "Song position"
+      field :date, "Song date from ID3 tag"
+      field :id, "Song ID in playlist"
+      field :genre, "Song genre"
+      field :album, "Song album"
+      field :file, "Filename of current song"
+      field :shortfile, "Filename without directory path and extension"
+      field :percentage, "Finished percentage of current song", 0.0
+      field :random, "True if playlist is on random"
+      field :repeat, "True if playlist is on repeat"
+      field :volume, "Volume of MPD mixer", 0
+
+      default do
+        track = "#{@artist && "#@artist - "}#@title"
+        track = @shortfile if track.empty?
+        case @state
+        when :playing
+          track
+        when :paused
+          "#{track} [paused]"
+        when :stopped
+          "[mpd not playing]"
+        end
+      end
+
+      init do
+        mpd = get_socket
+        status = send_command(:status)
+        @state = {:play => :playing, :pause => :paused, :stop => :stopped}[status["state"].to_sym]
+        @random = {0 => false, 1 => true}[status["random"].to_i]
+        @repeat = {0 => false, 1 => true}[status["repeat"].to_i]
+        @volume = status["volume"].to_i
+        if song = send_command(:currentsong)
+          @artist = song["Artist"]
+          @title = song["Title"]
+          len = song["Time"].to_i
+          minutes = (len / 60) % 60
+          seconds = len % 60
+          @length = "%d:%02d" % [minutes, seconds]
+          pos = status["time"].to_i
+          minutes = (pos / 60) % 60
+          seconds = pos % 60
+          @position = "%d:%02d" % [minutes, seconds]
+          @date = song["Date"]
+          @id = song["Id"].to_i
+          @genre = song["Genre"]
+          @album = song["Album"]
+          @file = song["file"]
+          @shortfile = ::File.basename(@file)
+          @shortfile = @shortfile[0..-::File.extname(@shortfile).length-1]
+          @percentage = pos == 0 ? 0.0 : pos / len.to_f * 100
+        end
+      end
+
+      private
+
+      def playing?
+        @state == :playing
+      end
+
+      def paused?
+        @state == :paused
+      end
+
+      def stopped?
+        @state == :stopped
+      end
+
+      def get_socket
+        @@connections ||= {}
+        mpd = nil
+        unless @@connections["#@identifier"]
+          @@connections["#@identifier"] = TCPSocket.new(@hostname, @port)
+          mpd = @@connections["#@identifier"]
+          mpd.gets
+          authenticate
+        else
+          mpd = @@connections["#@identifier"]
+          mpd.puts("ping")
+          unless mpd.gets
+            @@connections["#@identifier"] = TCPSocket.new(@hostname, @port)
+            mpd = @@connections["#@identifier"]
+            mpd.gets
+            authenticate
+          end
+        end
+        mpd
+      end
+
+      def authenticate
+        mpd = @@connections["#@identifier"]
+        if @password
+          mpd.puts("password #@password")
+          if mpd.gets.split[0] == "ACK"
+            raise WidgetError, "incorrect password"
+          end
+        end
+      end
+
+      def send_command(command)
+        mpd = @@connections["#@identifier"]
+        mpd.puts(command.to_s)
+        lines = []
+        loop do
+          line = mpd.gets
+          if line.split[0] == "ACK"
+            raise WidgetError, line.scan(/\{#{command}\} (.+)/).to_s
+          elsif line.split[0] == "OK"
+            return ProcFile.new(lines.join)[0]
+          else
+            lines << line
+          end
+        end
+      end
+    end
+  end
+end