// 
// Decompiled by Procyon v0.6.0
// 

package com.google.crypto.tink.hybrid;

import java.security.GeneralSecurityException;
import com.google.errorprone.annotations.CanIgnoreReturnValue;
import com.google.errorprone.annotations.Immutable;
import java.util.Objects;

public final class HpkeParameters extends HybridParameters
{
    private final KemId kem;
    private final KdfId kdf;
    private final AeadId aead;
    private final Variant variant;
    
    private HpkeParameters(final KemId kem, final KdfId kdf, final AeadId aead, final Variant variant) {
        this.kem = kem;
        this.kdf = kdf;
        this.aead = aead;
        this.variant = variant;
    }
    
    public static Builder builder() {
        return new Builder();
    }
    
    public KemId getKemId() {
        return this.kem;
    }
    
    public KdfId getKdfId() {
        return this.kdf;
    }
    
    public AeadId getAeadId() {
        return this.aead;
    }
    
    public Variant getVariant() {
        return this.variant;
    }
    
    @Override
    public boolean hasIdRequirement() {
        return this.variant != Variant.NO_PREFIX;
    }
    
    @Override
    public boolean equals(final Object o) {
        if (!(o instanceof HpkeParameters)) {
            return false;
        }
        final HpkeParameters other = (HpkeParameters)o;
        return this.kem == other.kem && this.kdf == other.kdf && this.aead == other.aead && this.variant == other.variant;
    }
    
    @Override
    public int hashCode() {
        return Objects.hash(HpkeParameters.class, this.kem, this.kdf, this.aead, this.variant);
    }
    
    @Override
    public String toString() {
        return "HPKE Parameters (Variant: " + this.variant + ", KemId: " + this.kem + ", KdfId: " + this.kdf + ", AeadId: " + this.aead + ")";
    }
    
    @Immutable
    public static final class Variant
    {
        public static final Variant TINK;
        public static final Variant CRUNCHY;
        public static final Variant NO_PREFIX;
        private final String name;
        
        private Variant(final String name) {
            this.name = name;
        }
        
        @Override
        public String toString() {
            return this.name;
        }
        
        static {
            TINK = new Variant("TINK");
            CRUNCHY = new Variant("CRUNCHY");
            NO_PREFIX = new Variant("NO_PREFIX");
        }
    }
    
    @Immutable
    private static class AlgorithmIdentifier
    {
        protected final String name;
        protected final int value;
        
        private AlgorithmIdentifier(final String name, final int value) {
            this.name = name;
            this.value = value;
        }
        
        public int getValue() {
            return this.value;
        }
        
        @Override
        public String toString() {
            return String.format("%s(0x%04x)", this.name, this.value);
        }
    }
    
    @Immutable
    public static final class KemId extends AlgorithmIdentifier
    {
        public static final KemId DHKEM_P256_HKDF_SHA256;
        public static final KemId DHKEM_P384_HKDF_SHA384;
        public static final KemId DHKEM_P521_HKDF_SHA512;
        public static final KemId DHKEM_X25519_HKDF_SHA256;
        
        private KemId(final String name, final int value) {
            super(name, value);
        }
        
        static {
            DHKEM_P256_HKDF_SHA256 = new KemId("DHKEM_P256_HKDF_SHA256", 16);
            DHKEM_P384_HKDF_SHA384 = new KemId("DHKEM_P384_HKDF_SHA384", 17);
            DHKEM_P521_HKDF_SHA512 = new KemId("DHKEM_P521_HKDF_SHA512", 18);
            DHKEM_X25519_HKDF_SHA256 = new KemId("DHKEM_X25519_HKDF_SHA256", 32);
        }
    }
    
    @Immutable
    public static final class KdfId extends AlgorithmIdentifier
    {
        public static final KdfId HKDF_SHA256;
        public static final KdfId HKDF_SHA384;
        public static final KdfId HKDF_SHA512;
        
        private KdfId(final String name, final int value) {
            super(name, value);
        }
        
        static {
            HKDF_SHA256 = new KdfId("HKDF_SHA256", 1);
            HKDF_SHA384 = new KdfId("HKDF_SHA384", 2);
            HKDF_SHA512 = new KdfId("HKDF_SHA512", 3);
        }
    }
    
    @Immutable
    public static final class AeadId extends AlgorithmIdentifier
    {
        public static final AeadId AES_128_GCM;
        public static final AeadId AES_256_GCM;
        public static final AeadId CHACHA20_POLY1305;
        
        private AeadId(final String name, final int value) {
            super(name, value);
        }
        
        static {
            AES_128_GCM = new AeadId("AES_128_GCM", 1);
            AES_256_GCM = new AeadId("AES_256_GCM", 2);
            CHACHA20_POLY1305 = new AeadId("CHACHA20_POLY1305", 3);
        }
    }
    
    public static final class Builder
    {
        private KemId kem;
        private KdfId kdf;
        private AeadId aead;
        private Variant variant;
        
        private Builder() {
            this.kem = null;
            this.kdf = null;
            this.aead = null;
            this.variant = Variant.NO_PREFIX;
        }
        
        @CanIgnoreReturnValue
        public Builder setKemId(final KemId kem) {
            this.kem = kem;
            return this;
        }
        
        @CanIgnoreReturnValue
        public Builder setKdfId(final KdfId kdf) {
            this.kdf = kdf;
            return this;
        }
        
        @CanIgnoreReturnValue
        public Builder setAeadId(final AeadId aead) {
            this.aead = aead;
            return this;
        }
        
        @CanIgnoreReturnValue
        public Builder setVariant(final Variant variant) {
            this.variant = variant;
            return this;
        }
        
        public HpkeParameters build() throws GeneralSecurityException {
            if (this.kem == null) {
                throw new GeneralSecurityException("HPKE KEM parameter is not set");
            }
            if (this.kdf == null) {
                throw new GeneralSecurityException("HPKE KDF parameter is not set");
            }
            if (this.aead == null) {
                throw new GeneralSecurityException("HPKE AEAD parameter is not set");
            }
            if (this.variant == null) {
                throw new GeneralSecurityException("HPKE variant is not set");
            }
            return new HpkeParameters(this.kem, this.kdf, this.aead, this.variant, null);
        }
    }
}