// platform/service/ScheduledPublishingTaskService.java
package com.aigeo.platform.service;

import com.aigeo.platform.entity.ScheduledPublishingTask;
import com.aigeo.platform.entity.PublishingTaskStatus;
import com.aigeo.platform.repository.ScheduledPublishingTaskRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

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

@Service
public class ScheduledPublishingTaskService {
    
    @Autowired
    private ScheduledPublishingTaskRepository scheduledPublishingTaskRepository;
    
    public List<ScheduledPublishingTask> getAllTasks() {
        return scheduledPublishingTaskRepository.findAll();
    }
    
    public List<ScheduledPublishingTask> getTasksByCompanyId(Integer companyId) {
        return scheduledPublishingTaskRepository.findByCompanyId(companyId);
    }
    
    public List<ScheduledPublishingTask> getTasksByStatus(PublishingTaskStatus status) {
        return scheduledPublishingTaskRepository.findByStatus(status);
    }
    
    public List<ScheduledPublishingTask> getTasksByCompanyIdAndStatus(Integer companyId, PublishingTaskStatus status) {
        return scheduledPublishingTaskRepository.findByCompanyIdAndStatus(companyId, status);
    }
    
    public List<ScheduledPublishingTask> getPendingTasksBeforeTime(Date scheduleTime) {
        return scheduledPublishingTaskRepository.findByScheduleTimeBeforeAndStatus(scheduleTime, PublishingTaskStatus.SCHEDULED);
    }
    
    public List<ScheduledPublishingTask> getTasksByPlatformConfigId(Integer platformConfigId) {
        return scheduledPublishingTaskRepository.findByPlatformConfigId(platformConfigId);
    }
    
    public Optional<ScheduledPublishingTask> getTaskById(Integer id) {
        return scheduledPublishingTaskRepository.findById(id);
    }
    
    public ScheduledPublishingTask saveTask(ScheduledPublishingTask task) {
        return scheduledPublishingTaskRepository.save(task);
    }
    
    public void deleteTask(Integer id) {
        scheduledPublishingTaskRepository.deleteById(id);
    }
}