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

package com.openisle.controller;

import com.openisle.service.OpenAiService;
import com.openisle.service.AiUsageService;
import lombok.RequiredArgsConstructor;
import org.springframework.http.ResponseEntity;
import org.springframework.security.core.Authentication;
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.RestController;
import io.swagger.v3.oas.annotations.Operation;
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.Map;

@RestController
@RequestMapping("/api/ai")
@RequiredArgsConstructor
public class AiController {

    private final OpenAiService openAiService;
    private final AiUsageService aiUsageService;

    @PostMapping("/format")
    @Operation(summary = "Format markdown", description = "Format text via AI")
    @ApiResponse(responseCode = "200", description = "Formatted content",
            content = @Content(schema = @Schema(implementation = Map.class)))
    @SecurityRequirement(name = "JWT")
    public ResponseEntity<Map<String, String>> format(@RequestBody Map<String, String> req,
                                                     Authentication auth) {
        String text = req.get("text");
        if (text == null) {
            return ResponseEntity.badRequest().build();
        }
        int limit = aiUsageService.getFormatLimit();
        int used = aiUsageService.getCount(auth.getName());
        if (limit > 0 && used >= limit) {
            return ResponseEntity.status(429).build();
        }
        aiUsageService.incrementAndGetCount(auth.getName());
        return openAiService.formatMarkdown(text)
                .map(t -> ResponseEntity.ok(Map.of("content", t)))
                .orElse(ResponseEntity.status(500).build());
    }
}