class Dvd

# File lib/dvd_ripper/dvd.rb, line 4
def initialize
  @volumes = '/Volumes'
end

# File lib/dvd_ripper/dvd.rb, line 113
def eject
  system('drutil tray eject')
end

# File lib/dvd_ripper/dvd.rb, line 26
def present?
  !volume.nil?
end

# File lib/dvd_ripper/dvd.rb, line 30
def rip(output)
  File.delete(output) if File.exist?(output)

  state = :scanning
  percent = '0.0'
  eta = 'unknown'
  dc = 0
  spinner = ['-', '\\', '|', '/']

  encoding_reg = /Encoding: task \d+ of \d+, (?<percent>\d+\.\d+) %.*ETA (?<eta>\d{2}h\d{2}m\d{2}s)/i
  status_update = Thread.new do
    loop do
      sleep(1)
      dc += 1

      if state == :encoding
        print "\r#{spinner[dc % 4]} Encoding #{percent} % (time left #{eta})"
      end

      if state == :scanning
        print("\r" + (' ' * 50)) if (dc % 20) == 0
        print "\r#{spinner[dc % 4]} Scanning Titles.#{'.' * (dc % 20)}"
      end
    end
  end

  command_output = []
  last_line = nil
  File.open('./encode.log', 'w+') do |log|
    IO.popen("HandBrakeCLI -Z \"AppleTV 3\" --main-feature -i \"#{volume}\" -o \"#{output}\" 2>&1", 'w+') do |f|
      begin
        while buf = f.read(255)
          lines = buf.split(/\r|\n/)
          lines.compact!

          if (lines.size > 1) && (!last_line.nil?)
            command_output << last_line + lines.shift
          end

          if buf[-1] == "\r" || buf[-1] == "\n"
            last_line = nil
          else
            last_line = lines.pop
          end

          command_output << lines
          command_output.flatten!

          while ln = command_output.shift
            ln = ln.strip unless ln.nil?

            # Change to encoding as soon as a matching line is found
            state = :encoding if encoding_reg =~ ln

            if state == :encoding
              if m = encoding_reg.match(ln)
                percent = m[:percent]
                eta = m[:eta]
              end
            end

            log.write "#{state}:#{percent}:#{eta} - #{ln}\n"
          end
          command_output = []
        end
      rescue StandardError => e
        puts e.message
        puts e.class
        puts e.backtrace
      end
    end
  end
  exit_code = $CHILD_STATUS

  if exit_code.exitstatus != 0
    `killall HandBrakeCLI`
    puts 'Encoding Failed. See encode.log'
  end
ensure
  status_update.kill
  status_update.join
end

# File lib/dvd_ripper/dvd.rb, line 21
def title
  fail "Invalid DVD Root or DVD not mounted at #{@volumes}" unless present?
  File.basename(volume).tr('_', ' ').capitalize
end

# File lib/dvd_ripper/dvd.rb, line 8
def volume
  vol = nil
  Dir.entries(@volumes).each do |video_dir|
    next if ['.', '..'].member?(video_dir)
    full_path = File.join(@volumes, video_dir)
    if Dir.exist?(File.join(full_path, 'VIDEO_TS'))
      vol = full_path
      break
    end
  end
  vol
end