/*
 *     .============.
 *    //  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 "module.h"

#include <algorithm>

#include <tbox/base/log.h>
#include <tbox/base/json.hpp>

namespace tbox {
namespace main {

Module::Module(const std::string &name, Context &ctx) :
    name_(name), ctx_(ctx)
{ }

Module::~Module()
{
    cleanup();

    for (const auto &item : children_)
        delete item.module_ptr;
    children_.clear();
}

bool Module::add(Module *child, bool required)
{
    if (state_ != State::kNone) {
        LogWarn("module %s's state is not State::kNone", name_.c_str());
        return false;
    }

    if (child == nullptr) {
        LogWarn("child == nullptr");
        return false;
    }

    //! 防止child被重复添加到多个Module上
    if (child->parent_ != nullptr) {
        LogWarn("module %s can't add %s, it has parent", name_.c_str(), child->name_.c_str());
        return false;
    }

    //! 检查名称有没有重复的
    auto iter = std::find_if(children_.begin(), children_.end(),
        [child] (const ModuleItem &item) {
            return item.module_ptr->name() == child->name();
        }
    );
    if (iter != children_.end()) {
        LogWarn("module %s can't add %s, name dupicated.", name_.c_str(), child->name().c_str());
        return false;
    }

    children_.emplace_back(ModuleItem{ child, required });
    child->vars_.setParent(&vars_);
    child->parent_ = this;

    return true;
}

bool Module::addAs(Module *child, const std::string &name, bool required)
{
    auto tmp = child->name_;
    child->name_ = name;

    if (!add(child, required)) {
        //! 如果失败了，还原之前的名称
        child->name_ = tmp;
        return false;
    }

    return true;
}

void Module::fillDefaultConfig(Json &js_parent) const
{
    Json &js_this = name_.empty() ? js_parent : js_parent[name_];

    onFillDefaultConfig(js_this);

    for (const auto &item : children_)
        item.module_ptr->fillDefaultConfig(js_this);
}

bool Module::initialize(const Json &js_parent)
{
    if (state_ != State::kNone) {
        LogWarn("module %s's state is not State::kNone", name_.c_str());
        return false;
    }

    if (!name_.empty() && !js_parent.contains(name_)) {
        LogWarn("missing `%s' field in config", name_.c_str());
        return false;
    }

    const Json &js_this = name_.empty() ? js_parent : js_parent[name_];

    if (!onInit(js_this)) {
        LogWarn("module %s init fail.", name_.c_str());
        return false;
    }

    for (const auto &item : children_) {
        if (!item.module_ptr->initialize(js_this) && item.required) {
            LogErr("required module `%s' initialize() fail", item.module_ptr->name().c_str());
            return false;
        }
    }

    state_ = State::kInited;
    return true;
}

bool Module::start()
{
    if (state_ != State::kInited) {
        LogWarn("module %s's state is not State::kInited", name_.c_str());
        return false;
    }

    if (!onStart()) {
        LogWarn("module %s start fail.", name_.c_str());
        return false;
    }

    for (const auto &item : children_) {
        if (!item.module_ptr->start() && item.required) {
            LogErr("required module `%s' start() fail", item.module_ptr->name().c_str());
            return false;
        }
    }

    state_ = State::kRunning;
    return true;
}

void Module::stop()
{
    if (state_ != State::kRunning)
        return;

    for (auto iter = children_.rbegin(); iter != children_.rend(); ++iter)
        iter->module_ptr->stop();

    onStop();

    state_ = State::kInited;
}

void Module::cleanup()
{
    if (state_ == State::kNone)
        return;

    stop();

    for (auto iter = children_.rbegin(); iter != children_.rend(); ++iter)
        iter->module_ptr->cleanup();

    onCleanup();

    state_ = State::kNone;
}

void Module::toJson(Json &js) const
{
    if (!vars_.empty())
        vars_.toJson(js["vars"]);

    if (!children_.empty()) {
        Json &js_children = js["children"];
        for (auto &item : children_) {
            Json &js_child = js_children[item.module_ptr->name()];
            js_child["required"] = item.required;
            item.module_ptr->toJson(js_child);
        }
    }
}

}
}