rum 0.0.1-universal-darwin-10

data/lib/rum/windows/system.rb

@@ -0,0 +1,310 @@
+require_relative 'system_foreign_functions'
+
+module Rum
+  module System
+    extend self
+    
+    def send_key_event key, down
+      extended, id = key.id.divmod 2**9
+      extended = (extended == 1)
+      keybd_event id, down, 0, extended
+    end
+
+    def send_keypress key
+      send_key_event key, true
+      send_key_event key, false
+    end
+
+    def active_window_handles
+      Enumerator.new(self, :enum_windows)
+    end
+
+    def active_windows
+      Enumerator.new do |yielder|
+        enum_windows { |handle| yielder.yield Window.new(handle) }
+      end
+    end
+
+    def active_window
+      Window.new get_foreground_window
+    end
+    
+    def terminal_window
+      Window.new get_console_window
+    end
+
+    def spawn_in_terminal(*args)
+      if args.delete :close_if_successful
+        close_window = '/c'
+        # if there's an error, ask for input to keep the shell window alive
+        args.concat %w(|| set /p name= Abort)
+      else
+        close_window = '/k'
+      end
+      if args.delete(:wait)
+        Rum.switch_worker_thread
+        wait = '/wait'
+      end
+      system('start', *wait, 'cmd', close_window, *args)
+    end
+
+    # returns a copy
+    def c_string obj
+      str = obj.to_s + "\0"
+      str.encode(Encoding::UTF_16LE) 
+    end
+
+    def message_box message, title=''
+      message_box_internal(c_string(message), c_string(title))
+    end
+
+    def input_box text, title, default_text
+      buffer_size = 2048
+      buffer = default_text.to_s.encode Encoding::UTF_16LE
+      buffer.force_encoding Encoding::BINARY
+      if buffer.length >= buffer_size
+        buffer << "\0\0"
+      else
+        buffer << "\0" * (buffer_size - buffer.length)
+      end
+
+      length = input_box_internal(c_string(text), c_string(title), buffer)
+      if length == 0
+        ''
+      else
+        result = buffer[0, length*2]
+        result.force_encoding(Encoding::UTF_16LE).encode(Encoding::UTF_8)
+      end
+    end
+
+    def get_selection
+      Clipboard.preserve { Clipboard.get_selection }
+    end
+
+    class Window
+      include System
+      
+      attr_reader :handle
+      
+      def initialize(handle)
+        @handle = handle
+      end
+
+      def == other
+        return false unless other.respond_to? :handle
+        @handle == other.handle
+      end
+      
+      def active?
+        self.handle == get_foreground_window
+      end
+
+      def close
+        post_message @handle, WM_SYSCOMMAND, SC_CLOSE, 0
+      end
+
+      def hide
+        show_window(@handle, SW_HIDE)
+      end
+
+      def zoomed?
+        is_zoomed(@handle) == 1 ? true : false
+      end
+
+      def toggle_maximized
+        zoomed? ? restore : maximize
+      end
+
+      def maximize
+        post_message @handle, WM_SYSCOMMAND, SC_MAXIMIZE, 0
+      end
+      
+      def minimize
+        post_message @handle, WM_SYSCOMMAND, SC_MINIMIZE, 0
+      end
+
+      def restore
+        post_message @handle, WM_SYSCOMMAND, SC_RESTORE, 0
+      end
+
+      def file_dialog?
+        class_name == '#32770'
+      end
+
+      def office_file_dialog?
+        class_name =~ /^bosa_sdm/
+      end
+
+      def toggle_always_on_top
+        window_style = get_window_long(@handle, GWL_EXSTYLE)
+        return if window_style == 0
+        topmost_or_not = if (window_style & WS_EX_TOPMOST == WS_EX_TOPMOST)
+                           HWND_NOTOPMOST
+                         else
+                           HWND_TOPMOST
+                         end
+        set_window_pos(@handle, topmost_or_not, 0, 0, 0, 0, SWP_NOMOVE | SWP_NOSIZE)
+      end
+
+      def move(x, y, width, height)
+        restore if zoomed?
+        move_window(@handle, x, y, width, height, 1)
+      end
+      
+      def title(max_length = 1024)
+        buffer = "\0" * (max_length * 2)
+        length = get_window_text_w(@handle, buffer, buffer.length)
+        result = (length == 0 ? '' : buffer[0..(length * 2 - 1)])
+        result.force_encoding(Encoding::UTF_16LE)
+        result.encode(Encoding::UTF_8, invalid: :replace)
+      end
+      
+      def class_name(max_length = 2048)
+        buffer = "\0" * max_length
+        length = get_class_name(@handle, buffer, buffer.length)
+        length == 0 ? '' : buffer[0..length - 1]
+      end
+      
+      def text(max_length = 2048)
+        buffer = '\0' * max_length
+        length = send_with_buffer @handle, WM_GETTEXT, buffer.length, buffer
+        length == 0 ? '' : buffer[0..length - 1]
+      end
+
+      def child(id)
+        result = case id
+                 when String
+                   by_title = find_window_ex @handle, 0, nil, id.gsub('_', '&')
+                   by_class = find_window_ex @handle, 0, id, nil
+                   by_title > 0 ? by_title : by_class
+                 when Fixnum
+                   get_dlg_item @handle, id
+                 else
+                   0
+                 end
+        
+        raise "Control '#{id}' not found" if result == 0
+        Window.new result
+      end
+
+      def kill_task
+        system("taskkill /F /IM #{exe_name}.exe")
+      end
+      
+      def report
+        "Title: #{title}\nClass: #{class_name}"
+      end
+
+      def self.[] arg, class_name=nil
+        if arg.is_a? Hash
+          WindowMatcher.new(arg[:title], arg[:class_name])
+        else
+          WindowMatcher.new(arg, class_name)
+        end
+      end
+    end
+
+    class WindowMatcher
+      include System
+      attr_reader :specs
+      
+      def initialize title=nil, class_name=nil
+        @specs = { title: title, class_name: class_name }
+        raise 'No specifications given.' unless @specs.values.any?
+        @specs.each do |spec, value|
+          if value.is_a? Regexp
+            @specs[spec] = Regexp.new(value.source, Regexp::IGNORECASE)
+          end
+        end
+      end
+
+      def comparison_method obj
+        if obj.is_a? Regexp then :=~ else :== end
+      end
+      
+      def matches? window
+        @specs.all? do |spec, value|
+          not value or value.send(comparison_method(value), window.send(spec))
+        end
+      end
+
+      def active?
+        matches? active_window
+      end
+
+      def to_matcher
+        self
+      end
+
+      def [] spec
+        @specs[spec]
+      end
+
+      def find
+        if @specs.values.any? { |spec| spec.is_a? Regexp }
+          active_windows.detect { |window| matches? window }
+        else
+          handle = find_window @specs[:class_name], @specs[:title]
+          Window.new(handle) unless handle == 0
+        end
+      end
+    end
+    
+    module Clipboard
+      extend self
+      
+      def get
+        Win32::Clipboard.get
+      end
+    
+      def set str
+        Win32::Clipboard.set str
+      end
+
+      def preserve
+        old_content = get
+        yield
+      ensure
+        set old_content
+      end
+
+      # def copy
+      #   watcher = install_watcher
+      #   Keyboard.type '(ctrl c)'
+      #   evaluate_watcher(watcher, 400) # Timeout: 400 msec
+      # end
+
+      def wait_for_change
+        watcher = install_watcher
+        begin
+          yield
+        ensure
+          result = evaluate_watcher(watcher, 400)
+        end
+        result
+      end
+
+      def copy
+        wait_for_change { Keyboard.type '(ctrl c)' }
+      end
+
+      def paste
+        Keyboard.type '(ctrl v)'
+      end
+      
+      def get_selection
+        Clipboard.get if Clipboard.copy
+      end
+
+      def append
+        clip = Clipboard.get
+        if (selection = get_selection)
+          clip << "\n" << selection
+        end
+      ensure
+        Clipboard.set clip
+      end
+    end
+  end
+end
+

