rkremap 0.3.0.1 → 0.5.0

data/lib/rkremap/evdev.rb

@@ -1,53 +1,11 @@
 class Rkremap
-  EV_KEY = 1
-  EV_SYN = 0
-  SYN_REPORT = 0
-
   # https://www.kernel.org/doc/html/v4.12/input/input_uapi.html
   class Evdev
     EVIOCGBIT_ANY = 2147566880     # EVIOCGBIT(0, 1)
-    EVIOCGBIT_EV_KEY = 2153792801  # EVIOCGBIT(EV_KEY, (KEY_MAX-1)/8+1)
+    EVIOCGBIT_EV_KEY = 2153792801  # EVIOCGBIT(EV_KEY, (KEY_CNT-1)/8+1)
     EVIOCGRAB = 1074021776
     EVIOCGNAME = 2164278534        # EVIOCGNAME(256)
 
-    @io2evdev = {}
-    def self.io2evdev
-      @io2evdev
-    end
-
-    @device_list = {}
-
-    # @return [Array<Evdev>]
-    def self.detect
-      devs = []
-      new_devs = Dir.children("/dev/input").map do |dev|
-        if dev =~ /\Aevent\d+\z/
-          path = "/dev/input/#{dev}"
-          devs.push path
-          next if @device_list[path]
-          @device_list[path] = true
-          Evdev.new(path)
-        end
-      end.compact
-      (@device_list.keys - devs).each do |path|
-        @device_list.delete path
-      end
-      new_devs
-    end
-
-    def self.remove_device(evdev)
-      @io2evdev.delete evdev.io
-      @device_list.delete evdev.path
-    end
-
-    # @param inputs [Array<Evdev>]
-    # @param timeout [Numeric]
-    # @return [Evdev]
-    def self.select(inputs, timeout)
-      r, = IO.select(inputs.map(&:io), nil, nil, timeout)
-      r && @io2evdev[r[0]]
-    end
-
     # @return [String]
     attr_reader :path
 
@@ -57,6 +15,11 @@ class Rkremap
       @io = nil
       @capa = nil
       @uinput = nil
+      @grab = false
+    end
+
+    def inspect
+      "#<Rkremap::Evdev: #{name} (#{path})>"
     end
 
     def eql?(other)
@@ -67,13 +30,18 @@ class Rkremap
     def io
       return @io if @io
       @io = File.open(@path)
-      self.class.io2evdev[@io] = self
-      @io
+    end
+
+    # @return [String]
+    def name
+      return @name if @name
+      buf = ''
+      io.ioctl(EVIOCGNAME, buf)
+      @name = buf.sub(/\0+$/, '')
     end
 
     def close
       @io&.close
-      self.class.remove_device(self)
     end
 
     # @param key [Integer]
@@ -89,18 +57,37 @@ class Rkremap
 
     # @return [Boolean]
     def keyboard?
-      buf = ' '
-      io.ioctl(EVIOCGNAME, buf)
-      return false if buf =~ /\Arkremap\0*\z/
+      return @is_keyboard unless @is_keyboard.nil?
+      @is_keyboard = false
+      return false if name =~ /\Arkremap\0*\z/
+
+      buf = ''
+      io.ioctl(EVIOCGBIT_ANY, buf)
+      return false if buf.unpack1('C')[EV_KEY] == 0
+      @is_keyboard = capable?(KeyCode::KEY_0) && capable?(KeyCode::KEY_9) &&
+                     capable?(KeyCode::KEY_A) && capable?(KeyCode::KEY_Z) && capable?(KeyCode::KEY_SPACE)
+    end
+
+    # @return [Boolean]
+    def mouse?
+      return @is_mouse unless @is_mouse.nil?
+      @is_mouse = false
+      return false if name =~ /\Arkremap\0*\z/
 
+      buf = ''
       io.ioctl(EVIOCGBIT_ANY, buf)
       return false if buf.unpack1('C')[EV_KEY] == 0
-      capable?(KeyCode::KEY_0) && capable?(KeyCode::KEY_9) &&
-        capable?(KeyCode::KEY_A) && capable?(KeyCode::KEY_Z) && capable?(KeyCode::KEY_SPACE)
+      @is_mouse = capable?(KeyCode::BTN_MOUSE)
     end
 
     def grab
       io.ioctl(EVIOCGRAB, 1)
