From d1da6880f6fe00947ad3f32c894f2f40ff51a9d0 Mon Sep 17 00:00:00 2001
From: ruibaby <i@ryanc.cc>
Date: Mon, 22 Apr 2019 17:36:12 +0800
Subject: [PATCH] Refactor GalleryTagDirective.

---
 .../halo/app/model/dto/GalleryOutputDTO.java  |  2 +
 .../run/halo/app/model/entity/Gallery.java    |  6 ++
 .../java/run/halo/app/model/entity/Link.java  |  2 +-
 .../freemarker/tag/GalleryTagDirective.java   | 11 +++-
 .../freemarker/tag/LinkTagDirective.java      |  7 ++-
 .../run/halo/app/model/vo/GalleryTeamVO.java  | 23 +++++++
 .../app/repository/GalleryRepository.java     | 13 +++-
 .../run/halo/app/service/GalleryService.java  | 23 ++++++-
 .../run/halo/app/service/LinkService.java     |  8 ---
 .../app/service/impl/GalleryServiceImpl.java  | 60 +++++++++++++++++--
 .../app/service/impl/LinkServiceImpl.java     | 11 ----
 11 files changed, 133 insertions(+), 33 deletions(-)
 create mode 100644 src/main/java/run/halo/app/model/vo/GalleryTeamVO.java

diff --git a/src/main/java/run/halo/app/model/dto/GalleryOutputDTO.java b/src/main/java/run/halo/app/model/dto/GalleryOutputDTO.java
index 2ac50069f..cfa28b06f 100644
--- a/src/main/java/run/halo/app/model/dto/GalleryOutputDTO.java
+++ b/src/main/java/run/halo/app/model/dto/GalleryOutputDTO.java
@@ -20,4 +20,6 @@ public class GalleryOutputDTO implements OutputConverter<GalleryOutputDTO, Galle
     private String thumbnail;
 
     private String url;
+
+    private String team;
 }
