/*
 Utility functions

 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 "Utils.h"
#import "OrderedDictionary.h"
#import "TSXAdditions.h"

#import <freerdp/input.h>
#import <freerdp/version.h>
#import "config.h"

#include <sys/types.h>
#include <sys/sysctl.h>
#include <sys/socket.h>
#include <ifaddrs.h>
#include <net/if_dl.h>

BOOL ScanHostNameAndPort(NSString* address, NSString** host,
                         unsigned short* port)
{
	*host = @"";
	*port = 0;

	if (![address length])
		return NO;

	NSURL* url = [NSURL URLWithString:[NSString stringWithFormat:@"rdp://%@",
	                                   address]];

	if (!url || ![[url host] length])
		return NO;

	*host = [url host];
	*port = [[url port] unsignedShortValue];
	return YES;
}

#pragma mark -
#pragma mark Working with Screen Resolutions

NSString* LocalizedFitScreen()
{
	return NSLocalizedString(@"Automatic",
	                         @"Screen resolution selector: Automatic resolution (Full Screen on iPad, reasonable size on iPhone)");
}

NSString* LocalizedCustom()
{
	return NSLocalizedString(@"Custom", @"Screen resolution selector: Custom");
}

BOOL ScanScreenResolution(NSString* description, int* width, int* height,
                          TSXScreenOptions* type)
{
	*height = 0;
	*width = 0;
	*type = TSXScreenOptionFixed;

	if ([description isEqualToString:LocalizedFitScreen()])
	{
		*type = TSXScreenOptionFitScreen;
		return YES;
	}
	else if ([description isEqualToString:LocalizedCustom()])
	{
		*type = TSXScreenOptionCustom;
		return YES;
	}

	NSArray* resolution_components = [description
	                                  componentsSeparatedByCharactersInSet:[NSCharacterSet
	                                          characterSetWithCharactersInString:@"x*×"]];

	if ([resolution_components count] != 2)
		return NO;

	*width = [[resolution_components objectAtIndex:0] intValue];
	*height = [[resolution_components objectAtIndex:1] intValue];
	return YES;
}

NSString* ScreenResolutionDescription(TSXScreenOptions type, int width,
                                      int height)
{
	if (type == TSXScreenOptionFitScreen)
		return LocalizedFitScreen();
	else if (type == TSXScreenOptionCustom)
		return LocalizedCustom();

	return [NSString stringWithFormat:@"%dx%d", width, height];
}


NSDictionary* SelectionForColorSetting()
{
	OrderedDictionary* dict = [OrderedDictionary dictionaryWithCapacity:3];
	[dict setValue:[NSNumber numberWithInt:8] forKey:NSLocalizedString(
	     @"Palette Color (8 Bit)", @"8 bit color selection")];
	[dict setValue:[NSNumber numberWithInt:15] forKey:NSLocalizedString(
	     @"High Color (15 Bit)", @"15 bit color selection")];
	[dict setValue:[NSNumber numberWithInt:16] forKey:NSLocalizedString(
	     @"High Color (16 Bit)", @"16 bit color selection")];
	[dict setValue:[NSNumber numberWithInt:24] forKey:NSLocalizedString(
	     @"True Color (24 Bit)", @"24 bit color selection")];
	[dict setValue:[NSNumber numberWithInt:32] forKey:NSLocalizedString(
	     @"Highest Quality (32 Bit)", @"32 bit color selection")];
	return dict;
}

NSArray* ResolutionModes()
{
	NSArray* array = [NSArray arrayWithObjects:ScreenResolutionDescription(
	                      TSXScreenOptionFitScreen, 0, 0),
	                  ScreenResolutionDescription(TSXScreenOptionFixed, 640, 480),
	                  ScreenResolutionDescription(TSXScreenOptionFixed, 800, 600),
	                  ScreenResolutionDescription(TSXScreenOptionFixed, 1024, 768),
	                  ScreenResolutionDescription(TSXScreenOptionFixed, 1280, 1024),
	                  ScreenResolutionDescription(TSXScreenOptionFixed, 1440, 900),
	                  ScreenResolutionDescription(TSXScreenOptionFixed, 1440, 1050),
	                  ScreenResolutionDescription(TSXScreenOptionFixed, 1600, 1200),
	                  ScreenResolutionDescription(TSXScreenOptionFixed, 1920, 1080),
	                  ScreenResolutionDescription(TSXScreenOptionFixed, 1920, 1200),
	                  ScreenResolutionDescription(TSXScreenOptionCustom, 0, 0), nil];
	return array;
}

#pragma mark Working with Security Protocols

NSString* LocalizedAutomaticSecurity()
{
	return NSLocalizedString(@"Automatic", @"Automatic protocl security selection");
}

NSString* ProtocolSecurityDescription(TSXProtocolSecurityOptions type)
{
	if (type == TSXProtocolSecurityNLA)
		return @"NLA";
	else if (type == TSXProtocolSecurityTLS)
		return @"TLS";
	else if (type == TSXProtocolSecurityRDP)
		return @"RDP";

	return LocalizedAutomaticSecurity();
}

BOOL ScanProtocolSecurity(NSString* description,
                          TSXProtocolSecurityOptions* type)
{
	*type = TSXProtocolSecurityRDP;

	if ([description isEqualToString:@"NLA"])
	{
		*type = TSXProtocolSecurityNLA;
		return YES;
	}
	else if ([description isEqualToString:@"TLS"])
	{
		*type = TSXProtocolSecurityTLS;
		return YES;
	}
	else if ([description isEqualToString:@"RDP"])
	{
		*type = TSXProtocolSecurityRDP;
		return YES;
	}
	else if ([description isEqualToString:LocalizedAutomaticSecurity()])
	{
		*type = TSXProtocolSecurityAutomatic;
		return YES;
	}

	return NO;
}

NSDictionary* SelectionForSecuritySetting()
{
	OrderedDictionary* dict = [OrderedDictionary dictionaryWithCapacity:4];
	[dict setValue:[NSNumber numberWithInt:TSXProtocolSecurityAutomatic] forKey:
	 ProtocolSecurityDescription(TSXProtocolSecurityAutomatic)];
	[dict setValue:[NSNumber numberWithInt:TSXProtocolSecurityRDP] forKey:
	 ProtocolSecurityDescription(TSXProtocolSecurityRDP)];
	[dict setValue:[NSNumber numberWithInt:TSXProtocolSecurityTLS] forKey:
	 ProtocolSecurityDescription(TSXProtocolSecurityTLS)];
	[dict setValue:[NSNumber numberWithInt:TSXProtocolSecurityNLA] forKey:
	 ProtocolSecurityDescription(TSXProtocolSecurityNLA)];
	return dict;
}


#pragma mark -
#pragma mark Bookmarks

#import "Bookmark.h"

NSMutableArray* FilterBookmarks(NSArray* bookmarks, NSArray* filter_words)
{
	NSMutableArray* matching_items = [NSMutableArray array];
	NSArray* searched_keys = [NSArray arrayWithObjects:@"label", @"params.hostname",
	                          @"params.username", @"params.domain", nil];

	for (ComputerBookmark * cur_bookmark in bookmarks)
	{
		double match_score = 0.0;

		for (int i = 0; i < [searched_keys count]; i++)
		{
			NSString* val = [cur_bookmark valueForKeyPath:[searched_keys objectAtIndex:i]];

			if (![val isKindOfClass:[NSString class]] || ![val length])
				continue;

			for (NSString * word in filter_words)
				if ([val rangeOfString:word options:(NSCaseInsensitiveSearch |
				                                     NSWidthInsensitiveSearch)].location != NSNotFound)
					match_score += (1.0 / [filter_words count]) * pow(2, [searched_keys count] - i);
		}

		if (match_score > 0.001)
			[matching_items addObject:[NSDictionary dictionaryWithObjectsAndKeys:
			                           cur_bookmark, @"bookmark",
			                           [NSNumber numberWithFloat:match_score], @"score",
			                           nil]];
	}

	[matching_items sortUsingComparator:^NSComparisonResult(NSDictionary * obj1,
	        NSDictionary * obj2)
	{
		return [[obj2 objectForKey:@"score"] compare:[obj1 objectForKey:@"score"]];
	}];
	return matching_items;
}

NSMutableArray* FilterHistory(NSArray* history, NSString* filterStr)
{
	NSMutableArray* result = [NSMutableArray array];

	for (NSString * item in history)
	{
		if ([item rangeOfString:filterStr].location != NSNotFound)
			[result addObject:item];
	}

	return result;
}

#pragma mark Version Info
NSString* TSXAppFullVersion()
{
	return [NSString stringWithUTF8String:GIT_REVISION];
}

#pragma mark iPad/iPhone detection

BOOL IsPad()
{
#ifdef UI_USER_INTERFACE_IDIOM
	return (UI_USER_INTERFACE_IDIOM() == UIUserInterfaceIdiomPad);
#else
	return NO;
#endif
}

BOOL IsPhone()
{
#ifdef UI_USER_INTERFACE_IDIOM
	return (UI_USER_INTERFACE_IDIOM() == UIUserInterfaceIdiomPhone);
#else
	return NO;
#endif
}

// set mouse buttons swapped flag
static BOOL g_swap_mouse_buttons = NO;
void SetSwapMouseButtonsFlag(BOOL swapped)
{
	g_swap_mouse_buttons = swapped;
}

// set invert scrolling flag
static BOOL g_invert_scrolling = NO;
void SetInvertScrollingFlag(BOOL invert)
{
	g_invert_scrolling = invert;
}

// return event value for left mouse button
int GetLeftMouseButtonClickEvent(BOOL down)
{
	if (g_swap_mouse_buttons)
		return (PTR_FLAGS_BUTTON2 | (down ? PTR_FLAGS_DOWN : 0));
	else
		return (PTR_FLAGS_BUTTON1 | (down ? PTR_FLAGS_DOWN : 0));
}

// return event value for right mouse button
int GetRightMouseButtonClickEvent(BOOL down)
{
	if (g_swap_mouse_buttons)
		return (PTR_FLAGS_BUTTON1 | (down ? PTR_FLAGS_DOWN : 0));
	else
		return (PTR_FLAGS_BUTTON2 | (down ? PTR_FLAGS_DOWN : 0));
}

// get mouse move event
int GetMouseMoveEvent()
{
	return (PTR_FLAGS_MOVE);
}

// return mouse wheel event
int GetMouseWheelEvent(BOOL down)
{
	if (g_invert_scrolling)
		down = !down;

	if (down)
		return (PTR_FLAGS_WHEEL | PTR_FLAGS_WHEEL_NEGATIVE | (0x0088));
	else
		return (PTR_FLAGS_WHEEL | (0x0078));
}

// scrolling gesture detection delta
CGFloat GetScrollGestureDelta()
{
	return 10.0f;
}

// this hack activates the iphone's WWAN interface in case it is offline
void WakeUpWWAN()
{
	NSURL* url = [[[NSURL alloc] initWithString:
	               @"http://www.nonexistingdummyurl.com"] autorelease];
	//NSData * data =
	[NSData dataWithContentsOfURL:
	 url]; // we don't need data but assigning one causes a "data not used" compiler warning
}


#pragma mark System Info functions

NSString* TSXGetPrimaryMACAddress(NSString* sep)
{
	NSString* macaddress = @"";
	struct ifaddrs* addrs;

	if (getifaddrs(&addrs) < 0)
	{
		NSLog(@"getPrimaryMACAddress: getifaddrs failed.");
		return macaddress;
	}

	for (struct ifaddrs* cursor = addrs; cursor != NULL; cursor = cursor->ifa_next)
	{
		if (strcmp(cursor->ifa_name, "en0"))
			continue;

		if ((cursor->ifa_addr->sa_family == AF_LINK)
		    && (((struct sockaddr_dl*) cursor->ifa_addr)->sdl_type == 0x6 /*IFT_ETHER*/))
		{
			struct sockaddr_dl* dlAddr = (struct sockaddr_dl*) cursor->ifa_addr;

			if (dlAddr->sdl_alen != 6)
				continue;

			unsigned char* base = (unsigned char*) &dlAddr->sdl_data[dlAddr->sdl_nlen];
			macaddress = [NSString hexStringFromData:base ofSize:6 withSeparator:sep
			              afterNthChar:1];
			break;
		}
	}

	freeifaddrs(addrs);
	return macaddress;
}

BOOL TSXDeviceHasJailBreak()
{
	if ([[NSFileManager defaultManager] fileExistsAtPath:
	     @"/Applications/Cydia.app/"])
		return YES;

	if ([[NSFileManager defaultManager] fileExistsAtPath:@"/etc/apt/"])
		return YES;

	return NO;
}

NSString* TSXGetPlatform()
{
	size_t size;
	sysctlbyname("hw.machine", NULL, &size, NULL, 0);
	char* machine = malloc(size);
	sysctlbyname("hw.machine", machine, &size, NULL, 0);
	NSString* platform = [NSString stringWithCString:machine encoding:
	                      NSASCIIStringEncoding];
	free(machine);
	return platform;
}