+      @grab = true
+    end
+
+    # @return [Boolean]
+    def grab?
+      @grab
     end
 
     # struct input_event {
@@ -109,46 +96,21 @@ class Rkremap
     #   unsigned short code;
     #   unsigned int value;
     # };
-    # @return [Array<Time, Integer, Integer, Integer>]
+    # @return [Rkremap::Event]
     def read_event
       raw = io.sysread(24)  # sizeof(struct input_event)
       sec, usec, type, code, value = raw.unpack('Q!Q!SSl')
       time = Time.at(sec, usec)
-      return time, type, code, value
-    end
-  end
-
-  # https://www.kernel.org/doc/html/v4.12/input/uinput.html
-  class Uinput
-    UI_SET_EVBIT = 1074025828
-    UI_SET_KEYBIT = 1074025829
-    UI_DEV_SETUP = 1079792899
-    UI_DEV_CREATE = 21761
-
-    def initialize
-      @dev = File.open('/dev/uinput', 'w')
-      setup
-    end
-
-    def setup
-      @dev.ioctl(UI_SET_EVBIT, EV_KEY)
-      KeyCode::KEY_CNT.times{|k| @dev.ioctl(UI_SET_KEYBIT, k)}
-      bustype = 0x03    # BUS_USB
-      vendor = 0x1234   # てきとー
-      product = 0x5678  # てきとー
-      version = 1       # てきとー
-      name = 'rkremap'
-      ff_efects_max = 0
-      setup = [bustype, vendor, product, version, name, ff_efects_max].pack("SSSSZ80L") # struct uinput_setup
-      @dev.ioctl(UI_DEV_SETUP, setup)
-      @dev.ioctl(UI_DEV_CREATE)
+      Event.new(self, time, type, code, value)
     end
 
-    # @param type [Integer] EV_KEY/ EV_SYN
-    # @param code [Integer] キーコード / SYN_REPORT
-    # @param value [Integer] 0:release / 1:press / 2:repeat
-    def write_event(type, code, value)
-      @dev.syswrite(['', type, code, value].pack('a16SSl'))
+    # @param pattern [Symbol, Regexp]
+    def match?(pattern)
+      return true if pattern == true
+      return true if pattern == :keyboard && keyboard?
+      return true if pattern == :mouse && mouse?
+      return true if pattern.is_a?(Regexp) && pattern =~ name
+      return false
     end
   end
 end

data/lib/rkremap/evdev_list.rb

@@ -0,0 +1,90 @@
+require 'tmtms/timer'
+
+class Rkremap
+  class EvdevList
+    include Enumerable
+
+    attr_reader :device_list
+    attr_accessor :grab
+
+    # @param device_files [Array<String>] path list (/dev/input/event*)
+    # @param auto_detect [Boolean]
+    # @param detect_mouse [Boolean]
+    # @param exclude [Regexp]
+    def initialize(device_files, auto_detect: false, detect_mouse: false, exclude: nil)
+      @device_files = device_files
+      @auto_detect = auto_detect
+      @detect_mouse = detect_mouse
+      @exclude = exclude
+      @io2evdev = {}     # {IO => Evdev}
+      @device_list = {}  # {path => Evdev}
+      @prev_devices = []
+    end
+
+    def each(&block)
+      @device_list.each_value do |evdev|
+        block.call evdev
+      end
+    end
+
+    def detect_loop
+      detect
+      timer = Tmtms::Timer.new
+      timer.repeat(3) do
+        detect
+      end
+    end
+
+    def detect
+      devs = Dir.glob("/dev/input/event*")
+      (@device_list.keys - devs).each{|path| remove_device(path)}
+      new_devs = devs - @prev_devices
+      @prev_devices = devs
+      new_devs &= @device_files unless @auto_detect
+      new_devs -= @device_list.keys
+      new_devs.each do |path|
+        ev = Evdev.new(path)
+        stat = nil
+        if @exclude && ev.name =~ @exclude
+          stat = 'excluded'
+        elsif ev.keyboard? || (@detect_mouse && ev.mouse?)
+          ev.grab if ev.match?(@grab)
+          @io2evdev[ev.io] = ev
+          @device_list[path] = ev
+          stat = ev.grab? ? 'grabbed' : 'not grabbed'
+        else
+          stat = 'ignored'
+        end
+        puts "detect: #{ev.name} (#{ev.path}) #{stat}" if $VERBOSE
+      end
+    end
+
+    # @param evdev [path]
+    def remove_device(path)
+      evdev = @device_list[path]
+      puts "disconnect: #{evdev.name} (#{evdev.path})" if $VERBOSE
+      @io2evdev.delete evdev.io
+      evdev.close
+      @device_list.delete path
+    end
+
+    # @param timeout [Numeric]
+    # @return [Evdev]
+    def select(timeout)
+      r, = IO.select(@device_list.values.map(&:io), nil, nil, timeout)
+      r && @io2evdev[r[0]]
+    end
+
+    def read_event
+      while true
+        evdev = select(3)
+        next unless evdev
+        begin
+          return evdev.read_event
+        rescue Errno::ENODEV
+          remove_device(evdev.path)
+        end
+      end
+    end
+  end
+end

