package com.airlock.iam.flow.shared.application.service.mtan;

import com.airlock.iam.common.api.domain.model.otp.UnverifiedOtp;
import com.airlock.iam.flow.api.domain.model.flow.FlowResult;
import com.airlock.iam.flow.api.domain.model.step.StepResult;
import com.airlock.iam.flow.domain.model.flow.FlowProvider;
import com.airlock.iam.flow.shared.domain.model.mtan.MtanOtpCheckingStep;
import jakarta.inject.Inject;
import java.util.function.Function;

public final class MtanOtpCheck {
   private final FlowProvider flowProvider;

   @Inject
   public MtanOtpCheck(FlowProvider flowProvider) {
      this.flowProvider = flowProvider;
   }

   public FlowResult checkOtp(UnverifiedOtp otp) {
      return this.executeStepMethod(step -> step.checkOtp(otp));
   }

   public FlowResult getNumberChoice() {
      return this.executeStepMethod(MtanOtpCheckingStep::getNumberChoice);
   }

   public FlowResult selectMtanToken(String mtanTokenId) {
      return this.executeStepMethod(step -> step.selectMtanToken(mtanTokenId));
   }

   public FlowResult resendOtp() {
      return this.executeStepMethod(MtanOtpCheckingStep::resendOtp);
   }

   public FlowResult resendOtpPossible() {
      return this.executeStepMethod(MtanOtpCheckingStep::resendOtpPossible);
   }

   private FlowResult executeStepMethod(Function<MtanOtpCheckingStep, StepResult> stepMethod) {
      return this.flowProvider.getCurrentOrStartNew().execute(stepMethod, MtanOtpCheckingStep.class);
   }
}