/*
** Copyright (C) 2008-2023 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 "config.h"

#include <glib.h>
#include <glib/gstdio.h>

#include <stdlib.h>
#include <unistd.h>
#include <string.h>

#include <langinfo.h>
#include <locale.h>

#include "utils/mu-utils.hh"
#include "utils/mu-test-utils.hh"
#include "utils/mu-utils-file.hh"
#include "utils/mu-error.hh"


using namespace Mu;

std::string
Mu::test_random_tmpdir()
{
	auto&& dir = mu_format("{}{}mu-test-{}{}test-{:x}",
			       g_get_tmp_dir(),
			       G_DIR_SEPARATOR,
			       getuid(),
			       G_DIR_SEPARATOR,
			       ::random() * getpid() * ::time({}));

	auto res = g_mkdir_with_parents(dir.c_str(), 0700);
	g_assert(res != -1);

	return dir;
}

const char*
Mu::set_tz(const char* tz)
{
	static const char* oldtz;

	oldtz = getenv("TZ");
	if (tz)
		setenv("TZ", tz, 1);
	else
		unsetenv("TZ");

	tzset();
	return oldtz;
}

bool
Mu::set_en_us_utf8_locale()
{
	setenv("LC_ALL", "en_US.UTF-8", 1);
	setlocale(LC_ALL, "en_US.UTF-8");

	if (strcmp(nl_langinfo(CODESET), "UTF-8") != 0) {
		mu_println("Note: Unit tests require the en_US.utf8 locale. "
			   "Ignoring test cases.");
		return false;
	}

	return true;
}

static void
black_hole(void)
{
	return; /* do nothing */
}

void
Mu::mu_test_init(int *argc, char ***argv)
{
	const auto tmpdir{test_random_tmpdir()};

	g_setenv("MU_TEST", "yes", TRUE);
	g_setenv("XDG_CACHE_HOME", tmpdir.c_str(), TRUE);

	g_test_init(argc, argv, NULL);

	g_test_bug_base("https://github.com/djcb/mu/issues/");

	if (!g_test_verbose())
		g_log_set_handler(
			NULL,
			(GLogLevelFlags)(G_LOG_LEVEL_MASK |
					 G_LOG_FLAG_FATAL | G_LOG_FLAG_RECURSION),
			(GLogFunc)black_hole, NULL);
}

void
Mu::allow_warnings()
{
	g_test_log_set_fatal_handler(
	    [](const char*, GLogLevelFlags, const char*, gpointer) { return FALSE; },
	    {});
}

Mu::TempDir::TempDir(bool autodelete): autodelete_{autodelete} {

	if (auto res{make_temp_dir()}; !res)
		throw res.error();
	else
		path_ = std::move(*res);

	mu_debug("created '{}'", path_);
}

Mu::TempDir::~TempDir()
{
	if (::access(path_.c_str(), F_OK) != 0)
		return; /* nothing to do */

	if (!autodelete_) {
		mu_debug("_not_ deleting {}", path_);
		return;
	}

	/* ugly */
	GError *err{};
	const auto cmd{mu_format("/bin/rm -rf '{}'", path_)};
	if (!g_spawn_command_line_sync(cmd.c_str(), NULL,
				       NULL, NULL, &err)) {
		mu_warning("error: {}", err ? err->message : "?");
		g_clear_error(&err);
	} else
		mu_debug("removed '{}'", path_);
}