// Copyright (C) 2016 xaizek <xaizek@posteo.net>
//
// This file is part of uncov.
//
// uncov is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// uncov 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 Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with uncov.  If not, see <http://www.gnu.org/licenses/>.

#include "TestUtils.hpp"

#include <boost/filesystem/operations.hpp>

#include "Settings.hpp"

namespace fs = boost::filesystem;

static void copyDir(const fs::path &src, const fs::path &dst);

TempDirCopy::TempDirCopy(const std::string &from, const std::string &to,
                         bool force)
    : to(to)
{
    if (force) {
        fs::remove_all(to);
    }
    copyDir(from, to);
}

static void
copyDir(const fs::path &src, const fs::path &dst)
{
    fs::create_directory(dst);
    for (fs::directory_entry &e : fs::directory_iterator(src)) {
        fs::path path = e.path();
        if (fs::is_directory(path)) {
            copyDir(path, dst/path.filename());
        } else {
            fs::copy_file(path, dst/path.filename());
        }
    }
}

TempDirCopy::~TempDirCopy()
{
    try {
        fs::remove_all(to);
    } catch (...) {
        // Don't throw from the destructor.
    }
}

FileRestorer::FileRestorer(const std::string &from, const std::string &to)
    : from(from), to(to)
{
    fs::copy_file(from, to);
}

FileRestorer::~FileRestorer()
{
    fs::remove(from);
    fs::rename(to, from);
}

Settings &
getSettings()
{
    static TestSettings settings;
    return settings;
}