From d12c76fe03de5a77a9bc6391ea1160f3f985837d Mon Sep 17 00:00:00 2001
From: Timi <imyeyu6@qq.com>
Date: Thu, 11 Dec 2025 18:41:29 +0800
Subject: [PATCH] add Attachment.mimeType,metadata

---
 .../api/modules/common/bean/Metadata.java     | 24 +++++++++++++++++++
 .../api/modules/common/entity/Attachment.java |  6 ++++-
 .../implement/AttachmentServiceImplement.java | 11 +++++++++
 3 files changed, 40 insertions(+), 1 deletion(-)
 create mode 100644 src/main/java/com/imyeyu/api/modules/common/bean/Metadata.java

diff --git a/src/main/java/com/imyeyu/api/modules/common/bean/Metadata.java b/src/main/java/com/imyeyu/api/modules/common/bean/Metadata.java
new file mode 100644
index 0000000..dfad00b
--- /dev/null
+++ b/src/main/java/com/imyeyu/api/modules/common/bean/Metadata.java
@@ -0,0 +1,24 @@
+package com.imyeyu.api.modules.common.bean;
+
+import lombok.Data;
+
+/**
+ * @author 夜雨
+ * @since 2025-12-11 18:14
+ */
+public class Metadata {
+
+	/**
+	 *
+	 *
+	 * @author 夜雨
+	 * @since 2025-12-11 18:15
+	 */
+	@Data
+	public static class Image {
+
+		private int width;
+
+		private int height;
+	}
+}
diff --git a/src/main/java/com/imyeyu/api/modules/common/entity/Attachment.java b/src/main/java/com/imyeyu/api/modules/common/entity/Attachment.java
index 859a5f0..d6b03a0 100644
--- a/src/main/java/com/imyeyu/api/modules/common/entity/Attachment.java
+++ b/src/main/java/com/imyeyu/api/modules/common/entity/Attachment.java
@@ -1,5 +1,6 @@
 package com.imyeyu.api.modules.common.entity;
 
+import com.google.gson.JsonObject;
 import com.imyeyu.api.bean.MultilingualHandler;
 import com.imyeyu.java.ref.Ref;
 import com.imyeyu.spring.entity.Entity;
@@ -67,6 +68,10 @@ public class Attachment extends Entity implements MultilingualHandler {
 
 	protected String name;
 
+	protected String mimeType;
+
+	protected JsonObject metadata;
+
 	protected Long size;
 
 	protected String md5;
@@ -82,5 +87,4 @@ public class Attachment extends Entity implements MultilingualHandler {
 	public <T extends Enum<T>> T getAttachTypeValue(Class<T> attachTypeClass) {
 		return Ref.toType(attachTypeClass, attachType);
 	}
-
 }
diff --git a/src/main/java/com/imyeyu/api/modules/common/service/implement/AttachmentServiceImplement.java b/src/main/java/com/imyeyu/api/modules/common/service/implement/AttachmentServiceImplement.java
index aed0371..14eaea0 100644
--- a/src/main/java/com/imyeyu/api/modules/common/service/implement/AttachmentServiceImplement.java
+++ b/src/main/java/com/imyeyu/api/modules/common/service/implement/AttachmentServiceImplement.java
@@ -3,6 +3,7 @@ package com.imyeyu.api.modules.common.service.implement;
 import com.google.gson.Gson;
 import com.imyeyu.api.config.dbsource.TimiServerDBConfig;
 import com.imyeyu.api.modules.common.bean.MediaAttach;
+import com.imyeyu.api.modules.common.bean.Metadata;
 import com.imyeyu.api.modules.common.entity.Attachment;
 import com.imyeyu.api.modules.common.mapper.AttachmentMapper;
 import com.imyeyu.api.modules.common.service.AttachmentService;
@@ -31,6 +32,8 @@ import org.springframework.data.mongodb.gridfs.GridFsTemplate;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+import javax.imageio.ImageIO;
+import java.awt.image.BufferedImage;
 import java.io.ByteArrayInputStream;
 import java.io.ByteArrayOutputStream;
 import java.io.File;
@@ -101,6 +104,14 @@ public class AttachmentServiceImplement extends AbstractEntityService<Attachment
 			request.setMongoId(mongoId);
 			request.setSize(gridFSFile.getLength());
 			request.setMd5(IO.md5(gridFSBucket.openDownloadStream(gridFSFile.getObjectId())));
+			request.setMimeType(new Tika().detect(gridFSBucket.openDownloadStream(gridFSFile.getObjectId())));
+			if (request.getMimeType().startsWith("image")) {
+				BufferedImage image = ImageIO.read(gridFSBucket.openDownloadStream(gridFSFile.getObjectId()));
+				Metadata.Image metadata = new Metadata.Image();
+				metadata.setWidth(image.getWidth());
+				metadata.setHeight(image.getHeight());
+				request.setMetadata(gson.toJsonTree(metadata).getAsJsonObject());
+			}
 			mapper.insert(request);
 		} catch (Exception e) {
 			if (mongoId != null) {