// statistics/service/CompanyUsageStatService.java
package com.aigeo.statistics.service;

import com.aigeo.statistics.entity.CompanyUsageStat;
import com.aigeo.statistics.repository.CompanyUsageStatRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.time.LocalDate;
import java.util.Date;
import java.util.List;
import java.util.Optional;

@Service
public class CompanyUsageStatService {
    
    @Autowired
    private CompanyUsageStatRepository companyUsageStatRepository;
    
    public List<CompanyUsageStat> getAllUsageStats() {
        return companyUsageStatRepository.findAll();
    }
    
    public List<CompanyUsageStat> getUsageStatsByCompanyId(Integer companyId) {
        return companyUsageStatRepository.findByCompanyId(companyId);
    }
    
    public Optional<CompanyUsageStat> getUsageStatByCompanyIdAndStatDate(Integer companyId, Date statDate) {
        return companyUsageStatRepository.findByCompanyIdAndStatDate(companyId, statDate);
    }
    
    public List<CompanyUsageStat> getUsageStatsByCompanyIdAndDateRange(Integer companyId, LocalDate startDate, Date endDate) {
        return companyUsageStatRepository.findByCompanyIdAndDateRange(companyId, startDate, endDate);
    }
    
    public Long getTotalArticleCountByCompanyId(Integer companyId) {
        return companyUsageStatRepository.getTotalArticleCountByCompanyId(companyId);
    }
    
    public Long getTotalAiTokenUsageByCompanyId(Integer companyId) {
        return companyUsageStatRepository.getTotalAiTokenUsageByCompanyId(companyId);
    }
    
    public Long getTotalWebsiteCountByCompanyId(Integer companyId) {
        return companyUsageStatRepository.getTotalWebsiteCountByCompanyId(companyId);
    }
    
    public Optional<CompanyUsageStat> getUsageStatById(Integer id) {
        return companyUsageStatRepository.findById(id);
    }
    
    public CompanyUsageStat saveUsageStat(CompanyUsageStat usageStat) {
        return companyUsageStatRepository.save(usageStat);
    }
    
    public void deleteUsageStat(Integer id) {
        companyUsageStatRepository.deleteById(id);
    }
}