/**
 * FreeRDP: A Remote Desktop Protocol Implementation
 * FreeRDP Windows Server
 *
 * Copyright 2012 Marc-Andre Moreau <marcandre.moreau@gmail.com>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

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

#include <winpr/tchar.h>
#include <winpr/windows.h>

#include "wf_interface.h"

#include "wfreerdp.h"

int IDcount = 0;

BOOL CALLBACK moncb(HMONITOR hMonitor, HDC hdcMonitor, LPRECT lprcMonitor, LPARAM dwData)
{
	DEBUG_MSG("%d\t(%d, %d), (%d, %d)\n",
		IDcount, lprcMonitor->left, lprcMonitor->top,
		lprcMonitor->right, lprcMonitor->bottom);

	IDcount++;

	return TRUE;
}

int main(int argc, char* argv[])
{
	BOOL screen_selected = FALSE;
	int index;
	wfServer* server;

	server = wfreerdp_server_new();

	set_screen_id(0);

	//handle args
	index = 1;
	while (index < argc)
	{
		//first the args that will cause the program to terminate
		if (strcmp("--list-screens", argv[index]) == 0)
		{
			_TCHAR name[128];
			int width;
			int height;
			int bpp;
			int i;

			_tprintf(_T("Detecting screens...\n"));
			_tprintf(_T("\nID\tResolution\t\tName (Interface)\n\n"));
		
			for (i=0; ; i++)
			{
				if (get_screen_info(i, name, &width, &height, &bpp) != 0)
				{
					if ( (width * height * bpp) == 0 )
						continue;

					_tprintf(_T("%d\t%dx%dx%d\t"), i, width, height, bpp);
					_tprintf(_T("%s\n"), name);
				}
				else
				{
					break;
				}
			}

			{
				int vscreen_w;
				int vscreen_h;
				vscreen_w = GetSystemMetrics(SM_CXVIRTUALSCREEN);
				vscreen_h = GetSystemMetrics(SM_CYVIRTUALSCREEN);

				DEBUG_MSG("\n");
				EnumDisplayMonitors(NULL, NULL, moncb, 0);
				IDcount = 0;
				DEBUG_MSG("\nVirtual Screen = %dx%d\n", vscreen_w, vscreen_h);
			}

			return 0;
		}
	
		if (strcmp("--screen", argv[index]) == 0)
		{
			screen_selected = TRUE;
			index++;
			if (index == argc)
			{
				DEBUG_MSG("missing screen id parameter\n");
				return 0;
			}

			set_screen_id(atoi(argv[index]));
			index++;
		}

		if (index == argc - 1)
		{
			server->port = (DWORD) atoi(argv[index]);
			break;
		}
	}

	if (screen_selected == FALSE)
	{
		_TCHAR name[128];
		int width;
		int height;
		int bpp;
		int i;

		_tprintf(_T("screen id not provided. attempting to detect...\n"));
		_tprintf(_T("Detecting screens...\n"));
		_tprintf(_T("\nID\tResolution\t\tName (Interface)\n\n"));

		for (i=0; ; i++)
		{
			if (get_screen_info(i, name, &width, &height, &bpp) != 0)
			{
				if ( (width * height * bpp) == 0 )
					continue;

				_tprintf(_T("%d\t%dx%dx%d\t"), i, width, height, bpp);
				_tprintf(_T("%s\n"), name);
				set_screen_id(i);
				break;
			}
			else
			{
				break;
			}
		}
	}

	DEBUG_MSG("Starting server\n");

	wfreerdp_server_start(server);

	WaitForSingleObject(server->thread, INFINITE);

	DEBUG_MSG("Stopping server\n");

	wfreerdp_server_stop(server);
	wfreerdp_server_free(server);

	return 0;
}