newcpms/master/src/main/java/com/ruoyi/project/invoice/controller/TApproveReserveInvoiceController.java

package com.ruoyi.project.invoice.controller;

import com.ruoyi.common.utils.DateUtils;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.framework.aspectj.lang.annotation.Log;
import com.ruoyi.framework.aspectj.lang.enums.BusinessType;
import com.ruoyi.framework.web.controller.BaseController;
import com.ruoyi.framework.web.domain.AjaxResult;
import com.ruoyi.framework.web.page.TableDataInfo;
import com.ruoyi.project.approve.damain.DevTask;
import com.ruoyi.project.invoice.domain.TApproveReserveInvoice;
import com.ruoyi.project.invoice.domain.TInvoiceBookingworkticket;
import com.ruoyi.project.invoice.domain.TInvoiceWorkcontent;
import com.ruoyi.project.invoice.mapper.TInvoiceWorkcontentMapper;
import com.ruoyi.project.invoice.service.ITApproveReserveInvoiceService;
import com.ruoyi.project.invoice.service.ITInvoiceBookingworkticketService;
import com.ruoyi.project.listener.invoiceApprove.*;
import org.activiti.engine.ProcessEngine;
import org.activiti.engine.ProcessEngines;
import org.activiti.engine.RuntimeService;
import org.activiti.engine.TaskService;
import org.activiti.engine.impl.identity.Authentication;
import org.activiti.engine.runtime.ProcessInstance;
import org.activiti.engine.task.Task;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.*;

/**
 * 预约开票申请Controller
 *
 * @author ruoyi
 * @date 2022-08-29
 */
@RestController
@RequestMapping("/approve/invoice")
public class TApproveReserveInvoiceController extends BaseController {
    @Autowired
    private ITApproveReserveInvoiceService tApproveReserveInvoiceService;
    @Autowired
    private ITInvoiceBookingworkticketService workService;

    @Autowired
    private TaskService taskService;

    @Autowired
    private RuntimeService runtimeService;

    @Autowired
    private TInvoiceWorkcontentMapper invoiceWorkcontentMapper;


    /**
     * 查询预约开票申请列表
     */

    @GetMapping("/list")
    public TableDataInfo list(TApproveReserveInvoice tApproveReserveInvoice) {
        startPage();
        List<TApproveReserveInvoice> list = tApproveReserveInvoiceService.selectTApproveReserveInvoiceList(tApproveReserveInvoice);
        for (TApproveReserveInvoice tApprove : list) {
            String[] ids;
            ids = tApprove.getInvoiceId().split(",");
            List<Object> devList = new ArrayList<>();
            for (String id : ids) {
                TInvoiceBookingworkticket book = workService.getById(id);
                devList.add(book);
            }
            tApprove.setDevList(devList);
        }
        return getDataTable(list);
    }

    /**
     * 导出预约开票申请列表
     */

    @Log(title = "预约开票申请", businessType = BusinessType.EXPORT)
    @GetMapping("/export")
    public AjaxResult export(TApproveReserveInvoice tApproveReserveInvoice) {
        List<TApproveReserveInvoice> list = tApproveReserveInvoiceService.selectTApproveReserveInvoiceList(tApproveReserveInvoice);
        ExcelUtil<TApproveReserveInvoice> util = new ExcelUtil<TApproveReserveInvoice>(TApproveReserveInvoice.class);
        return util.exportExcel(list, "invoice");
    }

    /**
     * 获取预约开票申请详细信息
     */

    @GetMapping(value = "/{id}")
    public AjaxResult getInfo(@PathVariable("id") Long id) {
        TApproveReserveInvoice tApprove = tApproveReserveInvoiceService.selectTApproveReserveInvoiceById(id);
        String[] ids;
        ids = tApprove.getInvoiceId().split(",");
        List<Object> devList = new ArrayList<>();
        for (String inId : ids) {
            TInvoiceBookingworkticket book = workService.getById(inId);
            devList.add(book);
        }
        tApprove.setDevList(devList);
        return AjaxResult.success(tApprove);
    }

    /**
     * 新增预约开票申请
     */

