Implement GraphQL mutations addMediaToUserLibrary and removeMediaFromUserLibrary with tests

Agent-Logs-Url: https://github.com/EspacoGeek-Teams/SpringAPI_EspacoGeek/sessions/18f5c822-c6fb-49ea-9fe5-aae0c61b4a39

Co-authored-by: AbigailGeovana <142514517+AbigailGeovana@users.noreply.github.com>

Author: Copilot

src/main/java/com/espacogeek/geek/controllers/UserLibraryController.java

@@ -0,0 +1,63 @@
+package com.espacogeek.geek.controllers;
+
+import java.util.List;
+
+import org.springframework.graphql.data.method.annotation.Argument;
+import org.springframework.graphql.data.method.annotation.MutationMapping;
+import org.springframework.graphql.data.method.annotation.QueryMapping;
+import org.springframework.security.access.prepost.PreAuthorize;
+import org.springframework.security.core.Authentication;
+import org.springframework.stereotype.Controller;
+
+import com.espacogeek.geek.models.UserLibraryModel;
+import com.espacogeek.geek.services.UserLibraryService;
+import com.espacogeek.geek.utils.UserUtils;
+
+import lombok.RequiredArgsConstructor;
+
+/**
+ * GraphQL controller for user media library operations.
+ * All endpoints require the user to be authenticated ({@code hasRole('user')}).
+ */
+@Controller
+@RequiredArgsConstructor
+public class UserLibraryController {
+
+    private final UserLibraryService userLibraryService;
+
+    /**
+     * Returns all media entries in the authenticated user's library.
+     */
+    @QueryMapping(name = "findUserMediaLibrary")
+    @PreAuthorize("hasRole('user')")
+    public List<UserLibraryModel> findUserMediaLibrary(Authentication authentication) {
+        Integer userId = UserUtils.getUserID(authentication);
+        return userLibraryService.findByUserId(userId);
+    }
+
+    /**
+     * Adds the specified media to the authenticated user's library.
+     *
+     * @param mediaId ID of the media to add
+     * @return the newly created library entry
+     */
+    @MutationMapping(name = "addMediaToUserLibrary")
+    @PreAuthorize("hasRole('user')")
+    public UserLibraryModel addMediaToUserLibrary(Authentication authentication, @Argument(name = "mediaId") Integer mediaId) {
+        Integer userId = UserUtils.getUserID(authentication);
+        return userLibraryService.addMedia(userId, mediaId);
+    }
+
+    /**
+     * Removes the specified media from the authenticated user's library.
+     *
+     * @param mediaId ID of the media to remove
+     * @return {@code true} if the entry was removed, {@code false} if it was not found
+     */
+    @MutationMapping(name = "removeMediaFromUserLibrary")
+    @PreAuthorize("hasRole('user')")
+    public Boolean removeMediaFromUserLibrary(Authentication authentication, @Argument(name = "mediaId") Integer mediaId) {
+        Integer userId = UserUtils.getUserID(authentication);
+        return userLibraryService.removeMedia(userId, mediaId);
+    }
+}

src/main/java/com/espacogeek/geek/repositories/TypeStatusRepository.java

@@ -0,0 +1,10 @@
+package com.espacogeek.geek.repositories;
+
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
+
+import com.espacogeek.geek.models.TypeStatusModel;
+
+@Repository
+public interface TypeStatusRepository extends JpaRepository<TypeStatusModel, Integer> {
+}

src/main/java/com/espacogeek/geek/repositories/UserLibraryRepository.java

@@ -1,9 +1,19 @@
 package com.espacogeek.geek.repositories;
 
+import java.util.List;
+import java.util.Optional;
+
 import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
 
 import com.espacogeek.geek.models.UserLibraryModel;
 
+@Repository
 public interface UserLibraryRepository extends JpaRepository<UserLibraryModel, Integer> {
-    
+
+    List<UserLibraryModel> findByUserId(Integer userId);
+
+    Optional<UserLibraryModel> findByUserIdAndMediaId(Integer userId, Integer mediaId);
+
+    void deleteByUserIdAndMediaId(Integer userId, Integer mediaId);
 }

src/main/java/com/espacogeek/geek/services/UserLibraryService.java

@@ -0,0 +1,39 @@
+package com.espacogeek.geek.services;
+
+import java.util.List;
+
+import com.espacogeek.geek.models.UserLibraryModel;
+
+/**
+ * Service interface for managing a user's personal media library.
+ */
+public interface UserLibraryService {
+
+    /**
+     * Retrieves all media entries in the given user's library.
+     *
+     * @param userId the ID of the user
+     * @return list of {@link UserLibraryModel} entries belonging to the user
+     */
+    List<UserLibraryModel> findByUserId(Integer userId);
+
+    /**
+     * Adds a media item to the user's library with default values
+     * (status = "Planning", progress = 0).
+     *
+     * @param userId  the ID of the authenticated user
+     * @param mediaId the ID of the media to add
+     * @return the newly created {@link UserLibraryModel} entry
+     */
+    UserLibraryModel addMedia(Integer userId, Integer mediaId);
+
+    /**
+     * Removes a media item from the user's library.
+     *
+     * @param userId  the ID of the authenticated user
+     * @param mediaId the ID of the media to remove
+     * @return {@code true} if the entry existed and was deleted,
+     *         {@code false} if no matching entry was found
+     */
+    boolean removeMedia(Integer userId, Integer mediaId);
+}

