import json
import base64
from Crypto.PublicKey import RSA
from Crypto.Hash import MD5
from Crypto.Signature import pkcs1_15
from Crypto.Cipher import PKCS1_OAEP

from Cryptodome.Cipher import PKCS1_v1_5
import os

class encrypt_rsa():

    
    def __init__(self, data):
        # replace with your public key
        public_key = ("-----BEGIN PUBLIC KEY-----\n"+
        "MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQCy7O9ULgdfc1SUXrU0W2qWg3l8VbvNpvq+ilwPDdq4EYzKwOe97Zd2wtW8HJQF7GNn2SaeHLCilsAJTYPLb+uRzXz3Aozxx8u6Bk5mGMVqi9rXXCNQCpRZYgM/7JDvtO5UhLCiMFHFO2f2c0QCmdR+yzdP6anJk9vLikuBwWxY6wIDAQAB"+
        "\n-----END PUBLIC KEY-----")
        self.message = data.encode()
        self.public_key = RSA.import_key(public_key)


    # Segmentation encryption
    def encrypt(self):
        try:
            cipher_rsa = PKCS1_v1_5.new(self.public_key)
            res = []
            for i in range(0, len(self.message), 64):
                enc_tmp = cipher_rsa.encrypt(self.message[i:i+64])
                res.append(enc_tmp)
            cipher_text = b''.join(res)
        except Exception as e:
            print(e)
        else:
            return base64.b64encode(cipher_text).decode()


if __name__ == '__main__':

    payload = "{providerName: 'KBZ Pay',methodName: PWA,orderId: 'test-23112021001',customerPhone: '959796564600', description:'Test',customerAddress:'Yangon, Myanmar' ,totalAmount: 2200, email: '[email protected]', customerName: 'test username', items:'[{\"name\":\"Apple\",\"amount\":\"1100\",\"quantity\":\"2\"}]'}"

    payload_encrypt = encrypt_rsa(payload).encrypt()
    print(str(payload_encrypt))


                                    

                    
public static byte[] encrypt(String data, String publicKey) throws BadPaddingException, IllegalBlockSizeException, InvalidKeyException, NoSuchPaddingException, NoSuchAlgorithmException {
    Cipher cipher = Cipher.getInstance("RSA/ECB/PKCS1Padding");
    cipher.init(Cipher.ENCRYPT_MODE, getPublicKey(publicKey));
    byte[] enBytes = null;
    int i = 0;
    while (i < data.length()) {
        byte[] doFinal = cipher.doFinal(ArrayUtils.subarray(data.getBytes(), i, i + 64));
        enBytes = ArrayUtils.addAll(enBytes, doFinal);
        i += 64;
    }
    return enBytes;
}

