package com.ruoyi.project.dc.securityDocuments.controller; import com.alibaba.fastjson.JSON; import com.ruoyi.common.constant.DualControlConstants; import com.ruoyi.common.utils.StringUtils; 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.config.RuoYiConfig; 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.dc.securityDocuments.domain.SecurityDocuments; import com.ruoyi.project.dc.securityDocuments.service.ISecurityDocumentsService; import com.ruoyi.project.enumerate.DcAttachmentTypeEnum; import com.ruoyi.project.system.attachment.domain.Attachment; import com.ruoyi.project.system.attachment.service.IAttachmentService; import com.ruoyi.project.system.user.domain.User; import org.apache.shiro.authz.annotation.RequiresPermissions; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Controller; import org.springframework.ui.ModelMap; import org.springframework.web.bind.annotation.*; import java.util.Date; import java.util.List; /** * 安全体系文件Controller * * @author wm * @date 2020-12-09 */ @Controller @RequestMapping("/dc/securityDocuments") public class SecurityDocumentsController extends BaseController { private String prefix = "dc/securityDocuments"; @Autowired private ISecurityDocumentsService securityDocumentsService; @Autowired private IAttachmentService attachmentService; @RequiresPermissions("dc:securityDocuments:view") @GetMapping() public String securityDocuments() { return prefix + "/securityDocuments"; } /** * 查询安全体系文件列表 */ @RequiresPermissions("dc:securityDocuments:list") @PostMapping("/list") @ResponseBody public TableDataInfo list(SecurityDocuments securityDocuments) { User user = getSysUser(); securityDocuments.setCompanyId(user.getCompanyId()); startPage(); List list = securityDocumentsService.selectSecurityDocumentsList(securityDocuments); return getDataTable(list); } /** * 导出安全体系文件列表 */ @RequiresPermissions("dc:securityDocuments:export") @Log(title = "安全体系文件", businessType = BusinessType.EXPORT) @PostMapping("/export") @ResponseBody public AjaxResult export(SecurityDocuments securityDocuments) { User user = getSysUser(); securityDocuments.setCompanyId(user.getCompanyId()); List list = securityDocumentsService.selectSecurityDocumentsList(securityDocuments); ExcelUtil util = new ExcelUtil(SecurityDocuments.class); return util.exportExcel(list, "securityDocuments"); } /** * 新增安全体系文件 */ @GetMapping("/add") public String add() { return prefix + "/add"; } /** * 新增保存安全体系文件 */ @RequiresPermissions("dc:securityDocuments:add") @Log(title = "安全体系文件", businessType = BusinessType.INSERT) @PostMapping("/add") @ResponseBody public AjaxResult addSave(SecurityDocuments securityDocuments) { User user = getSysUser(); securityDocuments.setCompanyId(user.getCompanyId()); int addNum = securityDocumentsService.insertSecurityDocuments(securityDocuments); //处理新增的附件 String ownedFile = securityDocuments.getOwnedFile(); if(StringUtils.isNotEmpty(ownedFile)&&addNum>0) { List attachmentList = JSON.parseArray(ownedFile, Attachment.class); if (attachmentList.size() > 0) { for (Attachment attachment : attachmentList) { attachment.setFileJoinId(securityDocuments.getDocumentsId()); attachment.setAttachmentType(DcAttachmentTypeEnum.Security_Documents.getType()); attachment.setCompanyId(user.getCompanyId()); attachment.setCreateBy(user.getUserName()); attachment.setCreateUserId(user.getUserId()); attachment.setCreateTime(new Date()); attachmentService.insertAttachment(attachment); } } } return toAjax(addNum); } /** * 修改安全体系文件 */ @GetMapping("/detail/{documentsId}") public String detail(@PathVariable("documentsId") Long documentsId, ModelMap mmap) { SecurityDocuments securityDocuments = securityDocumentsService.selectSecurityDocumentsById(documentsId); mmap.put("securityDocuments", securityDocuments); mmap.put("type", securityDocuments.getSecurityType()); Attachment attachment = new Attachment(); attachment.setFileJoinId(documentsId); attachment.setAttachmentType(DcAttachmentTypeEnum.Security_Documents.getType()); List attachmentList = attachmentService.selectAttachmentList(attachment); mmap.put("fileList",attachmentList); mmap.put("serviceDomainName", DualControlConstants.SERVICE_DOMAIN_NAME); return prefix + "/detail"; } /** * 修改安全体系文件 */ @GetMapping("/edit/{documentsId}") public String edit(@PathVariable("documentsId") Long documentsId, ModelMap mmap) { SecurityDocuments securityDocuments = securityDocumentsService.selectSecurityDocumentsById(documentsId); mmap.put("securityDocuments", securityDocuments); mmap.put("type", securityDocuments.getSecurityType()); Attachment attachment = new Attachment(); attachment.setFileJoinId(documentsId); attachment.setAttachmentType(DcAttachmentTypeEnum.Security_Documents.getType()); List attachmentList = attachmentService.selectAttachmentList(attachment); mmap.put("fileList",attachmentList); mmap.put("serviceDomainName", DualControlConstants.SERVICE_DOMAIN_NAME); return prefix + "/edit"; } /** * 修改保存安全体系文件 */ @RequiresPermissions("dc:securityDocuments:edit") @Log(title = "安全体系文件", businessType = BusinessType.UPDATE) @PostMapping("/edit") @ResponseBody public AjaxResult editSave(SecurityDocuments securityDocuments) { User user = getSysUser(); securityDocuments.setCompanyId(user.getCompanyId()); int updateNum = securityDocumentsService.updateSecurityDocuments(securityDocuments); //处理新增的附件 String ownedFile = securityDocuments.getOwnedFile(); if(StringUtils.isNotEmpty(ownedFile)&&updateNum>0) { List attachmentList = JSON.parseArray(ownedFile, Attachment.class); if (attachmentList.size() > 0) { for (Attachment attachment : attachmentList) { attachment.setFileJoinId(securityDocuments.getDocumentsId()); attachment.setAttachmentType(DcAttachmentTypeEnum.Security_Documents.getType()); attachment.setCompanyId(user.getCompanyId()); attachment.setCreateBy(user.getUserName()); attachment.setCreateUserId(user.getUserId()); attachment.setCreateTime(new Date()); attachmentService.insertAttachment(attachment); } } } return toAjax(updateNum); } /** * 删除安全体系文件 */ @RequiresPermissions("dc:securityDocuments:remove") @Log(title = "安全体系文件", businessType = BusinessType.DELETE) @PostMapping("/remove") @ResponseBody public AjaxResult remove(String ids) { return toAjax(securityDocumentsService.deleteSecurityDocumentsByIds(ids)); } }