/**
* FreeRDP: A Remote Desktop Protocol Client
* FreeRDP Mac OS X Server
*
* Copyright 2012 Corey Clayton <can.of.tuna@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 <stdlib.h>
#include <errno.h>

#include "mf_info.h"
#include "mf_mountain_lion.h"
//#include "mf_update.h"

static mfInfo* mfInfoInstance = NULL;
//static int _IDcount = 0;

int mf_info_lock(mfInfo* mfi)
{
    
    int status = pthread_mutex_lock(&mfi->mutex);
    
    switch (status) {
        case 0:
            return TRUE;
            break;
            
        default:
            printf("mf_info_lock failed with %#X\n", status);
            return -1;
            break;
    }
    
}

int mf_info_try_lock(mfInfo* mfi, UINT32 ms)
{
    
    int status = pthread_mutex_trylock(&mfi->mutex);
    
    switch (status) {
        case 0:
            return TRUE;
            break;
            
        case EBUSY:
            return FALSE;
            break;
            
        default:
            printf("mf_info_try_lock failed with %#X\n", status);
            return -1;
            break;
    }
    
}

int mf_info_unlock(mfInfo* mfi)
{
    int status = pthread_mutex_unlock(&mfi->mutex);
    
    switch (status) {
        case 0:
            return TRUE;
            break;
            
        default:
            printf("mf_info_unlock failed with %#X\n", status);
            return -1;
            break;
    }
    
}

mfInfo* mf_info_init()
{
	mfInfo* mfi;

	mfi = (mfInfo*) malloc(sizeof(mfInfo));
    memset(mfi, 0, sizeof(mfInfo));
	

	if (mfi != NULL)
	{
/*		HKEY hKey;
		LONG status;
		DWORD dwType;
		DWORD dwSize;
		DWORD dwValue;
*/
        
        int mutexInitStatus = pthread_mutex_init(&mfi->mutex, NULL);
        
		if (mutexInitStatus != 0)
		{
			printf(_T("CreateMutex error: %#X\n"), mutexInitStatus);
		}

		mfi->peers = (freerdp_peer**) malloc(sizeof(freerdp_peer*) * MF_INFO_MAXPEERS);
		memset(mfi->peers, 0, sizeof(freerdp_peer*) * MF_INFO_MAXPEERS);

		//Set FPS
		mfi->framesPerSecond = MF_INFO_DEFAULT_FPS;

		/*status = RegOpenKeyEx(HKEY_LOCAL_MACHINE, _T("Software\\FreeRDP\\Server"), 0, KEY_READ | KEY_WOW64_64KEY, &hKey);
		if (status == ERROR_SUCCESS)
		{
			if (RegQueryValueEx(hKey, _T("FramesPerSecond"), NULL, &dwType, (BYTE*) &dwValue, &dwSize) == ERROR_SUCCESS)
				mfi->framesPerSecond = dwValue;		
		}
		RegCloseKey(hKey);*/

		//Set input toggle
		mfi->input_disabled = FALSE;

		/*status = RegOpenKeyEx(HKEY_LOCAL_MACHINE, _T("Software\\FreeRDP\\Server"), 0, KEY_READ | KEY_WOW64_64KEY, &hKey);
		if (status == ERROR_SUCCESS)
		{
			if (RegQueryValueEx(hKey, _T("DisableInput"), NULL, &dwType, (BYTE*) &dwValue, &dwSize) == ERROR_SUCCESS)
			{
				if (dwValue != 0)
					mfi->input_disabled = TRUE;
			}
		}
		RegCloseKey(hKey);*/

		
	}

	return mfi;
}

mfInfo* mf_info_get_instance()
{
	if (mfInfoInstance == NULL)
		mfInfoInstance = mf_info_init();

	return mfInfoInstance;
}

