From 9d1dd77f6fa199b325a20c220746f287310e8418 Mon Sep 17 00:00:00 2001
From: tamguo <tamguo@Lenovo>
Date: Wed, 14 Nov 2018 18:29:30 +0800
Subject: [PATCH] =?UTF-8?q?=E4=B8=8A=E4=BC=A0=E9=99=84=E4=BB=B6=E5=8A=9F?=
 =?UTF-8?q?=E8=83=BD=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../tamguo/web/member/DocumentController.java |  86 +++++++++---
 .../src/main/resources/static/js/editor.js    |  10 +-
 .../src/main/resources/static/js/markdown.js  |   2 +-
 .../resources/templates/member/book/edit.html |   8 +-
 .../modules/book/dao/FileUploadMapper.java    |   8 ++
 .../modules/book/model/DocumentEntity.java    |   9 ++
 .../modules/book/model/FileUploadEntity.java  | 128 ++++++++++++++++++
 .../modules/book/model/enums/BizTypeEnum.java |  32 +++++
 .../model/enums/FileUploadStatusEnum.java     |  33 +++++
 .../book/service/IFileUploadService.java      |   8 ++
 .../service/impl/FileUploadServiceImpl.java   |  13 ++
 11 files changed, 305 insertions(+), 32 deletions(-)
 create mode 100644 tamguo-modules-core/src/main/java/com/tamguo/modules/book/dao/FileUploadMapper.java
 create mode 100644 tamguo-modules-core/src/main/java/com/tamguo/modules/book/model/FileUploadEntity.java
 create mode 100644 tamguo-modules-core/src/main/java/com/tamguo/modules/book/model/enums/BizTypeEnum.java
 create mode 100644 tamguo-modules-core/src/main/java/com/tamguo/modules/book/model/enums/FileUploadStatusEnum.java
 create mode 100644 tamguo-modules-core/src/main/java/com/tamguo/modules/book/service/IFileUploadService.java
 create mode 100644 tamguo-modules-core/src/main/java/com/tamguo/modules/book/service/impl/FileUploadServiceImpl.java

diff --git a/tamguo-bms/src/main/java/com/tamguo/web/member/DocumentController.java b/tamguo-bms/src/main/java/com/tamguo/web/member/DocumentController.java
index 6e8df1c..3076433 100644
--- a/tamguo-bms/src/main/java/com/tamguo/web/member/DocumentController.java
+++ b/tamguo-bms/src/main/java/com/tamguo/web/member/DocumentController.java
@@ -4,9 +4,11 @@ import java.io.File;
 import java.io.FileInputStream;
 import java.io.IOException;
 import java.util.Date;
+import java.util.List;
 
 import javax.servlet.http.HttpServletRequest;
 
+import org.apache.commons.collections.CollectionUtils;
 import org.apache.commons.io.FileUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -25,8 +27,12 @@ import com.tamguo.common.utils.DateUtil;
 import com.tamguo.common.utils.Result;
 import com.tamguo.modules.book.model.DocumentEntity;
 import com.tamguo.modules.book.model.FileEntity;
+import com.tamguo.modules.book.model.FileUploadEntity;
+import com.tamguo.modules.book.model.enums.BizTypeEnum;
+import com.tamguo.modules.book.model.enums.FileUploadStatusEnum;
 import com.tamguo.modules.book.service.IDocumentService;
 import com.tamguo.modules.book.service.IFileEntityService;
+import com.tamguo.modules.book.service.IFileUploadService;
 import com.tamguo.utils.FileMd5Utils;
 
 @Controller
