OpenIsle/backend/src/main/java/com/openisle/controller/NotificationController.java

package com.openisle.controller;

import com.openisle.dto.NotificationDto;
import com.openisle.dto.NotificationMarkReadRequest;
import com.openisle.dto.NotificationUnreadCountDto;
import com.openisle.dto.NotificationPreferenceDto;
import com.openisle.dto.NotificationPreferenceUpdateRequest;
import com.openisle.mapper.NotificationMapper;
import com.openisle.service.NotificationService;
import lombok.RequiredArgsConstructor;
import org.springframework.security.core.Authentication;
import org.springframework.web.bind.annotation.*;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.media.ArraySchema;
import io.swagger.v3.oas.annotations.media.Content;
import io.swagger.v3.oas.annotations.media.Schema;
import io.swagger.v3.oas.annotations.responses.ApiResponse;
import io.swagger.v3.oas.annotations.security.SecurityRequirement;

import java.util.List;
import java.util.stream.Collectors;

/** Endpoints for user notifications. */
@RestController
@RequestMapping("/api/notifications")
@RequiredArgsConstructor
public class NotificationController {
    private final NotificationService notificationService;
    private final NotificationMapper notificationMapper;

    @GetMapping
    @Operation(summary = "List notifications", description = "Retrieve notifications for the current user")
    @ApiResponse(responseCode = "200", description = "Notifications",
            content = @Content(array = @ArraySchema(schema = @Schema(implementation = NotificationDto.class))))
    @SecurityRequirement(name = "JWT")
    public List<NotificationDto> list(@RequestParam(value = "page", defaultValue = "0") int page,
                                      @RequestParam(value = "size", defaultValue = "30") int size,
                                      Authentication auth) {
        return notificationService.listNotifications(auth.getName(), null, page, size).stream()
                .map(notificationMapper::toDto)
                .collect(Collectors.toList());
    }

    @GetMapping("/unread")
    @Operation(summary = "List unread notifications", description = "Retrieve unread notifications for the current user")
    @ApiResponse(responseCode = "200", description = "Unread notifications",
            content = @Content(array = @ArraySchema(schema = @Schema(implementation = NotificationDto.class))))
    @SecurityRequirement(name = "JWT")
    public List<NotificationDto> listUnread(@RequestParam(value = "page", defaultValue = "0") int page,
                                            @RequestParam(value = "size", defaultValue = "30") int size,
                                            Authentication auth) {
        return notificationService.listNotifications(auth.getName(), false, page, size).stream()
                .map(notificationMapper::toDto)
                .collect(Collectors.toList());
    }

    @GetMapping("/unread-count")
    @Operation(summary = "Unread count", description = "Get count of unread notifications")
    @ApiResponse(responseCode = "200", description = "Unread count",
            content = @Content(schema = @Schema(implementation = NotificationUnreadCountDto.class)))
    @SecurityRequirement(name = "JWT")
    public NotificationUnreadCountDto unreadCount(Authentication auth) {
        long count = notificationService.countUnread(auth.getName());
        NotificationUnreadCountDto uc = new NotificationUnreadCountDto();
        uc.setCount(count);
        return uc;
    }

    @PostMapping("/read")
    @Operation(summary = "Mark notifications read", description = "Mark notifications as read")
    @ApiResponse(responseCode = "200", description = "Marked read")
    @SecurityRequirement(name = "JWT")
    public void markRead(@RequestBody NotificationMarkReadRequest req, Authentication auth) {
        notificationService.markRead(auth.getName(), req.getIds());
    }

    @GetMapping("/prefs")
    @Operation(summary = "List preferences", description = "List notification preferences")
    @ApiResponse(responseCode = "200", description = "Preferences",
            content = @Content(array = @ArraySchema(schema = @Schema(implementation = NotificationPreferenceDto.class))))
    @SecurityRequirement(name = "JWT")
    public List<NotificationPreferenceDto> prefs(Authentication auth) {
        return notificationService.listPreferences(auth.getName());
    }

    @PostMapping("/prefs")
    @Operation(summary = "Update preference", description = "Update notification preference")
    @ApiResponse(responseCode = "200", description = "Preference updated")
    @SecurityRequirement(name = "JWT")
    public void updatePref(@RequestBody NotificationPreferenceUpdateRequest req, Authentication auth) {
        notificationService.updatePreference(auth.getName(), req.getType(), req.isEnabled());
    }

    @GetMapping("/email-prefs")
    @Operation(summary = "List email preferences", description = "List email notification preferences")
    @ApiResponse(responseCode = "200", description = "Email preferences",
            content = @Content(array = @ArraySchema(schema = @Schema(implementation = NotificationPreferenceDto.class))))
    @SecurityRequirement(name = "JWT")
    public List<NotificationPreferenceDto> emailPrefs(Authentication auth) {
        return notificationService.listEmailPreferences(auth.getName());
    }

    @PostMapping("/email-prefs")
    @Operation(summary = "Update email preference", description = "Update email notification preference")
    @ApiResponse(responseCode = "200", description = "Email preference updated")
    @SecurityRequirement(name = "JWT")
    public void updateEmailPref(@RequestBody NotificationPreferenceUpdateRequest req, Authentication auth) {
        notificationService.updateEmailPreference(auth.getName(), req.getType(), req.isEnabled());
    }
}