From cd86c196f3500f20539754d347c8378349ca0974 Mon Sep 17 00:00:00 2001
From: vaxerski <43317083+vaxerski@users.noreply.github.com>
Date: Mon, 29 May 2023 18:35:41 +0200
Subject: [PATCH] core: disallow multiple instances of hyprpaper at once

---
 src/Hyprpaper.cpp | 46 ++++++++++++++++++++++++++++++++++++++++++++++
 src/Hyprpaper.hpp |  2 ++
 2 files changed, 48 insertions(+)

diff --git a/src/Hyprpaper.cpp b/src/Hyprpaper.cpp
index 373abcb..7cfd459 100644
--- a/src/Hyprpaper.cpp
+++ b/src/Hyprpaper.cpp
@@ -1,9 +1,18 @@
 #include "Hyprpaper.hpp"
+#include <filesystem>
+#include <fstream>
+#include <sys/types.h>
+#include <signal.h>
 
 CHyprpaper::CHyprpaper() = default;
 
 void CHyprpaper::init() {
 
+    if (!lockSingleInstance()) {
+        Debug::log(CRIT, "Cannot launch multiple instances of Hyprpaper at once!");
+        exit(1);
+    }
+
     removeOldHyprpaperImages();
 
     g_pConfigManager = std::make_unique<CConfigManager>();
@@ -29,6 +38,8 @@ void CHyprpaper::init() {
         std::lock_guard<std::mutex> lg(m_mtTickMutex);
         tick(true);
     }
+
+    unlockSingleInstance();
 }
 
 void CHyprpaper::tick(bool force) {
@@ -552,3 +563,38 @@ void CHyprpaper::renderWallpaperForMonitor(SMonitor* pMonitor) {
         }
     }
 }
+
+bool CHyprpaper::lockSingleInstance() {
+    const std::string XDG_RUNTIME_DIR = getenv("XDG_RUNTIME_DIR");
+
+    const auto LOCKFILE = XDG_RUNTIME_DIR + "/hyprpaper.lock";
+
+    if (std::filesystem::exists(LOCKFILE)) {
+        std::ifstream ifs(LOCKFILE);
+        std::string content((std::istreambuf_iterator<char>(ifs)), (std::istreambuf_iterator<char>()));
+
+        try {
+            kill(std::stoull(content), 0);
+
+            if (errno != ESRCH)
+                return false;
+        } catch (std::exception& e) {
+            ;
+        }
+    }
+
+    // create lockfile
+    std::ofstream ofs(LOCKFILE, std::ios::trunc);
+
+    ofs << std::to_string(getpid());
+
+    ofs.close();
+
+    return true;
+}
+
+void CHyprpaper::unlockSingleInstance() {
+    const std::string XDG_RUNTIME_DIR = getenv("XDG_RUNTIME_DIR");
+    const auto LOCKFILE = XDG_RUNTIME_DIR + "/hyprpaper.lock";
+    unlink(LOCKFILE.c_str());
+}
\ No newline at end of file
diff --git a/src/Hyprpaper.hpp b/src/Hyprpaper.hpp
index 9d6287c..16a173d 100644
--- a/src/Hyprpaper.hpp
+++ b/src/Hyprpaper.hpp
@@ -59,6 +59,8 @@ public:
     SPoolBuffer* getPoolBuffer(SMonitor*, CWallpaperTarget*);
     void        unloadWallpaper(const std::string&);
     void        createSeat(wl_seat*);
+    bool        lockSingleInstance(); // fails on multi-instance
+    void        unlockSingleInstance();
 
     std::mutex  m_mtTickMutex;