XuqmGroup-Server/push-service/src/main/java/com/xuqm/push/controller/PushManagementController.java

package com.xuqm.push.controller;

import com.xuqm.common.model.ApiResponse;
import com.xuqm.push.entity.DeviceLoginLogEntity;
import com.xuqm.push.service.PushDiagnosticsService;
import org.springframework.data.domain.Page;
import org.springframework.http.ResponseEntity;
import org.springframework.security.access.prepost.PreAuthorize;
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.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import java.util.Map;

@RestController
@RequestMapping("/api/push/admin")
@PreAuthorize("hasAnyAuthority('ROLE_OPS', 'ROLE_TENANT', 'ROLE_ADMIN')")
public class PushManagementController {

    private final PushDiagnosticsService diagnosticsService;

    public PushManagementController(PushDiagnosticsService diagnosticsService) {
        this.diagnosticsService = diagnosticsService;
    }

    @GetMapping("/user-status")
    public ResponseEntity<ApiResponse<PushDiagnosticsService.PushTokenDiagnostics>> userStatus(
            @RequestParam String appKey,
            @RequestParam String userId) {
        return ResponseEntity.ok(ApiResponse.success(diagnosticsService.searchByUserId(appKey, userId)));
    }

    @GetMapping("/device-logs")
    public ResponseEntity<ApiResponse<Map<String, Object>>> deviceLogs(
            @RequestParam String appKey,
            @RequestParam String userId,
            @RequestParam(defaultValue = "0") int page,
            @RequestParam(defaultValue = "20") int size) {
        Page<DeviceLoginLogEntity> result = diagnosticsService.deviceLogs(appKey, userId, page, size);
        return ResponseEntity.ok(ApiResponse.success(Map.of(
                "content", result.getContent(),
                "total", result.getTotalElements(),
                "totalPages", result.getTotalPages()
        )));
    }

    @PostMapping("/test-offline")
    public ResponseEntity<ApiResponse<PushDiagnosticsService.TestPushResult>> testOffline(
            @RequestBody TestOfflineRequest request) {
        PushDiagnosticsService.TestPushResult result = diagnosticsService.sendTestOfflineMessage(
                request.appKey(),
                request.userId(),
                request.title(),
                request.body(),
                request.payload());
        return ResponseEntity.ok(ApiResponse.success(result));
    }

    public record TestOfflineRequest(
            String appKey,
            String userId,
            String title,
            String body,
            String payload
    ) {}
}