//test.rs
//
// 测试在Rust侧生成钱包密钥对，转换成C侧的数据结构。
// 测试钱包在C侧调用接口存储和重新读出钱包密钥
//

use rustwallet::{
    free_cwallet, generate_sec_key, get_address, get_public_key, new_wallet, rdecrypt, rencrypt,
    sign, sign_for_tran, str_deflate, str_inflate, CWallet,
};
use std::ffi::{CStr, CString};
use std::os::raw::c_char;

macro_rules! print_cchar {
    ($p1:expr) => {
        let s = CStr::from_ptr($p1);
        let ps = s.to_str().unwrap();
        println!("{}=> {}", stringify!($p1), ps)
    };
    ($p1:expr, $p2:expr) => {
        let s = CStr::from_ptr($p2);
        let ps = s.to_str().unwrap();
        println!("{}{}", $p1, ps)
    };
}

macro_rules! cchar_to_str {
    ($p1:expr) => {{
        let s = CStr::from_ptr($p1);
        let ps = s.to_str().unwrap();
        ps
    }};
}

macro_rules! str_to_cchar {
    ($p1:expr) => {{
        let msgkey = CString::new($p1).unwrap();
        let c_msgkey: *mut c_char = msgkey.into_raw();
        c_msgkey
    }};
}

fn main() {
    unsafe {
        let msg_key = "a0f2022d15fd8072f12c76cc1596c4f1";
        // let wallet: CWallet = new_wallet(str_to_cchar!(msg));
        // println!("---- generated a wallet to be used on C-side ----");
        // print_wallet(&wallet);
        // let address = get_address(&wallet);
        // print_cchar!(address);
        // let address_str = cchar_to_str!(address);
        // println!("address: {}", address_str);
        // let sign_str = sign_for_tran(&wallet,str_to_cchar!("cc0dac9a2bd7125bbe9130b83053494860f7a444868a45a9c00a865631ba8894"));
        // print_cchar!(sign_str);
        // let s_key = generate_sec_key(&wallet);
        // print_cchar!(s_key);

        let key0 = "3f26e586111b5c4cab6a5dd5e0d13c3b13184ba74410ba064a3b485be4f9a2cd";
        let key1 = "";
        let key2 = "cd00eb0126aeed39762579ce94c90a04695ad17fbd5e79aa4e9fc4a34ba32a5";
        let address2 = get_address(
            str_to_cchar!(msg_key),
            str_to_cchar!(key0),
            str_to_cchar!(key1),
            str_to_cchar!(key2),
        );
        print_cchar!(address2);
        let key_public = get_public_key(
            str_to_cchar!(msg_key),
            str_to_cchar!(key0),
            str_to_cchar!(key1),
            str_to_cchar!(key2),
        );
        print_cchar!(key_public);
        let key_private = generate_sec_key(
            str_to_cchar!(msg_key),
            str_to_cchar!(key0),
            str_to_cchar!(key1),
            str_to_cchar!(key2),
        );
        print_cchar!(key_private);
        let msg = "hello world";
        let str_signed = sign(
            str_to_cchar!(msg_key),
            str_to_cchar!(key0),
            str_to_cchar!(key1),
            str_to_cchar!(key2),
            str_to_cchar!(msg),
        );
        print_cchar!(str_signed);
        let tran_str = "0cd00eb0126aeed39762579ce94c90a04695ad17fbd5e79aa4e9fc4a34ba32a5";
        let str_signed2 = sign_for_tran(
            str_to_cchar!(msg_key),
            str_to_cchar!(key0),
            str_to_cchar!(key1),
            str_to_cchar!(key2),
            str_to_cchar!(tran_str),
        );
        print_cchar!(str_signed2);

        let str_encrypt = rencrypt(
            str_to_cchar!(msg_key),
            str_to_cchar!(key0),
            str_to_cchar!(key1),
            str_to_cchar!(key2),
            str_to_cchar!(msg),
        );
        print_cchar!(str_encrypt);

        let str_decrypt = rdecrypt(
            str_to_cchar!(msg_key),
            str_to_cchar!(key0),
            str_to_cchar!(key1),
            str_to_cchar!(key2),
            str_encrypt,
        );
        print_cchar!(str_decrypt);

        let str_encode = str_deflate(key_public);
        print_cchar!(str_encode);

        let str_decode = str_inflate(str_encode);
        print_cchar!(str_decode);
    }
}

unsafe fn print_wallet(cwallet: &CWallet) {
    print_cchar!("msg=> ", cwallet.msg_key);
    print_cchar!("master key=> ", cwallet.master_key);
    print_cchar!("second key=> ", cwallet.second_key);
    print_cchar!("backup key=> ", cwallet.backup_key);
}