Source code
Revision control
Copy as Markdown
Other Tools
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* 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
#include "nsGNOMEShellService.h"
#include "nsIGIOService.h"
#include "nsCOMPtr.h"
#include "nsIServiceManager.h"
#include "prenv.h"
#include "nsIFile.h"
#include "nsIStringBundle.h"
#include "nsIPromptService.h"
#include "nsIPrefService.h"
#include "nsIPrefBranch.h"
#include "nsDirectoryServiceDefs.h"
#include "nsDirectoryServiceUtils.h"
#include "nsEmbedCID.h"
#include "mozilla/ArrayUtils.h"
#include "mozilla/Components.h"
#include <glib.h>
#include <limits.h>
#include <stdlib.h>
using mozilla::ArrayLength;
static const char* const sMailProtocols[] = {"mailto", "mid"};
static const char* const sNewsProtocols[] = {"news", "snews", "nntp"};
static const char* const sFeedProtocols[] = {"feed"};
static const char* const sCalendarProtocols[] = {"webcal", "webcals"};
struct AppTypeAssociation {
uint16_t type;
const char* const* protocols;
unsigned int protocolsLength;
const char* mimeType;
const char* extensions;
};
static bool IsRunningAsASnap() {
// SNAP holds the path to the snap, use SNAP_NAME
// which is easier to parse.
const char* snap_name = PR_GetEnv("SNAP_NAME");
// return early if not set.
if (snap_name == nullptr) {
return false;
}
return (strcmp(snap_name, "thunderbird") == 0);
}
static const AppTypeAssociation sAppTypes[] = {
{
nsIShellService::MAIL, sMailProtocols, ArrayLength(sMailProtocols),
"message/rfc822",
nullptr // don't associate .eml extension, as that breaks printing
// those
},
{nsIShellService::NEWS, sNewsProtocols, ArrayLength(sNewsProtocols),
nullptr, nullptr},
{nsIShellService::RSS, sFeedProtocols, ArrayLength(sFeedProtocols),
"application/rss+xml", "rss"},
{nsIShellService::CALENDAR, sCalendarProtocols,
ArrayLength(sCalendarProtocols), "text/calendar", "ics"}};
nsGNOMEShellService::nsGNOMEShellService()
: mUseLocaleFilenames(false),
mCheckedThisSession(false),
mAppIsInPath(false) {}
nsresult nsGNOMEShellService::Init() {
nsresult rv;
nsCOMPtr<nsIGIOService> giovfs = do_GetService(NS_GIOSERVICE_CONTRACTID);
if (!giovfs) return NS_ERROR_NOT_AVAILABLE;
// Check G_BROKEN_FILENAMES. If it's set, then filenames in glib use
// the locale encoding. If it's not set, they use UTF-8.
mUseLocaleFilenames = PR_GetEnv("G_BROKEN_FILENAMES") != nullptr;
if (GetAppPathFromLauncher()) return NS_OK;
nsCOMPtr<nsIFile> appPath;
rv = NS_GetSpecialDirectory(NS_XPCOM_CURRENT_PROCESS_DIR,
getter_AddRefs(appPath));
NS_ENSURE_SUCCESS(rv, rv);
rv = appPath->AppendNative(nsLiteralCString(MOZ_APP_NAME));
NS_ENSURE_SUCCESS(rv, rv);
rv = appPath->GetNativePath(mAppPath);
return rv;
}
NS_IMPL_ISUPPORTS(nsGNOMEShellService, nsIShellService, nsIToolkitShellService)
bool nsGNOMEShellService::GetAppPathFromLauncher() {
gchar* tmp;
const char* launcher = PR_GetEnv("MOZ_APP_LAUNCHER");
if (!launcher) return false;
if (g_path_is_absolute(launcher)) {
mAppPath = launcher;
tmp = g_path_get_basename(launcher);
gchar* fullpath = g_find_program_in_path(tmp);
if (fullpath && mAppPath.Equals(fullpath)) {
mAppIsInPath = true;
}
g_free(fullpath);
} else {
tmp = g_find_program_in_path(launcher);
if (!tmp) return false;
mAppPath = tmp;
mAppIsInPath = true;
}
g_free(tmp);
return true;
}
NS_IMETHODIMP
nsGNOMEShellService::IsDefaultClient(bool aStartupCheck, uint16_t aApps,
bool* aIsDefaultClient) {
*aIsDefaultClient = true;
for (unsigned int i = 0; i < MOZ_ARRAY_LENGTH(sAppTypes); i++) {
if (aApps & sAppTypes[i].type)
*aIsDefaultClient &=
checkDefault(sAppTypes[i].protocols, sAppTypes[i].protocolsLength);
}
// If this is the first mail window, maintain internal state that we've
// checked this session (so that subsequent window opens don't show the
// default client dialog).
if (aStartupCheck) mCheckedThisSession = true;
return NS_OK;
}
NS_IMETHODIMP
nsGNOMEShellService::SetDefaultClient(bool aForAllUsers, uint16_t aApps) {
nsresult rv = NS_OK;
for (unsigned int i = 0; i < MOZ_ARRAY_LENGTH(sAppTypes); i++) {
if (aApps & sAppTypes[i].type) {
nsresult tmp =
MakeDefault(sAppTypes[i].protocols, sAppTypes[i].protocolsLength,
sAppTypes[i].mimeType, sAppTypes[i].extensions);
if (NS_FAILED(tmp)) {
rv = tmp;
}
}
}
return rv;
}
NS_IMETHODIMP
nsGNOMEShellService::GetShouldCheckDefaultClient(bool* aResult) {
if (mCheckedThisSession) {
*aResult = false;
return NS_OK;
}
nsCOMPtr<nsIPrefBranch> prefs(do_GetService(NS_PREFSERVICE_CONTRACTID));
return prefs->GetBoolPref("mail.shell.checkDefaultClient", aResult);
}
NS_IMETHODIMP
nsGNOMEShellService::SetShouldCheckDefaultClient(bool aShouldCheck) {
nsCOMPtr<nsIPrefBranch> prefs(do_GetService(NS_PREFSERVICE_CONTRACTID));
return prefs->SetBoolPref("mail.shell.checkDefaultClient", aShouldCheck);
}
bool nsGNOMEShellService::KeyMatchesAppName(const char* aKeyValue) const {
gchar* commandPath;
if (mUseLocaleFilenames) {
gchar* nativePath = g_filename_from_utf8(aKeyValue, -1, NULL, NULL, NULL);
if (!nativePath) {
NS_ERROR("Error converting path to filesystem encoding");
return false;
}
commandPath = g_find_program_in_path(nativePath);
g_free(nativePath);
} else {
commandPath = g_find_program_in_path(aKeyValue);
}
if (!commandPath) return false;
bool matches = mAppPath.Equals(commandPath);
g_free(commandPath);
return matches;
}
bool nsGNOMEShellService::CheckHandlerMatchesAppName(
const nsACString& handler) const {
gint argc;
gchar** argv;
nsAutoCString command(handler);
if (g_shell_parse_argv(command.get(), &argc, &argv, NULL)) {
command.Assign(argv[0]);
g_strfreev(argv);
} else {
return false;
}
return KeyMatchesAppName(command.get());
}
bool nsGNOMEShellService::checkDefault(const char* const* aProtocols,
unsigned int aLength) {
nsCOMPtr<nsIGIOService> giovfs = do_GetService(NS_GIOSERVICE_CONTRACTID);
nsAutoCString handler;
nsresult rv;
for (unsigned int i = 0; i < aLength; ++i) {
if (IsRunningAsASnap()) {
const gchar* argv[] = {"xdg-settings", "get",
"default-url-scheme-handler", aProtocols[i],
nullptr};
GSpawnFlags flags = static_cast<GSpawnFlags>(G_SPAWN_SEARCH_PATH |
G_SPAWN_STDERR_TO_DEV_NULL);
gchar* output = nullptr;
gint exit_status = 0;
if (!g_spawn_sync(nullptr, (gchar**)argv, nullptr, flags, nullptr,
nullptr, &output, nullptr, &exit_status, nullptr)) {
return false;
}
if (exit_status != 0) {
g_free(output);
return false;
}
if (strcmp(output, "thunderbird.desktop\n") == 0) {
g_free(output);
return true;
}
g_free(output);
return false;
}
if (giovfs) {
handler.Truncate();
nsCOMPtr<nsIHandlerApp> handlerApp;
rv = giovfs->GetAppForURIScheme(nsDependentCString(aProtocols[i]),
getter_AddRefs(handlerApp));
if (NS_FAILED(rv) || !handlerApp) {
return false;
}
nsCOMPtr<nsIGIOMimeApp> app = do_QueryInterface(handlerApp, &rv);
if (NS_FAILED(rv) || !app) {
return false;
}
rv = app->GetCommand(handler);
if (NS_SUCCEEDED(rv) && !CheckHandlerMatchesAppName(handler)) {
return false;
}
}
}
return true;
}
nsresult nsGNOMEShellService::MakeDefault(const char* const* aProtocols,
unsigned int aProtocolsLength,
const char* aMimeType,
const char* aExtensions) {
nsAutoCString appKeyValue;
nsCOMPtr<nsIGIOService> giovfs = do_GetService(NS_GIOSERVICE_CONTRACTID);
if (mAppIsInPath) {
// mAppPath is in the users path, so use only the basename as the launcher
gchar* tmp = g_path_get_basename(mAppPath.get());
appKeyValue = tmp;
g_free(tmp);
} else {
appKeyValue = mAppPath;
}
appKeyValue.AppendLiteral(" %s");
if (IsRunningAsASnap()) {
for (unsigned int i = 0; i < aProtocolsLength; ++i) {
const gchar* argv[] = {"xdg-settings",
"set",
"default-url-scheme-handler",
aProtocols[i],
"thunderbird.desktop",
nullptr};
GSpawnFlags flags = static_cast<GSpawnFlags>(G_SPAWN_SEARCH_PATH |
G_SPAWN_STDOUT_TO_DEV_NULL |
G_SPAWN_STDERR_TO_DEV_NULL);
g_spawn_sync(nullptr, (gchar**)argv, nullptr, flags, nullptr, nullptr,
nullptr, nullptr, nullptr, nullptr);
}
}
nsresult rv;
if (giovfs) {
nsCOMPtr<nsIStringBundleService> bundleService =
mozilla::components::StringBundle::Service();
NS_ENSURE_TRUE(bundleService, NS_ERROR_UNEXPECTED);
nsCOMPtr<nsIStringBundle> brandBundle;
rv = bundleService->CreateBundle(BRAND_PROPERTIES,
getter_AddRefs(brandBundle));
NS_ENSURE_SUCCESS(rv, rv);
nsString brandShortName;
brandBundle->GetStringFromName("brandShortName", brandShortName);
// use brandShortName as the application id.
NS_ConvertUTF16toUTF8 id(brandShortName);
nsCOMPtr<nsIGIOMimeApp> app;
rv = giovfs->CreateAppFromCommand(mAppPath, id, getter_AddRefs(app));
NS_ENSURE_SUCCESS(rv, rv);
for (unsigned int i = 0; i < aProtocolsLength; ++i) {
rv = app->SetAsDefaultForURIScheme(nsDependentCString(aProtocols[i]));
NS_ENSURE_SUCCESS(rv, rv);
if (aMimeType)
rv = app->SetAsDefaultForMimeType(nsDependentCString(aMimeType));
NS_ENSURE_SUCCESS(rv, rv);
if (aExtensions)
rv =
app->SetAsDefaultForFileExtensions(nsDependentCString(aExtensions));
NS_ENSURE_SUCCESS(rv, rv);
}
}
return NS_OK;
}