//| Copyright: (C) 2020-2024 Kevin Larke <contact AT larke DOT org> 
//| License: GNU GPL version 3.0 or above. See the accompanying LICENSE file.
#include "cwCommon.h"
#include "cwLog.h"
#include "cwCommonImpl.h"
#include "cwTest.h"
#include "cwMem.h"
#include "cwTime.h"

#include "cwThread.h"
#include "cwTcpSocket.h"
#include "cwTcpSocketSrv.h"

#include "cwDnsSd.h"
#include "cwUtility.h"

#include "dns_sd/rpt.h"
#include "dns_sd/dns_sd.h"
#include "dns_sd/fader.h"

#define   MDNS_PORT 5353
#define   MDNS_IP   "224.0.0.251"


namespace cw
{
  namespace net
  {
    namespace dnssd
    {
      typedef struct text_str
      {
        char*            text;
        struct text_str* link;
      } text_t;

      typedef struct dnssd_str
      {
        srv::handle_t udpH;
        srv::handle_t tcpH;
        unsigned      udpRecvBufByteN;
        unsigned      tcpRecvBufByteN;
        text_t*       textL;
        dns_sd*       dnsSd;
        fader*        fdr;

        char*         serviceName;
        char*         serviceType;
        char*         serviceDomain;
        char*         hostName;
        char*         hostIpAddr;
        uint16_t      hostPort;
        unsigned char hostMac[6];
        unsigned      ticksPerHeartBeat;

        unsigned cbCnt;
        time::spec_t t0;
        
      } dnssd_t;

      void print_callback( const char* text )
      {
        printf("%s",text);
      }
      
      inline dnssd_t* _handleToPtr( handle_t h )
      {
        return handleToPtr<handle_t,dnssd_t>(h);
      }

      

      rc_t _destroy( dnssd_t* p )
      {
        rc_t rc = kOkRC;

        mem::release(p->serviceName);
        mem::release(p->serviceType);
        mem::release(p->serviceDomain);
        mem::release(p->hostName);
        mem::release(p->hostIpAddr);
        

        text_t* t0 = p->textL;
        while( t0 != nullptr )
        {
          text_t* t1 = t0->link;
          mem::release(t0->text);
          mem::release(t0);
          t0 = t1;
        }

        srv::destroy( p->tcpH );
        srv::destroy( p->udpH );

        if( p->fdr != nullptr )
        {
          delete p->fdr;
          p->fdr = nullptr;
        }
        
        if( p->dnsSd != nullptr )
        {
          delete p->dnsSd;
          p->dnsSd = nullptr;
        }
        
        if( p->dnsSd != nullptr )
          delete p->dnsSd;
        
        mem::release(p);
        return rc;
      }

      // Called by 'dns_sd' to send UDP messages.
      void udpSendCallback( void* arg, const void* buf, unsigned bufByteN )
      {
        rc_t rc;
        dnssd_t* p = (dnssd_t*)arg;
        //printHex(buf,bufByteN);
        if((rc = srv::send(p->udpH,buf,bufByteN,MDNS_IP,MDNS_PORT)) != kOkRC )
          cwLogError(rc,"UDP send failed.");
      }

      // Called by 'fader' to send TCP messages.
      void tcpSendCallback( void* arg, const void* buf, unsigned bufByteN )
      {
        rc_t rc;

        printf("Send:%i\n",bufByteN);
        
        dnssd_t* p = (dnssd_t*)arg;
        if((rc = srv::send(p->tcpH,buf,bufByteN)) != kOkRC )
          cwLogError(rc,"TCP send failed.");
      }

      // Called by UDP socket with incoming MDNS data.
      void udpReceiveCallback( void* arg, const void* data, unsigned dataByteCnt, const struct sockaddr_in* fromAddr )
      {
        dnssd_t* p = static_cast<dnssd_t*>(arg);
        p->dnsSd->receive(data,dataByteCnt);
      }

      // Called by TCP socket with incoming EuCon data
      void tcpReceiveCallback( void* arg, const void* data, unsigned dataByteCnt, const struct sockaddr_in* fromAddr )
      {
        dnssd_t* p = static_cast<dnssd_t*>(arg);

        if( dataByteCnt > 0 )
        {
          p->fdr->receive_from_eucon( data, dataByteCnt );  
        }

        time::spec_t t1;
        time::get(t1);
        unsigned ms = time::elapsedMs( p->t0, t1 );
        if( ms > 50 )
        {
          p->cbCnt+=1;
          p->fdr->tick();
          p->t0 = t1;
        }

        if( p->cbCnt > 20 )
        {
          printf(".");
          fflush(stdout);
          p->cbCnt = 1;
        }
      }

      // Called by 'fader' to send messages to change the state of the physical control.
      void physControlCallback( void* arg, const uint8_t* buf, uint8_t bufByteN )
      {
      }


      void _pushTextRecdField( dnssd_t* p, const char* text )
      {
        text_t* t = mem::allocZ<text_t>();
        t->text   = mem::duplStr(text);
        t->link   = p->textL;
        p->textL  = t;            
      }

