package com.ruoyi.task.domain.convert;

import com.ruoyi.task.api.enums.ExecuteTypeEnum;
import com.ruoyi.task.api.enums.StatusEnum;
import com.ruoyi.task.domain.model.Task;
import com.ruoyi.task.mapper.entity.TaskInfoEntity;

/**
 * 任务转换类
 *
 * @author ruoyi
 * @date 2026-03-05
 */
public class TaskConvert {

    /**
     * 将实体转换为领域模型
     *
     * @param entity 任务实体
     * @return 任务领域模型
     */
    public static Task toModel(TaskInfoEntity entity) {
        if (entity == null) {
            return null;
        }

        Task model = new Task();
        model.setId(entity.getId());
        model.setPlanId(entity.getPlanId());
        model.setTaskName(entity.getTaskName());
        model.setTaskCategory(entity.getTaskCategory());
        model.setTaskType(entity.getTaskType());
        model.setExecuteType(ExecuteTypeEnum.getByCode(entity.getExecuteType()));
        model.setRouteId(entity.getRouteId());
        model.setUavId(entity.getUavId());
        model.setStatus(StatusEnum.getByCode(entity.getStatus()));
        model.setStartTime(entity.getStartTime());
        model.setEndTime(entity.getEndTime());
        model.setDescription(entity.getDescription());
        model.setCreateBy(entity.getCreateBy());
        model.setCreateTime(entity.getCreateTime());
        model.setUpdateBy(entity.getUpdateBy());
        model.setUpdateTime(entity.getUpdateTime());
        model.setRemark(entity.getRemark());

        return model;
    }

    /**
     * 将领域模型转换为实体
     *
     * @param model 任务领域模型
     * @return 任务实体
     */
    public static TaskInfoEntity toEntity(Task model) {
        if (model == null) {
            return null;
        }

        TaskInfoEntity entity = new TaskInfoEntity();
        entity.setId(model.getId());
        entity.setPlanId(model.getPlanId());
        entity.setTaskName(model.getTaskName());
        entity.setTaskCategory(model.getTaskCategory());
        entity.setTaskType(model.getTaskType());
        entity.setExecuteType(model.getExecuteType() != null ? model.getExecuteType().getCode() : null);
        entity.setRouteId(model.getRouteId());
        entity.setUavId(model.getUavId());
        entity.setStatus(model.getStatus() != null ? model.getStatus().getCode() : null);
        entity.setStartTime(model.getStartTime());
        entity.setEndTime(model.getEndTime());
        entity.setDescription(model.getDescription());
        entity.setCreateBy(model.getCreateBy());
        entity.setCreateTime(model.getCreateTime());
        entity.setUpdateBy(model.getUpdateBy());
        entity.setUpdateTime(model.getUpdateTime());
        entity.setRemark(model.getRemark());

        return entity;
    }
}