package com.ruoyi.task.service.convert;

import com.ruoyi.task.domain.model.Task;
import com.ruoyi.task.service.dto.TaskDTO;

import java.util.List;
import java.util.stream.Collectors;

public class TaskDTOConvert {

    public static TaskDTO toDTO(Task task) {
        if (task == null) {
            return null;
        }

        TaskDTO dto = new TaskDTO();
        dto.setId(task.getId());
        dto.setTaskName(task.getTaskName());
        dto.setPlanId(task.getPlanId());
        dto.setTaskCategory(task.getTaskCategory());
        dto.setTaskType(task.getTaskType());
        dto.setExecuteType(task.getExecuteType());
        dto.setRouteId(task.getRouteId());
        dto.setUavId(task.getUavId());
        dto.setStartTime(task.getStartTime());
        dto.setEndTime(task.getEndTime());
        dto.setStatus(task.getStatus());
        dto.setDescription(task.getDescription());
        dto.setRemark(task.getRemark());

        return dto;
    }

    public static Task toDomain(TaskDTO dto) {
        if (dto == null) {
            return null;
        }

        Task task = new Task();
        task.setId(dto.getId());
        task.setTaskName(dto.getTaskName());
        task.setPlanId(dto.getPlanId());
        task.setTaskCategory(dto.getTaskCategory());
        task.setTaskType(dto.getTaskType());
        task.setExecuteType(dto.getExecuteType());
        task.setRouteId(dto.getRouteId());
        task.setUavId(dto.getUavId());
        task.setStartTime(dto.getStartTime());
        task.setEndTime(dto.getEndTime());
        task.setStatus(dto.getStatus());
        task.setDescription(dto.getDescription());
        task.setRemark(dto.getRemark());

        return task;
    }

    public static List<TaskDTO> toDTOList(List<Task> tasks) {
        return tasks.stream()
                .map(TaskDTOConvert::toDTO)
                .collect(Collectors.toList());
    }

    public static List<Task> toDomainList(List<TaskDTO> dtos) {
        return dtos.stream()
                .map(TaskDTOConvert::toDomain)
                .collect(Collectors.toList());
    }
}