#include <winpr/crt.h>
#include <winpr/tchar.h>
#include <winpr/path.h>
#include <winpr/wlog.h>

typedef struct
{
	UINT32 level;
	char* msg;
	char* channel;
} test_t;

static const char* function = NULL;
static const char* channels[] = { "com.test.channelA", "com.test.channelB" };

static const test_t messages[] = { { WLOG_INFO, "this is a test", "com.test.channelA" },
	                               { WLOG_INFO, "Just some info", "com.test.channelB" },
	                               { WLOG_WARN, "this is a %dnd %s", "com.test.channelA" },
	                               { WLOG_WARN, "we're warning a %dnd %s", "com.test.channelB" },
	                               { WLOG_ERROR, "this is an error", "com.test.channelA" },
	                               { WLOG_ERROR, "we've got an error", "com.test.channelB" },
	                               { WLOG_TRACE, "this is a trace output", "com.test.channelA" },
	                               { WLOG_TRACE, "leaving a trace behind", "com.test.channelB" } };

static BOOL success = TRUE;
static int pos = 0;

static BOOL check(const wLogMessage* msg)
{
	BOOL rc = TRUE;
	if (!msg)
		rc = FALSE;
	else if (strcmp(msg->FileName, __FILE__))
		rc = FALSE;
	else if (strcmp(msg->FunctionName, function))
		rc = FALSE;
	else if (strcmp(msg->PrefixString, messages[pos].channel))
		rc = FALSE;
	else if (msg->Level != messages[pos].level)
		rc = FALSE;
	else if (strcmp(msg->FormatString, messages[pos].msg))
		rc = FALSE;
	pos++;

	if (!rc)
	{
		fprintf(stderr, "Test failed!\n");
		success = FALSE;
	}
	return rc;
}

BOOL CallbackAppenderMessage(const wLogMessage* msg)
{
	check(msg);
	return TRUE;
}

BOOL CallbackAppenderData(const wLogMessage* msg)
{
	fprintf(stdout, "%s\n", __FUNCTION__);
	return TRUE;
}

BOOL CallbackAppenderImage(const wLogMessage* msg)
{
	fprintf(stdout, "%s\n", __FUNCTION__);
	return TRUE;
}

BOOL CallbackAppenderPackage(const wLogMessage* msg)
{
	fprintf(stdout, "%s\n", __FUNCTION__);
	return TRUE;
}

int TestWLogCallback(int argc, char* argv[])
{
	wLog* root;
	wLog* logA;
	wLog* logB;
	wLogLayout* layout;
	wLogAppender* appender;
	wLogCallbacks callbacks;

	function = __FUNCTION__;

	root = WLog_GetRoot();

	WLog_SetLogAppenderType(root, WLOG_APPENDER_CALLBACK);

	appender = WLog_GetLogAppender(root);

	callbacks.data = CallbackAppenderData;
	callbacks.image = CallbackAppenderImage;
	callbacks.message = CallbackAppenderMessage;
	callbacks.package = CallbackAppenderPackage;

	if (!WLog_ConfigureAppender(appender, "callbacks", (void*)&callbacks))
		return -1;

	layout = WLog_GetLogLayout(root);
	WLog_Layout_SetPrefixFormat(root, layout, "%mn");

	WLog_OpenAppender(root);

	logA = WLog_Get(channels[0]);
	logB = WLog_Get(channels[1]);

	WLog_SetLogLevel(logA, WLOG_TRACE);
	WLog_SetLogLevel(logB, WLOG_TRACE);

	WLog_Print(logA, messages[0].level, messages[0].msg);
	WLog_Print(logB, messages[1].level, messages[1].msg);
	WLog_Print(logA, messages[2].level, messages[2].msg, 2, "test");
	WLog_Print(logB, messages[3].level, messages[3].msg, 2, "time");
	WLog_Print(logA, messages[4].level, messages[4].msg);
	WLog_Print(logB, messages[5].level, messages[5].msg);
	WLog_Print(logA, messages[6].level, messages[6].msg);
	WLog_Print(logB, messages[7].level, messages[7].msg);

	WLog_CloseAppender(root);

	return success ? 0 : -1;
}