#include "xsync.hpp"
#define ENABLE_LOG
#ifdef ENABLE_LOG
#include "../src/logger.hpp"
#endif

#define TAG "XSYNC"

using namespace iflytop;

static uint32_t ipToUint32(const std::string &ipAddress, bool &suc) {
  uint32_t           result = 0;
  std::istringstream iss(ipAddress);
  std::string        segment;
  int                i = 0;

  while (std::getline(iss, segment, '.')) {
    uint32_t octet = std::stoi(segment);
    if (octet > 255) {
      suc = false;
      return 0;
    }
    result |= (octet << ((3 - i) * 8));
    i++;
  }
  if (i != 4) {
    suc = false;
    return 0;
  }
  suc = true;

  uint32_t result_n = 0;
  result_n |= ((result & 0xff000000) >> 24);
  result_n |= ((result & 0x00ff0000) >> 8);
  result_n |= ((result & 0x0000ff00) << 8);
  result_n |= ((result & 0x000000ff) << 24);

  return result_n;
}

Xsync::Xsync(/* args */) {}

Xsync &Xsync::Ins() {
  static Xsync xsync;
  return xsync;
}
void Xsync::initialize(I_XSUDPFactory *xsync_udp_factory) { m_xsync_udp_factory = xsync_udp_factory; }

xs_error_code_t Xsync::connect(string xsync_ip) {
  lock_guard<recursive_mutex> lock(lock_);

  m_xsync_ip = xsync_ip;
  disConnect();
  /**
   * @brief 创建 m_xsync_reg_udp
   */

  xs_error_code_t ecode = kxs_ec_success;

  auto xsync_reg_udp = m_xsync_udp_factory->createXSUDP();
  ecode              = xsync_reg_udp->initialize("0.0.0.0", IFLYTOP_XSYNC_SERVICE_PC_PORT);
  if (ecode != kxs_ec_success) {
    return ecode;
  }
  /**
   * @brief 创建 m_xsync_timecode_udp_listener
   */
  auto xsync_timecode_udp_listener = m_xsync_udp_factory->createXSUDP();

  ecode = xsync_timecode_udp_listener->initialize("0.0.0.0", IFLYTOP_XSYNC_TIMECODE_REPORT_PC_PORT);
  if (ecode != kxs_ec_success) {
    return ecode;
  }
  ecode = xsync_timecode_udp_listener->startReceive([this](XsyncNetAdd &from, uint8_t *data, size_t length) { parseTimecodeMsgAndReport(from, data, length); });
  if (ecode != kxs_ec_success) {
    return ecode;
  }

  /**
   * @brief 创建 m_xsync_camera_sync_udp_listener
   */
  auto xsync_camera_sync_udp_listener = m_xsync_udp_factory->createXSUDP();

  ecode = xsync_camera_sync_udp_listener->initialize("0.0.0.0", IFLYTOP_XSYNC_CAMERA_SYNC_PACKET_PC_PORT);
  if (ecode != kxs_ec_success) {
    return ecode;
  }
  ecode = xsync_camera_sync_udp_listener->startReceive([this](XsyncNetAdd &from, uint8_t *data, size_t length) { parseCameraSyncMsgAndReport(from, data, length); });
  if (ecode != kxs_ec_success) {
    return ecode;
  }

  m_xsync_reg_udp                  = xsync_reg_udp;
  m_xsync_timecode_udp_listener    = xsync_timecode_udp_listener;
  m_xsync_camera_sync_udp_listener = xsync_camera_sync_udp_listener;

  m_net_state = kxsync_net_state_connected;
  return ecode;
}
xs_error_code_t Xsync::disConnect() {
  lock_guard<recursive_mutex> lock(lock_);

  if (m_xsync_reg_udp != nullptr) {
    m_xsync_reg_udp->stopReceive();
    m_xsync_reg_udp = nullptr;
  }

  if (m_xsync_timecode_udp_listener != nullptr) {
    m_xsync_timecode_udp_listener->stopReceive();
    m_xsync_timecode_udp_listener = nullptr;
  }

  if (m_xsync_camera_sync_udp_listener != nullptr) {
    m_xsync_camera_sync_udp_listener->stopReceive();
    m_xsync_camera_sync_udp_listener = nullptr;
  }

  m_net_state = kxsync_net_state_disconnect;
  return kxs_ec_success;
}
xsync_net_state_t Xsync::getNetState() { return m_net_state; }

void Xsync::regOnTimecodeMsg(xsync_on_timecode_msg_t on_timecode_msg_cb) { m_on_timecode_msg_cb = on_timecode_msg_cb; }
void Xsync::regOnCameraSyncMsg(xsync_on_camera_sync_msg_t on_camera_sync_msg_cb) { m_on_camera_sync_msg_cb = on_camera_sync_msg_cb; }

