Merge pull request 'Update modules/sqlite_local.py' (!1) from swee-patch-1 into main

Reviewed-on: #1

modules/sqlite_local.py

@@ -1,24 +1,29 @@
 # IRCat module for local SQLite database (default)
-import sqlite3, os
+import sqlite3, os, traceback
+from cryptography import Fernet
 __ircat_type__ = "sql.provider" # The type of module
-__ircat_requires__ = ["data-path"] # The required config.yml entries.
+__ircat_requires__ = ["data-path", "fernet-key"] # The required config.yml entries.
 class broker:
-    def __init__(self, data_path):
+    def __init__(self, data_path, fernet_key):
         if not os.path.isfile(data_path):
             print("Creating database file...")
             open(data_path, "w").write("")
         self.conn = sqlite3.connect(data_path)
+        self.fnet = Fernet(fernet_key)
         db = self.conn.cursor()
         db.execute("""CREATE table IF NOT EXISTS bans (ip varchar(255), reason varchar(255)""")
         db.execute("""CREATE table IF NOT EXISTS nickserv (user varchar(255), modes varchar(255), hash varchar(255), cloak varchar(255))""")
         db.execute("""CREATE table IF NOT EXISTS groups (name varchar(255), owner varchar(255))""")
         db.execute("""CREATE table IF NOT EXISTS chanserv (name varchar(255), modes varchar(255), params varchar(255), owner varchar(255), usermodes varchar(255), optimodes varchar(255))""")
-    def nickserv_gethash(self, nick, password:str):
+    def nickserv_identify(self, nick, password:str):
         db = self.conn.cursor()
-        password = password.encode("UTF-8")
         db.execute("SELECT * FROM nickserv WHERE user=?;", [nick])
         e = db.fetchall()
         if e == []:
-            return ["Nickname doesn't exist."]
+            return False
         else:
-            return e
+            try:
+                return e[0] if self.fnet.decrypt(e[0][2]) == password else False
+            except:
+                print(traceback.format_exc())
+                return False