/**
 *
 * \section COPYRIGHT
 *
 * Copyright 2013-2021 Software Radio Systems Limited
 *
 * By using this file, you agree to the terms and conditions set
 * forth in the LICENSE file which can be found at the top level of
 * the distribution.
 *
 */

#ifndef SRSUE_TTCN3_UT_INTERFACE_H
#define SRSUE_TTCN3_UT_INTERFACE_H

#include "rapidjson/document.h"
#include "rapidjson/prettywriter.h"
#include "srsran/common/netsource_handler.h"
#include "ttcn3_interfaces.h"

using namespace rapidjson;

// The UpperTester interface
class ttcn3_ut_interface : public ttcn3_port_handler
{
public:
  explicit ttcn3_ut_interface(srslog::basic_logger& logger) : ttcn3_port_handler(logger) {}
  ~ttcn3_ut_interface(){};

  int init(ss_ut_interface* syssim_, std::string net_ip_, uint32_t net_port_)
  {
    syssim      = syssim_;
    net_ip      = net_ip_;
    net_port    = net_port_;
    initialized = true;

    // create socket and
    return port_listen();
  }

private:
  int handle_message(const unique_byte_array_t& rx_buf, const uint32_t n)
  {
    Document document;
    if (document.Parse((char*)rx_buf->begin()).HasParseError() || document.IsObject() == false) {
      logger.error(rx_buf->begin(), n, "Error parsing incoming data.");
      return SRSRAN_ERROR;
    }

    // Pretty-print
    StringBuffer               buffer;
    PrettyWriter<StringBuffer> writer(buffer);
    document.Accept(writer);
    logger.info("Received %d bytes\n%s.", n, (char*)buffer.GetString());

    // check for command
    assert(document.HasMember("Cmd"));
    assert(document["Cmd"].IsObject());

    // get Cmd
    const Value& a = document["Cmd"];

    if (a.HasMember("MMI")) {
      assert(a.HasMember("MMI"));

      // get MMI and make sure it has another Cmd nested
      const Value& mmi = a["MMI"];
      assert(mmi.HasMember("Cmd"));

      // get MMI cmd
      const Value& mmi_cmd = mmi["Cmd"];
      assert(mmi_cmd.IsString());

      // check for CnfRequired
      assert(document.HasMember("CnfRequired"));

      if (strcmp(mmi_cmd.GetString(), "POWER_OFF") == 0) {
        logger.info("Received POWER_OFF command.");
        handle_power_off(document);
      } else if (strcmp(mmi_cmd.GetString(), "SWITCH_ON") == 0) {
        logger.info("Received SWITCH_ON command.");
        syssim->switch_on_ue();
      } else if (strcmp(mmi_cmd.GetString(), "SWITCH_OFF") == 0) {
        logger.info("Received SWITCH_OFF command.");
        syssim->switch_off_ue();
      } else {
        logger.error("Received unknown command: %s", mmi_cmd.GetString());
      }
    } else if (a.HasMember("AT")) {
      handle_at_command(document);
    } else if (a.HasMember("TC_START")) {
      logger.info("Received TC_START command.");
      const Value& cmd = a["TC_START"];
      assert(cmd.HasMember("Name"));
      const Value& tc_name = cmd["Name"];
      syssim->tc_start(tc_name.GetString());
    } else if (a.HasMember("TC_END")) {
      logger.info("Received TC_END command.");
      syssim->tc_end();
    } else {
      logger.error("Unknown command type.");
    }

    return SRSRAN_SUCCESS;
  }

  void handle_power_off(Document& document)
  {
    syssim->power_off_ue();

    // Create response
    Document resp;
    resp.SetObject();
    resp.AddMember("Result", true, resp.GetAllocator());

    // Serialize and send to tester
    StringBuffer         buffer;
    Writer<StringBuffer> writer(buffer);
    resp.Accept(writer);

    logger.info("Sending %s to tester (%zd B)", buffer.GetString(), buffer.GetSize());
    send((const uint8_t*)buffer.GetString(), buffer.GetSize());
  }

  void handle_at_command(Document& document)
  {
    // We can assume the doc contains a AT CMD
    const Value& at = document["Cmd"]["AT"];

    // turn off data services
    if (std::string(at.GetString()) == "AT+CGATT=0<CR>") {
      logger.info("Disabling data services");
      syssim->disable_data();
    } else if (std::string(at.GetString()) == "AT+CGATT=1<CR>") {
      logger.info("Enabling data services");
      syssim->enable_data();
    } else {
      logger.error("Not handling AT command %s", at.GetString());
    }
  }

  ss_ut_interface* syssim = nullptr;
};

#endif // SRSUE_TTCN3_UT_INTERFACE_H