data/lib/rum/windows/system_foreign_functions.rb

@@ -0,0 +1,129 @@
+require_relative 'system.so'
+# The C-extension defines:
+# Rum::System (module methods)
+#        enum_windows
+#        keybd_event
+#        get_console_window
+# Rum::System::Window (class instance methods)
+#        show
+#        exe_path
+#        ...
+# Rum::System::Desktop (module methods)
+#        top
+#        right
+#        bottom
+#        left
+# Rum::System::Screen (module methods)
+#        width
+#        height
+# Rum::System::Clipboard (module methods)
+#        install_watcher
+#        evaluate_watcher
+
+require 'win32/api'
+require 'win32/clipboard'
+
+module Rum
+  module System
+    def self.snake_case str
+      str.gsub(/([a-z])([A-Z0-9])/, '\1_\2').downcase
+    end
+
+    def self.def_api(function_name, parameters, return_value, rename=nil)
+      api = Win32::API.new function_name, parameters, return_value, 'user32'
+      define_method(rename || snake_case(function_name)) do |*args|
+        api.call *args
+      end
+    end
+
+    def_api 'FindWindow',          'PP', 'L'
+    def_api 'FindWindowEx',        'LLPP', 'L'
+    def_api 'SendMessage',         'LLLP', 'L', :send_with_buffer
+    def_api 'SendMessage',         'LLLL', 'L'
+    def_api 'PostMessage',         'LLLL', 'L'
+    def_api 'GetDlgItem',          'LL', 'L' 
+    def_api 'GetWindowRect',       'LP', 'I'
+    def_api 'SetCursorPos',        'LL', 'I' 
+    def_api 'mouse_event',         'LLLLL', 'V'
+    def_api 'IsWindow',            'L', 'L'
+    def_api 'IsWindowVisible',     'L', 'L'
+    def_api 'SetForegroundWindow', 'L', 'L'
+    
+    def_api 'GetWindowLong',       'LI', 'L'
+    def_api 'GetForegroundWindow', 'V', 'L'
+    def_api 'GetClassName',        'LPI', 'I'
+    def_api 'GetWindowText',       'LPI', 'I'
+    def_api 'GetWindowTextW',      'LPI', 'I'
+    def_api 'ShowWindow',          'LI', 'I'
+    def_api 'SetForegroundWindow', 'L', 'I'
+    def_api 'IsIconic',            'L', 'L'
+    def_api 'IsZoomed',            'L', 'L'
+    def_api 'SetWindowPos',        'LLIIIII', 'I'
+    def_api 'MoveWindow',          'LIIIIB', 'L'
+    
+    ShellExecute = Win32::API.new('ShellExecute', 'LPPPPI', 'L', 'shell32')
+    def start command, parameters=0
+      ShellExecute.call(0, 'open', command, parameters, 0, 1)
+    end
+    
+    WM_COMMAND    = 0x0111 
+    WM_SYSCOMMAND = 0x0112
+    
+    SC_CLOSE    = 0xF060
+    SC_RESTORE  = 0xF120;
+    SC_MAXIMIZE = 0xF030;
+    SC_MINIMIZE = 0xF020;
+
+    WM_GETTEXT = 0x000D
+    EM_GETSEL  = 0x00B0
+    EM_SETSEL  = 0x00B1
+    SW_HIDE             = 0
+    SW_SHOWNORMAL       = 1
+    SW_NORMAL           = 1
+    SW_SHOWMINIMIZED    = 2
+    SW_SHOWMAXIMIZED    = 3
+    SW_MAXIMIZE         = 3
+    SW_SHOWNOACTIVATE   = 4
+    SW_SHOW             = 5
+    SW_MINIMIZE         = 6
+    SW_SHOWMINNOACTIVE  = 7
+    SW_SHOWNA           = 8
+    SW_RESTORE          = 9
+    SW_SHOWDEFAULT      = 10
+    SW_FORCEMINIMIZE    = 11
+    SW_MAX              = 11
+
+    SWP_NOMOVE     = 2
+    SWP_NOSIZE     = 1
+    HWND_TOPMOST   = -1
+    HWND_NOTOPMOST = -2
+    GWL_EXSTYLE    = -20
+    WS_EX_TOPMOST  = 8
+  end
+end
+
+module Win32
+  class Clipboard
+    def self.get
+      begin
+        self.open
+        return '' unless IsClipboardFormatAvailable(UNICODETEXT)
+        handle = GetClipboardData(UNICODETEXT)
+        clip_data = 0.chr * GlobalSize(handle)
+        memcpy(clip_data, handle, clip_data.size)
+        clip_data.force_encoding(Encoding::UTF_16LE)
+        clip_data.encode!(Encoding::UTF_8, invalid: :replace, replace: '', \
+                          universal_newline: true)
+        clip_data[ /^[^\0]*/ ]
+      ensure
+        self.close
+      end
+    end
+
+    def self.set str
+      data = str.encode(Encoding::UTF_16LE, crlf_newline: true)
+      data.force_encoding Encoding::BINARY
+      set_data(data, UNICODETEXT)
+    end
+  end
+end

data/rum.gemspec

@@ -0,0 +1,14 @@
+require 'rubygems'
+
+Gem::Specification.new do |spec|
+  spec.name       = 'rum'
+  spec.version    = '0.0.1'
+  spec.author     = 'nonsequitur'
+  spec.license    = 'MIT'
+  spec.homepage   = 'http://nonsequitur.github.com/rum'
+  spec.summary    = 'A cross-platform Hotkey and Macro utility, running on Windows and Mac OS.'
+  spec.required_ruby_version = '>= 1.9.1'
+
+  spec.executables << 'rum-client'
+  spec.extra_rdoc_files = ['README']
+end