      rc_t _setTextRecdFieldsV( dnssd_t* p, const char* text,  va_list vl )
      {
        rc_t  rc = kOkRC;
        char* s0;

        if( text != nullptr )
        {
          _pushTextRecdField( p, text );
          
          while( (s0 = va_arg(vl,char*)) != nullptr )
          {
            _pushTextRecdField( p, s0 );
          }
        }
        return rc;
      }

      char* _formatText( dnssd_t* p )
      {
        unsigned n = 0;
        text_t*  t = p->textL;
        for(; t != nullptr; t=t->link )
          n += strlen(t->text) + 1;

        char* s  = (char*)calloc(1,n);
        char* s0 = s;
        
        for(t=p->textL; t != nullptr; t=t->link, s++ )
        {
          strcpy(s,t->text);
          s += strlen(t->text);
          s[0] = '\n';         
        }

        // replace the ending \n with a zero-terminator
        s0[n-1] = '\0';

        return s0;
      }

      rc_t _init( dnssd_t* p )
      {
        rc_t               rc        = kOkRC;
        char*              formatStr = nullptr;
        struct sockaddr_in hostAddr;

        // get the 32bit host address 
        if((rc = socket::initAddr(  p->hostIpAddr, p->hostPort, &hostAddr )) != kOkRC )
          return rc;
        
        // Get the service 'TXT' fields in the format expected by dns_sd
        formatStr = _formatText(p);

        if( p->dnsSd != nullptr )
          delete p->dnsSd;

        // create the MDNS logic object
        p->dnsSd = new dns_sd(udpSendCallback,p,print_callback);
        
        // create the Surface logic object
        p->fdr   = new fader(print_callback, p->hostMac, hostAddr.sin_addr.s_addr, tcpSendCallback, p, physControlCallback, p, p->ticksPerHeartBeat );
  
        // Setup the internal dnsSd object
        p->dnsSd->setup( p->serviceName, p->serviceType, p->serviceDomain, p->hostName, hostAddr.sin_addr.s_addr, p->hostPort, formatStr );
        
        free(formatStr);

        //p->dnsSd->gen_question();
        p->dnsSd->gen_response();

        return rc;
      }
    }
  }
}

cw::rc_t cw::net::dnssd::createV(  handle_t& hRef, const char* name, const char* type, const char* domain, const char* hostName, const char* hostIpAddr, uint16_t hostPort, const unsigned char hostMac[6], const char* text, va_list vl )
{
  rc_t        rc              = kOkRC;
  unsigned    udpRecvBufByteN = 4096;
  unsigned    udpTimeOutMs    = 50;
  unsigned    tcpRecvBufByteN = 4096;
  unsigned    tcpTimeOutMs    = 50;
  
  if((rc = destroy(hRef)) != kOkRC )
    return rc;

  dnssd_t* p = mem::allocZ<dnssd_t>();
  p->udpRecvBufByteN = udpRecvBufByteN;
  
  // create the mDNS UDP socket server
  if((rc = srv::create(
        p->udpH,
        MDNS_PORT,
        socket::kBlockingFl | socket::kReuseAddrFl | socket::kReusePortFl | socket::kMultiCastTtlFl | socket::kMultiCastLoopFl,
        srv::kUseRecvFromFl,
        udpReceiveCallback,
        p,
        p->udpRecvBufByteN,
        udpTimeOutMs,
        NULL,
        socket::kInvalidPortNumber )) != kOkRC )
  {    
    return cwLogError(rc,"mDNS UDP socket create failed.");
  }

  // add the mDNS socket to the multicast group
  if((rc =  join_multicast_group( socketHandle(p->udpH), MDNS_IP )) != kOkRC )
    goto errLabel;
  
  // set the TTL for multicast 
  if((rc = set_multicast_time_to_live( socketHandle(p->udpH), 255 )) != kOkRC )
    goto errLabel;

  p->tcpRecvBufByteN = tcpRecvBufByteN;
  
  // create the service TCP socket server
  if((rc = srv::create(
        p->tcpH,
        hostPort,
        socket::kTcpFl | socket::kBlockingFl | socket::kStreamFl | socket::kListenFl,
        srv::kUseAcceptFl | srv::kRecvTimeOutFl,
        tcpReceiveCallback,
        p,
        p->tcpRecvBufByteN,
        tcpTimeOutMs,
        NULL,
        socket::kInvalidPortNumber,
        hostIpAddr)) != kOkRC )
  {    
    rc = cwLogError(rc,"mDNS TCP socket create failed.");
    goto errLabel;
  }

  p->serviceName       = mem::duplStr(name);
  p->serviceType       = mem::duplStr(type);
  p->serviceDomain     = mem::duplStr(domain);
  p->hostName          = mem::duplStr(hostName);
  p->hostIpAddr        = mem::duplStr(hostIpAddr);
  p->hostPort          = hostPort;
  p->dnsSd             = nullptr;
  p->ticksPerHeartBeat = 50;
  memcpy(p->hostMac,hostMac,6);
  
  if((rc = _setTextRecdFieldsV( p, text,  vl )) != kOkRC )
    goto errLabel;
  
  hRef.set(p);
  
 errLabel:
  if( rc != kOkRC )
    _destroy(p);
  
  return rc;
}

