Refactor file uploading.

src/main/java/org/codiki/account/AccountService.java

@@ -4,6 +4,7 @@ import java.io.IOException;
 import java.util.Optional;
 
 import javax.naming.AuthenticationException;
+import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 
 import org.codiki.core.entities.dto.PasswordWrapperDTO;
@@ -75,8 +76,24 @@ public class AccountService {
 		}
 	}
 	
-	public String uploadFile(final MultipartFile pFile) {
-		return fileUploadService.uploadProfileImage(pFile);
+	public String uploadFile(final MultipartFile pFile,
+			final HttpServletRequest pRequest, final HttpServletResponse pResponse) throws IOException {
+		final String avatarFileName = fileUploadService.uploadProfileImage(pFile);
+		
+		final Optional<User> connectedUser = tokenService.getAuthenticatedUserByToken(pRequest);
+		if(connectedUser.isPresent()) {
+			final Optional<User> userFromDb = userRepository.findById(connectedUser.get().getId());
+			if(userFromDb.isPresent()) {
+				userFromDb.get().setImage(avatarFileName);
+				userRepository.save(userFromDb.get());
+			} else {
+				pResponse.sendError(HttpServletResponse.SC_UNAUTHORIZED);
+			}
+		} else {
+			pResponse.sendError(HttpServletResponse.SC_UNAUTHORIZED);
+		}
+		
+		return avatarFileName;
 	}
 
 	public Resource loadAvatar(final String pAvatarFileName) {