/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
/**
 * Copyright (c) 2014, Regents of the University of California.
 *
 * This file is part of NDNS (Named Data Networking Domain Name Service).
 * See AUTHORS.md for complete list of NDNS authors and contributors.
 *
 * NDNS is free software: you can redistribute it and/or modify it under the terms
 * of the GNU General Public License as published by the Free Software Foundation,
 * either version 3 of the License, or (at your option) any later version.
 *
 * NDNS is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
 * PURPOSE.  See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along with
 * NDNS, e.g., in COPYING.md file.  If not, see <http://www.gnu.org/licenses/>.
 */
#include "name-server.hpp"


namespace ndn{
namespace ndns{
NameServer::NameServer(const char *programName, const char *prefix, const char *nameZone)
: NDNApp(programName, prefix)
, m_zone(Name(nameZone))
, m_zoneMgr(m_zone)
{
  //m_zoneMgr.lookupId();
}//NameServer Construction


void
NameServer::onInterest(const Name &name, const Interest &interest)
{

  cout<<"[* -> *] receive Interest: "<<interest.getName().toUri()<<std::endl;
  Query query;
  if (!query.fromInterest(interest))
  {
    cout<<"can resolve the Query from Interest: "<<endl;
    return;
  }

  /*
   * query.getAuthorityZone is routable name, not the zone's service name
  if (query.getAuthorityZone() != m_zoneMgr.getZone().getAuthorizedName())
  {
    cout<<"Query is intent to zone: "<<query.getAuthorityZone()
        <<". This is "<<m_zoneMgr.getZone().getAuthorizedName()<<endl;
    return;
  }
  */

  Response response;
  Name name2 = interest.getName();
  name2.appendVersion();
  response.setQueryName(name2);
  RRMgr mgr(m_zone, query, response);


  if (mgr.lookup()<0)
  {
    cout<<"[* !! *] lookup error, then exit: "<<mgr.getErr()<<endl;
    return;
  }

  if (response.getRrs().size() >0)
  {
    response.setResponseType(Response::NDNS_Resp);
  } else {

    if (query.getRrType() == RR::NS) {
      int count = mgr.count();
      if (count < 0)
      {
        cout<<"[* !! *] lookup error, then exit: "<<mgr.getErr()<<endl;
        return;
      } else if (count > 0)
      {
        response.setResponseType(Response::NDNS_Auth);
      } else{
        response.setResponseType(Response::NDNS_Nack);
      }
    } else {
      response.setResponseType(Response::NDNS_Nack);
    }
  }


  Data data = response.toData();
  data.setFreshnessPeriod(response.getFreshness());

  m_keyChain.sign(data);
  m_face.put(data);
  cout<<"[* <- *] send response: "<<response<<": "<<data<<endl;
}//onInterest


void
NameServer::run()
{
  //m_zoneMgr.lookupId();
  if (m_zoneMgr.getZone().getId() == 0)
  {
    m_hasError = true;
    m_error = "cannot get Zone.id from database for name="+m_zone.getAuthorizedName().toUri();
    stop();
  }

  boost::asio::signal_set signalSet(m_ioService, SIGINT, SIGTERM);
  signalSet.async_wait(boost::bind(&NDNApp::signalHandler, this));
 // boost::bind(&NdnTlvPingServer::signalHandler, this)
  Name name;
  name.set(m_prefix);
  name.append(Query::toString(Query::QUERY_DNS));

  m_face.setInterestFilter(name,
               bind(&NameServer::onInterest,
                  this, _1, _2),
               bind(&NDNApp::onRegisterFailed,
                  this, _1,_2));

  std::cout << "\n=== NDNS Server for Zone "
            << m_zoneMgr.getZone().getAuthorizedName().toUri()
            <<" with routable prefix "<< name.toUri()
            << " starts===\n" << std::endl;

  try {
    m_face.processEvents();
  }
  catch (std::exception& e) {
    m_hasError = true;
    m_error = "ERROR: ";
    m_error += e.what();
    stop();
  }

}//run

} //namespace ndns
} /* namespace ndn */