feat: basic api-key based authentication

src/modules/database/functions.py

@@ -3,14 +3,60 @@
 from datetime import datetime
 import sqlalchemy.orm
 
+import json
+
 from .schema import Base
 from .schema import Apikeys
 
 class functions():
-    def retrieve_api_keys(session: sqlalchemy.orm.Session) -> dict:
+    def retrieve_api_keys(session: sqlalchemy.orm.Session) -> tuple[list[dict], list[dict]]:
+        """ AI GENERATED
+        Retrieve all API keys from the database and separate them into valid and expired keys.
+
+        Parameters
+        ----------
+        session : sqlalchemy.orm.Session
+            An active SQLAlchemy session connected to the database.
+
+        Returns
+        -------
+        tuple[list[dict], list[dict]]
+            A tuple containing two lists of dictionaries:
+            - The first list contains valid API keys (not expired at the time of the call).
+            - The second list contains expired API keys (already expired at the time of the call).
+
+            Each dictionary has the following keys:
+            - 'id': int, the primary key of the API key record
+            - 'key': str, the API key string
+            - 'creation': str, timestamp of when the key was created
+            - 'expiration': str or None, timestamp of when the key expires, or None if no expiration
+
+        Notes
+        -----
+        - Keys with no expiration date are considered valid.
+        - Expiration comparison is performed against the current system time.
+        """
+
         time_now = datetime.now()
+        stored_keys = session.query(Apikeys).all()
 
-        api_keys = session.query(Apikeys).all()
-        print(dict(api_keys))
+        expired_keys = []
+        valid_keys = []
 
-        return {}
+        for key in stored_keys:
+            expiration_date = None
+
+            if key.key_expiration:
+                expiration_date = datetime.strptime(key.key_expiration, "%Y-%m-%d %H:%M:%S")
+
+            key_dict = key.to_dict()
+
+            if expiration_date is None or expiration_date > time_now:
+                valid_keys.append(key_dict)
+            else:
+                expired_keys.append(key_dict)
+        
+        print("VALID:", json.dumps(valid_keys,indent=4))
+        print("INVALID", json.dumps(expired_keys, indent=4))
+
+        return valid_keys, expired_keys

src/modules/database/schema.py

@@ -24,11 +24,19 @@ class User(Base):
 class Apikeys(Base):
     __tablename__ = 'apikeys'
 
-    apikey_id = sqlalchemy.Column(sqlalchemy.Integer, primary_key=True, index=True)
-    apikey_data = sqlalchemy.Column(sqlalchemy.String)
+    key_id = sqlalchemy.Column(sqlalchemy.Integer, primary_key=True, index=True)
+    key_data = sqlalchemy.Column(sqlalchemy.String)
 
-    apikey_creation = sqlalchemy.Column(sqlalchemy.String)
-    apikey_expiration = sqlalchemy.Column(sqlalchemy.String)
+    key_creation = sqlalchemy.Column(sqlalchemy.String)
+    key_expiration = sqlalchemy.Column(sqlalchemy.String)
+
+    def to_dict(self):
+        return {
+            'id': self.key_id,
+            'key': self.key_data,
+            'creation': self.key_creation,
+            'expiration': self.key_expiration,
+        }
 
 class Wireguard(Base):
     __tablename__ = 'wireguard_interfaces'