module Sportradar
  module Api
    module Football
      class Ncaafb
        class Drive < Sportradar::Api::Football::Drive

          def over?
            plays.last&.parsed_ending
          end

          def handle_plays_and_events(data, **opts)
            create_data(@plays_hash,  data['actions'],  klass: Play,  api: api, game: self) if data['actions']
            create_data(@events_hash, data['events'],   klass: Event, api: api, game: self) if data['events']
          end

          def play_count
            plays.count { |play| ['rush', 'pass'].include? play.play_type }
          end

          def gain
            Array(plays.select { |play| ['rush', 'pass'].include? play.play_type }).sum do |play|
              Array(play.players).sum do |p|
                p.dig('passing', 'yds').to_i + p.dig('rushing', 'yds').to_i
              end
            end
          end

        end
      end
    end
  end
end