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

package org.bouncycastle.pqc.jcajce.provider.kyber;

import org.bouncycastle.asn1.bc.BCObjectIdentifiers;
import org.bouncycastle.asn1.nist.NISTObjectIdentifiers;
import java.util.HashSet;
import java.security.PublicKey;
import org.bouncycastle.asn1.x509.SubjectPublicKeyInfo;
import java.io.IOException;
import java.security.PrivateKey;
import org.bouncycastle.asn1.pkcs.PrivateKeyInfo;
import java.security.InvalidKeyException;
import java.security.spec.X509EncodedKeySpec;
import java.security.spec.InvalidKeySpecException;
import java.security.spec.PKCS8EncodedKeySpec;
import java.security.spec.KeySpec;
import java.security.Key;
import org.bouncycastle.asn1.ASN1ObjectIdentifier;
import java.util.Set;
import org.bouncycastle.pqc.jcajce.provider.util.BaseKeyFactorySpi;

public class KyberKeyFactorySpi extends BaseKeyFactorySpi
{
    private static final Set<ASN1ObjectIdentifier> keyOids;
    
    public KyberKeyFactorySpi() {
        super(KyberKeyFactorySpi.keyOids);
    }
    
    public KyberKeyFactorySpi(final ASN1ObjectIdentifier asn1ObjectIdentifier) {
        super(asn1ObjectIdentifier);
    }
    
    public final KeySpec engineGetKeySpec(final Key key, final Class obj) throws InvalidKeySpecException {
        if (key instanceof BCKyberPrivateKey) {
            if (PKCS8EncodedKeySpec.class.isAssignableFrom(obj)) {
                return new PKCS8EncodedKeySpec(key.getEncoded());
            }
        }
        else {
            if (!(key instanceof BCKyberPublicKey)) {
                throw new InvalidKeySpecException("Unsupported key type: " + key.getClass() + ".");
            }
            if (X509EncodedKeySpec.class.isAssignableFrom(obj)) {
                return new X509EncodedKeySpec(key.getEncoded());
            }
        }
        throw new InvalidKeySpecException("Unknown key specification: " + obj + ".");
    }
    
    public final Key engineTranslateKey(final Key key) throws InvalidKeyException {
        if (key instanceof BCKyberPrivateKey || key instanceof BCKyberPublicKey) {
            return key;
        }
        throw new InvalidKeyException("Unsupported key type");
    }
    
    @Override
    public PrivateKey generatePrivate(final PrivateKeyInfo privateKeyInfo) throws IOException {
        return new BCKyberPrivateKey(privateKeyInfo);
    }
    
    @Override
    public PublicKey generatePublic(final SubjectPublicKeyInfo subjectPublicKeyInfo) throws IOException {
        return new BCKyberPublicKey(subjectPublicKeyInfo);
    }
    
    static {
        (keyOids = new HashSet<ASN1ObjectIdentifier>()).add(NISTObjectIdentifiers.id_alg_ml_kem_512);
        KyberKeyFactorySpi.keyOids.add(NISTObjectIdentifiers.id_alg_ml_kem_768);
        KyberKeyFactorySpi.keyOids.add(NISTObjectIdentifiers.id_alg_ml_kem_1024);
        KyberKeyFactorySpi.keyOids.add(BCObjectIdentifiers.kyber512_aes);
        KyberKeyFactorySpi.keyOids.add(BCObjectIdentifiers.kyber768_aes);
        KyberKeyFactorySpi.keyOids.add(BCObjectIdentifiers.kyber1024_aes);
    }
    
    public static class Kyber1024 extends KyberKeyFactorySpi
    {
        public Kyber1024() {
            super(NISTObjectIdentifiers.id_alg_ml_kem_1024);
        }
    }
    
    public static class Kyber1024_AES extends KyberKeyFactorySpi
    {
        public Kyber1024_AES() {
            super(BCObjectIdentifiers.kyber1024_aes);
        }
    }
    
    public static class Kyber512 extends KyberKeyFactorySpi
    {
        public Kyber512() {
            super(NISTObjectIdentifiers.id_alg_ml_kem_512);
        }
    }
    
    public static class Kyber512_AES extends KyberKeyFactorySpi
    {
        public Kyber512_AES() {
            super(BCObjectIdentifiers.kyber512_aes);
        }
    }
    
    public static class Kyber768 extends KyberKeyFactorySpi
    {
        public Kyber768() {
            super(NISTObjectIdentifiers.id_alg_ml_kem_768);
        }
    }
    
    public static class Kyber768_AES extends KyberKeyFactorySpi
    {
        public Kyber768_AES() {
            super(BCObjectIdentifiers.kyber768_aes);
        }
    }
}