package Utils;

import Pojo.TaskInfo;
import org.checkerframework.checker.units.qual.A;

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

public class TreeUtil {
    public  static Map<String,Object> mapArray = new LinkedHashMap<String, Object>();
    public List<TaskInfo> taskInfos;
    public List<Object> taskList = new ArrayList<>();

    public  List<Object> generateTaskTree(List<TaskInfo> tasks){
        this.taskInfos = tasks;
        for(TaskInfo task:tasks){
            Map<String,Object> mapArr =  new LinkedHashMap<String, Object>();
            if(task.getParentId() == 0){
                setTreeMap(mapArr,task);
                taskList.add(mapArr);
            }
        }
        return taskList;
    }

    public List<?> generateChildTask(Integer id){
        List<Object>  childTaskList =  new ArrayList<Object>();
        for(TaskInfo task:taskInfos){
            Map<String,Object> childMapArr =  new LinkedHashMap<String, Object>();
            if(task.getParentId() == id){
                setTreeMap(childMapArr,task);
                childTaskList.add(childMapArr);
            }
        }
        return childTaskList;
    }

    public void setTreeMap(Map<String,Object> mapArr,TaskInfo task){
        mapArr.put("rootId",task.getRootId());
        mapArr.put("taskId",task.getTaskId());
        mapArr.put("parentId",task.getParentId());
        mapArr.put("taskName",task.getTaskName());
        mapArr.put("childTask",generateChildTask(task.getTaskId()));
    }
}