procon_bypass_man 0.1.11 → 0.1.15

data/lib/procon_bypass_man/remote_pbm_action/base_action.rb

@@ -0,0 +1,53 @@
+module ProconBypassMan
+  module RemotePbmAction
+    class ActionUnexpectedError < StandardError; end
+    class NeedPbmVersionError < ActionUnexpectedError; end
+
+    class BaseAction
+      attr_accessor :pbm_job_uuid
+
+      # @param [String] pbm_job_uuid
+      def initialize(pbm_job_uuid: )
+        self.pbm_job_uuid = pbm_job_uuid
+      end
+
+      # @return [void]
+      def action_content(_args)
+        raise NotImplementedError, nil
+      end
+
+      # @param [Hash] args
+      # @return [void]
+      def run!(job_args: )
+        before_action_callback
+        action_content(args: job_args)
+        after_action_callback
+      rescue => e
+        be_failed
+        ProconBypassMan::SendErrorCommand.execute(error: e)
+      end
+
+      private
+
+      # @return [void]
+      def before_action_callback; end
+      # @return [void]
+      def after_action_callback; end
+
+      # @return [void]
+      def be_failed
+        ProconBypassMan::UpdateRemotePbmActionStatusCommand.new(pbm_job_uuid: pbm_job_uuid).execute(to_status: :failed)
+      end
+
+      # @return [void]
+      def be_in_progress
+        ProconBypassMan::UpdateRemotePbmActionStatusCommand.new(pbm_job_uuid: pbm_job_uuid).execute(to_status: :in_progress)
+      end
+
+      # @return [void]
+      def be_processed
+        ProconBypassMan::UpdateRemotePbmActionStatusCommand.new(pbm_job_uuid: pbm_job_uuid).execute(to_status: :processed)
+      end
+    end
+  end
+end

data/lib/procon_bypass_man/remote_pbm_action/change_pbm_version_action.rb

@@ -0,0 +1,25 @@
+module ProconBypassMan
+  module RemotePbmAction
+    class ChangePbmVersionAction < BaseAction
+
+      def action_content(args: )
+        require "pbmenv"
+        ProconBypassMan.logger.info "execute ChangePbmVersionAction!"
+        pbm_version = args["pbm_version"] or raise(ProconBypassMan::RemotePbmAction::NeedPbmVersionError, "pbm_versionが必要です, #{args.inspect}")
+        Pbmenv.install(pbm_version)
+        Pbmenv.use(pbm_version)
+        `reboot` # symlinkの参照先が変わるのでrebootする必要がある
+      end
+
+      private
+
+      def before_action_callback
+        be_processed
+      end
+
+      def after_action_callback
+        # no-op
+      end
+    end
+  end
+end

data/lib/procon_bypass_man/remote_pbm_action/lib/update_remote_pbm_action_status_command.rb

@@ -0,0 +1,24 @@
+module ProconBypassMan
+  class UpdateRemotePbmActionStatusCommand
+    # @param [String] pbm_job_uuid
+    def initialize(pbm_job_uuid: )
+      @pbm_job_uuid = pbm_job_uuid
+    end
+
+    # @param [String] to_status
+    # @return [void]
+    def execute(to_status: )
+      ProconBypassMan::UpdateRemotePbmActionStatusHttpClient.new(
+        path: path,
+        server_pool: ProconBypassMan.config.server_pool,
+      ).put(to_status: to_status)
+    end
+
+    private
+
+    # @return [String]
+    def path
+      "/api/devices/#{ProconBypassMan.device_id}/pbm_jobs/#{@pbm_job_uuid}"
+    end
+  end
+end

data/lib/procon_bypass_man/remote_pbm_action/reboot_os_action.rb

@@ -0,0 +1,21 @@
+module ProconBypassMan
+  module RemotePbmAction
+    class RebootOsAction < BaseAction
+
+      def action_content(_args)
+        ProconBypassMan.logger.info "execute RebootOsAction!"
+        `reboot`
+      end
+
+      private
+
+      def before_action_callback
+        be_processed
+      end
+
+      def after_action_callback
+        # no-op
+      end
+    end
+  end
+end

