/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this file,
 * You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef DOM_MEDIA_WEBRTC_SDP_SIPCCSDPMEDIASECTION_H_
#define DOM_MEDIA_WEBRTC_SDP_SIPCCSDPMEDIASECTION_H_

#include <map>

#include "mozilla/UniquePtr.h"
#include "sdp/SdpMediaSection.h"
#include "sdp/SipccSdpAttributeList.h"

extern "C" {
#include "sipcc_sdp.h"
}

namespace mozilla {

class SipccSdp;
class SdpParser;

using InternalResults = SdpParser::InternalResults;

class SipccSdpBandwidths final : public std::map<std::string, uint32_t> {
 public:
  bool Load(sdp_t* sdp, const uint16_t level, InternalResults& results);
  void Serialize(std::ostream& os) const;
};

class SipccSdpMediaSection final : public SdpMediaSection {
  friend class SipccSdp;

 public:
  ~SipccSdpMediaSection() = default;

  virtual MediaType GetMediaType() const override { return mMediaType; }

  virtual unsigned int GetPort() const override;
  virtual void SetPort(const unsigned int port) override;
  virtual unsigned int GetPortCount() const override;
  virtual Protocol GetProtocol() const override;
  virtual const SdpConnection& GetConnection() const override;
  virtual SdpConnection& GetConnection() override;
  virtual uint32_t GetBandwidth(const std::string& type) const override;
  virtual const std::vector<std::string>& GetFormats() const override;

  virtual const SdpAttributeList& GetAttributeList() const override;
  virtual SdpAttributeList& GetAttributeList() override;
  virtual SdpDirectionAttribute GetDirectionAttribute() const override;

  virtual void AddCodec(const std::string& pt, const std::string& name,
                        const uint32_t clockrate,
                        const uint16_t channels) override;
  virtual void ClearCodecs() override;

  virtual void AddDataChannel(const std::string& name, const uint16_t port,
                              const uint16_t streams,
                              const uint32_t message_size) override;

  virtual void Serialize(std::ostream&) const override;

 private:
  SipccSdpMediaSection(const size_t level,
                       const SipccSdpAttributeList* sessionLevel)
      : SdpMediaSection(level),
        mMediaType(static_cast<MediaType>(0)),
        mPort(0),
        mPortCount(0),
        mProtocol(static_cast<Protocol>(0)),
        mAttributeList(sessionLevel) {}

  SipccSdpMediaSection(const SipccSdpMediaSection& aOrig,
                       const SipccSdpAttributeList* sessionLevel);

  bool Load(sdp_t* sdp, const uint16_t level, InternalResults& results);
  bool LoadConnection(sdp_t* sdp, uint16_t level, InternalResults& results);
  bool LoadProtocol(sdp_t* sdp, const uint16_t level, InternalResults& results);
  bool LoadFormats(sdp_t* sdp, const uint16_t level, InternalResults& results);
  bool ValidateSimulcast(sdp_t* sdp, const uint16_t level,
                         InternalResults& results) const;
  bool ValidateSimulcastVersions(
      sdp_t* sdp, const uint16_t level,
      const SdpSimulcastAttribute::Versions& versions,
      const sdp::Direction direction, InternalResults& results) const;

  // the following values are cached on first get
  MediaType mMediaType;
  uint16_t mPort;
  uint16_t mPortCount;
  Protocol mProtocol;
  std::vector<std::string> mFormats;

  UniquePtr<SdpConnection> mConnection;
  SipccSdpBandwidths mBandwidths;

  SipccSdpAttributeList mAttributeList;
};
}  // namespace mozilla

#endif