From 80a46a58ac5f68a9bfb08ca0be8d165b57958afa Mon Sep 17 00:00:00 2001
From: dettlaff <dettlaff@riseup.net>
Date: Wed, 17 Jul 2024 19:44:56 +0400
Subject: [PATCH] feat: add prometheus migration

---
 selfprivacy_api/migrations/add_promiteus.py | 36 +++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 selfprivacy_api/migrations/add_promiteus.py

diff --git a/selfprivacy_api/migrations/add_promiteus.py b/selfprivacy_api/migrations/add_promiteus.py
new file mode 100644
index 0000000..52fa705
--- /dev/null
+++ b/selfprivacy_api/migrations/add_promiteus.py
@@ -0,0 +1,36 @@
+from selfprivacy_api.migrations.migration import Migration
+
+from selfprivacy_api.services.flake_service_manager import FlakeServiceManager
+from selfprivacy_api.utils import ReadUserData, WriteUserData
+
+
+class AddPrometheus(Migration):
+    """Adds the Prometheus if it is not present."""
+
+    def get_migration_name(self) -> str:
+        return "add_prometheus"
+
+    def get_migration_description(self) -> str:
+        return "Adds the Prometheus if it is not present."
+
+    def is_migration_needed(self) -> bool:
+        with FlakeServiceManager() as manager:
+            if "prometheus" not in manager.services:
+                return True
+        with ReadUserData() as data:
+            if "prometheus" not in data["modules"]:
+                return True
+        return False
+
+    def migrate(self) -> None:
+        with FlakeServiceManager() as manager:
+            if "prometheus" not in manager.services:
+                manager.services["prometheus"] = (
+                    "git+https://git.selfprivacy.org/SelfPrivacy/selfprivacy-nixos-config.git?ref=flakes&dir=sp-modules/prometheus"
+                )
+        with WriteUserData() as data:
+            if "prometheus" not in data["modules"]:
+                data["modules"]["prometheus"] = {
+                    "enable": False,
+                    "subdomain": "prometheus",
+                }