tamashii-agent 0.1.7

data/lib/tamashii/agent/device/fake_buzzer.rb

@@ -0,0 +1,30 @@
+require 'tamashii/agent/common'
+module Tamashii
+  module Agent
+    module Device
+      class FakeBuzzer
+        include Common::Loggable
+
+        def initialize
+          logger.debug "Initialized"
+        end
+
+        def play_ok
+          logger.debug "Played: OK"
+        end
+
+        def play_no
+          logger.debug "Played: No"
+        end
+
+        def play_error
+          logger.debug "Played: Error"
+        end
+
+        def stop
+          logger.debug "Stopped"
+        end
+      end
+    end
+  end
+end

data/lib/tamashii/agent/device/fake_card_reader.rb

@@ -0,0 +1,37 @@
+require 'tamashii/agent/common'
+module Tamashii
+  module Agent
+    module Device
+      class FakeCardReader
+        include Common::Loggable
+
+        def initialize(*args)
+          logger.debug "Initialized"
+          @last_time = Time.now
+        end
+
+        def picc_request(*args)
+          if Time.now - @last_time > 2
+            @last_time = Time.now
+            if rand > 0.5
+              logger.debug "Fake Card Generated"
+              return true
+            else
+              return false
+            end
+          else
+            return false
+          end
+        end
+
+        def picc_select(*args)
+          [Array.new(4){ rand(256)}, "sak"]
+        end
+
+        def picc_halt(*args)
+        end
+      end
+    end
+  end
+end
+

data/lib/tamashii/agent/device/pi_buzzer.rb

@@ -0,0 +1,70 @@
+require 'pi_piper'
+module Tamashii
+  module Agent
+    module Device
+      class PIBuzzer
+        SHORT_PLAY_TIME = 0.2
+        LONG_PLAY_TIME = 0.5
+        REPEAT_INTERVAL = 0.1
+        LOW_FREQ = 0.7
+
+        attr_reader :pwm
+
+        def initialize
+          @pwm = PiPiper::Pwm.new pin: 18 #, mode: :markspace
+          @pwm.off
+          @pwm.value = 1.0
+        end
+
+        def play(value = 1.0)
+          @pwm.value = value
+          @pwm.on
+        end
+
+        def stop
+          @pwm.off
+        end
+
+        def play_short(value = 1.0)
+          play_time(SHORT_PLAY_TIME,value)
+        end
+
+        def play_long(value = 1.0)
+          play_time(LONG_PLAY_TIME,value)
+        end
+
+        def play_time(time, value = 1.0)
+          play(value)
+          sleep time
+          stop
+        end
+
+        def play_repeat_short(repeat, repeat_interval = REPEAT_INTERVAL)
+          repeat.times do
+            play_short
+            sleep repeat_interval
+          end
+        end
+
+        def play_repeat_long(repeat, repeat_interval = REPEAT_INTERVAL)
+          repeat.times do
+            play_long
+            sleep repeat_interval
+          end
+        end
+
+        def play_ok
+          play_repeat_short(1)
+        end
+
+        def play_no
+          play_repeat_short(3)
+        end
+
+        def play_error
+          play_repeat_long(3)
+        end
+      end
+    end
+  end
+end

data/lib/tamashii/agent/handler.rb

@@ -0,0 +1,10 @@
+require 'tamashii/agent/handler/request_pool_response'
+require 'tamashii/agent/handler/system'
+require 'tamashii/agent/handler/buzzer'
+
+module Tamashii
+  module Agent
+    module Handler
+    end
+  end
+end

data/lib/tamashii/agent/handler/base.rb

@@ -0,0 +1,15 @@
+require 'tamashii/common'
+
+module Tamashii
+  module Agent
+    module Handler
+      class Base < Tamashii::Handler
+        def initialize(*args, &block)
+          super(*args, &block)
+          @connection = self.env[:connection]
+          @master = @connection.master
+        end
+      end
+    end
+  end
+end

data/lib/tamashii/agent/handler/buzzer.rb

@@ -0,0 +1,14 @@
+require 'tamashii/agent/common'
+require 'tamashii/agent/handler/base'
+
+module Tamashii
+  module Agent
+    module Handler
+      class Buzzer < Base
+        def resolve(data)
+          @master.send_event(EVENT_BEEP, data)
+        end
+      end
+    end
+  end
+end

data/lib/tamashii/agent/handler/request_pool_response.rb

@@ -0,0 +1,14 @@
+require 'tamashii/agent/handler/base'
+require 'tamashii/agent/request_pool'
+
+module Tamashii
+  module Agent
+    module Handler
+      class RequestPoolResponse < Base
+        def resolve(data)
+          @connection.request_pool.add_response(RequestPool::Response.new(self.type, data))
+        end
+      end
+    end
+  end
+end

data/lib/tamashii/agent/handler/system.rb

@@ -0,0 +1,14 @@
+require 'tamashii/agent/common'
+require 'tamashii/agent/handler/base'
+
+module Tamashii
+  module Agent
+    module Handler
+      class System < Base
+        def resolve(data)
+          @master.send_event(EVENT_SYSTEM_COMMAND, type.to_s)
+        end
+      end
+    end
+  end
+end

data/lib/tamashii/agent/master.rb

