package com.airlock.iam.saml2.application.configuration.assertion.attribute;

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.StringProperty;
import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;

@Plugin(description = "A SAML 2.0 attribute containing the username to propagate.", displayName = "Username SAML 2.0 Attribute")
@PluginLicenseRestriction(moduleAffiliation = "SamlIdp")
public class UsernameAttributeConfig extends SamlIdpAttributeConfigWithNameFormat implements SamlFlowIdpAttributeConfig {
   @StringProperty(
      displayName = "Attribute Name",
      description = "The name of the attribute to add to the assertion.",
      exampleValues = {"Username", "AUTH_USER"},
      optional = false
   )
   private String samlAttributeName;

   public UsernameAttributeConfig(String samlAttributeName, String nameFormat) {
      super(nameFormat);
      this.samlAttributeName = samlAttributeName;
   }

   @Override
   public String getSamlAttributeName() {
      return this.samlAttributeName;
   }

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

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

   @SuppressFBWarnings(justification = "generated code")
   public static class UsernameAttributeConfigBuilder {
      @SuppressFBWarnings(justification = "generated code")
      private String samlAttributeName;
      @SuppressFBWarnings(justification = "generated code")
      private String nameFormat;

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

      @SuppressFBWarnings(justification = "generated code")
      public UsernameAttributeConfig.UsernameAttributeConfigBuilder samlAttributeName(String samlAttributeName) {
         this.samlAttributeName = samlAttributeName;
         return this;
      }

      @SuppressFBWarnings(justification = "generated code")
      public UsernameAttributeConfig.UsernameAttributeConfigBuilder nameFormat(String nameFormat) {
         this.nameFormat = nameFormat;
         return this;
      }

      @SuppressFBWarnings(justification = "generated code")
      public UsernameAttributeConfig build() {
         return new UsernameAttributeConfig(this.samlAttributeName, this.nameFormat);
      }

      @SuppressFBWarnings(justification = "generated code")
      @Override
      public String toString() {
         return "UsernameAttributeConfig.UsernameAttributeConfigBuilder(samlAttributeName=" + this.samlAttributeName + ", nameFormat=" + this.nameFormat + ")";
      }
   }
}
