/*
 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 <freerdp/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 protocol 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:FREERDP_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;
}

#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 != nullptr; 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", nullptr, &size, nullptr, 0);
	char *machine = malloc(size);
	sysctlbyname("hw.machine", machine, &size, nullptr, 0);
	NSString *platform = [NSString stringWithCString:machine encoding:NSASCIIStringEncoding];
	free(machine);
	return platform;
}
