package com.ruoyi.task.domain.convert;

import com.ruoyi.task.api.enums.CycleTypeEnum;
import com.ruoyi.task.api.enums.ExecuteTypeEnum;
import com.ruoyi.task.api.enums.PlanTypeEnum;
import com.ruoyi.task.api.enums.StatusEnum;
import com.ruoyi.task.domain.model.TaskPlan;
import com.ruoyi.task.mapper.entity.TaskPlanEntity;

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

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

        TaskPlan model = new TaskPlan();
        model.setId(entity.getId());
        model.setPlanName(entity.getPlanName());
        model.setPlanType(PlanTypeEnum.getByCode(entity.getPlanType()));
        model.setExecuteType(ExecuteTypeEnum.getByCode(entity.getExecuteType()));
        model.setCycleType(CycleTypeEnum.getByCode(entity.getCycleType()));
        model.setCycleValue(entity.getCycleValue());
        model.setStartDate(entity.getStartDate());
        model.setEndDate(entity.getEndDate());
        model.setExecuteTime(entity.getExecuteTime());
        model.setDuration(entity.getDuration());
        model.setRouteId(entity.getRouteId());
        model.setUavId(entity.getUavId());
        model.setStatus(StatusEnum.getByCode(entity.getStatus()));
        model.setDescription(entity.getDescription());
        model.setCreateBy(entity.getCreateBy());
        model.setCreateTime(entity.getCreateTime());
        model.setUpdateBy(entity.getUpdateBy());
        model.setUpdateTime(entity.getUpdateTime());
        model.setRemark(entity.getRemark());
        model.setRouteUrl(entity.getRouteUrl());

        return model;
    }

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

        TaskPlanEntity entity = new TaskPlanEntity();
        entity.setId(model.getId());
        entity.setPlanName(model.getPlanName());
        entity.setPlanType(model.getPlanType() != null ? model.getPlanType().getCode() : null);
        entity.setExecuteType(model.getExecuteType() != null ? model.getExecuteType().getCode() : null);
        entity.setCycleType(model.getCycleType() != null ? model.getCycleType().getCode() : null);
        entity.setCycleValue(model.getCycleValue());
        entity.setStartDate(model.getStartDate());
        entity.setEndDate(model.getEndDate());
        entity.setExecuteTime(model.getExecuteTime());
        entity.setDuration(model.getDuration());
        entity.setRouteId(model.getRouteId());
        entity.setUavId(model.getUavId());
        entity.setStatus(model.getStatus() != null ? model.getStatus().getCode() : null);
        entity.setDescription(model.getDescription());
        entity.setCreateBy(model.getCreateBy());
        entity.setCreateTime(model.getCreateTime());
        entity.setUpdateBy(model.getUpdateBy());
        entity.setUpdateTime(model.getUpdateTime());
        entity.setRemark(model.getRemark());
        entity.setRouteUrl(model.getRouteUrl());

        return entity;
    }
}