cw::rc_t cw::net::dnssd::create(  handle_t& hRef, const char* name, const char* type, const char* domain, const char* hostname, const char* hostIpAddr, uint16_t hostPort, const unsigned char hostMac[6], const char* text, ... )
{
  va_list vl;
  va_start(vl,text);
  rc_t rc = createV( hRef, name, type, domain, hostname, hostIpAddr, hostPort, hostMac, text, vl );
  va_end(vl);
  return rc;
}

cw::rc_t cw::net::dnssd::destroy( handle_t& hRef )
{
  rc_t rc = kOkRC;
  
  if( !hRef.isValid() )
    return rc;

  dnssd_t* p = _handleToPtr(hRef);
  if((rc = _destroy(p)) != kOkRC )
    return rc;
  
  hRef.clear();
  return rc;
}

cw::net::srv::handle_t cw::net::dnssd::tcpHandle( handle_t h )
{
  dnssd_t* p = _handleToPtr(h);
  return p->tcpH;
}

cw::net::srv::handle_t cw::net::dnssd::udpHandle( handle_t h )
{
  dnssd_t* p = _handleToPtr(h);
  return p->udpH;
}

cw::rc_t cw::net::dnssd::setTextRecdFields(  handle_t h, const char* text, ... )
{
  va_list vl;
  va_start(vl,text);
  dnssd_t* p  = _handleToPtr(h);
  rc_t     rc = _setTextRecdFieldsV(p,text,vl);
  va_end(vl);
  return rc;
}


cw::rc_t cw::net::dnssd::start( handle_t h )
{
  rc_t     rc = kOkRC;
  dnssd_t* p  = _handleToPtr(h);

  time::get(p->t0);
  
  if((rc = _init(p)) != kOkRC )
    return rc;
  
  // start the mDNS socket server
  if((rc = srv::start( p->udpH )) != kOkRC )
    return rc;
  
  // start the TCP socket server
  if((rc = srv::start( p->tcpH )) != kOkRC )
    return rc;
  
  return rc;
}

cw::rc_t cw::net::dnssd::test()
{
  rc_t               rc            = kOkRC;
  const char*        netIFace      = "ens9";
  const char*        serviceName   = "MC Mix - 1";
  const char*        serviceType   = "_EuConProxy._tcp";
  const char*        serviceDomain = "local";
  uint16_t           hostPort      = 49168;
  const unsigned     sbufN         = 31;
  handle_t           h;
  unsigned char      hostMac[6];
  char               sbuf[ sbufN+1 ];
  char*              text0;
  struct sockaddr_in addr;

  // Get the host name and address from the selected network interface
  char hostname[ _POSIX_HOST_NAME_MAX+1 ];
  char hostIpAddr[ INET_ADDRSTRLEN+1 ];

  //memset(hostMac,0,6);
  //memset(hostname,0,_POSIX_HOST_NAME_MAX+1);
  //memset(&addr, 0, sizeof(struct sockaddr_in));
  //memset(hostIpAddr,0,INET_ADDRSTRLEN+1);
  
  if( socket::get_info( netIFace, hostMac, hostname, sizeof(hostname), &addr ) == kOkRC )
  {
    char ip[128];
    memset(ip,0,128);
    socket::addrToString(&addr,hostIpAddr,sizeof(hostIpAddr));
    printf("%s %s 0x%x %02x:%02x:%02x:%02x:%02x:%02x\n", hostname, ip, addr.sin_addr.s_addr, hostMac[0],hostMac[1],hostMac[2],hostMac[3],hostMac[4],hostMac[5] );
  }

  //
  // Override the host name and mac address to match the example Wireshark captures
  //
  //strcpy(hostname,"Euphonix-MC-0090D580F4DE");
  //unsigned char tmp_mac[] = { 0x00, 0x90, 0xd5, 0x80, 0xf4, 0xde };
  //memcpy(hostMac,tmp_mac,6);
      

  // create the DNS-SD server
  if((rc = create(  h, serviceName, serviceType, serviceDomain, hostname, hostIpAddr, hostPort, hostMac, nullptr, nullptr )) != kOkRC )
    return cwLogError(rc,"Unable to create DNS-SD server.");

  // form the 'lmac=38-C9-86-37-44-E7'
  text0 = mem::printf<char>(nullptr,"lmac=%02x-%02x-%02x-%02x-%02x-%02x",hostMac[0],hostMac[1],hostMac[2],hostMac[3],hostMac[4],hostMac[5]);

  // set the DNS-SD 'SRV' text fields
  if((rc = setTextRecdFields( h, "dummy=0", text0, nullptr )) != kOkRC )
    goto errLabel;

  // start the DNS-SD server
  if((rc = start( h )) != kOkRC )
    goto errLabel;
  
  while( true )
  {
    printf("? ");
    if( std::fgets(sbuf,sbufN,stdin) == sbuf )
    {

      if( strcmp(sbuf,"quit\n") == 0 )
        break;
        
    }
  }

 errLabel:
  mem::release(text0);
  rc = destroy(h);
  
  return rc;
}