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

package org.bouncycastle.openssl.jcajce;

import org.bouncycastle.openssl.PEMException;
import org.bouncycastle.openssl.PEMEncryptor;
import org.bouncycastle.jcajce.util.NamedJcaJceHelper;
import org.bouncycastle.jcajce.util.ProviderJcaJceHelper;
import java.security.Provider;
import org.bouncycastle.jcajce.util.DefaultJcaJceHelper;
import java.security.SecureRandom;
import org.bouncycastle.jcajce.util.JcaJceHelper;

public class JcePEMEncryptorBuilder
{
    private final String algorithm;
    private JcaJceHelper helper;
    private SecureRandom random;
    
    public JcePEMEncryptorBuilder(final String algorithm) {
        this.helper = new DefaultJcaJceHelper();
        this.algorithm = algorithm;
    }
    
    public JcePEMEncryptorBuilder setProvider(final Provider provider) {
        this.helper = new ProviderJcaJceHelper(provider);
        return this;
    }
    
    public JcePEMEncryptorBuilder setProvider(final String s) {
        this.helper = new NamedJcaJceHelper(s);
        return this;
    }
    
    public JcePEMEncryptorBuilder setSecureRandom(final SecureRandom random) {
        this.random = random;
        return this;
    }
    
    public PEMEncryptor build(final char[] array) {
        if (this.random == null) {
            this.random = new SecureRandom();
        }
        final byte[] bytes = new byte[this.algorithm.startsWith("AES-") ? 16 : 8];
        this.random.nextBytes(bytes);
        return new PEMEncryptor() {
            @Override
            public String getAlgorithm() {
                return JcePEMEncryptorBuilder.this.algorithm;
            }
            
            @Override
            public byte[] getIV() {
                return bytes;
            }
            
            @Override
            public byte[] encrypt(final byte[] array) throws PEMException {
                return PEMUtilities.crypt(true, JcePEMEncryptorBuilder.this.helper, array, array, JcePEMEncryptorBuilder.this.algorithm, bytes);
            }
        };
    }
}