package com.ruoyi.common.utils;

import org.apache.commons.codec.binary.Base64;

import javax.crypto.Cipher;
import java.security.*;
import java.security.interfaces.RSAPrivateKey;
import java.security.interfaces.RSAPublicKey;
import java.security.spec.PKCS8EncodedKeySpec;
import java.security.spec.X509EncodedKeySpec;

public class RsaUtils {
    // Rsa 私钥
    public static String privateKey = "MIIEvgIBADANBgkqhkiG9w0BAQEFAASCBKgwggSkAgEAAoIBAQC5XbmsZ3Fy9BM66up/roCBNSwNHiiETfi0+EHYHw1WQWhgPj4u0LQR+SfPsC8BdfQ6TJino+CytqkFLdf5Pvoor4VMIytR/TVIm8kibXpuejqcicrkt4D66O2CHMLzJDzDsO4/fF/ybn5ShnbXEuar/2ocIjNAIp9kqebJXofntuv4/0wmZWrwbxPKe7tDboML1RYCFJaamzKCBrmM7a5je68C+7YE+fN7/EMQIR/dSqzcx/Q+qDyDs22k+waZPjL/XiGDSTxnUBj5A+PNeo3eKQDrVkQa6gZxnfe1MBn8jbu8kjjSg5Nj3LDYOXGuk1bydpER6jkX8A1qCw20OO4HAgMBAAECggEAU7laoN1EJ7jIascqTatizXlXwUEK6d9R42NyUgyMeJYob7gNXBXWD3wrGCqqwI0nV6G7HZTua6oHI5i2QRTJ5tCIc4A7E9g7VeVPbqZmMzov+P5fzvh3YF0O0cLt5uZKL9Vddv1VVOnFUr77NF+MIsnFgpmL7n704YHABNoWpnh6cxDpm+sjv1S/Y7Mvq1X+r/D8ocbkd2MKaWxK1OCB64hFE3Cv+9rgp/f9hFg0meAhRQd6+vAo8zm8Y1w7xqFEWEvMgmGgyat5cA8aqZINzRo8fmhN2mj7THvMa32pGIMuOKSgHX8pF6cyaEydNYICljSoxyIeGShU5Wao4LQeYQKBgQDnAd9qgfZcR8EwnjUz6MSQGyCiOHZxUUmKP0IHbu96w+f1CGtSYyjy6jex3qGSxYBa1K2k16GDQFiSUHYu2CFM8TPH/JStRSjiLRcokC8aR7FwCEohI4X9TvvWbHFwgJ5yu8vs0/TJ+X6BoZ7r6nBkz4iSib9Ixbn6Fy2bUHohkQKBgQDNa76XGE6B+wzxo0Wb2tpLCHBVN7leWr/qumLl4Ei1gQeIB0MI3Oc3JogB1dYzEk0KAbspLoZ8oEr5GwlcQcqZ4mdL3m7cTNSfDnpQ6SPY9CvyvwLutyktb+NkL3KYPOCM+NP9QjhZmj4iEnh4NFjXchykxbx9FFQaJBKWrMxKFwKBgQCDtYeifFLwrmCxzaJb5F0eJLlohHTTixs43XRSNKgKAOShypkWMj9IsM3sdhj9S+ow343ZrLAD244kbgfUEZp6+hMh9Rr357S1sCvCDuMsQrAiRjetyUk98eiYit/q2wp8NCLC9t5cwgmk83de3x4R/FnL4wnS/hlZRBnWsf3QYQKBgQDHKu4xpTmJQRY9uuYkxL7d/SKdHVg3KZnmRizZYOpeQOpCc3gFQdlUWaGI1gHGprnXN/J1bUyrzOakPBiJQKjlJuSsZM/r/Pox9WGWzVUaZzOtOpHnTeM40MESRAbBoFS/mgpABVKJXclcU7hGI8r3qQYopCeqxmj8p3j2ajfwpQKBgHKsWjZaULE9+sFwAYOn9eU59/CNqbpSskdUFXdaVGNFIunNxusoWL3x9W5Xjo2IXcEGaI4ch3vOC+Ha3+7vv3ty81JZeGFfiW+VGf8bv3Md+6K+x78/DE69HRYu4fCfa4OhFL+fp3IkPY+HwxKPRwS46K4JLG1jp1vdkn7WG4mf";

    /**
     * 私钥解密
     *
     * @param text 私钥
     * @param text 待解密的文本
     * @return 解密后的文本
     */
    public static String decryptByPrivateKey(String text) throws Exception {
        return decryptByPrivateKey(privateKey, text);
    }

    /**
     * 公钥解密
     *
     * @param publicKeyString 公钥
     * @param text            待解密的信息
     * @return 解密后的文本
     */
    public static String decryptByPublicKey(String publicKeyString, String text) throws Exception {
        X509EncodedKeySpec x509EncodedKeySpec = new X509EncodedKeySpec(Base64.decodeBase64(publicKeyString));
        KeyFactory keyFactory = KeyFactory.getInstance("RSA");
        PublicKey publicKey = keyFactory.generatePublic(x509EncodedKeySpec);
        Cipher cipher = Cipher.getInstance("RSA");
        cipher.init(Cipher.DECRYPT_MODE, publicKey);
        byte[] result = cipher.doFinal(Base64.decodeBase64(text));
        return new String(result);
    }

    /**
     * 私钥加密
     *
     * @param privateKeyString 私钥
     * @param text             待加密的信息
     * @return 加密后的文本
     */
    public static String encryptByPrivateKey(String privateKeyString, String text) throws Exception {
        PKCS8EncodedKeySpec pkcs8EncodedKeySpec = new PKCS8EncodedKeySpec(Base64.decodeBase64(privateKeyString));
        KeyFactory keyFactory = KeyFactory.getInstance("RSA");
        PrivateKey privateKey = keyFactory.generatePrivate(pkcs8EncodedKeySpec);
        Cipher cipher = Cipher.getInstance("RSA");
        cipher.init(Cipher.ENCRYPT_MODE, privateKey);
        byte[] result = cipher.doFinal(text.getBytes());
        return Base64.encodeBase64String(result);
    }

    /**
     * 私钥解密
     *
     * @param privateKeyString 私钥
     * @param text             待解密的文本
     * @return 解密后的文本
     */
    public static String decryptByPrivateKey(String privateKeyString, String text) throws Exception {
        PKCS8EncodedKeySpec pkcs8EncodedKeySpec5 = new PKCS8EncodedKeySpec(Base64.decodeBase64(privateKeyString));
        KeyFactory keyFactory = KeyFactory.getInstance("RSA");
        PrivateKey privateKey = keyFactory.generatePrivate(pkcs8EncodedKeySpec5);
        Cipher cipher = Cipher.getInstance("RSA");
        cipher.init(Cipher.DECRYPT_MODE, privateKey);
        byte[] result = cipher.doFinal(Base64.decodeBase64(text));
        return new String(result);
    }

    /**
     * 公钥加密
     *
     * @param publicKeyString 公钥
     * @param text            待加密的文本
     * @return 加密后的文本
     */
    public static String encryptByPublicKey(String publicKeyString, String text) throws Exception {
        X509EncodedKeySpec x509EncodedKeySpec2 = new X509EncodedKeySpec(Base64.decodeBase64(publicKeyString));
        KeyFactory keyFactory = KeyFactory.getInstance("RSA");
        PublicKey publicKey = keyFactory.generatePublic(x509EncodedKeySpec2);
        Cipher cipher = Cipher.getInstance("RSA");
        cipher.init(Cipher.ENCRYPT_MODE, publicKey);
        byte[] result = cipher.doFinal(text.getBytes());
        return Base64.encodeBase64String(result);
    }

    /**
     * 构建RSA密钥对
     *
     * @return 生成后的公私钥信息
     */
    public static RsaKeyPair generateKeyPair() throws NoSuchAlgorithmException {
        KeyPairGenerator keyPairGenerator = KeyPairGenerator.getInstance("RSA");
        keyPairGenerator.initialize(1024);
        KeyPair keyPair = keyPairGenerator.generateKeyPair();
        RSAPublicKey rsaPublicKey = (RSAPublicKey) keyPair.getPublic();
        RSAPrivateKey rsaPrivateKey = (RSAPrivateKey) keyPair.getPrivate();
        String publicKeyString = Base64.encodeBase64String(rsaPublicKey.getEncoded());
        String privateKeyString = Base64.encodeBase64String(rsaPrivateKey.getEncoded());
        return new RsaKeyPair(publicKeyString, privateKeyString);
    }

    /**
     * RSA密钥对对象
     */
    public static class RsaKeyPair {
        private final String publicKey;
        private final String privateKey;

        public RsaKeyPair(String publicKey, String privateKey) {
            this.publicKey = publicKey;
            this.privateKey = privateKey;
        }

        public String getPublicKey() {
            return publicKey;
        }

        public String getPrivateKey() {
            return privateKey;
        }
    }
}
