ripcd.rb

require 'tempfile'
require 'optparse'
require 'yaml'


ARGV.options do |opts|
  opts.on("-o", "--output-dir=val", String)  { |val| ProjectDir = val.tr("\\", "/") }
  opts.on("-d", "--discs=val", String) { |val| @discTotal = val.to_i }
  opts.on("-b", "--burst") { @mode = '-B' }
  opts.on("-s", "--secure") { @mode = '-S' }
  opts.on("-p", "--paranoid") { @mode = '-P' }
  opts.parse!
end

#confirm inputs

if defined?(ProjectDir).nil?
  puts "Please enter a valid output directory for rip!"
  exit
elsif ! File.exist?(ProjectDir)
  puts "Please enter a valid output directory for rip!"
  exit
end

if defined?(@discTotal).nil?
  puts "Please enter the amount of discs to be ripped!"
  exit
end

if defined?(@mode).nil?
  puts "Defaulting to secure rip"
  @mode = '-S'
else
  ripOptions = ['-B', '-S', '-P']
  if ! ripOptions.include? @mode
    puts 'Non-valid option entered for rip mode - using default of Secure'
    @mode = '-S'
  end
end



scriptPath = __dir__
configPath = scriptPath + "/ripcd.config"
configOptions = YAML.load_file(configPath)

CLI_Tools_Path = configOptions['CLI_Tools_Path']
CueToolsPath = configOptions['CueToolsPath']
LoadPath =  CLI_Tools_Path + configOptions['LoadPath']
UnloadPath =  CLI_Tools_Path + configOptions['UnloadPath']
Drive = configOptions['Drive']

def loadDisc()
  tempFile1 = Tempfile.new('batchRipping')
  tempFile2 = Tempfile.new('batchRipping')
  system(LoadPath, "--drive=#{Drive}", '--rejectifnodisc' "--logfile=#{tempFile1.path}", "--passerrorsback=#{tempFile2.path}")
  puts "\n"
end

def unloadDisc()
  tempFile1 = Tempfile.new('batchRipping')
  tempFile2 = Tempfile.new('batchRipping')
  system(UnloadPath, "--drive=#{Drive}", '--rejectifnodisc' "--logfile=#{tempFile1.path}", "--passerrorsback=#{tempFile2.path}")
  puts "\n"
end

def ripDisc()
  puts "Ripping disc: #{@discNumber}"
  `powershell "Start-Transcript -Append #{ProjectDir}/cdimage.consolelog ; #{CueToolsPath} -D #{Drive} #{@mode} ; Stop-Transcript"`
end

def checkRipError()
  ripLog = File.readlines("#{ProjectDir}/cdimage.consolelog")
  if ripLog.any? {|line| line.include?('Results')}
    ripExit = 5
  else
    ripExit = 1
  end
end
def updateCue(cuePath,outputFile)
  oldCue = File.readlines(cuePath)
  temp = Tempfile.new
  newCuePath = outputFile
  newWavPath = File.basename(outputFile, ".*") + '.wav'
  oldCue.each do|line|
    if (line.include?('FILE') && line.include?('WAVE'))
      newLine = "FILE " + '"' + newWavPath + '"' + " WAVE\n"
      temp << newLine
    else
      temp << line
    end
  end
  temp.rewind
  temp.close
  FileUtils.mv(temp.path,newCuePath)
  if ! File.readlines(newCuePath).empty?
    FileUtils.rm(cuePath)
  end
end

def renameOutput(file,time,status)
  outName = 'cdrip-'
  outName.prepend('FAIL_') if status == 'fail'
  dir = File.dirname(file)
  ext = File.extname(file)
  outputFile = dir + '/' + outName + time + ext
  if ext == '.cue'
    updateCue(file,outputFile)
  else
    File.rename(file,outputFile)
  end
end

# Start process
@discNumber = 1
Dir.chdir(ProjectDir)
while @discNumber <= @discTotal do
  ripAttempt = 1
  loadDisc()
  while ripAttempt <=5
    ripDisc()
    ripAttempt += checkRipError()
    sleep 5
  end
  time = Time.now.strftime("%m%d%H%M%S")
  outputFiles = Dir.glob("#{ProjectDir}/cdimage*")
  if outputFiles.length == 4
    ripLog = File.readlines("#{ProjectDir}/cdimage.consolelog")
    ripResults = ripLog["#{ripLog.count - 5}".to_i]
    puts ripResults
    outputFiles.each {|file| renameOutput(file, time, 'pass')}
    csvLine = "#{@discNumber}, cdrip-#{time}, #{ripResults}"
    open("#{ProjectDir}/rip-log.txt", 'a') { |f| f.puts csvLine}
    @discNumber += 1
  else
    sleep 20
    csvLine = "#{@discNumber}, FAIL, FAIL"
    puts "Rip Failed!"
    outputFiles.each {|file| renameOutput(file, time, 'fail')}
    @discNumber += 1
    open("#{ProjectDir}/rip-log.txt", 'a') { |f| f.puts csvLine}
  end
  unloadDisc()
  sleep 5
end
sleep 5
puts "All done!!"