armorpaint/base/sources/libs/nfd.m

// This software is provided 'as-is', without any express or implied
// warranty.  In no event will the authors be held liable for any damages
// arising from the use of this software.
// Permission is granted to anyone to use this software for any purpose,
// including commercial applications, and to alter it and redistribute it
// freely, subject to the following restrictions:
// 1. The origin of this software must not be misrepresented; you must not
//    claim that you wrote the original software. If you use this software
//    in a product, an acknowledgment in the product documentation would be
//    appreciated but is not required.
// 2. Altered source versions must be plainly marked as such, and must not be
//    misrepresented as being the original software.
// 3. This notice may not be removed or altered from any source distribution.
// https://github.com/mlabbe/nativefiledialog

#include <AppKit/AppKit.h>
#include "nfd.h"

static NSArray *BuildAllowedFileTypes( const char *filterList ) {
	// Commas and semicolons are the same thing on this platform

	NSMutableArray *buildFilterList = [[NSMutableArray alloc] init];

	char typebuf[NFD_MAX_STRLEN] = {0};

	size_t filterListLen = strlen(filterList);
	char *p_typebuf = typebuf;
	for ( size_t i = 0; i < filterListLen+1; ++i )
	{
		if ( filterList[i] == ',' || filterList[i] == ';' || filterList[i] == '\0' )
		{
			*p_typebuf = '\0';
			NSString *thisType = [NSString stringWithUTF8String: typebuf];
			[buildFilterList addObject:thisType];
			p_typebuf = typebuf;
			*p_typebuf = '\0';
		}
		else
		{
			*p_typebuf = filterList[i];
			++p_typebuf;

		}
	}

	NSArray *returnArray = [NSArray arrayWithArray:buildFilterList];

//    [buildFilterList release];
	return returnArray;
}

static void AddFilterListToDialog( NSSavePanel *dialog, const char *filterList ) {
	if ( !filterList || strlen(filterList) == 0 )
		return;

	NSArray *allowedFileTypes = BuildAllowedFileTypes( filterList );
	if ( [allowedFileTypes count] != 0 )
	{
		[dialog setAllowedFileTypes:allowedFileTypes];
	}
}

static void SetDefaultPath( NSSavePanel *dialog, const nfdchar_t *defaultPath ) {
	if ( !defaultPath || strlen(defaultPath) == 0 )
		return;

	NSString *defaultPathString = [NSString stringWithUTF8String: defaultPath];
	NSURL *url = [NSURL fileURLWithPath:defaultPathString isDirectory:YES];
	[dialog setDirectoryURL:url];
}

/* fixme: pathset should be pathSet */
static nfdresult_t AllocPathSet( NSArray *urls, nfdpathset_t *pathset ) {
	assert(pathset);
	assert([urls count]);

	pathset->count = (size_t)[urls count];
	pathset->indices = NFDi_Malloc( sizeof(size_t)*pathset->count );
	if ( !pathset->indices )
	{
		return NFD_ERROR;
	}

	// count the total space needed for buf
	size_t bufsize = 0;
	for ( NSURL *url in urls )
	{
		NSString *path = [url path];
		bufsize += [path lengthOfBytesUsingEncoding:NSUTF8StringEncoding] + 1;
	}

	pathset->buf = NFDi_Malloc( sizeof(nfdchar_t) * bufsize );
	if ( !pathset->buf )
	{
		return NFD_ERROR;
	}

	// fill buf
	nfdchar_t *p_buf = pathset->buf;
	size_t count = 0;
	for ( NSURL *url in urls )
	{
		NSString *path = [url path];
		const nfdchar_t *utf8Path = [path UTF8String];
		size_t byteLen = [path lengthOfBytesUsingEncoding:NSUTF8StringEncoding] + 1;
		memcpy( p_buf, utf8Path, byteLen );

		ptrdiff_t index = p_buf - pathset->buf;
		assert( index >= 0 );
		pathset->indices[count] = (size_t)index;

		p_buf += byteLen;
		++count;
	}

	return NFD_OKAY;
}

