/*
This Source Code Form is subject to the
terms of the Mozilla Public License, v.
2.0. If a copy of the MPL was not
distributed with this file, You can
obtain one at
http://mozilla.org/MPL/2.0/.
*/
/*
MaSzyna EU07 locomotive simulator
Copyright (C) 2001-2004  Marcin Wozniak, Maciej Czapkiewicz and others
*/
/*
Authors:
MarcinW, McZapkie, Shaxbee, ABu, nbmx, youBy, Ra, winger, mamut, Q424,
Stele, firleju, szociu, hunter, ZiomalCl, OLI_EU and others
*/

#include "stdafx.h"

#include "application/application.h"
#include "simulation/simulationtest.h"
#include "utilities/Globals.h"
#include "utilities/Logs.h"
#include "utilities/crashdump.h"
#include <cstdlib>
#include <exception>
#include <string_view>
#include <thread>
#include <vector>

#ifdef _MSC_VER
#ifdef EU07_CPU_PREFLIGHT
// checks the processor before the C runtime starts, see utilities/cpu_preflight.cpp
#pragma comment(linker, "/subsystem:windows /ENTRY:eu07_preflight_entry")
#else
#pragma comment(linker, "/subsystem:windows /ENTRY:mainCRTStartup")
#endif
#endif

void export_e3d_standalone(std::string in, std::string out, int flags, bool dynamic);
int export_btf_standalone(std::string in, std::string out);
int generate_nodebank_previews(int Argc, char *Argv[]);

#include <string>
#include <utilities/Globals.h>

// lets the log service write what is still queued and stops it; does nothing once Application.exit() has done that
static void stop_log_service()
{
	Global.applicationQuitOrder = true;
	auto const it = Global.threads.find("LogService");
	if (it != Global.threads.end() && it->second.joinable())
		it->second.join();
}

namespace
{

// the log is written by a thread of its own, which the closing of the application stops and waits for. on the ways
// out that skip the closing (a failed start, an exception) what was logged last would be lost with the process
void finish_log()
{
	StopLogService();
	if (auto const logservice{Global.threads.find("LogService")};
	    logservice != Global.threads.end() && logservice->second.joinable() && logservice->second.get_id() != std::this_thread::get_id())
		logservice->second.join();
}

// ends a test run with its verdict on the standard output and in the exit code. a test the simulator didn't end
// (a failed start, an exception) ends with the error and Reason
[[noreturn]] void end_test_run(std::string_view const Reason)
{
	simulation::test().conclude(simulation::test_result::error, Reason);
	finish_log();
	fflush(stdout);
	fflush(stderr);
	// skip destructors, there are ordering errors which causes segfaults
	std::_Exit(simulation::test().exit_code());
}

// a test run (--test): whatever stops it, it ends with its verdict. any other run never gets here and ends as
// it always has
[[noreturn]] void run_test(int argc, char *argv[])
{
	// for what doesn't reach the catch below: an exception leaving a noexcept function, or on Linux one leaving
	// a thread (on Windows the handler is the main thread's only)
	std::set_terminate([] { end_test_run(simulation::describe_exception(std::current_exception())); });
	try
	{
		if (Application.init(argc, argv) == 0)
		{
			Application.run();
			Application.exit();
		}
	}
	catch (...)
	{
		end_test_run(simulation::describe_exception(std::current_exception()));
	}
	end_test_run("the simulator stopped before the test ended");
}

} // namespace

int main(int argc, char *argv[])
{
	crashdump_install();
	// init start timestamp
	Global.startTimestamp = std::chrono::steady_clock::now();

	int exitcode{0};

	// quick short-circuit for standalone e3d export
	if (argc == 6 && std::string(argv[1]) == "-e3d")
	{
		std::string in(argv[2]);
		std::string out(argv[3]);
		int flags = std::stoi(std::string(argv[4]));
		int dynamic = std::stoi(std::string(argv[5]));
		export_e3d_standalone(in, out, flags, dynamic);
	}
	// standalone generation of preview images of the node bank models: -generate-nodebank-previews [options]
	else if (argc >= 2 && std::string(argv[1]) == "-generate-nodebank-previews")
	{
		auto const result{generate_nodebank_previews(argc, argv)};
		fflush(stdout);
		fflush(stderr);
		std::_Exit(result);
	}
	// standalone conversion of text terrain file to binary format: -btf -s <text file> [-o <binary file>]
	else if (argc >= 4 && std::string(argv[1]) == "-btf")
	{
		std::string in;
		std::string out;
		for (int i = 2; i + 1 < argc; i += 2)
		{
			std::string const option(argv[i]);
			if (option == "-s")
			{
				in = argv[i + 1];
			}
			else if (option == "-o")
			{
				out = argv[i + 1];
			}
		}
		auto const result{export_btf_standalone(in, out)};
		fflush(stdout);
		fflush(stderr);
		std::_Exit(result);
	}
	// unattended test run of a scenery, which says its result in the exit code (eu07/maszyna#541)
	else if (simulation::parse_test_command_line(std::vector<char const *>(argv, argv + argc)).enabled)
	{
		run_test(argc, argv);
	}
	else
	{
		try
		{
			auto result{Application.init(argc, argv)};
			if (result == 0)
			{
				result = Application.run();
				Application.exit();
			}
			else
			{
				exitcode = result;
			}
		}
		catch (std::bad_alloc const &Error)
		{
			ErrorLog("Critical error, memory allocation failure: " + std::string(Error.what()));
		}
#ifdef _WIN32
		catch (std::runtime_error const &Error)
		{
			std::string msg = "Simulator crash occured :(\n";
			msg += Error.what();
			MessageBoxA(nullptr, msg.c_str(), "Simulator crashed :(", MB_ICONERROR);
		}
#endif
	}
	stop_log_service();
#ifndef _WIN32
	fflush(stdout);
	fflush(stderr);
#endif
	std::_Exit(exitcode); // skip destructors, there are ordering errors which causes segfaults
}