/**
 * Copyright 2013-2021 Software Radio Systems Limited
 *
 * This file is part of srsRAN.
 *
 * srsRAN is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of
 * the License, or (at your option) any later version.
 *
 * srsRAN is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * A copy of the GNU Affero General Public License can be found in
 * the LICENSE file in the top-level directory of this distribution
 * and at http://www.gnu.org/licenses/.
 *
 */

#include "srsran/interfaces/gnb_interfaces.h"
#include "srsran/interfaces/ue_gw_interfaces.h"
#include "srsran/interfaces/ue_rlc_interfaces.h"
#include "srsran/upper/pdcp.h"
#include <map>

#ifndef SRSENB_PDCP_NR_H
#define SRSENB_PDCP_NR_H

namespace srsenb {

struct pdcp_nr_args_t {
  std::string log_level;
  uint32_t    log_hex_limit;
};

class pdcp_nr : public pdcp_interface_rlc_nr, public pdcp_interface_sdap_nr, public pdcp_interface_rrc_nr
{
public:
  explicit pdcp_nr(srsran::task_sched_handle task_sched_, const char* logname);
  virtual ~pdcp_nr() = default;
  void init(const pdcp_nr_args_t&   args_,
            rlc_interface_pdcp_nr*  rlc_,
            rrc_interface_pdcp_nr*  rrc_,
            sdap_interface_pdcp_nr* gtpu_);
  void stop();

  // pdcp_interface_rlc_nr
  void write_pdu(uint16_t rnti, uint32_t lcid, srsran::unique_byte_buffer_t sdu);
  void notify_delivery(uint16_t rnti, uint32_t lcid, const srsran::pdcp_sn_vector_t& pdcp_sn);
  void notify_failure(uint16_t rnti, uint32_t lcid, const srsran::pdcp_sn_vector_t& pdcp_sn);
  void write_pdu_mch(uint32_t lcid, srsran::unique_byte_buffer_t sdu) {}

  // pdcp_interface_rrc_nr
  void reset(uint16_t rnti) final;
  void add_user(uint16_t rnti) final;
  void rem_user(uint16_t rnti) final;
  void write_sdu(uint16_t rnti, uint32_t lcid, srsran::unique_byte_buffer_t sdu) final;
  void add_bearer(uint16_t rnti, uint32_t lcid, srsran::pdcp_config_t cnfg) final;
  void config_security(uint16_t rnti, uint32_t lcid, srsran::as_security_config_t sec_cfg) final;
  void enable_integrity(uint16_t rnti, uint32_t lcid) final;
  void enable_encryption(uint16_t rnti, uint32_t lcid) final;

private:
  class user_interface_rlc final : public srsue::rlc_interface_pdcp
  {
  public:
    uint16_t                       rnti;
    srsenb::rlc_interface_pdcp_nr* rlc;
    // rlc_interface_pdcp_nr
    void write_sdu(uint32_t lcid, srsran::unique_byte_buffer_t sdu) final;
    void discard_sdu(uint32_t lcid, uint32_t discard_sn) final;
    bool rb_is_um(uint32_t lcid) final;
    bool sdu_queue_is_full(uint32_t lcid) final;
  };

  class user_interface_sdap : public srsue::gw_interface_pdcp
  {
  public:
    uint16_t                        rnti;
    srsenb::sdap_interface_pdcp_nr* sdap;
    // gw_interface_pdcp
    void write_pdu(uint32_t lcid, srsran::unique_byte_buffer_t pdu) final;
    void write_pdu_mch(uint32_t lcid, srsran::unique_byte_buffer_t sdu) final {}
  };

  class user_interface_rrc : public srsue::rrc_interface_pdcp
  {
  public:
    uint16_t                       rnti;
    srsenb::rrc_interface_pdcp_nr* rrc;
    // rrc_interface_pdcp_nr
    void        write_pdu(uint32_t lcid, srsran::unique_byte_buffer_t pdu) final;
    void        write_pdu_bcch_bch(srsran::unique_byte_buffer_t pdu) final;
    void        write_pdu_bcch_dlsch(srsran::unique_byte_buffer_t pdu) final;
    void        write_pdu_pcch(srsran::unique_byte_buffer_t pdu) final;
    void        write_pdu_mch(uint32_t lcid, srsran::unique_byte_buffer_t pdu) final {}
    void        notify_pdcp_integrity_error(uint32_t lcid) final {}
    const char* get_rb_name(uint32_t lcid) final;
  };

  class user_interface
  {
  public:
    user_interface_rlc            rlc_itf;
    user_interface_sdap           sdap_itf;
    user_interface_rrc            rrc_itf;
    std::unique_ptr<srsran::pdcp> pdcp;
  };

  // args
  pdcp_nr_args_t          m_args = {};
  rlc_interface_pdcp_nr*  m_rlc  = nullptr;
  rrc_interface_pdcp_nr*  m_rrc  = nullptr;
  sdap_interface_pdcp_nr* m_sdap = nullptr;

  std::map<uint32_t, user_interface> users;

  srsran::task_sched_handle task_sched;
  srslog::basic_logger&     logger;
};

} // namespace srsenb

#endif // SRSENB_PDCP_NR_H