#include "LibSip_Interface.hh"
#include "LibSip_SIPTypesAndValues.hh"

#include "diameter_layer_factory.hh"

#include "loggers.hh"

#include "converter.hh"

diameter_layer::diameter_layer(const std::string & p_type, const std::string & p_param) : t_layer<LibDiameter__Interface::DiameterPort>(p_type), _params(), _codec_request(), _codec_response(), _codec_register(), _codec_invite(), _codec_ack(), _codec_subscribe(), _codec_message(), _codec_options(), _codec_cancel(), _codec_notify(), _codec_info(), _codec_bye() {
  loggers::get_instance().log(">>> diameter_layer::diameter_layer: %s, %s", to_string().c_str(), p_param.c_str());
  // Setup parameters
  params::convert(_params, p_param);
}

void diameter_layer::sendMsg(const LibDiameter__SIPTypesAndValues::REGISTER__Request& p_diameter_message, params& p_param) {
  loggers::get_instance().log_msg(">>> diameter_layer::sendMsg: ", p_diameter_message);

  // Encode SipMessage
  OCTETSTRING data;
  _codec_register.encode(p_diameter_message, data);
  send_data(data, _params);
}

void diameter_layer::sendMsg(const LibDiameter__SIPTypesAndValues::INVITE__Request& p_diameter_message, params& p_param) {
  loggers::get_instance().log_msg(">>> diameter_layer::sendMsg: ", p_diameter_message);

  // Encode SipMessage
  OCTETSTRING data;
  _codec_invite.encode(p_diameter_message, data);
  send_data(data, _params);
}

void diameter_layer::sendMsg(const LibDiameter__SIPTypesAndValues::ACK__Request& p_diameter_message, params& p_param) {
  loggers::get_instance().log_msg(">>> diameter_layer::sendMsg: ", p_diameter_message);

  // Encode SipMessage
  OCTETSTRING data;
  _codec_ack.encode(p_diameter_message, data);
  send_data(data, _params);
}

void diameter_layer::sendMsg(const LibDiameter__SIPTypesAndValues::SUBSCRIBE__Request& p_diameter_message, params& p_param) {
  loggers::get_instance().log_msg(">>> diameter_layer::sendMsg: ", p_diameter_message);

  // Encode SipMessage
  OCTETSTRING data;
  _codec_subscribe.encode(p_diameter_message, data);
  send_data(data, _params);
}

void diameter_layer::sendMsg(const LibDiameter__SIPTypesAndValues::MESSAGE__Request& p_diameter_message, params& p_param) {
  loggers::get_instance().log_msg(">>> diameter_layer::sendMsg: ", p_diameter_message);

  // Encode SipMessage
  OCTETSTRING data;
  _codec_message.encode(p_diameter_message, data);
  send_data(data, _params);
}

void diameter_layer::sendMsg(const LibDiameter__SIPTypesAndValues::OPTIONS__Request& p_diameter_message, params& p_param) {
  loggers::get_instance().log_msg(">>> diameter_layer::sendMsg: ", p_diameter_message);

  // Encode SipMessage
  OCTETSTRING data;
  _codec_options.encode(p_diameter_message, data);
  send_data(data, _params);
}

void diameter_layer::sendMsg(const LibDiameter__SIPTypesAndValues::BYE__Request& p_diameter_message, params& p_param) {
  loggers::get_instance().log_msg(">>> diameter_layer::sendMsg: ", p_diameter_message);

  // Encode SipMessage
  OCTETSTRING data;
  _codec_bye.encode(p_diameter_message, data);
  send_data(data, _params);
}

void diameter_layer::sendMsg(const LibDiameter__SIPTypesAndValues::CANCEL__Request& p_diameter_message, params& p_param) {
  loggers::get_instance().log_msg(">>> diameter_layer::sendMsg: ", p_diameter_message);

  // Encode SipMessage
  OCTETSTRING data;
  _codec_cancel.encode(p_diameter_message, data);
  send_data(data, _params);
}

void diameter_layer::sendMsg(const LibDiameter__SIPTypesAndValues::NOTIFY__Request& p_diameter_message, params& p_param) {
  loggers::get_instance().log_msg(">>> diameter_layer::sendMsg: ", p_diameter_message);

  // Encode SipMessage
  OCTETSTRING data;
  _codec_notify.encode(p_diameter_message, data);
  send_data(data, _params);
}

void diameter_layer::sendMsg(const LibDiameter__SIPTypesAndValues::INFO__Request& p_diameter_message, params& p_param) {
  loggers::get_instance().log_msg(">>> diameter_layer::sendMsg: ", p_diameter_message);

  // Encode SipMessage
  OCTETSTRING data;
  _codec_info.encode(p_diameter_message, data);
  send_data(data, _params);
}

void diameter_layer::sendMsg(const LibDiameter__SIPTypesAndValues::Response& p_diameter_message, params& p_param) {
  loggers::get_instance().log_msg(">>> diameter_layer::sendMsg: ", p_diameter_message);

  // Encode SipMessage
  OCTETSTRING data;
  _codec_response.encode(p_diameter_message, data);
  send_data(data, _params);
}

void diameter_layer::send_data(OCTETSTRING& data, params& p_params) {
  loggers::get_instance().log_msg(">>> diameter_layer::send_data: ", data);

  send_to_all_layers(data, p_params);
}

void diameter_layer::receive_data(OCTETSTRING& data, params& p_params)
{
  loggers::get_instance().log_msg(">>> diameter_layer::receive_data: ", data);

  // Decode SIP message
  if ((data[0].get_octet() == 'S') && (data[1].get_octet() == 'I') && (data[2].get_octet() == 'P')) { // SIP... Assume it is a responce
    LibDiameter__SIPTypesAndValues::Response diameter_response;
    if (_codec_response.decode(data, diameter_response) == -1) {
      loggers::get_instance().warning("diameter_layer::receive_data: Failed to decode response data");
      return;
    }
    // Pass it to the ports
    to_all_upper_ports(diameter_response, p_params);
  } else { // Assume it is a request
    LibDiameter__SIPTypesAndValues::Request diameter_request;
    if (_codec_request.decode(data, diameter_request) == -1) {
      loggers::get_instance().warning("diameter_layer::receive_data: Failed to decode request data");
      return;
    }
    // Pass it to the ports
    to_all_upper_ports(diameter_request, p_params);
  }
}

diameter_layer_factory diameter_layer_factory::_f;