epitools 0.5.4 → 0.5.5

data/VERSION

@@ -1 +1 @@
-0.5.4
+0.5.5

data/epitools.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = "epitools"
-  s.version = "0.5.4"
+  s.version = "0.5.5"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["epitron"]
-  s.date = "2012-04-28"
+  s.date = "2012-05-01"
   s.description = "Miscellaneous utility libraries to make my life easier."
   s.email = "chris@ill-logic.com"
   s.extra_rdoc_files = [
@@ -60,6 +60,8 @@ Gem::Specification.new do |s|
     "lib/epitools/sys.rb",
     "lib/epitools/term.rb",
     "lib/epitools/trie.rb",
+    "lib/epitools/typed_struct.rb",
+    "lib/epitools/wm.rb",
     "lib/epitools/zopen.rb",
     "spec/autoreq_spec.rb",
     "spec/browser_spec.rb",
@@ -78,6 +80,7 @@ Gem::Specification.new do |s|
     "spec/spec_helper.rb",
     "spec/sys_spec.rb",
     "spec/term_spec.rb",
+    "spec/typed_struct_spec.rb",
     "spec/zopen_spec.rb"
   ]
   s.homepage = "http://github.com/epitron/epitools"

data/lib/epitools/autoloads.rb

@@ -43,12 +43,14 @@ autoload :Ezdb,         'epitools/ezdb'
 autoload :Browser,      'epitools/browser'
 autoload :Rash,         'epitools/rash'
 autoload :Ratio,        'epitools/ratio'
-autoload :Sys,          'epitools/sys'
 autoload :ProgressBar,  'epitools/progressbar'
 autoload :Trie,         'epitools/trie'
 autoload :MimeMagic,    'epitools/mimemagic'
 autoload :Term,         'epitools/term'
 autoload :Iter,         'epitools/iter'
+autoload :WM,           'epitools/wm'
+autoload :TypedStruct,  'epitools/typed_struct'
+autoload :Sys,          'epitools/sys'
 
 ## Gems (common)
 autoreq  :Nokogiri,     'nokogiri'

data/lib/epitools/core_ext/enumerable.rb

@@ -193,6 +193,8 @@ module Enumerable
   def deep_select(depth=nil, &block)
     map do |*args|
 
+      obj = args.last
+
       if depth.nil? or depth > 0
 
         case obj

data/lib/epitools/core_ext/string.rb

@@ -257,6 +257,15 @@ class String
     end
   end
   
+  #
+  # Converts time duration strings (mm:ss, hh:mm:ss, or dd:hh:mm:ss) to seconds.
+  # (The reverse of Integer#to_hms)
+  #
+  def from_hms
+    nums = split(':').map(&:to_i)
+    nums_and_units = nums.reverse.zip %w[seconds minutes hours days]
+    nums_and_units.map { |num, units| num.send(units) }.sum
+  end
   
   unless public_method_defined? :to_proc
   

data/lib/epitools/typed_struct.rb

@@ -0,0 +1,67 @@
+#
+#
+#
+class TypedStruct < Struct
+
+  ## TODO: Compact syntax: "a,b,c:int x:string y:date"
+  ## TODO: Optional commas separating fields: "a, b, c:int, d:bool"
+  ## TODO: booleans fields add "field?" methods
+
+  #
+  # A perhaps-too-clever table of { "typename" => convert_proc } mappings.
+  #
+  CONVERTERS = Hash[ *{
+
+    ["str", "string"]  => :passthru,
+    ["int", "integer"] => proc { |me| me.to_i },
+    ["hex"]            => proc { |me| me.to_i(16) },
+    ["bool", "boolean"] => proc { |me| 
+      case me
+      when false, 0, "0", "off", "no",  "false", nil
+        false
+      when true,  1, "1", "on",  "yes", "true"
+        true
+      else
+        raise "Invalid boolean type: #{me.inspect}"
+      end
+    },
+    ["date", "time", "datetime"] => proc { |me| DateTime.parse me },
+    ["timestamp"] => proc { |me| Time.at me },
+
+  }.map { |names, converter| names.map { |n| [n, converter] } }.flatten ]
+
+  #
+  # Initialize a new struct.
+  #
+  def self.[](specs)
+    # create [name,type] pairs
+    pairs = specs.split.map do |spec|
+      name, type = spec.split(":")
+
+      type ||= "string"
+      unless converter = CONVERTERS[type]
+        raise "Unknown type: #{type}"
+      end
+
+      [name.to_sym, converter]
+    end
+
+    # initialize the C Struct
+    struct = new(*pairs.transpose.first)
+
+    # overload setter methods to call the proc
+    pairs.each do |field, converter| 
+      next if converter == :passthru
+      struct.send(:define_method, "#{field}=") do |val|
+        self[field] = ( val and converter.call val )
+      end
+    end
+
+    struct
+  end
+
+  def initialize(*args)    
+    members.zip(args).each { |field,value| send "#{field}=", value }
+  end
+
+end

data/lib/epitools/wm.rb