nfdresult_t NFD_OpenDialog( const nfdchar_t *filterList,
							const nfdchar_t *defaultPath,
							nfdchar_t **outPath ) {
//    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];

	NSWindow *keyWindow = [[NSApplication sharedApplication] keyWindow];
	NSOpenPanel *dialog = [NSOpenPanel openPanel];
	[dialog setAllowsMultipleSelection:NO];

	// Build the filter list
	AddFilterListToDialog(dialog, filterList);

	// Set the starting directory
	SetDefaultPath(dialog, defaultPath);

	nfdresult_t nfdResult = NFD_CANCEL;
	if ( [dialog runModal] == NSModalResponseOK )
	{
		NSURL *url = [dialog URL];
		const char *utf8Path = [[url path] UTF8String];

		// byte count, not char count
		size_t len = strlen(utf8Path);//NFDi_UTF8_Strlen(utf8Path);

		*outPath = NFDi_Malloc( len+1 );
		if ( !*outPath )
		{
//            [pool release];
			[keyWindow makeKeyAndOrderFront:nil];
			return NFD_ERROR;
		}
		memcpy( *outPath, utf8Path, len+1 ); /* copy null term */
		nfdResult = NFD_OKAY;
	}
//    [pool release];

	[keyWindow makeKeyAndOrderFront:nil];
	return nfdResult;
}

nfdresult_t NFD_OpenDialogMultiple( const nfdchar_t *filterList,
									const nfdchar_t *defaultPath,
									nfdpathset_t *outPaths ) {
//    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
	NSWindow *keyWindow = [[NSApplication sharedApplication] keyWindow];

	NSOpenPanel *dialog = [NSOpenPanel openPanel];
	[dialog setAllowsMultipleSelection:YES];

	// Build the fiter list.
	AddFilterListToDialog(dialog, filterList);

	// Set the starting directory
	SetDefaultPath(dialog, defaultPath);

	nfdresult_t nfdResult = NFD_CANCEL;
	if ( [dialog runModal] == NSModalResponseOK )
	{
		NSArray *urls = [dialog URLs];

		if ( [urls count] == 0 )
		{
//            [pool release];
			[keyWindow makeKeyAndOrderFront:nil];
			return NFD_CANCEL;
		}

		if ( AllocPathSet( urls, outPaths ) == NFD_ERROR )
		{
//            [pool release];
			[keyWindow makeKeyAndOrderFront:nil];
			return NFD_ERROR;
		}

		nfdResult = NFD_OKAY;
	}
//    [pool release];

	[keyWindow makeKeyAndOrderFront:nil];
	return nfdResult;
}

nfdresult_t NFD_SaveDialog( const nfdchar_t *filterList,
							const nfdchar_t *defaultPath,
							nfdchar_t **outPath ) {
//    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
	NSWindow *keyWindow = [[NSApplication sharedApplication] keyWindow];

	NSSavePanel *dialog = [NSSavePanel savePanel];
	[dialog setExtensionHidden:NO];

	// Build the filter list.
	AddFilterListToDialog(dialog, filterList);

	// Set the starting directory
	SetDefaultPath(dialog, defaultPath);

	nfdresult_t nfdResult = NFD_CANCEL;
	if ( [dialog runModal] == NSModalResponseOK )
	{
		NSURL *url = [dialog URL];
		const char *utf8Path = [[url path] UTF8String];

		size_t byteLen = [url.path lengthOfBytesUsingEncoding:NSUTF8StringEncoding] + 1;

		*outPath = NFDi_Malloc( byteLen );
		if ( !*outPath )
		{
//            [pool release];
			[keyWindow makeKeyAndOrderFront:nil];
			return NFD_ERROR;
		}
		memcpy( *outPath, utf8Path, byteLen );
		nfdResult = NFD_OKAY;
	}

//    [pool release];
	[keyWindow makeKeyAndOrderFront:nil];
	return nfdResult;
}

nfdresult_t NFD_PickFolder(const nfdchar_t *defaultPath,
	nfdchar_t **outPath) {
//    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];

	NSWindow *keyWindow = [[NSApplication sharedApplication] keyWindow];
	NSOpenPanel *dialog = [NSOpenPanel openPanel];
	[dialog setAllowsMultipleSelection:NO];
	[dialog setCanChooseDirectories:YES];
	[dialog setCanCreateDirectories:YES];
	[dialog setCanChooseFiles:NO];

	// Set the starting directory
	SetDefaultPath(dialog, defaultPath);

	nfdresult_t nfdResult = NFD_CANCEL;
	if ( [dialog runModal] == NSModalResponseOK )
	{
		NSURL *url = [dialog URL];
		const char *utf8Path = [[url path] UTF8String];

		// byte count, not char count
		size_t len = strlen(utf8Path);//NFDi_UTF8_Strlen(utf8Path);

		*outPath = NFDi_Malloc( len+1 );
		if ( !*outPath )
		{
//            [pool release];
			[keyWindow makeKeyAndOrderFront:nil];
			return NFD_ERROR;
		}
		memcpy( *outPath, utf8Path, len+1 ); /* copy null term */
		nfdResult = NFD_OKAY;
	}
//    [pool release];

	[keyWindow makeKeyAndOrderFront:nil];
	return nfdResult;
}