rkremap 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 945483dcd11240d5c54b359a15055219532f2f311275633503e7d485aa68917b
-  data.tar.gz: d594c9b549696f964cde750094169ba7198c0ac9ab5f5dfc59a7936d9610a09d
+  metadata.gz: b9cb72d3bf69bf2d0641eb7856903d110f0a253d5377c2d97ea452a1f0fd5c13
+  data.tar.gz: 1dd89d3569fc5396a6c314618bcc975e04b63b0f2a55d9c3dd8cd5497f1e2164
 SHA512:
-  metadata.gz: a1412a373a0673172a9eba0449613ab9a76debf400b622977887031c294f8381a58a7c060ef6451e6264703690974aeddb305ce35619c07958a2b5a1239ecf12
-  data.tar.gz: 619ee1de0814c2542c2a4f347dca7c48e640590a01ff58f2b7362ed80ef447acaf1576df2a2a659eaac95c9a7ec0a8f52d732d8024356613fbef43f055b16dd2
+  metadata.gz: 8cf20b635b0aa1584e81cc0317040931f62f011e004189e30428534c04456159cde7ac37c5f790cc1af06d396b9d60c36c4398a5d51947dcf99309eb3edb584b
+  data.tar.gz: cde725ba50b0d5b2fd3daf837af9e1244a1cc5cb3d40f497d9dbf00e6655edc0e436dcdc94056b8bc9efa3887ec2419fd7e53b11e8289c142217be2dd577f8ef

data/README.md

@@ -25,7 +25,9 @@ KERNEL=="uinput", GROUP="input"
 
 ### `Rkremap.new`
 
-引数に `/dev/input/event*` を指定する。キーボードデバイス以外を指定してもたぶん無意味。
+引数を指定しなければ自動的にキーボードデバイスの検出を試みる。
+
+引数を指定する場合は `/dev/input/event*` を指定する。キーボードデバイス以外を指定してもたぶん無意味。
 キーボードのデバイスファイル名を調べるには
 ```
 cat /proc/bus/input/devices
@@ -79,6 +81,7 @@ ThinkPad 本体のキーボードは `/dev/input/event3` だった。
   * [xkeysnail - もうひとつの Linux 向けキーリマッパ - Qiita](https://qiita.com/mooz@github/items/c5f25f27847333dd0b37)
 * [k0kubun/xremap: Dynamic key remapper for X11 and Wayland](https://github.com/k0kubun/xremap)
   * [Linux向けの最強のキーリマッパーを作った - k0kubun's blog](https://k0kubun.hatenablog.com/entry/xkremap)
+  * [Linux用キーリマッパーxremapをRustで書き直した - k0kubun's blog](https://k0kubun.hatenablog.com/entry/xremap)
 * [kui/rbindkeys: key remap with ruby](https://github.com/kui/rbindkeys)
   * [Ruby で設定できる Linux 環境向けキーリマッパー作った — 電卓片手に](http://k-ui.jp/blog/2012/06/06/rbindkeys-configurable-key-remapper-in-ruby/)
 * [Linux の入力デバイスをカスタマイズ - Qiita](https://qiita.com/propella/items/a73fc333c95d14d06835)

data/example/henkan-alt.rb

@@ -4,7 +4,7 @@
 require 'rkremap'
 include Rkremap::KeyCode
 
-rk = Rkremap.new('/dev/input/event3')
+rk = Rkremap.new
 rk.grab = true
 
 class State

data/example/keylogger.rb

@@ -6,7 +6,7 @@ def code2key(code)
   Rkremap::CODE_KEY[code].to_s.sub(/\AKEY_/, '')
 end
 
-rk = Rkremap.new('/dev/input/event3')
+rk = Rkremap.new
 rk.grab = false
 rk.x11 = true
 rk.start do |code, mod, app|

data/example/mac-shortcut.rb

@@ -33,7 +33,7 @@ mod_disable_meta = {
 }
 mod_disable_all = {}.merge(mod_disable_ctrl, mod_disable_shift, mod_disable_alt, mod_disable_meta)
 
-rk = Rkremap.new("/dev/input/event3")
+rk = Rkremap.new
 rk.grab = true
 rk.x11 = true
 rk.start do |code, mod, app|

data/example/tmtms.rb

@@ -40,7 +40,7 @@ class State
   end
 end
 
-rk = Rkremap.new("/dev/input/event3")
+rk = Rkremap.new
 rk.grab = true
 rk.x11 = true
 
@@ -160,10 +160,15 @@ rk.start do |code, mod, app|
   # Ctrl は他のキーに変換
   if mod[KEY_LEFTCTRL] || mod[KEY_RIGHTCTRL]
     # Ctrl+K は行末まで削除 (日本語変換時はそのまま)
-    if code == KEY_K && !File.exist?('/tmp/fcitx-enabled')
-      rk.key(KEY_END, mod_disable_all.merge({KEY_LEFTSHIFT => true}))
-      rk.key(KEY_X, mod_disable_all.merge({KEY_LEFTCTRL => true}))
-      next
+    if code == KEY_K
+      if File.exist?('/tmp/fcitx-enabled')
+        rk.key(code, mod)
+        next
+      else
+        rk.key(KEY_END, mod_disable_all.merge({KEY_LEFTSHIFT => true}))
+        rk.key(KEY_X, mod_disable_all.merge({KEY_LEFTCTRL => true}))
+        next
+      end
     end
     # 単純な変換
     if ctrl_map[code]

data/lib/rkremap/evdev.rb

@@ -0,0 +1,124 @@
+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)
+    EVIOCGRAB = 1074021776
+
+    @io2evdev = {}
+    def self.io2evdev
+      @io2evdev
+    end
+
+    # @return [Array<Evdev>]
+    def self.devices
+      Dir.children("/dev/input").map do |dev|
+        dev =~ /\Aevent\d+\z/ ? Evdev.new("/dev/input/#{dev}") : nil
+      end.compact
+    end
+
+    # @param inputs [Array<Evdev>]
+    # @return [Evdev]
+    def self.select(inputs)
+      h = inputs.map{|ev| [ev.io, ev]}.to_h
+      r, = IO.select(inputs.map(&:io), nil, nil, nil)
+      @io2evdev[r[0]]
+    end
+
+    # @param path [String]
+    def initialize(path)
+      @path = path
+      @io = nil
+      @capa = nil
+      @uinput = nil
+    end
+
+    # @return [IO]
+    def io
+      return @io if @io
+      @io = File.open(@path)
+      self.class.io2evdev[@io] = self
+      @io
+    end
+
+    def close
+      @io&.close
+    end
+
+    # @param key [Integer]
+    # @return [Boolean]
+    def capable?(key)
+      unless @capa
+        buf = ' ' * ((KeyCode::KEY_MAX-1)/8+1)
+        io.ioctl(EVIOCGBIT_EV_KEY, buf)
+        @capa = buf.unpack('C*')
+      end
+      @capa[key/8][key%8] != 0
+    end
+
+    # @return [Boolean]
+    def keyboard?
+      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)
+    end
+
+    def grab
+      io.ioctl(EVIOCGRAB, 1)
+    end
+
+    # struct input_event {
+    #   struct timeval time;
+    #   unsigned short type;
+    #   unsigned short code;
+    #   unsigned int value;
+    # };
+    # @return [Array<Time, Integer, Integer, Integer>]
+    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)
+    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.1.0"
+  VERSION = "0.2.0"
 end

data/lib/rkremap/winattr.rb

@@ -1,12 +1,4 @@
-require 'ffi'
-
-module FFI
-  typedef :pointer, :Display
-  typedef :ulong, :XID
-  typedef :XID, :Window
-  typedef :int, :Status
-  typedef :ulong, :Atom
-end
+require 'fiddle/import'
 
 class Rkremap
   # アプリの name, class, title を取得する
@@ -14,38 +6,39 @@ class Rkremap
     def initialize
       @app_win = {}
       @display = X11.XOpenDisplay(nil) or raise "Cannot open display: #{ENV['DISPLAY']}"
+      @buf = ' '*8
     end
 
+    # @return [Integer] Window
     def focus_win
-      win = FFI::MemoryPointer.new(:Window)
-      revert_to = FFI::MemoryPointer.new(:int)
-      X11.XGetInputFocus(@display, win, revert_to)
-      win.read(:Window)
+      win_buf = ' ' * X11.sizeof('Window')
+      X11.XGetInputFocus(@display, win_buf, @buf)
+      win_buf.unpack1('L')
     end
 
     # @return [Array<window, name, class>]
     def app_win(window)
       return @app_win[window] if @app_win[window]
-      class_hint = XClassHint.new
-      root = FFI::MemoryPointer.new(:Window)
-      parent = FFI::MemoryPointer.new(:Window)
-      children = FFI::MemoryPointer.new(:pointer)
-      nchildren = FFI::MemoryPointer.new(:int)
+      class_hint = X11::XClassHint.malloc
+      parent = ' ' * X11.sizeof('Window')
+      children = ' ' * X11.sizeof('void*')
       win = window
       while win > 0
+        class_hint.name = nil
+        class_hint.class_name = nil
         X11.XGetClassHint(@display, win, class_hint)
-        X11.XQueryTree(@display, win, root, parent, children, nchildren)
-        X11.XFree(children.read_pointer)
-        break unless class_hint[:name].null? && class_hint[:class].null?
-        win = parent.read(:Window)
+        X11.XQueryTree(@display, win, @buf, parent, children, @buf)
+        X11.XFree(Fiddle::Pointer[children].ptr)
+        break unless class_hint.name.null? && class_hint.class_name.null?
+        win = parent.unpack1('L')
       end
-      unless class_hint[:name].null?
-        win_name = class_hint[:name].read_string
-        X11.XFree(class_hint[:name])
+      unless class_hint.name.null?
+        win_name = class_hint.name.to_s
+        X11.XFree(class_hint.name)
       end
-      unless class_hint[:class].null?
-        win_class = class_hint[:class].read_string
-        X11.XFree(class_hint[:class])
+      unless class_hint.class_name.null?
+        win_class = class_hint.class_name.to_s
+        X11.XFree(class_hint.class_name)
       end
       @app_win[window] = [win, win_name, win_class]
     end
@@ -53,43 +46,33 @@ class Rkremap
     # @return [String]
     def app_title(window)
       win = app_win(window)[0]
-      prop = XTextProperty.new
-      text_list = FFI::MemoryPointer.new(:pointer)
-      count = FFI::MemoryPointer.new(:int)
+      prop = X11::XTextProperty.malloc
+      text_list = ' ' * X11.sizeof('void*')
       X11.XGetWMName(@display, win, prop)
-      X11.Xutf8TextPropertyToTextList(@display, prop, text_list, count)
-      title = text_list.read(:pointer).read(:string)&.force_encoding('utf-8')
-      X11.XFreeStringList(text_list.read(:pointer))
+      X11.Xutf8TextPropertyToTextList(@display, prop, text_list, @buf)
+      title = Fiddle::Pointer[text_list].ptr.ptr.to_s.force_encoding('utf-8')
+      X11.XFreeStringList(Fiddle::Pointer[text_list].ptr)
       title
     end
   end
 
-  class XClassHint < FFI::Struct
-    layout(
-      :name, :pointer,
-      :class, :pointer,
-    )
-  end
-
-  class XTextProperty < FFI::Struct
-    layout(
-      :value, :pointer,
-      :encoding, :Atom,
-      :format, :int,
-      :nitems, :ulong,
-    )
-  end
-
   module X11
-    extend FFI::Library
-    ffi_lib 'X11'
-    attach_function :XOpenDisplay, [:string], :Display
-    attach_function :XGetInputFocus, [:Display, :pointer, :pointer], :int
-    attach_function :XGetClassHint, [:Display, :Window, XClassHint.by_ref], :int
-    attach_function :XQueryTree, [:Display, :Window, :pointer, :pointer, :pointer, :pointer], :Status
-    attach_function :XGetWMName, [:Display, :Window, XTextProperty.by_ref], :void
-    attach_function :Xutf8TextPropertyToTextList, [:Display, XTextProperty.by_ref, :pointer, :pointer], :int
-    attach_function :XFree, [:pointer], :int
-    attach_function :XFreeStringList, [:pointer], :void
+    extend Fiddle::Importer
+    dlload 'libX11.so'
+    typealias 'XID', 'unsigned long'
+    typealias 'Window', 'XID'
+    typealias 'Status', 'int'
+    typealias 'Atom', 'unsigned long'
+    XClassHint = struct(['char *name', 'char *class_name'])
+    XTextProperty = struct(['unsigned char *value', 'Atom encoding', 'int format', 'unsigned long nitems'])
+
+    extern 'Display* XOpenDisplay(char*)'
+    extern 'int XGetInputFocus(Display*, Window*, int*)'
+    extern 'int XGetClassHint(Display*, Window, XClassHint*)'
+    extern 'Status XQueryTree(Display*, Window, Window*, Window*, Window**, unsigned int*)'
+    extern 'Status XGetWMName(Display*, Window, XTextProperty*)'
+    extern 'int Xutf8TextPropertyToTextList(Display*, XTextProperty*, char***, int*)'
+    extern 'int XFree(void*)'
+    extern 'void XFreeStringList(char**)'
   end
 end

data/lib/rkremap.rb

@@ -3,20 +3,11 @@
 require_relative "rkremap/version"
 require_relative 'rkremap/keycode'
 require_relative 'rkremap/winattr'
+require_relative 'rkremap/evdev'
 
 class Rkremap
   include KeyCode
 
-  # ioctl constants
-  EVIOCGRAB = 1074021776
-  UI_SET_EVBIT = 1074025828
-  UI_SET_KEYBIT = 1074025829
-  UI_DEV_SETUP = 1079792899
-  UI_DEV_CREATE = 21761
-  EV_KEY = 1
-  EV_SYN = 0
-  SYN_REPORT = 0
-
   EVENT_TYPE_VALUE = {
     release: 0,
     press:   1,
@@ -27,11 +18,12 @@ class Rkremap
   attr_accessor :modifiers
   attr_accessor :x11
 
-  # @param device [String]
-  def initialize(device)
-    @input = File.open(device)
-    @uinput = File.open('/dev/uinput', 'w')
-    setup_uinput
+  # @param devices [Array<String>, String]
+  def initialize(devices=nil)
+    devices = Array(devices) if devices.is_a? String
+    @inputs = devices ? devices.map{|d| Evdev.new(d)} : Evdev.devices.select(&:keyboard?)
+    raise 'Unable to detect keyboard device' if @inputs.empty?
+    @uinput = Uinput.new
     @grab = false
     @x11 = false
     @modifiers = [
@@ -53,13 +45,12 @@ class Rkremap
   def start(&block)
     @mod_state = @modifiers.map{|m| [m, false]}.to_h
     @winattr = WinAttr.new if @x11
-    @input.ioctl(EVIOCGRAB, 1) if @grab
+    @inputs.each(&:grab) if @grab
     while true
       @keys = []
-      raw = @input.sysread(24)
-      sec, usec, type, code, value = raw.unpack('Q!Q!SSl')
+      input = Evdev.select(@inputs)
+      time, type, code, value = input.read_event
       next if type != EV_KEY
-      time = Time.at(sec, usec)
       event = Event.new(time, code, value, @winattr)
       @events.each do |cond, b|
         synchronize{ b.call event } if event.match?(**cond)
@@ -100,20 +91,6 @@ class Rkremap
 
   private
 
-  def setup_uinput
-    @uinput.ioctl(UI_SET_EVBIT, EV_KEY)
-    KEY_CNT.times{|k| @uinput.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
-    @uinput.ioctl(UI_DEV_SETUP, setup)
-    @uinput.ioctl(UI_DEV_CREATE)
-  end
-
   def proc_event(code, value, event)
     if @mod_state.include?(code)
       @mod_state[code] = value != 0
@@ -130,8 +107,8 @@ class Rkremap
   # @param code [Integer]
   # @param value [Integer] 0:release / 1:press / 2:repeat
   def write_event(code, value)
-    @uinput.syswrite(['', EV_KEY, code, value].pack('a16SSl'))
-    @uinput.syswrite(['', EV_SYN, SYN_REPORT, 0].pack('a16SSl'))
+    @uinput.write_event(EV_KEY, code, value)
+    @uinput.write_event(EV_SYN, SYN_REPORT, 0)
   end
 
   class Event

metadata

@@ -1,29 +1,15 @@
 --- !ruby/object:Gem::Specification
 name: rkremap
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - TOMITA Masahiro
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-01-10 00:00:00.000000000 Z
-dependencies:
-- !ruby/object:Gem::Dependency
-  name: ffi
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '1.0'
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '1.0'
+date: 2022-01-12 00:00:00.000000000 Z
+dependencies: []
 description: key remapper
 email:
 - tommy@tmtm.org
@@ -37,6 +23,7 @@ files:
 - example/mac-shortcut.rb
 - example/tmtms.rb
 - lib/rkremap.rb
+- lib/rkremap/evdev.rb
 - lib/rkremap/keycode.rb
 - lib/rkremap/version.rb
 - lib/rkremap/winattr.rb