rakemkv 0.2.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: cd54a538d15fb78a5905d935a82afd6015cb9c5d
+  data.tar.gz: dae6fb5b06d867b7174c9f5d8783cbad34d3f37d
+SHA512:
+  metadata.gz: 9af5fa59cabaa281f2f7a8fdbd26f2468896333f9209b037156a01e1037e6fce90a7dceb5791112ca3d02a6a1adcad6f731f62bdbdbc5ef165d350416453a56d
+  data.tar.gz: f4295cb3583406b97e197c99e8efe29d3e245dbdc509b6de6537a1413102b54828c338b946288dbee3bf23b732df146be85d8ba9668083fd2b01be9f4606423a

data/.gitignore

@@ -0,0 +1,20 @@
+*.gem
+*.orig
+*.rbc
+.#*
+.bundle
+.config
+.rspec
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/.travis.yml

@@ -0,0 +1,5 @@
+language: ruby
+rvm:
+  - 2.0.0
+  - 1.9.3
+  - 1.9.2

data/Gemfile

@@ -0,0 +1,3 @@
+source 'https://rubygems.org'
+
+gemspec

data/Guardfile

@@ -0,0 +1,8 @@
+# A sample Guardfile
+# More info at https://github.com/guard/guard#readme
+
+guard :rspec do
+  watch(%r{^spec/.+_spec\.rb$})
+  watch(%r{^lib/(.+)\.rb$})     { |m| "spec/lib/#{m[1]}_spec.rb" }
+  watch('spec/spec_helper.rb')  { "spec" }
+end

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013-2015 Eric Collins
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,51 @@
+[![Build Status](https://travis-ci.org/tabfugnic/rakemkv.png)](https://travis-ci.org/tabfugnic/rakemkv)
+
+# Rakemkv
+
+A fully object oriented wrapper around MakeMKV to help facilitate more
+programmable backups.
+
+## Requirements
+
+- MakeMKV
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'rakemkv'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install rakemkv
+
+## Usage
+
+```
+disc = RakeMKV::Disc.new('location/to/disc')
+disc.transcode!
+```
+
+## Configuration
+
+You can configure RakeMKV by doing the following:
+
+```ruby
+RakeMKV.configure do |config|
+  config.binary = 'new_makemkv_binary'
+  config.destination = 'new/destination/path'
+  config.minimum_title_length = 120 # Number in seconds
+end
+```
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,5 @@
+require "bundler/gem_tasks"
+require 'rspec/core/rake_task'
+
+task default: :spec
+RSpec::Core::RakeTask.new

data/lib/rakemkv/code.rb

@@ -0,0 +1,62 @@
+#  Code
+class RakeMKV::Code
+  CODES = [
+    :unknown,
+    :type,
+    :name,
+    :language_code,
+    :language_name,
+    :codec_id,
+    :codec_short,
+    :codec_long,
+    :chapter_count,
+    :duration,
+    :disk_size,
+    :disk_size_bytes,
+    :stream_type_extension,
+    :bitrate,
+    :audio_channels_count,
+    :angle_info,
+    :source_file_name,
+    :audio_sample_rate,
+    :audio_sample_size,
+    :video_size,
+    :video_aspect_ratio,
+    :video_frame_rate,
+    :stream_flags,
+    :date_time,
+    :original_title_id,
+    :segments_count,
+    :segments_map,
+    :output_file_name,
+    :metadata_language_code,
+    :metadata_language_name,
+    :tree_info,
+    :panel_title,
+    :volume_name,
+    :order_weight,
+    :output_format,
+    :output_format_description,
+    :seamless_info,
+    :panel_text,
+    :mkv_flags,
+    :mkv_flags_text
+  ]
+
+  attr_reader :index
+
+  #  Initialize code
+  def initialize(index)
+    @index = index.to_i
+  end
+
+  #  Take code and convert it to the proper symbol
+  def to_sym
+    CODES[index]
+  end
+
+  #  Short hand to initialize code and convert it to symbol
+  def self.[](index)
+    new(index).to_sym
+  end
+end

data/lib/rakemkv/command.rb

@@ -0,0 +1,35 @@
+#  Command Object
+class RakeMKV::Command
+  #  Initialize with path
+  def initialize(path)
+    @path = path
+  end
+
+  #  Check if mkv is installed
+  def self.installed?
+    output = Cocaine::CommandLine.new('which', RakeMKV.config.binary).run
+    !output.empty?
+  end
+
+  #  Call info command on disc
+  def info
+    @info ||= execute("info #{@path}")
+  end
+
+  #  Call mkv command on disc
+  def mkv(title_id, destination)
+    @mkv ||= execute("mkv #{@path} #{title_id} #{destination}")
+  end
+
+  private
+
+  def execute(command)
+    Cocaine::CommandLine.new(
+      "#{RakeMKV.config.binary} -r", full_command(command)
+    ).run
+  end
+
+  def full_command(command)
+    RakeMKV::CommandBuilder.new(command).build
+  end
+end

data/lib/rakemkv/command_builder.rb

@@ -0,0 +1,19 @@
+class RakeMKV::CommandBuilder
+  def initialize(command)
+    @command = command
+  end
+
+  def build
+    [command, minimum_length].join(' ').strip
+  end
+
+  private
+
+  attr_reader :command
+
+  def minimum_length
+    if RakeMKV.config.minimum_title_length
+      "--minlength=#{RakeMKV.config.minimum_title_length}"
+    end
+  end
+end

data/lib/rakemkv/configuration.rb

@@ -0,0 +1,20 @@
+class RakeMKV::Configuration
+  include Singleton
+
+  attr_accessor :minimum_title_length
+  attr_writer :binary, :destination
+
+  def binary
+    @binary || 'makemkvcon'
+  end
+
+  def destination
+    @destination || Dir.pwd
+  end
+
+  def reset!
+    self.binary = nil
+    self.destination = nil
+    self.minimum_title_length = nil
+  end
+end

data/lib/rakemkv/disc.rb

@@ -0,0 +1,64 @@
+#  Disc object
+class RakeMKV::Disc
+  attr_reader :location
+
+  #  Initialize disc
+  def initialize(location)
+    @location = location
+  end
+
+  # Get path from location
+  def path
+    if location =~ /^\/dev/
+      "dev:#{location}"
+    elsif location =~ /iso$/
+      "iso:#{location}"
+    elsif location.is_a?(Integer)
+      "disc:#{location}"
+    elsif location =~ /^disc/
+      location
+    else
+      raise RuntimeError
+    end
+  end
+
+  # parse file info from command
+  def info
+    @info ||= RakeMKV::Parser.new(command.info)
+  end
+
+  #  Transcode information on disc
+  def transcode!(options = {})
+    destination = options[:destination] || RakeMKV.config.destination
+    title_id = options[:title_id] || titles.longest.id
+    command.mkv(title_id, destination)
+  end
+
+  # Get titles for disc
+  def titles
+    RakeMKV::Titles.new(build_titles)
+  end
+
+  #  Meta disc information
+  def method_missing(method, *args)
+    info.cinfo[method.to_sym] || super
+  end
+
+  private
+
+  def command
+    RakeMKV::Command.new(path)
+  end
+
+  def build_titles
+    info.tinfo.each_with_index.map do |title, title_id|
+      RakeMKV::Title.new(title_id, title)
+    end
+  end
+
+  def check_directory!(destination)
+    unless File.directory? destination
+      raise StandardError
+    end
+  end
+end

data/lib/rakemkv/parser.rb

@@ -0,0 +1,85 @@
+# Parser
+class RakeMKV::Parser
+  CINFO_REGEX = /CINFO:(\d+),\d+,"(.+)"/
+  DRIVES_REGEX = /DRV:\d+,\d+,\d+,(\d+),"(.*)","(.*)","(.*)"/
+  MSG_REGEX = /MSG:[^"]*"([^"]*)"/
+  SINFO_REGEX = /SINFO:(\d+),(\d+),(\d+),\d+,"(.*)"/
+  TINFO_REGEX = /TINFO:(\d+),(\d+),\d+,"(.*)"/
+
+  attr_reader :raw
+
+  #  Initialize using info received from disc
+  def initialize(raw_info)
+    @raw = raw_info
+  end
+
+  #  Grab information from cinfo
+  def cinfo
+    cinfo = {}
+    parse(CINFO_REGEX) do |code, info|
+      code = RakeMKV::Code[code]
+      cinfo[code] = info
+    end
+    cinfo
+  end
+
+  #  Grab information from tinfo
+  def tinfo
+    tinfo = []
+    parse(TINFO_REGEX) do |title_id, code, info|
+      code = RakeMKV::Code[code]
+      tinfo[title_id.to_i] ||= Hash.new
+      tinfo[title_id.to_i][code] = info
+    end
+    tinfo
+  end
+
+  #  Grab information from sinfo
+  def sinfo
+    sinfo = []
+    parse(SINFO_REGEX) do |title_id, section_id, code, info|
+      code = RakeMKV::Code[code]
+      title = title_id.to_i
+      section = section_id.to_i
+      sinfo[title] ||= Array.new
+      sinfo[title][section] ||= Hash.new
+      sinfo[title][section][code] = info
+    end
+    sinfo
+  end
+
+  #  Grab information from messages
+  def messages
+    messages = []
+    parse(MSG_REGEX) do |info|
+      messages << info.first
+    end
+    messages
+  end
+
+  #  Grab information from discs
+  def drives
+    drives = []
+    parse(DRIVES_REGEX) do |accessible, drive_name, disc_name, location|
+      drives << {
+        accessible: accessible(accessible),
+        drive_name: drive_name,
+        disc_name: disc_name,
+        location: location
+      }
+    end
+    drives
+  end
+
+  private
+
+  def accessible(accessible)
+    accessible != '0'
+  end
+
+  def parse(regex, &block)
+    raw.split('\n').each do |line|
+      line.scan(regex, &block)
+    end
+  end
+end

data/lib/rakemkv/title.rb

@@ -0,0 +1,40 @@
+#  Title
+class RakeMKV::Title
+  attr_reader :id, :code, :chapter_count, :size
+
+  MINUTE = 60
+  HOUR = 3600
+
+  #  Initialize with a required id
+  def initialize(id, options = {})
+    # These claim to start at 1, but the CLI treats them as starting from 0
+    @id = id
+    @duration = options[:duration]
+    @chapter_count = options[:chapter_count].to_i
+    @size = options[:disk_size_bytes].to_i
+  end
+
+  # Find short lengthed title
+  def short_length?
+    time > 900 && time < 2100
+  end
+
+  def time
+    time_from_integer || time_from_string
+  end
+
+  private
+
+  attr_reader :duration
+
+  def time_from_integer
+    if duration.is_a? Integer
+      duration
+    end
+  end
+
+  def time_from_string
+    times = duration.split(':')
+    (times[0].to_i * HOUR) + (times[1].to_i * MINUTE) + (times[2].to_i)
+  end
+end

data/lib/rakemkv/titles.rb

@@ -0,0 +1,22 @@
+# Titles
+class RakeMKV::Titles
+  include Enumerable
+
+  def initialize(titles)
+    @titles = titles
+  end
+
+  def each(&block)
+    @titles.each(&block)
+  end
+
+  #  Find title by id
+  def at_id(id)
+    detect { |title| title.id == id }
+  end
+
+  #  Get longest title
+  def longest
+    max { |a, b| a.time <=> b.time }
+  end
+end

data/lib/rakemkv/version.rb

@@ -0,0 +1,3 @@
+module RakeMKV
+  VERSION = '0.2.0'
+end

data/lib/rakemkv.rb

@@ -0,0 +1,20 @@
+module RakeMKV
+  def self.config
+    Configuration.instance
+  end
+
+  def self.configure
+    yield config
+  end
+end
+
+require "cocaine"
+require "singleton"
+require "rakemkv/code"
+require "rakemkv/command"
+require "rakemkv/command_builder"
+require "rakemkv/configuration"
+require "rakemkv/disc"
+require "rakemkv/parser"
+require "rakemkv/title"
+require "rakemkv/titles"

data/rakemkv.gemspec

@@ -0,0 +1,25 @@
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'rakemkv/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "rakemkv"
+  spec.version       = RakeMKV::VERSION
+  spec.authors       = ["Eric Collins"]
+  spec.email         = ["eric@tabfugni.cc"]
+  spec.description   = %q{A fully object oriented wrapper around MakeMKV to help facilitate more programmable backups.}
+  spec.summary       = %q{Object oriented wrapper around MakeMKV}
+  spec.homepage      = "https://github.com/tabfugnic/rakemkv"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.test_files    = spec.files.grep(%r{spec/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "json", "~> 1.7"
+  spec.add_dependency "cocaine", "~> 0.5"
+
+  spec.add_development_dependency "bundler"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "rspec", "~> 3.2"
+end

data/spec/integration/transcodes_disc_spec.rb

@@ -0,0 +1,17 @@
+require 'spec_helper'
+
+describe 'transcoding disc' do
+  it 'copies disc to destination' do
+    movie = File.new('./spec/fixtures/movie.iso')
+    movie_path = File.expand_path(movie.path)
+
+    Dir.mktmpdir do |directory|
+      RakeMKV.config.minimum_title_length = 2
+      RakeMKV.config.destination = directory
+
+      RakeMKV::Disc.new(movie_path).transcode!
+
+      expect(Dir.entries(directory)).to include('title00.mkv')
+    end
+  end
+end

data/spec/lib/rakemkv/code_spec.rb

@@ -0,0 +1,21 @@
+require 'spec_helper'
+
+describe RakeMKV::Code do
+  describe '#new' do
+    it 'takes the index' do
+      expect(RakeMKV::Code.new(5)).to be_a RakeMKV::Code
+      expect(RakeMKV::Code.new(5).index).to eq 5
+    end
+  end
+  describe '#to_sym' do
+    it 'returns the code' do
+      expect(RakeMKV::Code.new(5).to_sym).to eq :codec_id
+    end
+  end
+
+  describe '.[]' do
+    it 'gets the code' do
+      expect(RakeMKV::Code[5]).to eq :codec_id
+    end
+  end
+end

data/spec/lib/rakemkv/command_builder_spec.rb

@@ -0,0 +1,15 @@
+require 'spec_helper'
+
+describe RakeMKV::CommandBuilder do
+  describe '#build' do
+    it 'adds minimum length from configuration' do
+      RakeMKV.configure do |config|
+        config.minimum_title_length = 5
+      end
+
+      full_command = RakeMKV::CommandBuilder.new("command").build
+
+      expect(full_command).to eq "command --minlength=5"
+    end
+  end
+end

data/spec/lib/rakemkv/command_spec.rb

@@ -0,0 +1,33 @@
+require 'spec_helper'
+
+describe RakeMKV::Command do
+  describe '.installed?' do
+
+    it 'verifies makemkv is installed' do
+      allow(Cocaine::CommandLine).to receive(:new)
+        .with('which', 'makemkvcon')
+        .and_return(double('command', run: '/something/great'))
+      expect(RakeMKV::Command).to be_installed
+    end
+  end
+
+  describe '#info' do
+    it 'gets the info for the object' do
+      command = RakeMKV::Command.new('disc:0')
+      allow(Cocaine::CommandLine).to receive(:new)
+        .with('makemkvcon -r', 'info disc:0')
+        .and_return(double('cocaine', run: 'info'))
+      expect(command.info).to eq 'info'
+    end
+  end
+
+  describe '#mkv' do
+    it 'takes title and destination' do
+      command = RakeMKV::Command.new('disc:0')
+      allow(Cocaine::CommandLine).to receive(:new)
+        .with('makemkvcon -r', 'mkv disc:0 5 /path/to/heart')
+        .and_return(double('cocaine', run: 'mkv'))
+      expect(command.mkv(5,'/path/to/heart')).to eq 'mkv'
+    end
+  end
+end

data/spec/lib/rakemkv/configuration_spec.rb

@@ -0,0 +1,47 @@
+require 'spec_helper'
+
+describe RakeMKV::Configuration do
+  before do
+    config.reset!
+  end
+
+  describe '#binary' do
+    it 'returns default makemkvcon' do
+      expect(config.binary).to eq 'makemkvcon'
+    end
+
+    it 'set a new binary' do
+      config.binary = 'new_binary'
+      expect(config.binary).to eq 'new_binary'
+    end
+  end
+
+  describe '#destination' do
+    it 'defaults the to current directory' do
+      expect(config.destination).to eq Dir.pwd
+    end
+
+    it 'sets a new destination' do
+      config.binary = 'new_destination'
+      expect(config.binary).to eq 'new_destination'
+    end
+  end
+
+  describe '#reset!' do
+    it 'resets all the things' do
+      config.minimum_title_length = 3
+      config.binary = 'stuff'
+      config.destination = '/path/somewhere'
+
+      config.reset!
+
+      expect(config.minimum_title_length).to be_nil
+      expect(config.binary).to eq 'makemkvcon'
+      expect(config.destination).to eq Dir.pwd
+    end
+  end
+
+  def config
+    RakeMKV::Configuration.instance
+  end
+end