qt: Add Settings -> Integrate... menu item

Author: hebasto

src/qt/bitcoingui.cpp

@@ -8,6 +8,7 @@
 #include <qt/clientmodel.h>
 #include <qt/createwalletdialog.h>
 #include <qt/guiconstants.h>
+#include <qt/guifileutil.h>
 #include <qt/guiutil.h>
 #include <qt/modaloverlay.h>
 #include <qt/networkstyle.h>
@@ -479,6 +480,21 @@ void BitcoinGUI::createMenuBar()
         settings->addSeparator();
     }
     settings->addAction(optionsAction);
+#ifdef Q_OS_LINUX
+    if (gArgs.GetChainName() != CBaseChainParams::REGTEST) {
+        settings->addSeparator();
+        QAction* integrate_with_DE = settings->addAction(tr("&Integrate with desktop environment"));
+        connect(integrate_with_DE, &QAction::triggered, [this] {
+            if (GUIUtil::IntegrateWithDesktopEnvironment(m_network_style->getTrayAndWindowIcon())) {
+                QMessageBox::information(this, tr("Application registered"),
+                                         tr("Now you are able to launch " PACKAGE_NAME " from the desktop menu."));
+            } else {
+                QMessageBox::warning(this, tr("Application registration failed"),
+                                     tr("" PACKAGE_NAME " failed integration with your desktop environment."));
+            }
+        });
+    }
+#endif // Q_OS_LINUX
 
     QMenu* window_menu = appMenuBar->addMenu(tr("&Window"));
 

src/qt/guifileutil.cpp

@@ -22,6 +22,7 @@
 #endif // WIN32
 
 #ifdef Q_OS_LINUX
+#include <cassert>
 #include <stdlib.h>
 #include <unistd.h>
 #endif // Q_OS_LINUX
@@ -30,6 +31,7 @@
 #include <QDir>
 #include <QFileDialog>
 #include <QFileInfo>
+#include <QPixmap>
 #include <QRegExp>
 #include <QStandardPaths>
 #include <QString>
@@ -49,11 +51,88 @@ std::string GetExecutablePathAsString()
     exe_path[r] = '\0';
     return exe_path;
 }
+
+fs::path GetUserApplicationsDir()
+{
+    const char* home_dir = getenv("HOME");
+    if (!home_dir) return fs::path();
+    return fs::path(home_dir) / ".local" / "share" / "applications";
+}
+
+fs::path GetDesktopFilePath(std::string chain)
+{
+    const auto app_dir = GetUserApplicationsDir();
+    if (app_dir.empty()) return fs::path();
+    if (!chain.empty()) {
+        chain = "_" + chain;
+    }
+    return app_dir / strprintf("org.bitcoincore.BitcoinQt%s.desktop", chain);
+}
+
+fs::path GetUserIconsDir()
+{
+    const char* home_dir = getenv("HOME");
+    if (!home_dir) return fs::path();
+    return fs::path(home_dir) / ".local" / "share" / "icons";
+}
+
+fs::path GetIconPath(std::string chain)
+{
+    const auto icons_dir = GetUserIconsDir();
+    if (icons_dir.empty()) return fs::path();
+    if (!chain.empty()) {
+        chain = "-" + chain;
+    }
+    return icons_dir / strprintf("bitcoin%s.png", chain);
+}
 #endif // Q_OS_LINUX
 } // namespace
 
 namespace GUIUtil {
 
+#ifdef Q_OS_LINUX
+bool IntegrateWithDesktopEnvironment(QIcon icon)
+{
+    std::string chain = gArgs.GetChainName();
+    assert(chain == CBaseChainParams::MAIN || chain == CBaseChainParams::TESTNET);
+    if (chain == CBaseChainParams::MAIN) {
+        chain.clear();
+    }
+
+    const auto icon_path = GetIconPath(chain);
+    if (icon_path.empty() || !icon.pixmap(256).save(boostPathToQString(icon_path))) return false;
+    const auto exe_path = GetExecutablePathAsString();
+    if (exe_path.empty()) return false;
+    const auto desktop_file_path = GetDesktopFilePath(chain);
+    if (desktop_file_path.empty()) return false;
+    fsbridge::ofstream desktop_file(desktop_file_path, std::ios_base::out | std::ios_base::trunc);
+    if (!desktop_file.good()) return false;
+
+    desktop_file << "[Desktop Entry]\n";
+    desktop_file << "Type=Application\n";
+    desktop_file << "Version=1.1\n";
+    desktop_file << "GenericName=Bitcoin client\n";
+    desktop_file << "Comment=Bitcoin full node and wallet\n";
+    desktop_file << strprintf("Icon=%s\n", icon_path.stem().string());
+    desktop_file << strprintf("TryExec=%s\n", exe_path);
+    desktop_file << "Categories=Network;Office;Finance;\n";
+    if (chain.empty()) {
+        desktop_file << "Name=" PACKAGE_NAME "\n";
+        desktop_file << strprintf("Exec=%s %%u\n", exe_path);
+        desktop_file << "Actions=Testnet;\n";
+        desktop_file << "[Desktop Action Testnet]\n";
+        desktop_file << strprintf("Exec=%s -testnet\n", exe_path);
+        desktop_file << "Name=Testnet mode\n";
+    } else {
+        desktop_file << "Name=" PACKAGE_NAME " - Testnet\n";
+        desktop_file << strprintf("Exec=%s -testnet %%u\n", exe_path);
+    }
+
+    desktop_file.close();
+    return true;
+}
+#endif // Q_OS_LINUX
+
 QString getDefaultDataDirectory()
 {
     return boostPathToQString(GetDefaultDataDir());

src/qt/guifileutil.h

@@ -7,7 +7,9 @@
 
 #include <fs.h>
 
+#include <QIcon>
 #include <QString>
+#include <QtGlobal>
 
 QT_BEGIN_NAMESPACE
 class QWidget;
@@ -16,6 +18,11 @@ QT_END_NAMESPACE
 /** Filesystem utility functions used by the GUI.
  */
 namespace GUIUtil {
+
+#ifdef Q_OS_LINUX
+bool IntegrateWithDesktopEnvironment(QIcon icon);
+#endif // Q_OS_LINUX
+
 /**
  * Determine default data directory for operating system.
  */