/********************************************************************************
 *    Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             *
 *         GNU Lesser General Public Licence version 3 (LGPL) version 3,        *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/
/*
 * File:   FairMQProgOptions.cxx
 * Author: winckler
 * 
 * Created on March 11, 2015, 10:20 PM
 */

#include "FairMQProgOptions.h"
#include <algorithm>
#include "FairMQParser.h"
using namespace std;

FairMQProgOptions::FairMQProgOptions()
    : FairProgOptions()
    , fMQParserOptions("MQ-Device parser options")
    , fMQOptionsInCfg("MQ-Device options")
    , fMQOptionsInCmd("MQ-Device options")
    , fMQtree()
    , fFairMQMap()
    , fHelpTitle("***** FAIRMQ Program Options ***** ")
    , fVersion("Beta version 0.1")
{
}

FairMQProgOptions::~FairMQProgOptions()
{
}

int FairMQProgOptions::ParseAll(const int argc, char** argv, bool allowUnregistered)
{
    // init description
    InitOptionDescription();
    // parse command line options
    if (ParseCmdLine(argc, argv, fCmdLineOptions, fVarMap, allowUnregistered))
    {
        return 1;
    }

    // if txt/INI configuration file enabled then parse it as well
    if (fUseConfigFile)
    {
        // check if file exist
        if (fs::exists(fConfigFile))
        {
            if (ParseCfgFile(fConfigFile.string(), fConfigFileOptions, fVarMap, allowUnregistered))
            {
                return 1;
            }
        }
        else
        {
            LOG(ERROR) << "config file '" << fConfigFile << "' not found";
            return 1;
        }
    }

    // // set log level before printing (default is 0 = DEBUG level)
    // std::string verbose = GetValue<std::string>("verbose");
    // bool color = GetValue<bool>("log-color");
    // if (!color)
    // {
    //     reinit_logger(false);
    // }
    // //SET_LOG_LEVEL(DEBUG);
    // if (fSeverityMap.count(verbose))
    // {
    //     set_global_log_level(log_op::operation::GREATER_EQ_THAN, fSeverityMap.at(verbose));
    // }
    // else
    // {
    //     LOG(ERROR) << " verbosity level '" << verbose << "' unknown, it will be set to DEBUG";
    //     set_global_log_level(log_op::operation::GREATER_EQ_THAN, fSeverityMap.at("DEBUG"));
    // }

    PrintOptions();

    // check if one of required MQ config option is there  
    auto parserOption_shptr = fMQParserOptions.options();
    bool optionExists = false;
    vector<string> MQParserKeys;
    for (const auto& p : parserOption_shptr)
    {
        MQParserKeys.push_back(p->canonical_display_name());
        if (fVarMap.count(p->canonical_display_name()))
        {
            optionExists = true;
            break;
        }
    }

    if (!optionExists)
    {
        LOG(WARN) << "Options to configure FairMQ channels are not provided.";
        LOG(WARN) << "Please provide the value for one of the following keys:";
        for (const auto& p : MQParserKeys)
        {
            LOG(WARN) << p;
        }
        LOG(WARN) << "No channels will be created (You can create them manually).";
        // return 1;
    }
    else
    {
        if (fVarMap.count("mq-config"))
        {
            LOG(DEBUG) << "mq-config command line called : default xml/json parser will be used";
            std::string file = fVarMap["mq-config"].as<std::string>();
            std::string id;
            if (fVarMap.count("config-key"))
            {
                id = fVarMap["config-key"].as<std::string>();
            }
            else
            {
                id = fVarMap["id"].as<std::string>();
            }

            std::string file_extension = boost::filesystem::extension(file);

            std::transform(file_extension.begin(), file_extension.end(), file_extension.begin(), ::tolower);

            if (file_extension == ".json")
            {
                UserParser<FairMQParser::JSON>(file, id);
            }
            else
            {
                if (file_extension == ".xml")
                {
                    UserParser<FairMQParser::XML>(file, id);
                }
                else
                {
                    LOG(ERROR)  << "mq-config command line called but file extension '"
                                << file_extension
                                << "' not recognized. Program will now exit";
                    return 1;
                }
            }
        }
    }

    return 0;
}

int FairMQProgOptions::NotifySwitchOption()
{
    if (fVarMap.count("help"))
    {
        LOG(INFO) << fHelpTitle << "\n" << fVisibleOptions;
        return 1;
    }

    if (fVarMap.count("version")) 
    {
        LOG(INFO) << fVersion << "\n";
        return 1;
    }

    return 0;
}

void FairMQProgOptions::InitOptionDescription()
{
    // Id required in command line if config txt file not enabled
    if (fUseConfigFile)
    {
        fMQOptionsInCmd.add_options()
            ("id",                po::value<string>(),                          "Device ID (required argument).")
            ("io-threads",        po::value<int>()->default_value(1),           "Number of I/O threads.")
            ("transport",         po::value<string>()->default_value("zeromq"), "Transport ('zeromq'/'nanomsg').")
            ("deployment",        po::value<string>()->default_value("static"), "Deployment ('static'/'dds').")
            ("network-interface", po::value<string>()->default_value("eth0"),   "Network interface to bind on (e.g. eth0, ib0, wlan0, en0, lo...).")
            ;

        fMQOptionsInCfg.add_options()
            ("id",                po::value<string>()->required(),              "Device ID (required argument).")
            ("io-threads",        po::value<int>()->default_value(1),           "Number of I/O threads.")
            ("transport",         po::value<string>()->default_value("zeromq"), "Transport ('zeromq'/'nanomsg').")
            ("deployment",        po::value<string>()->default_value("static"), "Deployment ('static'/'dds').")
            ("network-interface", po::value<string>()->default_value("eth0"),   "Network interface to bind on (e.g. eth0, ib0, wlan0, en0, lo...).")
            ;
    }
    else
    {
        fMQOptionsInCmd.add_options()
            ("id",                po::value<string>()->required(),              "Device ID (required argument)")
            ("io-threads",        po::value<int>()->default_value(1),           "Number of I/O threads")
            ("transport",         po::value<string>()->default_value("zeromq"), "Transport ('zeromq'/'nanomsg').")
            ("deployment",        po::value<string>()->default_value("static"), "Deployment ('static'/'dds').")
            ("network-interface", po::value<string>()->default_value("eth0"),   "Network interface to bind on (e.g. eth0, ib0, wlan0, en0, lo...).")
            ;
    }

    fMQParserOptions.add_options()
        ("config-xml-string",  po::value<vector<string>>()->multitoken(), "XML input as command line string.")
        ("config-xml-file",    po::value<string>(),                       "XML input as file.")
        ("config-json-string", po::value<vector<string>>()->multitoken(), "JSON input as command line string.")
        ("config-json-file",   po::value<string>(),                       "JSON input as file.")
        ("mq-config",          po::value<string>(),                       "JSON/XML input as file. The configuration object will check xml or json file extention and will call the json or xml parser accordingly")
        ("config-key",         po::value<string>(),                       "Use provided value instead of device id for fetching the configuration from the config file")
        ;

    AddToCmdLineOptions(fGenericDesc);
    AddToCmdLineOptions(fMQOptionsInCmd);
    AddToCmdLineOptions(fMQParserOptions);

    if (fUseConfigFile)
    {
        AddToCfgFileOptions(fMQOptionsInCfg, false);
        AddToCfgFileOptions(fMQParserOptions, false);
    }
}