videoreg 0.1

data/.gitignore

@@ -0,0 +1,10 @@
+.idea
+*.iml
+coverage
+.bundle
+tmp
+old
+chromedriver.log
+atlassian-ide-plugin.xml
+//*.gem
+*.log

data/Gemfile

@@ -0,0 +1,12 @@
+source :rubygems
+
+gem 'dante'
+gem 'open4'
+gem 'god'
+gem 'amqp'
+gem 'json'
+
+group :test do
+  gem 'rspec'
+  gem 'simplecov', :require => false
+end

data/Gemfile.lock

@@ -0,0 +1,42 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    amq-client (0.9.3)
+      amq-protocol (>= 0.9.1)
+      eventmachine
+    amq-protocol (0.9.3)
+    amqp (0.9.6)
+      amq-client (~> 0.9.3)
+      amq-protocol (>= 0.9.1)
+      eventmachine
+    dante (0.1.5)
+    diff-lcs (1.1.3)
+    eventmachine (0.12.10)
+    god (0.12.1)
+    json (1.7.3)
+    multi_json (1.3.6)
+    open4 (1.3.0)
+    rspec (2.10.0)
+      rspec-core (~> 2.10.0)
+      rspec-expectations (~> 2.10.0)
+      rspec-mocks (~> 2.10.0)
+    rspec-core (2.10.1)
+    rspec-expectations (2.10.0)
+      diff-lcs (~> 1.1.3)
+    rspec-mocks (2.10.1)
+    simplecov (0.6.4)
+      multi_json (~> 1.0)
+      simplecov-html (~> 0.5.3)
+    simplecov-html (0.5.3)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  amqp
+  dante
+  god
+  json
+  open4
+  rspec
+  simplecov

data/README.md

