/*
 * .============.
 * // 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)
 return false;

 if (child == nullptr)
 return false;

 auto iter = std::find_if(children_.begin(), children_.end(),
 [child] (const ModuleItem &item) {
 return item.module_ptr == child;
 }
 );
 if (iter != children_.end())
 return false;

 children_.emplace_back(ModuleItem{ child, required });
 return true;
}

void Module::fillDefaultConfig(Json &js_parent)
{
 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;
}

}
}