#include <cstdlib>
#include <string>
#include <sstream>
#include <ndn-cpp-dev/face.hpp>
#include <ndn-cpp-dev/security/key-chain.hpp>
#include <ndn-cpp-dev/security/identity-certificate.hpp>
#include <ndn-cpp-dev/util/scheduler.hpp>


#include "nlsr.hpp"
#include "nlsr_conf_processor.hpp"
#include "utility/nlsr_logger.hpp"
#include "security/nlsr_km.hpp"
#include "security/nlsr_cert_store.hpp"
#include "security/nlsr_cse.hpp"


namespace nlsr
{

  using namespace ndn;
  using namespace std;

  void
  Nlsr::nlsrRegistrationFailed(const ndn::Name& name)
  {
    cerr << "ERROR: Failed to register prefix in local hub's daemon" << endl;
    getNlsrFace()->shutdown();
  }


  void
  Nlsr::setInterestFilterNlsr(const string& name)
  {
    getNlsrFace()->setInterestFilter(name,
                                     func_lib::bind(&interestManager::processInterest, &im,
                                         boost::ref(*this), _1, _2),
                                     func_lib::bind(&Nlsr::nlsrRegistrationFailed, this, _1));
  }

  void
  Nlsr::initNlsr()
  {
    confParam.buildRouterPrefix();
    nlsrLogger.initNlsrLogger(confParam.getLogDir());
    nlsrLsdb.setLsaRefreshTime(confParam.getLsaRefreshTime());
    nlsrLsdb.setThisRouterPrefix(confParam.getRouterPrefix());
    fib.setFibEntryRefreshTime(2*confParam.getLsaRefreshTime());
    if( ! km.initKeyManager(confParam) )
    {
      std::cerr<<"Can not initiate/load certificate"<<endl;
    }
    sm.setSeqFileName(confParam.getSeqFileDir());
    sm.initiateSeqNoFromFile();
    /* debugging purpose start */
    cout <<	confParam;
    adl.printAdl();
    npl.printNpl();
    /* debugging purpose end */
    nlsrLsdb.buildAndInstallOwnNameLsa(boost::ref(*this));
    nlsrLsdb.buildAndInstallOwnCorLsa(boost::ref(*this));
    setInterestFilterNlsr(confParam.getRouterPrefix());
    setInterestFilterNlsr(confParam.getChronosyncLsaPrefix()+
                          confParam.getRouterPrefix());
    setInterestFilterNlsr(confParam.getRootKeyPrefix());
    slh.setSyncPrefix(confParam.getChronosyncSyncPrefix());
    slh.createSyncSocket(boost::ref(*this));
    slh.publishKeyUpdate(km);
    im.scheduleInfoInterest(boost::ref(*this),10);
  }

  void
  Nlsr::startEventLoop()
  {
    io->run();
  }

  int
  Nlsr::usage(const string& progname)
  {
    cout << "Usage: " << progname << " [OPTIONS...]"<<endl;
    cout << "   NDN routing...." << endl;
    cout << "       -d, --daemon        Run in daemon mode" << endl;
    cout << "       -f, --config_file   Specify configuration file name" <<endl;
    cout << "       -p, --api_port      port where api client will connect" <<endl;
    cout << "       -h, --help          Display this help message" << endl;
    exit(EXIT_FAILURE);
  }


} // namespace nlsr

using namespace nlsr;

int
main(int argc, char **argv)
{
  nlsr::Nlsr nlsr_;
  string programName(argv[0]);
  nlsr_.setConfFileName("nlsr.conf");
  int opt;
  while ((opt = getopt(argc, argv, "df:p:h")) != -1)
  {
    switch (opt)
    {
      case 'f':
        nlsr_.setConfFileName(optarg);
        break;
      case 'd':
        nlsr_.setIsDaemonProcess(optarg);
        break;
      case 'p':
        {
          stringstream sst(optarg);
          int ap;
          sst>>ap;
          nlsr_.setApiPort(ap);
        }
        break;
      case 'h':
      default:
        nlsr_.usage(programName);
        return EXIT_FAILURE;
    }
  }
  ConfFileProcessor cfp(nlsr_.getConfFileName());
  int res=cfp.processConfFile(nlsr_);
  if ( res < 0 )
  {
    return EXIT_FAILURE;
  }
  nlsr_.initNlsr();
  try
  {
    nlsr_.startEventLoop();
  }
  catch(std::exception &e)
  {
    std::cerr << "ERROR: " << e.what() << std::endl;
  }
  return EXIT_SUCCESS;
}