package com.airlock.iam.oauth2.application.configuration.claims;

import com.airlock.iam.common.api.domain.model.claims.Claim;
import com.airlock.iam.oauth2.api.application.configuration.claims.ClaimConditionConfig;
import com.airlock.iam.oauth2.api.application.configuration.claims.CustomClaimConfig;
import com.airlock.iam.oauth2.api.application.configuration.claims.CustomClaimExtractorSettings;
import com.airlock.iam.oauth2.api.domain.model.claims.UserClaimExtractor;
import com.airlock.iam.oauth2.domain.model.claims.custom.CustomClaimSetClaimExtractor;
import com.airlock.iam.plugin.api.annotation.config.Plugin;
import com.airlock.iam.plugin.api.annotation.license.PluginLicenseRestriction;
import com.airlock.iam.plugin.api.annotation.property.PluginListProperty;
import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
import java.util.List;
import java.util.stream.Collectors;

@Plugin(description = "A custom claim introducing a JSON Object in the response.", displayName = "Claim Set Custom Claim")
@PluginLicenseRestriction(moduleAffiliation = "OAuthServer")
public final class CustomClaimSetClaimConfig extends GenericCustomClaimConfig {
   @PluginListProperty(description = "Custom claims to include in the object.", minElements = 1, optional = false)
   private List<GenericCustomClaimConfig> customClaims;

   @Override
   public UserClaimExtractor<?> logic() {
      List<UserClaimExtractor<?>> extractors = this.customClaims.stream().map(CustomClaimConfig::logic).collect(Collectors.toList());
      return new CustomClaimSetClaimExtractor(this::getClaimName, this::getClaimCondition, () -> extractors);
   }

   @Override
   public UserClaimExtractor<? extends Claim<?>> logic(CustomClaimExtractorSettings settings) {
      return new CustomClaimSetClaimExtractor(
         this::getClaimName,
         this::getClaimCondition,
         () -> this.getCustomClaims().stream().map(customClaimConfig -> customClaimConfig.logic(settings)).collect(Collectors.toList())
      );
   }

   public List<GenericCustomClaimConfig> getCustomClaims() {
      return this.customClaims;
   }

   public CustomClaimSetClaimConfig(String claimName, List<GenericCustomClaimConfig> customClaims, ClaimConditionConfig claimCondition) {
      super(claimName, claimCondition);
      this.customClaims = customClaims;
   }

   @SuppressFBWarnings(justification = "generated code")
   public static CustomClaimSetClaimConfig.CustomClaimSetClaimConfigBuilder builder() {
      return new CustomClaimSetClaimConfig.CustomClaimSetClaimConfigBuilder();
   }

   @SuppressFBWarnings(justification = "generated code")
   public CustomClaimSetClaimConfig() {
   }

   @SuppressFBWarnings(justification = "generated code")
   public static class CustomClaimSetClaimConfigBuilder {
      @SuppressFBWarnings(justification = "generated code")
      private String claimName;
      @SuppressFBWarnings(justification = "generated code")
      private List<GenericCustomClaimConfig> customClaims;
      @SuppressFBWarnings(justification = "generated code")
      private ClaimConditionConfig claimCondition;

      @SuppressFBWarnings(justification = "generated code")
      CustomClaimSetClaimConfigBuilder() {
      }

      @SuppressFBWarnings(justification = "generated code")
      public CustomClaimSetClaimConfig.CustomClaimSetClaimConfigBuilder claimName(String claimName) {
         this.claimName = claimName;
         return this;
      }

      @SuppressFBWarnings(justification = "generated code")
      public CustomClaimSetClaimConfig.CustomClaimSetClaimConfigBuilder customClaims(List<GenericCustomClaimConfig> customClaims) {
         this.customClaims = customClaims;
         return this;
      }

      @SuppressFBWarnings(justification = "generated code")
      public CustomClaimSetClaimConfig.CustomClaimSetClaimConfigBuilder claimCondition(ClaimConditionConfig claimCondition) {
         this.claimCondition = claimCondition;
         return this;
      }

      @SuppressFBWarnings(justification = "generated code")
      public CustomClaimSetClaimConfig build() {
         return new CustomClaimSetClaimConfig(this.claimName, this.customClaims, this.claimCondition);
      }

      @SuppressFBWarnings(justification = "generated code")
      @Override
      public String toString() {
         return "CustomClaimSetClaimConfig.CustomClaimSetClaimConfigBuilder(claimName="
            + this.claimName
            + ", customClaims="
            + this.customClaims
            + ", claimCondition="
            + this.claimCondition
            + ")";
      }
   }
}