/*

 RDP Keyboard helper

 Copyright 2013 Thincast Technologies GmbH, Author: Martin Fleisz

 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 "RDPKeyboard.h"

#include <freerdp/locale/keyboard.h>

@interface RDPKeyboard (Private)

- (void)sendVirtualKey:(int)vKey up:(BOOL)up;

- (void)handleSpecialKey:(int)character;

- (void)handleAlphaNumChar:(int)character;

- (void)notifyDelegateModifiersChanged;

@end

@implementation RDPKeyboard

@synthesize delegate = _delegate, ctrlPressed = _ctrl_pressed, altPressed = _alt_pressed,

            shiftPressed = _shift_pressed, winPressed = _win_pressed;

- (id)init

{

	if ((self = [super init]) != nil)

	{

		[self initWithSession:nil delegate:nil];

		memset(_virtual_key_map, 0, sizeof(_virtual_key_map));

		memset(_unicode_map, 0, sizeof(_unicode_map));

		// init vkey map - used for alpha-num characters

		_virtual_key_map['0'] = VK_KEY_0;

		_virtual_key_map['1'] = VK_KEY_1;

		_virtual_key_map['2'] = VK_KEY_2;

		_virtual_key_map['3'] = VK_KEY_3;

		_virtual_key_map['4'] = VK_KEY_4;

		_virtual_key_map['5'] = VK_KEY_5;

		_virtual_key_map['6'] = VK_KEY_6;

		_virtual_key_map['7'] = VK_KEY_7;

		_virtual_key_map['8'] = VK_KEY_8;

		_virtual_key_map['9'] = VK_KEY_9;

		_virtual_key_map['a'] = VK_KEY_A;

		_virtual_key_map['b'] = VK_KEY_B;

		_virtual_key_map['c'] = VK_KEY_C;

		_virtual_key_map['d'] = VK_KEY_D;

		_virtual_key_map['e'] = VK_KEY_E;

		_virtual_key_map['f'] = VK_KEY_F;

		_virtual_key_map['g'] = VK_KEY_G;

		_virtual_key_map['h'] = VK_KEY_H;

		_virtual_key_map['i'] = VK_KEY_I;

		_virtual_key_map['j'] = VK_KEY_J;

		_virtual_key_map['k'] = VK_KEY_K;

		_virtual_key_map['l'] = VK_KEY_L;

		_virtual_key_map['m'] = VK_KEY_M;

		_virtual_key_map['n'] = VK_KEY_N;

		_virtual_key_map['o'] = VK_KEY_O;

		_virtual_key_map['p'] = VK_KEY_P;

		_virtual_key_map['q'] = VK_KEY_Q;

		_virtual_key_map['r'] = VK_KEY_R;

		_virtual_key_map['s'] = VK_KEY_S;

		_virtual_key_map['t'] = VK_KEY_T;

		_virtual_key_map['u'] = VK_KEY_U;

		_virtual_key_map['v'] = VK_KEY_V;

		_virtual_key_map['w'] = VK_KEY_W;

		_virtual_key_map['x'] = VK_KEY_X;

		_virtual_key_map['y'] = VK_KEY_Y;

		_virtual_key_map['z'] = VK_KEY_Z;

		// init scancode map - used for special characters

		_unicode_map['-'] = 45;

		_unicode_map['/'] = 47;

		_unicode_map[':'] = 58;

		_unicode_map[';'] = 59;

		_unicode_map['('] = 40;

		_unicode_map[')'] = 41;

		_unicode_map['&'] = 38;

		_unicode_map['@'] = 64;

		_unicode_map['.'] = 46;

		_unicode_map[','] = 44;

		_unicode_map['?'] = 63;

		_unicode_map['!'] = 33;

		_unicode_map['\''] = 39;

		_unicode_map['\"'] = 34;

		_unicode_map['['] = 91;

		_unicode_map[']'] = 93;

		_unicode_map['{'] = 123;

		_unicode_map['}'] = 125;

		_unicode_map['#'] = 35;

		_unicode_map['%'] = 37;

		_unicode_map['^'] = 94;

		_unicode_map['*'] = 42;

		_unicode_map['+'] = 43;

		_unicode_map['='] = 61;

		_unicode_map['_'] = 95;

		_unicode_map['\\'] = 92;

		_unicode_map['|'] = 124;

		_unicode_map['~'] = 126;

		_unicode_map['<'] = 60;

		_unicode_map['>'] = 62;

		_unicode_map['$'] = 36;

	}

	return self;

}

- (void)dealloc

{

	[super dealloc];

}

#pragma mark -

#pragma mark class methods

// return a keyboard instance

+ (RDPKeyboard *)getSharedRDPKeyboard

{

	static RDPKeyboard *_shared_keyboard = nil;

	if (_shared_keyboard == nil)

	{

		@synchronized(self)

		{

			if (_shared_keyboard == nil)

				_shared_keyboard = [[RDPKeyboard alloc] init];

		}

	}

	return _shared_keyboard;

}

// reset the keyboard instance and assign the given rdp instance

- (void)initWithSession:(RDPSession *)session delegate:(NSObject<RDPKeyboardDelegate> *)delegate

{

	_alt_pressed = NO;

	_ctrl_pressed = NO;

	_shift_pressed = NO;

	_win_pressed = NO;

	_session = session;

	_delegate = delegate;

}

- (void)reset

{

	// reset pressed ctrl, alt, shift or win key

	if (_shift_pressed)

		[self toggleShiftKey];

	if (_alt_pressed)

		[self toggleAltKey];

	if (_ctrl_pressed)

		[self toggleCtrlKey];

	if (_win_pressed)

		[self toggleWinKey];

}

// handles button pressed input event from the iOS keyboard

// performs all conversions etc.

- (void)sendUnicode:(int)character

{

	if (isalnum(character))

		[self handleAlphaNumChar:character];

	else

		[self handleSpecialKey:character];

	[self reset];

}

// send a backspace key press

- (void)sendVirtualKeyCode:(int)keyCode

{

	[self sendVirtualKey:keyCode up:NO];

	[self sendVirtualKey:keyCode up:YES];

}

#pragma mark modifier key handling

// toggle ctrl key, returns true if pressed, otherwise false

- (void)toggleCtrlKey

{

	[self sendVirtualKey:VK_LCONTROL up:_ctrl_pressed];

	_ctrl_pressed = !_ctrl_pressed;

	[self notifyDelegateModifiersChanged];

}

// toggle alt key, returns true if pressed, otherwise false

- (void)toggleAltKey

{

	[self sendVirtualKey:VK_LMENU up:_alt_pressed];

	_alt_pressed = !_alt_pressed;

	[self notifyDelegateModifiersChanged];

}

// toggle shift key, returns true if pressed, otherwise false

- (void)toggleShiftKey

{

	[self sendVirtualKey:VK_LSHIFT up:_shift_pressed];

	_shift_pressed = !_shift_pressed;

	[self notifyDelegateModifiersChanged];

}

// toggle windows key, returns true if pressed, otherwise false

- (void)toggleWinKey

{

	[self sendVirtualKey:(VK_LWIN | KBDEXT) up:_win_pressed];

	_win_pressed = !_win_pressed;

	[self notifyDelegateModifiersChanged];

}

#pragma mark Sending special key strokes

- (void)sendEnterKeyStroke

{

	[self sendVirtualKeyCode:(VK_RETURN | KBDEXT)];

}

- (void)sendEscapeKeyStroke

{

	[self sendVirtualKeyCode:VK_ESCAPE];

}

- (void)sendBackspaceKeyStroke

{

	[self sendVirtualKeyCode:VK_BACK];

}

@end

#pragma mark -

@implementation RDPKeyboard (Private)

- (void)handleAlphaNumChar:(int)character

{

	// if we recive an uppercase letter - make it lower and send an shift down event to server

	BOOL shift_was_sent = NO;

	if (isupper(character) && _shift_pressed == NO)

	{

		character = tolower(character);

		[self sendVirtualKey:VK_LSHIFT up:NO];

		shift_was_sent = YES;

	}

	// convert the character to a VK

	int vk = _virtual_key_map[character];

	if (vk != 0)

	{

		// send key pressed

		[self sendVirtualKey:vk up:NO];

		[self sendVirtualKey:vk up:YES];

	}

	// send the missing shift up if we had a shift down

	if (shift_was_sent)

		[self sendVirtualKey:VK_LSHIFT up:YES];

}

- (void)handleSpecialKey:(int)character

{

	NSDictionary *eventDescriptor = nil;

	if (character < 256)

	{

		// convert the character to a unicode character

		int code = _unicode_map[character];

		if (code != 0)

			eventDescriptor = [NSDictionary

			    dictionaryWithObjectsAndKeys:@"keyboard", @"type", @"unicode", @"subtype",

			                                 [NSNumber numberWithUnsignedShort:0], @"flags",

			                                 [NSNumber numberWithUnsignedShort:code],

			                                 @"unicode_char", nil];

	}

	if (eventDescriptor == nil)

		eventDescriptor = [NSDictionary

		    dictionaryWithObjectsAndKeys:@"keyboard", @"type", @"unicode", @"subtype",

		                                 [NSNumber numberWithUnsignedShort:0], @"flags",

		                                 [NSNumber numberWithUnsignedShort:character],

		                                 @"unicode_char", nil];

	[_session sendInputEvent:eventDescriptor];

}

// sends the vk code to the session

- (void)sendVirtualKey:(int)vKey up:(BOOL)up

{

	DWORD scancode = GetVirtualScanCodeFromVirtualKeyCode(vKey, 4);

	int flags = (up ? KBD_FLAGS_RELEASE : KBD_FLAGS_DOWN);

	flags |= ((scancode & KBDEXT) ? KBD_FLAGS_EXTENDED : 0);

	[_session

	    sendInputEvent:[NSDictionary

	                       dictionaryWithObjectsAndKeys:@"keyboard", @"type", @"scancode",

	                                                    @"subtype",

	                                                    [NSNumber numberWithUnsignedShort:flags],

	                                                    @"flags",

	                                                    [NSNumber

	                                                        numberWithUnsignedShort:(scancode &

	                                                                                 0xFF)],

	                                                    @"scancode", nil]];

}

- (void)notifyDelegateModifiersChanged

{

	if ([[self delegate] respondsToSelector:@selector(modifiersChangedForKeyboard:)])

		[[self delegate] modifiersChangedForKeyboard:self];

}

@end