feat: Add profile images taken from the nft

website/models.py

@@ -12,6 +12,7 @@ db = SQLAlchemy()
 class User(db.Model):
     id = db.Column(db.Integer, primary_key=True)
     username = db.Column(db.String(40), unique=True)
+    profile_picture = db.Column(db.String(255), nullable=True)
 
     def __str__(self):
         return self.username
@@ -19,6 +20,9 @@ class User(db.Model):
     def get_user_id(self):
         return self.id
     
+    def get_user_profile_picture(self):
+        return self.profile_picture
+    
     def check_password(self, password):
         return password == 'valid'
 

website/routes.py

@@ -119,7 +119,8 @@ def hnsid_domain(domain):
             # Add domain to the session
             user = User.query.filter_by(username=domain).first()
             if not user:
-                user = User(username=domain)
+                # Create user with username and profile picture
+                user = User(username=domain, profile_picture=nft["image_url"])
                 db.session.add(user)
                 db.session.commit()
             session["id"] = user.id
@@ -230,6 +231,11 @@ def revoke_token():
 @require_oauth(["profile", "openid"])
 def api_me():
     user = current_token.user
+    picture = f"https://login.hns.au/u/{user.username}/avatar.png"
+
+    if user.profile_picture:
+        picture = user.profile_picture
+
     userInfo = {
         "id": user.id,
         "uid": user.id,
@@ -243,7 +249,7 @@ def api_me():
         "nickname": user.username,
         "preferred_username": user.username,
         "profile": f"https://login.hns.au/u/{user.username}",
-        "picture": f"https://login.hns.au/u/{user.username}/avatar.png",
+        "picture": picture,
         "website": f"https://{user.username}",
         "email_verified": True
     }
@@ -275,10 +281,39 @@ def autodiscovery():
 @bp.route("/u/<username>")
 def profile(username):
     user = User.query.filter_by(username=username).first()
-    return jsonify({"name": user.username, "id": user.id})
+    if not user:
+        return jsonify({"error": "User not found"})
+    picture = f"https://login.hns.au/u/{user.username}/avatar.png"
+
+    if user.profile_picture:
+        picture = user.profile_picture
+
+    userInfo = {
+        "id": user.id,
+        "uid": user.id,
+        "username": user.username,
+        "email": f"{user.username}@login.hns.au",
+        "displayName": user.username + "/",
+        "sub": user.id,
+        "name": user.username,
+        "given_name": user.username,
+        "family_name": user.username,
+        "nickname": user.username,
+        "preferred_username": user.username,
+        "profile": f"https://login.hns.au/u/{user.username}",
+        "picture": picture,
+        "website": f"https://{user.username}",
+        "email_verified": True
+    }
+    return jsonify(userInfo)
 
 @bp.route("/u/<username>/avatar.png")
 def avatar(username):
+    user = User.query.filter_by(username=username).first()
+    if not user:
+        print("User not found")
+        return send_from_directory("templates", "favicon.png", mimetype="image/png")
+    
     # Check if file exists
     if os.path.exists(f"website/avatars/{username}.png"):
         return send_from_directory("avatars", f"{username}.png", mimetype="image/png")