    @Log(title = "预约开票申请", businessType = BusinessType.INSERT)
    @PostMapping
    public AjaxResult add(@RequestBody TApproveReserveInvoice tApproveReserveInvoice) {
        Long userid = getUserId();
        tApproveReserveInvoice.setUserId(userid);
        tApproveReserveInvoice.setApNo(DateUtils.dateTimeNow() + userid);
        Authentication.setAuthenticatedUserId(userid.toString());
        tApproveReserveInvoiceService.insertTApproveReserveInvoice(tApproveReserveInvoice);
        long bussniseeKey = tApproveReserveInvoice.getId();
        String[] ids;
        ids = tApproveReserveInvoice.getInvoiceId().split(",");
        String devSupId = "";
        for (String id : ids) {
            TInvoiceBookingworkticket invoice = workService.getById(id);
            invoice.setStatus(1L);
            workService.updateById(invoice);
            switch (invoice.getWorkArea()) {
                case "310裂解区":
                case "700开关锅炉":
                case "320裂解区":
                case "250管廊-700":
//                    devSupId="20253,20257";
//                    break;
                case "330压缩区":
                case "废水处理":
                case "600废碱":
                case "250管廊-600":
//                    devSupId="20249,20317";
//                    break;
                case "380分离区":
                case "360分离区":
                case "350分离区":
                case "340分离区":
//                    devSupId="20252";
//                    break;
                case "400汽油加氢":
                case "250管廊-400":
//                    devSupId="20248";
//                    break;
                case "500芳烃抽提":
                case "250管廊-500":
//                    devSupId="20244";
//                    break;
                case "D301控制楼":
                case " D201变电所":
                case " 维修厂房":
                case "危废车库":
//                    devSupId="20332,20250,20276";
                    devSupId = "10081,10082";
                    break;
            }
        }
        //开始工作流、监听
        Map<String, Object> variables = new HashMap<>();
        variables.put("applyUser", userid.toString());
        variables.put("yhzgusers", tApproveReserveInvoice.getUserSupId().toString());//用户主管
        variables.put("zzzgusers", devSupId);//装置主管
        if ("1".equals(tApproveReserveInvoice.getIsToday())) {
            variables.put("zzgcsusers", tApproveReserveInvoice.getDevEngineerId().toString());//装置工程师
        }
        if ("1".equals(tApproveReserveInvoice.getIsSpecial())) {
            variables.put("zzjlusers", tApproveReserveInvoice.getManagerconId().toString());//装置经理
        }
        variables.put("yhzgTaskCreateListener", new YhzgTaskCreateListener());//用户主管监听发送邮件
        variables.put("zzzgTaskCreateListener", new ZzzgTaskCreateListener());//装置主管监听发送邮件
        variables.put("zzgcsTaskCreateListener", new ZzgcsTaskCreateListener());//装置工程师监听发送邮件
        variables.put("zzjlTaskCreateListener", new ZzjlTaskCreateListener());//装置经理监听发送邮件
        variables.put("invoiceEndSuccessListener", new InvoiceEndSuccessListener());//流程结束监听发送邮件
        //采用key来启动流程定义并设置流程变量，返回流程实例
        ProcessInstance pi = runtimeService.startProcessInstanceByKey("reserveInvoice", String.valueOf(bussniseeKey), variables);
        logger.info("流程部署id：" + pi.getDeploymentId());
        logger.info("流程定义id：" + pi.getProcessDefinitionId());
        logger.info("流程实例id：" + pi.getProcessInstanceId());
        tApproveReserveInvoice.setProcessId(pi.getProcessInstanceId());
        tApproveReserveInvoiceService.updateTApproveReserveInvoice(tApproveReserveInvoice);
        return AjaxResult.success();
    }

    @Log(title = "预约开票审核处理", businessType = BusinessType.UPDATE)
    @PutMapping("/handle")
    public AjaxResult handle(@RequestBody DevTask devTask) {
        //流程审批意见
        String symbol = "";
        if (devTask.getComment() != "") {
            symbol = "，";
        }
        //使用任务服务完成任务(提交任务)
        String taskId = devTask.getTaskId();
        // 使用任务id,获取任务对象，获取流程实例id
        Task task = taskService.createTaskQuery().taskId(taskId).singleResult();
        //利用任务对象，获取流程实例id
        String processInstancesId = task.getProcessInstanceId();
        TApproveReserveInvoice invoice = tApproveReserveInvoiceService.selectTApproveReserveInvoiceById(Long.parseLong(devTask.getBusinessKey()));
        if (devTask.getCondition().equals("1")) {
            devTask.setComment("通过" + symbol + devTask.getComment());
            invoice.setStatus(2L);
            invoice.setEnddate(new Date());
        } else if (devTask.getCondition().equals("0")) {
            devTask.setComment("未通过" + symbol + devTask.getComment());
            invoice.setStatus(3L);
            invoice.setEnddate(new Date());
        }
        Map<String, Object> param = new HashMap<>();

        TInvoiceBookingworkticket invoiceBook = workService.getById(invoice.getInvoiceId());
//        if (devTask.getCondition().equals("1") && task.getName().equals("装置经理签字确认")) {
//            invoiceBook.setStatus(3L);
//        } else if (devTask.getCondition().equals("0")) {
//            invoiceBook.setStatus(5L);
//        }
//        workService.updateById(invoiceBook);
        if (DateUtils.dateTime(invoiceBook.getCreatedate()).equals(DateUtils.dateTime(invoiceBook.getWorkStartTime()))) {
            param.put("istoday", "1");
        } else {
            param.put("istoday", "0");
        }
        List<TInvoiceWorkcontent> workcontents = invoiceWorkcontentMapper.selectTInvoiceWorkcontentBybookingticketId(Long.valueOf(invoice.getInvoiceId()));
        for (TInvoiceWorkcontent workcontent : workcontents) {
            if (workcontent.getWorkType() == 1 || workcontent.getWorkType() == 2 || workcontent.getWorkType() == 3 || workcontent.getWorkType() == 15) {
                param.put("isspecial", "1");
                break;
            } else {
                param.put("isspecial", "0");
            }
        }
        param.put("condition", devTask.getCondition());
        ProcessEngine processEngine = ProcessEngines.getDefaultProcessEngine();
        TaskService taskService = processEngine.getTaskService();
        //认领任务
        processEngine.getTaskService()
                .claim(taskId, getUserId().toString());
        taskService.addComment(taskId, processInstancesId, devTask.getComment());
        taskService.complete(taskId, param);
        tApproveReserveInvoiceService.updateTApproveReserveInvoice(invoice);
        return AjaxResult.success();
    }

    /**
     * 修改预约开票申请
     */

    @Log(title = "预约开票申请", businessType = BusinessType.UPDATE)
    @PutMapping
    public AjaxResult edit(@RequestBody TApproveReserveInvoice tApproveReserveInvoice) {
        return toAjax(tApproveReserveInvoiceService.updateTApproveReserveInvoice(tApproveReserveInvoice));
    }

    /**
     * 删除预约开票申请
     */

    @Log(title = "预约开票申请", businessType = BusinessType.DELETE)
    @DeleteMapping("/{ids}")
    public AjaxResult remove(@PathVariable Long[] ids) {
        return toAjax(tApproveReserveInvoiceService.deleteTApproveReserveInvoiceByIds(ids));
    }
}