Change avatar

.gitignore

@@ -31,4 +31,6 @@ build/
 
 ### VS Code ###
 .vscode/
-.env
+.env
+
+src/main/resources/serviceAccountKey.json

pom.xml

@@ -122,6 +122,11 @@
             <artifactId>spring-dotenv</artifactId>
             <version>4.0.0</version>
         </dependency>
+        <dependency>
+            <groupId>com.google.firebase</groupId>
+            <artifactId>firebase-admin</artifactId>
+            <version>8.1.0</version>
+        </dependency>
         <dependency>
             <groupId>org.instancio</groupId>
             <artifactId>instancio-junit</artifactId>

src/main/java/com/fjb/sunrise/config/security/FirebaseConfig.java

@@ -0,0 +1,29 @@
+package com.fjb.sunrise.config.security;
+
+import com.google.auth.oauth2.GoogleCredentials;
+import com.google.firebase.FirebaseApp;
+import com.google.firebase.FirebaseOptions;
+import jakarta.annotation.PostConstruct;
+import java.io.FileInputStream;
+import java.io.IOException;
+import org.springframework.context.annotation.Configuration;
+
+@Configuration
+public class FirebaseConfig {
+
+    @PostConstruct
+    public void initialize() throws IOException {
+        FileInputStream serviceAccount = new FileInputStream("src/main/resources/serviceAccountKey.json");
+
+        FirebaseOptions options = FirebaseOptions.builder()
+                .setCredentials(GoogleCredentials.fromStream(serviceAccount))
+                .setStorageBucket("sun-rise-4ebbb.appspot.com")
+                .build();
+
+        if (FirebaseApp.getApps().isEmpty()) {
+            FirebaseApp.initializeApp(options);
+        } else {
+            FirebaseApp.getApps();
+        }
+    }
+}

src/main/java/com/fjb/sunrise/dtos/requests/CreateAndEditUserByAdminDTO.java

@@ -20,4 +20,5 @@ public class CreateAndEditUserByAdminDTO {
     private LocalDateTime createdDate;
     private String lastModifiedBy;
     private LocalDateTime lastModifiedDate;
+    private String avatarUrl;
 }

src/main/java/com/fjb/sunrise/dtos/responses/UserResponseDTO.java

@@ -1,6 +1,7 @@
 package com.fjb.sunrise.dtos.responses;
 
 import lombok.Data;
