bstick 0.2.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 860d1e7835ee172ba587158584502eddb9b9c48b
+  data.tar.gz: 123ef2faa52bb9c4a128159cdfb032c0a2f009ee
+SHA512:
+  metadata.gz: c4370a38df2b414f4e929dacc388f13f14554ded32aa9091f842606b662ec680ae350480c52195e8ebc76b45faa9f2e4b55d4e96854c0c617e6ae8a1425eafa5
+  data.tar.gz: 933faedc1f7815b69fee2d5966d68bacd407fa730ac57444b788c4216db243d7b156ebafcceaf6ea8fae0f545b84d7da3373b50cb78c49f7f1dc32ee076651a5

data/bin/bstick

@@ -0,0 +1,27 @@
+#!/usr/bin/env ruby
+
+if !ARGV[0]
+  puts 'no argument'
+  exit 0
+end
+
+def write(state)
+  file = File.expand_path("/var/bstick/led.state")
+  File.truncate(file, 0)
+  File.write(file, state, 0)
+end
+
+case ARGV[0]
+when 'on', 'off', 'blink', 'random' then
+  write(ARGV[0])
+when 'ping', 'alarm' then
+  write("#{ARGV[0]} #{ARGV[1]}")
+when 'run' then
+  write('blink')
+  system(ARGV.drop(1).join(' '))
+  write('off')
+else
+  puts 'invalid argument'
+  exit 0
+end
+exit 1

data/bin/bstick_server

@@ -0,0 +1,13 @@
+#!/usr/bin/env ruby
+
+require 'bstick'
+require 'daemons'
+
+if ARGV[0] && ARGV[0] == '-v'
+  puts Bstick::VERSION
+  exit 1
+end
+
+Daemons.run_proc('server.rb') do
+  Bstick::Server.new.run
+end

data/lib/blinkstick.rb

