diff options
Diffstat (limited to 'comm/mail/components/shell/nsGNOMEShellService.cpp')
| -rw-r--r-- | comm/mail/components/shell/nsGNOMEShellService.cpp | 341 |
1 files changed, 341 insertions, 0 deletions
diff --git a/comm/mail/components/shell/nsGNOMEShellService.cpp b/comm/mail/components/shell/nsGNOMEShellService.cpp new file mode 100644 index 0000000000..f7381b2adc --- /dev/null +++ b/comm/mail/components/shell/nsGNOMEShellService.cpp @@ -0,0 +1,341 @@ +/* -*- 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 + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ + +#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; + } + + // snap_name as defined on https://snapcraft.io/thunderbird + 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; +} |