public static PublicKey getPublicKey(String base64PublicKey){
        PublicKey publicKey = null;
        try{
            X509EncodedKeySpec keySpec = new X509EncodedKeySpec(Base64.getDecoder().decode(base64PublicKey.getBytes()));
            KeyFactory keyFactory = KeyFactory.getInstance("RSA");
            publicKey = keyFactory.generatePublic(keySpec);
            return publicKey;
        } catch (NoSuchAlgorithmException e) {
            e.printStackTrace();
        } catch (InvalidKeySpecException e) {
            e.printStackTrace();
        }
        return publicKey;


                                    

                    
 ** if you want to put ' in json array, you can use this way to make a correct json format or you can replace ' and " characters in your product name before sending to Dinger gateway **
const items = `[{\"name\":\"Dinger\\'s\",\"amount\": ${totalAmount},\"quantity\":\"1\"}]`;

const data = "{providerName: 'Onepay',methodName: PIN,orderId: 'hnt-11092021001',customerPhone: '959796564600',customerName: 'Hsu Nyeint Thu', description:'Dinger Campaign Test',customerAddress:'Yangon, Myanmar' ,totalAmount: 100,items:'[{\"name\":\"Dinger\",\"amount\":\"100\",\"quantity\":\"1\"}]'}"


const pubKey = "-----BEGIN PUBLIC KEY-----\n"+
            "MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQCp0/5oszMaTRzFL/M7yaO2+B9yjGO2u3grq5dNzH+xnJn7IViC5oA//WdHuF6lDvcGHbhcIpQX8tGV5juOK7N/xBj9mrcE4QYAdVAMsnpx8bpTD9sUjg/QbXkK79JDIDbPjdgHWKKfKfo0EWgHkGpYmZDmDy0JG6KUiiUOrFaSQQIDAQAB"
        "-----END PUBLIC KEY-----";

const publicKey = new NodeRSA();
publicKey.importKey(pubKey, 'pkcs8-public')
publicKey.setOptions({encryptionScheme: 'pkcs1'})
const encrytpStr = publicKey.encrypt(data,'base64')
console.log("encrytpStr "+encrytpStr)
const param = {payload: encrytpStr};
const payloadData = new URLSearchParams(param);
// in fetch:
return await fetch(url, {
   method,
   body: payloadData })


                                    

val payload = data.toString()
val encryptedPayload = Base64.getEncoder().encodeToString(
                segmentationEncryption(
                    data.toString().toByteArray(),
                    publicKey
                )
            )           

  fun segmentationEncryption(data: ByteArray, rsaPublicKeyBase64: String): ByteArray? {

        val cipher: Cipher = Cipher.getInstance("RSA/ECB/PKCS1Padding")
        cipher.init(Cipher.ENCRYPT_MODE, getRsaPublicKey(rsaPublicKeyBase64))

        var enBytes: ByteArray? = null

        var i = 0
        while (i < data.size) {
            val doFinal = cipher.doFinal(ArrayUtils.subarray(data, i, i + 64))
            enBytes = ArrayUtils.addAll(enBytes, doFinal)
            i += 64
        }

        return enBytes
    }


                                    

 $items_data = array(
        "name" => "Dinger University Donation",
        "amount" => "$total",
        "quantity" => "1"
    );

            if ($payment == "Visa") {
             $data_pay = json_encode(array(
            "providerName" => $payment,
            "methodName" => $method,
            "totalAmount" => "$total",
            "items" => json_encode(array($items_data)),
            "orderId" => $donationID,
            "customerName" => $name,
            "email" => $email,
            "billToForeName" => "Customer name",
            "billToSurName" => $name,
            "billAddress" => $address,
            "billCity" => $city,
            "customerPhone" => $phone
        ));
    } elseif ($payment == "Master") {
        $data_pay = json_encode(array(
            "providerName" => $payment,
            "methodName" => $method,
            "totalAmount" => "$total",
            "items" => json_encode(array($items_data)),
            "orderId" => $donationID,
            "customerName" => $name,
            "email" => $email,
            "billToForeName" => "Customer name",
            "billToSurName" => $name,
            "billAddress" => $address,
            "billCity" => $city,
            "customerPhone" => $phone
        ));
    } elseif ($payment == "JCB") {
        $data_pay = json_encode(array(
            "providerName" => $payment,
            "methodName" => $method,
            "totalAmount" => "$total",
            "items" => json_encode(array($items_data)),
            "orderId" => $donationID,
            "customerName" => $name,
            "email" => $email,
            "billToForeName" => "Customer name",
            "billToSurName" => $name,
            "billAddress" => $address,
            "billCity" => $city,
            "customerPhone" => $phone
        ));
    } elseif ($payment != "Visa" && $payment != "Master" && $payment != "JCB") {
        $data_pay = json_encode(array(
            "providerName" => $payment,
            "methodName" => $method,
            "totalAmount" => "$total",
            "items" => json_encode(array($items_data)),
            "orderId" => $donationID,
            "customerName" => $name,
            "customerPhone" => $phone
        ));
    }
   
    $publicKey = '-----BEGIN PUBLIC KEY-----MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQCJtu2coOqkFaaxLtlnb6DAQRvw+6l9iwm6RZlGrAf6IUnZiJavYi60hTveLkFbeYLvvLcFyIGddQDUJBCvEOIk7GwgF6pPRlV9k5g7CDyHbqsjudOix+ElD2XkAiUeYWAK++uRVBqcE/xxwNMDoRwyYqoC/OifZf0pH7PA3XCUyQIDAQAB-----END PUBLIC KEY-----';

    $rsa = new \phpseclib\Crypt\RSA();

    extract($rsa->createKey(1024));
    $rsa->loadKey($publicKey); // public key
    $rsa->setEncryptionMode(2);

    $ciphertext = $rsa->encrypt($data_pay);

    $value = base64_encode($ciphertext)