Adam Lum's solution for RPCFN #1 (Shift Subtitle)

shift_subtitle.rb

# Adam Lum's solution for RPCFN #1
# http://rubylearning.com/blog/2009/09/24/rpcfn-shift-subtitle-1/

require 'optparse'
 
# Get the command line arguments
options = {}
OptionParser.new do |opts|
  opts.banner = "Usage: shift_subtitle.rb [options] input_file output_file"
  opts.on("--operation [OPERATION]") do |o|
    options[:operation] = o
  end
  opts.on("--time [TIME]") do |t|
    time_split = t.split(",")
    options[:seconds] = time_split[0].to_i
    options[:milliseconds] = time_split[1].to_i
    if options[:operation] == "sub"
      options[:seconds] = options[:seconds] * -1
      options[:milliseconds] = options[:milliseconds] * -1
    end
  end
end.parse!
 
# Read in the file
subtitle_entries = File.read(ARGV[0]).split("\n\n")
output_string = ""
 
# Process the shift
subtitle_entries.each do |subtitle_entry|
  subtitle_parts = subtitle_entry.split("\n")
  time_range = subtitle_parts[1].split(" --> ")
  start_time = time_range[0].gsub(",", ":").split(":")
  end_time = time_range[1].gsub(",", ":").split(":")
 
  start_total_milliseconds = start_time[3].to_i + options[:milliseconds]
  start_time[3] = start_total_milliseconds % 1000
  start_time[2] = start_time[2].to_i + (start_total_milliseconds / 1000)
  start_total_seconds = start_time[2].to_i + options[:seconds].to_i
  start_time[2] = start_total_seconds % 60
  start_time[1] = start_time[1].to_i + (start_total_seconds / 60)
  start_time[0] = start_time[0].to_i + (start_time[1] / 60)
  start_time[1] = start_time[1] % 60
 
  end_total_milliseconds = end_time[3].to_i + options[:milliseconds]
  end_time[3] = end_total_milliseconds % 1000
  end_time[2] = end_time[2].to_i + (end_total_milliseconds / 1000)
  end_total_seconds = end_time[2].to_i + options[:seconds].to_i
  end_time[2] = end_total_seconds % 60
  end_time[1] = end_time[1].to_i + (end_total_seconds / 60)
  end_time[0] = end_time[0].to_i + (end_time[1] / 60)
  end_time[1] = (end_time[1] % 60)
 
  output_string += "#{subtitle_parts[0]}\n%02d:%02d:%02d,%03d --> %02d:%02d:%02d,%03d" % [start_time[0], start_time[1], start_time[2], start_time[3], end_time[0], end_time[1], end_time[2], end_time[3]]
  (2..subtitle_parts.size - 1).each do |i|
    output_string += "\n#{subtitle_parts[i]}"
  end
  output_string += "\n\n"
end
 
# Write the file
File.open(ARGV[1], 'w') {|f| f.write(output_string.strip)}