package com.airlock.iam.authentication.infrastructure.idpropagation;

import com.airlock.iam.common.domain.model.request.ContextPath;
import com.airlock.iam.oauth2.api.domain.model.OpenIdConnectSessionManagementSettings;
import com.airlock.iam.oauth2.domain.service.OpenIdProviderUserAgentStateService;
import com.airlock.iam.oauth2.infrastructure.provider.OpenIDConnectOpuasCookieWriter;
import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
import jakarta.inject.Provider;
import jakarta.servlet.http.HttpServletResponse;

public class OpenIdSessionManagementStatePropagator implements OAuth2AdditionalStatePropagator {
   private final OpenIdConnectSessionManagementSettings sessionManagementSettings;
   private final Provider<OpenIdProviderUserAgentStateService> stateService;
   private final HttpServletResponse response;
   private final ContextPath contextPath;

   @Override
   public void propagate() {
      ((OpenIdProviderUserAgentStateService)this.stateService.get())
         .state()
         .ifPresent(
            state -> OpenIDConnectOpuasCookieWriter.writeCookie(this.response, this.contextPath, state, this.sessionManagementSettings.getSameSitePolicy())
         );
   }

   @SuppressFBWarnings(justification = "generated code")
   OpenIdSessionManagementStatePropagator(
      OpenIdConnectSessionManagementSettings sessionManagementSettings,
      Provider<OpenIdProviderUserAgentStateService> stateService,
      HttpServletResponse response,
      ContextPath contextPath
   ) {
      this.sessionManagementSettings = sessionManagementSettings;
      this.stateService = stateService;
      this.response = response;
      this.contextPath = contextPath;
   }

   @SuppressFBWarnings(justification = "generated code")
   public static OpenIdSessionManagementStatePropagator.OpenIdSessionManagementStatePropagatorBuilder builder() {
      return new OpenIdSessionManagementStatePropagator.OpenIdSessionManagementStatePropagatorBuilder();
   }

   @SuppressFBWarnings(justification = "generated code")
   public static class OpenIdSessionManagementStatePropagatorBuilder {
      @SuppressFBWarnings(justification = "generated code")
      private OpenIdConnectSessionManagementSettings sessionManagementSettings;
      @SuppressFBWarnings(justification = "generated code")
      private Provider<OpenIdProviderUserAgentStateService> stateService;
      @SuppressFBWarnings(justification = "generated code")
      private HttpServletResponse response;
      @SuppressFBWarnings(justification = "generated code")
      private ContextPath contextPath;

      @SuppressFBWarnings(justification = "generated code")
      OpenIdSessionManagementStatePropagatorBuilder() {
      }

      @SuppressFBWarnings(justification = "generated code")
      public OpenIdSessionManagementStatePropagator.OpenIdSessionManagementStatePropagatorBuilder sessionManagementSettings(
         OpenIdConnectSessionManagementSettings sessionManagementSettings
      ) {
         this.sessionManagementSettings = sessionManagementSettings;
         return this;
      }

      @SuppressFBWarnings(justification = "generated code")
      public OpenIdSessionManagementStatePropagator.OpenIdSessionManagementStatePropagatorBuilder stateService(
         Provider<OpenIdProviderUserAgentStateService> stateService
      ) {
         this.stateService = stateService;
         return this;
      }

      @SuppressFBWarnings(justification = "generated code")
      public OpenIdSessionManagementStatePropagator.OpenIdSessionManagementStatePropagatorBuilder response(HttpServletResponse response) {
         this.response = response;
         return this;
      }

      @SuppressFBWarnings(justification = "generated code")
      public OpenIdSessionManagementStatePropagator.OpenIdSessionManagementStatePropagatorBuilder contextPath(ContextPath contextPath) {
         this.contextPath = contextPath;
         return this;
      }

      @SuppressFBWarnings(justification = "generated code")
      public OpenIdSessionManagementStatePropagator build() {
         return new OpenIdSessionManagementStatePropagator(this.sessionManagementSettings, this.stateService, this.response, this.contextPath);
      }

      @SuppressFBWarnings(justification = "generated code")
      @Override
      public String toString() {
         return "OpenIdSessionManagementStatePropagator.OpenIdSessionManagementStatePropagatorBuilder(sessionManagementSettings="
            + this.sessionManagementSettings
            + ", stateService="
            + this.stateService
            + ", response="
            + this.response
            + ", contextPath="
            + this.contextPath
            + ")";
      }
   }
}