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

package com.xuqm.push.controller;

import com.xuqm.common.model.ApiResponse;
import com.xuqm.push.entity.DeviceTokenEntity;
import com.xuqm.push.service.PushDispatcher;
import jakarta.validation.constraints.NotBlank;
import jakarta.validation.constraints.NotNull;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

@RestController
@RequestMapping("/api/push")
public class PushController {

    private final PushDispatcher pushDispatcher;

    public PushController(PushDispatcher pushDispatcher) {
        this.pushDispatcher = pushDispatcher;
    }

    @PostMapping("/register")
    public ResponseEntity<ApiResponse<Void>> register(
            @RequestParam @NotBlank String appKey,
            @RequestParam @NotBlank String userId,
            @RequestParam @NotNull DeviceTokenEntity.Vendor vendor,
            @RequestParam @NotBlank String token,
            @RequestParam(required = false) String platform,
            @RequestParam(required = false) String deviceId,
            @RequestParam(required = false) String brand,
            @RequestParam(required = false) String model,
            @RequestParam(required = false) String osVersion,
            @RequestParam(required = false) String appVersion) {
        pushDispatcher.registerToken(appKey, userId, vendor, token, platform, deviceId, brand, model, osVersion, appVersion);
        return ResponseEntity.ok(ApiResponse.ok());
    }

    @PostMapping("/receive-push")
    public ResponseEntity<ApiResponse<Void>> receivePush(
            @RequestParam @NotBlank String appKey,
            @RequestParam @NotBlank String userId,
            @RequestParam(required = false) String deviceId,
            @RequestParam boolean enabled) {
        pushDispatcher.setReceivePush(appKey, userId, deviceId, enabled);
        return ResponseEntity.ok(ApiResponse.ok());
    }

    @PostMapping("/send")
    public ResponseEntity<ApiResponse<Void>> send(
            @RequestParam @NotBlank String appKey,
            @RequestParam @NotBlank String userId,
            @RequestParam @NotBlank String title,
            @RequestParam @NotBlank String body,
            @RequestParam(required = false) String payload) {
        pushDispatcher.pushToUser(appKey, userId, title, body, payload);
        return ResponseEntity.ok(ApiResponse.ok());
    }

    @DeleteMapping("/unregister")
    public ResponseEntity<ApiResponse<Void>> unregister(
            @RequestParam @NotBlank String appKey,
            @RequestParam @NotBlank String userId,
            @RequestParam @NotNull DeviceTokenEntity.Vendor vendor,
            @RequestParam(required = false) String deviceId) {
        pushDispatcher.unregisterToken(appKey, userId, vendor, deviceId);
        return ResponseEntity.ok(ApiResponse.ok());
    }
}