xs_error_code_t Xsync::xsync_send_cmd_block(iflytop_xsync_packet_header_t *cmd, iflytop_xsync_packet_header_t *rx_data, int32_t buffersize, int32_t overtime_ms) {
  lock_guard<recursive_mutex> lock(lock_);
  if (!m_xsync_reg_udp) return kxs_ec_lose_connect;
  m_xsync_reg_udp->clearRxBuffer();

  cmd->index = txpacket_index++;

  XsyncNetAdd     toadd = {m_xsync_ip, IFLYTOP_XSYNC_SERVICE_XSYNC_PORT};
  xs_error_code_t ecode =  //
      m_xsync_reg_udp->sendto(toadd, (const char *)cmd, sizeof(iflytop_xsync_packet_header_t) + cmd->ndata * 4, nullptr);
  if (ecode != kxs_ec_success) {
    return ecode;
  }

  XsyncNetAdd fromadd;
  while (true) {
    // ZLOGI(TAG, "start rx wait for rxdata");
    ecode = m_xsync_reg_udp->receive((char *)rx_data, buffersize, fromadd, overtime_ms);
    // ZLOGI(TAG, "end rx wait for rxdata");
    if (ecode != kxs_ec_success) {
      return ecode;
    }
    if (rx_data->index != cmd->index) {
      // ZLOGI(TAG, "packet index error %d %d", cmd->index, rx_data->index);
      continue;
    }
    break;
  }

  return (xs_error_code_t)rx_data->data[0];
}

xs_error_code_t Xsync::reg_write(uint32_t regadd, uint32_t regvalue, uint32_t &regbackvalue, int32_t overtime_ms) {
  /**
   * @brief
   *      协议说明
   *
   * kxsync_packet_type_reg_write
   *  tx: regadd,regdata
   *  rx: ecode,regdata
   */

  uint8_t txdata[128] = {0};
  uint8_t rxdata[128] = {0};

  iflytop_xsync_packet_header_t *txpacket = (iflytop_xsync_packet_header_t *)txdata;
  iflytop_xsync_packet_header_t *rxpacket = (iflytop_xsync_packet_header_t *)rxdata;

  txpacket->type    = kxsync_packet_type_cmd;
  txpacket->index   = txpacket_index++;
  txpacket->cmd     = kxsync_packet_type_reg_write;
  txpacket->ndata   = 2;
  txpacket->data[0] = regadd;
  txpacket->data[1] = regvalue;

  auto ecode = xsync_send_cmd_block(txpacket, rxpacket, sizeof(rxdata), overtime_ms);
  if (ecode != kxs_ec_success) {
    return ecode;
  }

  regbackvalue = rxpacket->data[1];
  return ecode;
}

xs_error_code_t Xsync::reg_read(uint32_t regadd, uint32_t &regvalue, int32_t overtime_ms) {
  /**
   * @brief
   *      协议说明
   *
   * kxsync_packet_type_reg_write
   *  tx: regadd,regdata
   *  rx: ecode,regdata
   */

  uint8_t txdata[128] = {0};
  uint8_t rxdata[128] = {0};

  iflytop_xsync_packet_header_t *txpacket = (iflytop_xsync_packet_header_t *)txdata;
  iflytop_xsync_packet_header_t *rxpacket = (iflytop_xsync_packet_header_t *)rxdata;

  txpacket->type    = kxsync_packet_type_cmd;
  txpacket->index   = txpacket_index++;
  txpacket->cmd     = kxsync_packet_type_reg_read;
  txpacket->ndata   = 2;
  txpacket->data[0] = regadd;
  txpacket->data[1] = regvalue;

  auto ecode = xsync_send_cmd_block(txpacket, rxpacket, sizeof(rxdata), overtime_ms);
  if (ecode != kxs_ec_success) {
    return ecode;
  }
  regvalue = rxpacket->data[1];
  return ecode;
}
xs_error_code_t Xsync::reg_read_muti(uint32_t regadd, uint32_t nreg, vector<uint32_t> &regvalues, int32_t overtime_ms) {
  /**
   * @brief
   *      协议说明
   *
   * kxsync_packet_type_reg_read_regs
   *  tx: regstartadd,nreg
   *  rx: ecode,regdatas
   */

  uint8_t txdata[128]  = {0};
  uint8_t rxdata[1280] = {0};

  iflytop_xsync_packet_header_t *txpacket = (iflytop_xsync_packet_header_t *)txdata;
  iflytop_xsync_packet_header_t *rxpacket = (iflytop_xsync_packet_header_t *)rxdata;

  txpacket->type    = kxsync_packet_type_cmd;
  txpacket->index   = txpacket_index++;
  txpacket->cmd     = kxsync_packet_type_reg_read_regs;
  txpacket->ndata   = 2;
  txpacket->data[0] = regadd;
  txpacket->data[1] = nreg;

  auto ecode = xsync_send_cmd_block(txpacket, rxpacket, sizeof(rxdata), overtime_ms);
  if (ecode != kxs_ec_success) {
    return ecode;
  }

  if (rxpacket->ndata > 0) {
    for (int i = 0; i < rxpacket->ndata - 1; i++) {
      regvalues.push_back(rxpacket->data[i + 1]);
    }
  }
  return ecode;
}

