mod utils;

use wasm_bindgen::prelude::*;

use std::io::{Read, Write};

use age::x25519::Recipient;

//use rand::{rngs::OsRng, RngCore};

// When the `wee_alloc` feature is enabled, use `wee_alloc` as the global
// allocator.
#[cfg(feature = "wee_alloc")]
#[global_allocator]
static ALLOC: wee_alloc::WeeAlloc = wee_alloc::WeeAlloc::INIT;

#[wasm_bindgen]
extern "C" {
    fn alert(s: &str);
}

#[wasm_bindgen]
pub fn greet() {
    alert("Hello, {{project-name}}!");
}

#[wasm_bindgen]
pub fn age_encrypt(plaintext: String, key: String) -> Option<Vec<u8>> {
    utils::set_panic_hook();

    let pubkey = key.parse::<Recipient>().ok()?;

    let encryptor = age::Encryptor::with_recipients(vec![Box::new(pubkey)]);

    let mut encrypted = vec![];

    let mut writer = encryptor.wrap_output(&mut encrypted).ok()?;

    writer.write_all(&plaintext.as_bytes()).ok()?;
    writer.finish().ok()?;

    Some(encrypted)
}

#[wasm_bindgen]
pub fn age_decrypt_passphrase(blob: Vec<u8>, passphrase: String) -> Option<String> {
    utils::set_panic_hook();


    let decryptor = match age::Decryptor::new(&blob[..]).unwrap() {
        age::Decryptor::Passphrase(d) => d,
        _ => panic!("something very wrong happened!"),
    };

    let mut decrypted = vec![];
    let mut reader = decryptor
        .decrypt(&secrecy::Secret::new(passphrase), None)
        .ok()?;
    reader.read_to_end(&mut decrypted).ok()?;
    Some(std::str::from_utf8(&decrypted).ok()?.to_owned())
}