content: /*
   * Licensed to the Apache Software Foundation (ASF) under one or more
   * contributor license agreements.  See the NOTICE file distributed with
   * this work for additional information regarding copyright ownership.
   * The ASF licenses this file to You 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 org.apache.dolphinscheduler.api.configuration;
  
  import org.apache.dolphinscheduler.api.interceptor.LocaleChangeInterceptor;
  import org.apache.dolphinscheduler.api.interceptor.LoginHandlerInterceptor;
  import org.apache.dolphinscheduler.api.interceptor.RateLimitInterceptor;
  
  import java.util.Locale;
  
  import org.apache.dolphinscheduler.api.interceptor.UserHandlerInterceptor;
  import org.springframework.beans.factory.annotation.Autowired;
  import org.springframework.context.annotation.Bean;
  import org.springframework.context.annotation.Configuration;
  import org.springframework.web.cors.CorsConfiguration;
  import org.springframework.web.cors.UrlBasedCorsConfigurationSource;
  import org.springframework.web.filter.CorsFilter;
  import org.springframework.web.servlet.LocaleResolver;
  import org.springframework.web.servlet.config.annotation.ContentNegotiationConfigurer;
  import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
  import org.springframework.web.servlet.config.annotation.ResourceHandlerRegistry;
  import org.springframework.web.servlet.config.annotation.ViewControllerRegistry;
  import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
  import org.springframework.web.servlet.i18n.CookieLocaleResolver;
  
  /**
   * application configuration
   */
  @Configuration
  public class AppConfiguration implements WebMvcConfigurer {
  
      public static final String LOGIN_INTERCEPTOR_PATH_PATTERN = "/**/*";
      public static final String LOGIN_PATH_PATTERN = "/login/**";
      public static final String REGISTER_PATH_PATTERN = "/users/register";
      public static final String PATH_PATTERN = "/**";
      public static final String LOCALE_LANGUAGE_COOKIE = "language";
  
      @Autowired
      private ApiConfig apiConfig;
  
      @Bean
      public CorsFilter corsFilter() {
          CorsConfiguration config = new CorsConfiguration();
          config.addAllowedOrigin("*");
          config.addAllowedMethod("*");
          config.addAllowedHeader("*");
          UrlBasedCorsConfigurationSource configSource = new UrlBasedCorsConfigurationSource();
          configSource.registerCorsConfiguration(PATH_PATTERN, config);
          return new CorsFilter(configSource);
      }
  
  //    @Bean
      public LoginHandlerInterceptor loginInterceptor() {
  
          return new LoginHandlerInterceptor();
      }
  
      @Bean
      public UserHandlerInterceptor userInterceptor() {
          return new UserHandlerInterceptor();
      }
  
      /**
       * Cookie
       * @return local resolver
       */
      @Bean(name = "localeResolver")
      public LocaleResolver localeResolver() {
          CookieLocaleResolver localeResolver = new CookieLocaleResolver();
          localeResolver.setCookieName(LOCALE_LANGUAGE_COOKIE);
          // set default locale
          localeResolver.setDefaultLocale(Locale.US);
          // set language tag compliant
          localeResolver.setLanguageTagCompliant(false);
          return localeResolver;
      }
  
      @Bean
      public LocaleChangeInterceptor localeChangeInterceptor() {
          return new LocaleChangeInterceptor();
      }
  
  //    @Bean
      public RateLimitInterceptor createRateLimitInterceptor() {
          return new RateLimitInterceptor(apiConfig.getTrafficControl());
      }
  
      @Override
      public void addInterceptors(InterceptorRegistry registry) {
          // i18n
          registry.addInterceptor(localeChangeInterceptor());
          ApiConfig.TrafficConfiguration trafficControl = apiConfig.getTrafficControl();
          if (trafficControl.isGlobalSwitch() || trafficControl.isTenantSwitch()) {
              registry.addInterceptor(createRateLimitInterceptor());
          }
          registry.addInterceptor(userInterceptor())
                  .addPathPatterns(LOGIN_INTERCEPTOR_PATH_PATTERN)
                  .excludePathPatterns(LOGIN_PATH_PATTERN, REGISTER_PATH_PATTERN,
                          "/swagger-resources/**", "/webjars/**", "/v3/api-docs/**", "/api-docs/**", "/swagger-ui.html",
                          "/doc.html", "/swagger-ui/**", "*.html", "/ui/**", "/error", "/oauth2-provider",
                          "/redirect/login/oauth2", "/cookies");
      }
  
      @Override
      public void addResourceHandlers(ResourceHandlerRegistry registry) {
          registry.addResourceHandler("/static/**").addResourceLocations("classpath:/static/");
          registry.addResourceHandler("doc.html").addResourceLocations("classpath:/META-INF/resources/");
          registry.addResourceHandler("/webjars/**").addResourceLocations("classpath:/META-INF/resources/webjars/");
          registry.addResourceHandler("/ui/**").addResourceLocations("file:ui/");
      }
  
      @Override
      public void addViewControllers(ViewControllerRegistry registry) {
          registry.addViewController("/").setViewName("redirect:/ui/");
          registry.addViewController("/ui/").setViewName("forward:/ui/index.html");
      }
  
      /**
       * Turn off suffix-based content negotiation
       *
       * @param configurer configurer
       */
      @Override
      public void configureContentNegotiation(final ContentNegotiationConfigurer configurer) {
          configurer.favorPathExtension(false);
      }
  }
  