#include <winpr/crt.h>

#include <winpr/error.h>

#include <winpr/wtsapi.h>

#include <winpr/environment.h>

int TestWtsApiSessionNotification(int argc, char* argv[])

{

	HWND hWnd = NULL;

	BOOL bSuccess;

	DWORD dwFlags;

#ifndef _WIN32

	if (!GetEnvironmentVariableA("WTSAPI_LIBRARY", NULL, 0))

	{

		printf("%s: No RDS environment detected, skipping test\n", __FUNCTION__);

		return 0;

	}

#else

	/* We create a message-only window and use the predefined class name "STATIC" for simplicity */

	hWnd = CreateWindowA("STATIC", "TestWtsApiSessionNotification", 0, 0, 0, 0, 0, HWND_MESSAGE,

	                     NULL, NULL, NULL);

	if (!hWnd)

	{

		printf("%s: error creating message-only window: %" PRIu32 "\n", __FUNCTION__,

		       GetLastError());

		return -1;

	}

#endif

	dwFlags = NOTIFY_FOR_ALL_SESSIONS;

	bSuccess = WTSRegisterSessionNotification(hWnd, dwFlags);

	if (!bSuccess)

	{

		printf("%s: WTSRegisterSessionNotification failed: %" PRIu32 "\n", __FUNCTION__,

		       GetLastError());

		return -1;

	}

	bSuccess = WTSUnRegisterSessionNotification(hWnd);

#ifdef _WIN32

	if (hWnd)

	{

		DestroyWindow(hWnd);

		hWnd = NULL;

	}

#endif

	if (!bSuccess)

	{

		printf("%s: WTSUnRegisterSessionNotification failed: %" PRIu32 "\n", __FUNCTION__,

		       GetLastError());

		return -1;

	}

	return 0;

}