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

          def handle_plays_and_events(data, **opts)
            things = (Array(data['events']) + Array(data['actions'])).group_by { |hash| hash['type'] }

            create_data(@plays_hash,  things['play'],  klass: Play,  api: api, drive: self) if things['play']
            create_data(@events_hash, things['event'], klass: Event, api: api, drive: self) if things['setup']
            create_data(@events_hash, things['event'], klass: Event, api: api, drive: self) if things['event']
          end

          def team_id
            plays.last&.start_situation&.team_id
          end

        end
      end
    end
  end
end