e14fd62f46
When returning false in the notification callback the ProcessSingleton will assume current process is quitting, we should met its expectation.
428 lines
13 KiB
C++
428 lines
13 KiB
C++
// 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/api/atom_api_app.h"
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#if defined(OS_WIN)
|
|
#include <shlobj.h>
|
|
#endif
|
|
|
|
#include "atom/browser/api/atom_api_menu.h"
|
|
#include "atom/browser/api/atom_api_session.h"
|
|
#include "atom/browser/atom_browser_context.h"
|
|
#include "atom/browser/atom_browser_main_parts.h"
|
|
#include "atom/browser/browser.h"
|
|
#include "atom/browser/api/atom_api_web_contents.h"
|
|
#include "atom/common/native_mate_converters/callback.h"
|
|
#include "atom/common/native_mate_converters/file_path_converter.h"
|
|
#include "atom/common/native_mate_converters/command_line_converter.h"
|
|
#include "atom/common/node_includes.h"
|
|
#include "atom/common/options_switches.h"
|
|
#include "base/command_line.h"
|
|
#include "base/environment.h"
|
|
#include "base/files/file_path.h"
|
|
#include "base/path_service.h"
|
|
#include "brightray/browser/brightray_paths.h"
|
|
#include "content/public/browser/client_certificate_delegate.h"
|
|
#include "content/public/browser/gpu_data_manager.h"
|
|
#include "content/public/common/content_switches.h"
|
|
#include "native_mate/dictionary.h"
|
|
#include "native_mate/object_template_builder.h"
|
|
#include "net/ssl/ssl_cert_request_info.h"
|
|
#include "ui/base/l10n/l10n_util.h"
|
|
|
|
#if defined(OS_WIN)
|
|
#include "base/strings/utf_string_conversions.h"
|
|
#endif
|
|
|
|
using atom::Browser;
|
|
|
|
namespace mate {
|
|
|
|
#if defined(OS_WIN)
|
|
template<>
|
|
struct Converter<Browser::UserTask> {
|
|
static bool FromV8(v8::Isolate* isolate, v8::Local<v8::Value> val,
|
|
Browser::UserTask* out) {
|
|
mate::Dictionary dict;
|
|
if (!ConvertFromV8(isolate, val, &dict))
|
|
return false;
|
|
if (!dict.Get("program", &(out->program)) ||
|
|
!dict.Get("title", &(out->title)))
|
|
return false;
|
|
if (dict.Get("iconPath", &(out->icon_path)) &&
|
|
!dict.Get("iconIndex", &(out->icon_index)))
|
|
return false;
|
|
dict.Get("arguments", &(out->arguments));
|
|
dict.Get("description", &(out->description));
|
|
return true;
|
|
}
|
|
};
|
|
#endif
|
|
|
|
template<>
|
|
struct Converter<scoped_refptr<net::X509Certificate>> {
|
|
static v8::Local<v8::Value> ToV8(
|
|
v8::Isolate* isolate,
|
|
const scoped_refptr<net::X509Certificate>& val) {
|
|
mate::Dictionary dict(isolate, v8::Object::New(isolate));
|
|
std::string encoded_data;
|
|
net::X509Certificate::GetPEMEncoded(
|
|
val->os_cert_handle(), &encoded_data);
|
|
dict.Set("data", encoded_data);
|
|
dict.Set("issuerName", val->issuer().GetDisplayName());
|
|
return dict.GetHandle();
|
|
}
|
|
};
|
|
|
|
} // namespace mate
|
|
|
|
|
|
namespace atom {
|
|
|
|
namespace api {
|
|
|
|
namespace {
|
|
|
|
// Return the path constant from string.
|
|
int GetPathConstant(const std::string& name) {
|
|
if (name == "appData")
|
|
return brightray::DIR_APP_DATA;
|
|
else if (name == "userData")
|
|
return brightray::DIR_USER_DATA;
|
|
else if (name == "cache")
|
|
return brightray::DIR_CACHE;
|
|
else if (name == "userCache")
|
|
return brightray::DIR_USER_CACHE;
|
|
else if (name == "home")
|
|
return base::DIR_HOME;
|
|
else if (name == "temp")
|
|
return base::DIR_TEMP;
|
|
else if (name == "userDesktop")
|
|
return base::DIR_USER_DESKTOP;
|
|
else if (name == "exe")
|
|
return base::FILE_EXE;
|
|
else if (name == "module")
|
|
return base::FILE_MODULE;
|
|
else
|
|
return -1;
|
|
}
|
|
|
|
// Run the NotificationCallback and returns whether browser is shuting down.
|
|
bool NotificationCallbackWrapper(
|
|
const ProcessSingleton::NotificationCallback& callback,
|
|
const base::CommandLine& command_line,
|
|
const base::FilePath& current_directory) {
|
|
callback.Run(command_line, current_directory);
|
|
return !Browser::Get()->is_shutting_down();
|
|
}
|
|
|
|
void OnClientCertificateSelected(
|
|
v8::Isolate* isolate,
|
|
std::shared_ptr<content::ClientCertificateDelegate> delegate,
|
|
mate::Arguments* args) {
|
|
v8::Locker locker(isolate);
|
|
v8::HandleScope handle_scope(isolate);
|
|
mate::Dictionary cert_data;
|
|
if (!(args->Length() == 1 && args->GetNext(&cert_data))) {
|
|
args->ThrowError();
|
|
return;
|
|
}
|
|
|
|
std::string encoded_data;
|
|
cert_data.Get("data", &encoded_data);
|
|
|
|
auto certs =
|
|
net::X509Certificate::CreateCertificateListFromBytes(
|
|
encoded_data.data(), encoded_data.size(),
|
|
net::X509Certificate::FORMAT_AUTO);
|
|
|
|
delegate->ContinueWithCertificate(certs[0].get());
|
|
}
|
|
|
|
} // namespace
|
|
|
|
App::App() {
|
|
Browser::Get()->AddObserver(this);
|
|
content::GpuDataManager::GetInstance()->AddObserver(this);
|
|
}
|
|
|
|
App::~App() {
|
|
Browser::Get()->RemoveObserver(this);
|
|
content::GpuDataManager::GetInstance()->RemoveObserver(this);
|
|
}
|
|
|
|
void App::OnBeforeQuit(bool* prevent_default) {
|
|
*prevent_default = Emit("before-quit");
|
|
}
|
|
|
|
void App::OnWillQuit(bool* prevent_default) {
|
|
*prevent_default = Emit("will-quit");
|
|
}
|
|
|
|
void App::OnWindowAllClosed() {
|
|
Emit("window-all-closed");
|
|
}
|
|
|
|
void App::OnQuit() {
|
|
Emit("quit");
|
|
|
|
if (process_singleton_.get()) {
|
|
process_singleton_->Cleanup();
|
|
process_singleton_.reset();
|
|
}
|
|
}
|
|
|
|
void App::OnOpenFile(bool* prevent_default, const std::string& file_path) {
|
|
*prevent_default = Emit("open-file", file_path);
|
|
}
|
|
|
|
void App::OnOpenURL(const std::string& url) {
|
|
Emit("open-url", url);
|
|
}
|
|
|
|
void App::OnActivate(bool has_visible_windows) {
|
|
Emit("activate", has_visible_windows);
|
|
}
|
|
|
|
void App::OnWillFinishLaunching() {
|
|
Emit("will-finish-launching");
|
|
}
|
|
|
|
void App::OnFinishLaunching() {
|
|
// Create the defaultSession.
|
|
v8::Locker locker(isolate());
|
|
v8::HandleScope handle_scope(isolate());
|
|
auto browser_context = static_cast<AtomBrowserContext*>(
|
|
AtomBrowserMainParts::Get()->browser_context());
|
|
auto handle = Session::CreateFrom(isolate(), browser_context);
|
|
default_session_.Reset(isolate(), handle.ToV8());
|
|
|
|
if (process_singleton_startup_lock_.get())
|
|
process_singleton_startup_lock_->Unlock();
|
|
|
|
Emit("ready");
|
|
}
|
|
|
|
void App::OnSelectCertificate(
|
|
content::WebContents* web_contents,
|
|
net::SSLCertRequestInfo* cert_request_info,
|
|
scoped_ptr<content::ClientCertificateDelegate> delegate) {
|
|
std::shared_ptr<content::ClientCertificateDelegate>
|
|
shared_delegate(delegate.release());
|
|
bool prevent_default =
|
|
Emit("select-certificate",
|
|
api::WebContents::CreateFrom(isolate(), web_contents),
|
|
cert_request_info->host_and_port.ToString(),
|
|
cert_request_info->client_certs,
|
|
base::Bind(&OnClientCertificateSelected,
|
|
isolate(),
|
|
shared_delegate));
|
|
|
|
// Default to first certificate from the platform store.
|
|
if (!prevent_default)
|
|
shared_delegate->ContinueWithCertificate(
|
|
cert_request_info->client_certs[0].get());
|
|
}
|
|
|
|
void App::OnGpuProcessCrashed(base::TerminationStatus exit_code) {
|
|
Emit("gpu-process-crashed");
|
|
}
|
|
|
|
base::FilePath App::GetPath(mate::Arguments* args, const std::string& name) {
|
|
bool succeed = false;
|
|
base::FilePath path;
|
|
int key = GetPathConstant(name);
|
|
if (key >= 0)
|
|
succeed = PathService::Get(key, &path);
|
|
if (!succeed)
|
|
args->ThrowError("Failed to get path");
|
|
return path;
|
|
}
|
|
|
|
void App::SetPath(mate::Arguments* args,
|
|
const std::string& name,
|
|
const base::FilePath& path) {
|
|
bool succeed = false;
|
|
int key = GetPathConstant(name);
|
|
if (key >= 0)
|
|
succeed = PathService::Override(key, path);
|
|
if (!succeed)
|
|
args->ThrowError("Failed to set path");
|
|
}
|
|
|
|
void App::SetDesktopName(const std::string& desktop_name) {
|
|
#if defined(OS_LINUX)
|
|
scoped_ptr<base::Environment> env(base::Environment::Create());
|
|
env->SetVar("CHROME_DESKTOP", desktop_name);
|
|
#endif
|
|
}
|
|
|
|
void App::SetAppUserModelId(const std::string& app_id) {
|
|
#if defined(OS_WIN)
|
|
base::string16 app_id_utf16 = base::UTF8ToUTF16(app_id);
|
|
SetCurrentProcessExplicitAppUserModelID(app_id_utf16.c_str());
|
|
#endif
|
|
}
|
|
|
|
void App::AllowNTLMCredentialsForAllDomains(bool should_allow) {
|
|
auto browser_context = static_cast<AtomBrowserContext*>(
|
|
AtomBrowserMainParts::Get()->browser_context());
|
|
browser_context->AllowNTLMCredentialsForAllDomains(should_allow);
|
|
}
|
|
|
|
std::string App::GetLocale() {
|
|
return l10n_util::GetApplicationLocale("");
|
|
}
|
|
|
|
v8::Local<v8::Value> App::DefaultSession(v8::Isolate* isolate) {
|
|
if (default_session_.IsEmpty())
|
|
return v8::Null(isolate);
|
|
else
|
|
return v8::Local<v8::Value>::New(isolate, default_session_);
|
|
}
|
|
|
|
bool App::MakeSingleInstance(
|
|
const ProcessSingleton::NotificationCallback& callback) {
|
|
#if defined(OS_MACOSX)
|
|
LOG(ERROR) << "MakeSingleInstance is not implemnted on OS X";
|
|
return false;
|
|
#endif
|
|
if (process_singleton_.get())
|
|
return false;
|
|
|
|
base::FilePath user_dir;
|
|
PathService::Get(brightray::DIR_USER_DATA, &user_dir);
|
|
|
|
process_singleton_startup_lock_.reset(
|
|
new ProcessSingletonStartupLock(
|
|
base::Bind(NotificationCallbackWrapper, callback)));
|
|
process_singleton_.reset(
|
|
new ProcessSingleton(
|
|
user_dir,
|
|
process_singleton_startup_lock_->AsNotificationCallback()));
|
|
|
|
if (Browser::Get()->is_ready())
|
|
process_singleton_startup_lock_->Unlock();
|
|
|
|
switch (process_singleton_->NotifyOtherProcessOrCreate()) {
|
|
case ProcessSingleton::NotifyResult::PROCESS_NONE:
|
|
return false;
|
|
case ProcessSingleton::NotifyResult::LOCK_ERROR:
|
|
case ProcessSingleton::NotifyResult::PROFILE_IN_USE:
|
|
case ProcessSingleton::NotifyResult::PROCESS_NOTIFIED:
|
|
process_singleton_.reset();
|
|
process_singleton_startup_lock_.reset();
|
|
return true;
|
|
}
|
|
}
|
|
|
|
mate::ObjectTemplateBuilder App::GetObjectTemplateBuilder(
|
|
v8::Isolate* isolate) {
|
|
auto browser = base::Unretained(Browser::Get());
|
|
return mate::ObjectTemplateBuilder(isolate)
|
|
.SetMethod("quit", base::Bind(&Browser::Quit, browser))
|
|
.SetMethod("focus", base::Bind(&Browser::Focus, browser))
|
|
.SetMethod("getVersion", base::Bind(&Browser::GetVersion, browser))
|
|
.SetMethod("setVersion", base::Bind(&Browser::SetVersion, browser))
|
|
.SetMethod("getName", base::Bind(&Browser::GetName, browser))
|
|
.SetMethod("setName", base::Bind(&Browser::SetName, browser))
|
|
.SetMethod("isReady", base::Bind(&Browser::is_ready, browser))
|
|
.SetMethod("addRecentDocument",
|
|
base::Bind(&Browser::AddRecentDocument, browser))
|
|
.SetMethod("clearRecentDocuments",
|
|
base::Bind(&Browser::ClearRecentDocuments, browser))
|
|
#if defined(OS_WIN)
|
|
.SetMethod("setUserTasks",
|
|
base::Bind(&Browser::SetUserTasks, browser))
|
|
#endif
|
|
.SetMethod("setPath", &App::SetPath)
|
|
.SetMethod("getPath", &App::GetPath)
|
|
.SetMethod("setDesktopName", &App::SetDesktopName)
|
|
.SetMethod("setAppUserModelId", &App::SetAppUserModelId)
|
|
.SetMethod("allowNTLMCredentialsForAllDomains",
|
|
&App::AllowNTLMCredentialsForAllDomains)
|
|
.SetMethod("getLocale", &App::GetLocale)
|
|
.SetMethod("makeSingleInstance", &App::MakeSingleInstance)
|
|
.SetProperty("defaultSession", &App::DefaultSession);
|
|
}
|
|
|
|
// static
|
|
mate::Handle<App> App::Create(v8::Isolate* isolate) {
|
|
return CreateHandle(isolate, new App);
|
|
}
|
|
|
|
} // namespace api
|
|
|
|
} // namespace atom
|
|
|
|
|
|
namespace {
|
|
|
|
void AppendSwitch(const std::string& switch_string, mate::Arguments* args) {
|
|
auto command_line = base::CommandLine::ForCurrentProcess();
|
|
|
|
if (switch_string == atom::switches::kPpapiFlashPath ||
|
|
switch_string == atom::switches::kClientCertificate ||
|
|
switch_string == switches::kLogNetLog) {
|
|
base::FilePath path;
|
|
args->GetNext(&path);
|
|
command_line->AppendSwitchPath(switch_string, path);
|
|
return;
|
|
}
|
|
|
|
std::string value;
|
|
if (args->GetNext(&value))
|
|
command_line->AppendSwitchASCII(switch_string, value);
|
|
else
|
|
command_line->AppendSwitch(switch_string);
|
|
}
|
|
|
|
#if defined(OS_MACOSX)
|
|
int DockBounce(const std::string& type) {
|
|
int request_id = -1;
|
|
if (type == "critical")
|
|
request_id = Browser::Get()->DockBounce(Browser::BOUNCE_CRITICAL);
|
|
else if (type == "informational")
|
|
request_id = Browser::Get()->DockBounce(Browser::BOUNCE_INFORMATIONAL);
|
|
return request_id;
|
|
}
|
|
|
|
void DockSetMenu(atom::api::Menu* menu) {
|
|
Browser::Get()->DockSetMenu(menu->model());
|
|
}
|
|
#endif
|
|
|
|
void Initialize(v8::Local<v8::Object> exports, v8::Local<v8::Value> unused,
|
|
v8::Local<v8::Context> context, void* priv) {
|
|
v8::Isolate* isolate = context->GetIsolate();
|
|
auto command_line = base::CommandLine::ForCurrentProcess();
|
|
|
|
mate::Dictionary dict(isolate, exports);
|
|
dict.Set("app", atom::api::App::Create(isolate));
|
|
dict.SetMethod("appendSwitch", &AppendSwitch);
|
|
dict.SetMethod("appendArgument",
|
|
base::Bind(&base::CommandLine::AppendArg,
|
|
base::Unretained(command_line)));
|
|
#if defined(OS_MACOSX)
|
|
auto browser = base::Unretained(Browser::Get());
|
|
dict.SetMethod("dockBounce", &DockBounce);
|
|
dict.SetMethod("dockCancelBounce",
|
|
base::Bind(&Browser::DockCancelBounce, browser));
|
|
dict.SetMethod("dockSetBadgeText",
|
|
base::Bind(&Browser::DockSetBadgeText, browser));
|
|
dict.SetMethod("dockGetBadgeText",
|
|
base::Bind(&Browser::DockGetBadgeText, browser));
|
|
dict.SetMethod("dockHide", base::Bind(&Browser::DockHide, browser));
|
|
dict.SetMethod("dockShow", base::Bind(&Browser::DockShow, browser));
|
|
dict.SetMethod("dockSetMenu", &DockSetMenu);
|
|
#endif
|
|
}
|
|
|
|
} // namespace
|
|
|
|
NODE_MODULE_CONTEXT_AWARE_BUILTIN(atom_browser_app, Initialize)
|