data/lib/procon_bypass_man/remote_pbm_action/restore_pbm_setting.rb

@@ -0,0 +1,28 @@
+module ProconBypassMan
+  module RemotePbmAction
+    class RestorePbmSettingAction < BaseAction
+
+      def action_content(args: )
+        require "pbmenv"
+        ProconBypassMan.logger.info "execute RestorePbmSettingAction!"
+        setting = args["setting"] or raise(ProconBypassMan::RemotePbmAction::NeedPbmVersionError, "settingが必要です, #{args.inspect}")
+        File.write(
+          ProconBypassMan::ButtonsSettingConfiguration.instance.setting_path,
+          setting.to_yaml,
+        )
+        ProconBypassMan.hot_reload!
+      end
+
+      private
+
+      def before_action_callback
+        be_in_progress
+      end
+
+      def after_action_callback
+        be_processed
+      end
+    end
+  end
+end
+

data/lib/procon_bypass_man/remote_pbm_action/stop_pbm_action.rb

@@ -0,0 +1,21 @@
+module ProconBypassMan
+  module RemotePbmAction
+    class StopPbmAction < BaseAction
+
+      def action_content(_args)
+        ProconBypassMan.logger.info "execute StopPbmAction!"
+        Process.kill("TERM", ProconBypassMan.pid)
+      end
+
+      private
+
+      def before_action_callback
+        be_processed
+      end
+
+      def after_action_callback
+        # no-op
+      end
+    end
+  end
+end

data/lib/procon_bypass_man/remote_pbm_action.rb

@@ -0,0 +1,32 @@
+module ProconBypassMan
+  module RemotePbmAction
+    require "procon_bypass_man/remote_pbm_action/base_action"
+    require "procon_bypass_man/remote_pbm_action/change_pbm_version_action"
+    require "procon_bypass_man/remote_pbm_action/reboot_os_action"
+    require "procon_bypass_man/remote_pbm_action/stop_pbm_action"
+    require "procon_bypass_man/remote_pbm_action/restore_pbm_setting.rb"
+    require "procon_bypass_man/remote_pbm_action/lib/update_remote_pbm_action_status_command"
+
+    ACTION_CHANGE_PBM_VERSION = "change_pbm_version"
+    ACTION_REBOOT_OS = "reboot_os"
+    ACTION_STOP_PBM = "stop_pbm"
+    ACTION_RESTORE_SETTING = "restore_pbm_setting"
+
+    ACTIONS = [
+      ACTION_CHANGE_PBM_VERSION,
+      ACTION_REBOOT_OS,
+      ACTION_STOP_PBM,
+      ACTION_RESTORE_SETTING,
+    ]
+
+    STATUS_FAILED = :failed
+    STATUS_IN_PROGRESS = :in_progress
+    STATUS_PROCESSED = :processed
+
+    ACTION_STATUSES = [
+      STATUS_FAILED,
+      STATUS_IN_PROGRESS,
+      STATUS_PROCESSED,
+    ]
+  end
+end

data/lib/procon_bypass_man/runner.rb

@@ -1,15 +1,19 @@
 require_relative "io_monitor"
-require_relative "uptime"
 require_relative "boot_message"
-require_relative "background/report_thread"
 
 class ProconBypassMan::Runner
   class InterruptForRestart < StandardError; end
 
-  def run
-    first_negotiation
-    print_booted_message
+  include ProconBypassMan::SignalHandler
+
+  def initialize(gadget: , procon: )
+    @gadget = gadget
+    @procon = procon
 
+    ProconBypassMan::PrintBootMessageCommand.execute
+  end
+
+  def run
     self_read, self_write = IO.pipe
     %w(TERM INT USR1 USR2).each do |sig|
       begin
@@ -17,13 +21,14 @@ class ProconBypassMan::Runner
           self_write.puts(sig)
         end
       rescue ArgumentError
