rubyVDRconvert 0.0.1

data/bin/rubyVDRconvert

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+
+# TODO: Lower my own process priority to idle class
+# TODO: Write the daemon

data/lib/file_dir.rb

@@ -0,0 +1,31 @@
+require 'find'
+
+class FileDir
+  def self.delete!(filedir)
+	case Dir.exist?(filedir)
+	  when true
+		begin
+		  Dir.rmdir(filedir)
+		rescue
+		  Dir.foreach(filedir) { | entry |
+			delete!(entry) unless (entry == ".") or (entry == "..")
+		  }
+		  retry
+		end
+	  else
+		File.delete(filedir)
+	end
+  end
+  def self.mkdirtree(directory)
+	begin
+	  Dir.mkdir(directory)
+	rescue SystemCallError => ex
+	  parent = File.dirname(directory)
+	  # Prevent an endless loop
+	  return true if Dir.exist?(directory)
+	  raise ex if parent.length == '/'
+	  self.mkdirtree(parent)
+	  retry
+	end
+  end
+end

data/lib/processing_thread.rb

@@ -0,0 +1,8 @@
+class Processing_thread
+  def start
+	# TODO: Run the processing thread
+  end
+  def stop
+	# TODO: Stop the processing thread
+  end
+end

data/lib/recording.rb

@@ -0,0 +1,76 @@
+require 'rubygems'
+
+require 'recording_ts'
+require 'recording_vdr'
+require 'file_dir'
+
+class Recording
+  attr_reader :directory, :target_dir, :processed, :friendly_name
+  def initialize(directory)
+	@mutex = Mutex.new
+	@directory = directory
+	@recording = nil
+	@title = nil
+	@subtitle = nil
+	@friendly_name = nil
+	@target_dir = nil
+	@processed = false
+	@timestamp = File.mtime(@directory + '/index')
+	load_info
+	update_target_dir
+	determine_stream_type
+  end
+  def delete!()
+	@mutex.synchronize {
+	  @recording = nil
+	  GC.start
+	  FileDir.delete!(@target_dir)
+	}
+  end
+  def load_info()
+	#begin
+	  info = File.read(@directory + '/info')
+	  info.force_encoding('BINARY')
+	  cd = CharDet.detect(info)
+	  encoding = cd['encoding']
+	  info.each_line { | line |
+		case line[0..1]
+		  when 'T '
+			@title = line[2..256].chomp
+			@title.force_encoding(encoding)
+			p "Found title: #{@title}"
+		  when 'S '
+			@subtitle = line[2..256].chomp
+			@subtitle.force_encoding(encoding)
+			p "Found subtitle: #{@subtitle}"
+		end
+	  }
+	#rescue
+	#end
+	@title = 'Unknown' if not @title
+	@subtitle = 'Unknown' if not @subtitle
+  end
+  def determine_stream_type()
+	case
+	  when File.file?(@directory + '/00001.ts')
+		@recording = RecordingTS.new(self)
+	  when File.file?(@directory + '/00001.vdr')
+		@recording = RecordingVDR.new(self)
+	  else
+		raise 'Unknown recording'
+	end
+  end
+  def update_target_dir()
+	title = @title
+	subtitle = @subtitle
+	title.gsub!('/', '!')
+	subtitle.gsub!('/', '!')
+	@target_dir = "/video/#{title}/#{subtitle}"
+	@friendly_name = "#{title} - #{subtitle}"	
+  end
+  def process!()
+	@mutex.synchronize{
+	  @processed = true if @recording.process!
+	}
+  end
+end

data/lib/recording_ts.rb

