From 02eae0a10769b1dc2f57a3545b9d82aa960ac0b8 Mon Sep 17 00:00:00 2001
From: Vaxry <vaxry@vaxry.net>
Date: Wed, 6 Dec 2023 17:23:02 +0000
Subject: [PATCH] ensure state store always exists

---
 hyprpm/src/core/DataState.cpp | 24 ++++++++++++++++++++++++
 hyprpm/src/core/DataState.hpp |  1 +
 2 files changed, 25 insertions(+)

diff --git a/hyprpm/src/core/DataState.cpp b/hyprpm/src/core/DataState.cpp
index 7b29b862..562dc880 100644
--- a/hyprpm/src/core/DataState.cpp
+++ b/hyprpm/src/core/DataState.cpp
@@ -4,7 +4,21 @@
 #include <filesystem>
 #include <fstream>
 
+void DataState::ensureStateStoreExists() {
+    const auto HOME = getenv("HOME");
+    if (!HOME) {
+        std::cerr << "DataState: no $HOME\n";
+        return;
+    }
+    const auto PATH = std::string(HOME) + "/.hyprpm/";
+
+    if (!std::filesystem::exists(PATH))
+        std::filesystem::create_directories(PATH);
+}
+
 void DataState::addNewPluginRepo(const SPluginRepository& repo) {
+    ensureStateStoreExists();
+
     const auto HOME = getenv("HOME");
     if (!HOME) {
         std::cerr << "DataState: no $HOME\n";
@@ -38,6 +52,8 @@ void DataState::addNewPluginRepo(const SPluginRepository& repo) {
 }
 
 bool DataState::pluginRepoExists(const std::string& urlOrName) {
+    ensureStateStoreExists();
+
     const auto HOME = getenv("HOME");
     if (!HOME) {
         std::cerr << "DataState: no $HOME\n";
@@ -62,6 +78,8 @@ bool DataState::pluginRepoExists(const std::string& urlOrName) {
 }
 
 void DataState::removePluginRepo(const std::string& urlOrName) {
+    ensureStateStoreExists();
+
     const auto HOME = getenv("HOME");
     if (!HOME) {
         std::cerr << "DataState: no $HOME\n";
@@ -86,6 +104,8 @@ void DataState::removePluginRepo(const std::string& urlOrName) {
 }
 
 void DataState::updateGlobalState(const SGlobalState& state) {
+    ensureStateStoreExists();
+
     const auto HOME = getenv("HOME");
     if (!HOME) {
         std::cerr << "DataState: no $HOME\n";
@@ -108,6 +128,8 @@ void DataState::updateGlobalState(const SGlobalState& state) {
 }
 
 std::vector<SPluginRepository> DataState::getAllRepositories() {
+    ensureStateStoreExists();
+
     const auto HOME = getenv("HOME");
     if (!HOME) {
         std::cerr << "DataState: no $HOME\n";
@@ -149,6 +171,8 @@ std::vector<SPluginRepository> DataState::getAllRepositories() {
 }
 
 bool DataState::setPluginEnabled(const std::string& name, bool enabled) {
+    ensureStateStoreExists();
+
     const auto HOME = getenv("HOME");
     if (!HOME) {
         std::cerr << "DataState: no $HOME\n";
diff --git a/hyprpm/src/core/DataState.hpp b/hyprpm/src/core/DataState.hpp
index be8792bb..f789c530 100644
--- a/hyprpm/src/core/DataState.hpp
+++ b/hyprpm/src/core/DataState.hpp
@@ -8,6 +8,7 @@ struct SGlobalState {
 };
 
 namespace DataState {
+    void                           ensureStateStoreExists();
     void                           addNewPluginRepo(const SPluginRepository& repo);
     void                           removePluginRepo(const std::string& urlOrName);
     bool                           pluginRepoExists(const std::string& urlOrName);