Add JWT and Bcrypt utility classes for authentication

src/utils/__init__.py

@@ -1,6 +1,14 @@
+from ._jwt_util import JWTUtil
+from ._bcrypt_util import BcryptUtil
 from ._openai_client import OpenAIClient
 from ._pinecone_client import PineconeClient
 
-__all__ = ["OpenAIClient", "PineconeClient"]
+__all__ = [
+  "JWTUtil",
+  "BcryptUtil",
+  "OpenAIClient",
+  "PineconeClient",
+]
+
 __version__ = "0.1.0"
 __author__ = "Ramanjit Singh"

src/utils/_bcrypt_util.py

@@ -0,0 +1,17 @@
+import bcrypt
+
+
+class BcryptUtil:
+    @staticmethod
+    def hash_password(password: str) -> str:
+
+        salt = bcrypt.gensalt()
+        hashed_password = bcrypt.hashpw(password.encode('utf-8'), salt)
+        return hashed_password.decode('utf-8')
+
+    @staticmethod
+    def compare_password(stored_password: str, provided_password: str) -> bool:
+
+        stored_password = stored_password.encode('utf-8')
+        provided_password = provided_password.encode('utf-8')
+        return bcrypt.checkpw(provided_password, stored_password)

src/utils/_jwt_util.py

@@ -0,0 +1,20 @@
+import os
+import jwt
+from datetime import datetime, timedelta
+
+class JWTUtil:
+  def __init__(self, algorithm='HS256'):
+    self.secret_key = os.getenv('JWT_SECRET_KEY') 
+    self.algorithm = algorithm
+
+  def generate_jwt(self, payload, expiration_minutes=60):
+    payload['exp'] = datetime.now() + timedelta(minutes=expiration_minutes)
+    return jwt.encode(payload, self.secret_key, algorithm=self.algorithm)
+
+  def validate_jwt(self, token):
+    try:
+      return jwt.decode(token, self.secret_key, algorithms=[self.algorithm])
+    except jwt.ExpiredSignatureError:
+      return None
+    except jwt.InvalidTokenError:
+      return None