237 lines
8.2 KiB
Text
237 lines
8.2 KiB
Text
// Copyright (c) 2013 GitHub, Inc.
|
|
// Use of this source code is governed by the MIT license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include "atom/browser/ui/file_dialog.h"
|
|
|
|
#import <Cocoa/Cocoa.h>
|
|
#import <CoreServices/CoreServices.h>
|
|
|
|
#include "atom/browser/native_window.h"
|
|
#include "base/files/file_util.h"
|
|
#include "base/mac/foundation_util.h"
|
|
#include "base/mac/mac_util.h"
|
|
#include "base/mac/scoped_cftyperef.h"
|
|
#include "base/strings/sys_string_conversions.h"
|
|
|
|
namespace file_dialog {
|
|
|
|
namespace {
|
|
|
|
void SetAllowedFileTypes(NSSavePanel* dialog, const Filters& filters) {
|
|
NSMutableSet* file_type_set = [NSMutableSet set];
|
|
for (size_t i = 0; i < filters.size(); ++i) {
|
|
const Filter& filter = filters[i];
|
|
for (size_t j = 0; j < filter.second.size(); ++j) {
|
|
// If we meet a '*' file extension, we allow all the file types and no
|
|
// need to set the specified file types.
|
|
if (filter.second[j] == "*") {
|
|
[dialog setAllowsOtherFileTypes:YES];
|
|
return;
|
|
}
|
|
base::ScopedCFTypeRef<CFStringRef> ext_cf(
|
|
base::SysUTF8ToCFStringRef(filter.second[j]));
|
|
[file_type_set addObject:base::mac::CFToNSCast(ext_cf.get())];
|
|
}
|
|
}
|
|
|
|
// Passing empty array to setAllowedFileTypes will cause exception.
|
|
NSArray* file_types = nil;
|
|
if ([file_type_set count])
|
|
file_types = [file_type_set allObjects];
|
|
|
|
[dialog setAllowedFileTypes:file_types];
|
|
}
|
|
|
|
void SetupDialog(NSSavePanel* dialog,
|
|
const std::string& title,
|
|
const std::string& button_label,
|
|
const base::FilePath& default_path,
|
|
const Filters& filters,
|
|
const std::string& message,
|
|
const std::string& name_field_label,
|
|
const bool& shows_tag_field) {
|
|
if (!title.empty())
|
|
[dialog setTitle:base::SysUTF8ToNSString(title)];
|
|
|
|
if (!button_label.empty())
|
|
[dialog setPrompt:base::SysUTF8ToNSString(button_label)];
|
|
|
|
if (!message.empty())
|
|
[dialog setMessage:base::SysUTF8ToNSString(message)];
|
|
|
|
if (!name_field_label.empty())
|
|
[dialog setNameFieldLabel:base::SysUTF8ToNSString(name_field_label)];
|
|
|
|
[dialog setShowsTagField:shows_tag_field];
|
|
|
|
NSString* default_dir = nil;
|
|
NSString* default_filename = nil;
|
|
if (!default_path.empty()) {
|
|
if (base::DirectoryExists(default_path)) {
|
|
default_dir = base::SysUTF8ToNSString(default_path.value());
|
|
} else {
|
|
default_dir = base::SysUTF8ToNSString(default_path.DirName().value());
|
|
default_filename =
|
|
base::SysUTF8ToNSString(default_path.BaseName().value());
|
|
}
|
|
}
|
|
|
|
if (default_dir)
|
|
[dialog setDirectoryURL:[NSURL fileURLWithPath:default_dir]];
|
|
if (default_filename)
|
|
[dialog setNameFieldStringValue:default_filename];
|
|
|
|
if (filters.empty())
|
|
[dialog setAllowsOtherFileTypes:YES];
|
|
else
|
|
SetAllowedFileTypes(dialog, filters);
|
|
}
|
|
|
|
void SetupDialogForProperties(NSOpenPanel* dialog, int properties) {
|
|
[dialog setCanChooseFiles:(properties & FILE_DIALOG_OPEN_FILE)];
|
|
if (properties & FILE_DIALOG_OPEN_DIRECTORY)
|
|
[dialog setCanChooseDirectories:YES];
|
|
if (properties & FILE_DIALOG_CREATE_DIRECTORY)
|
|
[dialog setCanCreateDirectories:YES];
|
|
if (properties & FILE_DIALOG_MULTI_SELECTIONS)
|
|
[dialog setAllowsMultipleSelection:YES];
|
|
if (properties & FILE_DIALOG_SHOW_HIDDEN_FILES)
|
|
[dialog setShowsHiddenFiles:YES];
|
|
}
|
|
|
|
// Run modal dialog with parent window and return user's choice.
|
|
int RunModalDialog(NSSavePanel* dialog, atom::NativeWindow* parent_window) {
|
|
__block int chosen = NSFileHandlingPanelCancelButton;
|
|
if (!parent_window || !parent_window->GetNativeWindow()) {
|
|
chosen = [dialog runModal];
|
|
} else {
|
|
NSWindow* window = parent_window->GetNativeWindow();
|
|
|
|
[dialog beginSheetModalForWindow:window
|
|
completionHandler:^(NSInteger c) {
|
|
chosen = c;
|
|
[NSApp stopModal];
|
|
}];
|
|
[NSApp runModalForWindow:window];
|
|
}
|
|
|
|
return chosen;
|
|
}
|
|
|
|
void ReadDialogPaths(NSOpenPanel* dialog, std::vector<base::FilePath>* paths) {
|
|
NSArray* urls = [dialog URLs];
|
|
for (NSURL* url in urls)
|
|
if ([url isFileURL])
|
|
paths->push_back(base::FilePath(base::SysNSStringToUTF8([url path])));
|
|
}
|
|
|
|
} // namespace
|
|
|
|
bool ShowOpenDialog(atom::NativeWindow* parent_window,
|
|
const std::string& title,
|
|
const std::string& button_label,
|
|
const base::FilePath& default_path,
|
|
const Filters& filters,
|
|
int properties,
|
|
const std::string& message,
|
|
std::vector<base::FilePath>* paths) {
|
|
DCHECK(paths);
|
|
NSOpenPanel* dialog = [NSOpenPanel openPanel];
|
|
|
|
// NSOpenPanel does not support name_field_label and shows_tag_field
|
|
SetupDialog(dialog, title, button_label, default_path, filters, message, "", false);
|
|
SetupDialogForProperties(dialog, properties);
|
|
|
|
int chosen = RunModalDialog(dialog, parent_window);
|
|
if (chosen == NSFileHandlingPanelCancelButton)
|
|
return false;
|
|
|
|
ReadDialogPaths(dialog, paths);
|
|
return true;
|
|
}
|
|
|
|
void ShowOpenDialog(atom::NativeWindow* parent_window,
|
|
const std::string& title,
|
|
const std::string& button_label,
|
|
const base::FilePath& default_path,
|
|
const Filters& filters,
|
|
int properties,
|
|
const std::string& message,
|
|
const OpenDialogCallback& c) {
|
|
NSOpenPanel* dialog = [NSOpenPanel openPanel];
|
|
|
|
// NSOpenPanel does not support name_field_label and shows_tag_field
|
|
SetupDialog(dialog, title, button_label, default_path, filters, message, "", false);
|
|
SetupDialogForProperties(dialog, properties);
|
|
|
|
// Duplicate the callback object here since c is a reference and gcd would
|
|
// only store the pointer, by duplication we can force gcd to store a copy.
|
|
__block OpenDialogCallback callback = c;
|
|
|
|
NSWindow* window = parent_window ? parent_window->GetNativeWindow() : NULL;
|
|
[dialog beginSheetModalForWindow:window
|
|
completionHandler:^(NSInteger chosen) {
|
|
if (chosen == NSFileHandlingPanelCancelButton) {
|
|
callback.Run(false, std::vector<base::FilePath>());
|
|
} else {
|
|
std::vector<base::FilePath> paths;
|
|
ReadDialogPaths(dialog, &paths);
|
|
callback.Run(true, paths);
|
|
}
|
|
}];
|
|
}
|
|
|
|
bool ShowSaveDialog(atom::NativeWindow* parent_window,
|
|
const std::string& title,
|
|
const std::string& button_label,
|
|
const base::FilePath& default_path,
|
|
const Filters& filters,
|
|
const std::string& message,
|
|
const std::string& name_field_label,
|
|
bool shows_tag_field,
|
|
base::FilePath* path) {
|
|
DCHECK(path);
|
|
NSSavePanel* dialog = [NSSavePanel savePanel];
|
|
|
|
SetupDialog(dialog, title, button_label, default_path, filters, message,
|
|
name_field_label, shows_tag_field);
|
|
|
|
int chosen = RunModalDialog(dialog, parent_window);
|
|
if (chosen == NSFileHandlingPanelCancelButton || ![[dialog URL] isFileURL])
|
|
return false;
|
|
|
|
*path = base::FilePath(base::SysNSStringToUTF8([[dialog URL] path]));
|
|
return true;
|
|
}
|
|
|
|
void ShowSaveDialog(atom::NativeWindow* parent_window,
|
|
const std::string& title,
|
|
const std::string& button_label,
|
|
const base::FilePath& default_path,
|
|
const Filters& filters,
|
|
const std::string& message,
|
|
const std::string& name_field_label,
|
|
bool shows_tag_field,
|
|
const SaveDialogCallback& c) {
|
|
NSSavePanel* dialog = [NSSavePanel savePanel];
|
|
|
|
SetupDialog(dialog, title, button_label, default_path, filters, message,
|
|
name_field_label, shows_tag_field);
|
|
[dialog setCanSelectHiddenExtension:YES];
|
|
|
|
__block SaveDialogCallback callback = c;
|
|
|
|
NSWindow* window = parent_window ? parent_window->GetNativeWindow() : NULL;
|
|
[dialog beginSheetModalForWindow:window
|
|
completionHandler:^(NSInteger chosen) {
|
|
if (chosen == NSFileHandlingPanelCancelButton) {
|
|
callback.Run(false, base::FilePath());
|
|
} else {
|
|
std::string path = base::SysNSStringToUTF8([[dialog URL] path]);
|
|
callback.Run(true, base::FilePath(path));
|
|
}
|
|
}];
|
|
}
|
|
|
|
} // namespace file_dialog
|