@@ -0,0 +1,205 @@
+## Videoreg: script for continuous video recording from the plugged webcams 
+
+### Prerequisites: you will need to install the required libraries
+
+RabbitMQ and Ruby are required to use this script.
+
+```bash
+sudo echo "deb http://www.rabbitmq.com/debian/ testing main" >> /etc/apt/sources.list
+sudo wget http://www.rabbitmq.com/rabbitmq-signing-key-public.asc
+sudo apt-key add rabbitmq-signing-key-public.asc
+sudo apt-get update
+apt-get install -y mencoder ffmpeg qc-usb-utils v4l-utils rabbitmq-server ruby1.9.1 rubygems ruby1.9.1-dev
+```
+
+### Then write the configuration file according to your setup:
+
+```ruby
+	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'
+```
+
+### After the installation you are able to manage the process using 'videoreg' command:
+
+```
+	$ videoreg -h
+	Usage: videoreg [options]
+	    -c, --config CONFIG              Use the specified config
+	    -d, --device DEVICE              Run only the specified device
+	    -H, --halt DEVICE                Halt (SIGKILL) recording process
+	    -P, --pause DEVICE               Pause (SIGSTOP) recording process
+	    -R, --resume DEVICE              Resume (SIGCONT) recording process
+	    -r, --recover DEVICE             Recover the recording process
+	    -p, --pid PID                    Set the PID file for the capture process
+	    -l, --log LOGFILE                Set the logfile for daemon
+	    -e, --ensure                     Check the state of the daemon
+	    -C, --clear                      Clear the state (remove lock-files)
+	    -k, --kill                       Kills the capture processes
+	    -v, --version                    Show the version
+	    -h, --help                       Show this help message
+	    -U, --udev                       Generate udev rules for current config
+	    -I, --info                       Show info about plugged devices
+```
+### To get the current daemon status:
+
+```
+	$  videoreg -c config.rb -e
+	Running command 'ensure' for device(s): 'all'...
+	DAEMON:		     [RUNNING]
+	Uptime:		     4sec
+	/dev/webcam0: 	 [RUNNING]
+	/dev/webcam1: 	 [RUNNING]
+	/dev/webcam2: 	 [NO DEVICE]
+```
+### To run the daemon:
+
+```
+	$ videoreg -c config.rb
+	Daemon has started successfully
+```
+### To stop the daemon:
+
+```
+	$ videoreg -c config.rb --k
+	Stopped PID: 3870 at /tmp/videoreg.pid
+```
+### To stop recording from the specified device:
+
+```
+	$ videoreg -c config.rb -H /dev/webcam0
+	Running command 'halt' for device(s): '/dev/webcam0'...
+	I, [2012-07-03T14:00:35.802428 #8115]  INFO -- : Publish message to RabbitMQ 'HALT' with arg '/dev/webcam0' to 'com.ifree.videoreg.daemon'...
+	I, [2012-07-03T14:00:36.347704 #8115]  INFO -- : Disconnecting from RabbitMQ...
+
+	$ videoreg -c config.rb -e
+	Running command 'ensure' for device(s): 'all'...
+	DAEMON:		     [RUNNING]
+	Uptime:		     1min 18sec
+	/dev/webcam0: 	 [NOT RUNNING]
+	/dev/webcam1: 	 [RUNNING]
+	/dev/webcam2: 	 [NO DEVICE]
+```
+### To run recording from the specified device:
+
+```
+	$ videoreg -c config.rb -r /dev/webcam0
+	Running command 'recover' for device(s): '/dev/webcam0'...
+	I, [2012-07-03T14:02:12.375352 #8193]  INFO -- : Publish message to RabbitMQ 'RECOVER' with arg '/dev/webcam0' to 'com.ifree.videoreg.daemon'...
+	I, [2012-07-03T14:02:12.920787 #8193]  INFO -- : Disconnecting from RabbitMQ...
+
+	$ videoreg -c config.rb -e
+	Running command 'ensure' for device(s): 'all'...
+	DAEMON:		     [RUNNING]
+	Uptime:		     2min 34sec
+	/dev/webcam0: 	 [RUNNING]
+	/dev/webcam1: 	 [RUNNING]
+	/dev/webcam2: 	 [NO DEVICE]
+```
+### To pause the device:
+
+```
+	$ videoreg -c config.rb -P /dev/webcam0
+	Running command 'pause' for device(s): '/dev/webcam0'...
+	I, [2012-07-03T14:03:28.840693 #8254]  INFO -- : Publish message to RabbitMQ 'PAUSE' with arg '/dev/webcam0' to 'com.ifree.videoreg.daemon'...
+	I, [2012-07-03T14:03:29.386111 #8254]  INFO -- : Disconnecting from RabbitMQ...
+
+	$ videoreg -c config.rb -e
+	Running command 'ensure' for device(s): 'all'...
+	DAEMON:		     [RUNNING]
+	Uptime:		     3min 50sec
+	/dev/webcam0: 	 [PAUSED]
+	/dev/webcam1: 	 [RUNNING]
+	/dev/webcam2: 	 [NO DEVICE]
+```
+### To resume the device:
+
+```
+	$ videoreg -c config.rb -R /dev/webcam0
+	Running command 'resume' for device(s): '/dev/webcam0'...
+	I, [2012-07-03T14:04:34.870130 #8308]  INFO -- : Publish message to RabbitMQ 'RESUME' with arg '/dev/webcam0' to 'com.ifree.videoreg.daemon'...
+	I, [2012-07-03T14:04:35.415852 #8308]  INFO -- : Disconnecting from RabbitMQ...
+
+	$ videoreg -c config.rb -e
+	Running command 'ensure' for device(s): 'all'...
+	DAEMON:		     [RUNNING]
+	Uptime:		     4min 58sec
+	/dev/webcam0: 	 [RUNNING]
+	/dev/webcam1: 	 [RUNNING]
+	/dev/webcam2: 	 [NO DEVICE]
+```
+### Get the information about the plugged webcams:
+
+```
+	$ videoreg -I
+	usb1[1-1] --> /dev/webcam0
+	usb2[2-1] --> /dev/webcam1
+```
+### To generate the udev rules file (create the user-friendly symlinks to the webcam devices):
+
+```
+	$ videoreg -U
+	Writing the rules to /etc/udev/rules.d/50-udev-videoreg.rules file:
+	            BUS=="usb" ACTION=="add" DEVPATH=="/devices/pci0000:00/0000:00:1a.0/usb1/1-1/*/video4linux/video*" SYMLINK+="webcam0" GROUP="video"
+	            BUS=="usb" ACTION=="add" DEVPATH=="/devices/pci0000:00/0000:00:1d.0/usb2/2-1/*/video4linux/video*" SYMLINK+="webcam1" GROUP="video"
+	File has been written successfully! Now restart udev (/etc/init.d/udev restart) and replug the webcams!
+```
+
+# Changelog
+
+## 03.07.2012
+### Current status
+* Реализация правил udev (добавление симлинков на девайсы по определённым правилам)
+
+## 29.06.2012
+### Current status
+* Проверка наличия всех девайсов (камер)
+* По команде прервать запись с определенной камеры и начать заново (это нужно чтобы вытащить кадр из текущего кусочка записи)
+* "Ленивая" инициализация заданной камеры: пауза/запуск процесса записи
+
+## 28.06.2012
+### Current status
+* Запуск видеозаписи
+    * Параметры для всех камер хранятся в одном конфиге (разрешение, фпс, длительность каждого файла)
+    * Запись по кускам (по 10минут в один файл например)
+    * Слежение
+
+### TODO
+* Состояние системы (свободное место на винте, время работы с момента последней перезагрузки и тп) – тоже в конфиге как-то хранить
+* Слежение за свободным местом на диске (например записи с камер 1 и 2 должны хранится 12 часов, но все вместе не более 15Гб)
+* Алерты об ошибках в Приложение1
+* Запуск Приложения3 – при появлении файлов с определенной камеры надо запускать нашу программку (Приложение2), потом файлик удалять (или не удалять – параметр в конфиге)

