set up auth

app/db/sqlite/auth.py

@@ -0,0 +1,93 @@
+import json
+from app.models.user import User
+from app.utils.auth import encode_password
+from app.db.sqlite.utils import SQLiteManager
+
+
+class SQLiteAuthMethods:
+    """
+    Methods for authenticating users.
+    """
+
+    @staticmethod
+    def insert_default_user():
+        """
+        Inserts the default admin user.
+        """
+        user = {
+            "username": "admin",
+            "password": encode_password("admin"),
+            "roles": json.dumps(["admin"]),
+        }
+        user_tuple = tuple(user.values())
+
+        sql = """INSERT INTO users(
+        username,
+        password,
+        roles
+        ) VALUES(:username, :password, :roles)
+        """
+
+        with SQLiteManager(userdata_db=True) as cur:
+            cur.execute(sql, user_tuple)
+            cur.close()
+
+    @staticmethod
+    def insert_guest_user():
+        """
+        Inserts the default guest user.
+        """
+        user = {
+            "username": "guest",
+            "password": encode_password("guest"),
+            "firstname": "Guest",
+            "lastname": "User",
+            "roles": json.dumps(["guest"]),
+        }
+        user_tuple = tuple(user.values())
+
+        sql = """INSERT INTO users(
+        username,
+        password,
+        firstname,
+        lastname,
+        roles
+        ) VALUES(:username, :password, :firstname, :lastname, :roles)
+        """
+
+        with SQLiteManager(userdata_db=True) as cur:
+            cur.execute(sql, user_tuple)
+            cur.close()
+
+    @staticmethod
+    def get_all_users():
+        """
+        Check if there are any users in the database.
+        """
+        sql = "SELECT * FROM users"
+
+        with SQLiteManager(userdata_db=True) as cur:
+            cur.execute(sql)
+
+            data = cur.fetchall()
+            cur.close()
+
+            return [User(*user) for user in data]
+
+    @staticmethod
+    def get_user_by_username(username: str):
+        """
+        Get a user by username.
+        """
+        sql = "SELECT * FROM users WHERE username = ?"
+
+        with SQLiteManager(userdata_db=True) as cur:
+            cur.execute(sql, (username,))
+
+            data = cur.fetchone()
+            cur.close()
+
+            if data is not None:
+                return User(*data)
+
+            return None

app/db/sqlite/queries.py

@@ -54,6 +54,17 @@ CREATE TABLE IF NOT EXISTS track_logger (
     timestamp integer NOT NULL,
     source text,
     userid integer NOT NULL DEFAULT 0
+);
+
+CREATE TABLE IF NOT EXISTS users (
+    id integer PRIMARY KEY,
+    username text NOT NULL UNIQUE,
+    firstname text,
+    lastname text,
+    password text NOT NULL,
+    email text,
+    image text,
+    roles text NOT NULL DEFAULT '["user"]'
 )
 """