diff --git a/src/main/java/run/halo/app/model/entity/Gallery.java b/src/main/java/run/halo/app/model/entity/Gallery.java
index 9314ea0c8..1cca462a9 100644
--- a/src/main/java/run/halo/app/model/entity/Gallery.java
+++ b/src/main/java/run/halo/app/model/entity/Gallery.java
@@ -66,6 +66,12 @@ public class Gallery extends BaseEntity {
     @Column(name = "url", columnDefinition = "varchar(1023) not null")
     private String url;
 
+    /**
+     * Gallery team name.
+     */
+    @Column(name = "team", columnDefinition = "varchar(255) default ''")
+    private String team;
+
     @Override
     public void prePersist() {
         super.prePersist();
diff --git a/src/main/java/run/halo/app/model/entity/Link.java b/src/main/java/run/halo/app/model/entity/Link.java
index 8b0c59f3a..e5d3deac1 100644
--- a/src/main/java/run/halo/app/model/entity/Link.java
+++ b/src/main/java/run/halo/app/model/entity/Link.java
@@ -53,7 +53,7 @@ public class Link extends BaseEntity {
     private String description;
 
     /**
-     * Link group name.
+     * Link team name.
      */
     @Column(name = "team", columnDefinition = "varchar(255) default ''")
     private String team;
diff --git a/src/main/java/run/halo/app/model/freemarker/tag/GalleryTagDirective.java b/src/main/java/run/halo/app/model/freemarker/tag/GalleryTagDirective.java
index 4a5544b47..ab824aa8b 100644
--- a/src/main/java/run/halo/app/model/freemarker/tag/GalleryTagDirective.java
+++ b/src/main/java/run/halo/app/model/freemarker/tag/GalleryTagDirective.java
@@ -2,6 +2,7 @@ package run.halo.app.model.freemarker.tag;
 
 import freemarker.core.Environment;
 import freemarker.template.*;
+import org.springframework.data.domain.Sort;
 import org.springframework.stereotype.Component;
 import run.halo.app.model.support.HaloConst;
 import run.halo.app.service.GalleryService;
@@ -9,6 +10,8 @@ import run.halo.app.service.GalleryService;
 import java.io.IOException;
 import java.util.Map;
 
+import static org.springframework.data.domain.Sort.Direction.DESC;
+
 /**
  * Freemarker custom tag of gallery.
  *
@@ -30,12 +33,16 @@ public class GalleryTagDirective implements TemplateDirectiveModel {
 
         if (params.containsKey(HaloConst.METHOD_KEY)) {
             String method = params.get(HaloConst.METHOD_KEY).toString();
+            String team = params.get("team").toString();
             switch (method) {
                 case "list":
                     env.setVariable("galleries", builder.build().wrap(galleryService.listAll()));
                     break;
-                case "listTeamVos":
-                    env.setVariable("galleries", builder.build().wrap(null));
+                case "listTeams":
+                    env.setVariable("teams", builder.build().wrap(galleryService.listDtos(Sort.by(DESC, "createTime"))));
+                    break;
+                case "listByTeam":
+                    env.setVariable("galleries", builder.build().wrap(galleryService.listByTeam(team, Sort.by(DESC, "createTime"))));
                     break;
                 case "count":
                     env.setVariable("count", builder.build().wrap(galleryService.count()));
diff --git a/src/main/java/run/halo/app/model/freemarker/tag/LinkTagDirective.java b/src/main/java/run/halo/app/model/freemarker/tag/LinkTagDirective.java
index 186c991a0..c1a41a7b6 100644
--- a/src/main/java/run/halo/app/model/freemarker/tag/LinkTagDirective.java
+++ b/src/main/java/run/halo/app/model/freemarker/tag/LinkTagDirective.java
@@ -2,6 +2,7 @@ package run.halo.app.model.freemarker.tag;
 
 import freemarker.core.Environment;
 import freemarker.template.*;
+import org.springframework.data.domain.Sort;
 import org.springframework.stereotype.Component;
 import run.halo.app.model.support.HaloConst;
 import run.halo.app.service.LinkService;
@@ -9,6 +10,8 @@ import run.halo.app.service.LinkService;
 import java.io.IOException;
 import java.util.Map;
 
+import static org.springframework.data.domain.Sort.Direction.DESC;
+
 /**
  * Freemarker custom tag of link.
  *
@@ -34,8 +37,8 @@ public class LinkTagDirective implements TemplateDirectiveModel {
                 case "list":
                     env.setVariable("links", builder.build().wrap(linkService.listAll()));
                     break;
-                case "listTeamVos":
-                    env.setVariable("links", builder.build().wrap(null));
+                case "listTeams":
+                    env.setVariable("teams", builder.build().wrap(linkService.listTeamVos(Sort.by(DESC, "createTime"))));
                     break;
                 case "count":
                     env.setVariable("count", builder.build().wrap(linkService.count()));
diff --git a/src/main/java/run/halo/app/model/vo/GalleryTeamVO.java b/src/main/java/run/halo/app/model/vo/GalleryTeamVO.java
new file mode 100644
index 000000000..83c28ebf5
--- /dev/null
+++ b/src/main/java/run/halo/app/model/vo/GalleryTeamVO.java
@@ -0,0 +1,23 @@
+package run.halo.app.model.vo;
+
+import lombok.Data;
+import lombok.ToString;
+import run.halo.app.model.dto.GalleryOutputDTO;
+import run.halo.app.model.dto.LinkOutputDTO;
+
+import java.util.List;
+
+/**
+ * Link team vo.
+ *
+ * @author : RYAN0UP
+ * @date : 2019/3/22
+ */
+@Data
+@ToString
+public class GalleryTeamVO {
+
+    private String team;
+
+    private List<GalleryOutputDTO> galleries;
+}
diff --git a/src/main/java/run/halo/app/repository/GalleryRepository.java b/src/main/java/run/halo/app/repository/GalleryRepository.java
index be4711910..7000ba945 100644
--- a/src/main/java/run/halo/app/repository/GalleryRepository.java
+++ b/src/main/java/run/halo/app/repository/GalleryRepository.java
@@ -1,8 +1,10 @@
 package run.halo.app.repository;
 
+import org.springframework.data.domain.Sort;
 import run.halo.app.model.entity.Gallery;
 import run.halo.app.repository.base.BaseRepository;
-import run.halo.app.repository.base.BaseRepository;
+
+import java.util.List;
 
 /**
  * Gallery repository.
@@ -10,4 +12,13 @@ import run.halo.app.repository.base.BaseRepository;
  * @author johnniang
  */
 public interface GalleryRepository extends BaseRepository<Gallery, Integer> {
+
+    /**
+     * Query galleries by team
+     *
+     * @param team team
+     * @param sort sort
+     * @return list of gallery
+     */
+    List<Gallery> findByTeam(String team, Sort sort);
 }
diff --git a/src/main/java/run/halo/app/service/GalleryService.java b/src/main/java/run/halo/app/service/GalleryService.java
index 4cec5fe60..445b0c782 100644
--- a/src/main/java/run/halo/app/service/GalleryService.java
+++ b/src/main/java/run/halo/app/service/GalleryService.java
@@ -1,10 +1,10 @@
 package run.halo.app.service;
 
-import run.halo.app.model.dto.GalleryOutputDTO;
-import run.halo.app.model.entity.Gallery;
-import run.halo.app.service.base.CrudService;
 import org.springframework.data.domain.Sort;
 import org.springframework.lang.NonNull;
+import run.halo.app.model.dto.GalleryOutputDTO;
+import run.halo.app.model.entity.Gallery;
+import run.halo.app.model.vo.GalleryTeamVO;
 import run.halo.app.service.base.CrudService;
 
 import java.util.List;
@@ -23,4 +23,21 @@ public interface GalleryService extends CrudService<Gallery, Integer> {
      * @return all galleries
      */
     List<GalleryOutputDTO> listDtos(@NonNull Sort sort);
+
+    /**
+     * Lists gallery team vos.
+     *
+     * @param sort must not be null
+     * @return a list of gallery team vo
+     */
+    List<GalleryTeamVO> listTeamVos(@NonNull Sort sort);
+
+    /**
+     * List galleries by team.
+     *
+     * @param team team
+     * @param sort sort
+     * @return list of galleries
+     */
+    List<GalleryOutputDTO> listByTeam(@NonNull String team, Sort sort);
 }
diff --git a/src/main/java/run/halo/app/service/LinkService.java b/src/main/java/run/halo/app/service/LinkService.java
index 3098f314a..7b5732c73 100755
--- a/src/main/java/run/halo/app/service/LinkService.java
+++ b/src/main/java/run/halo/app/service/LinkService.java
@@ -27,14 +27,6 @@ public interface LinkService extends CrudService<Link, Integer> {
     @NonNull
     List<LinkOutputDTO> listDtos(@NonNull Sort sort);
 
-    /**
-     * List link by group
-     *
-     * @return a list of link team vo
-     */
-    @NonNull
-    List<LinkTeamVO> listTeamVos();
-
     /**
      * Lists link team vos.
      *
diff --git a/src/main/java/run/halo/app/service/impl/GalleryServiceImpl.java b/src/main/java/run/halo/app/service/impl/GalleryServiceImpl.java
index 4f64283a8..731f42209 100644
--- a/src/main/java/run/halo/app/service/impl/GalleryServiceImpl.java
+++ b/src/main/java/run/halo/app/service/impl/GalleryServiceImpl.java
@@ -1,17 +1,20 @@
 package run.halo.app.service.impl;
 
-import run.halo.app.model.dto.GalleryOutputDTO;
-import run.halo.app.model.entity.Gallery;
-import run.halo.app.repository.GalleryRepository;
-import run.halo.app.service.GalleryService;
-import run.halo.app.service.base.AbstractCrudService;
 import org.springframework.data.domain.Sort;
 import org.springframework.stereotype.Service;
 import org.springframework.util.Assert;
+import run.halo.app.model.dto.GalleryOutputDTO;
+import run.halo.app.model.entity.Gallery;
+import run.halo.app.model.vo.GalleryTeamVO;
 import run.halo.app.repository.GalleryRepository;
+import run.halo.app.service.GalleryService;
 import run.halo.app.service.base.AbstractCrudService;
+import run.halo.app.utils.ServiceUtils;
 
+import java.util.LinkedList;
 import java.util.List;
+import java.util.Map;
+import java.util.Set;
 import java.util.stream.Collectors;
 
 /**
@@ -42,4 +45,51 @@ public class GalleryServiceImpl extends AbstractCrudService<Gallery, Integer> im
 
         return listAll(sort).stream().map(gallery -> (GalleryOutputDTO) new GalleryOutputDTO().convertFrom(gallery)).collect(Collectors.toList());
     }
+
+    /**
+     * Lists gallery team vos.
+     *
+     * @param sort must not be null
+     * @return a list of gallery team vo
+     */
+    @Override
+    public List<GalleryTeamVO> listTeamVos(Sort sort) {
+        Assert.notNull(sort, "Sort info must not be null");
+
+        // List all galleries
+        List<GalleryOutputDTO> galleries = listDtos(sort);
+
+        // Get teams
+        Set<String> teams = ServiceUtils.fetchProperty(galleries, GalleryOutputDTO::getTeam);
+
+        Map<String, List<GalleryOutputDTO>> teamGalleryListMap = ServiceUtils.convertToListMap(teams, galleries, GalleryOutputDTO::getTeam);
+
+        List<GalleryTeamVO> result = new LinkedList<>();
+
+        // Wrap gallery team vo list
+        teamGalleryListMap.forEach((team, galleryList) -> {
+            // Build gallery team vo
+            GalleryTeamVO galleryTeamVO = new GalleryTeamVO();
+            galleryTeamVO.setTeam(team);
+            galleryTeamVO.setGalleries(galleryList);
+
+            // Add it to result
+            result.add(galleryTeamVO);
+        });
+
+        return result;
+    }
+
+    /**
+     * List galleries by team.
+     *
+     * @param team team
+     * @param sort sort
+     * @return list of galleries
+     */
+    @Override
+    public List<GalleryOutputDTO> listByTeam(String team, Sort sort) {
+        List<Gallery> galleries = galleryRepository.findByTeam(team, sort);
+        return galleries.stream().map(gallery -> (GalleryOutputDTO) new GalleryOutputDTO().convertFrom(gallery)).collect(Collectors.toList());
+    }
 }
diff --git a/src/main/java/run/halo/app/service/impl/LinkServiceImpl.java b/src/main/java/run/halo/app/service/impl/LinkServiceImpl.java
index 4efd3f9bf..157aa2c7c 100644
--- a/src/main/java/run/halo/app/service/impl/LinkServiceImpl.java
+++ b/src/main/java/run/halo/app/service/impl/LinkServiceImpl.java
@@ -52,17 +52,6 @@ public class LinkServiceImpl extends AbstractCrudService<Link, Integer> implemen
         return convertTo(listAll(sort));
     }
 
-    /**
-     * List link by group
-     *
-     * @return List<LinkTeamVO>
-     */
-    @Override
-    public List<LinkTeamVO> listTeamVos() {
-        // TODO list team
-        return null;
-    }
-
     @Override
     public List<LinkTeamVO> listTeamVos(Sort sort) {
         Assert.notNull(sort, "Sort info must not be null");