data/Rakefile

@@ -0,0 +1,34 @@
+require 'rspec/core/rake_task'
+require_relative 'lib/videoreg'
+
+task :default => :help
+
+desc "Build gem"
+task :install do
+  %x[gem build videoreg.gemspec]
+  %x[sudo gem install #{Dir["*.gem"].sort.last}]
+end
+
+desc "Run specs"
+task :spec do
+  RSpec::Core::RakeTask.new(:spec) do |t|
+    t.pattern = './spec/**/*_spec.rb'
+  end
+end
+
+desc "Show help menu"
+task :help do
+  puts "Available rake tasks: "
+  puts "rake spec - Run specs and calculate coverage"
+end
+
+
+desc "Capture the video from the device"
+task :videoreg, :device, :duration, :storage do |t, args|
+  Videoreg::Registrar.new do |c|
+    c.device = args[:device]
+    c.duration = args[:duration]
+    c.storage = args[:storage]
+  end.continuous
+end
+

data/bin/videoreg

@@ -0,0 +1,109 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'ostruct'
+require 'optparse'
+require 'dante'
+require_relative '../lib/videoreg'
+
+
+def colorize(text, color)
+  "\e[#{color}m#{text}\e[0m"
+end
+
+def red(text)
+  colorize(text, 31)
+end
+
+def green(text)
+  colorize(text, 32)
+end
+
+def blue(text)
+  colorize(text, 34)
+end
+
+def running_state(is_running)
+  is_running ? green("RUNNING") : red("NOT RUNNING")
+end
+
+
+MAX_DEVICES_COUNT = 10
+
+ARGV.options do |parser|
+  Videoreg.class_eval {
+    parser.banner = "Usage: #{File.basename($0)} [options]"
+    parser.on('-c', '--config CONFIG', 'Use the specified config') { |s| opt.script = s }
+    parser.on('-d', '--device DEVICE', 'Run only the specified device') { |d| opt.device = d }
+    parser.on('-H', '--halt DEVICE', 'Halt (SIGKILL) recording process') { |d| opt.action= :halt; opt.device = d }
+    parser.on('-P', '--pause DEVICE', 'Pause (SIGSTOP) recording process') { |d| opt.action= :pause; opt.device= d }
+    parser.on('-R', '--resume DEVICE', 'Resume (SIGCONT) recording process') { |d| opt.action= :resume; opt.device= d }
+    parser.on('-r', '--recover DEVICE', 'Recover the recording process') { |d| opt.action= :recover; opt.device = d }
+    parser.on('-p', '--pid PID', 'Set the PID file for the capture process') { |pid| opt.pid_path = pid }
+    parser.on('-l', '--log LOGFILE', 'Set the logfile for daemon') { |log| opt.log_path = log }
+    parser.on('-e', '--ensure', 'Check the state of the daemon') { opt.action = :ensure }
+    parser.on('-C', '--clear', 'Clear the state (remove lock-files)') { opt.action = :reset }
+    parser.on('-k', '--kill', 'Kills the capture processes') { opt.action = :kill }
+    parser.on('-v', '--version', 'Show the version') { puts "#{Videoreg.version_info}"; exit }
+    parser.on('-h', '--help', 'Show this help message') { puts parser; exit }
+    parser.on('-U', '--udev', 'Generate udev rules for current config') {
+      opt.action = :info
+      config = Videoreg::Util.udevinfo(MAX_DEVICES_COUNT).each_with_index.map do |info, index|
+        <<-RULE
+            BUS=="usb" ACTION=="add" DEVPATH=="#{info[:prefix]}#{info[:usbhub]}/#{info[:usbport]}/*/video4linux/video*" SYMLINK+="#{Videoreg::DEV_SYMLINK}#{index}" GROUP="video"
+        RULE
+      end.join
+      puts "Writing the rules to #{green(Videoreg::UDEV_RULES_FILE)} file:"
+      puts config
+      begin
+        File.open(Videoreg::UDEV_RULES_FILE, "w+") do |f|
+          f.puts(config)
+        end
+        puts green("File has been written successfully! Now restart udev (/etc/init.d/udev restart) and replug the webcams!")
+      rescue => e
+        puts red("Failed to write file: #{e.message}. Please do this manually!")
+      end
+    }
+    parser.on('-I', '--info', 'Show info about plugged devices') {
+      opt.action = :info
+      Videoreg::Util.udevinfo(MAX_DEVICES_COUNT).each_with_index do |info|
+        puts "#{green(info[:usbhub])}[#{blue(info[:usbport])}] --> #{blue("/dev/#{Videoreg::DEV_SYMLINK}#{info[:devnum]}")}"
+      end
+    }
+    parser.parse!
+
+    case
+      when opt.action == :info
+        # do nothing
+      when opt.script then
+        unless File.exists?(opt.script)
+          puts "File #{opt.script} does not exist!"
+          exit
+        end
+        begin
+          eval(File.open(File.expand_path(opt.script)).read)
+          res = run(opt.device, opt.action)
+
+          case opt.action
+            when :ensure
+              puts "#{blue("DAEMON")}:\t\t [#{running_state(res[0][:daemon_running?])}]"
+              puts "Uptime:\t\t #{Videoreg::Util.humanize(res[0][:uptime])}" if res[0][:daemon_running?]
+              res[1..-1].each do |status|
+                proc_status = if status[:device_exists?] then
+                                status[:paused?] ? red("PAUSED") : running_state(status[:process_alive?])
+                              else
+                                red("NO DEVICE")
+                              end
+                puts "#{status[:device]}: \t [#{proc_status}]"
+              end
+            else
+          end
+        rescue => e
+          puts "An error has occured during config processing: #{e.message} \n #{e.backtrace.join("\n")}"
+        end
+      else
+        puts parser
+    end
+  }
+end
+

data/install.sh

@@ -0,0 +1,39 @@
+#!/bin/bash
+CURDIR=`pwd`
+cd /tmp
+if grep -q "deb http://www.rabbitmq.com/debian/ testing main" /etc/apt/sources.list
+then
+    echo "It seems you are installing env second time..."
+else
+    echo "deb http://www.rabbitmq.com/debian/ testing main" >> /etc/apt/sources.list
+    wget http://www.rabbitmq.com/rabbitmq-signing-key-public.asc
+    apt-key add rabbitmq-signing-key-public.asc
+    apt-get update
+fi
+apt-get install -y mencoder ffmpeg qc-usb-utils v4l-utils rabbitmq-server ruby1.9.1 rubygems ruby1.9.1-dev
+ln -sf /usr/bin/ruby1.9.1 /usr/bin/ruby
+ln -sf /usr/bin/rake1.9.1 /usr/bin/rake
+ln -sf /usr/bin/gem1.9.1 /usr/bin/gem
+ln -sf /usr/bin/irb1.9.1 /usr/bin/irb
+export GEM_BIN=/var/lib/gems/1.9.1/bin
+export PATH=$PATH:$GEM_BIN
+cd $CURDIR
+gem install bundler --no-rdoc --no-ri
+if [ -f $GEM_BIN/bundle ];
+then
+    export BUNDLER=$GEM_BIN/bundle
+else
+    export BUNDLER=/var/lib/gems/1.8/bin/bundle
+fi
+
+if [ -f $BUNDLER ];
+then
+    $BUNDLER install
+    $BUNDLER exec gem install dist/videoreg-0.1.gem
+
+    echo "Information about plugged devices:"
+    videoreg -I
+    videoreg -U
+else
+    echo "Bundler executable not found ($BUNDLER)! Installation FAILED!"
+fi