@@ -0,0 +1,186 @@
+#https://github.com/arvydas/blinkstick-ruby
+
+require "libusb"
+require "color"
+
+class BlinkStick
+  @@VENDOR_ID = 0X20A0
+  @@PRODUCT_ID = 0x41E5
+
+  def open(device = nil)
+    @@usb ||= LIBUSB::Context.new
+
+    if (device)
+      @device = device
+    else
+      @device = @@usb.devices(:idVendor => @@VENDOR_ID, :idProduct => @@PRODUCT_ID).first
+    end
+
+    @handle = @device.open
+  end
+
+  def self.find_all
+    @@usb ||= LIBUSB::Context.new
+
+    result = []
+
+    @@usb.devices(:idVendor => @@VENDOR_ID, :idProduct => @@PRODUCT_ID).each { | device |
+      b = BlinkStick.new
+      b.open(device)
+
+      result.push(b)
+    }
+
+    result
+  end
+
+  def self.find_by_serial(serial)
+    @@usb ||= LIBUSB::Context.new
+
+    @@usb.devices(:idVendor => @@VENDOR_ID, :idProduct => @@PRODUCT_ID).each { | device |
+      if device.serial_number == serial
+        b = BlinkStick.new
+        b.open(device)
+        return b
+      end
+    }
+
+    nil
+  end
+
+  def color=(value)
+    @handle.control_transfer(:bmRequestType => 0x20, :bRequest => 0x9, :wValue => 0x1, :wIndex => 0x0000, :dataOut => 1.chr + value.red.to_i.chr + value.green.to_i.chr + value.blue.to_i.chr)
+  end
+
+  def color
+    result = @handle.control_transfer(:bmRequestType => 0x80 | 0x20, :bRequest => 0x1, :wValue => 0x1, :wIndex => 0x0000, :dataIn => 4)
+    Color::RGB.new(result[1].ord, result[2].ord, result[3].ord)
+  end
+
+  def set_color(channel, index, value)
+    attempts = 0
+    while attempts < 5
+      attempts += 1
+
+      begin
+        @handle.control_transfer(:bmRequestType => 0x20,
+                                 :bRequest => 0x9,
+                                 :wValue => 0x5,
+                                 :wIndex => 0x0000,
+                                 :dataOut => 1.chr + channel.to_i.chr + index.to_i.chr + value.red.to_i.chr + value.green.to_i.chr + value.blue.to_i.chr)
+        break
+      rescue
+        if attempts == 5
+          raise
+        end
+      end
+    end
+  end
+
+  def set_colors(channel, data)
+    report_id = 9
+    max_leds = 64
+
+    if data.size <= 8 * 3
+        max_leds = 8
+        report_id = 6
+    elsif data.size <= 16 * 3
+        max_leds = 16
+        report_id = 7
+    elsif data.size <= 32 * 3
+        max_leds = 32
+        report_id = 8
+    elsif data.size <= 64 * 3
+        max_leds = 64
+        report_id = 9
+    end
+
+    report = report_id.chr + channel.to_i.chr
+
+    (0..max_leds * 3 - 1).each do | i |
+      if data.size > i
+        report += data[i].to_i.chr
+      else
+        report += 0.chr
+      end
+    end
+
+    #Debug code
+    #puts report.unpack('U'*report.length).collect {|x| x.to_s 16}.join(" ")
+
+    attempts = 0
+    while attempts < 5
+      attempts += 1
+
+      begin
+        @handle.control_transfer(:bmRequestType => 0x20,
+                                 :bRequest => 0x9,
+                                 :wValue => report_id,
+                                 :wIndex => 0,
+                                 :dataOut => report)
+        break
+      rescue
+        if attempts == 5
+          raise
+        end
+      end
+    end
+  end
+
+  def off
+    self.color = Color::RGB.new(0, 0, 0)
+  end
+
+  def get_info_block(id)
+    bytes = @handle.control_transfer(:bmRequestType => 0x80 | 0x20, :bRequest => 0x1, :wValue => id + 1, :wIndex => 0x0000, :dataIn => 33)
+
+    result = ""
+    for i in 1..(bytes.length-1)
+      if i == "\x00"
+        break
+      end
+      result += bytes[i]
+    end
+
+    result
+  end
+
+  def set_info_block(id, data)
+    data = (id + 1).chr + data
+    data = data + 0.chr while data.length < 33
+    @handle.control_transfer(:bmRequestType => 0x20, :bRequest => 0x9, :wValue => id + 1, :wIndex => 0x0000, :dataOut => data)
+  end
+
+  def random_color
+    r = Random.new
+    self.color = Color::RGB.new(r.rand(255), r.rand(255), r.rand(255))
+  end
+
+  def serial
+    @device.serial_number
+  end
+
+  def manufacturer
+    @device.manufacturer
+  end
+
+  def description
+    @device.product
+  end
+
+  def info_block1
+    get_info_block(1)
+  end
+
+  def info_block1=(value)
+    set_info_block(1, value)
+  end
+
+  def info_block2
+    get_info_block(2)
+  end
+
+  def info_block2=(value)
+    set_info_block(2, value)
+  end
+end

data/lib/bstick.rb

