OpenIsle/backend/src/main/java/com/openisle/service/AiUsageService.java

package com.openisle.service;

import com.openisle.model.AiFormatUsage;
import com.openisle.model.User;
import com.openisle.repository.AiFormatUsageRepository;
import com.openisle.repository.UserRepository;
import java.time.LocalDate;
import lombok.RequiredArgsConstructor;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

@Service
@RequiredArgsConstructor
public class AiUsageService {

  private final AiFormatUsageRepository usageRepository;
  private final UserRepository userRepository;

  @Value("${app.ai.format-limit:3}")
  private int formatLimit;

  public int getFormatLimit() {
    return formatLimit;
  }

  public void setFormatLimit(int formatLimit) {
    this.formatLimit = formatLimit;
  }

  public int incrementAndGetCount(String username) {
    User user = userRepository
      .findByUsername(username)
      .orElseThrow(() -> new com.openisle.exception.NotFoundException("User not found"));
    LocalDate today = LocalDate.now();
    AiFormatUsage usage = usageRepository
      .findByUserAndUseDate(user, today)
      .orElseGet(() -> {
        AiFormatUsage u = new AiFormatUsage();
        u.setUser(user);
        u.setUseDate(today);
        u.setCount(0);
        return u;
      });
    usage.setCount(usage.getCount() + 1);
    usageRepository.save(usage);
    return usage.getCount();
  }

  public int getCount(String username) {
    User user = userRepository
      .findByUsername(username)
      .orElseThrow(() -> new com.openisle.exception.NotFoundException("User not found"));
    return usageRepository
      .findByUserAndUseDate(user, LocalDate.now())
      .map(AiFormatUsage::getCount)
      .orElse(0);
  }
}