/**
 * 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.
 */

#ifndef __ADDFILE_COMMAND_H_
#define __ADDFILE_COMMAND_H_

#include "ServerCommand.h"

/**
 *自config服务下载文件 
 *
 */
class CommandAddFile : public ServerCommand
{
    
public:
    CommandAddFile(const ServerObjectPtr &pServerObjectPtr,const string &sFile,bool bByNode = false);
    ExeStatus canExecute(string &sResult);
    int execute(string &sResult);
private:
    int getScriptFile(string &sResult);
    bool                _byNode;
    string              _file; 
    string              _serverId;               
    ServerDescriptor    _desc;
    StatExChangePtr     _statExChange;
    ServerObjectPtr     _serverObjectPtr;
};

//////////////////////////////////////////////////////////////
//
inline CommandAddFile::CommandAddFile(const ServerObjectPtr &pServerObjectPtr,const string &sFile,bool bByNode)
:_byNode(bByNode)
,_file(sFile)
,_serverObjectPtr(pServerObjectPtr)
{ 
    _desc      = _serverObjectPtr->getServerDescriptor();
    _serverId  = _serverObjectPtr->getServerId();
}

//////////////////////////////////////////////////////////////////////////////
// 
inline ServerCommand::ExeStatus CommandAddFile::canExecute(string &sResult)
{
  
    TC_ThreadRecLock::Lock lock( *_serverObjectPtr );
    
    TLOGDEBUG("CommandAddFile::canExecute "<<_desc.application<< "." << _desc.serverName << " beging addFilesing------"<< endl);
    ServerObject::InternalServerState eState = _serverObjectPtr->getInternalState();
    
    if(_desc.application == "" || _desc.serverName == "" )
    {
        TLOGERROR("CommandAddFile::canExecute app or server name is empty"<<endl);
        return DIS_EXECUTABLE;
    }
    if(_file.empty())
    {
        TLOGERROR("CommandAddFile::canExecute file is empty. file:"<<_file<<endl);
        return DIS_EXECUTABLE;
    }
    //设当前状态为正在AddFilesing
    _statExChange = new StatExChange(_serverObjectPtr,ServerObject::AddFilesing, eState);
    
    return EXECUTABLE;      
}
 
//////////////////////////////////////////////////////////////
//
inline int CommandAddFile::execute(string &sResult)
{
  try
  {       
        //若File不合法采用默认路径
        if( TC_File::isAbsolute(_file ) ==  false )
        {
            _file = _serverObjectPtr->getExePath()+"/"+_file;
        }
        _file               = TC_File::simplifyDirectory(TC_Common::trim(_file));

        string sFilePath    = TC_File::extractFilePath(_file); 
        
        string sFileName    = TC_File::extractFileName(_file);      
        //创建目录
        if(!TC_File::makeDirRecursive(sFilePath ) )
        {

            TLOGERROR("CommandAddFile::execute cannot create dir:"<<sFilePath<<" erro:"<<strerror(errno)<<endl);
            return -1;
        }
        bool bRet = false;
        if(_serverObjectPtr->isScriptFile(sFileName) == true)
        {
            bRet = getScriptFile(sResult);
        }  
        else
        {
            TarsRemoteConfig tTarsRemoteConfig;
            tTarsRemoteConfig.setConfigInfo(Application::getCommunicator(), ServerConfig::Config,_desc.application, _desc.serverName, sFilePath,_desc.setId);
//            tTarsRemoteConfig.setConfigInfo(ServerConfig::Config,_desc.application, _desc.serverName, sFilePath,_desc.setId);
            bRet = tTarsRemoteConfig.addConfig(sFileName,sResult);
            g_app.reportServer(_serverObjectPtr->getServerId(),sResult); 
        }
        return bRet;               
  } 
  catch(exception &e)
  {
      TLOGERROR("CommandAddFile::execute get file: "<<_file<<" from configServer fail."<<e.what());
  }
  return -1;
}

inline int CommandAddFile::getScriptFile(string &sResult)
{
    try
    {     
        string sFileName    = TC_File::extractFileName(_file); 
                  
        ConfigPrx pPtr = Application::getCommunicator()->stringToProxy<ConfigPrx>(ServerConfig::Config);
        string sStream;
        int ret;
        if(_desc.setId.empty())
        {
            ret = pPtr->loadConfig(_desc.application, _desc.serverName, sFileName, sStream);
        }
        else
        {
            struct ConfigInfo confInfo;
            confInfo.appname     = _desc.application;
            confInfo.servername  = _desc.serverName;
            confInfo.filename    = sFileName;
            confInfo.setdivision = _desc.setId;
            ret = pPtr->loadConfigByInfo(confInfo,sStream);
        }

        if (ret != 0)
        {
            sResult = FILE_FUN_STR+"[fail] get remote file :" + sFileName;
            TLOGERROR(sResult<<endl);
            g_app.reportServer(_serverId,sResult); 
            return -1;
        }
        sStream = _serverObjectPtr->decodeMacro(sStream);
        string sTem;
        string::size_type pos = sStream.find('\n');
        while(pos != string::npos)
        {            
            sTem        = sTem + TC_Common::trim(sStream.substr(0,pos)," \r\t") +"\n";
            sStream     = sStream.substr(pos+1); 
            pos         = sStream.find('\n');     
        }               
        sStream = sTem + sStream;
        string sEndMark = "echo 'end-"+sFileName+"'";
        if(sStream.find(sEndMark) == string::npos)
        {
            sStream = sStream + "\n"+sEndMark ;   //tarsnode 生成的脚本都以 'echo 'end filename'' 结束 
        }                   
        if(TC_File::isFileExist(_file) && TC_File::load2str(_file) != sStream)
        {
            TC_File::copyFile(_file,_file+".bak");
        }
        ofstream out(_file.c_str());
        if(!out.good())
        {
            TLOGERROR("CommandAddFile::getScriptFile cannot create file: "<<(_file + " erro:"+ strerror(errno))<<endl);
            return -1;
        }
        out<<sStream;
        out.close();  
        sResult = FILE_FUN_STR+"[succ] get remote file :"+sFileName;
        g_app.reportServer(_serverObjectPtr->getServerId(),sResult); 
        TLOGDEBUG(sResult<<endl);
        return 0;          
  } 
  catch(exception &e)
  {

      TLOGERROR("CommandAddFile::getScriptFile get file"<<(_file+" from configServer fail."+ e.what())<<endl);
  }  
  return -1; 
}
#endif