@@ -0,0 +1,146 @@
+require "bstick/version"
+require "net/ping"
+require "logger"
+require_relative "blinkstick.rb"
+
+module Bstick
+  class Server
+    def initialize
+      init
+    end
+
+    def init
+      system 'sudo mkdir -p /var/bstick'
+      system 'sudo chmod 777 /var/bstick'
+      system 'sudo touch /var/bstick/led.state'
+      system 'sudo chmod 777 /var/bstick/led.state'
+      @logger = ::Logger.new('/var/bstick/bstick.log', 1, 1024000)
+      @state  = ''
+      @values = nil
+      @b      = nil
+    end
+
+    def stick
+      @b ||= BlinkStick.find_all.first
+    end
+
+    def black
+      @black ||= Color::RGB.from_html("#000000")
+    end
+
+    def green
+      @green = Color::RGB.new(0x00, 0xFF, 0x00)
+    end
+
+    def red
+      @red = Color::RGB.new(0xFF, 0x00, 0x00)
+    end
+
+    def read_state
+      @old_state = @state
+      file    = File.expand_path("/var/bstick/led.state", __FILE__)
+      @file   = File.open(file, "a+")
+      File.chmod(0777, file)
+      @file.seek(0)
+      @state = @file.readline.strip rescue 'off'
+      @file.close
+    end
+
+    def set_colors(color_1, color_2)
+      if stick
+        if (color_1 != @color_1 || color_2 != @color_2)
+          @logger.info @state
+          stick.set_color(0, 0, color_1)
+          stick.set_color(0, 1, color_2)
+          @color_1 = color_1
+          @color_2 = color_2
+        end
+      else
+        @logger.error 'no stick found'
+        sleep 10
+        init
+      end
+    end
+
+    def off_state
+      set_colors(black, black)
+    end
+
+    def on_state
+      set_colors(green, green)
+    end
+
+    def blink_state
+      @values = { green: 0, direction: 20 } if !@values
+      color_1 = Color::RGB.new(0xFF - @values[:green], @values[:green], 0x00)
+      color_2 = Color::RGB.new(@values[:green], 0xFF - @values[:green], 0x00)
+      @values[:green] += @values[:direction]
+      if(@values[:green] < 0 || @values[:green] > 0xFF)
+        @values[:direction] *= -1
+      end
+      set_colors(color_1, color_2)
+    end
+
+    def ping_state
+      @wait += 1
+      return if @wait < 15 * 33
+      @wait = 0
+      url = @state.split(' ')[1] || "http://www.undefine.io"
+      ping = Net::Ping::HTTP.new(url)
+      @logger.info "ping #{url} #{ping.ping?}"
+      if result = ping.ping?
+        set_colors(green, green)
+      else
+        set_colors(red, red)
+      end
+    end
+
+    def random_state
+      @wait += 1
+      return if @wait < 0.5 * 33
+      @wait = 0
+      color_1 = Color::RGB.new(rand(0..0xFF), rand(0..0xFF), rand(0..0xFF))
+      color_2 = Color::RGB.new(rand(0..0xFF), rand(0..0xFF), rand(0..0xFF))
+      set_colors(color_1, color_2)
+    end
+
+    def alarm_state
+      time = Time.parse(@state.split(' ')[1])
+      if Time.now < time
+        on_state
+      else
+        random_state
+      end
+    end
+
+    def handle_state
+      if @old_state != @state
+        @values = nil
+        @wait = 100000000000000
+      end
+      case @state.split(' ').first
+      when 'off'   then off_state
+      when 'random'then random_state
+      when 'on'    then on_state
+      when 'blink' then blink_state
+      when 'ping'  then ping_state
+      when 'alarm' then alarm_state
+      end
+    end
+
+    def run
+      @logger.info 'started'
+      loop do
+        begin
+          read_state
+          handle_state
+          sleep 1/33.0
+        rescue => e
+          @logger.error e.message
+          sleep 10
+          init
+        end
+      end
+    end
+  end
+end

data/lib/bstick/version.rb

@@ -0,0 +1,3 @@
+module Bstick
+  VERSION = "0.2.0"
+end

metadata

@@ -0,0 +1,149 @@
+--- !ruby/object:Gem::Specification
+name: bstick
+version: !ruby/object:Gem::Version
+  version: 0.2.0
+platform: ruby
+authors:
+- bseverac
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-09-29 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+- !ruby/object:Gem::Dependency
+  name: color
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.8'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.8'
+- !ruby/object:Gem::Dependency
+  name: libusb
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.5'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.5'
+- !ruby/object:Gem::Dependency
+  name: net-ping
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: daemons
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.2'
+description: ''
+email:
+- bseverac@gmail.com
+executables:
+- bstick
+- bstick_server
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/bstick
+- bin/bstick_server
+- lib/blinkstick.rb
+- lib/bstick.rb
+- lib/bstick/version.rb
+homepage: ''
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5.1
+signing_key: 
+specification_version: 4
+summary: Blinkstick server watching file state and utilities
+test_files: []