Add user preferences logic to server

Author: esadakcam

server/user/build.gradle.kts

@@ -1,3 +1,7 @@
+dependencies {
+	implementation("org.springframework.boot:spring-boot-starter-data-mongodb")
+}
+
 jib {
     from {
         image = "eclipse-temurin:21-jre-alpine"

server/user/src/main/java/com/continiousdisappointment/user/controller/UserPreferencesController.java

@@ -0,0 +1,39 @@
+package com.continiousdisappointment.user.controller;
+
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
+import org.springframework.stereotype.Controller;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestHeader;
+import org.springframework.web.bind.annotation.RequestMapping;
+
+import com.continiousdisappointment.user.dto.SaveUserPreferencesRequestDto;
+import com.continiousdisappointment.user.service.UserService;
+
+import jakarta.websocket.server.PathParam;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.log4j.Log4j2;
+
+@Controller
+@RequestMapping("/preferences")
+@Log4j2
+@RequiredArgsConstructor
+public class UserPreferencesController {
+    private final UserService userService;
+
+    @PostMapping
+    public ResponseEntity<Object> saveUserPreferences(@RequestHeader("Authorization") String authorization,
+            @RequestBody SaveUserPreferencesRequestDto request) {
+        try {
+
+            userService.saveUserPreferences(authorization, request.dietaryPreferences());
+
+        } catch (Exception e) {
+            log.warn("Error while saving user preferences", e);
+            return ResponseEntity.status(HttpStatus.UNAUTHORIZED).build();
+        }
+        return null;
+    }
+}

server/user/src/main/java/com/continiousdisappointment/user/domain/DietaryPreference.java

@@ -0,0 +1,10 @@
+package com.continiousdisappointment.user.domain;
+
+public enum DietaryPreference {
+    VEGETARIAN,
+    VEGAN,
+    GLUTEN_FREE,
+    DAIRY_FREE,
+    NUT_FREE,
+    SPICY_FOOD,
+}

server/user/src/main/java/com/continiousdisappointment/user/domain/OAuthUser.java

@@ -0,0 +1,11 @@
+package com.continiousdisappointment.user.domain;
+
+import lombok.Data;
+import lombok.RequiredArgsConstructor;
+
+@Data
+@RequiredArgsConstructor
+public class OAuthUser {
+    private final String username;
+    private final Integer id;
+}

server/user/src/main/java/com/continiousdisappointment/user/domain/User.java

@@ -1,12 +1,18 @@
 package com.continiousdisappointment.user.domain;
 
-public record User(Integer id, String username) {
-    public User {
-        if (id == null || id < 0) {
-            throw new IllegalArgumentException("ID must be a non-negative integer.");
-        }
-        if (username == null || username.isBlank()) {
-            throw new IllegalArgumentException("Username cannot be null or blank.");
-        }
+import java.util.Set;
+
+import org.springframework.lang.NonNull;
+
+public class User extends OAuthUser {
+    private final Set<DietaryPreference> dietaryPreferences;
+
+    public User(String username, Integer id, @NonNull Set<DietaryPreference> dietaryPreferences) {
+        super(username, id);
+        this.dietaryPreferences = dietaryPreferences;
+    }
+
+    public Set<DietaryPreference> getDietaryPreferences() {
+        return dietaryPreferences;
     }
 }

server/user/src/main/java/com/continiousdisappointment/user/dto/SaveUserPreferencesRequestDto.java

@@ -0,0 +1,14 @@
+package com.continiousdisappointment.user.dto;
+
+import java.util.Set;
+
+import com.continiousdisappointment.user.domain.DietaryPreference;
+
+public record SaveUserPreferencesRequestDto(Set<DietaryPreference> dietaryPreferences) {
+
+    public SaveUserPreferencesRequestDto {
+        if (dietaryPreferences == null || dietaryPreferences.isEmpty()) {
+            throw new IllegalArgumentException("Dietary preferences cannot be null or empty");
+        }
+    }
+}

server/user/src/main/java/com/continiousdisappointment/user/model/UserPreferencesModel.java

@@ -0,0 +1,24 @@
+package com.continiousdisappointment.user.model;
+
+import java.util.Set;
+import java.util.UUID;
+
+import org.springframework.data.annotation.Id;
+import org.springframework.data.mongodb.core.mapping.Document;
+
+import com.continiousdisappointment.user.domain.DietaryPreference;
+
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+@Data
+@NoArgsConstructor
+@AllArgsConstructor
+@Document(collection = "user-preferences")
+public class UserPreferencesModel {
+    @Id
+    public UUID id;
+    public int userId;
+    public Set<DietaryPreference> dietaryPreferences;
+}

server/user/src/main/java/com/continiousdisappointment/user/repository/UserPreferencesRepository.java

@@ -0,0 +1,17 @@
+package com.continiousdisappointment.user.repository;
+
+import java.util.List;
+import java.util.Optional;
+import java.util.UUID;
+
+import org.springframework.data.mongodb.repository.MongoRepository;
+import org.springframework.lang.NonNull;
+
+import com.continiousdisappointment.user.model.UserPreferencesModel;
+
+public interface UserPreferencesRepository extends MongoRepository<UserPreferencesModel, UUID> {
+
+    List<UserPreferencesModel> findByUserId(int userId);
+
+    Optional<UserPreferencesModel> findById(@NonNull UUID id);
+}

server/user/src/main/java/com/continiousdisappointment/user/service/UserService.java

@@ -1,20 +1,52 @@
 package com.continiousdisappointment.user.service;
 
+import lombok.RequiredArgsConstructor;
 import lombok.extern.log4j.Log4j2;
+
+import java.util.Set;
+
 import org.springframework.http.*;
 import org.springframework.stereotype.Service;
 import org.springframework.web.client.RestTemplate;
 
+import com.continiousdisappointment.user.domain.DietaryPreference;
+import com.continiousdisappointment.user.domain.OAuthUser;
 import com.continiousdisappointment.user.domain.User;
+import com.continiousdisappointment.user.model.UserPreferencesModel;
+import com.continiousdisappointment.user.repository.UserPreferencesRepository;
 
 @Log4j2
 @Service
+@RequiredArgsConstructor
 public class UserService {
 
     private final RestTemplate restTemplate = new RestTemplate();
+    private final UserPreferencesRepository userPreferencesRepository;
 
     public User getUserInfo(String authorization) {
-        String url = "https://gitlab.lrz.de/api/v4/user";
+        OAuthUser oAuthUser = getOAuthUserInfo(authorization);
+        var userPreferences = getUserPreferences(authorization);
+        return new User(oAuthUser.getUsername(), oAuthUser.getId(), userPreferences);
+    }
+
+    public void saveUserPreferences(String authorization, Set<DietaryPreference> dietaryPreferences) {
+        OAuthUser oAuthUser = getOAuthUserInfo(authorization);
+        var userPreferences = new UserPreferencesModel();
+        userPreferences.setUserId(oAuthUser.getId());
+        userPreferences.setDietaryPreferences(dietaryPreferences);
+        userPreferencesRepository.save(userPreferences);
+    }
+
+    private Set<DietaryPreference> getUserPreferences(String authorization) {
+        OAuthUser oAuthUser = getOAuthUserInfo(authorization);
+        return userPreferencesRepository.findByUserId(oAuthUser.getId()).stream()
+                .findFirst()
+                .map(UserPreferencesModel::getDietaryPreferences)
+                .orElse(Set.of());
+    }
+
+    private OAuthUser getOAuthUserInfo(String authorization) {
+        String url = "https://gitlab.lrz.de/api{/v4/user";
         if (authorization == null) {
             log.warn("No access token found in security context");
             throw new IllegalStateException("No access token found in security context");
@@ -24,11 +56,11 @@ public User getUserInfo(String authorization) {
 
         HttpEntity<Void> requestEntity = new HttpEntity<>(headers);
 
-        ResponseEntity<User> response = restTemplate.exchange(
+        ResponseEntity<OAuthUser> response = restTemplate.exchange(
                 url,
                 HttpMethod.GET,
                 requestEntity,
-                User.class);
+                OAuthUser.class);
         return response.getBody();
     }
 }

server/user/src/main/resources/application-dev.yaml

@@ -0,0 +1,4 @@
+spring:
+  data:
+    mongodb:
+      uri: mongodb://admin:admin@127.0.0.1:27017/recipai?authSource=admin