/*
 *     .============.
 *    //  M A K E  / \
 *   //  C++ DEV  /   \
 *  //  E A S Y  /  \/ \
 * ++ ----------.  \/\  .
 *  \\     \     \ /\  /
 *   \\     \     \   /
 *    \\     \     \ /
 *     -============'
 *
 * Copyright (c) 2018 Hevake and contributors, all rights reserved.
 *
 * This file is part of cpp-tbox (https://github.com/cpp-main/cpp-tbox)
 * Use of this source code is governed by MIT license that can be found
 * in the LICENSE file in the root of the source tree. All contributing
 * project authors may be found in the CONTRIBUTORS.md file in the root
 * of the source tree.
 */
#include <tbox/main/main.h>
#include <tbox/base/log.h>
#include <tbox/base/log_output.h>
#include <tbox/util/argument_parser.h>
#include <tbox/util/fs.h>

#include <iostream>
#include <dlfcn.h>

namespace {

typedef void(*RegisterAppsFunc) (tbox::main::Module &, tbox::main::Context &);

std::vector<void*> _dl_handle_vec;
std::vector<RegisterAppsFunc> _register_func_vec;

/// 解决参数中的'-l <module>' 与 '--load <module>' 参数作为模块动态库列表
void ParseArgs(int argc, char **argv,
               std::vector<std::string> &module_file_vec)
{
    tbox::util::ArgumentParser parser(
        [&] (char short_option, const std::string &long_option,
             tbox::util::ArgumentParser::OptionValue &option_value) {
            if (short_option == 'l' || long_option == "load")
                module_file_vec.push_back(option_value.get());
            return true;
        }
    );

    parser.parse(argc, argv);
}

/// 导入模块动态库
void Load(int argc, char **argv)
{
    std::vector<std::string> module_file_vec;
    ParseArgs(argc, argv, module_file_vec);

    for (auto &module_file : module_file_vec) {
        if (!tbox::util::fs::IsFileExist(module_file)) {
            LogWarn("file %s not exist.", module_file.c_str());
            continue;
        }

        void *dl_handle = dlopen(module_file.c_str(), RTLD_NOW);
        if (dl_handle == nullptr) {
            LogWarn("load %s faild.", module_file.c_str());
            const char *err_str = dlerror();
            if (err_str != nullptr)
                LogNotice("reason: %s", err_str);
            continue;
        }

        auto register_func = (RegisterAppsFunc)dlsym(dl_handle, "RegisterApps");
        if (register_func == nullptr) {
            dlclose(dl_handle);
            LogWarn("can't find 'RegisterApps' symbol in %s.", module_file.c_str());
            continue;
        }

        _dl_handle_vec.push_back(dl_handle);
        _register_func_vec.push_back(register_func);

        LogInfo("load %s success.", module_file.c_str());
    }
}

/// 关闭模块动态库
void Release()
{
    for (void *dl_handle : _dl_handle_vec)
        ::dlclose(dl_handle);
    _dl_handle_vec.clear();
}

}

namespace tbox {
namespace main {

void RegisterApps(Module &apps, Context &ctx)
{
    if (_register_func_vec.empty()) {
        LogWarn("no module");
        return;
    }

    for (auto register_func : _register_func_vec)
        register_func(apps, ctx);
    _register_func_vec.clear();
}

std::string GetAppDescribe() { return "modules runner"; }

std::string GetAppBuildTime() { return __DATE__ " " __TIME__; }

void GetAppVersion(int &major, int &minor, int &rev, int &build)
{
    major = 0;
    minor = 0;
    rev = 2;
    build = 0;
}

}
}

int main(int argc, char **argv)
{
    LogOutput_Enable();
    Load(argc, argv);
    LogOutput_Disable();

    auto ret = tbox::main::Main(argc, argv);

    Release();
    return ret;
}