From 3cd86042ffed0c5f28c403b6b97850e10b62293f Mon Sep 17 00:00:00 2001
From: shahaibo <1023316923@qq.com>
Date: Thu, 12 Jun 2025 13:12:44 +0800
Subject: [PATCH] =?UTF-8?q?TASK:=E4=B8=8A=E4=BC=A0=E5=9B=9E=E6=98=BE;?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/main/java/com/ai/da/model/enums/Module.java           | 4 +++-
 src/main/java/com/ai/da/model/vo/ModuleChooseVO.java      | 3 +++
 .../com/ai/da/service/impl/UserLikeGroupServiceImpl.java  | 8 ++++++++
 3 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/ai/da/model/enums/Module.java b/src/main/java/com/ai/da/model/enums/Module.java
index 78c24bc5..8f9a2eed 100644
--- a/src/main/java/com/ai/da/model/enums/Module.java
+++ b/src/main/java/com/ai/da/model/enums/Module.java
@@ -14,7 +14,9 @@ public enum Module implements IEnumDisplay {
     poseTransfer("Pose Transfer"),
     canvas("Canvas"),
     patternMaking3D("3D Pattern Making"),
-    deReconstruction("De/Reconstruction");
+    deReconstruction("De/Reconstruction"),
+    uploadElement("UploadElement"),
+    ;
 
     private final String value;
 
diff --git a/src/main/java/com/ai/da/model/vo/ModuleChooseVO.java b/src/main/java/com/ai/da/model/vo/ModuleChooseVO.java
index 0d899a9f..ae73e6f7 100644
--- a/src/main/java/com/ai/da/model/vo/ModuleChooseVO.java
+++ b/src/main/java/com/ai/da/model/vo/ModuleChooseVO.java
@@ -1,5 +1,6 @@
 package com.ai.da.model.vo;
 
+import com.ai.da.mapper.primary.entity.ToProductElement;
 import com.ai.da.model.dto.PortfolioDTO;
 import com.ai.da.model.dto.ToProductImageDTO;
 import io.swagger.annotations.ApiModel;
@@ -37,4 +38,6 @@ public class ModuleChooseVO {
     private ThreeDModuleVO patternMaking3D;
 
     private List<CollectionElementVO> mannequin;
+
+    private List<ToProductElement> uploadElement;
 }
diff --git a/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java b/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
index fcffafa2..3518f410 100644
--- a/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
@@ -1747,6 +1747,14 @@ public class UserLikeGroupServiceImpl extends ServiceImpl<UserLikeGroupMapper, U
                     }
                 }
                 moduleChooseVO.setMannequin(list);
+            }else if (module.equals(Module.uploadElement.name())) {
+                QueryWrapper<ToProductElement> qw = new QueryWrapper<>();
+                qw.lambda().eq(ToProductElement::getProjectId, projectDTO.getId());
+                List<ToProductElement> toProductElements = toProductElementMapper.selectList(qw);
+                for (ToProductElement toProductElement : toProductElements) {
+                    toProductElement.setUrl(minioUtil.getPreSignedUrl(toProductElement.getUrl(), 24 * 60));
+                }
+                moduleChooseVO.setUploadElement(toProductElements);
             }
         }
         return moduleChooseVO;