2018-12-24 11:58:45 +08:00
|
|
|
|
/**
|
|
|
|
|
* Copyright (c) 2018-2028, lengleng (wangiegie@gmail.com).
|
|
|
|
|
* <p>
|
|
|
|
|
* Licensed under the GNU LESSER GENERAL PUBLIC LICENSE;
|
|
|
|
|
* you may not use this file except in compliance with the License.
|
|
|
|
|
* You may obtain a copy of the License at
|
|
|
|
|
* <p>
|
|
|
|
|
* http://www.gnu.org/licenses/lgpl.html
|
|
|
|
|
* <p>
|
|
|
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
|
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
|
* See the License for the specific language governing permissions and
|
|
|
|
|
* limitations under the License.
|
|
|
|
|
*/
|
|
|
|
|
package org.springblade.core.swagger;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
import com.github.xiaoymin.swaggerbootstrapui.annotations.EnableSwaggerBootstrapUI;
|
|
|
|
|
import com.google.common.base.Predicate;
|
|
|
|
|
import com.google.common.base.Predicates;
|
|
|
|
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
|
2018-12-24 12:47:07 +08:00
|
|
|
|
import org.springframework.boot.context.properties.EnableConfigurationProperties;
|
2018-12-24 11:58:45 +08:00
|
|
|
|
import org.springframework.context.annotation.Bean;
|
|
|
|
|
import org.springframework.context.annotation.Configuration;
|
|
|
|
|
import org.springframework.context.annotation.Profile;
|
|
|
|
|
import springfox.documentation.builders.ApiInfoBuilder;
|
|
|
|
|
import springfox.documentation.builders.PathSelectors;
|
|
|
|
|
import springfox.documentation.builders.RequestHandlerSelectors;
|
|
|
|
|
import springfox.documentation.service.*;
|
|
|
|
|
import springfox.documentation.spi.DocumentationType;
|
|
|
|
|
import springfox.documentation.spi.service.contexts.SecurityContext;
|
|
|
|
|
import springfox.documentation.spring.web.plugins.Docket;
|
|
|
|
|
import springfox.documentation.swagger2.annotations.EnableSwagger2;
|
|
|
|
|
|
|
|
|
|
import java.util.ArrayList;
|
|
|
|
|
import java.util.Collections;
|
|
|
|
|
import java.util.List;
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* swagger配置
|
2018-12-27 13:29:11 +08:00
|
|
|
|
*
|
2018-12-29 18:43:11 +08:00
|
|
|
|
* @author Chill
|
2018-12-24 11:58:45 +08:00
|
|
|
|
*/
|
|
|
|
|
@Configuration
|
|
|
|
|
@EnableSwagger2
|
|
|
|
|
@EnableSwaggerBootstrapUI
|
|
|
|
|
@Profile({"dev", "test"})
|
2018-12-24 12:47:07 +08:00
|
|
|
|
@EnableConfigurationProperties(SwaggerProperties.class)
|
2018-12-24 11:58:45 +08:00
|
|
|
|
public class SwaggerAutoConfiguration {
|
|
|
|
|
|
2018-12-27 13:29:11 +08:00
|
|
|
|
private static final String DEFAULT_EXCLUDE_PATH = "/error";
|
|
|
|
|
private static final String BASE_PATH = "/**";
|
2018-12-24 11:58:45 +08:00
|
|
|
|
|
2018-12-27 13:29:11 +08:00
|
|
|
|
@Bean
|
|
|
|
|
@ConditionalOnMissingBean
|
|
|
|
|
public SwaggerProperties swaggerProperties() {
|
|
|
|
|
return new SwaggerProperties();
|
|
|
|
|
}
|
2018-12-24 11:58:45 +08:00
|
|
|
|
|
2018-12-27 13:29:11 +08:00
|
|
|
|
@Bean
|
|
|
|
|
public Docket api(SwaggerProperties swaggerProperties) {
|
|
|
|
|
// base-path处理
|
|
|
|
|
if (swaggerProperties.getBasePath().size() == 0) {
|
|
|
|
|
swaggerProperties.getBasePath().add(BASE_PATH);
|
|
|
|
|
}
|
|
|
|
|
//noinspection unchecked
|
|
|
|
|
List<Predicate<String>> basePath = new ArrayList();
|
|
|
|
|
swaggerProperties.getBasePath().forEach(path -> basePath.add(PathSelectors.ant(path)));
|
2018-12-24 11:58:45 +08:00
|
|
|
|
|
2018-12-27 13:29:11 +08:00
|
|
|
|
// exclude-path处理
|
|
|
|
|
if (swaggerProperties.getExcludePath().size() == 0) {
|
|
|
|
|
swaggerProperties.getExcludePath().add(DEFAULT_EXCLUDE_PATH);
|
|
|
|
|
}
|
|
|
|
|
List<Predicate<String>> excludePath = new ArrayList<>();
|
|
|
|
|
swaggerProperties.getExcludePath().forEach(path -> excludePath.add(PathSelectors.ant(path)));
|
2018-12-24 11:58:45 +08:00
|
|
|
|
|
2018-12-27 13:29:11 +08:00
|
|
|
|
//noinspection Guava
|
|
|
|
|
return new Docket(DocumentationType.SWAGGER_2)
|
|
|
|
|
.host(swaggerProperties.getHost())
|
|
|
|
|
.apiInfo(apiInfo(swaggerProperties)).select()
|
|
|
|
|
.apis(RequestHandlerSelectors.basePackage(swaggerProperties.getBasePackage()))
|
|
|
|
|
.paths(Predicates.and(Predicates.not(Predicates.or(excludePath)), Predicates.or(basePath)))
|
|
|
|
|
.build()
|
|
|
|
|
.securitySchemes(Collections.singletonList(securitySchema()))
|
|
|
|
|
.securityContexts(Collections.singletonList(securityContext()))
|
|
|
|
|
.pathMapping("/");
|
|
|
|
|
}
|
2018-12-24 11:58:45 +08:00
|
|
|
|
|
2018-12-27 13:29:11 +08:00
|
|
|
|
/**
|
|
|
|
|
* 配置默认的全局鉴权策略的开关,通过正则表达式进行匹配;默认匹配所有URL
|
|
|
|
|
*
|
|
|
|
|
* @return
|
|
|
|
|
*/
|
|
|
|
|
private SecurityContext securityContext() {
|
|
|
|
|
return SecurityContext.builder()
|
|
|
|
|
.securityReferences(defaultAuth())
|
|
|
|
|
.forPaths(PathSelectors.regex(swaggerProperties().getAuthorization().getAuthRegex()))
|
|
|
|
|
.build();
|
|
|
|
|
}
|
2018-12-24 11:58:45 +08:00
|
|
|
|
|
2018-12-27 13:29:11 +08:00
|
|
|
|
/**
|
|
|
|
|
* 默认的全局鉴权策略
|
|
|
|
|
*
|
|
|
|
|
* @return
|
|
|
|
|
*/
|
|
|
|
|
private List<SecurityReference> defaultAuth() {
|
|
|
|
|
ArrayList<AuthorizationScope> authorizationScopeList = new ArrayList<>();
|
|
|
|
|
swaggerProperties().getAuthorization().getAuthorizationScopeList().forEach(authorizationScope -> authorizationScopeList.add(new AuthorizationScope(authorizationScope.getScope(), authorizationScope.getDescription())));
|
|
|
|
|
AuthorizationScope[] authorizationScopes = new AuthorizationScope[authorizationScopeList.size()];
|
|
|
|
|
return Collections.singletonList(SecurityReference.builder()
|
|
|
|
|
.reference(swaggerProperties().getAuthorization().getName())
|
|
|
|
|
.scopes(authorizationScopeList.toArray(authorizationScopes))
|
|
|
|
|
.build());
|
|
|
|
|
}
|
2018-12-24 11:58:45 +08:00
|
|
|
|
|
|
|
|
|
|
2018-12-27 13:29:11 +08:00
|
|
|
|
private OAuth securitySchema() {
|
|
|
|
|
ArrayList<AuthorizationScope> authorizationScopeList = new ArrayList<>();
|
|
|
|
|
swaggerProperties().getAuthorization().getAuthorizationScopeList().forEach(authorizationScope -> authorizationScopeList.add(new AuthorizationScope(authorizationScope.getScope(), authorizationScope.getDescription())));
|
|
|
|
|
ArrayList<GrantType> grantTypes = new ArrayList<>();
|
|
|
|
|
swaggerProperties().getAuthorization().getTokenUrlList().forEach(tokenUrl -> grantTypes.add(new ResourceOwnerPasswordCredentialsGrant(tokenUrl)));
|
|
|
|
|
return new OAuth(swaggerProperties().getAuthorization().getName(), authorizationScopeList, grantTypes);
|
|
|
|
|
}
|
2018-12-24 11:58:45 +08:00
|
|
|
|
|
2018-12-27 13:29:11 +08:00
|
|
|
|
private ApiInfo apiInfo(SwaggerProperties swaggerProperties) {
|
|
|
|
|
return new ApiInfoBuilder()
|
|
|
|
|
.title(swaggerProperties.getTitle())
|
|
|
|
|
.description(swaggerProperties.getDescription())
|
|
|
|
|
.license(swaggerProperties.getLicense())
|
|
|
|
|
.licenseUrl(swaggerProperties.getLicenseUrl())
|
|
|
|
|
.termsOfServiceUrl(swaggerProperties.getTermsOfServiceUrl())
|
|
|
|
|
.contact(new Contact(swaggerProperties.getContact().getName(), swaggerProperties.getContact().getUrl(), swaggerProperties.getContact().getEmail()))
|
|
|
|
|
.version(swaggerProperties.getVersion())
|
|
|
|
|
.build();
|
|
|
|
|
}
|
2018-12-24 11:58:45 +08:00
|
|
|
|
|
|
|
|
|
}
|