// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0

use s2n_quic_core::{
    frame::{ack_elicitation::AckElicitation, path_validation, FrameTrait},
    inet::DatagramInfo,
    packet::number::PacketNumber,
};

/// Tracks information about a packet that has been processed
#[derive(Clone, Copy, Debug)]
pub struct ProcessedPacket<'a> {
    pub(crate) packet_number: PacketNumber,
    pub(crate) datagram: &'a DatagramInfo,
    pub(crate) ack_elicitation: AckElicitation,
    pub(crate) path_challenge_on_active_path: bool,
    pub(crate) frames: usize,
    pub(crate) path_validation_probing: path_validation::Probe,
    pub(crate) bytes_progressed: usize,
}

impl<'a> ProcessedPacket<'a> {
    /// Creates a processed packet tracker
    pub fn new(packet_number: PacketNumber, datagram: &'a DatagramInfo) -> Self {
        Self {
            packet_number,
            datagram,
            ack_elicitation: AckElicitation::default(),
            path_challenge_on_active_path: false,
            frames: 0,
            path_validation_probing: path_validation::Probe::default(),
            bytes_progressed: 0,
        }
    }

    /// Records information about a processed frame
    pub fn on_processed_frame<F: FrameTrait>(&mut self, frame: &F) {
        self.ack_elicitation |= frame.ack_elicitation();
        self.frames += 1;
        self.path_validation_probing |= frame.path_validation();
    }

    /// Returns `true` if any of the processed frames are ack eliciting
    pub fn is_ack_eliciting(&self) -> bool {
        self.ack_elicitation.is_ack_eliciting()
    }
}