videoreg 0.1

data/lib/videoreg/util.rb

@@ -0,0 +1,80 @@
+module Videoreg
+  class Util
+
+    class << self
+      def which(cmd)
+        exts = ENV['PATHEXT'] ? ENV['PATHEXT'].split(';') : ['']
+        ENV['PATH'].split(File::PATH_SEPARATOR).each do |path|
+          exts.each { |ext|
+            exe = "#{path}/#{cmd}#{ext}"
+            return exe if File.executable? exe
+          }
+        end
+        nil
+      end
+
+      def proc_alive?(pid)
+        return false unless pid
+        begin
+          Process.kill(0, pid)
+          true
+        rescue Errno::ESRCH
+          false
+        end
+      end
+
+      # Extracts the connection string for the rabbitmq service from the
+      # service information provided by Cloud Foundry in an environment
+      # variable.
+      def amqp_url
+        services = JSON.parse(ENV['VCAP_SERVICES'], :symbolize_names => true)
+        url = services.values.map do |srvs|
+          srvs.map do |srv|
+            if srv[:label] =~ /^rabbitmq-/
+              srv[:credentials][:url]
+            else
+              []
+            end
+          end
+        end.flatten!.first
+      end
+
+
+      def humanize(secs)
+        [[60, :sec], [60, :min], [24, :hr], [1000, :days]].map { |count, name|
+          if secs > 0
+            secs, n = secs.divmod(count)
+            "#{n.to_i}#{name}"
+          end
+        }.compact.reverse.join(' ')
+      end
+
+
+      def udevinfo(maxcount)
+        res = []
+        maxcount.times do |dnum|
+          devpath_parts = nil
+          Open4::popen4("udevadm info --query all --name video#{dnum} | grep DEVPATH") do |pid, stdin, stdout, stderr|
+            devpath = stdout.read.match(/DEVPATH=(.*)\n$/)
+            unless devpath.nil?
+              devpath_parts = devpath[1].match(/(\/.*\/)(usb\d+)\/(\d*-\d*)((?:\/.*)?\/video4linux\/)(video(\d+))/)
+            end
+          end
+          unless devpath_parts.nil?
+            res << {
+                :prefix => devpath_parts[1],
+                :usbhub => devpath_parts[2],
+                :usbport => devpath_parts[3],
+                :postfix => devpath_parts[4],
+                :device => "/dev/#{devpath_parts[5]}",
+                :devnum => devpath_parts[6]
+            }
+          end
+        end
+        res
+      end
+
+    end
+
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,8 @@
+require 'rspec'
+require_relative '../lib/videoreg'
+Dir["#{File.dirname(__FILE__)}/../lib/**/*.rb"].each { |f| require f }
+
+RSpec.configure do |config|
+  config.color_enabled = true
+  config.formatter     = 'documentation'
+end

data/spec/videoreg/lockfile_spec.rb

@@ -0,0 +1,26 @@
+require_relative '../spec_helper'
+require 'tempfile'
+
+
+describe Videoreg::Lockfile do
+
+  it "should create lockfile " do
+
+    PID = 12345
+    tempfile = Tempfile.new('lockfile')
+    lock_path = tempfile.path
+    tempfile.unlink
+    g = Videoreg::Lockfile.new(lock_path)
+    g.lock(PID).should be_true
+    g.verify.should be_true
+    h = Videoreg::Lockfile.new(lock_path)
+    h.lock.should be_false
+    h.verify.should be_false
+    h.release.should be_false
+    g.verify.should be_true
+    g.lockcode.should == PID.to_s
+    g.release.should be_true
+    File.exists?(lock_path).should be_false
+  end
+
+end

data/spec/videoreg/registrar_spec.rb

@@ -0,0 +1,33 @@
+require_relative '../spec_helper'
+require 'tmpdir'
+
+describe Videoreg::Registrar do
+
+
+  it "should clean up old files" do
+
+    tmpdir = Pathname.new(Dir.tmpdir)
+    reg = Videoreg::Registrar.new { |c|
+      c.storage = tmpdir.to_s
+      c.store_max = 5
+    }
+
+    (reg.config.store_max + 10).times {
+      File.open(tmpdir.join("#{Time.now.to_i}.#{Time.now.usec}.avi").to_s, "w+") { |f| f.write('') }
+    }
+
+    lambda {
+      reg.clean_old_files!
+    }.should change {
+      Dir[tmpdir.join('*.avi').to_s].length
+    }.to(reg.config.store_max)
+
+
+    lambda {
+      reg.clean_old_files!
+    }.should_not change {
+      Dir[tmpdir.join('*.avi').to_s].length
+    }
+
+  end
+end

data/spec/videoreg_spec.rb

