feat: limit daily AI formatting usage

src/main/java/com/openisle/controller/AdminConfigController.java

@@ -4,6 +4,7 @@ import com.openisle.model.PasswordStrength;
 import com.openisle.model.PublishMode;
 import com.openisle.service.PasswordValidator;
 import com.openisle.service.PostService;
+import com.openisle.service.AiUsageService;
 import lombok.Data;
 import lombok.RequiredArgsConstructor;
 import org.springframework.web.bind.annotation.*;
@@ -14,12 +15,14 @@ import org.springframework.web.bind.annotation.*;
 public class AdminConfigController {
     private final PostService postService;
     private final PasswordValidator passwordValidator;
+    private final AiUsageService aiUsageService;
 
     @GetMapping
     public ConfigDto getConfig() {
         ConfigDto dto = new ConfigDto();
         dto.setPublishMode(postService.getPublishMode());
         dto.setPasswordStrength(passwordValidator.getStrength());
+        dto.setAiFormatLimit(aiUsageService.getFormatLimit());
         return dto;
     }
 
@@ -31,6 +34,9 @@ public class AdminConfigController {
         if (dto.getPasswordStrength() != null) {
             passwordValidator.setStrength(dto.getPasswordStrength());
         }
+        if (dto.getAiFormatLimit() != null) {
+            aiUsageService.setFormatLimit(dto.getAiFormatLimit());
+        }
         return getConfig();
     }
 
@@ -38,5 +44,6 @@ public class AdminConfigController {
     public static class ConfigDto {
         private PublishMode publishMode;
         private PasswordStrength passwordStrength;
+        private Integer aiFormatLimit;
     }
 }

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

@@ -1,8 +1,10 @@
 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;
@@ -16,13 +18,21 @@ import java.util.Map;
 public class AiController {
 
     private final OpenAiService openAiService;
+    private final AiUsageService aiUsageService;
 
     @PostMapping("/format")
-    public ResponseEntity<Map<String, String>> format(@RequestBody Map<String, String> req) {
+    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());

src/main/java/com/openisle/controller/ConfigController.java

@@ -25,6 +25,9 @@ public class ConfigController {
     @Value("${app.captcha.comment-enabled:false}")
     private boolean commentCaptchaEnabled;
 
+    @Value("${app.ai.format-limit:3}")
+    private int aiFormatLimit;
+
     @GetMapping("/config")
     public ConfigResponse getConfig() {
         ConfigResponse resp = new ConfigResponse();
@@ -33,6 +36,7 @@ public class ConfigController {
         resp.setLoginCaptchaEnabled(loginCaptchaEnabled);
         resp.setPostCaptchaEnabled(postCaptchaEnabled);
         resp.setCommentCaptchaEnabled(commentCaptchaEnabled);
+        resp.setAiFormatLimit(aiFormatLimit);
         return resp;
     }
 
@@ -43,5 +47,6 @@ public class ConfigController {
         private boolean loginCaptchaEnabled;
         private boolean postCaptchaEnabled;
         private boolean commentCaptchaEnabled;
+        private int aiFormatLimit;
     }
 }