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

Restructure api interceptor

pull/137/head
johnniang 2019-03-08 19:32:47 +08:00
parent 4ec6604dcb
commit e654e96db6
3 changed files with 24 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/main/java/cc/ryanc/halo/config/SwaggerConfiguration.java
View File

@ -30,6 +30,8 @@ import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import static cc.ryanc.halo.model.support.HaloConst.HALO_VERSION;
import static cc.ryanc.halo.model.support.HaloConst.TOKEN_HEADER;
import static springfox.documentation.schema.AlternateTypeRules.newRule;
/**
@ -101,7 +103,7 @@ public class SwaggerConfiguration {
}
private ApiKey apiKeys() {
return new ApiKey("TOKEN ACCESS", HttpHeaders.AUTHORIZATION, In.HEADER.name());
return new ApiKey("TOKEN ACCESS", TOKEN_HEADER, In.HEADER.name());
}
private SecurityContext securityContext() {
@ -121,7 +123,7 @@ public class SwaggerConfiguration {
return new ApiInfoBuilder()
.title("Halo API Documentation")
.description("Documentation for Halo API")
.version("v0.4.2")
.version(HALO_VERSION)
.termsOfServiceUrl("https://ryanc.cc/")
.contact(new Contact("RYAN0UP", "https://ryanc.cc/", "i#ryanc.cc"))
.build();

5
src/main/java/cc/ryanc/halo/model/support/HaloConst.java
View File

@ -20,6 +20,11 @@ public class HaloConst {
*/
public static final String HALO_VERSION = "0.4.2";
/**
* Token key
*/
public static final String TOKEN_HEADER = "token";
/**
* 所有设置选项（key,value）
*/

20
src/main/java/cc/ryanc/halo/web/interceptor/ApiInterceptor.java
View File

@ -1,8 +1,8 @@
package cc.ryanc.halo.web.interceptor;
import cc.ryanc.halo.model.support.JsonResult;
import cc.ryanc.halo.model.enums.BlogPropertiesEnum;
import cc.ryanc.halo.model.enums.TrueFalseEnum;
import cc.ryanc.halo.model.support.JsonResult;
import cn.hutool.core.util.StrUtil;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.springframework.http.HttpStatus;
@ -16,6 +16,7 @@ import javax.servlet.http.HttpServletResponse;
import java.nio.charset.StandardCharsets;
import static cc.ryanc.halo.model.support.HaloConst.OPTIONS;
import static cc.ryanc.halo.model.support.HaloConst.TOKEN_HEADER;
/**
* <pre>
@ -28,8 +29,6 @@ import static cc.ryanc.halo.model.support.HaloConst.OPTIONS;
@Component
public class ApiInterceptor implements HandlerInterceptor {
private static final String TOKEN = "token";
private final ObjectMapper objectMapper;
public ApiInterceptor(ObjectMapper objectMapper) {
@ -38,19 +37,20 @@ public class ApiInterceptor implements HandlerInterceptor {
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
if (StrUtil.equals(TrueFalseEnum.TRUE.getDesc(), OPTIONS.get(BlogPropertiesEnum.API_STATUS.getProp()))) {
if (StrUtil.equals(request.getHeader(TOKEN), OPTIONS.get(BlogPropertiesEnum.API_TOKEN.getProp()))) {
return true;
} else {
if (!StrUtil.equals(TrueFalseEnum.TRUE.getDesc(), OPTIONS.get(BlogPropertiesEnum.API_STATUS.getProp()))) {
response.sendRedirect("/404");
return false;
}
if (!StrUtil.equals(request.getHeader(TOKEN_HEADER), OPTIONS.get(BlogPropertiesEnum.API_TOKEN.getProp()))) {
response.setCharacterEncoding(StandardCharsets.UTF_8.name());
response.setContentType(MediaType.APPLICATION_JSON_UTF8_VALUE);
JsonResult result = new JsonResult(HttpStatus.BAD_REQUEST.value(), "Invalid Token");
response.getWriter().write(objectMapper.writeValueAsString(result));
return false;
}
}
response.sendRedirect("/404");
return false;
return true;
}
@Override