github
/
halo
mirror of https://github.com/halo-dev/halo
1
0
Fork 0
Code Issues Releases Wiki Activity

Refactor PostTagDirective.

pull/146/head
ruibaby 2019-04-29 15:39:13 +08:00
parent 683444b66f
commit 53eee6cb82
1 changed files with 20 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/main/java/run/halo/app/model/freemarker/tag/PostTagDirective.java
View File

@ -4,7 +4,9 @@ import freemarker.core.Environment;
import freemarker.template.*; import freemarker.template.*;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import run.halo.app.model.support.HaloConst; import run.halo.app.model.support.HaloConst;
import run.halo.app.service.PostCategoryService;
import run.halo.app.service.PostService; import run.halo.app.service.PostService;
import run.halo.app.service.PostTagService;
import java.io.IOException; import java.io.IOException;
import java.util.Map; import java.util.Map;
@ -20,10 +22,17 @@ public class PostTagDirective implements TemplateDirectiveModel {
private final PostService postService; private final PostService postService;
public PostTagDirective(Configuration configuration, private final PostTagService postTagService;
PostService postService) {
this.postService = postService;
private final PostCategoryService postCategoryService;
public PostTagDirective(Configuration configuration,
PostService postService,
PostTagService postTagService,
PostCategoryService postCategoryService) {
this.postService = postService;
this.postTagService = postTagService;
this.postCategoryService = postCategoryService;
configuration.setSharedVariable("postTag", this); configuration.setSharedVariable("postTag", this);
} }
@ -32,6 +41,8 @@ public class PostTagDirective implements TemplateDirectiveModel {
final DefaultObjectWrapperBuilder builder = new DefaultObjectWrapperBuilder(Configuration.VERSION_2_3_25); final DefaultObjectWrapperBuilder builder = new DefaultObjectWrapperBuilder(Configuration.VERSION_2_3_25);
if (params.containsKey(HaloConst.METHOD_KEY)) { if (params.containsKey(HaloConst.METHOD_KEY)) {
String method = params.get(HaloConst.METHOD_KEY).toString(); String method = params.get(HaloConst.METHOD_KEY).toString();
Integer categoryId = Integer.parseInt(params.get("categoryId").toString());
Integer tagId = Integer.parseInt(params.get("tagId").toString());
switch (method) { switch (method) {
case "count": case "count":
env.setVariable("count", builder.build().wrap(postService.count())); env.setVariable("count", builder.build().wrap(postService.count()));
@ -42,6 +53,12 @@ public class PostTagDirective implements TemplateDirectiveModel {
case "archiveMonth": case "archiveMonth":
env.setVariable("archives", builder.build().wrap(postService.listMonthArchives())); env.setVariable("archives", builder.build().wrap(postService.listMonthArchives()));
break; break;
case "listByCategoryId":
env.setVariable("posts", builder.build().wrap(postCategoryService.listPostBy(categoryId)));
break;
case "listByTagId":
env.setVariable("posts", builder.build().wrap(postTagService.listPostsBy(tagId)));
break;
default: default:
break; break;
} }