#include "MapemSpatemLayerFactory.hh"

#include "MapemSpatemTypes.hh"

#include "loggers.hh"

MapemSpatemLayer::MapemSpatemLayer(const std::string & p_type, const std::string & param) : TLayer<LibItsMapemSpatem__TestSystem::MapemSpatemPort>(p_type), _params(), _mapem_codec(), _spatem_codec() {
  loggers::get_instance().log(">>> MapemSpatemLayer::MapemSpatemLayer: %s, %s", to_string().c_str(), param.c_str());
  // Setup parameters
  Params::convert(_params, param);
}

void mapemLayer::sendMsg(const LibItsmapem__TestSystem::mapemReq& p, Params& params){
  loggers::get_instance().log_msg(">>> mapemLayer::sendMsg: ", p);

  // Encode mapem PDU
  OCTETSTRING data;
  if (_mapem_codec.encode(p.msgOut(), data) == -1) {
    loggers::get_instance().warning("mapemLayer::sendMsg: Encodeing failure");
    return;
  }
  // Update parameters
  params.insert(std::pair<std::string, std::string>(
                                                    Params::packetize,
                                                    std::string("")
                                                   ));
  sendData(data, params);
}

void spatemLayer::sendMsg(const LibItsspatem__TestSystem::spatemReq& p, Params& params){
  loggers::get_instance().log_msg(">>> spatemLayer::sendMsg: ", p);

  // Encode spatem PDU
  OCTETSTRING data;
  if (_spatem_codec.encode(p.msgOut(), data) == -1) {
    loggers::get_instance().warning("spatemLayer::sendMsg: Encodeing failure");
    return;
  }
  // Update parameters
  params.insert(std::pair<std::string, std::string>(
                                                  Params::packetize,
                                                  std::string("")
                                                  ));
  sendData(data, params);
}

void MapemSpatemLayer::sendData(OCTETSTRING& data, Params& params) {
  loggers::get_instance().log_msg(">>> MapemSpatemLayer::sendData: ", data);
  //params.log();
  sendToAllLayers(data, params);
}

void MapemSpatemLayer::receiveData(OCTETSTRING& data, Params& params)
{
  loggers::get_instance().log_msg(">>> MapemSpatemLayer::receiveData: ", data);
  
  // Sanity check
  if (*(static_cast<const unsigned char*>(data)+ 1) == 0x04) { // Check that received packet has MAPE message id 
    // Decode the MAPEM payload
    LibItsMapemSpatem__TestSystem::MapemInd p;
    _mapem_codec.decode(data, p.msgIn());
    if (!p.msgIn().is_bound()) {
      // Discard it
      return;
    } // else, continue
  } else if (*(static_cast<const unsigned char*>(data)+ 1) == 0x05) { // Check that received packet has SPATE message id 
    // Decode the SPATEM payload
    LibItsMapemSpatem__TestSystem::SpatemInd p;
    _spatem_codec.decode(data, p.msgIn());
    if (!p.msgIn().is_bound()) {
      // Discard it
      return;
    } // else, continue
  } else {
    // Not a DENM message, discard it
    loggers::get_instance().warning("MapemSpatemLayer::receiveData: Wrong message id: 0x%02x", *(static_cast<const unsigned char*>(data)+ 1));
    return;
  }
  
  // Process lower layer data
  // gnNextHeader
  Params::const_iterator it = params.find(Params::gn_next_header);
  if (it != params.cend()) {
    p.gnNextHeader() = std::stoi(it->second);
  } else {
    p.gnNextHeader().set_to_omit();
  }
  // gnHeaderType
  it = params.find(Params::gn_header_type);
  if (it != params.cend()) {
    p.gnHeaderType() = std::stoi(it->second);
  } else {
    p.gnHeaderType().set_to_omit();
  }
  // gnHeaderSubtype
  it = params.find(Params::gn_header_sub_type);
  if (it != params.cend()) {
    p.gnHeaderSubtype() = std::stoi(it->second);
  } else {
    p.gnHeaderSubtype().set_to_omit();
  }
  // gnLifetime
  it = params.find(Params::gn_lifetime);
  if (it != params.cend()) {
    p.gnLifetime() = std::stoi(it->second);
  } else {
    p.gnLifetime().set_to_omit();
  }
  // gnTrafficClass
  it = params.find(Params::gn_traffic_class);
  if (it != params.cend()) {
    p.gnTrafficClass() = std::stoi(it->second);
  } else {
    p.gnTrafficClass().set_to_omit();
  }
  // btpDestinationPort
  it = params.find(Params::btp_destination_port);
  if (it != params.cend()) {
    p.btpDestinationPort() = std::stoi(it->second);
  } else {
    p.btpDestinationPort().set_to_omit();
  }
  // btpInfo
  it = params.find(Params::btp_info);
  if (it != params.cend()) {
    p.btpInfo() = std::stoi(it->second);
  } else {
    p.btpInfo().set_to_omit();
  }
  // ssp
  it = params.find(Params::ssp);
  if (it != params.cend()) {
    OCTETSTRING os(it->second.length(), (const unsigned char *)(it->second.c_str()));
    p.ssp() = oct2bit(os);
  } else {
    p.ssp().set_to_omit();
  }
  // its_aid
  it = params.find(Params::its_aid);
  if (it != params.cend()) {
    p.its__aid() = std::stoi(it->second);
  } else {
    p.its__aid().set_to_omit();
  }
  
  // Pass it to the ports if amy
  toAllUpperPorts(p, params);
}

MapemSpatemLayerFactory MapemSpatemLayerFactory::_f;