submerge.rb - Pasteur

Filename:
Private:
Filetype:
Indentation:
#!/usr/bin/env ruby

if ARGV.first == '-h' || ARGV.first == '--help'
  puts "Usage: #{$0} [file.srt...]"
  exit!
end

subs = if ARGV.any?
  ARGV.map { |fn| [fn, File.read(fn)] }
else
  [['-', STDIN.read]]
end.to_h

subs.each do |fn, data|
  last_from = nil
  last_to = nil

  frames = []

  data.split("\n\n").each do |frame|
    _, from, to, line = *frame.strip.match( /(\d+:\d+:\d+,\d+) --> (\d+:\d+:\d+,\d+)(.*)/m )

    if from == last_from && to == last_to
      frames.last << "\r\n#{line.strip}"
    else
      STDERR.puts("!! WARNING: Possibly buggy frame #{from} --> #{to}") if from == last_from || to == last_to

      frames << "#{from} --> #{to}\n#{line.strip}"
      last_from = from
      last_to = to
    end
  end

  sub = frames.each_with_index.map { |frame, idx| "#{idx + 1}\n#{frame}\n" }.join("\n")
  if fn == '-'
    puts(sub)
  else
    File.write(fn, sub)
  end
end