// Copyright (C) 2015 xaizek <xaizek@openmailbox.org>
//
// This file is part of dit.
//
// dit is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// dit is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with dit.  If not, see <http://www.gnu.org/licenses/>.

#include "Config.hpp"

#include <string>
#include <utility>
#include <vector>

#include <boost/property_tree/ptree.hpp>
#include <boost/property_tree/info_parser.hpp>
#include <boost/filesystem.hpp>

#include "utils/Passkey.hpp"
#include "utils/propsRange.hpp"

namespace fs = boost::filesystem;
namespace pt = boost::property_tree;

Config::Config(std::string path, Config *parent)
    : path(std::move(path)), parent(parent)
{
}

std::string
Config::get(const std::string &key)
{
    ensureLoaded();

    std::string val;
    try {
        val = props.get<std::string>(key);
    } catch (boost::property_tree::ptree_bad_path &) {
        if (parent == nullptr) {
            throw;
        }

        // Do nothing, just execute the code below.
    }

    if (parent == nullptr || !val.empty()) {
        return val;
    }
    return parent->get(key);
}

std::string
Config::get(const std::string &key, const std::string &def)
{
    ensureLoaded();
    std::string actualDef = (parent == nullptr) ? def : parent->get(key, def);
    std::string val = props.get<std::string>(key, actualDef);
    return val.empty() ? std::move(actualDef) : std::move(val);
}

std::vector<std::string>
Config::list(const std::string &path)
{
    ensureLoaded();

    boost::optional<boost::property_tree::ptree &> subtree =
        props.get_child_optional(path);
    if (!subtree) {
        return {};
    }

    std::vector<std::string> list;
    for (const std::string &propName : propsRange(*subtree)) {
        const std::string &fullPath = path.empty()
                                    ? propName
                                    : path + '.' + propName;
        if (propName[0] != '!' && !get(fullPath, std::string()).empty()) {
            list.push_back(propName);
        }
    }
    return list;
}

void
Config::set(const std::string &key, const std::string &val)
{
    ensureLoaded();

    try {
        if (get(key) == val) {
            return;
        }
    } catch (pt::ptree_bad_path &) {
        // Skip the check if field doesn't exist.
    }

    props.put(key, val);
    markModified();
}

void
Config::save()
{
    if (isModified()) {
        write_info(path, props);
    }
}

bool
Config::isModified(pk<Tests>) const
{
    return isModified();
}

void
Config::load()
{
    try {
        read_info(path, props);
    } catch (pt::info_parser_error &) {
        // It's OK if specific configuration doesn't exist yet.
        if (fs::exists(path)) {
            throw;
        }
    }
}