@@ -0,0 +1,83 @@
+module WM
+
+  raise "Error: wmctrl not found." unless Path.which("wmctrl")
+
+  def self.windows;           @windows    ||= Window.all; end
+  def self.desktops;          @desktops   ||= Desktop.all; end
+  def self.processes;         @processes  ||= Hash[ Sys.ps.map { |process| [process.pid, process] } ] ; end
+  def self.current_desktop;   @current    = desktops.find { |d| d.current? }; end
+
+  class Desktop < TypedStruct["num:int current:bool resolution viewport desktop_geometry name"]
+    def self.all
+      # 0  - DG: 1680x1050  VP: N/A  WA: 0,25 1680x974  Workspace 1
+      # 1  - DG: 1680x1050  VP: N/A  WA: 0,25 1680x974  Workspace 2
+      # 2  * DG: 1680x1050  VP: 0,0  WA: 0,25 1680x974  Workspace 3
+      # 3  - DG: 1680x1050  VP: N/A  WA: 0,25 1680x974  Workspace 4
+      # 4  - DG: 1680x1050  VP: N/A  WA: 0,25 1680x974  Workspace 5
+      # 5  - DG: 1680x1050  VP: N/A  WA: 0,25 1680x974  Workspace 6
+      # 0  1 2   3          4   5    6   7    8         9
+      `wmctrl -d`.lines.map(&:strip).map { |line| Desktop.from_line(line) }
+    end
+
+    def self.from_line(line)
+      fields = line.split
+      fields[1] = (fields[1] == "*")
+      fields[5] = nil if fields[5] == "N/A"
+      
+      name = fields[9..-1].join(" ")
+      
+      new *(fields.values_at(0,1,3,5,8) + [name])
+    end    
+
+    def current?
+      current
+    end
+
+    def windows
+      #binding.pry
+      @windows ||= WM.windows.select { |w| w.desktop_id == num }
+    end
+  end
+
+
+  class Window < TypedStruct["addr desktop_id:int pid:int x:int y:int w:int h:int hostname title"]
+
+    def self.all
+      `wmctrl -lpG`.lines.map(&:strip).map { |line| Window.from_line(line) }
+    end
+
+    def self.from_line(line)
+      # 0x01600031 -1 2562   0    0    1680 25   fizz Top Expanded Edge Panel
+      # 0x01600003 -1 2562   0    1998 1680 51   fizz Bottom Expanded Edge Panel
+      # 0x02c0001f  0 3012   849  173  783  667  fizz Terminal
+      # 0x048001f8  5 4080   311  186  1316 835  fizz Gorillaz - Highway (Under Construction)
+      # 0x02c28577  4 3012   66   461  1143 548  fizz Terminal
+      # 0x07c00003  0 14117  12   73   1298 948  fizz tr1984001_comp_soft.pdf
+      # 0x02d767d8  2 3012   520  470  1143 548  fizz Terminal    
+
+      fields = line.split
+      title  = fields[8..-1].join ' '
+
+      new *(fields[0..7] + [title])
+    end
+
+    def desktop
+      WM.desktops[desktop_id]
+    end
+
+    def sticky?
+      desktop == -1
+    end
+
+    alias_method :name, :title
+
+    def process
+      WM.processes[pid]
+    end
+
+    def inspect
+      "{ ::#{name}:: [#{desktop_id}]}"
+    end
+  end
+
+end

data/spec/core_ext_spec.rb

@@ -613,3 +613,9 @@ describe "to_jsons and to_yamls" do
   data.to_yaml.from_yaml.should == data  
 end
 
+describe "to_hms and from_hms" do
+  60.to_hms.should == "01:00"
+  60.to_hms.from_hms.should == 60
+  "1:20:33".from_hms.to_hms.should == "01:20:33"
+  "5:01:20:33".from_hms.to_hms.should == "05:01:20:33"
+end

data/spec/typed_struct_spec.rb

@@ -0,0 +1,16 @@
+require 'epitools'
+
+describe TypedStruct do
+  
+  it "works" do
+    t = TypedStruct["a:int b c:boolean d:timestamp"].new
+
+                 t.a.should == nil
+    t.a = "111"; t.a.should == 111
+    t.b = "111"; t.b.should == "111"
+    t.c = "yes"; t.c.should == true
+    #t.c?.should == true
+  end    
+  
+end
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: epitools
 version: !ruby/object:Gem::Version
-  version: 0.5.4
+  version: 0.5.5
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-04-28 00:00:00.000000000 Z
+date: 2012-05-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
@@ -110,6 +110,8 @@ files:
 - lib/epitools/sys.rb
 - lib/epitools/term.rb
 - lib/epitools/trie.rb
+- lib/epitools/typed_struct.rb
+- lib/epitools/wm.rb
 - lib/epitools/zopen.rb
 - spec/autoreq_spec.rb
 - spec/browser_spec.rb
@@ -128,6 +130,7 @@ files:
 - spec/spec_helper.rb
 - spec/sys_spec.rb
 - spec/term_spec.rb
+- spec/typed_struct_spec.rb
 - spec/zopen_spec.rb
 homepage: http://github.com/epitron/epitools
 licenses: