procon_bypass_man 0.1.2

data/lib/procon_bypass_man/processor.rb

@@ -0,0 +1,17 @@
+class ProconBypassMan::Processor
+  # @return [String] binary
+  def initialize(binary)
+    @binary = binary
+  end
+
+  # @return [String] 加工後の入力データ
+  def process
+    unless @binary[0] == "\x30".b
+      return @binary
+    end
+
+    procon = ProconBypassMan::Procon.new(@binary)
+    procon.apply!
+    procon.to_binary
+  end
+end

data/lib/procon_bypass_man/procon.rb

@@ -0,0 +1,129 @@
+class ProconBypassMan::Procon
+  require "procon_bypass_man/procon/mode_registry"
+  require "procon_bypass_man/procon/macro_registry"
+  require "procon_bypass_man/procon/layer_changeable"
+  require "procon_bypass_man/procon/button_collection"
+  require "procon_bypass_man/procon/pressed_button_helper"
+  require "procon_bypass_man/procon/user_operation"
+
+  attr_accessor :user_operation
+
+  def self.reset_cvar!
+    @@status = {
+      buttons: {},
+      current_layer_key: :up,
+      ongoing_macro: MacroRegistry.load(:null),
+      ongoing_mode: ModeRegistry.load(:manual),
+    }
+  end
+  def self.reset!; reset_cvar!; end
+  reset!
+
+  def initialize(binary)
+    self.user_operation = ProconBypassMan::Procon::UserOperation.new(binary.dup)
+  end
+
+  def status; @@status[:buttons]; end
+  def ongoing_macro; @@status[:ongoing_macro]; end
+  def ongoing_mode; @@status[:ongoing_mode]; end
+  def current_layer_key; @@status[:current_layer_key]; end
+
+  def current_layer
+    ProconBypassMan::Configuration.instance.layers[current_layer_key]
+  end
+
+  def apply!
+    if user_operation.change_layer?
+      @@status[:current_layer_key] = user_operation.next_layer_key if user_operation.pressed_next_layer?
+      user_operation.set_no_action!
+      return
+    end
+
+    if ongoing_macro.finished?
+      current_layer.macros.each do |macro_name, options|
+        if options[:if_pressed].all? { |b| user_operation.pressed_button?(b) }
+          @@status[:ongoing_macro] = MacroRegistry.load(macro_name)
+        end
+      end
+    end
+
+    case current_layer.mode
+    when :manual
+      @@status[:ongoing_mode] = ModeRegistry.load(:manual)
+      current_layer.flip_buttons.each do |button, options|
+        unless options[:if_pressed]
+          status[button] = !status[button]
+          next
+        end
+
+        if options[:if_pressed] && options[:if_pressed].all? { |b| user_operation.pressed_button?(b) }
+          status[button] = !status[button]
+        else
+          status[button] = false
+        end
+      end
+    else
+      unless ongoing_mode.name == current_layer.mode
+        @@status[:ongoing_mode] = ProconBypassMan::Procon::ModeRegistry.load(current_layer.mode)
+      end
+      if(binary = ongoing_mode.next_binary)
+        self.user_operation.merge(target_binary: binary)
+      end
+      return
+    end
+
+    status
+  end
+
+  # @return [String<binary>]
+  def to_binary
+    if ongoing_mode.name != :manual
+      return user_operation.binary
+    end
+
+    if ongoing_macro.ongoing?
+      step = ongoing_macro.next_step or return(user_operation.binary)
+      user_operation.press_button_only(step)
+      return user_operation.binary
+    end
+
+    current_layer.flip_buttons.each do |button, options|
+      # 何もしないで常に連打
+      if !options[:if_pressed] && status[button]
+        user_operation.press_button(button) unless user_operation.pressed_button?(button)
+        next
+      end
+
+      # 押している時だけ連打
+      if options[:if_pressed] && options[:if_pressed].all? { |b| user_operation.pressed_button?(b) }
+        if !status[button]
+          user_operation.unpress_button(button)
+        end
+        if options[:force_neutral] && user_operation.pressed_button?(options[:force_neutral])
+          button = options[:force_neutral]
+          user_operation.unpress_button(button)
+        end
+      end
+    end
+
+    current_layer.remaps.each do |from_button, to_button|
+      if user_operation.pressed_button?(from_button)
+        user_operation.unpress_button(from_button)
+        # TODO 2重でpressしないようにしたい
+        user_operation.press_button(to_button) unless user_operation.pressed_button?(to_button)
+      end
+    end
+
+    user_operation.binary
+  end
+
+  private
+
+  def method_missing(name)
+    if name.to_s =~ /\Apressed_[a-z]+\?\z/
+      user_operation.public_send(name)
+    else
+      super
+    end
+  end
+end

data/lib/procon_bypass_man/procon/button_collection.rb

@@ -0,0 +1,40 @@
+class ProconBypassMan::Procon::ButtonCollection
+  class Button
+    attr_accessor :byte_position, :bit_position
+    def initialize(key)
+      b = BUTTONS_MAP[key] or raise("undefined button")
+      self.byte_position = b[:byte_position]
+      self.bit_position = b[:bit_position]
+    end
+  end
+
+  #3)  ZR	R	SR(right)	SL(right)	A	B	X	Y
+  #4)  Grip	(none)	Cap	Home	ThumbL	ThumbR	+	-
+  #5)  ZL	L	SL(left)	SR(left)	Left	Right	Up	Down
+  #6)  analog[0]
+  #7)  analog[1]
+  #8)  analog[2]
+  #9)  analog[3]
+  #a)  analog[4]
+  #b)  analog[5]
+  BYTES_MAP = {
+    0 => nil,
+    1 => nil,
+    2 => nil,
+    3 => [:zr, :r, :sr, :sl, :a, :b, :x, :y],
+    4 => [:grip, :_undefined_key, :cap, :home, :thumbl, :thumbr, :plus, :minus],
+    5 => [:zl, :l, :sl, :sr, :left, :right, :up, :down],
+  }
+
+  BUTTONS_MAP = BYTES_MAP.reduce({}) { |acc, value|
+    next acc if value[1].nil?
+    value[1].reverse.each.with_index do |button, index|
+      acc[button] = { byte_position: value[0], bit_position: index }
+    end
+    acc
+  }
+
+  def self.load(button_key)
+    Button.new(button_key)
+  end
+end

data/lib/procon_bypass_man/procon/data.rb

@@ -0,0 +1,5 @@
+class ProconBypassMan::Procon
+  module Data
+    NO_ACTION = "30f28100800078c77448287509550274ff131029001b0022005a0271ff191028001e00210064027cff1410280020002100000000000000000000000000000000".freeze
+  end
+end

data/lib/procon_bypass_man/procon/layer_changeable.rb

@@ -0,0 +1,28 @@
+module ProconBypassMan::Procon::LayerChangeable
+  def next_layer_key
+    case
+    when pressed_up?
+      :up
+    when pressed_right?
+      :right
+    when pressed_left?
+      :left
+    when pressed_down?
+      :down
+    else
+      ProconBypassMan.logger.warn("next_layer_key is unknown")
+      :up
+    end
+  end
+
+  def change_layer?
+    if ProconBypassMan::Configuration.instance.prefix_keys.empty?
+      raise "prefix_keysが未設定です"
+    end
+    ProconBypassMan::Configuration.instance.prefix_keys.map { |b| pressed_button?(b) }.all?
+  end
+
+  def pressed_next_layer?
+    change_layer? && (pressed_up? || pressed_right? || pressed_left? || pressed_down?)
+  end
+end

data/lib/procon_bypass_man/procon/macro_registry.rb

@@ -0,0 +1,48 @@
+class ProconBypassMan::Procon::MacroRegistry
+  class Macro
+    attr_accessor :name, :steps
+
+    def initialize(name: , steps: )
+      self.name = name
+      self.steps = steps
+    end
+
+    def next_step
+      steps.shift
+    end
+
+    def finished?
+      steps.empty?
+    end
+
+    def ongoing?
+      !finished?
+    end
+  end
+
+  PRESETS = {
+    null: [],
+  }
+
+  def self.install_plugin(klass)
+    if plugins[klass.name]
+      raise "すでに登録済みです"
+    end
+    plugins[klass.name] = klass.steps
+  end
+
+  def self.load(name)
+    steps = PRESETS[name] || plugins[name] || raise("unknown macro")
+    Macro.new(name: name, steps: steps.dup)
+  end
+
+  def self.reset!
+    ProconBypassMan::Configuration.instance.macro_plugins = {}
+  end
+
+  def self.plugins
+    ProconBypassMan::Configuration.instance.macro_plugins
+  end
+
+  reset!
+end

data/lib/procon_bypass_man/procon/mode_registry.rb

@@ -0,0 +1,46 @@
+class ProconBypassMan::Procon::ModeRegistry
+  class Mode
+    attr_accessor :name, :binaries, :source_binaries
+
+    def initialize(name: , binaries: )
+      self.name = name
+      self.binaries = binaries
+      self.source_binaries = binaries.dup
+    end
+
+    def next_binary
+      binary = binaries.shift
+      unless binary
+        self.binaries = source_binaries.dup
+        return binaries.shift
+      end
+      return binary
+    end
+  end
+
+  PRESETS = {
+    manual: [],
+  }
+
+  def self.install_plugin(klass)
+    if plugins[klass.name]
+      raise "すでに登録済みです"
+    end
+    plugins[klass.name] = klass.binaries
+  end
+
+  def self.load(name)
+    b = PRESETS[name] || plugins[name] || raise("unknown mode")
+    Mode.new(name: name, binaries: b.dup)
+  end
+
+  def self.reset!
+    ProconBypassMan::Configuration.instance.mode_plugins = {}
+  end
+
+  def self.plugins
+    ProconBypassMan::Configuration.instance.mode_plugins
+  end
+
+  reset!
+end

data/lib/procon_bypass_man/procon/pressed_button_helper.rb

@@ -0,0 +1,25 @@
+module ProconBypassMan::Procon::PushedButtonHelper
+  module Static
+    def pressed_button?(button)
+      binary[
+        ::ProconBypassMan::Procon::ButtonCollection.load(button).byte_position
+      ].unpack("H*").first.to_i(16).to_s(2).reverse[
+        ::ProconBypassMan::Procon::ButtonCollection.load(button).bit_position
+      ] == '1'
+    end
+  end
+
+  module Dynamic
+    @@compiled = false
+    def compile_if_not_compile_yet!
+      unless @@compiled
+        ::ProconBypassMan::Procon::ButtonCollection::BUTTONS_MAP.each do |button, value|
+          define_method "pressed_#{button}?" do
+            pressed_button?(button)
+          end
+        end
+      end
+      @@compiled = true
+    end
+  end
+end

data/lib/procon_bypass_man/procon/user_operation.rb

@@ -0,0 +1,72 @@
+class ProconBypassMan::Procon
+  class UserOperation
+    include LayerChangeable
+    include PushedButtonHelper::Static
+    extend PushedButtonHelper::Dynamic
+
+    attr_reader :binary
+
+    def initialize(binary)
+      self.class.compile_if_not_compile_yet!
+      unless binary.encoding.name == ASCII_ENCODING
+        raise "おかしいです"
+      end
+      @binary = binary
+    end
+
+    ZERO_BIT = ["0"].pack("H*").freeze
+    ASCII_ENCODING = "ASCII-8BIT"
+
+    # @depilicate
+    def binary=(binary)
+      unless binary.encoding.name == ASCII_ENCODING
+        raise "おかしいです"
+      end
+      @binary = binary
+    end
+
+    def set_no_action!
+      binary[3] = ZERO_BIT
+      binary[4] = ZERO_BIT
+      binary[5] = ZERO_BIT
+    end
+
+    def unpress_button(button)
+      byte_position = ButtonCollection.load(button).byte_position
+      value = binary[byte_position].unpack("H*").first.to_i(16) - 2**ButtonCollection.load(button).bit_position
+      binary[byte_position] = ["%02X" % value.to_s].pack("H*")
+    end
+
+    def press_button(button)
+      byte_position = ButtonCollection.load(button).byte_position
+      value = binary[byte_position].unpack("H*").first.to_i(16) + 2**ButtonCollection.load(button).bit_position
+      binary[byte_position] = ["%02X" % value.to_s].pack("H*")
+    end
+
+    def press_button_only(button)
+      [ProconBypassMan::Procon::Data::NO_ACTION.dup].pack("H*").tap do |no_action_binary|
+        ButtonCollection.load(button).byte_position
+        byte_position = ButtonCollection.load(button).byte_position
+        value = 2**ButtonCollection.load(button).bit_position
+        no_action_binary[byte_position] = ["%02X" % value.to_s].pack("H*")
+        binary[3] = no_action_binary[3]
+        binary[4] = no_action_binary[4]
+        binary[5] = no_action_binary[5]
+      end
+    end
+
+    def merge(target_binary: )
+      tb = [target_binary].pack("H*")
+      binary[3] = tb[3]
+      binary[4] = tb[4]
+      binary[5] = tb[5]
+      binary[6] = tb[6]
+      binary[7] = tb[7]
+      binary[8] = tb[8]
+      binary[9] = tb[9]
+      binary[10] = tb[10]
+      binary[11] = tb[11]
+      self.binary
+    end
+  end
+end

data/lib/procon_bypass_man/runner.rb

@@ -0,0 +1,171 @@
+require_relative "io_monitor"
+
+class ProconBypassMan::Runner
+  class InterruptForRestart < StandardError; end
+  class InterruptForCouldNotConnect < StandardError; end
+
+  def initialize(gadget: , procon: )
+    @gadget = gadget
+    @procon = procon
+
+    $will_interval_0_0_0_5 = 0
+    $will_interval_1_6 = 0
+  end
+
+  def run
+    first_negotiation
+    $is_stable = false
+
+    self_read, self_write = IO.pipe
+    %w(TERM INT USR1 USR2).each do |sig|
+      begin
+        trap sig do
+          self_write.puts(sig)
+        end
+      rescue ArgumentError
+        ProconBypassMan.logger.info("Signal #{sig} not supported")
+      end
+    end
+
+    FileUtils.mkdir_p "tmp"
+    File.write "tmp/pid", $$
+
+    loop do
+      $will_terminate_token = false
+      main_loop_pid = fork { main_loop }
+
+      begin
+        while readable_io = IO.select([self_read])
+          signal = readable_io.first[0].gets.strip
+          handle_signal(signal)
+        end
+      rescue InterruptForCouldNotConnect
+        $will_terminate_token = true
+        Process.kill("TERM", main_loop_pid)
+        Process.wait
+        raise ProconBypassMan::CouldNotConnectDeviceError
+      rescue InterruptForRestart
+        $will_terminate_token = true
+        Process.kill("TERM", main_loop_pid)
+        Process.wait
+        ProconBypassMan.logger.info("Reloading config file")
+        begin
+          ProconBypassMan::Configuration::Loader.reload_setting
+          puts "設定ファイルの再読み込みができました"
+        rescue ProconBypassMan::CouldNotLoadConfigError
+          ProconBypassMan.logger.error "設定ファイルが不正です。再読み込みができませんでした"
+        end
+        ProconBypassMan.logger.info("バイパス処理を再開します")
+      rescue Interrupt
+        $will_terminate_token = true
+        Process.kill("TERM", main_loop_pid)
+        Process.wait
+        @gadget&.close
+        @procon&.close
+        exit 1
+      end
+    end
+  end
+
+  private
+
+  def main_loop
+    # TODO 接続確立完了をswitchを読み取るようにして、この暫定で接続完了sleepを消す
+    Thread.new do
+      sleep(10)
+      $will_interval_0_0_0_5 = 0.005
+      $will_interval_1_6 = 1.6
+      $is_stable = true
+    end
+
+    ProconBypassMan::IOMonitor.start!
+    # gadget => procon
+    # 遅くていい
+    monitor1 = ProconBypassMan::IOMonitor.new(label: "switch -> procon")
+    monitor2 = ProconBypassMan::IOMonitor.new(label: "procon -> switch")
+    ProconBypassMan.logger.info "Thread1を起動します"
+    t1 = Thread.new do
+      bypass = ProconBypassMan::Bypass.new(gadget: @gadget, procon: @procon, monitor: monitor1)
+      begin
+        loop do
+          break if $will_terminate_token
+          bypass.send_gadget_to_procon!
+        rescue Errno::EIO, Errno::ENODEV, Errno::EPROTO, IOError => e
+          ProconBypassMan.logger.error "Proconが切断されました.終了処理を開始します"
+          Process.kill "TERM", Process.ppid
+        end
+        ProconBypassMan.logger.info "Thread1を終了します"
+      end
+    end
+
+    # procon => gadget
+    # シビア
+    ProconBypassMan.logger.info "Thread2を起動します"
+    t2 = Thread.new do
+      bypass = ProconBypassMan::Bypass.new(gadget: @gadget, procon: @procon, monitor: monitor2)
+      begin
+        loop do
+          break if $will_terminate_token
+          bypass.send_procon_to_gadget!
+        rescue Errno::EIO, Errno::ENODEV, Errno::EPROTO, IOError => e
+          ProconBypassMan.logger.error "Proconが切断されました.終了処理を開始します"
+          Process.kill "TERM", Process.ppid
+        end
+        ProconBypassMan.logger.info "Thread2を終了します"
+      end
+    end
+
+    self_read, self_write = IO.pipe
+    %w(TERM INT).each do |sig|
+      begin
+        trap sig do
+          self_write.puts(sig)
+        end
+      rescue ArgumentError
+        puts "プロセスでSignal #{sig} not supported"
+      end
+    end
+
+    ProconBypassMan.logger.info "子プロセスでgraceful shutdownの準備ができました"
+    begin
+      while readable_io = IO.select([self_read])
+        signal = readable_io.first[0].gets.strip
+        handle_signal(signal)
+      end
+    rescue Interrupt
+      $will_terminate_token = true
+      [t1, t2].each(&:join)
+      @gadget&.close
+      @procon&.close
+      exit 1
+    end
+  end
+
+  def first_negotiation
+    loop do
+      begin
+        input = @gadget.read_nonblock(128)
+        ProconBypassMan.logger.debug { ">>> #{input.unpack("H*")}" }
+        @procon.write_nonblock(input)
+        if input[0] == "\x80".b && input[1] == "\x01".b
+          ProconBypassMan.logger.info("first negotiation is over")
+          break
+        end
+        break if $will_terminate_token
+      rescue IO::EAGAINWaitReadable
+      end
+    end
+  end
+
+  def handle_signal(sig)
+    ProconBypassMan.logger.info "#{$$}で#{sig}を受け取りました"
+    case sig
+    when 'USR1'
+      raise InterruptForCouldNotConnect
+    when 'USR2'
+      raise InterruptForRestart
+    when 'INT', 'TERM'
+      raise Interrupt
+    end
+  end
+end