commando-munger-spec

#!/usr/bin/env ruby
require_relative 'commando-helpers.rb'

require 'csv'
require 'pp'
require 'edl'
require 'timecode'

program :description, 'WIP'
program :version, '1.0.0'

command :run do |c|
  c.description = 'WIP'
  c.option '--edl EDL', String, 'EDL file'
  c.option '--sheet CSV', String, 'Old Google Docs sheet'
  c.option '--frameio CSV', String, 'Frame.io comments export'
  c.option '--fps FPS', Integer, 'FPS to process EDL file with'
  c.action do |args, options|
    options.default \
      fps: 30

    crash("EDL is required") unless options.edl && File.exists?(options.edl)
    crash("Sheet is required") unless options.sheet && File.exists?(options.sheet)

    options.edl = File.expand_path(options.edl)
    options.sheet = File.expand_path(options.sheet)
    options.frameio = File.expand_path(options.frameio) if options.frameio && File.exists?(options.frameio)

    command_header(c, options)

    output = {}
    dupes = {}
    stub = {
      start_tc: nil,
      start_frame: nil,
      end_tc: nil,
      end_frame: nil,
      description: nil,
      notes: nil,
    }

    edl = EDL::Parser.new(fps=options.fps).parse(File.open(options.edl))
    edl.each do |evt|
      # if evt.num == "072"
      #   pp evt
      #   pp evt.comments
      #   pp evt.clip_name
      # end

      next if evt.track == "AA"

      # use TO NAME for transitions
      if evt.has_transition?
        evt.comments.each do |comment|
          next unless comment.start_with? "* TO CLIP NAME"
          evt.clip_name = comment.gsub("* TO CLIP NAME:", '').strip
          break
        end
      end

      next unless evt.clip_name

      clip_id = evt.clip_name.split('.')[0]

      if output[clip_id]
        dupes[clip_id] ||= 1
        dupes[clip_id] += 1

        clip_id = "#{clip_id}_dupe#{dupes[clip_id]}"
      end

      output[clip_id] = stub.clone

      output[clip_id][:start_tc] = evt.rec_start_tc.to_s
      output[clip_id][:start_frame] = evt.rec_start_tc.total
      output[clip_id][:end_tc] = evt.rec_end_tc.to_s
      output[clip_id][:end_frame] = evt.rec_end_tc.total
    end

    sheet = CSV.parse(File.read(options.sheet), headers: true)
    sheet.each do |row|
      shot = row.to_hash
      id = shot['Shot ID'].split('.')[0]

      # find existing via start_with
      unless output[id]
        output.each do |k,v|
          if k.start_with?(id)
            id = k
            break
          end
        end
      end

      unless output[id]
        output[id] = stub.clone
        puts "still didnt find entry for #{id}"
      end

      output[id][:description] = shot['Description']
      output[id][:notes] = shot['Notes']
    end

    if options.frameio
      frameio = File.open(options.frameio, "rb")
      comments = frameio.read.split("\n\n")

      comments.each_with_index do |comment, i|
        next if i == 0
        next if comment.start_with? "\t"

        meta,msg = comment.split("\n")
        meta = meta.split(' - ', 2)[1]
        
        tc,msg = msg.split(' - ', 2)

        frame = Timecode.parse(tc, fps = options.fps).total.to_i

        output.each do |id,shot|
          next if shot.nil? || shot[:start_frame].nil? || shot[:end_frame].nil?

          next if shot[:start_frame] > frame
          next if shot[:end_frame] < frame

          msg = "#{meta}\n#{msg}"

          if shot[:notes]
            shot[:notes] = "#{shot[:notes]}\n\n====\n\n#{msg}"
          else
            shot[:notes] = msg
          end
        end
      end
    end

    CSV.open("data.csv", "wb") do |csv|
      csv << ["Shot ID", "Start TC", "Start Frame", "End TC", "End Frame", "Description", "Notes"]
      
      output.each do |id, shot|
        csv << [
          "#{id}.mov",
          shot[:start_tc],
          shot[:start_frame],
          shot[:end_tc],
          shot[:end_frame],
          shot[:description],
          shot[:notes]
        ]
      end
    end

    # pp output
  end
end
default_command :run