data/lib/rkremap/event.rb

@@ -0,0 +1,48 @@
+class Rkremap
+  class Event
+    attr_accessor :app
+    attr_reader :device
+    attr_reader :time
+    attr_reader :ev_type
+    attr_accessor :code
+    attr_accessor :value
+
+    # @param evdev [Rkremap::Evdev]
+    # @param time [Time]
+    # @param type [Integer]
+    # @param code [Integer]
+    # @param value [Integer]
+    def initialize(evdev, time, type, code, value)
+      @device = evdev
+      @time = time
+      @ev_type = type
+      @code = code
+      @value = value
+      @app = nil
+      @skipped = false
+    end
+
+    def inspect
+      "#<Rkremap::Event: #{code}(#{CODE_KEY[code]}) #{value} #{device.inspect}>"
+    end
+
+    def type
+      @type ||= value == 1 ? :press : value == 0 ? :release : :repeat
+    end
+
+    def match?(device:, code:, type:, app:)
+      (device.nil? || self.device.match?(device)) &&
+        (code.nil? || Array(code).include?(self.code)) &&
+        (type.nil? || Array(type).include?(self.type)) &&
+        (app.nil? || @app.nil? || @app.match?(app))
+    end
+
+    def skip
+      @skipped = true
+    end
+
+    def skipped?
+      @skipped
+    end
+  end
+end

data/lib/rkremap/keycode.rb

@@ -249,6 +249,87 @@ class Rkremap
     KEY_WIMAX                    = KEY_WWAN
     KEY_RFKILL                   = 247 # Key that controls all radios
     KEY_MICMUTE                  = 248 # Mute / unmute the microphone
