nfd2 0.3.1

/*
  Native File Dialog

  http://www.frogtoss.com/labs
 */

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

#include "ftg_core.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') {
            if (filterList[i] != '\0')
                ++p_typebuf;
            *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 || defaultPath[0] == '\0')
        return;

    // If the file in defaultPath doesn't exist, fall back to just using the directory.
    // Cocoa behaviour here ignores the entire defaultPath if the file at the end of
    // the path doesn't exist, falling back to an internally computed path.
    //
    // We override this behaviour to be consistent on all platforms:
    // just open the directory in question in the dialog and have no file selected.
    const char *defaultDir, *defaultFilename;
    NFDi_SplitPath(defaultPath, &defaultDir, &defaultFilename);

    bool onlyUseDirectory = true;
    if (defaultFilename) {
        if (ftg_path_exists(defaultPath)) {
            onlyUseDirectory = false;
        }
    }

    NSString* defaultPathString;
    if (defaultFilename && onlyUseDirectory) {
        assert(defaultFilename > defaultDir);
        defaultPathString = [NSString
            stringWithFormat:@"%.*s", (int)(defaultFilename - defaultDir), defaultDir];
    } else {
        defaultPathString = [NSString stringWithUTF8String:defaultPath];
    }

    NSURL* url = [NSURL fileURLWithPath:defaultPathString
                            isDirectory:ftg_is_dir(defaultPath)];
    [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;
}

void
SetAppPolicy(void)
{
    NSApplication* app = [NSApplication sharedApplication];

    NSApplicationActivationPolicy initial_policy = [app activationPolicy];
    if (initial_policy == NSApplicationActivationPolicyProhibited) {
        [app setActivationPolicy:NSApplicationActivationPolicyAccessory];
    }
}

/* public */


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

    SetAppPolicy();

    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];

    SetAppPolicy();

    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];

    SetAppPolicy();

    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];

    SetAppPolicy();

    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;
}