-        ProconBypassMan.logger.error("Signal #{sig} not supported")
+        ProconBypassMan::SendErrorCommand.execute(error: "Signal #{sig} not supported")
       end
     end
 
     loop do
       $will_terminate_token = false
-      main_loop_pid = fork { main_loop }
+      # NOTE メインプロセスではThreadをいくつか起動しているので念のためパフォーマンスを優先するためにforkしていく
+      main_loop_pid = Kernel.fork { ProconBypassMan::BypassCommand.new(gadget: @gadget, procon: @procon).execute }
 
       begin
         while(readable_io = IO.select([self_read]))
@@ -37,9 +42,9 @@ class ProconBypassMan::Runner
         ProconBypassMan.logger.info("Reloading config file")
         begin
           ProconBypassMan::ButtonsSettingConfiguration::Loader.reload_setting
-          puts "設定ファイルの再読み込みができました"
+          ProconBypassMan::SendReloadConfigEventCommand.execute
         rescue ProconBypassMan::CouldNotLoadConfigError
-          ProconBypassMan.logger.error "設定ファイルが不正です。再読み込みができませんでした"
+          ProconBypassMan::SendErrorCommand.execute(error: "設定ファイルが不正です。再読み込みができませんでした")
         end
         ProconBypassMan.logger.info("バイパス処理を再開します")
       rescue Interrupt
@@ -54,110 +59,4 @@ class ProconBypassMan::Runner
       end
     end
   end
-
-  private
-
-  def main_loop
-    ProconBypassMan::IOMonitor.start!
-    ProconBypassMan::Background::Reporter.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
-      timer = ProconBypassMan::Timer.new(timeout: Time.now + 10)
-      bypass = ProconBypassMan::Bypass.new(gadget: @gadget, procon: @procon, monitor: monitor1)
-      loop do
-        break if $will_terminate_token
-        timer.throw_if_timeout!
-        bypass.send_gadget_to_procon!
-        sleep(0.005)
-      rescue ProconBypassMan::Timer::Timeout
-        ProconBypassMan.logger.info "10秒経過したのでThread1を終了します"
-        puts "10秒経過したのでThread1を終了します"
-        break
-      rescue Errno::EIO, Errno::ENODEV, Errno::EPROTO, IOError => e
-        ProconBypassMan.logger.error "Proconが切断されました.終了処理を開始します"
-        Process.kill "TERM", Process.ppid
-      rescue Errno::ETIMEDOUT => e
-        # TODO まれにこれが発生する. 再接続したい
-        ProconBypassMan::ErrorReporter.report(body: e)
-        ProconBypassMan.logger.error "Switchとの切断されました.終了処理を開始します"
-        Process.kill "TERM", Process.ppid
-      end
-      ProconBypassMan.logger.info "Thread1を終了します"
-    end
-
-    # procon => gadget
-    # シビア
-    ProconBypassMan.logger.info "Thread2を起動します"
-    t2 = Thread.new do
-      bypass = ProconBypassMan::Bypass.new(gadget: @gadget, procon: @procon, monitor: monitor2)
-      loop do
-        break if $will_terminate_token
-        bypass.send_procon_to_gadget!
-      rescue EOFError => e
-        ProconBypassMan.logger.error "Proconと通信ができませんでした.終了処理を開始します"
-        Process.kill "TERM", Process.ppid
-      rescue Errno::EIO, Errno::ENODEV, Errno::EPROTO, IOError => e
-        ProconBypassMan.logger.error "Proconが切断されました。終了処理を開始します"
-        Process.kill "TERM", Process.ppid
-      end
-      ProconBypassMan.logger.info "Thread2を終了します"
-    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
-    @gadget, @procon = ProconBypassMan::DeviceConnector.connect
-  rescue ProconBypassMan::Timer::Timeout
-    ::ProconBypassMan.logger.error "デバイスとの通信でタイムアウトが起きて接続ができませんでした。"
-    @gadget&.close
-    @procon&.close
-    raise ::ProconBypassMan::EternalConnectionError
-  end
-
-  def handle_signal(sig)
-    ProconBypassMan.logger.info "#{$$}で#{sig}を受け取りました"
-    case sig
-    when 'USR2'
-      raise InterruptForRestart
-    when 'INT', 'TERM'
-      raise Interrupt
-    end
-  end
-
-  # @return [void]
-  def print_booted_message
-    message = ProconBypassMan::BootMessage.new
-    ProconBypassMan.logger.info(message.to_s)
-    Thread.new { ProconBypassMan::Reporter.report(body: message.to_hash) }
-    puts message.to_s
-  end
 end

