From ac26ca1b962e5fb76e3149750dc835bc6145c069 Mon Sep 17 00:00:00 2001
From: Frisk <git@frisk.space>
Date: Sun, 8 Sep 2024 00:08:02 +0200
Subject: [PATCH] Added db_connection hook as an example (MarkusRost's request)

---
 extensions/hooks/__init__.py      |  1 +
 extensions/hooks/db_connection.py | 80 +++++++++++++++++++++++++++++++
 2 files changed, 81 insertions(+)
 create mode 100644 extensions/hooks/db_connection.py

diff --git a/extensions/hooks/__init__.py b/extensions/hooks/__init__.py
index 727931b..9f65bf4 100644
--- a/extensions/hooks/__init__.py
+++ b/extensions/hooks/__init__.py
@@ -16,3 +16,4 @@
 #import extensions.hooks.example_hook
 #import extensions.hooks.usertalk
 #import extensions.hooks.edit_alerts
+#import extensions.hooks.db_connection
\ No newline at end of file
diff --git a/extensions/hooks/db_connection.py b/extensions/hooks/db_connection.py
new file mode 100644
index 0000000..e8dfc1c
--- /dev/null
+++ b/extensions/hooks/db_connection.py
@@ -0,0 +1,80 @@
+#  This file is part of Recent changes Goat compatible Discord webhook (RcGcDw).
+#
+#  RcGcDw is free software: you can redistribute it and/or modify
+#  it under the terms of the GNU General Public License as published by
+#  the Free Software Foundation, either version 3 of the License, or
+#  (at your option) any later version.
+#
+#  RcGcDw is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU General Public License for more details.
+#
+#  You should have received a copy of the GNU General Public License
+#  along with RcGcDw.  If not, see <http://www.gnu.org/licenses/>.
+import psycopg2
+
+# from src.api.context import Context
+# from src.discord.message import DiscordMessage, DiscordMessageMetadata
+from src.api.hook import post_hook
+from src.configloader import settings
+from contextlib import contextmanager
+
+# {
+#     "hooks": {
+#         "dbconn": {
+#                "database": "database",
+#                "host": "localhost",
+#                "port": 5432,
+#                "user": "rcgcdw",
+#                "password": "secret",
+#                "socket": null
+#             }
+#     }
+# }
+
+dbconn = settings.get("hooks", {}).get("dbconn", [])
+
+
+class DBConnector:
+    def __init__(self):
+        self.session = psycopg2.connect(dbname=dbconn.get("database", "rcgcdw"), user=dbconn.get("user", "rcgcdw"), password=dbconn.get("password", "rcgcdw"),
+                                        host=dbconn.get("host", "localhost"), port=dbconn.get("port", "5432"))
+
+    @contextmanager
+    def obtain_cursor(self) -> psycopg2._psycopg.cursor:
+        """Obtain cursor for database connection"""
+        cur = self.session.cursor()
+        try:
+            yield cur
+        finally:
+            self.session.commit()
+            cur.close()
+
+    def shutdown(self):  # placeholder, rcgcdw doesn't signal to its extensions when shutdown occurs so we cannot cleanup
+        self.session.close()
+
+
+db_singleton = DBConnector()
+
+@post_hook
+def db_interaction_hook(message, metadata, context, change):
+    # Check if user is in the database
+    if not dbconn:
+        return
+    with db_singleton.obtain_cursor() as cursor:
+        cursor.execute(
+            "INSERT INTO rcgcdw (username, edit_amount) VALUES (%s, 1) ON conflict (username) DO UPDATE SET edit_amount = rcgcdw.edit_amount + 1;",
+            (change["user"],))
+        cursor.execute("SELECT edit_amount FROM rcgcdw WHERE username = %s;", (change["user"],))
+        record = cursor.fetchone()
+        print(record)
+        if record is None or record[0] == 1:
+            if isinstance(message["description"], str):
+                message["description"] = message["description"] + "\n**This user is new! Let them know they are awesome! :)**"
+            else:
+                message["description"] = "\n**This user is new! Let them know they are awesome! :)**"
+                # if not record:
+        #     cursor.execute("INSERT INTO rcgcdw (user, edit_amount) VALUES (%s, %s) ON CONFLICT DO UPDATE SET edit_amount = edit_amount + 1", (change["user"], 1))
+        # else:
+        #     cursor.execute("UPDATE rcgcdw SET edit_amount = edit_amount + 1 WHERE user = %s;", (change["user"],))