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

import com.airlock.iam.base.api.domain.model.plugin.LogicForConfig;
import com.airlock.iam.flow.shared.domain.model.attributes.AuthTokenIdFlowSessionAttribute;
import com.airlock.iam.saml2.application.configuration.assertion.attribute.AuthTokenIdAttributeConfig;
import com.airlock.iam.saml2.domain.model.assertion.FlowSaml2Attribute;
import jakarta.inject.Inject;
import jakarta.inject.Provider;
import java.util.Optional;

@LogicForConfig(AuthTokenIdAttributeConfig.class)
public class Saml2AuthTokenIdAttributeImpl implements Saml2AttributeProvider {
   private final AuthTokenIdAttributeConfig config;
   private final Provider<Optional<AuthTokenIdFlowSessionAttribute>> authTokenId;

   @Inject
   public Saml2AuthTokenIdAttributeImpl(AuthTokenIdAttributeConfig config, Provider<Optional<AuthTokenIdFlowSessionAttribute>> authTokenId) {
      this.config = config;
      this.authTokenId = authTokenId;
   }

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