/**
 * 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 "tars_patch.h"
#include "util/tc_option.h"
#include "servant/Application.h"

#include <iostream>

using namespace tars;

void usage(int argc, char *argv[])
{
    cout << "Usage:" << argv[0] << "  --server=[server] --remote=[server-dir] --local=[local] [--remove]" << endl;
    cout << "server: PatchObj@tcp -h ip -p port -t timeout" << endl;
    cout << "remote: directory or file sync from server" << endl;
    cout << "local:  directory sync to local" << endl;
    cout << "remove: remove local dir data" << endl;
    exit(0);
}

class PatchClientInterface : public TarsPatchNotifyInterface
{
public:
    virtual void onDownload(const FileInfo &fi)
    {
        cout << "------------------------------------" << endl;
        cout << "downloading [" << fi.path << "] [" << fi.size*1.0/1024 << "K]" << endl;
    }
    virtual void onDownloading(const FileInfo &fi, size_t pos, const string &dest)
    {
        cout << "downloading [" << fi.path << " ->] [" << dest << "] [" << pos*100/fi.size << "/100%]" << endl;
    }
    virtual void onDownloadOK(const FileInfo &fi, const string &dest)
    {
        cout << "download ok [" << fi.path << "] -> [" << dest << "]" << endl;
    }
    virtual void onListFile()
    {
        cout << "listing files" << endl;
    }
    virtual void onListFileOK(const vector<FileInfo> &vf)
    {
        cout << "list ok total [" << vf.size() << "] files" << endl;
    }
    virtual void onRemoveLocalDirectory(const string &sDir)
    {
        cout << "------------------------------------" << endl;
        cout << "remove local directory [" << sDir << "]" << endl;
    }
    virtual void onRemoveLocalDirectoryOK(const string &sDir)
    {
        cout << "remove local directory ok [" << sDir << "]"<< endl;
        cout << "------------------------------------" << endl;
    }
    virtual void onSetExecutable(const FileInfo &fi)
    {
        cout << "executable  [" << fi.path << "]" << endl;
    }
    virtual void onDownloadAllOK(const vector<FileInfo> &vf, time_t timeBegin, time_t timeEnd)
    {
        cout << "------------------------------------" << endl;
        cout << "download all files succ." << endl;
    }

    virtual void onReportTime(const string & sFile, const time_t timeBegin, const time_t timeEnd)
    {
        cout << "------------------------------------" << endl;
        cout << "download '" << sFile << "', begin:" << timeBegin << ", end:" << timeEnd << endl;
    }
};

typedef TC_AutoPtr<PatchClientInterface> PatchClientInterfacePtr;

int main(int argc, char *argv[])
{

    if(argc < 2)
    {
        usage(argc, argv);
    }

    try
    {

        TC_Option option;
        option.decode(argc, argv);

        //远程目录的相对路径
        string server = option.getValue("server");
        if(server.empty())
        {
            cout << "server must not be empty." << endl;
            return 0;
        }

        CommunicatorFactory::getInstance()->getCommunicator()->setProperty("sync-invoke-timeout", "60000");
        PatchPrx pPtr;
        CommunicatorFactory::getInstance()->getCommunicator()->stringToProxy(server, pPtr);

        //远程目录的相对路径
        string remote_dir = option.getValue("remote");
        if(remote_dir.empty())
        {
            cout << "remote_dir must not be empty." << endl;
            return 0;
        }

        //本地同步路径
        string local_dir  = option.getValue("local");
        if(local_dir.empty())
        {
            cout << "local_dir must not be empty." << endl;
            return 0;
        }

        //是否先清空本地目录
        bool bRemove = option.hasParam("remove");

        cout << "connecting to patch server" << endl;
        TarsPatch op;
        op.init(pPtr, remote_dir, local_dir);
        cout << "connect ok to patch server" << endl;
        op.setRemove(bRemove);

        PatchClientInterfacePtr p = new PatchClientInterface();
        op.download(p);
    }
    catch(exception &ex)
    {
        cout << ex.what() << endl;
    }

    return 0;
}