package com.airlock.iam.saml2.infrastructure.jersey;

import com.airlock.iam.common.api.application.configuration.RestApplicationConfiguration;
import com.airlock.iam.common.api.infrastructure.ApplicationInitializationHook;
import com.airlock.iam.common.infrastructure.event.NoConfiguredEventSubscribersModule;
import com.airlock.iam.common.infrastructure.filter.general.CharsetResponseFilter;
import com.airlock.iam.common.infrastructure.filter.general.CombinedRequestDelayAndIamSessionFilter;
import com.airlock.iam.common.infrastructure.filter.general.ConfigurationLicensedFilter;
import com.airlock.iam.common.infrastructure.filter.general.ConfigurationPresentFilter;
import com.airlock.iam.common.infrastructure.filter.general.ContentSecurityPolicyJerseyFilter;
import com.airlock.iam.common.infrastructure.filter.general.EnterExitFilter;
import com.airlock.iam.common.infrastructure.filter.general.LegacyInjectorInitFilter;
import com.airlock.iam.common.infrastructure.filter.general.LogContextFilter;
import com.airlock.iam.common.infrastructure.filter.general.MediaTypeNormalizationFilter;
import com.airlock.iam.common.infrastructure.filter.general.NoCacheFilter;
import com.airlock.iam.common.infrastructure.filter.general.SessionTerminationResponseFilter;
import com.airlock.iam.common.infrastructure.filter.general.SessionTimeoutResponseFilter;
import com.airlock.iam.common.infrastructure.filter.general.SystemAvailabilityFilter;
import com.airlock.iam.flow.shared.infrastructure.NoFlowModule;
import com.airlock.iam.flow.shared.infrastructure.factory.FlowAttributesModule;
import com.airlock.iam.flow.shared.infrastructure.factory.SingleRequestPerUserServiceModule;
import com.airlock.iam.flow.shared.infrastructure.factory.UserSessionModule;
import com.airlock.iam.login.api.infrastructure.Saml2Extension;
import com.airlock.iam.saml2.infrastructure.jersey.filter.Saml2FlowConfigPresentFilter;
import com.airlock.iam.saml2.infrastructure.jersey.filter.Saml2RequestThreadContextFilter;
import com.airlock.iam.saml2.infrastructure.resource.Saml2IdpArtifactResolutionResource;
import com.airlock.iam.saml2.infrastructure.resource.Saml2IdpSloResource;
import com.airlock.iam.saml2.infrastructure.resource.Saml2IdpSsoResource;
import com.airlock.iam.saml2.infrastructure.resource.Saml2SpSloResource;
import com.airlock.iam.saml2.infrastructure.resource.Saml2SpSsoResource;
import com.airlock.iam.saml2.infrastructure.state.Saml2StateStoreMapsInitializationHook;
import com.google.inject.Module;
import java.util.Collection;
import java.util.List;
import java.util.stream.Stream;

@RestApplicationConfiguration(
   dependencyProviders = {
         "com.airlock.iam.base.infrastructure.factory",
         "com.airlock.iam.common.infrastructure.factory",
         "com.airlock.iam.login.app.infrastructure.factory.shared",
         "com.airlock.iam.login.rest.infrastructure.factory.shared",
         "com.airlock.iam.saml2.infrastructure.factory",
         "com.airlock.iam.saml2.infrastructure.factory"
   },
   exceptionMappers = "com.airlock.iam.saml2.infrastructure.exception"
)
public final class MainSaml2Extension implements Saml2Extension {
   @Override
   public Collection<Class<?>> jaxRsComponents() {
      return Stream.concat(restFilters(), restResources()).toList();
   }

   private static Stream<Class<?>> restFilters() {
      return Stream.of(
         CharsetResponseFilter.class,
         ConfigurationLicensedFilter.class,
         ConfigurationPresentFilter.class,
         ContentSecurityPolicyJerseyFilter.class,
         EnterExitFilter.class,
         CombinedRequestDelayAndIamSessionFilter.class,
         LegacyInjectorInitFilter.class,
         LogContextFilter.class,
         MediaTypeNormalizationFilter.class,
         NoCacheFilter.class,
         Saml2FlowConfigPresentFilter.class,
         Saml2RequestThreadContextFilter.class,
         SessionTerminationResponseFilter.class,
         SessionTimeoutResponseFilter.class,
         SystemAvailabilityFilter.class
      );
   }

   private static Stream<Class<?>> restResources() {
      return Stream.of(
         Saml2IdpArtifactResolutionResource.class, Saml2IdpSloResource.class, Saml2IdpSsoResource.class, Saml2SpSloResource.class, Saml2SpSsoResource.class
      );
   }

   @Override
   public Collection<Class<? extends Module>> guiceModules() {
      return List.of(
         FlowAttributesModule.class,
         NoConfiguredEventSubscribersModule.class,
         NoFlowModule.class,
         SingleRequestPerUserServiceModule.class,
         UserSessionModule.class
      );
   }

   @Override
   public Collection<Class<? extends ApplicationInitializationHook>> initializationHooks() {
      return List.of(Saml2StateStoreMapsInitializationHook.class);
   }
}