package com.airlock.iam.authentication.infrastructure.factory;

import com.airlock.iam.authentication.api.domain.model.risk.RiskTagFactory;
import com.airlock.iam.authentication.api.domain.model.risk.extractor.MatchingRiskTags;
import com.airlock.iam.authentication.api.domain.model.risk.extractor.MismatchingRiskTags;
import com.airlock.iam.authentication.application.configuration.risk.extractor.anomaly.AnomalyShieldStateRiskExtractorConfig;
import com.airlock.iam.base.api.infrastructure.plugin.ModuleForConfig;
import com.airlock.iam.common.domain.model.session.WafMlAnomaly;
import com.google.inject.Exposed;
import com.google.inject.PrivateModule;
import com.google.inject.Provides;
import java.util.List;

@ModuleForConfig(AnomalyShieldStateRiskExtractorConfig.class)
public final class AnomalyShieldStateRiskExtractorModule extends PrivateModule {
   @Exposed
   @Provides
   List<WafMlAnomaly> provideWafMlAnomalies(AnomalyShieldStateRiskExtractorConfig config) {
      return config.getExpectedAnomalyStates();
   }

   @Exposed
   @Provides
   MatchingRiskTags provideMatchingRiskTags(AnomalyShieldStateRiskExtractorConfig config, RiskTagFactory factory) {
      return factory.getMatchingRiskTags(config.getTagsIfOneOfExpectedAnomalyStates());
   }

   @Exposed
   @Provides
   MismatchingRiskTags provideMismatchingRiskTags(AnomalyShieldStateRiskExtractorConfig config, RiskTagFactory factory) {
      return factory.getMismatchingRiskTags(config.getTagsIfNoneOfExpectedAnomalyStates());
   }

   protected void configure() {
   }
}