claudiob/yt-annotations

require 'yt/annotations/base'

module Yt
  module Annotations
    # An end screen annotation shows in the last 5 to seconds of the video.
    class EndScreen < Base
      # @param [Hash] json the Hash representation of the XML data returned by
      #   YouTube for each end screen of a video.
      def initialize(json = {})
        @text = text_in json
        @starts_at = json['startMs'].to_i / 1000.0
        @ends_at = ends_at_in json
        @link = to_link json
      end

    private

      def text_in(json)
        json['title']['simpleText']
      end

      def ends_at_in(json)
        json['endMs'].to_i / 1000.0
      end

      def to_link(json)
        target_url = case json['style']
          when 'WEBSITE'
            json['endpoint']['urlEndpoint']['url']
          when 'PLAYLIST'
            if json['endpoint']['watchEndpoint']
              "https://www.youtube.com/watch?v=" +
              json['endpoint']['watchEndpoint']['videoId'] +
              "&list=" +
              json['endpoint']['watchEndpoint']['playlistId']
            else
              "https://www.youtube.com" + json['endpoint']['urlEndpoint']['url']
            end
          when 'VIDEO'
            if json['endpoint']['watchEndpoint']
              "https://www.youtube.com/watch?v=" +
              json['endpoint']['watchEndpoint']['videoId']
            else
              "https://www.youtube.com" + json['endpoint']['urlEndpoint']['url']
            end
          when 'CHANNEL'
            if json['isSubscribe']
              "https://www.youtube.com/channel/" +
              json['hovercardButton']['subscribeButtonRenderer']['channelId']
            else
              "https://www.youtube.com/channel/" +
              json['endpoint']['browseEndpoint']["browseId"]
            end
          end
        {
          url: target_url, new_window: new_window(json['style']),
          type: link_type(json)
        }
      end

      def link_type(json)
        case json['style']
          when 'WEBSITE' then :website
          when 'PLAYLIST' then :playlist
          when 'VIDEO' then :video
          when 'CHANNEL' then (json['isSubscribe'] ? :subscribe : :channel)
        end
      end

      def new_window(type)
        %w(WEBSITE CHANNEL).include? type
      end
    end
  end
end