Remade Image Selector for Profil Picture in UserSettings

Author: JadnK

src/main/java/de/jadenk/springcloud/controller/UserSettingController.java

@@ -61,7 +61,7 @@ private Map<String, Object> getUserAttributes(User user) {
 
 
     @GetMapping("/settings")
-    public String userSettingsPage(Model model) {
+    public String userSettingsPage(@RequestParam(value = "error", required = false) String error,Model model) {
         Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
         String username = authentication.getName();
 
@@ -107,6 +107,8 @@ public String updateUserSettings(@RequestParam("username") String username,
             user.setEmail(email);
         }
 
+
+
         boolean wantsToChangePassword =
                 currentPassword != null && !currentPassword.trim().isEmpty() &&
                         newPassword != null && !newPassword.trim().isEmpty() &&
@@ -139,6 +141,16 @@ public String updateUserSettings(@RequestParam("username") String username,
         user.setNotificationsEnabled(notificationsEnabled);
 
         if (profileImage != null && !profileImage.isEmpty()) {
+            String originalFilename = profileImage.getOriginalFilename();
+            if (originalFilename != null &&
+                    !originalFilename.toLowerCase().endsWith(".png") &&
+                    !originalFilename.toLowerCase().endsWith(".jpg") &&
+                    !originalFilename.toLowerCase().endsWith(".jpeg")) {
+                model.addAttribute("error", messageService.getError("settings.upload.wrong.ending"));
+                model.addAttribute("userAttributes", getUserAttributes(user));
+                return "user-settings";
+            }
+
             try {
                 InputStream is = profileImage.getInputStream();
                 BufferedImage originalImage = ImageIO.read(is);

src/main/java/de/jadenk/springcloud/util/MessageService.java

@@ -14,6 +14,7 @@ public class MessageService {
             "changepassword.same.password", "You cant Use the same Password as before!",
             "auth.invalid", "Username or password is incorrect.",
             "settings.upload.error", "Error while uploading Profile Picture, try again later.",
+            "settings.upload.wrong.ending", "You can only Upload .jpg, .png and .jpeg files.",
             "settings.username.exists", "This username is already in use.",
             "settings.email.exists", "This mail is already in use.",
             "register.username.exists", "Username already exists."

src/main/resources/templates/user-settings.html

@@ -63,7 +63,7 @@ <h2>Profil Information</h2>
                 <label>Profilpicture:</label>
                 <div class="profile-picture-upload">
                     <img th:src="${userAttributes.profileImageData}" alt="Profilbild" class="profile-preview">
-                    <input type="file" name="profileImage" accept="image/*">
+                    <input type="file" name="profileImage" accept="image/png, image/jpeg, image/jpg">
                 </div>
             </div>