data/lib/procon_bypass_man/scheduler.rb

@@ -0,0 +1,92 @@
+module ProconBypassMan
+  class Scheduler
+    class Schedule
+      attr_accessor :klass, :args, :interval, :next_enqueue_at
+
+      # @param [any] klass
+      # @param [Array] args
+      # @param [Integer] interval
+      def initialize(klass: , args: , interval: )
+        self.klass = klass
+        self.args = args
+        self.interval = interval
+        self.next_enqueue_at = Time.now
+      end
+
+      # @return [void]
+      def enqueue
+        klass.perform_async(*unwrap_args(args))
+        set_next_enqueue_at!
+      end
+
+      # @return [boolean]
+      def past_interval?
+        next_enqueue_at < Time.now
+      end
+
+      private
+
+      # @return [void]
+      def set_next_enqueue_at!
+        self.next_enqueue_at = Time.now + interval
+      end
+
+      # @param [Array] args
+      # @return [void]
+      def unwrap_args(args)
+        args.map do |arg|
+          case arg
+          when Proc
+            arg.call
+          else
+            arg
+          end
+        end
+      end
+    end
+
+    # @return [Hash]
+    def self.schedules
+      @@schedules
+    end
+
+    @@schedules = {}
+
+    def self.start!
+      register_schedules
+
+      @@thread = Thread.start do
+        loop do
+          schedules.each do |_key, schedule|
+            if schedule.past_interval?
+              schedule.enqueue
+            end
+          end
+          sleep 10
+        end
+      end
+    end
+
+    def self.register_schedules
+      register(
+        schedule: Schedule.new(
+          klass: ProconBypassMan::FetchAndRunRemotePbmActionJob,
+          args: [],
+          interval: 3,
+        )
+      )
+      register(
+        schedule: Schedule.new(
+          klass: ProconBypassMan::SyncDeviceStatsJob,
+          args: [->{ ProconBypassMan::DeviceStatus.current }],
+          interval: 60,
+        )
+      )
+    end
+
+    # @param [Schedule] schedule
+    def self.register(schedule: )
+      schedules[schedule.klass] = schedule
+    end
+  end
+end

data/lib/procon_bypass_man/support/compress_array.rb

@@ -0,0 +1,56 @@
+module ProconBypassMan
+  class CompressArray
+    class CompressibleValue
+      # @params [String] prev
+      # @params [String] current
+      def initialize(prev, current)
+        @prev = prev
+        @current = current
+      end
+
+      # @return [Boolean]
+      def compress?
+        @prev.include?(@current)
+      end
+
+      # @return [String]
+      def to_s_with_mark
+        if /^(.+) \* (\d+)/ =~ @prev
+          value = $1
+          count = $2
+          return "#{value} * #{count.to_i + 1}"
+        end
+        if /^(.+)/ =~ @prev
+          value = $1
+          return "#{value} * 1"
+        end
+      end
+    end
+
+    def initialize(array)
+      @array = array
+    end
+
+    # @return [Array<String>]
+    def compress
+      previous_value = nil
+      @array.reduce([]) do |acc, item|
+        if previous_value.nil?
+          acc << item
+          previous_value = item
+          next acc
+        end
+
+        if CompressibleValue.new(previous_value, item).compress?
+          registered_value = acc.pop
+          acc << CompressibleValue.new(registered_value, item).to_s_with_mark
+        else
+          acc << item
+        end
+
+        previous_value = item
+        next acc
+      end
+    end
+  end
+end

