From 20b42fee2c7353da068c5904191d491e8b5314fb Mon Sep 17 00:00:00 2001
From: Dryusdan <contact@dryusdan.fr>
Date: Fri, 11 Dec 2020 13:11:28 +0100
Subject: [PATCH] Add discord notification

---
 .gitignore                 |  2 +-
 config/kimsufi.sample.yaml |  3 +++
 src/notifications.py       | 12 ++++++++----
 3 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/.gitignore b/.gitignore
index efc2cde..abf1084 100644
--- a/.gitignore
+++ b/.gitignore
@@ -130,5 +130,5 @@ dmypy.json
 .pyre/
 
 # Ignore config
-conf/kimsufi.yaml
+config/kimsufi.yaml
 db.sqlite
diff --git a/config/kimsufi.sample.yaml b/config/kimsufi.sample.yaml
index c4068d7..74bdaf0 100644
--- a/config/kimsufi.sample.yaml
+++ b/config/kimsufi.sample.yaml
@@ -34,4 +34,7 @@ notifications:
     host: domain.tld
     thumbnail: True
     room_id: !12546789:domain.tld
+  discord:
+    webhookid: 123
+    webhooktoken: 567
 db: db.sqlite
diff --git a/src/notifications.py b/src/notifications.py
index b7b00bc..2e00c79 100644
--- a/src/notifications.py
+++ b/src/notifications.py
@@ -35,8 +35,8 @@ class Notifications:
     def run(self, server, datacenter):
         refOfServer = Settings().get("config/refToName.yml")
         self.apprise.notify(
-            body="{} is available {}".format(refOfServer[server], datacenter),
-            title="{} is available \o/".format(refOfServer[server], datacenter),
+            body="",
+            title="Le serveur {} disponible au datacenter de {}".format(refOfServer[server], datacenter),
             notify_type=apprise.NotifyType.INFO,
         )
 
@@ -77,7 +77,12 @@ class Notifications:
         )
 
     def _addDiscord(self):
-        return True
+        self.apprise.add(
+            "discord://{}/{}/".format(
+                self.config["discord"]["webhookid"],
+                self.config["discord"]["webhooktoken"],
+            )
+        )
 
     def _addMatrix(self):
         self.apprise.add(
@@ -88,4 +93,3 @@ class Notifications:
                 self.config["matrix"]["room_id"],
             )
         )
-        return True