@@ -0,0 +1,22 @@
+require 'file_dir'
+
+class RecordingTS
+  def initialize(recording)
+	@parent = recording
+  end
+  def process!()
+	target_file = @parent.target_dir + "/#{@parent.friendly_name}.ts"
+	FileDir.mkdirtree(@parent.target_dir)
+	tf = File.new(target_file, 'wb:ascii-8bit')
+	source_files = Dir.glob("#{@parent.directory}/*.ts")
+	source_files.sort!
+	source_files.each { | source_file |
+	  sf = File.new(source_file, 'rb:ascii-8bit')
+	  while not sf.eof?
+		tf.write(sf.read(1*(2**20))) # Only read some bytes at once
+		sleep(0.01) # Be friendly to other processes
+	  end
+	  sf.close
+	}
+  end
+end

data/lib/recording_vdr.rb

@@ -0,0 +1,21 @@
+class RecordingVDR
+  def initialize(recording)
+	@parent = recording
+  end
+  def process!()
+	return false
+	target_file = @parent.target_dir + "/#{@parent.friendly_name}.ts"
+	FileDir.mkdirtree(@parent.target_dir)
+	tf = File.new(target_file, 'wb:ascii-8bit')
+	source_files = Dir.glob("#{@parent.directory}/*.ts")
+	source_files.sort!
+	source_files.each { | source_file |
+	  sf = File.new(source_file, 'rb:ascii-8bit')
+	  while not sf.eof?
+		tf.write(sf.read(1*(2**20))) # Only read some bytes at once
+		sleep(0.01) # Be friendly to other processes
+	  end
+	  sf.close
+	}
+  end
+end

data/lib/recordings.rb

@@ -0,0 +1,59 @@
+require 'rubygems'
+require 'rchardet'
+require 'find'
+
+require 'recording'
+
+class Recordings
+  def initialize()
+	@recordings_db = Hash.new
+  end
+  def scan()
+	basedir = '/var/vdr/video'
+	found_directories = Array.new
+	Find.find(basedir) { | entry |
+	  entry.force_encoding('BINARY')
+	  # Guess encoding of filename
+	  cd = CharDet.detect(entry)
+	  encoding = cd['encoding']
+	  #p encoding
+	  entry.force_encoding(encoding)
+	  # Look for the index file created by VDR
+	  if File.basename(entry) == "index" then
+		directory = File.dirname(entry)
+		found_directories << directory
+		if @recordings_db[directory] == nil then
+		  # Found a new entry
+		  p directory
+		  add(directory)
+		end
+	  end
+	}
+	@recordings_db.each { | recording |
+	  recording.delete! unless found_directories.include?(recording.directory)
+	}
+  end
+  def dump()
+	File.new('/var/vdr/video/rubyVDRconvert.recordings.marshal', 'wb:ascii-8bit') { | f |
+	  f.write(Marshal.dump(@recordings_db))
+	}
+  end
+  def load()
+	begin
+	  @recordings_db = Marshal.load(File.read('/var/vdr/video/rubyVDRconvert.recordings.marshal'))
+	rescue
+	  @recordings_db = Hash.new
+	end
+  end
+  def add(directory)
+	# TODO: Add a new recording that was found in the specified directory
+	# Ensure that VDR has finished writing the recording by checking the mtime of the index file
+	# Wait at least 30 seconds after VDR has finished
+	return false if (Time.now - File.mtime(directory + '/index')) > 30
+	@recordings_db[directory] = Recording.new(directory)
+	@recordings_db[directory].process!
+  end
+  def delete(recording)
+	# TODO: Delete the recording
+  end
+end

metadata

@@ -0,0 +1,82 @@
+--- !ruby/object:Gem::Specification 
+name: rubyVDRconvert
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Daniel Frank
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-11-07 00:00:00 +01:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rchardet
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 1
+        - 3
+        - 0
+        version: 1.3.0
+  type: :runtime
+  version_requirements: *id001
+description: 
+email: scripts@danielfrank.net
+executables: 
+- rubyVDRconvert
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/file_dir.rb
+- lib/processing_thread.rb
+- lib/recording.rb
+- lib/recording_ts.rb
+- lib/recording_vdr.rb
+- lib/recordings.rb
+has_rdoc: true
+homepage: 
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 1
+      - 9
+      - 2
+      version: 1.9.2
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: background maintenance script for VDR
+test_files: []
+