src/main/java/com/espacogeek/geek/services/impl/UserLibraryServiceImpl.java

@@ -0,0 +1,74 @@
+package com.espacogeek.geek.services.impl;
+
+import java.util.List;
+import java.util.Optional;
+
+import org.springframework.http.HttpStatus;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+import com.espacogeek.geek.exception.GenericException;
+import com.espacogeek.geek.models.MediaModel;
+import com.espacogeek.geek.models.TypeStatusModel;
+import com.espacogeek.geek.models.UserLibraryModel;
+import com.espacogeek.geek.models.UserModel;
+import com.espacogeek.geek.repositories.MediaRepository;
+import com.espacogeek.geek.repositories.TypeStatusRepository;
+import com.espacogeek.geek.repositories.UserLibraryRepository;
+import com.espacogeek.geek.repositories.UserRepository;
+import com.espacogeek.geek.services.UserLibraryService;
+
+import lombok.RequiredArgsConstructor;
+
+/**
+ * Implementation of {@link UserLibraryService}.
+ */
+@Service
+@RequiredArgsConstructor
+public class UserLibraryServiceImpl implements UserLibraryService {
+
+    /** ID of the default "Planning" status in the {@code types_status} table. */
+    private static final int DEFAULT_STATUS_ID = 1;
+
+    private final UserLibraryRepository userLibraryRepository;
+    private final UserRepository userRepository;
+    private final MediaRepository mediaRepository;
+    private final TypeStatusRepository typeStatusRepository;
+
+    @Override
+    public List<UserLibraryModel> findByUserId(Integer userId) {
+        return userLibraryRepository.findByUserId(userId);
+    }
+
+    @Override
+    @Transactional
+    public UserLibraryModel addMedia(Integer userId, Integer mediaId) {
+        UserModel user = userRepository.findById(userId)
+                .orElseThrow(() -> new GenericException(HttpStatus.NOT_FOUND.toString()));
+
+        MediaModel media = mediaRepository.findById(mediaId)
+                .orElseThrow(() -> new GenericException(HttpStatus.NOT_FOUND.toString()));
+
+        TypeStatusModel defaultStatus = typeStatusRepository.findById(DEFAULT_STATUS_ID)
+                .orElseThrow(() -> new GenericException(HttpStatus.INTERNAL_SERVER_ERROR.toString()));
+
+        UserLibraryModel entry = new UserLibraryModel();
+        entry.setUser(user);
+        entry.setMedia(media);
+        entry.setTypeStatus(defaultStatus);
+        entry.setProgress(0);
+
+        return userLibraryRepository.save(entry);
+    }
+
+    @Override
+    @Transactional
+    public boolean removeMedia(Integer userId, Integer mediaId) {
+        Optional<UserLibraryModel> entry = userLibraryRepository.findByUserIdAndMediaId(userId, mediaId);
+        if (entry.isEmpty()) {
+            return false;
+        }
+        userLibraryRepository.deleteByUserIdAndMediaId(userId, mediaId);
+        return true;
+    }
+}

src/main/resources/graphql/entity/UserMediaList.graphqls

@@ -0,0 +1,25 @@
+"""
+Represents a media entry in the user's personal library
+"""
+type UserMediaList {
+    "Unique library entry identifier"
+    id: ID
+    "Watching/reading progress"
+    progress: Int
+    "Date the media was added to the library"
+    addedAt: Date
+    "Current tracking status (e.g. Planning, Watching, Completed)"
+    typeStatus: TypeStatus
+    "The media associated with this entry"
+    media: Media
+}
+
+"""
+Represents a media tracking status (e.g. Planning, Watching, Completed)
+"""
+type TypeStatus {
+    "Unique status identifier"
+    id: ID
+    "Status name"
+    name: String
+}

src/main/resources/graphql/mutation.graphqls

@@ -76,4 +76,19 @@ type Mutation {
     Returns: Success message
     """
     verifyEmailChange(token: String!): String
+
+    """
+    Add a media item to the authenticated user's library. Requires authentication.
+    Creates a new library entry with default status (Planning) and progress 0.
+    Example: addMediaToUserLibrary(mediaId: 42)
+    Returns: The newly created UserMediaList entry.
+    """
+    addMediaToUserLibrary(mediaId: Int!): UserMediaList
+
+    """
+    Remove a media item from the authenticated user's library. Requires authentication.
+    Example: removeMediaFromUserLibrary(mediaId: 42)
+    Returns: true if the entry was removed, false if it was not found.
+    """
+    removeMediaFromUserLibrary(mediaId: Int!): Boolean
 }

src/main/resources/graphql/query.graphqls

@@ -75,4 +75,10 @@ type Query {
     Example: getBatchJobs(page: 0, size: 10, status: "FAILED")
     """
     getBatchJobs(page: Int!, size: Int!, status: String): BatchJobPage!
+
+    """
+    Get the authenticated user's media library. Requires authentication via Authorization: Bearer header.
+    Returns a list of all media entries tracked by the current user.
+    """
+    findUserMediaLibrary: [UserMediaList]
 }