package com.airlock.iam.flow.shared.domain.model.cronto;

import com.airlock.iam.core.domain.model.cronto.CrontoOtp;
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 jakarta.inject.Inject;
import java.util.function.Function;

public final class CrontoActivation {
   private final FlowProvider flowProvider;

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

   public FlowResult retrieveFirstActivationChallenge() {
      return this.executeStepMethod(CrontoActivationStep::retrieveFirstActivationChallenge);
   }

   public FlowResult checkFirstActivationOtp(CrontoOtp otp) {
      return this.executeStepMethod(step -> step.checkFirstActivationOtp(otp));
   }

   public FlowResult checkSecondActivationOtp(CrontoOtp otp, String deviceName) {
      return this.executeStepMethod(step -> step.checkSecondActivationOtp(otp, deviceName));
   }

   public FlowResult retrieveSecondActivationChallenge() {
      return this.executeStepMethod(CrontoActivationStep::retrieveSecondActivationChallenge);
   }

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