content: package com.ruoyi.web.controller.flowable;
  
  
  import cn.hutool.core.bean.BeanUtil;
  import com.ruoyi.common.annotation.Log;
  import com.ruoyi.common.core.domain.AjaxResult;
  import com.ruoyi.common.core.domain.R;
  import com.ruoyi.common.enums.BusinessType;
  import com.ruoyi.common.utils.poi.ExcelUtil;
  import com.ruoyi.flowable.base.BaseController;
  import com.ruoyi.flowable.domain.WfCategory;
  import com.ruoyi.flowable.domain.bo.WfModelBo;
  import com.ruoyi.flowable.domain.vo.WfCategoryVo;
  import com.ruoyi.flowable.domain.vo.WfModelExportVo;
  import com.ruoyi.flowable.domain.vo.WfModelVo;
  import com.ruoyi.flowable.page.PageQuery;
  import com.ruoyi.flowable.page.TableDataInfo;
  import com.ruoyi.flowable.service.IWfCategoryService;
  import com.ruoyi.flowable.service.IWfModelService;
  import lombok.RequiredArgsConstructor;
  import lombok.extern.slf4j.Slf4j;
  import org.springframework.security.access.prepost.PreAuthorize;
  import org.springframework.web.bind.annotation.*;
  
  import javax.servlet.http.HttpServletResponse;
  import javax.validation.constraints.NotEmpty;
  import javax.validation.constraints.NotNull;
  import java.util.Arrays;
  import java.util.List;
  import java.util.Map;
  import java.util.stream.Collectors;
  
  /**
   * 工作流流程模型管理
   *
   * @author fengcheng
   */
  @Slf4j
  @RequiredArgsConstructor
  @RestController
  @RequestMapping("/flowable/workflow/model")
  public class WfModelController extends BaseController {
  
      /**
       * 流程模型服务
       */
      private final IWfModelService modelService;
  
      /**
       * 流程分类服务
       */
      private final IWfCategoryService categoryService;
  
      /**
       * 查询流程模型列表
       *
       * @param modelBo 流程模型对象
       */
      @PreAuthorize("@ss.hasPermi('workflow:model:list')")
      @GetMapping("/list")
      public R<List<WfModelVo>> list(WfModelBo modelBo) {
          List<WfModelVo> list = modelService.selectList(modelBo);
          return R.ok(list);
      }
  
      /**
       * 查询流程模型列表
       *
       * @param modelBo   流程模型对象
       * @param pageQuery 分页参数
       */
      @PreAuthorize("@ss.hasPermi('workflow:model:list')")
      @GetMapping("/historyList")
      public TableDataInfo<WfModelVo> historyList(WfModelBo modelBo, PageQuery pageQuery) {
          return modelService.historyList(modelBo, pageQuery);
      }
  
      /**
       * 获取流程模型详细信息
       *
       * @param modelId 模型主键
       */
      @PreAuthorize("@ss.hasPermi('workflow:model:query')")
      @GetMapping(value = "/{modelId}")
      public R<WfModelVo> getInfo(@NotNull(message = "主键不能为空") @PathVariable("modelId") String modelId) {
          return R.ok(modelService.getModel(modelId));
      }
  
      /**
       * 获取流程表单详细信息
       *
       * @param modelId 模型主键
       */
      @PreAuthorize("@ss.hasPermi('workflow:model:query')")
      @GetMapping(value = "/bpmnXml/{modelId}")
      public R<String> getBpmnXml(@NotNull(message = "主键不能为空") @PathVariable("modelId") String modelId) {
          return R.ok(modelService.queryBpmnXmlById(modelId), "操作成功");
      }
  
      /**
       * 新增流程模型
       */
      @PreAuthorize("@ss.hasPermi('workflow:model:add')")
      @Log(title = "流程模型", businessType = BusinessType.INSERT)
      @PostMapping
      public R<String> add(@RequestBody WfModelBo modelBo) {
          return R.ok(modelService.insertModel(modelBo));
      }
  
      /**
       * 修改流程模型
       */
      @PreAuthorize("@ss.hasPermi('workflow:model:edit')")
      @Log(title = "流程模型", businessType = BusinessType.UPDATE)
      @PutMapping
      public R<Void> edit(@RequestBody WfModelBo modelBo) {
          modelService.updateModel(modelBo);
          return R.ok();
      }
  
      /**
       * 保存流程模型
       */
      @PreAuthorize("@ss.hasPermi('workflow:model:save')")
      @Log(title = "保存流程模型", businessType = BusinessType.INSERT)
      @PostMapping("/save")
      public R<String> save(@RequestBody WfModelBo modelBo) {
          modelService.saveModel(modelBo);
          return R.ok();
      }
  
      /**
       * 设为最新流程模型
       *
       * @param modelId
       * @return
       */
      @PreAuthorize("@ss.hasPermi('workflow:model:save')")
      @Log(title = "设为最新流程模型", businessType = BusinessType.INSERT)
      @PostMapping("/latest")
      public R<?> latest(@RequestParam String modelId) {
          modelService.latestModel(modelId);
          return R.ok();
      }
  
      /**
       * 删除流程模型
       *
       * @param modelIds 流程模型主键串
       */
      @PreAuthorize("@ss.hasPermi('workflow:model:remove')")
      @Log(title = "删除流程模型", businessType = BusinessType.DELETE)
      @DeleteMapping("/{modelIds}")
      public R<String> remove(@NotEmpty(message = "主键不能为空") @PathVariable String[] modelIds) {
          modelService.deleteByIds(Arrays.asList(modelIds));
          return R.ok();
      }
  
      /**
       * 部署流程模型
       *
       * @param modelId 流程模型主键
       */
      @PreAuthorize("@ss.hasPermi('workflow:model:deploy')")
      @Log(title = "部署流程模型", businessType = BusinessType.INSERT)
      @PostMapping("/deploy")
      public AjaxResult deployModel(@RequestParam String modelId) {
          return toAjax(modelService.deployModel(modelId));
      }
  
      /**
       * 导出流程模型数据
       */
      @Log(title = "导出流程模型数据", businessType = BusinessType.EXPORT)
      @PreAuthorize("@ss.hasPermi('workflow:model:export')")
      @PostMapping("/export")
      public void export(WfModelBo modelBo, HttpServletResponse response) {
          List<WfModelVo> list = modelService.list(modelBo);
          List<WfModelExportVo> listVo = BeanUtil.copyToList(list, WfModelExportVo.class);
          List<WfCategoryVo> categoryVos = categoryService.queryList(new WfCategory());
          Map<String, String> categoryMap = categoryVos.stream()
                  .collect(Collectors.toMap(WfCategoryVo::getCode, WfCategoryVo::getCategoryName));
          for (WfModelExportVo exportVo : listVo) {
              exportVo.setCategoryName(categoryMap.get(exportVo.getCategory()));
          }
          ExcelUtil<WfModelExportVo> util = new ExcelUtil<WfModelExportVo>(WfModelExportVo.class);
          util.exportExcel(response, listVo, "流程模型数据");
      }
  }
  