package com.airlock.iam.oauth2.domain.model.session;

import com.airlock.iam.base.misc.json.JsonCoding;
import com.airlock.iam.common.api.domain.model.claims.Claim;
import com.airlock.iam.common.api.domain.model.claims.ClaimValue;
import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public final class OAuth2SessionClaims {
   private Map<String, Object> idTokenSnapshot = Map.of();
   private Map<String, Object> accessTokenSnapshot = Map.of();

   public static OAuth2SessionClaims sessionClaims(List<Claim<?>> idTokenClaims, List<Claim<?>> accessTokenClaims) {
      return new OAuth2SessionClaims(serializeClaims(idTokenClaims), serializeClaims(accessTokenClaims));
   }

   public static OAuth2SessionClaims noClaims() {
      return new OAuth2SessionClaims();
   }

   public String asJsonString() {
      Map<String, Map<String, Object>> tokenContainer = new HashMap<>();
      tokenContainer.put("idTokenSnapshot", this.idTokenSnapshot);
      tokenContainer.put("accessTokenSnapshot", this.accessTokenSnapshot);
      return JsonCoding.toJson(tokenContainer);
   }

   public static List<Claim<?>> toClaims(Map<String, Object> snapshot) {
      return snapshot.entrySet().stream().map(entry -> new Claim<ClaimValue>() {
         @Override
         public String name() {
            return (String)entry.getKey();
         }

         @Override
         public ClaimValue value() {
            return entry::getValue;
         }
      }).toList();
   }

   private static Map<String, Object> serializeClaims(List<Claim<?>> claims) {
      return claims.stream().collect(Collectors.toMap(Claim::name, claim -> claim.value().toJSONSerializable()));
   }

   @SuppressFBWarnings(justification = "generated code")
   public Map<String, Object> getIdTokenSnapshot() {
      return this.idTokenSnapshot;
   }

   @SuppressFBWarnings(justification = "generated code")
   public Map<String, Object> getAccessTokenSnapshot() {
      return this.accessTokenSnapshot;
   }

   @SuppressFBWarnings(justification = "generated code")
   public void setIdTokenSnapshot(Map<String, Object> idTokenSnapshot) {
      this.idTokenSnapshot = idTokenSnapshot;
   }

   @SuppressFBWarnings(justification = "generated code")
   public void setAccessTokenSnapshot(Map<String, Object> accessTokenSnapshot) {
      this.accessTokenSnapshot = accessTokenSnapshot;
   }

   @SuppressFBWarnings(justification = "generated code")
   @Override
   public boolean equals(Object o) {
      if (o == this) {
         return true;
      } else if (!(o instanceof OAuth2SessionClaims other)) {
         return false;
      } else {
         Object this$idTokenSnapshot = this.idTokenSnapshot;
         Object other$idTokenSnapshot = other.idTokenSnapshot;
         if (this$idTokenSnapshot == null ? other$idTokenSnapshot == null : this$idTokenSnapshot.equals(other$idTokenSnapshot)) {
            Object this$accessTokenSnapshot = this.accessTokenSnapshot;
            Object other$accessTokenSnapshot = other.accessTokenSnapshot;
            return this$accessTokenSnapshot == null ? other$accessTokenSnapshot == null : this$accessTokenSnapshot.equals(other$accessTokenSnapshot);
         } else {
            return false;
         }
      }
   }

   @SuppressFBWarnings(justification = "generated code")
   @Override
   public int hashCode() {
      int PRIME = 59;
      int result = 1;
      Object $idTokenSnapshot = this.idTokenSnapshot;
      result = result * 59 + ($idTokenSnapshot == null ? 43 : $idTokenSnapshot.hashCode());
      Object $accessTokenSnapshot = this.accessTokenSnapshot;
      return result * 59 + ($accessTokenSnapshot == null ? 43 : $accessTokenSnapshot.hashCode());
   }

   @SuppressFBWarnings(justification = "generated code")
   @Override
   public String toString() {
      return "OAuth2SessionClaims(idTokenSnapshot=" + this.getIdTokenSnapshot() + ", accessTokenSnapshot=" + this.getAccessTokenSnapshot() + ")";
   }

   @SuppressFBWarnings(justification = "generated code")
   private OAuth2SessionClaims() {
   }

   @SuppressFBWarnings(justification = "generated code")
   public OAuth2SessionClaims(Map<String, Object> idTokenSnapshot, Map<String, Object> accessTokenSnapshot) {
      this.idTokenSnapshot = idTokenSnapshot;
      this.accessTokenSnapshot = accessTokenSnapshot;
   }
}