/*
 * @Author: DI JUNKUN
 * @Date: 2025-01-03
 * Copyright (c) 2025 by DI JUNKUN, All Rights Reserved.
 */

#ifndef _DATA_CHANNEL_RECEIVE_H_
#define _DATA_CHANNEL_RECEIVE_H_

#include "ice_agent.h"
#include "rtp_data_receiver.h"

class DataChannelReceive {
 public:
  DataChannelReceive();
  DataChannelReceive(std::shared_ptr<IceAgent> ice_agent,
                     std::shared_ptr<IOStatistics> ice_io_statistics,
                     std::function<void(const char *, size_t)> on_receive_data);
  ~DataChannelReceive();

 public:
  void Initialize(rtp::PAYLOAD_TYPE payload_type);
  void Destroy();

  uint32_t GetSsrc() {
    if (rtp_data_receiver_) {
      return rtp_data_receiver_->GetSsrc();
    }
    return 0;
  }

  uint32_t GetRemoteSsrc() {
    if (rtp_data_receiver_) {
      return rtp_data_receiver_->GetRemoteSsrc();
    }
    return 0;
  }

  int OnReceiveRtpPacket(const char *data, size_t size);

  void OnSenderReport(int64_t now_time, uint64_t ntp_time) {
    if (rtp_data_receiver_) {
      rtp_data_receiver_->OnSenderReport(now_time, ntp_time);
    }
  }

 private:
  std::shared_ptr<IceAgent> ice_agent_ = nullptr;
  std::shared_ptr<IOStatistics> ice_io_statistics_ = nullptr;
  std::unique_ptr<RtpDataReceiver> rtp_data_receiver_ = nullptr;
  std::function<void(const char *, size_t)> on_receive_data_ = nullptr;
};

#endif