com.taotao.cloud.useragent.configuration.UserAgentConfiguration Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2020-2030, Shuigedeng ([email protected] & https://blog.taotaocloud.top/).
 *
 * 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
 *
 *      https://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.taotao.cloud.useragent.configuration;

import com.taotao.cloud.useragent.properties.UserAgentProperties;
import com.taotao.cloud.useragent.reactive.ReactiveUserAgentFilter;
import com.taotao.cloud.useragent.reactive.ReactiveUserAgentResolver;
import com.taotao.cloud.useragent.servlet.UserAgentFilter;
import com.taotao.cloud.useragent.servlet.UserAgentResolver;
import com.taotao.cloud.useragent.support.UserAgentConverter;
import java.util.List;
import org.springframework.boot.autoconfigure.AutoConfiguration;
import org.springframework.boot.autoconfigure.condition.ConditionalOnBean;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.boot.autoconfigure.condition.ConditionalOnWebApplication;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.reactive.config.WebFluxConfigurer;
import org.springframework.web.reactive.result.method.annotation.ArgumentResolverConfigurer;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;

/**
 * The type User agent auto configuration.
 *
 * @author shuigedeng
 * @version 2023.04
 * @since 2023-06-15 09:21:48
 */
@AutoConfiguration
@ConditionalOnBean(UserAgentConverter.class)
@ConditionalOnProperty(prefix = UserAgentProperties.PREFIX, name = "enabled", havingValue = "true")
public class UserAgentConfiguration {

    /**
     * The type User agent mvc auto configuration.
     */
    @AutoConfiguration
    @ConditionalOnWebApplication(type = ConditionalOnWebApplication.Type.SERVLET)
    public static class UserAgentMvcAutoConfiguration implements WebMvcConfigurer {

        /**
         * Filter registration bean filter registration bean.
         *
         * @return the filter registration bean
         */
        @Bean
        public FilterRegistrationBean filterRegistrationBean() {
            FilterRegistrationBean registrationBean = new FilterRegistrationBean<>();
            registrationBean.setFilter(new UserAgentFilter());
            registrationBean.addUrlPatterns("/*");
            registrationBean.setName("userAgentFilter");
            registrationBean.setOrder(1);
            return registrationBean;
        }

        @Override
        public void addArgumentResolvers(List resolvers) {
            resolvers.add(new UserAgentResolver());
        }
    }

    /**
     * The type User agent reactive auto configuration.
     */
    @AutoConfiguration
    @ConditionalOnWebApplication(type = ConditionalOnWebApplication.Type.REACTIVE)
    public static class UserAgentReactiveAutoConfiguration implements WebFluxConfigurer {

        /**
         * Reactive user agent filter reactive user agent filter.
         *
         * @return the reactive user agent filter
         */
        @Bean
        public ReactiveUserAgentFilter reactiveUserAgentFilter() {
            return new ReactiveUserAgentFilter();
        }

        @Override
        public void configureArgumentResolvers(ArgumentResolverConfigurer configurer) {
            configurer.addCustomResolver(new ReactiveUserAgentResolver());
        }
    }
}