package com.airlock.iam.saml2.domain.model.assertion.attribute;

import com.airlock.iam.base.api.domain.model.plugin.LogicForConfig;
import com.airlock.iam.saml2.application.configuration.assertion.attribute.AuthnClassRefAttributeConfig;
import com.airlock.iam.saml2.domain.model.assertion.FlowSaml2Attribute;
import com.airlock.iam.saml2.domain.model.idp.Saml2AuthenticationContextClassRef;
import jakarta.inject.Inject;
import jakarta.inject.Provider;
import java.util.Optional;

@LogicForConfig(AuthnClassRefAttributeConfig.class)
public class Saml2AuthnClassRefAttributeImpl implements Saml2AttributeProvider {
   private final AuthnClassRefAttributeConfig config;
   private final Provider<Optional<Saml2AuthenticationContextClassRef>> authenticationContext;

   @Inject
   public Saml2AuthnClassRefAttributeImpl(AuthnClassRefAttributeConfig config, Provider<Optional<Saml2AuthenticationContextClassRef>> authenticationContext) {
      this.config = config;
      this.authenticationContext = authenticationContext;
   }

   @Override
   public Optional<FlowSaml2Attribute> getAttribute() {
      return ((Optional)this.authenticationContext.get())
         .map(classRef -> new FlowSaml2Attribute(this.config.getSamlAttributeName(), classRef.value(), this.config.getNameFormat()));
   }
}