From 475a6f77cfb7b28d081985ccedf6ed0782556813 Mon Sep 17 00:00:00 2001
From: heshunme <heshunme@outlook.com>
Date: Sun, 14 Jul 2024 13:34:40 +0800
Subject: [PATCH] add loadUserInfo method; set automatically call loadUserInfo
 after login

---
 .../java/com/web/web_assignment/ProjectBean.java    | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/main/java/com/web/web_assignment/ProjectBean.java b/src/main/java/com/web/web_assignment/ProjectBean.java
index 5e66427..e1c9279 100644
--- a/src/main/java/com/web/web_assignment/ProjectBean.java
+++ b/src/main/java/com/web/web_assignment/ProjectBean.java
@@ -20,6 +20,7 @@ public class ProjectBean implements Serializable {
     private final DatabaseService databaseService = new DatabaseService();
     private List<Project> projects; // List to store all projects
     private Project project; // Current project being worked on
+    private UserInfo userInfo; // User info for the current user
     private String username;
     private String password;
     public String projectKeywords = ""; // String to hold project keywords
@@ -57,6 +58,17 @@ public class ProjectBean implements Serializable {
         databaseService.close();
     }
 
+    public void loadUserInfo() {
+        try {
+            UserInfo info = databaseService.getUserInfo(username);
+            if (userInfo != null) {
+                userInfo = info;
+            }
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+    }
+
     public void loadProjects() {
         projects = databaseService.getProjectsByUser(Objects.requireNonNullElse(username, "admin"));
     }
@@ -131,6 +143,7 @@ public class ProjectBean implements Serializable {
         try {
             if (databaseService.validateUser(username, password)) {
                 FacesContext.getCurrentInstance().getExternalContext().getSessionMap().put("username", username);
+                loadUserInfo();
                 loadProjects();
                 return "project";
             } else {