package org.springblade.mdm.flow.controller;

import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.tags.Tag;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.flowable.engine.TaskService;
import org.flowable.task.api.Task;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.support.Kv;
import org.springblade.core.tool.utils.DateUtil;
import org.springblade.core.tool.utils.Func;
import org.springblade.core.tool.utils.StringUtil;
import org.springblade.mdm.flow.entity.ApproveRecord;
import org.springblade.mdm.flow.excution.StartDispatcher;
import org.springblade.mdm.flow.service.ApproveRecordService;
import org.springblade.mdm.flow.vo.TaskAssignVO;
import org.springblade.mdm.program.service.ProcessProgRefService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

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

@Slf4j
@RestController
@RequestMapping("/flow/dispatch")
@Tag(name = "派工流程", description = "派工流程")
public class DispatchController {

	@Autowired
	private StartDispatcher dispatcher;
	@Autowired
	private TaskService taskService;
	@Autowired
	private ProcessProgRefService processProgRefService;

	@Autowired
	private ApproveRecordService approveRecordService;
	/**
	 * 新增
	 */
	@PostMapping("/start")
	@Operation(summary = "任务计划(派工流程)", description = "启动派工流程")
	public R<Boolean> start(@RequestBody TaskAssignVO startVO) {
		dispatcher.start(startVO);
		return R.status(true);
	}

	@Operation(summary = "完成任务", description = "流向下一个节点")
	@PostMapping("completeTask")
	public R<Void> completeTask(String taskId, String processInstanceId, String comment,@Parameter(description = "对应的程序id,逗号分隔") String programIds, @Parameter(name = "variables", description = "任务变量") @RequestBody Map<String, Object> variables) {
		// 增加评论
		if (StringUtil.isNoneBlank(processInstanceId, comment)) {
			taskService.addComment(taskId, processInstanceId, comment);
		}
		// 非空判断
		if (Func.isEmpty(variables)) {
			variables = Kv.create();
		}
		if(StringUtils.isNotEmpty(programIds)) {
			processProgRefService.addRelations(processInstanceId,Func.toLongList(programIds));
		}

		//加入审批用户
		variables.put("approveUserNickName",AuthUtil.getNickName());

		if(variables.containsKey("assignee")){
			addApproveRecord(taskId,processInstanceId,comment,programIds,variables);
			//指定了下一步执行人
			taskService.complete(taskId, variables);

			return R.success("流程提交成功");
		}else {
			// 完成任务,给默认用户
			return R.fail("请指定流程下一步处理人");
		}

	}

	void addApproveRecord(String taskId, String processInstanceId, String comment,String programIds,Map<String, Object> variables){
		List<Long> programIdList = new ArrayList<>();
		if(programIds != null){
			programIdList = Func.toLongList(programIds);
		}else{
			programIdList.add(null);
		}
		Task task = taskService.createTaskQuery()
			.taskId(taskId)
			.singleResult();
		for(Long programId : programIdList){
			ApproveRecord rec = new ApproveRecord();
			rec.setTaskName(task.getName());
			rec.setComment(comment);
			rec.setNcProgramId(programId);
			rec.setOperateResult(variables.get("approve")+"");//审批结果
			rec.setOperateTime(DateUtil.now());
			rec.setProcessInstanceId(processInstanceId);
			rec.setUserId(AuthUtil.getUserId());
			rec.setUserNickname(AuthUtil.getNickName());

			approveRecordService.save(rec);
		}

	}
}