+
+    BTN_MISC                     = 0x100
+    BTN_0                        = 0x100
+    BTN_1                        = 0x101
+    BTN_2                        = 0x102
+    BTN_3                        = 0x103
+    BTN_4                        = 0x104
+    BTN_5                        = 0x105
+    BTN_6                        = 0x106
+    BTN_7                        = 0x107
+    BTN_8                        = 0x108
+    BTN_9                        = 0x109
+
+    BTN_LEFT                     = 0x110
+    BTN_MOUSE                    = 0x110
+    BTN_RIGHT                    = 0x111
+    BTN_MIDDLE                   = 0x112
+    BTN_SIDE                     = 0x113
+    BTN_EXTRA                    = 0x114
+    BTN_FORWARD                  = 0x115
+    BTN_BACK                     = 0x116
+    BTN_TASK                     = 0x117
+
+    BTN_JOYSTICK                 = 0x120
+    BTN_TRIGGER                  = 0x120
+    BTN_THUMB                    = 0x121
+    BTN_THUMB2                   = 0x122
+    BTN_TOP                      = 0x123
+    BTN_TOP2                     = 0x124
+    BTN_PINKIE                   = 0x125
+    BTN_BASE                     = 0x126
+    BTN_BASE2                    = 0x127
+    BTN_BASE3                    = 0x128
+    BTN_BASE4                    = 0x129
+    BTN_BASE5                    = 0x12a
+    BTN_BASE6                    = 0x12b
+    BTN_DEAD                     = 0x12f
+
+    BTN_GAMEPAD                  = 0x130
+    BTN_SOUTH                    = 0x130
+    BTN_A                        = BTN_SOUTH
+    BTN_EAST                     = 0x131
+    BTN_B                        = BTN_EAST
+    BTN_C                        = 0x132
+    BTN_NORTH                    = 0x133
+    BTN_X                        = BTN_NORTH
+    BTN_WEST                     = 0x134
+    BTN_Y                        = BTN_WEST
+    BTN_Z                        = 0x135
+    BTN_TL                       = 0x136
+    BTN_TR                       = 0x137
+    BTN_TL2                      = 0x138
+    BTN_TR2                      = 0x139
+    BTN_SELECT                   = 0x13a
+    BTN_START                    = 0x13b
+    BTN_MODE                     = 0x13c
+    BTN_THUMBL                   = 0x13d
+    BTN_THUMBR                   = 0x13e
+
+    BTN_DIGI                     = 0x140
+    BTN_TOOL_PEN                 = 0x140
+    BTN_TOOL_RUBBER              = 0x141
+    BTN_TOOL_BRUSH               = 0x142
+    BTN_TOOL_PENCIL              = 0x143
+    BTN_TOOL_AIRBRUSH            = 0x144
+    BTN_TOOL_FINGER              = 0x145
+    BTN_TOOL_MOUSE               = 0x146
+    BTN_TOOL_LENS                = 0x147
+    BTN_TOOL_QUINTTAP            = 0x148 # Five fingers on trackpad
+    BTN_STYLUS3                  = 0x149
+    BTN_TOUCH                    = 0x14a
+    BTN_STYLUS                   = 0x14b
+    BTN_STYLUS2                  = 0x14c
+    BTN_TOOL_DOUBLETAP           = 0x14d
+    BTN_TOOL_TRIPLETAP           = 0x14e
+    BTN_TOOL_QUADTAP             = 0x14f # Four fingers on trackpad
+
+    BTN_WHEEL                    = 0x150
+    BTN_GEAR_DOWN                = 0x150
+    BTN_GEAR_UP                  = 0x151
+
     KEY_OK                       = 0x160
     KEY_SELECT                   = 0x161
     KEY_GOTO                     = 0x162
@@ -502,7 +583,6 @@ class Rkremap
 
   CODE_KEY = {}
   KeyCode.constants.each do |k|
-    next unless k =~ /\AKEY_/
     code = KeyCode.const_get(k)
     CODE_KEY[code] = k unless CODE_KEY[code]
   end

data/lib/rkremap/remap.rb

@@ -0,0 +1,105 @@
+class Rkremap
+  class Remap
+    # @param map [Hash] from => to
+    # @param hold [Numeric] seconds
+    def initialize(rk, map:, hold:)
+      @rk = rk
+      @pressed = {}
+
+      if hold
+        setup_holding_event(map, hold)
+      else
+        setup_event(map)
+      end
+    end
+
+    # @param map [Hash] {form => to}
+    def setup_event(map)
+      keycodes = map.keys.flatten
+
+      @rk.match(code: keycodes, type: :press) do |event, current_mod|
+        map.each do |from, to|
+          if event_match?(from, event, current_mod)
+            make_press_event(to, event)
+            break
+          end
+        end
+      end
+
+      @rk.match(code: keycodes, type: :release) do |event|
+        if @pressed[event.code]
+          @rk.event(code: @pressed[event.code], type: :release)
+          @pressed[event.code] = nil
+          event.skip
+        end
+      end
+    end
+
+    # @param map [Hash] {form => to}
+    # @param hold [Numeric] seconds
+    def setup_holding_event(map, hold)
+      keycodes = map.keys.flatten
+      timer = Tmtms::Timer.new
+      jobs = map.transform_values{nil}
+
+      @rk.match(code: keycodes, type: :press) do |event|
+        map.each do |from, to|
+          if from == event.code
+            jobs[event.code] = timer.set(hold) do
+              @rk.synchronize{ make_press_event(to) }
+            end
+            event.skip
+            break
+          end
+        end
+      end
+
+      @rk.match(code: keycodes, type: :repeat) do |event|
+        event.skip
+      end
+
+      @rk.match(code: keycodes, type: :release) do |event|
+        if jobs[event.code]&.cancel
+          @rk.key(event.code)
+        elsif map[event.code]
+          @rk.event(code: map[event.code], type: :release)
+        end
+        event.skip
+      end
+
+      @rk.match(type: :press) do
+        jobs.each do |code, job|
+          make_press_event(map[code]) if job&.cancel
+        end
+      end
+    end
+
+    # @param keys [Integer, Array<Integer>] KEY_A, [KEY_LEFTCTRL, KEY_A], [:SHIFT, KEY_A], ...
+    # @param event [Rkremap::Event]
+    # @param mod [Hash] current modifiers status
+    def event_match?(keys, event, mod)
+      if keys == event.code
+        return true
+      end
+      keys = Array(keys)
+      return false unless keys.include? event.code
+      (keys - [event.code]).all?{mod[_1]}
+    end
+
+    # @param to [Integer, Array<Integer>]
+    # @param event [Rkremap::Event, nil]
+    def make_press_event(to, event=nil)
+      Array(to).each do |key|
+        @pressed[event.code] = key if event
+        if @rk.modifiers.include? key
+          @rk.event(code: key, type: :press)
+          event&.skip
+        elsif event
+          event.code = key
+        else
+          @rk.key(code: key, type: :press)
+        end
+      end
+    end
+  end
+end