@@ -0,0 +1,16 @@
+require 'spec_helper'
+
+describe Videoreg::Registrar do
+
+  it "should be configurable" do
+
+    subject.configure do |c|
+      c.resolution = '640x480'
+      c.command = 'test command #{resolution}'
+    end
+
+    subject.resolution.should == '640x480'
+    subject.cmd.should == 'test command 640x480'
+
+  end
+end

data/videoreg-sample-config.rb

@@ -0,0 +1,38 @@
+reg {
+  device      '/dev/webcam0'
+  resolution  '640x480'
+  fps         25
+  duration    60
+  filename    '#{time}-webcam0.avi'
+  storage     '/tmp/webcam0'
+  lockfile    '/tmp/videoreg.webcam0.lock'
+  store_max   5
+}
+
+reg {
+  device      '/dev/webcam1'
+  resolution  '640x480'
+  fps         25
+  duration    60
+  filename    '#{time}-webcam1.avi'
+  storage     '/tmp/webcam1'
+  lockfile    '/tmp/videoreg.webcam1.lock'
+  store_max   3
+}
+
+reg {
+  device      '/dev/webcam2'
+  resolution  '640x480'
+  fps         25
+  duration    60
+  filename    '#{time}-webcam2.avi'
+  storage     '/tmp/webcam2'
+  lockfile    '/tmp/videoreg.webcam2.lock'
+  store_max   3
+}
+
+
+log_path    '/tmp/videoreg.log'
+pid_path    '/tmp/videoreg.pid'
+mq_queue    'com.ifree.videoreg.daemon'
+mq_host     'localhost'

data/videoreg-sample.god

@@ -0,0 +1,22 @@
+pid_path = '/tmp/videoreg.pid'
+log_path = '/tmp/videoreg.log'
+
+God.watch do |w|
+  w.name            = "videoreg"
+  w.interval        = 30.seconds
+  w.start           = "videoreg -l #{log_path} -P #{pid_path} -c videoreg-sample-config.rb"
+  w.stop            = "videoreg -l #{log_path} -P #{pid_path} -c videoreg-sample-config.rb -k"
+  w.start_grace     = 15.seconds
+  w.restart_grace   = 15.seconds
+  w.pid_file        = pid_path
+  w.keepalive
+
+  w.behavior(:clean_pid_file)
+
+  w.start_if do |start|
+    start.condition(:process_running) do |c|
+      c.interval = 5.seconds
+      c.running = false
+    end
+  end
+end

data/videoreg.gemspec

@@ -0,0 +1,26 @@
+require 'pathname'
+
+require Pathname.new(File.dirname(File.expand_path(__FILE__))).join('lib/videoreg.rb')
+
+Gem::Specification.new do |s|
+  s.name = 'videoreg'
+  s.version = Videoreg::VERSION
+  s.date = '2012-06-22'
+  s.authors = ["Ilya Sadykov"]
+  s.email = 'i.sadykov@i-free.com'
+  s.homepage = ""
+  s.summary = %q{Video registration script}
+  s.description = %q{Video Registration scripts. Allows to register continuously}
+
+  s.add_development_dependency 'rspec', '~> 2.10.0'
+  s.add_runtime_dependency 'dante'
+  s.add_runtime_dependency 'open4'
+  s.add_runtime_dependency 'god'
+  s.add_runtime_dependency 'amqp'
+  s.add_runtime_dependency 'json'
+
+  s.files = `git ls-files`.split("\n")
+  s.test_files = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables = `git ls-files -- bin/*`.split("\n").map { |f| File.basename(f) }
+  s.require_paths = %w(lib)
+end

metadata

@@ -0,0 +1,162 @@
+--- !ruby/object:Gem::Specification
+name: videoreg
+version: !ruby/object:Gem::Version
+  version: '0.1'
+  prerelease: 
+platform: ruby
+authors:
+- Ilya Sadykov
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-06-22 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.10.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.10.0
+- !ruby/object:Gem::Dependency
+  name: dante
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: open4
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: god
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: amqp
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Video Registration scripts. Allows to register continuously
+email: i.sadykov@i-free.com
+executables:
+- videoreg
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Gemfile.lock
+- README.md
+- Rakefile
+- bin/videoreg
+- dist/videoreg-0.1.gem
+- install.sh
+- lib/videoreg.rb
+- lib/videoreg/base.rb
+- lib/videoreg/config.rb
+- lib/videoreg/lockfile.rb
+- lib/videoreg/registrar.rb
+- lib/videoreg/util.rb
+- spec/spec_helper.rb
+- spec/videoreg/lockfile_spec.rb
+- spec/videoreg/registrar_spec.rb
+- spec/videoreg_spec.rb
+- videoreg-sample-config.rb
+- videoreg-sample.god
+- videoreg.gemspec
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.25
+signing_key: 
+specification_version: 3
+summary: Video registration script
+test_files: []