package com.airlock.iam.publicselfservice.infrastructure.factory;

import com.airlock.iam.base.api.infrastructure.plugin.ModuleForConfig;
import com.airlock.iam.common.application.configuration.continuation.FlowContinuationRepositoryConfig;
import com.airlock.iam.database.infrastructure.DatabaseType;
import com.airlock.iam.database.infrastructure.JooqConfig;
import com.airlock.iam.database.infrastructure.TenantId;
import com.airlock.iam.flow.api.domain.model.flow.FlowId;
import com.airlock.iam.flow.shared.domain.model.continuation.FlowContinuationToken;
import com.airlock.iam.flow.shared.domain.service.continuation.FlowContinuationTokenRepository;
import com.airlock.iam.flow.shared.domain.service.continuation.FlowContinuationTokenService;
import com.airlock.iam.flow.shared.domain.service.continuation.FlowContinuationTokenServiceImpl;
import com.airlock.iam.flow.shared.infrastructure.repository.FlowContinuationTokenRepositoryImpl;
import com.airlock.iam.publicselfservice.application.configuration.flow.PublicSelfServiceFlowConfig;
import com.airlock.iam.publicselfservice.application.configuration.steps.FlowContinuationStepConfig;
import com.airlock.iam.publicselfservice.domain.model.step.FlowContinuationStepSettings;
import com.google.inject.Exposed;
import com.google.inject.PrivateModule;
import com.google.inject.Provides;
import jakarta.servlet.http.HttpServletRequest;
import java.util.List;
import java.util.Optional;

@ModuleForConfig(FlowContinuationStepConfig.class)
public final class FlowContinuationStepModule extends PrivateModule {
   @Provides
   @Exposed
   Optional<FlowContinuationToken> provideFlowContinuationToken(HttpServletRequest request) {
      return Optional.ofNullable(request.getHeader("X-Flow-Continuation-Token")).map(FlowContinuationToken::new);
   }

   @Provides
   @Exposed
   FlowContinuationStepSettings provideFlowContinuationStepSettings(FlowContinuationStepConfig config) {
      return new FlowContinuationStepSettings(config.isConsumesToken());
   }

   @Provides
   JooqConfig provideJooqConfig(FlowContinuationStepConfig config) {
      FlowContinuationRepositoryConfig repositoryConfig = config.getRepository();
      return new JooqConfig(
         repositoryConfig.getSqlDataSource(),
         repositoryConfig.getSqlDataSource().getDbType(),
         repositoryConfig.getSqlDataSource().getDbType() == DatabaseType.ORACLE ? "flow_continuation_token_seq" : null,
         List.of(),
         repositoryConfig.isLogQueries(),
         TenantId.tenantId(repositoryConfig.getTenantId())
      );
   }

   @Provides
   @Exposed
   FlowId provideCurrentFlowId(PublicSelfServiceFlowConfig currentFlowConfig) {
      return currentFlowConfig.getFlowId();
   }

   protected void configure() {
      this.bind(FlowContinuationTokenService.class).to(FlowContinuationTokenServiceImpl.class);
      this.bind(FlowContinuationTokenRepository.class).to(FlowContinuationTokenRepositoryImpl.class);
      this.expose(FlowContinuationTokenService.class);
      this.expose(FlowContinuationTokenRepository.class);
   }
}