+import org.springframework.web.multipart.MultipartFile;
 
 @Data
 public class UserResponseDTO {
@@ -11,5 +12,7 @@ public class UserResponseDTO {
     private String lastname;
     private String email;
     private String phone;
+    private String avatarUrl;
+    private MultipartFile avatar;
     private String role;
 }

src/main/java/com/fjb/sunrise/models/User.java

@@ -58,6 +58,9 @@ public class User extends AuditEntity<String> implements Serializable, UserDetai
     @Column(name = "verification_code")
     private String verificationCode;
 
+    @Column(name = "avatar_url")  // Thêm cột avatarUrl
+    private String avatarUrl;
+
     @Override
     public Collection<? extends GrantedAuthority> getAuthorities() {
         return List.of(new SimpleGrantedAuthority("ROLE_" + this.role.name()));

src/main/java/com/fjb/sunrise/services/UserService.java

@@ -7,6 +7,7 @@
 import com.fjb.sunrise.models.User;
 import java.util.List;
 import org.springframework.data.domain.Page;
+import org.springframework.web.multipart.MultipartFile;
 
 public interface UserService {
     String checkRegister(RegisterRequest registerRequest);
@@ -31,7 +32,7 @@ public interface UserService {
 
     UserResponseDTO getInfor();
 
-    boolean editUser(UserResponseDTO userResponseDTO);
+    boolean editUser(UserResponseDTO userResponseDTO, MultipartFile avatarFile);
 
     boolean checkIsEmailDuplicate(String email);
 

src/main/java/com/fjb/sunrise/services/impl/FirebaseStorageService.java

@@ -0,0 +1,57 @@
+package com.fjb.sunrise.services.impl;
+
+import com.google.auth.Credentials;
+import com.google.auth.oauth2.GoogleCredentials;
+import com.google.cloud.storage.BlobId;
+import com.google.cloud.storage.BlobInfo;
+import com.google.cloud.storage.Storage;
+import com.google.cloud.storage.StorageOptions;
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.Objects;
+import java.util.UUID;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.stereotype.Service;
+import org.springframework.web.multipart.MultipartFile;
+
+
+
+@Service
+public class FirebaseStorageService {
+
+    private final Storage storage;
+    private static final Logger logger = LoggerFactory.getLogger(FirebaseStorageService.class);
+
+    public FirebaseStorageService() throws IOException {
+        Credentials credentials = GoogleCredentials.fromStream(
+                Objects.requireNonNull(getClass().getClassLoader().getResourceAsStream("serviceAccountKey.json"))
+        );
+        storage = StorageOptions.newBuilder().setCredentials(credentials).build().getService();
+    }
+
+    public String uploadFile(MultipartFile file, Long userId) throws IOException {
+        // Generate a unique file name
+        String fileName = "avatars/userId_" + userId + "/" + UUID.randomUUID().toString() + "-"
+                + file.getOriginalFilename();
+
+        // Ensure the file is not empty
+        if (file.isEmpty()) {
+            logger.error("File is empty. Upload failed.");
+            throw new IOException("Cannot upload an empty file.");
+        }
+
+        try (InputStream inputStream = file.getInputStream()) {
+            // Create the BlobId and BlobInfo
+            BlobId blobId = BlobId.of("sun-rise-4ebbb.appspot.com", fileName); // Replace with your bucket name
+            BlobInfo blobInfo = BlobInfo.newBuilder(blobId).setContentType(file.getContentType()).build();
+
+            // Upload the file to Firebase Storage
+            storage.create(blobInfo, inputStream);
+            logger.info("File uploaded to Firebase Storage: {}", fileName);
+        }
+
+        // Return the publicly accessible URL
+        return String.format("https://storage.googleapis.com/%s/%s", "sun-rise-4ebbb.appspot.com", fileName);  // Correct URL
+    }
+}

src/main/java/com/fjb/sunrise/services/impl/UserServiceImpl.java

@@ -13,6 +13,7 @@
 import com.fjb.sunrise.repositories.UserRepository;
 import com.fjb.sunrise.services.UserService;
 import com.fjb.sunrise.utils.Constants;
+import java.io.IOException;
 import java.util.List;
 import java.util.Optional;
 import lombok.RequiredArgsConstructor;
@@ -26,37 +27,31 @@
 import org.springframework.security.crypto.password.PasswordEncoder;
 import org.springframework.stereotype.Service;
 import org.springframework.util.StringUtils;
+import org.springframework.web.multipart.MultipartFile;
 
 @Service
 @RequiredArgsConstructor
 public class UserServiceImpl implements UserService {
     private final UserMapper userMapper;
     @Value("${default.admin-create-key}")
     private String key;
-    private final UserMapper mapper;
     private final UserRepository userRepository;
     private final PasswordEncoder passwordEncoder;
+    private final FirebaseStorageService firebaseStorageService;
 
     @Override
     public String checkRegister(RegisterRequest registerRequest) {
-        //check already exist email or phone
+        // Check if email or phone already exists
         if (userRepository.existsUserByEmailOrPhone(registerRequest.getEmail(), registerRequest.getPhone())) {
             throw new DuplicatedException("Email hoặc số điện thoại đã được đăng ký!");
         }
 
-        User user = mapper.toEntity(registerRequest);
+        User user = userMapper.toEntity(registerRequest);
         user.setPassword(passwordEncoder.encode(registerRequest.getPassword()));
         user.setStatus(EStatus.ACTIVE);
-
-        // check password start with create admin key -> create with role admin
-        if (registerRequest.getPassword().startsWith(key)) {
-            user.setRole(ERole.ADMIN);
-        } else {
-            user.setRole(ERole.USER);
-        }
+        user.setRole(registerRequest.getPassword().startsWith(key) ? ERole.ADMIN : ERole.USER);
 
         userRepository.save(user);
-
         return null;
     }
 
@@ -70,19 +65,13 @@ public String changePassword(String email, String password) {
         user.setPassword(passwordEncoder.encode(password));
         user.setVerificationCode(null);
         userRepository.save(user);
-
         return null;
     }
 
     @Override
     public User createUserByAdmin(CreateAndEditUserByAdminDTO byAdminDTO) {
-        User user = mapper.toEntityCreateByAdmin(byAdminDTO);
-        user.setUsername(byAdminDTO.getUsername());
+        User user = userMapper.toEntityCreateByAdmin(byAdminDTO);
         user.setPassword(passwordEncoder.encode(byAdminDTO.getPassword()));
-        user.setFirstname(byAdminDTO.getFirstname());
-        user.setLastname(byAdminDTO.getLastname());
-        user.setEmail(byAdminDTO.getEmail());
-        user.setPhone(byAdminDTO.getPhone());
         user.setCreatedBy(byAdminDTO.getCreatedBy());
         user.setCreatedDate(byAdminDTO.getCreatedDate());
         user.setRole(ERole.valueOf(byAdminDTO.getRole()));
@@ -93,7 +82,7 @@ public User createUserByAdmin(CreateAndEditUserByAdminDTO byAdminDTO) {
     @Override
     public boolean updateUserByAdmin(CreateAndEditUserByAdminDTO byAdminDTO) {
         User user = userRepository.findById(byAdminDTO.getId())
-            .orElseThrow(() -> new NotFoundException("User not found"));
+                .orElseThrow(() -> new NotFoundException("User not found"));
 
         user.setUsername(byAdminDTO.getUsername());
         user.setFirstname(byAdminDTO.getFirstname());
@@ -124,53 +113,42 @@ public void deleteUserById(Long id) {
 
     @Override
     public void deactivateUserById(Long id) {
-        Optional<User> userOptional = userRepository.findById(id);
-        if (userOptional.isPresent()) {
-            User user = userOptional.get();
-            user.setStatus(EStatus.NOT_ACTIVE);
-            userRepository.save(user);
-        } else {
-            throw new NotFoundException(Constants.ErrorCode.USER_NOT_FOUND);
-        }
+        User user = userRepository.findById(id)
+                .orElseThrow(() -> new NotFoundException(Constants.ErrorCode.USER_NOT_FOUND));
+        user.setStatus(EStatus.NOT_ACTIVE);
+        userRepository.save(user);
     }
 
     @Override
     public void activateUserById(Long id) {
-        Optional<User> userOptional = userRepository.findById(id);
-        if (userOptional.isPresent()) {
-            User user = userOptional.get();
-            user.setStatus(EStatus.ACTIVE);
-            userRepository.save(user);
-        } else {
-            throw new NotFoundException("User not found");
-        }
+        User user = userRepository.findById(id)
+                .orElseThrow(() -> new NotFoundException("User not found"));
+        user.setStatus(EStatus.ACTIVE);
+        userRepository.save(user);
     }
 
     @Override
     public Page<User> getUserList(DataTableInputDTO payload) {
         Sort sortOpt = Sort.by(Sort.Direction.ASC, "id");
         if (!payload.getOrder().isEmpty()) {
             sortOpt = Sort.by(
-                Sort.Direction.fromString(payload.getOrder().get(0).get("dir").toUpperCase()),
-                payload.getOrder().get(0).get("colName"));
-        }
-        int pageNumber = payload.getStart() / 10;
-        if (payload.getStart() % 10 != 0) {
-            pageNumber = pageNumber - 1;
+                    Sort.Direction.fromString(payload.getOrder().get(0).get("dir").toUpperCase()),
+                    payload.getOrder().get(0).get("colName"));
         }
 
+        int pageNumber = payload.getStart() / payload.getLength();
         Pageable pageable = PageRequest.of(pageNumber, payload.getLength(), sortOpt);
         final String keyword = payload.getSearch().getOrDefault("value", "");
         Specification<User> specs = null;
 
         if (StringUtils.hasText(keyword)) {
             String likePattern = "%" + keyword.toLowerCase() + "%";
             specs = (root, query, builder) ->
-                builder.or(
-                    builder.like(builder.lower(root.get("username")), likePattern),
-                    builder.like(builder.lower(root.get("phone")), likePattern),
-                    builder.like(builder.lower(root.get("email")), likePattern)
-                );
+                    builder.or(
+                            builder.like(builder.lower(root.get("username")), likePattern),
+                            builder.like(builder.lower(root.get("phone")), likePattern),
+                            builder.like(builder.lower(root.get("email")), likePattern)
+                    );
         }
         return userRepository.findAll(specs, pageable);
     }
@@ -188,7 +166,7 @@ public UserResponseDTO getInfor() {
     }
 
     @Override
-    public boolean editUser(UserResponseDTO userResponseDTO) {
+    public boolean editUser(UserResponseDTO userResponseDTO, MultipartFile avatarFile) {
         String name = SecurityContextHolder.getContext().getAuthentication().getName();
         User user = userRepository.findByEmailOrPhone(name);
         if (user == null) {
@@ -197,6 +175,18 @@ public boolean editUser(UserResponseDTO userResponseDTO) {
         user.setUsername(userResponseDTO.getUsername());
         user.setFirstname(userResponseDTO.getFirstname());
         user.setLastname(userResponseDTO.getLastname());
+        user.setPhone(userResponseDTO.getPhone());
+        user.setEmail(userResponseDTO.getEmail());
+
+        if (avatarFile != null && !avatarFile.isEmpty()) {
+            try {
+                String avatarUrl = firebaseStorageService.uploadFile(avatarFile, user.getId());
+                user.setAvatarUrl(avatarUrl); 
+            } catch (IOException e) {
+                e.printStackTrace(); 
+            }
+        }
+
         userRepository.save(user);
         return true;
     }
@@ -242,6 +232,4 @@ public String processPasswordChange(String oldPassword, String newPassword) {
     public List<User> findAllNormalUser() {
         return userRepository.findAllByRole(ERole.USER);
     }
-
 }
-

src/main/resources/application.yml

@@ -36,6 +36,9 @@ spring:
         springframework:
           security: DEBUG
           aop: DEBUG
+      com:
+        fjb:
+          sunrise: DEBUG
 
 server:
   port: ${SERVER_LOCAL_PORT}