data/lib/procon_bypass_man/support/http_client.rb

@@ -0,0 +1,102 @@
+module ProconBypassMan
+  class HttpClient
+    class HttpRequest
+      class Get
+        def self.request!(uri: )
+          http = Net::HTTP.new(uri.host, uri.port)
+          http.use_ssl = uri.scheme === "https"
+          http.get(uri.path, { "Content-Type" => "application/json" })
+        end
+      end
+
+      class Post
+        def self.request!(uri: , request_body: {})
+          http = Net::HTTP.new(uri.host, uri.port)
+          http.use_ssl = uri.scheme === "https"
+          http.post(uri.path, request_body.to_json, { "Content-Type" => "application/json" })
+        end
+      end
+
+      class Put
+        def self.request!(uri: , request_body: {})
+          http = Net::HTTP.new(uri.host, uri.port)
+          http.use_ssl = uri.scheme === "https"
+          http.put(uri.path, request_body.to_json, { "Content-Type" => "application/json" })
+        end
+      end
+    end
+
+    def initialize(path: , server_pool: , retry_on_connection_error: false)
+      @server_pool = server_pool
+      @uri = URI.parse("#{server_pool.server}#{path}")
+      @retry_on_connection_error = retry_on_connection_error
+    end
+
+    def get
+      handle_request do
+        response = HttpRequest::Get.request!(
+          uri: @uri,
+        )
+        break process_response(response)
+      end
+    end
+
+    def post(request_body: )
+      handle_request do
+        body = {}.merge!(request_body)
+        response = HttpRequest::Post.request!(
+          uri: @uri,
+          request_body: body,
+        )
+        break process_response(response)
+      end
+    end
+
+    def put(request_body: )
+      handle_request do
+        body = {
+          hostname: `hostname`.chomp,
+        }.merge!(request_body)
+        response = HttpRequest::Put.request!(
+          uri: @uri,
+          request_body: body,
+        )
+        break process_response(response)
+      end
+    end
+
+    private
+
+    def process_response(response)
+      case response.code
+      when /^200/
+        begin
+          return JSON.parse(response.body)
+        rescue JSON::ParserError
+          return response.body
+        end
+      else
+        @server_pool.next!
+        ProconBypassMan.logger.error("#{@uri}から200以外(#{response.code})が帰ってきました. #{response.body}")
+      end
+    end
+
+    def handle_request
+      raise "need block" unless block_given?
+      if @server_pool.server.nil?
+        ProconBypassMan.logger.info('送信先が未設定なのでスキップしました')
+        return
+      end
+      return yield
+    rescue SocketError => e
+      ProconBypassMan.logger.error(e)
+      if @retry_on_connection_error
+        sleep(10)
+        retry
+      end
+    rescue => e
+      puts e
+      ProconBypassMan.logger.error(e)
+    end
+  end
+end

data/lib/procon_bypass_man/support/report_http_client.rb

@@ -0,0 +1,19 @@
+module ProconBypassMan
+  class ReportHttpClient < HttpClient
+    def post(body: , event_type: )
+      if body.is_a?(Hash)
+        b = body
+      else
+        b = { text: body }
+      end
+
+      super(request_body: {
+        session_id: ProconBypassMan.session_id,
+        device_id: ProconBypassMan.device_id,
+        hostname: `hostname`.chomp,
+        event_type: event_type,
+        body: b,
+      })
+    end
+  end
+end

data/lib/procon_bypass_man/{timer.rb → support/safe_timeout.rb}

@@ -1,5 +1,5 @@
 module ProconBypassMan
-  class Timer
+  class SafeTimeout
     class Timeout < StandardError; end
 
     # 5秒後がタイムアウト

data/lib/procon_bypass_man/support/send_device_stats_http_client.rb

@@ -0,0 +1,9 @@
+module ProconBypassMan
+  class SendDeviceStatsHttpClient < HttpClient
+    def post(status: , pbm_session_id: )
+      super(request_body: {
+        body: { status: status, pbm_session_id: pbm_session_id },
+      })
+    end
+  end
+end