/**
 * Tencent is pleased to support the open source community by making Tars available.
 *
 * Copyright (C) 2016THL A29 Limited, a Tencent company. All rights reserved.
 *
 * Licensed under the BSD 3-Clause License (the "License"); you may not use this file except 
 * in compliance with the License. You may obtain a copy of the License at
 *
 * https://opensource.org/licenses/BSD-3-Clause
 *
 * Unless required by applicable law or agreed to in writing, software distributed 
 * under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR 
 * CONDITIONS OF ANY KIND, either express or implied. See the License for the 
 * specific language governing permissions and limitations under the License.
 */

#include "NodeServer.h"
#include "servant/Communicator.h"
#include "util/tc_option.h"
#include "util/tc_file.h"
#include "util/tc_config.h"
#include <iostream>

using namespace std;
using namespace tars;

NodeServer g_app;
TC_Config* g_pconf;

static string outfill(const string& s, char c = ' ', int n = 29)
{
    return (s + string(abs(n - (int)s.length()), c));
}

bool getConfig(const string &sLocator,const string &sRegistryObj,string &sNodeId,string &sLocalIp, string &sConfigFile, bool bCloseCout)
{  
    try
    {
        //string          sLocalIp;
        string          sResult;
        string          sTemplate; 
        TC_Config       tConf;   
        CommunicatorFactory::getInstance()->getCommunicator()->setProperty("locator",sLocator);       
        RegistryPrx pRegistryPrx = CommunicatorFactory::getInstance()->getCommunicator()->stringToProxy<RegistryPrx>(sRegistryObj);  
        if( sLocalIp.empty() && pRegistryPrx->getClientIp(sLocalIp) != 0)
        {
            cerr<<"cannot get localip: " <<sLocalIp << endl;
            return false;
        }

        if(sNodeId == "" )
        {            
            sNodeId = sLocalIp;
        }  

        pRegistryPrx->getNodeTemplate(sNodeId,sTemplate);
        if(TC_Common::trim(sTemplate) == "" )
        {
            cerr << "cannot get node Template nodeid:"<<sNodeId <<",Template:"<<sTemplate<< endl; 
            return false;
        }

        sTemplate = TC_Common::replace(sTemplate, "${locator}",sLocator);
        sTemplate = TC_Common::replace(sTemplate, "${registryObj}",sRegistryObj);
        sTemplate = TC_Common::replace(sTemplate, "${localip}",sLocalIp);
        //sTemplate = TC_Common::replace(sTemplate, "${nodeid}",sNodeId);
        tConf.parseString(sTemplate); 
        if (!bCloseCout)
        {
            string sCloseCoutValue = tConf.get("/tars/application/server<closecout>");
            if (sCloseCoutValue == "1")
            {
                sTemplate = TC_Common::replace(sTemplate, "closecout=1", "closecout=0");
                tConf.parseString(sTemplate);
            }
            else if (sCloseCoutValue.empty())
            {
                map<string, string> m;
                m["closecout"] = "0";
                tConf.insertDomainParam("/tars/application/server", m,true);
            }
            else
            {
                cerr << "failed to set closeout value" << endl;
            }
        }
    
        string sConfigPath    = TC_File::extractFilePath(sConfigFile); 
        if(!TC_File::makeDirRecursive( sConfigPath ))
        {
            cerr<<"cannot create dir: " <<sConfigPath << endl;
            exit( 0 );
        }
        string sFileTemp    = sConfigPath + "/config.conf.tem";
        ofstream configfile(sFileTemp.c_str());
        if(!configfile.good())
        {        
            cerr << "cannot create configuration file: "<<sConfigFile<< endl;
            return false;
        } 
        configfile <<TC_Common::replace(tConf.tostr(),"\\s"," ") << endl;
        configfile.close(); 
               
        string sFileBak     = sConfigFile+"."+TC_Common::now2str() + ".bak";
        if(TC_File::isFileExist(sConfigFile))
        {
            TC_File::copyFile(sConfigFile,sFileBak,true);
        }  

        TC_File::copyFile(sFileTemp,sConfigFile,true);
        TC_File::removeFile(sFileTemp,false); 

        return true;          
    } 
    catch(exception &e)
    {
       cerr<< "load config  erro:" <<e.what()<< endl; 
    }
    return false;
}