data/lib/rkremap/uinput.rb

@@ -0,0 +1,40 @@
+class Rkremap
+  # https://www.kernel.org/doc/html/v4.12/input/uinput.html
+  class Uinput
+    UI_SET_EVBIT = 1074025828
+    UI_SET_KEYBIT = 1074025829
+    UI_SET_RELBIT = 1074025830
+    UI_DEV_SETUP = 1079792899
+    UI_DEV_CREATE = 21761
+
+    REL_CNT = 0x10
+
+    def initialize
+      @dev = File.open('/dev/uinput', 'w')
+      setup
+    end
+
+    def setup
+      @dev.ioctl(UI_SET_EVBIT, EV_KEY)
+      KeyCode::KEY_CNT.times{|k| @dev.ioctl(UI_SET_KEYBIT, k)}
+      @dev.ioctl(UI_SET_EVBIT, EV_REL)
+      REL_CNT.times{|r| @dev.ioctl(UI_SET_RELBIT, r)}
+      bustype = 0x03    # BUS_USB
+      vendor = 0x1234   # てきとー
+      product = 0x5678  # てきとー
+      version = 1       # てきとー
+      name = 'rkremap'
+      ff_efects_max = 0
+      setup = [bustype, vendor, product, version, name, ff_efects_max].pack("SSSSZ80L") # struct uinput_setup
+      @dev.ioctl(UI_DEV_SETUP, setup)
+      @dev.ioctl(UI_DEV_CREATE)
+    end
+
+    # @param type [Integer] EV_KEY/ EV_SYN
+    # @param code [Integer] キーコード / SYN_REPORT
+    # @param value [Integer] 0:release / 1:press / 2:repeat
+    def write_event(type, code, value)
+      @dev.syswrite(['', type, code, value].pack('a16SSl'))
+    end
+  end
+end

data/lib/rkremap/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 class Rkremap
-  VERSION = "0.3.0.1"
+  VERSION = "0.5.0"
 end

data/lib/rkremap/winattr.rb

@@ -46,11 +46,14 @@ class Rkremap
     # @return [String]
     def app_title(window)
       win = app_win(window)[0]
+      return '' if win == 0
       prop = X11::XTextProperty.malloc(Fiddle::RUBY_FREE)
       text_list = X11::Pointer.malloc(Fiddle::RUBY_FREE)
       X11.XGetWMName(@display, win, prop)
       X11.Xutf8TextPropertyToTextList(@display, prop, text_list, @buf)
-      title = text_list.ptr.ptr.to_s.force_encoding('utf-8')
+      ptr = text_list.ptr
+      return '' if ptr.null?
+      title = ptr.ptr.to_s.force_encoding('utf-8')
       X11.XFreeStringList(text_list.ptr)
       title
     end