void mf_info_peer_register(mfInfo* mfi, mfPeerContext* context)
{
	if (mf_info_lock(mfi) > 0)
	{
		int i;
		int peerId;
		if (mfi->peerCount == MF_INFO_MAXPEERS)
		{
            printf("TODO: socketClose on OS X\n");
			//context->socketClose = TRUE;
			mf_info_unlock(mfi);
			return;
		}

		context->info = mfi;
		
		//get the offset of the top left corner of selected screen
		//EnumDisplayMonitors(NULL, NULL, mf_info_monEnumCB, 0);
		//_IDcount = 0;

        //initialize screen capture
		if (mfi->peerCount == 0)
        {
            mf_mlion_display_info(&mfi->servscreen_width, &mfi->servscreen_height, &mfi->scale);
            mf_mlion_screen_updates_init();
            mf_mlion_start_getting_screen_updates();
        }
			
		//look trhough the array of peers until an empty slot
		for(i=0; i<MF_INFO_MAXPEERS; ++i)
		{
			//empty index will be our peer id
			if (mfi->peers[i] == NULL)
			{
				peerId = i;
				break;
			}
		}

		mfi->peers[peerId] = ((rdpContext*) context)->peer;
		mfi->peers[peerId]->pId = peerId;
		mfi->peerCount++;
		printf("Registering Peer: id=%d #=%d\n", peerId, mfi->peerCount);

		mf_info_unlock(mfi);

		//mfreerdp_server_peer_callback_event(peerId, MF_SRV_CALLBACK_EVENT_CONNECT);
	}
}

void mf_info_peer_unregister(mfInfo* mfi, mfPeerContext* context)
{
	if (mf_info_lock(mfi) > 0)
	{
		int peerId;

		peerId = ((rdpContext*) context)->peer->pId;
		mfi->peers[peerId] = NULL;
		mfi->peerCount--;

		printf("Unregistering Peer: id=%d, #=%d\n", peerId, mfi->peerCount);

        //screen capture cleanup
		if (mfi->peerCount == 0)
        {
            mf_mlion_stop_getting_screen_updates();
        }
		mf_info_unlock(mfi);

		//mfreerdp_server_peer_callback_event(peerId, MF_SRV_CALLBACK_EVENT_DISCONNECT);
	}
}

BOOL mf_info_have_updates(mfInfo* mfi)
{
	if(mfi->framesWaiting == 0)
		return FALSE;

	return TRUE;
}

void mf_info_update_changes(mfInfo* mfi)
{
/*#ifdef WITH_WIN8
	mf_dxgi_nextFrame(mfi, mfi->framesPerSecond * 1000);
#else
	GETCHANGESBUF* buf;

	buf = (GETCHANGESBUF*) mfi->changeBuffer;
	mfi->nextUpdate = buf->buffer->counter;
#endif*/
}

void mf_info_find_invalid_region(mfInfo* mfi)
{
    mf_mlion_get_dirty_region(&mfi->invalid);
}

void mf_info_clear_invalid_region(mfInfo* mfi)
{
    mf_mlion_clear_dirty_region();
    mfi->invalid.height = 0;
    mfi->invalid.width = 0;
}

void mf_info_invalidate_full_screen(mfInfo* mfi)
{
    mfi->invalid.x = 0;
    mfi->invalid.y = 0;
    mfi->invalid.height = mfi->servscreen_height;
    mfi->invalid.height = mfi->servscreen_width;
}

BOOL mf_info_have_invalid_region(mfInfo* mfi)
{
    if (mfi->invalid.width * mfi->invalid.height == 0) {
        return FALSE;
    }
    return TRUE;
}

void mf_info_getScreenData(mfInfo* mfi, long* width, long* height, BYTE** pBits, int* pitch)
{
    *width = mfi->invalid.width / mfi->scale;
    *height = mfi->invalid.height / mfi->scale;
    *pitch = mfi->servscreen_width * mfi->scale * 4;
    
    mf_mlion_get_pixelData(mfi->invalid.x / mfi->scale, mfi->invalid.y / mfi->scale, *width, *height, pBits);
    
    *pBits = *pBits + (mfi->invalid.x * 4) + (*pitch * mfi->invalid.y);

}

/*
BOOL CALLBACK mf_info_monEnumCB(HMONITOR hMonitor, HDC hdcMonitor, LPRECT lprcMonitor, LPARAM dwData)
{
	mfInfo * mfi;

	mfi = mf_info_get_instance();

	if(_IDcount == mfi->screenID)
	{
		mfi->servscreen_xoffset = lprcMonitor->left;
		mfi->servscreen_yoffset = lprcMonitor->top;
	}
	
	_IDcount++;	

	return TRUE;
}
*/