| | |
|---|
| | | package com.ruoyi.framework.config; |
|---|
| | | |
|---|
| | | import cn.dev33.satoken.SaManager; |
|---|
| | | import com.github.xiaoymin.knife4j.spring.annotations.EnableKnife4j; |
|---|
| | | import com.ruoyi.common.utils.StringUtils; |
|---|
| | | import com.ruoyi.common.utils.spring.SpringUtils; |
|---|
| | | import com.ruoyi.framework.config.properties.SwaggerProperties; |
|---|
| | | import io.swagger.models.auth.In; |
|---|
| | | import org.springframework.beans.factory.annotation.Autowired; |
|---|
| | | import io.swagger.v3.oas.models.OpenAPI; |
|---|
| | | import io.swagger.v3.oas.models.info.Info; |
|---|
| | | import lombok.RequiredArgsConstructor; |
|---|
| | | import org.springdoc.core.SpringDocConfiguration; |
|---|
| | | import org.springframework.boot.autoconfigure.AutoConfigureBefore; |
|---|
| | | import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean; |
|---|
| | | import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; |
|---|
| | | import org.springframework.context.annotation.Bean; |
|---|
| | | import org.springframework.context.annotation.Configuration; |
|---|
| | | 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 javax.annotation.PostConstruct; |
|---|
| | | import java.util.ArrayList; |
|---|
| | | import java.util.List; |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * Swagger 文档配置 |
|---|
| | | * |
|---|
| | | * @author Lion Li |
|---|
| | | */ |
|---|
| | | @RequiredArgsConstructor |
|---|
| | | @Configuration |
|---|
| | | @EnableKnife4j |
|---|
| | | @AutoConfigureBefore(SpringDocConfiguration.class) |
|---|
| | | @ConditionalOnProperty(name = "swagger.enabled", havingValue = "true", matchIfMissing = true) |
|---|
| | | public class SwaggerConfig { |
|---|
| | | |
|---|
| | | @Autowired |
|---|
| | | private SwaggerProperties swaggerProperties; |
|---|
| | | private final SwaggerProperties swaggerProperties; |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * 创建API |
|---|
| | | */ |
|---|
| | | @PostConstruct |
|---|
| | | public void createRestApi() { |
|---|
| | | for (SwaggerProperties.Groups group : swaggerProperties.getGroups()) { |
|---|
| | | String basePackage = group.getBasePackage(); |
|---|
| | | Docket docket = new Docket(DocumentationType.OAS_30) |
|---|
| | | .enable(swaggerProperties.getEnabled()) |
|---|
| | | // 用来创建该API的基本信息，展示在文档的页面中（自定义展示的信息） |
|---|
| | | .apiInfo(apiInfo()) |
|---|
| | | // 设置哪些接口暴露给Swagger展示 |
|---|
| | | .select() |
|---|
| | | // 扫描所有有注解的api，用这种方式更灵活 |
|---|
| | | //.apis(RequestHandlerSelectors.withMethodAnnotation(ApiOperation.class)) |
|---|
| | | // 扫描指定包中的swagger注解 |
|---|
| | | .apis(RequestHandlerSelectors.basePackage(basePackage)) |
|---|
| | | // 扫描所有 .apis(RequestHandlerSelectors.any()) |
|---|
| | | .paths(PathSelectors.any()) |
|---|
| | | .build() |
|---|
| | | .groupName(group.getName()) |
|---|
| | | // 设置安全模式，swagger可以设置访问token |
|---|
| | | .securitySchemes(securitySchemes()) |
|---|
| | | .securityContexts(securityContexts()) |
|---|
| | | .pathMapping(swaggerProperties.getPathMapping()); |
|---|
| | | String beanName = StringUtils.substringAfterLast(basePackage, ".") + "Docket"; |
|---|
| | | SpringUtils.registerBean(beanName, docket); |
|---|
| | | } |
|---|
| | | @Bean |
|---|
| | | @ConditionalOnMissingBean(OpenAPI.class) |
|---|
| | | public OpenAPI openApi() { |
|---|
| | | OpenAPI openApi = new OpenAPI(); |
|---|
| | | // 文档基本信息 |
|---|
| | | SwaggerProperties.InfoProperties infoProperties = swaggerProperties.getInfo(); |
|---|
| | | Info info = convertInfo(infoProperties); |
|---|
| | | openApi.info(info); |
|---|
| | | // 扩展文档信息 |
|---|
| | | openApi.externalDocs(swaggerProperties.getExternalDocs()); |
|---|
| | | openApi.tags(swaggerProperties.getTags()); |
|---|
| | | openApi.paths(swaggerProperties.getPaths()); |
|---|
| | | openApi.components(swaggerProperties.getComponents()); |
|---|
| | | |
|---|
| | | return openApi; |
|---|
| | | } |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * 安全模式，这里指定token通过Authorization头请求头传递 |
|---|
| | | */ |
|---|
| | | private List<SecurityScheme> securitySchemes() { |
|---|
| | | List<SecurityScheme> apiKeyList = new ArrayList<SecurityScheme>(); |
|---|
| | | String header = SaManager.getConfig().getTokenName(); |
|---|
| | | apiKeyList.add(new ApiKey(header, header, In.HEADER.toValue())); |
|---|
| | | return apiKeyList; |
|---|
| | | } |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * 安全上下文 |
|---|
| | | */ |
|---|
| | | private List<SecurityContext> securityContexts() { |
|---|
| | | List<SecurityContext> securityContexts = new ArrayList<>(); |
|---|
| | | securityContexts.add( |
|---|
| | | SecurityContext.builder() |
|---|
| | | .securityReferences(defaultAuth()) |
|---|
| | | .operationSelector(o -> o.requestMappingPattern().matches("/.*")) |
|---|
| | | .build()); |
|---|
| | | return securityContexts; |
|---|
| | | } |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * 默认的安全上引用 |
|---|
| | | */ |
|---|
| | | private List<SecurityReference> defaultAuth() { |
|---|
| | | AuthorizationScope authorizationScope = new AuthorizationScope("global", "accessEverything"); |
|---|
| | | AuthorizationScope[] authorizationScopes = new AuthorizationScope[1]; |
|---|
| | | authorizationScopes[0] = authorizationScope; |
|---|
| | | List<SecurityReference> securityReferences = new ArrayList<>(); |
|---|
| | | securityReferences.add(new SecurityReference(SaManager.getConfig().getTokenName(), authorizationScopes)); |
|---|
| | | return securityReferences; |
|---|
| | | } |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * 添加摘要信息 |
|---|
| | | */ |
|---|
| | | private ApiInfo apiInfo() { |
|---|
| | | // 用ApiInfoBuilder进行定制 |
|---|
| | | SwaggerProperties.Contact contact = swaggerProperties.getContact(); |
|---|
| | | return new ApiInfoBuilder() |
|---|
| | | // 设置标题 |
|---|
| | | .title(swaggerProperties.getTitle()) |
|---|
| | | // 描述 |
|---|
| | | .description(swaggerProperties.getDescription()) |
|---|
| | | // 作者信息 |
|---|
| | | .contact(new Contact(contact.getName(), contact.getUrl(), contact.getEmail())) |
|---|
| | | // 版本 |
|---|
| | | .version(swaggerProperties.getVersion()) |
|---|
| | | .build(); |
|---|
| | | private Info convertInfo(SwaggerProperties.InfoProperties infoProperties) { |
|---|
| | | Info info = new Info(); |
|---|
| | | info.setTitle(infoProperties.getTitle()); |
|---|
| | | info.setDescription(infoProperties.getDescription()); |
|---|
| | | info.setContact(infoProperties.getContact()); |
|---|
| | | info.setLicense(infoProperties.getLicense()); |
|---|
| | | info.setVersion(infoProperties.getVersion()); |
|---|
| | | return info; |
|---|
| | | } |
|---|
| | | } |
|---|
