/*
 RDP ui callbacks

 Copyright 2013 Thincast Technologies GmbH, Authors: Martin Fleisz, Dorian Johnson

 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/.
 */

#import <Foundation/Foundation.h>

#import <freerdp/gdi/gdi.h>
#import "ios_freerdp_ui.h"

#import "RDPSession.h"

#pragma mark -
#pragma mark Certificate authentication

static void ios_resize_display_buffer(mfInfo* mfi);
static BOOL ios_ui_authenticate_raw(freerdp* instance, char** username,
                                    char** password,
                                    char** domain, const char* title)
{
	mfInfo* mfi = MFI_FROM_INSTANCE(instance);
	NSMutableDictionary* params = [NSMutableDictionary dictionaryWithObjectsAndKeys:
	                               (*username) ? [NSString stringWithUTF8String:*username] : @"", @"username",
	                               (*password) ? [NSString stringWithUTF8String:*password] : @"", @"password",
	                               (*domain) ? [NSString stringWithUTF8String:*domain] : @"", @"domain",
	                               [NSString stringWithUTF8String:instance->settings->ServerHostname],
	                               @"hostname", // used for the auth prompt message; not changed
	                               nil];
	// request auth UI
	[mfi->session performSelectorOnMainThread:@selector(
	     sessionRequestsAuthenticationWithParams:) withObject:params waitUntilDone:YES];
	// wait for UI request to be completed
	[[mfi->session uiRequestCompleted] lock];
	[[mfi->session uiRequestCompleted] wait];
	[[mfi->session uiRequestCompleted] unlock];

	if (![[params valueForKey:@"result"] boolValue])
	{
		mfi->unwanted = YES;
		return FALSE;
	}

	// Free old values
	free(*username);
	free(*password);
	free(*domain);
	// set values back
	*username = strdup([[params objectForKey:@"username"] UTF8String]);
	*password = strdup([[params objectForKey:@"password"] UTF8String]);
	*domain = strdup([[params objectForKey:@"domain"] UTF8String]);

	if (!(*username) || !(*password) || !(*domain))
	{
		free(*username);
		free(*password);
		free(*domain);
		return FALSE;
	}

	return TRUE;
}

BOOL ios_ui_authenticate(freerdp* instance, char** username, char** password,
                         char** domain)
{
	return ios_ui_authenticate_raw(instance, username, password, domain, "");
}

BOOL ios_ui_gw_authenticate(freerdp* instance, char** username, char** password,
                            char** domain)
{
	return ios_ui_authenticate_raw(instance, username, password, domain, "gateway");
}


DWORD ios_ui_verify_certificate(freerdp* instance,
                                const char* common_name,
                                const char* subject,
                                const char* issuer,
                                const char* fingerprint,
                                BOOL host_mismatch)
{
	// check whether we accept all certificates
	if ([[NSUserDefaults standardUserDefaults] boolForKey:
	     @"security.accept_certificates"] == YES)
		return 2;

	mfInfo* mfi = MFI_FROM_INSTANCE(instance);
	NSMutableDictionary* params = [NSMutableDictionary dictionaryWithObjectsAndKeys:
	                               (subject) ? [NSString stringWithUTF8String:subject] : @"", @"subject",
	                               (issuer) ? [NSString stringWithUTF8String:issuer] : @"", @"issuer",
	                               (fingerprint) ? [NSString stringWithUTF8String:subject] : @"", @"fingerprint",
	                               nil];
	// request certificate verification UI
	[mfi->session performSelectorOnMainThread:@selector(
	     sessionVerifyCertificateWithParams:) withObject:params waitUntilDone:YES];
	// wait for UI request to be completed
	[[mfi->session uiRequestCompleted] lock];
	[[mfi->session uiRequestCompleted] wait];
	[[mfi->session uiRequestCompleted] unlock];

	if (![[params valueForKey:@"result"] boolValue])
	{
		mfi->unwanted = YES;
		return 0;
	}

	return 1;
}

DWORD ios_ui_verify_changed_certificate(freerdp* instance,
                                        const char* common_name,
                                        const char* subject,
                                        const char* issuer,
                                        const char* new_fingerprint,
                                        const char* old_subject,
                                        const char* old_issuer,
                                        const char* old_fingerprint)
{
	return ios_ui_verify_certificate(instance, common_name, subject, issuer,
	                                 new_fingerprint, FALSE);
}


#pragma mark -
#pragma mark Graphics updates

BOOL ios_ui_begin_paint(rdpContext* context)
{
	rdpGdi* gdi = context->gdi;
	gdi->primary->hdc->hwnd->invalid->null = TRUE;
	return TRUE;
}

BOOL ios_ui_end_paint(rdpContext* context)
{
	mfInfo* mfi = MFI_FROM_INSTANCE(context->instance);
	rdpGdi* gdi = context->gdi;
	CGRect dirty_rect = CGRectMake(gdi->primary->hdc->hwnd->invalid->x,
	                               gdi->primary->hdc->hwnd->invalid->y, gdi->primary->hdc->hwnd->invalid->w,
	                               gdi->primary->hdc->hwnd->invalid->h);

	if (!gdi->primary->hdc->hwnd->invalid->null)
		[mfi->session performSelectorOnMainThread:@selector(
		     setNeedsDisplayInRectAsValue:) withObject:[NSValue valueWithCGRect:dirty_rect]
		 waitUntilDone:NO];

	return TRUE;
}


BOOL ios_ui_resize_window(rdpContext* context)
{
	rdpSettings* settings;
	rdpGdi* gdi;

	if (!context || !context->settings)
		return FALSE;

	settings = context->settings;
	gdi = context->gdi;

	if (!gdi_resize(gdi, settings->DesktopWidth, settings->DesktopHeight))
		return FALSE;

	ios_resize_display_buffer(MFI_FROM_INSTANCE(context->instance));
	return TRUE;
}


#pragma mark -
#pragma mark Exported

static void ios_create_bitmap_context(mfInfo* mfi)
{
	[mfi->session performSelectorOnMainThread:@selector(
	     sessionBitmapContextWillChange) withObject:nil waitUntilDone:YES];
	rdpGdi* gdi = mfi->instance->context->gdi;
	CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();

	if (GetBytesPerPixel(gdi->dstFormat) == 2)
		mfi->bitmap_context = CGBitmapContextCreate(gdi->primary_buffer, gdi->width,
		                      gdi->height, 5, gdi->stride, colorSpace,
		                      kCGBitmapByteOrder16Little | kCGImageAlphaNoneSkipFirst);
	else
		mfi->bitmap_context = CGBitmapContextCreate(gdi->primary_buffer, gdi->width,
		                      gdi->height, 8, gdi->stride, colorSpace,
		                      kCGBitmapByteOrder32Little | kCGImageAlphaNoneSkipFirst);

	CGColorSpaceRelease(colorSpace);
	[mfi->session performSelectorOnMainThread:@selector(
	     sessionBitmapContextDidChange) withObject:nil waitUntilDone:YES];
}

void ios_allocate_display_buffer(mfInfo* mfi)
{
	ios_create_bitmap_context(mfi);
}

void ios_resize_display_buffer(mfInfo* mfi)
{
	// Release the old context in a thread-safe manner
	CGContextRef old_context = mfi->bitmap_context;
	mfi->bitmap_context = NULL;
	CGContextRelease(old_context);
	// Create the new context
	ios_create_bitmap_context(mfi);
}