com.github.xiaoymin.knife4j.spring.configuration.Knife4jAutoConfiguration Maven / Gradle / Ivy
/*
* Copyright © 2017-2023 Knife4j([email protected])
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* 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 com.github.xiaoymin.knife4j.spring.configuration;
import com.github.xiaoymin.knife4j.core.conf.GlobalConstants;
import com.github.xiaoymin.knife4j.extend.filter.basic.ServletSecurityBasicAuthFilter;
import com.github.xiaoymin.knife4j.spring.extension.Knife4jOpenApiCustomizer;
import com.github.xiaoymin.knife4j.spring.extension.Knife4jOperationCustomizer;
import com.github.xiaoymin.knife4j.spring.filter.ProductionSecurityFilter;
import com.github.xiaoymin.knife4j.spring.util.EnvironmentUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springdoc.core.SpringDocConfigProperties;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.env.Environment;
import org.springframework.web.cors.CorsConfiguration;
import org.springframework.web.cors.UrlBasedCorsConfigurationSource;
import org.springframework.web.filter.CorsFilter;
/***
* Knife4j 基础自动配置类
* @since 2.0.0
* @author [email protected]
* 2019/08/28 21:08
*/
@Configuration
@EnableConfigurationProperties({Knife4jProperties.class, Knife4jHttpBasic.class, Knife4jSetting.class})
@ConditionalOnProperty(name = "knife4j.enable", havingValue = "true")
public class Knife4jAutoConfiguration {
@Autowired
private Environment environment;
Logger logger = LoggerFactory.getLogger(Knife4jAutoConfiguration.class);
/**
* 增强自定义配置
* @param knife4jProperties
* @param docProperties
* @return
*/
@Bean
@ConditionalOnMissingBean
public Knife4jOpenApiCustomizer knife4jOpenApiCustomizer(Knife4jProperties knife4jProperties,
SpringDocConfigProperties docProperties) {
logger.debug("Register Knife4jOpenApiCustomizer");
return new Knife4jOpenApiCustomizer(knife4jProperties, docProperties);
}
@Bean
@ConditionalOnMissingBean
public Knife4jOperationCustomizer knife4jOperationCustomizer() {
return new Knife4jOperationCustomizer();
}
/**
* 配置Cors
* @since 2.0.4
* @return
*/
@Bean("knife4jCorsFilter")
@ConditionalOnMissingBean(CorsFilter.class)
@ConditionalOnProperty(name = "knife4j.cors", havingValue = "true")
public CorsFilter corsFilter() {
logger.info("init CorsFilter...");
UrlBasedCorsConfigurationSource source = new UrlBasedCorsConfigurationSource();
CorsConfiguration corsConfiguration = new CorsConfiguration();
corsConfiguration.setAllowCredentials(true);
corsConfiguration.addAllowedOrigin("*");
corsConfiguration.addAllowedHeader("*");
corsConfiguration.addAllowedMethod("*");
corsConfiguration.setMaxAge(10000L);
// 匹配所有API
source.registerCorsConfiguration("/**", corsConfiguration);
CorsFilter corsFilter = new CorsFilter(source);
return corsFilter;
}
/**
* Security with Basic Http
* @param knife4jProperties Basic Properties
* @return BasicAuthFilter
*/
@Bean
@ConditionalOnMissingBean(ServletSecurityBasicAuthFilter.class)
@ConditionalOnProperty(name = "knife4j.basic.enable", havingValue = "true")
public ServletSecurityBasicAuthFilter securityBasicAuthFilter(Knife4jProperties knife4jProperties) {
ServletSecurityBasicAuthFilter authFilter = new ServletSecurityBasicAuthFilter();
if (knife4jProperties == null) {
authFilter.setEnableBasicAuth(EnvironmentUtils.resolveBool(environment, "knife4j.basic.enable", Boolean.FALSE));
authFilter.setUserName(EnvironmentUtils.resolveString(environment, "knife4j.basic.username", GlobalConstants.BASIC_DEFAULT_USERNAME));
authFilter.setPassword(EnvironmentUtils.resolveString(environment, "knife4j.basic.password", GlobalConstants.BASIC_DEFAULT_PASSWORD));
} else {
// 判断非空
if (knife4jProperties.getBasic() == null) {
authFilter.setEnableBasicAuth(Boolean.FALSE);
authFilter.setUserName(GlobalConstants.BASIC_DEFAULT_USERNAME);
authFilter.setPassword(GlobalConstants.BASIC_DEFAULT_PASSWORD);
} else {
authFilter.setEnableBasicAuth(knife4jProperties.getBasic().isEnable());
authFilter.setUserName(knife4jProperties.getBasic().getUsername());
authFilter.setPassword(knife4jProperties.getBasic().getPassword());
authFilter.addRule(knife4jProperties.getBasic().getInclude());
}
}
return authFilter;
}
@Bean
@ConditionalOnMissingBean(ProductionSecurityFilter.class)
@ConditionalOnProperty(name = "knife4j.production", havingValue = "true")
public ProductionSecurityFilter productionSecurityFilter(Knife4jProperties knife4jProperties) {
boolean prod = false;
ProductionSecurityFilter p = null;
if (knife4jProperties == null) {
if (environment != null) {
String prodStr = environment.getProperty("knife4j.production");
if (logger.isDebugEnabled()) {
logger.debug("swagger.production:{}", prodStr);
}
prod = Boolean.valueOf(prodStr);
}
p = new ProductionSecurityFilter(prod);
} else {
p = new ProductionSecurityFilter(knife4jProperties.isProduction());
}
return p;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy