package com.m2pool.common.core.utils.sign;


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;

/**
 * @Description TODO
 * @Date 2024/6/12 17:35
 * @Author dy
 */
public class Rsa2Utils {

    //Rsa 公钥
    public static  String publicKey ="MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQDbMEkGLB+J/SSAOp7iCp2J6g7k\n" +
            "zI5lNXEe6vqSkPLCKfUCAw94FI8RwTQYh3HIR778JZkLskxgm8PZjAWknWPKr4T7\n" +
            "Ma7cUWo9aWswtCFaa8ojTaKbJ7OqABpEIRoucrmrvna5Z0XwHmli9XgReJVg9gv3\n" +
            "pvHZnrmAXNQI9f7rCQIDAQAB";

    // Rsa 私钥
    public static String privateKey = "MIICdwIBADANBgkqhkiG9w0BAQEFAASCAmEwggJdAgEAAoGBANswSQYsH4n9JIA6nuIKnYnqDuTMjmU1cR7q+pKQ8sIp9QIDD3gUjxHBNBiHcchHvvwlmQuyTGCbw9mMBaSdY8qvhPsxrtxRaj1pazC0IVpryiNNopsns6oAGkQhGi5yuau+drlnRfAeaWL1eBF4lWD2C/em8dmeuYBc1Aj1/usJAgMBAAECgYEAywVzbhsqb2ahC4DAr8CDYT4B450w+7+/cpLV2zIVGRFB1kY9as0oI5rgBSRXmNoTpXuxwaq5ofZFNtjCVVJPHbQUmKaoHm2M6BRgPwP6OySlCaqQnyMnb0V8GP0QtrRiJDz1X+jklXtwyQ6MWIbvjlz7szKUmlCCkwpz3TLG2QECQQDu/go7C27FxQKb2k4T7rVJv5MJYDv6OvA8foHcC5BQtbrz1t2pLvBhyU2x6jVx/SaTR81vwKSXUoqj3OplLDapAkEA6sl2DgDBYbY8OAMmbgp998VdkpxEnkjyUnDp7ZnWs+UkwY8ZfHnISLaESkZVozXhZBpV1nMBVEDxdXF4y56tYQJATAw+SSeMKhZUjC9dJO6SdVMmgJdEvo0+oKFIxTJQy73oLWszwYAUMamStYhnVUxOmBMDBgpw1U4Im7fSRjtZcQJBALcn5A12b2U/WWjEpFURoEUKVT3K5AiqlUbUyNhOu1vo9Kx+an5dLm3y2+5pQeMpZCPIG5BIdQ/5/aMFmxsVNOECQG4CvMu7NTwx5ugKVpT112Z4LJCwGa/Xd25L/5ps4yLey0DonlkT/jr0p5aKFQ0e/ufinEmVHrAXsLLwg8iQZlc=";

    /**
     * 私钥解密
     *
     * @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;
        }
    }

}
