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

#include "nlsr.hpp"
#include "nlsr_conf_processor.hpp"
#include "utility/nlsr_logger.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());
        km.initKeyManager(confParam);
        sm.setSeqFileName(confParam.getSeqFileDir());
        sm.initiateSeqNoFromFile();
        slh.setSyncPrefix(confParam.getChronosyncSyncPrefix());
    }

    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();
    /* debugging purpose start */
    cout <<	nlsr_.getConfParameter();
    nlsr_.getAdl().printAdl();
    nlsr_.getNpl().printNpl();
    /* debugging purpose end */
    nlsr_.getLsdb().buildAndInstallOwnNameLsa(nlsr_);
    nlsr_.getLsdb().buildAndInstallOwnCorLsa(nlsr_);
    nlsr_.setInterestFilterNlsr(nlsr_.getConfParameter().getRouterPrefix());
    nlsr_.setInterestFilterNlsr(nlsr_.getConfParameter().getChronosyncLsaPrefix()+
                                nlsr_.getConfParameter().getRouterPrefix());
    nlsr_.setInterestFilterNlsr(nlsr_.getConfParameter().getRootKeyPrefix());
    nlsr_.getSlh().createSyncSocket(nlsr_);
    nlsr_.getSlh().publishKeyUpdate(nlsr_.getKeyManager());
    nlsr_.getSlh().publishRoutingUpdate(nlsr_.getSm(),
                                        nlsr_.getConfParameter().getChronosyncLsaPrefix()
                                        + nlsr_.getConfParameter().getRouterPrefix());
    nlsr_.getIm().scheduleInfoInterest(nlsr_,1);
    try
    {
        nlsr_.startEventLoop();
    }
    catch(std::exception &e)
    {
        std::cerr << "ERROR: " << e.what() << std::endl;
    }
    return EXIT_SUCCESS;
}