// website/service/WebsiteGenerationTaskService.java
package com.aigeo.website.service;

import com.aigeo.website.entity.WebsiteGenerationTask;
import com.aigeo.common.enums.TaskStatus;
import com.aigeo.website.repository.WebsiteGenerationTaskRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

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

@Service
public class WebsiteGenerationTaskService {
    
    @Autowired
    private WebsiteGenerationTaskRepository websiteGenerationTaskRepository;
    
    public List<WebsiteGenerationTask> getTasksByProjectId(Integer projectId) {
        return websiteGenerationTaskRepository.findByProjectId(projectId);
    }
    
    public List<WebsiteGenerationTask> getTasksByUserId(Integer userId) {
        return websiteGenerationTaskRepository.findByUserId(userId);
    }
    
    public List<WebsiteGenerationTask> getTasksByParentTaskId(Integer parentTaskId) {
        return websiteGenerationTaskRepository.findByParentTaskId(parentTaskId);
    }
    
    public List<WebsiteGenerationTask> getTasksByStatus(TaskStatus status) {
        return websiteGenerationTaskRepository.findByStatus(status);
    }
    
    public List<WebsiteGenerationTask> getTasksByProjectIdAndStatus(Integer projectId, TaskStatus status) {
        return websiteGenerationTaskRepository.findByProjectIdAndStatus(projectId, status);
    }
    
    public Optional<WebsiteGenerationTask> getTaskById(Integer id) {
        return websiteGenerationTaskRepository.findById(id);
    }
    
    public WebsiteGenerationTask saveTask(WebsiteGenerationTask task) {
        return websiteGenerationTaskRepository.save(task);
    }
    
    public void deleteTask(Integer id) {
        websiteGenerationTaskRepository.deleteById(id);
    }
}