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

package com.nimbusds.jose.crypto.factories;

import java.util.Collections;
import java.util.Collection;
import java.util.LinkedHashSet;
import com.nimbusds.jose.JOSEException;
import com.nimbusds.jose.crypto.Ed25519Signer;
import com.nimbusds.jose.jwk.OctetKeyPair;
import com.nimbusds.jose.crypto.ECDSASigner;
import com.nimbusds.jose.jwk.ECKey;
import com.nimbusds.jose.crypto.RSASSASigner;
import com.nimbusds.jose.jwk.RSAKey;
import com.nimbusds.jose.crypto.MACSigner;
import com.nimbusds.jose.jwk.OctetSequenceKey;
import com.nimbusds.jose.jwk.KeyUse;
import com.nimbusds.jose.jwk.JWKException;
import com.nimbusds.jose.JWSSigner;
import com.nimbusds.jose.jwk.JWK;
import com.nimbusds.jose.JWSAlgorithm;
import java.util.Set;
import com.nimbusds.jose.jca.JCAContext;
import com.nimbusds.jose.produce.JWSSignerFactory;

public class DefaultJWSSignerFactory implements JWSSignerFactory
{
    private final JCAContext jcaContext;
    public static final Set<JWSAlgorithm> SUPPORTED_ALGORITHMS;
    
    public DefaultJWSSignerFactory() {
        this.jcaContext = new JCAContext();
    }
    
    @Override
    public Set<JWSAlgorithm> supportedJWSAlgorithms() {
        return DefaultJWSSignerFactory.SUPPORTED_ALGORITHMS;
    }
    
    @Override
    public JCAContext getJCAContext() {
        return this.jcaContext;
    }
    
    @Override
    public JWSSigner createJWSSigner(final JWK key) throws JOSEException {
        if (!key.isPrivate()) {
            throw JWKException.expectedPrivate();
        }
        if (key.getKeyUse() != null && !KeyUse.SIGNATURE.equals(key.getKeyUse())) {
            throw new JWKException("The JWK use must be sig (signature) or unspecified");
        }
        JWSSigner signer;
        if (key instanceof OctetSequenceKey) {
            signer = new MACSigner((OctetSequenceKey)key);
        }
        else if (key instanceof RSAKey) {
            signer = new RSASSASigner((RSAKey)key);
        }
        else if (key instanceof ECKey && ECDSASigner.SUPPORTED_CURVES.contains(((ECKey)key).getCurve())) {
            signer = new ECDSASigner((ECKey)key);
        }
        else {
            if (!(key instanceof OctetKeyPair) || !Ed25519Signer.SUPPORTED_CURVES.contains(((OctetKeyPair)key).getCurve())) {
                throw new JOSEException("Unsupported JWK type and / or curve");
            }
            signer = new Ed25519Signer((OctetKeyPair)key);
        }
        signer.getJCAContext().setSecureRandom(this.jcaContext.getSecureRandom());
        signer.getJCAContext().setProvider(this.jcaContext.getProvider());
        return signer;
    }
    
    @Override
    public JWSSigner createJWSSigner(final JWK key, final JWSAlgorithm alg) throws JOSEException {
        if (!key.isPrivate()) {
            throw JWKException.expectedPrivate();
        }
        if (key.getKeyUse() != null && !KeyUse.SIGNATURE.equals(key.getKeyUse())) {
            throw new JWKException("The JWK use must be sig (signature) or unspecified");
        }
        JWSSigner signer;
        if (MACSigner.SUPPORTED_ALGORITHMS.contains(alg) && key instanceof OctetSequenceKey) {
            signer = new MACSigner((OctetSequenceKey)key);
        }
        else if (RSASSASigner.SUPPORTED_ALGORITHMS.contains(alg) && key instanceof RSAKey) {
            signer = new RSASSASigner((RSAKey)key);
        }
        else if (ECDSASigner.SUPPORTED_ALGORITHMS.contains(alg) && key instanceof ECKey && ECDSASigner.SUPPORTED_CURVES.contains(((ECKey)key).getCurve())) {
            signer = new ECDSASigner((ECKey)key);
        }
        else {
            if (!Ed25519Signer.SUPPORTED_ALGORITHMS.contains(alg) || !(key instanceof OctetKeyPair) || !Ed25519Signer.SUPPORTED_CURVES.contains(((OctetKeyPair)key).getCurve())) {
                throw new JOSEException("Unsupported JWK type, JWK curve and / or JWS algorithm");
            }
            signer = new Ed25519Signer((OctetKeyPair)key);
        }
        signer.getJCAContext().setSecureRandom(this.jcaContext.getSecureRandom());
        signer.getJCAContext().setProvider(this.jcaContext.getProvider());
        return signer;
    }
    
    static {
        final Set<JWSAlgorithm> algs = new LinkedHashSet<JWSAlgorithm>();
        algs.addAll(MACSigner.SUPPORTED_ALGORITHMS);
        algs.addAll(RSASSASigner.SUPPORTED_ALGORITHMS);
        algs.addAll(ECDSASigner.SUPPORTED_ALGORITHMS);
        algs.addAll(Ed25519Signer.SUPPORTED_ALGORITHMS);
        SUPPORTED_ALGORITHMS = Collections.unmodifiableSet((Set<? extends JWSAlgorithm>)algs);
    }
}