@@ -0,0 +1,113 @@
+require 'tamashii/agent/connection'
+require 'tamashii/agent/buzzer'
+require 'tamashii/agent/card_reader'
+
+require 'thread'
+
+module Tamashii
+  module Agent
+    class Master < Component
+
+      attr_reader :serial_number
+
+      def initialize(host, port)
+        super()
+        logger.info "Starting Tamashii::Agent #{Tamashii::Agent::VERSION} in #{Config.env} mode"
+        @host = host
+        @port = port
+        @serial_number = get_serial_number
+        logger.info "Serial number: #{@serial_number}"
+        create_components
+      end
+
+      def get_serial_number
+        File.open("/proc/cpuinfo") do |f|
+          content = f.read
+          if content =~ /Serial\s*:\s*(\w+)/
+            return $1
+          end
+        end
+        # Cannot get serial number
+        if Config.env == "test"
+          return "TEST_PID_#{Process.pid}"
+        else
+          return nil
+        end
+      end
+
+      def create_components
+        @components = {}
+        @components[:connection] = create_component(Connection, self, @host, @port)
+        @components[:buzzer] = create_component(Buzzer)
+        @components[:card_reader] = create_component(CardReader, self)
+      end
+
+      def create_component(class_name, *args)
+        c = class_name.new(*args)
+        logger.info "Starting component: #{class_name}"
+        yield c if block_given?
+        c.run
+        c
+      end
+
+      # override
+      def process_event(ev_type, ev_body)
+        super
+        case ev_type
+        when EVENT_SYSTEM_COMMAND
+          logger.info "System command code: #{ev_body}"
+          case ev_body.to_i
+          when Tamashii::Type::REBOOT
+            system_reboot
+          when Tamashii::Type::POWEROFF
+            system_poweroff
+          when Tamashii::Type::RESTART
+            system_restart
+          when Tamashii::Type::UPDATE
+            system_update
+          end
+        when EVENT_CONNECTION_NOT_READY
+          broadcast_event(EVENT_BEEP, "error")
+        else
+          broadcast_event(ev_type, ev_body)
+        end
+      end
+
+      def system_reboot
+        logger.info "Rebooting..."
+        system("reboot &")
+      end
+
+      def system_poweroff
+        logger.info "Powering Off..."
+        system("poweroff &")
+      end
+
+      def system_restart
+        logger.info "Restarting..."
+        system("systemctl restart tamashii-agent.service &")
+      end
+
+      def system_update
+        logger.info "Updating..."
+        system("gem update tamashii-agent")
+        system_restart
+      end
+
+      # override
+      def stop
+        super
+        @components.each_value do |c|
+          c.stop
+        end
+      end
+
+
+      def broadcast_event(ev_type, ev_body)
+        @components.each_value do |c|
+          c.send_event(ev_type, ev_body)
+        end
+      end
+    end
+  end
+end

data/lib/tamashii/agent/request_pool.rb

@@ -0,0 +1,80 @@
+require 'tamashii/agent/common'
+require 'tamashii/agent/request_pool/request'
+require 'tamashii/agent/request_pool/response'
+
+
+module Tamashii
+  module Agent
+    class RequestPool
+      include Common::Loggable
+      def initialize
+        @pool = {}
+        @handlers = {}
+      end
+
+      def set_handler(sym, method)
+         @handlers[sym] = method      
+      end
+
+      def call_handler(sym, *args)
+        if handle?(sym)
+          @handlers[sym].call(*args)
+        else
+          logger.warn "WARN: un-handled event: #{sym}"
+        end
+      end
+
+      def handle?(sym)
+        @handlers.has_key? sym
+      end
+
+      def add_request(req, timedout = 3)
+        @pool[req.id] = {req: req, timestamp: Time.now, timedout: timedout}
+        try_send_request(req)
+      end
+
+      def add_response(res)
+        # find the same id
+        req_data = @pool[res.id]
+        if req_data
+          @pool.delete(res.id)
+          call_handler(:request_meet, req_data[:req], res)
+        else
+          # unmatched response
+          # discard
+          logger.warn "WARN: un-matched response (id=#{res.id}): #{res.inspect}"
+        end
+      end
+
+      def update
+        process_pending
+        check_timedout
+      end
+      
+      def check_timedout
+        now = Time.now
+        @pool.each do |id, req_data|
+          if now - req_data[:timestamp] >= req_data[:timedout]
+            # timedout
+            @pool.delete(id)
+            call_handler(:request_timedout, req_data[:req])
+          end
+        end
+      end
+
+      def process_pending
+        @pool.each_value do |data|
+          try_send_request(data[:req]) unless data[:req].sent?
+        end
+      end
+
+      def try_send_request(req)
+        if handle?(:send_request)
+          req.sent! if call_handler(:send_request, req)
+        end
+      end
+    end
+  end
+end
+
+

data/lib/tamashii/agent/request_pool/request.rb

@@ -0,0 +1,38 @@
+require 'json'
+module Tamashii
+  module Agent
+    class RequestPool
+      class Request
+        attr_accessor :id
+        attr_accessor :ev_type
+        attr_accessor :ev_body
+        attr_accessor :state
+
+        STATE_PENDING = :pending
+        STATE_SENT = :sent
+
+        def initialize(ev_type, ev_body, id)
+          @ev_type = ev_type
+          @ev_body = ev_body
+          @id = id
+          @state = STATE_PENDING
+        end
+
+        def wrap_body
+          {
+            id: @id,
+            ev_body: @ev_body
+          }.to_json
+        end
+
+        def sent!
+          @state = STATE_SENT
+        end
+
+        def sent?
+          @state == STATE_SENT
+        end
+      end
+    end
+  end
+end