void Xsync::parseTimecodeMsgAndReport(XsyncNetAdd &from, uint8_t *data, size_t length) {
  //
  iflytop_timecode_report_packet_t *packet = (iflytop_timecode_report_packet_t *)data;
  xysnc_timecode_t                  timecode;
  /**
   * @brief
   *
   * 01 02 03 04 01 02 03 04
   *
   */

  uint8_t frameuints  = packet->timecode0 & 0x0f;
  uint8_t frame10s    = (packet->timecode0 >> 8) & 0x3;
  uint8_t seconduints = (packet->timecode0 >> 16) & 0x0f;
  uint8_t second10s   = (packet->timecode0 >> 24) & 0x03;

  uint8_t minuteuints = packet->timecode1 & 0x0f;
  uint8_t minute10s   = (packet->timecode1 >> 8) & 0x03;
  uint8_t houruints   = (packet->timecode1 >> 16) & 0x0f;
  uint8_t hour10s     = (packet->timecode1 >> 24) & 0x03;

  timecode.hour   = hour10s * 10 + houruints;
  timecode.minute = minute10s * 10 + minuteuints;
  timecode.second = second10s * 10 + seconduints;
  timecode.frame  = frame10s * 10 + frameuints;

  if (m_on_timecode_msg_cb) m_on_timecode_msg_cb(&timecode);
}
void Xsync::parseCameraSyncMsgAndReport(XsyncNetAdd &from, uint8_t *data, size_t length) {
  uint32_t count = 0;

  uint32_t data0 = data[7];
  uint32_t data1 = data[6];
  uint32_t data2 = data[5];
  uint32_t data3 = data[4];

  count = data0 + (data1 << 8) + (data2 << 16) + (data3 << 24);

  xysnc_camera_sync_data_t camera_sync_data;
  camera_sync_data.frameIndex = count;

  if (m_on_camera_sync_msg_cb) m_on_camera_sync_msg_cb(&camera_sync_data);
}

// xsync_stm32_action_generator_new_mac,  //
// xsync_stm32_action_factory_reset,      //
// xsync_stm32_action_reboot,             //
// xsync_stm32_action_storage_cfg,
xs_error_code_t Xsync::generatorNewMac() { return doaction(xsync_stm32_action_generator_new_mac, 0, nullptr, 2000); }
xs_error_code_t Xsync::factoryReset() { return doaction(xsync_stm32_action_factory_reset, 0, nullptr, 1000); }
xs_error_code_t Xsync::reboot() { return doaction(xsync_stm32_action_reboot, 0, nullptr); }
xs_error_code_t Xsync::storageConfig() { return doaction(xsync_stm32_action_storage_cfg, 0, nullptr, 1000); }
xs_error_code_t Xsync::changeNetworkConfig(string ip, string mask, string gateway) {
  uint32_t        ip32      = 0;
  uint32_t        mask32    = 0;
  uint32_t        gateway32 = 0;
  xs_error_code_t ecode;
  bool            suc = false;

  ip32 = (uint32_t)ipToUint32(ip.c_str(), suc);
  if (!suc) return kxs_ec_param_error;
  mask32 = (uint32_t)ipToUint32(mask.c_str(), suc);
  if (!suc) return kxs_ec_param_error;
  gateway32 = (uint32_t)ipToUint32(gateway.c_str(), suc);
  if (!suc) return kxs_ec_param_error;

  uint32_t readbak = 0;

  ecode = reg_write(kxsync_reg_stm32_ip, ip32, readbak);
  if (ecode != kxs_ec_success) return ecode;
  ecode = reg_write(kxsync_reg_stm32_netmask, mask32, readbak);
  if (ecode != kxs_ec_success) return ecode;
  ecode = reg_write(kxsync_reg_stm32_gw, gateway32, readbak);
  if (ecode != kxs_ec_success) return ecode;

  ecode = storageConfig();
  if (ecode != kxs_ec_success) return ecode;

  return kxs_ec_success;
}

xs_error_code_t Xsync::clearXsyncCameraSyncIndexCount() {
  uint32_t readbak = 0;
  return reg_write(kxsync_reg_stm32_camera_sync_signal_count, 0, readbak);
}

xs_error_code_t Xsync::doaction(uint32_t action, uint32_t actionval, uint32_t *ackreturn, int32_t overtime_ms) {
  //
  uint32_t        readbak = 0;
  xs_error_code_t ecode;
  ecode = reg_write(kxsync_reg_stm32_action_val0, actionval, readbak);
  if (ecode != kxs_ec_success) return ecode;

  ecode = reg_write(kxsync_reg_stm32_action0, action, readbak, overtime_ms);
  if (ecode != kxs_ec_success) return ecode;
  if (ackreturn) *ackreturn = readbak;
  return ecode;
}