@@ -39,18 +45,32 @@ public class DocumentController {
 	IDocumentService iDocumentService;
 	@Autowired
 	IFileEntityService iFileEntityService;
+	@Autowired
+	IFileUploadService iFileUploadService;
 	
 	@Value("${file.storage.path}")
 	private String fileStoragePath;
 	@Value("${domain.name}")
 	private String domainName;
 	
+	@SuppressWarnings("unchecked")
 	@RequestMapping(value = "{id}" , method = RequestMethod.GET)
 	@ResponseBody
 	public Result getDocument(@PathVariable String id) {
 		DocumentEntity document = null;
 		try {
 			document = iDocumentService.selectById(id);
+			// 查询附件
+			List<FileUploadEntity> fileUploads = iFileUploadService.selectList(Condition.create().eq("biz_key", document.getId()).eq("biz_type", BizTypeEnum.DOCUMENT.getValue()));
+			if(!CollectionUtils.isEmpty(fileUploads)) {
+				for(int i=0 ; i<fileUploads.size() ; i++) {
+					FileUploadEntity fileUpload = fileUploads.get(i);
+					FileEntity fileEntity = iFileEntityService.selectOne(Condition.create().eq("file_id", fileUpload.getFileId()));
+					fileUpload.setFilePath(domainName + "files/" + fileEntity.getFilePath());
+					fileUpload.setFileSize(fileEntity.getFileSize());
+				}
+				document.setFileUploads(fileUploads);
+			}
 		} catch (Exception e) {
 			logger.error(e.getMessage() , e );
 			return Result.failResult("查询失败");
@@ -130,36 +150,58 @@ public class DocumentController {
 	@SuppressWarnings("unchecked")
 	@RequestMapping(value = "uploadFile" , method = RequestMethod.POST)
 	@ResponseBody
-	public Result uploadFile(@RequestParam("editormd-file-file") MultipartFile file, String bookId , HttpServletRequest request) {
+	public Result uploadFile(@RequestParam("editormd-file-file") MultipartFile file , String documentId , String bookId , HttpServletRequest request) {
 		try {
 			String fileMd5 = FileMd5Utils.getMD5((FileInputStream)file.getInputStream());
 			FileEntity sysFile = iFileEntityService.selectOne(Condition.create().eq("file_md5", fileMd5));
-			if(sysFile != null) {
-				sysFile.setFilePath(domainName + "files/" + sysFile.getFilePath());
-				return Result.successResult(sysFile);
+			// 文件不存在
+			if(sysFile == null) {
+				String filePath  = fileStoragePath + "book/" + DateUtil.fomatDate(new Date(), "yyyyMM") + "/" + bookId;
+				File dest = new File(filePath);
+				if(!dest.exists()) {
+					dest.mkdirs();
+				}
+				// save 文件
+				FileUtils.writeByteArrayToFile(new File(filePath + "/" + file.getOriginalFilename()) , file.getBytes());
+				sysFile = new FileEntity();
+				sysFile.setFileContentType(file.getContentType());
+				sysFile.setFileExtension(file.getOriginalFilename());
+				sysFile.setFileMd5(FileMd5Utils.getMD5((FileInputStream)file.getInputStream()));
+				sysFile.setFileSize(file.getSize());
+				sysFile.setFilePath("book/" + DateUtil.fomatDate(new Date(), "yyyyMM") + "/" + bookId + "/" + file.getOriginalFilename());
+				iFileEntityService.insert(sysFile);
 			}
-			String filePath  = fileStoragePath + "book/" + DateUtil.fomatDate(new Date(), "yyyyMM") + "/" + bookId;
-			File dest = new File(filePath);
-			if(!dest.exists()) {
-				dest.mkdirs();
-			}
-			// save 文件
-			FileUtils.writeByteArrayToFile(new File(filePath + "/" + file.getOriginalFilename()) , file.getBytes());
-			
-			FileEntity fileEntity = new FileEntity();
-			fileEntity.setFileContentType(file.getContentType());
-			fileEntity.setFileExtension(file.getOriginalFilename());
-			fileEntity.setFileMd5(FileMd5Utils.getMD5((FileInputStream)file.getInputStream()));
-			fileEntity.setFileSize(file.getSize());
-			fileEntity.setFilePath("book/" + DateUtil.fomatDate(new Date(), "yyyyMM") + "/" + bookId + "/" + file.getOriginalFilename());
-			iFileEntityService.insert(fileEntity);
-			
-			fileEntity.setFilePath(domainName + "files/" + fileEntity.getFilePath());
+			// 创建上传记录
+			FileUploadEntity fileUpload = new FileUploadEntity();
+			fileUpload.setBizKey(documentId);
+			fileUpload.setBizType(BizTypeEnum.DOCUMENT);
+			fileUpload.setCreateBy("system");
+			fileUpload.setCreateDate(new Date());
+			fileUpload.setFileId(sysFile.getFileId());
+			fileUpload.setFileName(sysFile.getFileExtension());
+			fileUpload.setFileType(file.getContentType());
+			fileUpload.setUpdateBy("system");
+			fileUpload.setUpdateDate(new Date());
+			fileUpload.setStatus(FileUploadStatusEnum.NORMAL);
+			iFileUploadService.insert(fileUpload);
 			
-			return Result.successResult(fileEntity);
+			fileUpload.setFilePath(domainName + "files/" + sysFile.getFilePath());
+			fileUpload.setFileSize(sysFile.getFileSize());
+			return Result.successResult(fileUpload);
 		} catch (IOException e) {
 			e.printStackTrace();
 			return Result.failResult("上传失败");
 		}
 	}
+	
+	@RequestMapping(value = "removeFile" , method = RequestMethod.POST)
+	@ResponseBody
+	public Result removeFile(String id) {
+		try {
+			iFileUploadService.deleteById(id);
+		} catch (Exception e) {
+			return Result.result(1, null, "删除失败");
+		}
+		return Result.successResult("删除成功");
+	}
 }
diff --git a/tamguo-bms/src/main/resources/static/js/editor.js b/tamguo-bms/src/main/resources/static/js/editor.js
index 6111a21..f006f71 100644
--- a/tamguo-bms/src/main/resources/static/js/editor.js
+++ b/tamguo-bms/src/main/resources/static/js/editor.js
@@ -404,23 +404,23 @@ $(function () {
             removeAttach : function ($attach_id) {
                 var $this = this;
                 var item = $this.lists.filter(function ($item) {
-                    return $item.attachment_id == $attach_id;
+                    return $item.id == $attach_id;
                 });
 
                 if(item && item[0].hasOwnProperty("state")){
                     $this.lists = $this.lists.filter(function ($item) {
-                        return $item.attachment_id != $attach_id;
+                        return $item.id != $attach_id;
                     });
                     return;
                 }
                 $.ajax({
                     url : window.removeAttachURL,
                     type : "post",
-                    data : { "attach_id" : $attach_id},
+                    data : { "id" : $attach_id},
                     success : function (res) {
-                        if(res.errcode === 0){
+                        if(res.code === 0){
                             $this.lists = $this.lists.filter(function ($item) {
-                                return $item.attachment_id != $attach_id;
+                                return $item.id != $attach_id;
                             });
                         }else{
                             layer.msg(res.message);
diff --git a/tamguo-bms/src/main/resources/static/js/markdown.js b/tamguo-bms/src/main/resources/static/js/markdown.js
index 9b794b5..ba62378 100644
--- a/tamguo-bms/src/main/resources/static/js/markdown.js
+++ b/tamguo-bms/src/main/resources/static/js/markdown.js
@@ -154,7 +154,7 @@ $(function () {
                 var node = { "id": res.result.id, 'parent': res.result.parentId === "0" ? '#' : res.result.parentId, "text": res.result.name, "identify": res.result.id};
                 pushDocumentCategory(node);
                 window.selectNode = node;
-                pushVueLists(null);
+                pushVueLists(res.result.fileUploads);
                 setLastSelectNode($node);
             } else {
                 layer.msg("文档加载失败");
diff --git a/tamguo-bms/src/main/resources/templates/member/book/edit.html b/tamguo-bms/src/main/resources/templates/member/book/edit.html
index 81c758a..111a5e4 100644
--- a/tamguo-bms/src/main/resources/templates/member/book/edit.html
+++ b/tamguo-bms/src/main/resources/templates/member/book/edit.html
@@ -21,7 +21,7 @@
         window.releaseURL = "[[${domainName}]]" + "/book/xingzheng/release";
         window.sortURL = "[[${domainName}]]" + "/book/xingzheng/sort";
         window.historyURL = "[[${domainName}]]" + "/history/get";
-        window.removeAttachURL =  "[[${domainName}]]" + "/api/attach/remove";
+        window.removeAttachURL =  "[[${domainName}]]" + "member/document/removeFile";
         window.highlightStyle = "github";
         window.template = { "getUrl":"[[${domainName}]]" + "/api/template/get", "listUrl" : "[[${domainName}]]" + "/api/template/list", "deleteUrl" : "[[${domainName}]]" + "/api/template/remove", "saveUrl" :"[[${domainName}]]" + "/api/template/add"}
         
@@ -215,15 +215,15 @@
                                 </template>
                                 <template v-else-if="item.state == 'error'">
                                     <span class="error-message">${item.message}</span>
-                                    <button type="button" class="btn btn-sm close" @click="removeAttach(item.fileId)">
+                                    <button type="button" class="btn btn-sm close" @click="removeAttach(item.id)">
                                         <i class="fa fa-remove" aria-hidden="true"></i>
                                     </button>
                                 </template>
                                 <template v-else>
-                                    <a :href="item.filePath" target="_blank" :title="item.file_name">${item.fileExtension}</a>
+                                    <a :href="item.filePath" target="_blank" :title="item.fileName">${item.fileName}</a>
                                     <span class="text">(${ formatBytes(item.fileSize) })</span>
                                     <span class="error-message">${item.message}</span>
-                                    <button type="button" class="btn btn-sm close" @click="removeAttach(item.attachment_id)">
+                                    <button type="button" class="btn btn-sm close" @click="removeAttach(item.id)">
                                         <i class="fa fa-remove" aria-hidden="true"></i>
                                     </button>
                                     <div class="clearfix"></div>
diff --git a/tamguo-modules-core/src/main/java/com/tamguo/modules/book/dao/FileUploadMapper.java b/tamguo-modules-core/src/main/java/com/tamguo/modules/book/dao/FileUploadMapper.java
new file mode 100644
index 0000000..f0c2b0e
--- /dev/null
+++ b/tamguo-modules-core/src/main/java/com/tamguo/modules/book/dao/FileUploadMapper.java
@@ -0,0 +1,8 @@
+package com.tamguo.modules.book.dao;
+
+import com.baomidou.mybatisplus.mapper.BaseMapper;
+import com.tamguo.modules.book.model.FileUploadEntity;
+
+public interface FileUploadMapper extends BaseMapper<FileUploadEntity>{
+
+}
diff --git a/tamguo-modules-core/src/main/java/com/tamguo/modules/book/model/DocumentEntity.java b/tamguo-modules-core/src/main/java/com/tamguo/modules/book/model/DocumentEntity.java
index b0fa44e..a89f269 100644
--- a/tamguo-modules-core/src/main/java/com/tamguo/modules/book/model/DocumentEntity.java
+++ b/tamguo-modules-core/src/main/java/com/tamguo/modules/book/model/DocumentEntity.java
@@ -44,6 +44,9 @@ public class DocumentEntity extends Model<DocumentEntity>{
 	@TableField(exist=false)
 	private String cover; 
 	
+	@TableField(exist=false)
+	private List<FileUploadEntity> fileUploads;
+	
 	public String getId() {
 		return id;
 	}
@@ -144,5 +147,11 @@ public class DocumentEntity extends Model<DocumentEntity>{
 	public void setStatus(DocumentStatusEnum status) {
 		this.status = status;
 	}
+	public List<FileUploadEntity> getFileUploads() {
+		return fileUploads;
+	}
+	public void setFileUploads(List<FileUploadEntity> fileUploads) {
+		this.fileUploads = fileUploads;
+	}
 	
 }
diff --git a/tamguo-modules-core/src/main/java/com/tamguo/modules/book/model/FileUploadEntity.java b/tamguo-modules-core/src/main/java/com/tamguo/modules/book/model/FileUploadEntity.java
new file mode 100644
index 0000000..eef88a1
--- /dev/null
+++ b/tamguo-modules-core/src/main/java/com/tamguo/modules/book/model/FileUploadEntity.java
@@ -0,0 +1,128 @@
+package com.tamguo.modules.book.model;
+
+import java.io.Serializable;
+import java.util.Date;
+
+import com.alibaba.fastjson.annotation.JSONField;
+import com.alibaba.fastjson.serializer.SerializerFeature;
+import com.baomidou.mybatisplus.activerecord.Model;
+import com.baomidou.mybatisplus.annotations.TableField;
+import com.baomidou.mybatisplus.annotations.TableName;
+import com.tamguo.modules.book.model.enums.BizTypeEnum;
+import com.tamguo.modules.book.model.enums.FileUploadStatusEnum;
+
+@TableName(value="b_file_upload")
+public class FileUploadEntity extends Model<FileUploadEntity>{
+	
+	private static final long serialVersionUID = 1L;
+	
+	private String id;
+	private String fileId;
+	private String fileName;
+	private String fileType;
+	private String bizKey;	
+	@JSONField(serialzeFeatures= SerializerFeature.WriteEnumUsingToString)
+	private BizTypeEnum bizType;
+	@JSONField(serialzeFeatures= SerializerFeature.WriteEnumUsingToString)
+	private FileUploadStatusEnum status;
+	private String createBy;
+	private Date createDate;
+	private String updateBy;
+	private Date updateDate;
+	private String remarks;
+	
+	@TableField(exist=false)
+	private String filePath;
+	@TableField(exist=false)
+	private Long fileSize;
+	
+	public String getId() {
+		return id;
+	}
+	public void setId(String id) {
+		this.id = id;
+	}
+	public String getFileId() {
+		return fileId;
+	}
+	public void setFileId(String fileId) {
+		this.fileId = fileId;
+	}
+	public String getFileName() {
+		return fileName;
+	}
+	public void setFileName(String fileName) {
+		this.fileName = fileName;
+	}
+	public String getFileType() {
+		return fileType;
+	}
+	public void setFileType(String fileType) {
+		this.fileType = fileType;
+	}
+	public String getBizKey() {
+		return bizKey;
+	}
+	public void setBizKey(String bizKey) {
+		this.bizKey = bizKey;
+	}
+	public String getCreateBy() {
+		return createBy;
+	}
+	public void setCreateBy(String createBy) {
+		this.createBy = createBy;
+	}
+	public Date getCreateDate() {
+		return createDate;
+	}
+	public void setCreateDate(Date createDate) {
+		this.createDate = createDate;
+	}
+	public String getUpdateBy() {
+		return updateBy;
+	}
+	public void setUpdateBy(String updateBy) {
+		this.updateBy = updateBy;
+	}
+	public Date getUpdateDate() {
+		return updateDate;
+	}
+	public void setUpdateDate(Date updateDate) {
+		this.updateDate = updateDate;
+	}
+	public String getRemarks() {
+		return remarks;
+	}
+	public void setRemarks(String remarks) {
+		this.remarks = remarks;
+	}
+	@Override
+	protected Serializable pkVal() {
+		return getId();
+	}
+	public BizTypeEnum getBizType() {
+		return bizType;
+	}
+	public void setBizType(BizTypeEnum bizType) {
+		this.bizType = bizType;
+	}
+	public FileUploadStatusEnum getStatus() {
+		return status;
+	}
+	public void setStatus(FileUploadStatusEnum status) {
+		this.status = status;
+	}
+	public String getFilePath() {
+		return filePath;
+	}
+	public void setFilePath(String filePath) {
+		this.filePath = filePath;
+	}
+	public Long getFileSize() {
+		return fileSize;
+	}
+	public void setFileSize(Long fileSize) {
+		this.fileSize = fileSize;
+	}
+	
+}
diff --git a/tamguo-modules-core/src/main/java/com/tamguo/modules/book/model/enums/BizTypeEnum.java b/tamguo-modules-core/src/main/java/com/tamguo/modules/book/model/enums/BizTypeEnum.java
new file mode 100644
index 0000000..e3258dd
--- /dev/null
+++ b/tamguo-modules-core/src/main/java/com/tamguo/modules/book/model/enums/BizTypeEnum.java
@@ -0,0 +1,32 @@
+package com.tamguo.modules.book.model.enums;
+
+import java.io.Serializable;
+import com.baomidou.mybatisplus.enums.IEnum;
+
+/**
+ * 用户状态
+ */
+public enum BizTypeEnum implements IEnum {
+	DOCUMENT("document", "文档");
+
+    private String value;
+    private String desc;
+
+    BizTypeEnum(final String value, final String desc) {
+        this.value = value;
+        this.desc = desc;
+    }
+
+    public Serializable getValue() {
+        return this.value;
+    }
+
+    public String getDesc(){
+        return this.desc;
+    }
+    
+    @Override
+    public String toString() {
+    	return this.value;
+    }
+}
diff --git a/tamguo-modules-core/src/main/java/com/tamguo/modules/book/model/enums/FileUploadStatusEnum.java b/tamguo-modules-core/src/main/java/com/tamguo/modules/book/model/enums/FileUploadStatusEnum.java
new file mode 100644
index 0000000..ff95cc3
--- /dev/null
+++ b/tamguo-modules-core/src/main/java/com/tamguo/modules/book/model/enums/FileUploadStatusEnum.java
@@ -0,0 +1,33 @@
+package com.tamguo.modules.book.model.enums;
+
+import java.io.Serializable;
+import com.baomidou.mybatisplus.enums.IEnum;
+
+/**
+ * 用户状态
+ */
+public enum FileUploadStatusEnum implements IEnum {
+	NORMAL("normal", "正常"),
+	DELETE("delete", "删除");
+
+    private String value;
+    private String desc;
+
+    FileUploadStatusEnum(final String value, final String desc) {
+        this.value = value;
+        this.desc = desc;
+    }
+
+    public Serializable getValue() {
+        return this.value;
+    }
+
+    public String getDesc(){
+        return this.desc;
+    }
+    
+    @Override
+    public String toString() {
+    	return this.value;
+    }
+}
diff --git a/tamguo-modules-core/src/main/java/com/tamguo/modules/book/service/IFileUploadService.java b/tamguo-modules-core/src/main/java/com/tamguo/modules/book/service/IFileUploadService.java
new file mode 100644
index 0000000..593892d
--- /dev/null
+++ b/tamguo-modules-core/src/main/java/com/tamguo/modules/book/service/IFileUploadService.java
@@ -0,0 +1,8 @@
+package com.tamguo.modules.book.service;
+
+import com.baomidou.mybatisplus.service.IService;
+import com.tamguo.modules.book.model.FileUploadEntity;
+
+public interface IFileUploadService extends IService<FileUploadEntity>{
+
+}
diff --git a/tamguo-modules-core/src/main/java/com/tamguo/modules/book/service/impl/FileUploadServiceImpl.java b/tamguo-modules-core/src/main/java/com/tamguo/modules/book/service/impl/FileUploadServiceImpl.java
new file mode 100644
index 0000000..a4f1aad
--- /dev/null
+++ b/tamguo-modules-core/src/main/java/com/tamguo/modules/book/service/impl/FileUploadServiceImpl.java
@@ -0,0 +1,13 @@
+package com.tamguo.modules.book.service.impl;
+
+import org.springframework.stereotype.Service;
+
+import com.baomidou.mybatisplus.service.impl.ServiceImpl;
+import com.tamguo.modules.book.dao.FileUploadMapper;
+import com.tamguo.modules.book.model.FileUploadEntity;
+import com.tamguo.modules.book.service.IFileUploadService;
+
+@Service
+public class FileUploadServiceImpl extends ServiceImpl<FileUploadMapper, FileUploadEntity> implements IFileUploadService{
+
+}