/*
** Copyright (C) 2019-2020 Dirk-Jan C. Binnema <djcb@djcbsoftware.nl>
**
** This program 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, or (at your option) any
** later version.
**
** This program 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 this program; if not, write to the Free Software Foundation,
** Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
**
*/

#include "mu-runtime.hh"
#include "utils/mu-util.h"
#include "utils/mu-logger.hh"

#include <locale.h> /* for setlocale() */

#include <string>
#include <unordered_map>
static std::unordered_map<MuRuntimePath, std::string> RuntimePaths;

constexpr auto PartsDir  = "parts";
constexpr auto LogDir    = "log";
constexpr auto XapianDir = "xapian";
constexpr auto MuName    = "mu";
constexpr auto Bookmarks = "bookmarks";

static const std::string Sepa{G_DIR_SEPARATOR_S};

static void
init_paths_xdg()
{
	RuntimePaths.emplace(MU_RUNTIME_PATH_XAPIANDB,
	                     g_get_user_cache_dir() + Sepa + MuName + Sepa + XapianDir);
	RuntimePaths.emplace(MU_RUNTIME_PATH_CACHE, g_get_user_cache_dir() + Sepa + MuName);
	RuntimePaths.emplace(MU_RUNTIME_PATH_MIMECACHE,
	                     g_get_user_cache_dir() + Sepa + MuName + Sepa + PartsDir);
	RuntimePaths.emplace(MU_RUNTIME_PATH_LOGDIR, g_get_user_cache_dir() + Sepa + MuName);
	RuntimePaths.emplace(MU_RUNTIME_PATH_BOOKMARKS, g_get_user_config_dir() + Sepa + MuName);
}

static void
init_paths_muhome(const char* muhome)
{
	RuntimePaths.emplace(MU_RUNTIME_PATH_XAPIANDB, muhome + Sepa + XapianDir);
	RuntimePaths.emplace(MU_RUNTIME_PATH_CACHE, muhome);
	RuntimePaths.emplace(MU_RUNTIME_PATH_MIMECACHE, muhome + Sepa + PartsDir);
	RuntimePaths.emplace(MU_RUNTIME_PATH_LOGDIR, muhome + Sepa + LogDir);
	RuntimePaths.emplace(MU_RUNTIME_PATH_BOOKMARKS, muhome + Sepa + Bookmarks);
}

gboolean
mu_runtime_init(const char* muhome, const char* name, gboolean debug)
{
	g_return_val_if_fail(RuntimePaths.empty(), FALSE);
	g_return_val_if_fail(name, FALSE);

	setlocale(LC_ALL, "");

	if (muhome)
		init_paths_muhome(muhome);
	else
		init_paths_xdg();

	for (const auto& d : RuntimePaths) {
		char* dir;
		if (d.first == MU_RUNTIME_PATH_BOOKMARKS) // special case
			dir = g_path_get_dirname(d.second.c_str());
		else
			dir = g_strdup(d.second.c_str());

		auto ok = mu_util_create_dir_maybe(dir, 0700, TRUE);
		if (!ok) {
			g_critical("failed to create %s", dir);
			g_free(dir);
			mu_runtime_uninit();
			return FALSE;
		}
		g_free(dir);
	}

	const auto log_path = RuntimePaths[MU_RUNTIME_PATH_LOGDIR] + Sepa + name + ".log";

	using namespace Mu;
	LogOptions opts{LogOptions::None};
	if (debug)
		opts |= (LogOptions::Debug | LogOptions::None);

	Mu::log_init(log_path, opts);

	return TRUE;
}

void
mu_runtime_uninit(void)
{
	RuntimePaths.clear();
	Mu::log_uninit();
}

const char*
mu_runtime_path(MuRuntimePath path)
{
	const auto it = RuntimePaths.find(path);
	if (it == RuntimePaths.end())
		return NULL;
	else
		return it->second.c_str();
}