void monitorNode(const string &configFile)
{
    TC_Config conf;
    conf.parseFile(configFile);

    string nodeObj = "AdminObj@"+conf["/tars/application/server<local>"];

    ServantPrx prx = CommunicatorFactory::getInstance()->getCommunicator()->stringToProxy<ServantPrx>(nodeObj);
    prx->tars_ping();
}

void parseConfig(int argc, char *argv[])
{

    TC_Option tOp;//consider
    tOp.decode(argc, argv);

     if (tOp.hasParam("nodeversion"))
    {
        cout << "Node:" << TARS_VERSION << "_" << NODE_VERSION << endl;
        exit(0);
    }

    if (tOp.hasParam("monitor"))
    {
        try
        {
            string configFile = tOp.getValue("config");

            monitorNode(configFile);
        }
        catch (exception &ex)
        {
            cout << "failed:" << ex.what() << endl;
            exit(-1);
        }
        exit(0);
        return;
    }

    string sLocator         = tOp.getValue("locator");
    string sNodeId          = tOp.getValue("nodeid"); 
    string sConfigFile      = tOp.getValue("config");   
    string sRegistryObj     = tOp.getValue("registryObj");
    string sLocalIp         = tOp.getValue("localip");
    
    if(sConfigFile == "")
    {
        cerr << endl;
        cerr <<"start server with locator config, for example: "<<endl;
        cerr << argv[0] << " --locator=tars.tarsregistry.QueryObj@tcp -h 172.25.38.67 -p 17890"  " --config=config.conf [--nodeid = 172.25.38.67 --registryObj=tars.tarsregistry.RegistryObj]" << endl;
        cerr << argv[0] << " --config=config.conf" << endl;
        exit(0);
    }   
    
    if(!TC_File::isAbsolute(sConfigFile))
    {
        char sCwd[PATH_MAX];
        if ( getcwd( sCwd, PATH_MAX ) == NULL )
        {
            TLOGERROR("cannot get the current directory:\n" << endl);
            exit( 0 );
        }
        sConfigFile = string(sCwd) +"/"+ sConfigFile;
    }
    sConfigFile = TC_File::simplifyDirectory(sConfigFile);
    if(sLocator != "")
    {
        if(sRegistryObj == "")
        {
            sRegistryObj = "tars.tarsregistry.RegistryObj";
        }
        bool bCloseOut=true;
        if(tOp.hasParam("closecout"))
        {
            if(TC_Common::lower(tOp.getValue("closecout"))=="n")
            {
                bCloseOut = false;
            }
        }
        bool bRet = getConfig(sLocator,sRegistryObj,sNodeId, sLocalIp, sConfigFile, bCloseOut);
        if(bRet == false)
        {
            cerr<<"reload config erro. use old config";
        }
    }
    cout <<endl;
    map<string,string> mOp = tOp.getMulti();
    for(map<string,string>::const_iterator it = mOp.begin(); it != mOp.end(); ++it)
    {
        cout << outfill( it->first)<< it->second << endl;  
    } 

}

int main( int argc, char* argv[] )
{
    try
    {   
        TC_Common::daemon();

        parseConfig(argc,argv);

        g_pconf = &g_app.getConfig();

        g_app.main( argc, argv );
        g_app.waitForShutdown();
    }
    catch ( exception& ex )
    {
        cout<< ex.what() << endl;
    }
    catch ( ... )
    {
        cout<< "main unknow exception cathed" << endl;
    }

    return 0;
}