package org.activiti.demo.engine;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Collections;
import java.util.List;
import java.util.Map;

import org.activiti.bpmn.model.BpmnModel;
import org.activiti.engine.repository.ProcessDefinition;
import org.activiti.engine.runtime.ProcessInstance;
import org.activiti.engine.task.Task;
import org.activiti.image.impl.DefaultProcessDiagramGenerator;

public class CompleProcessEngine extends ActivitiEngine {
	public CompleProcessEngine() {
		super();
	}

	private static Map<String, Object> proMap = null;

	public static void main(String[] args) throws Exception {
		CompleProcessEngine processEngine = new CompleProcessEngine();
		/************************** 启动流程引擎begin *****************************/
		// processEngine.deploymentInstance("D:\\workspace\\activiti-demo\\src\\main\\resources\\gaojiqingjiaProcess.bpmn20.xml");
		/************************** 启动流程引擎end *******************************/
		/************************** 启动流程实例start *****************************/
		// proMap = new HashMap<String, Object>();
		// proMap.put("leaveUser", "zhangsan1");
		// // 启动流程
		// ProcessInstance processInstance =
		// processEngine.startInstance("leaveProcess", proMap);
		/************************** 启动流程实例end *******************************/
		/************************** 流程任务start ********************************/
		// // 查询任务
		// Task task =
		// processEngine.queryByExecutionIdSingle(processInstance.getProcessInstanceId());
		// proMap = new HashMap<String, Object>();
		// proMap.put("day", 3);
		// // 完成任务
		// processEngine.completeUserTask(task.getId(), proMap);
		/************************** 流程任务end **********************************/
		/************************** 获取下一节点审批人start ************************/
		// // 获取下一节点的审批人
		// String nextPerson =
		// processEngine.getNextAppersonal(task.getExecutionId());
		// System.out.println("下一节点审批人是:" + nextPerson);
		/************************** 获取下一节点审批人end *************************/
		/************************** 查询用户待办 start ***************************/
		List<Task> taskList = processEngine.queryUserTaskList("hrRecord");
		System.out.println("用户共有待办:" + taskList.size() + "条");
		for (Task task : taskList) {
			System.out.println("用户  任务ID[" + task.getId() + "] 流程ID[" + task.getExecutionId() + "]");
			// 完成用户下所有的任务
			processEngine.completeUserTask(task.getId(), null);
		}
		// /*************************** 查询用户待办 end ***************************/
		// /*************************** 查看流程是否结束 start **********************/
		boolean processIsEnd = processEngine.queryProcessIsEnd("5");
		System.out.println(processIsEnd);
		/*************************** 查看流程是否结束 end ************************/
		processEngine.outProcessPic(processEngine.getDiagram("5"));
	}

	/**
	 * 显示流程图(仅显示部署的图片)
	 * 
	 * @return
	 * @throws Exception
	 */
	public InputStream getProcessPic(String procDefId) throws Exception {
		ProcessDefinition procDef = repositoryService.createProcessDefinitionQuery().processDefinitionId(procDefId).singleResult();
		String diagramResourceName = procDef.getDiagramResourceName();
		InputStream inputStream = repositoryService.getResourceAsStream(procDef.getDeploymentId(), diagramResourceName);
		return inputStream;
	}

	/**
	 * 查询当前流程流程图(当前流程高亮)
	 * 
	 * @param processInstanceId
	 * @return
	 */
	public InputStream getDiagram(String processInstanceId) {
		// 查询流程实例
		ProcessInstance pi = runtimeService.createProcessInstanceQuery().processInstanceId(processInstanceId).singleResult();
		BpmnModel bpmnModel = repositoryService.getBpmnModel(pi.getProcessDefinitionId());
		// 得到正在执行的环节
		List<String> activeIds = runtimeService.getActiveActivityIds(pi.getId());
		InputStream is = new DefaultProcessDiagramGenerator().generateDiagram(bpmnModel, "png", activeIds, Collections.<String> emptyList(), processEngine.getProcessEngineConfiguration()
				.getActivityFontName(), processEngine.getProcessEngineConfiguration().getLabelFontName(), null, 1.0);
		return is;
	}

	/**
	 * 输出流程图片
	 * 
	 * @param inputStream
	 * @throws IOException
	 */
	public void outProcessPic(InputStream inputStream) throws IOException {
		ByteArrayOutputStream outStream = new ByteArrayOutputStream();
		// 创建一个Buffer字符串
		byte[] buffer = new byte[1024];
		// 每次读取的字符串长度，如果为-1，代表全部读取完毕
		int len = 0;
		// 使用一个输入流从buffer里把数据读取出来
		while ((len = inputStream.read(buffer)) != -1) {
			// 用输出流往buffer里写入数据，中间参数代表从哪个位置开始读，len代表读取的长度
			outStream.write(buffer, 0, len);
		}
		// 关闭输入流
		inputStream.close();
		// 把outStream里的数据写入内存
		// 得到图片的二进制数据，以二进制封装得到数据，具有通用性
		byte[] data = outStream.toByteArray();
		// new一个文件对象用来保存图片，默认保存当前工程根目录
		File imageFile = new File("D:\\myprocess.jpg");
		// 创建输出流
		FileOutputStream fileOutputStream = new FileOutputStream(imageFile);
		// 写入数据
		fileOutputStream.write(data);
	}
}