apple-tv-converter 0.0.1

data/CHANGELOG

@@ -0,0 +1,3 @@
+v0.0.1
+------
+* Mac OSX implementation

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2012 Pedro Rodrigues
+
+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,16 @@
+apple-tv-converter
+=================
+
+Media converter for AppleTV.
+
+## On Mac OSX
+
+- Uses ffmpeg for encoding (http://ffmpeg.org)
+- Uses Subler for adding the subtitles and setting the iTunes metadata tags (http://code.google.com/p/subler/)
+- Uses MKVToolNix for extracting the embedded subtitles from MKV files (http://www.bunkus.org/videotools/mkvtoolnix/)
+
+# TODO
+
+- Windows implementation
+- Tests!
+- Create a Mac OSX executable for running the script

data/lib/apple_tv_converter.rb

@@ -0,0 +1,72 @@
+$LOAD_PATH.unshift File.dirname(__FILE__)
+
+require 'logger'
+require 'stringio'
+require 'shellwords'
+require 'open3'
+require 'streamio-ffmpeg'
+require 'mkv'
+require 'awesome_print'
+require 'fileutils'
+require 'language_list'
+
+require 'apple_tv_converter/version'
+require 'apple_tv_converter/io_patch'
+require 'apple_tv_converter/media_converter'
+require 'apple_tv_converter/media'
+require 'apple_tv_converter/media_converter_adapter'
+require 'apple_tv_converter/media_converter_windows_adapter' if RUBY_PLATFORM =~ /(win|w)(32|64)$/
+require 'apple_tv_converter/media_converter_mac_adapter' if RUBY_PLATFORM =~ /(darwin)/
+
+module AppleTvConverter
+  # AppleTvConverter logs information about its progress when it's transcoding.
+  # Jack in your own logger through this method if you wish to.
+  #
+  # @param [Logger] log your own logger
+  # @return [Logger] the logger you set
+  def self.logger=(log)
+    @logger = log
+  end
+  
+  # Get AppleTvConverter logger.
+  #
+  # @return [Logger]
+  def self.logger
+    return @logger if @logger
+    logger = Logger.new(STDOUT)
+    logger.level = Logger::INFO
+    @logger = logger
+  end
+
+  # Set the path of the mp4box binary.
+  # Can be useful if you need to specify a path such as /usr/local/bin/mp4box
+  #
+  # @param [String] path to the mp4box binary
+  # @return [String] the path you set
+  def self.mp4box_binary=(bin)
+    @mp4box_binary = bin
+  end
+
+  # Get the path to the mp4box binary, defaulting to 'MP4Box'
+  #
+  # @return [String] the path to the mp4box binary
+  def self.mp4box_binary
+    @mp4box_binary.nil? ? 'MP4Box' : @mp4box_binary
+  end
+
+  # Set the path of the atomic parsley binary.
+  # Can be useful if you need to specify a path such as /usr/local/bin/atomic_parsley
+  #
+  # @param [String] path to the atomic parsley binary
+  # @return [String] the path you set
+  def self.atomic_parsley_binary=(bin)
+    @atomic_parsley_binary = bin
+  end
+
+  # Get the path to the atomic_parsley binary, defaulting to 'AtomicParsley'
+  #
+  # @return [String] the path to the atomic_parsley binary
+  def self.atomic_parsley_binary
+    @atomic_parsley_binary.nil? ? 'AtomicParsley' : @atomic_parsley_binary
+  end
+end

data/lib/apple_tv_converter/io_patch.rb

@@ -0,0 +1,59 @@
+if RUBY_VERSION =~ /1\.8/
+  # Useful when `timeout.rb`, which, on M.R.I 1.8, relies on green threads, does not work consistently.
+  begin
+    require 'system_timer'
+    AppleTvConverter::Timer = SystemTimer
+  rescue LoadError
+    require 'timeout'
+    AppleTvConverter::Timer = Timeout
+  end
+else
+  require 'timeout'
+  AppleTvConverter::Timer = Timeout
+end
+
+require 'win32/process' if RUBY_PLATFORM =~ /(win|w)(32|64)$/
+
+#
+# Monkey Patch timeout support into the IO class
+#
+class IO
+  def each_with_timeout(pid, seconds, sep_string=$/)
+    sleeping_queue = Queue.new
+    thread = nil
+
+    timer_set = lambda do
+      thread = new_thread(pid) { AppleTvConverter::Timer.timeout(seconds) { sleeping_queue.pop } }
+    end
+
+    timer_cancel = lambda do
+      thread.kill if thread rescue nil
+    end
+
+    timer_set.call
+    each(sep_string) do |buffer|
+      timer_cancel.call
+      yield buffer
+      timer_set.call
+    end
+  ensure
+    timer_cancel.call
+  end
+
+  private
+  def new_thread(pid, &block)
+    current_thread = Thread.current
+    Thread.new do
+      begin
+        block.call
+      rescue Exception => e
+        current_thread.raise e
+        if RUBY_PLATFORM =~ /(win|w)(32|64)$/
+          Process.kill(1, pid)
+        else
+          Process.kill('SIGKILL', pid)
+        end
+      end
+    end
+  end
+end

data/lib/apple_tv_converter/media.rb

@@ -0,0 +1,58 @@
+module AppleTvConverter
+  class Media
+    attr_accessor :show, :season, :number, :genre
+    attr_reader :original_filename, :quality
+
+    def original_filename=(value)
+      @original_filename = value
+
+      if @original_filename =~ /.*?\.m4v$/
+        if File.exists?(@original_filename.gsub(/\.m4v/, '.avi'))
+          @original_filename = @original_filename.gsub(/\.m4v/, '.avi')
+        elsif File.exists?(@original_filename.gsub(/\.m4v/, '.mkv'))
+          @original_filename = @original_filename.gsub(/\.m4v/, '.mkv')
+        end
+      end
+    end
+
+    def subtitle_filename
+      @subtitle_filename ||= self.original_filename.gsub(/\.(mkv|avi|mp4)/, '.srt')
+    end
+
+    def converted_filename
+      @converted_filename ||= self.original_filename.gsub(/\.(mkv|avi)/, '.m4v')
+    end
+
+    def converted_filename_with_subtitles
+      @converted_filename_with_subtitles ||= self.original_filename.gsub(/\.(mkv|avi|mp4)/, '_subtitled.m4v')
+    end
+
+    def ffmpeg_data
+      @ffmpeg_data ||= FFMPEG::Movie.new(original_filename)
+    end
+
+    def mkv_data
+      @mkv_data ||= MKV::Movie.new(original_filename)
+    end
+
+    def is_tv_show_episode?
+      !season.nil? && !number.nil?
+    end
+
+    def is_movie?
+      !is_tv_show_episode?
+    end
+
+    def is_mkv?
+      ffmpeg_data.container =~ /matroska/i rescue File.extname(original_filename).downcase == '.mkv'
+    end
+
+    def is_mp4?
+      ffmpeg_data.container =~ /mp4/ rescue File.extname(original_filename) =~ /\.(mp4|m4v)$/
+    end
+
+    def is_valid?
+      ffmpeg_data.valid?
+    end
+  end
+end

data/lib/apple_tv_converter/media_converter.rb

@@ -0,0 +1,49 @@
+module AppleTvConverter
+  class MediaConverter
+    @@timeout = 200
+
+    def is_windows? ; RUBY_PLATFORM =~/.*?mingw.*?/i ; end
+    def is_macosx? ; RUBY_PLATFORM =~/.*?darwin.*?/i ; end
+
+    def initialize
+      @adapter = is_windows? ? AppleTvConverter::MediaConverterWindowsAdapter.new : AppleTvConverter::MediaConverterMacAdapter.new
+    end
+
+    def process_media(media)
+      AppleTvConverter.logger.debug "**** #{File.basename(media.original_filename)}"
+      AppleTvConverter.logger.debug "* Video codec: #{media.ffmpeg_data.video_codec}"
+      AppleTvConverter.logger.debug "* Audio codec: #{media.ffmpeg_data.audio_codec}"
+      AppleTvConverter.logger.debug "* Container: #{media.ffmpeg_data.container}" rescue nil
+
+      extract_subtitles(media) if media.is_mkv?
+
+      if @adapter.transcode(media)
+        @adapter.add_subtitles media
+        @adapter.tag media
+
+        # add_to_itunes media
+        @adapter.clean_up media
+      end
+    end
+
+    private
+
+      # HELPERS
+      def process_all_medias(medias)
+        medias.each_with_index do |media, index|
+          puts '*' * 80
+          puts "* [#{(index + 1).to_s.rjust(medias.length.to_s.length, ' ')}/#{medias.length}] #{File.basename(media.original_filename)}"
+          
+          process_media media
+        end
+      end
+
+      def process_file(file)
+        puts '*' * 80
+        puts "* #{file}"
+        
+        media = parse_filename(file)
+        process_media media
+      end
+  end
+end

data/lib/apple_tv_converter/media_converter_adapter.rb

@@ -0,0 +1,87 @@
+module AppleTvConverter
+  class MediaConverterAdapter
+    def extract_subtitles(media)
+      printf "*** Extracting subtitles"
+      media.mkv_data.extract_subtitles(File.dirname(media.original_filename)) do |progress|
+        printf "\r* Progress: #{progress}%%"
+      end
+      puts "\r* Progress: [DONE]"
+    end
+
+    def transcode(media)
+      if convert?(media)
+        puts "*** Encoding"
+
+        options = {
+          :video_codec => convert_video?(media) ? 'mpeg4' : 'copy',
+          :audio_codec => convert_audio?(media) ? 'libfaac' : 'copy'
+        }
+
+        options[:custom] = "-qscale 1" if convert_video?(media)
+
+        transcoded = media.ffmpeg_data.transcode(media.converted_filename, options) do |progress|
+          printf %Q[\r* Progress: #{(progress * 100).round(2)}%%]
+        end
+
+        status transcoded.valid?
+
+        if status
+          puts "\r* Encoding: [DONE]#{' ' * 20}"
+        else
+          puts "\r* Encoding: [ERROR]#{' ' * 20}"
+        end
+      else
+        puts "*** Encoding: [UNNECESSARY]"
+        status = true
+      end
+
+      return status
+    end
+
+    def add_subtitles(media)
+      raise NotImplementedYetException
+    end
+
+    def tag(media)
+      raise NotImplementedYetException
+    end
+
+    def add_to_itunes(media)
+      raise NotImplementedYetException
+    end
+
+    def clean_up(media)
+      printf "*** Cleaning up"
+      begin
+        FileUtils.rm(media.original_filename) unless media.original_filename == media.converted_filename
+        FileUtils.rm_r list_files(File.join(File.dirname(media.original_filename), '*.srt'))
+        puts " [DONE]"
+      rescue
+        puts " [ERROR]"
+      end
+    end
+
+    protected
+
+      def list_files(ls)
+        raise NotImplementedYetException
+      end
+
+      def convert_audio?(media)
+        return media.ffmpeg_data.audio_codec !~ /(?:ac3|aac)/i
+      end
+
+      def convert_video?(media)
+        return media.ffmpeg_data.video_codec !~ /.*(?:mpeg4).*/i
+      end
+
+      def convert?(media)
+        !(media.is_valid? && media.is_mp4?)# && !convert_video?(media) && !convert_audio?(media))
+      end
+
+      def has_subtitles?(media)
+        list_files(File.join(File.dirname(media.original_filename), '*.srt')).any?
+      end
+
+  end
+end

data/lib/apple_tv_converter/media_converter_mac_adapter.rb

@@ -0,0 +1,87 @@
+module AppleTvConverter
+  class MediaConverterMacAdapter < MediaConverterAdapter
+    def add_subtitles(media)
+      puts "*** Adding subtitles"
+
+      printf "* Removing any previous subtitles"
+
+      command_line = %Q[./SublerCLI -remove -dest "#{media.converted_filename}"]
+
+      AppleTvConverter.logger.debug "Executing:"
+      AppleTvConverter.logger.debug command_line
+
+      output = Open3.popen3(command_line) { |stdin, stdout, stderr, wait_thr| stdout.read }
+
+      puts output.strip.empty? ? " [DONE]" : (output.strip == 'Error: (null)' ? " [NONE FOUND]" : " [ERROR]")
+
+      if has_subtitles?(media)
+        list_files(File.join(File.dirname(media.original_filename), '*.srt')).map do |subtitle_filename|
+          subtitle_filename =~ /(\w{3})\.srt$/
+          language_code = $1 || 'eng'
+
+          language = ::LanguageList::LanguageInfo.find_by_iso_639_3(language_code)
+          language ||= ::LanguageList::LanguageInfo.find_by_iso_639_3('eng')
+
+
+          command_line = "./SublerCLI "
+          command_line << %Q[-source "#{subtitle_filename}" ]
+          command_line << %Q[-language "#{language.name}" ]
+          command_line << %Q[-dest "#{media.converted_filename}"]
+
+          AppleTvConverter.logger.debug "Executing:"
+          AppleTvConverter.logger.debug command_line
+
+          printf "* Adding #{language.name} subtitles"
+          output = Open3.popen3(command_line) { |stdin, stdout, stderr, wait_thr| stdout.read }
+
+          puts output.strip.empty? ? " [DONE]" : " [ERROR]"
+        end
+      else
+        puts "* No new subtitles found"
+      end
+    end
+
+    def tag(media)
+      metadata = ''
+      metadata << %Q[{Name: #{media.show} S#{media.season.to_s.rjust(2, '0')}E#{media.number.to_s.rjust(2, '0')}}] if media.is_tv_show_episode?     
+      metadata << %Q[{Name: #{media.show}}] if media.is_movie?
+      metadata << %Q[{Genre: #{media.show}}{TV Show: #{media.show}}{TV Season: #{media.season}}{TV Episode #: #{media.number}}] if media.is_tv_show_episode?
+      metadata << %Q[{Genre: #{media.quality} Movie}] if media.is_movie?
+
+      command_line = %Q[./SublerCLI -metadata "#{metadata}" -dest "#{media.converted_filename}"]
+
+      AppleTvConverter.logger.debug "Executing:"
+      AppleTvConverter.logger.debug command_line
+
+      printf "*** Tagging"
+
+      output = Open3.popen3(command_line) { |stdin, stdout, stderr, wait_thr| stdout.read }
+
+      puts output.strip.empty? ? " [DONE]" : " [ERROR]"
+
+      return output.strip.empty?
+    end
+
+    def add_to_itunes(media)
+      printf "*** Adding to iTunes"
+
+      command_line = [
+        "ln -s",
+        "#{media.converted_filename}".gsub(/\s/, '\ ').gsub(/\[/, '\[').gsub(/\]/, '\]'),
+        "#{File.expand_path(File.join('~', 'Music', 'iTunes', 'iTunes Media', 'Automatically Add to iTunes.localized'))}".gsub(/\s/, '\ ').gsub(/\[/, '\[').gsub(/\]/, '\]')
+      ].join(' ')
+
+      AppleTvConverter.logger.debug "Executing:"
+      AppleTvConverter.logger.debug command_line
+
+      `#{command_line}`
+
+      puts ' [DONE]'
+      return true
+    end
+
+    def list_files(ls)
+      `ls -1 #{ls.gsub(/\s/, '\ ').gsub(/\[/, '\[').gsub(/\]/, '\]')} 2>/dev/null`.split("\n")
+    end
+  end
+end

data/lib/apple_tv_converter/media_converter_windows_adapter.rb

@@ -0,0 +1,43 @@
+module AppleTvConverter
+  class MediaConverterWindowsAdapter < MediaConverterAdapter
+    require 'win32ole'
+
+    def handbrake_location
+      return File.expand_path("./HandBrakeCLI#{'_x64' if is_windows_64bit?}.exe")
+    end
+
+    def atomic_parsley_location
+      return File.expand_path('./AtomicParsley.exe')
+    end
+
+    def clean_up_command_line(command_line)
+      return command_line.gsub(/\//, '\\')
+    end
+    
+    def add_to_itunes(episode)
+      printf "* Adding to iTunes"
+      @itunes ||= WIN32OLE.new("iTunes.Application")
+      @itunes.PlayFile(episode.converted_filename)
+      @itunes.Stop
+
+      puts " [DONE]"
+    end
+
+    def line_ending
+    end
+
+    def execute_command(command_line, &block)
+      `#{command_line}`
+    end
+
+    private
+
+      def is_windows_32bit?
+        !is_windows_64bit?
+      end
+
+      def is_windows_64bit?
+        ENV.has_key?('ProgramFiles(x86)')
+      end
+  end
+end

data/lib/apple_tv_converter/version.rb

@@ -0,0 +1,3 @@
+module AppleTvConverter
+  VERSION = "0.0.1"
+end

metadata

@@ -0,0 +1,137 @@
+--- !ruby/object:Gem::Specification
+name: apple-tv-converter
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Pedro Rodrigues
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-11-15 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.7'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.7'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.2
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.2
+- !ruby/object:Gem::Dependency
+  name: mkv
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.0.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.0.1
+- !ruby/object:Gem::Dependency
+  name: streamio-ffmpeg
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.0
+- !ruby/object:Gem::Dependency
+  name: language_list
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.0.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.0.3
+description: Converts movies to a format playable on Apple TV. Supporting multiple
+  subtitles.
+email:
+- pedro@bbde.org
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/apple_tv_converter/io_patch.rb
+- lib/apple_tv_converter/media.rb
+- lib/apple_tv_converter/media_converter.rb
+- lib/apple_tv_converter/media_converter_adapter.rb
+- lib/apple_tv_converter/media_converter_mac_adapter.rb
+- lib/apple_tv_converter/media_converter_windows_adapter.rb
+- lib/apple_tv_converter/version.rb
+- lib/apple_tv_converter.rb
+- README.md
+- LICENSE
+- CHANGELOG
+homepage: http://github.com/gokuu/apple-tv-converter
+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.24
+signing_key: 
+specification_version: 3
+summary: Converts movies to a format playable on Apple TV.
+test_files: []