fosdem-recorder.rb/lib/fosdem_recorder/cli.rb

# frozen_string_literal: true

# Cli part of FosdemRecorder
module FosdemRecorder
  # Fosdem Cli - Download and cut streams video
  class Cli < Thor
    desc 'info URL', 'Get information about URL'
    def info(url)
      meta = _get_meta(url)
      puts meta[:title].green
      puts "* event start  = #{meta[:event_start]}"
      puts "* event stop   = #{meta[:event_stop]}"
      puts "* event length = #{meta[:duration_str]}"
      puts "* stream url   = #{meta[:stream_url]}"
    end

    desc 'download URL', 'Download conference described at URL'
    def download(url)
      meta = _get_meta(url)

      localtime = meta[:event_start].localtime
      timeformat = format(
        '%<hours>02d:%<minutes>02d %<year>d-%<month>02d-%<day>02d',
        hours: localtime.hour,
        minutes: localtime.min,
        year: localtime.year,
        month: localtime.month,
        day: localtime.day
      )
      cmd = "echo ffmpeg -i #{meta[:stream_url]} -c copy -t #{meta[:duration_str]} \"#{meta[:title_sane]}.mp4\" | at #{timeformat}"
      puts "Command: #{cmd}".yellow
      system cmd
    end

    private

    def _validate_url(url)
      return if url =~ %r{^https://fosdem.org/\d+/schedule/event/.*}

      if url =~ %r{^https://fosdem.org/.*}
        warn 'ERROR: not a schedule page. URL must contain .../schedule/event/...'
        exit 1
      end

      warn 'ERROR: not a fosdem stream. URL must start with https://fosdem.org/...'
      exit 1
    end

    def _get_meta(url)
      puts "Loading data from #{url}".yellow
      mechanize = Mechanize.new
      page = mechanize.get(url)

      # body_class = page.at('body').attr('class')
      # if body_class != 'schedule-event'
      #   STDERR.puts "ERROR: Not an event schedule page!"
      #   exit 1
      # end

      puts body_class

      title = page.title
      title_sane =
        title
        .gsub(/[^a-zA-Z0-9]/, '-')
        .gsub(/--*/, '-')
        .gsub(/-$/, '')
        .gsub(/^-/, '')

      play_start_str =
        page
        .at('.side-box .icon-play')
        .parent
        .at('.value-title')
        .attr('title')
        .strip

      play_start = Time.parse(play_start_str)

      play_stop_str =
        page
        .at('.side-box .icon-stop')
        .parent
        .at('.value-title')
        .attr('title')
        .strip

      play_stop = Time.parse(play_stop_str)

      duration = (play_stop - play_start) / 3600
      duration_h = duration.to_i
      duration_m = ((duration - duration_h) * 60).to_i
      duration_str = format('%<hours>02d:%<minutes>02d:00', hours: duration_h, minutes: duration_m)

      stream_page =
        page
        .links
        .select { |link| link.href =~ /live.fosdem.org/ }
        .first
        .href

      stream_url =
        stream_page
        .gsub(%r{.*watch/}, 'https://stream.fosdem.org/')
        .gsub(/$/, '.m3u8')

      {
        title: title,
        title_sane: title_sane,
        stream_url: stream_url,
        event_start: play_start,
        event_stop: play_stop,
        event_start_str: play_start_str,
        event_stop_str: play_stop_str,
        duration_str: duration_str
      }
    end
  end
end