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

import com.airlock.iam.base.api.domain.model.exception.FailureException;
import com.airlock.iam.base.api.domain.model.plugin.LogicForConfig;
import com.airlock.iam.login.domain.model.session.AuditToken;
import com.airlock.iam.saml2.application.configuration.assertion.attribute.AuditTokenAttributeConfig;
import com.airlock.iam.saml2.domain.model.assertion.FlowSaml2Attribute;
import jakarta.inject.Inject;
import jakarta.inject.Provider;
import java.util.Optional;

@LogicForConfig(AuditTokenAttributeConfig.class)
public class Saml2AuditTokenAttributeImpl implements Saml2AttributeProvider {
   private final AuditTokenAttributeConfig config;
   private final Provider<Optional<AuditToken>> auditToken;

   @Inject
   public Saml2AuditTokenAttributeImpl(AuditTokenAttributeConfig config, Provider<Optional<AuditToken>> auditToken) {
      this.config = config;
      this.auditToken = auditToken;
   }

   @Override
   public Optional<FlowSaml2Attribute> getAttribute() {
      return Optional.of(
         new FlowSaml2Attribute(
            this.config.getSamlAttributeName(),
            ((AuditToken)((Optional)this.auditToken.get()).orElseThrow(() -> new FailureException("No AuditToken available"))).getValue(),
            this.config.getNameFormat()
         )
      );
   }
}