IPass-library/src/lib.rs

use std::io::{Read, Write};
use aes_gcm::{
    aead::{Aead, KeyInit},
    Aes256Gcm, Nonce
};
use sha2::{Sha256, Digest};
use std::fs::{
    File,
    read_dir,
    read_to_string
};
use rand::{
    rngs::OsRng,
    RngCore
};

pub fn get_args() -> Vec<String> {
    //! [0] = file path; [n>0] = argument
    std::env::args().collect()
}

#[derive(serde::Deserialize, Debug)]
struct Saved {
    success: bool,
    _status: u16,
    data: Vec<u8>,
    _errorcode: u64
}

fn get_token() -> String {
    let mut token: String = String::new();
    //check if file exists
    if !std::path::Path::new(&(get_ipass_folder()+"token.ipasst")).exists() {
        return "".to_string();
    }
    File::open(get_ipass_folder()+"token.ipasst").unwrap().read_to_string(&mut token).unwrap();
    token
}


pub async fn isync_get() -> bool {
    let token = get_token();
    let client = reqwest::Client::builder().https_only(true).build().unwrap();
    let req = client.get("https://ipass.ipost.rocks/saved").header("ipass-auth-token", token).build().unwrap();
    let res = client.execute(req).await.unwrap();
    let body = res.json::<Saved>().await.unwrap();
    if body.success {
        File::create(get_ipass_folder()+"temp.ipassx").unwrap().write_all(&body.data).unwrap();
        import_file(&(get_ipass_folder()+"temp.ipassx"));
        std::fs::remove_file(get_ipass_folder()+"temp.ipassx").unwrap();
        return true;
    }

    false
}

pub async fn isync_save() -> bool {
    let possible_data = export_data();
    match possible_data {
        Some(data) => {
            let token = get_token();
            let client = reqwest::Client::builder().https_only(true).build().unwrap();
            let req = client.post("https://ipass.ipost.rocks/saved")
                .header("ipass-auth-token", token)
                .body(data)
                .build()
                .unwrap();
            let _res = client.execute(req).await.unwrap();
            //sent data
            true
        },
        None => {
            false
        }
    }
}

pub fn import_data<R: Read>(mut reader: brotli::Decompressor<R>) {
    let mut content: String = String::new();
    let mut buf = [0u8; 4096];
    loop {
        match reader.read(&mut buf[..]) {
            Err(e) => {
                if let std::io::ErrorKind::Interrupted = e.kind() {
                    continue;
                }
                panic!("{}", e);
            }
            Ok(size) => {
                if size == 0 {
                    break;
                }
                content += std::str::from_utf8(&buf[..size]).unwrap();
            }
        }
    }

    let lines = content.lines();
    let mut name = "";
    for i in lines {
        if name.is_empty() {
            name = i;
            continue;
        }

        let mut file = File::create(format!("{}/{}.ipass",get_ipass_folder(), name)).unwrap();
        file.write_all(i.as_bytes()).unwrap();
        name = "";
    }
}

pub fn import_file(file_path:&String) -> bool {
    if std::path::Path::new(file_path).exists() {
        let reader = brotli::Decompressor::new(
            File::open(file_path).unwrap(),
            4096, // buffer size
        );
        import_data(reader);
        true
    } else {
        false
    }
}

pub fn export_data() -> Option<Vec<u8>> {
    let mut collected_data = String::new();
    let paths = std::fs::read_dir(get_ipass_folder()).ok()?;

    for path in paths.flatten() {
        let file_name = path.file_name().into_string().ok()?.replace(".ipass", "");
        let content = std::fs::read_to_string(get_ipass_folder() + &path.file_name().to_string_lossy()).ok()?;
        collected_data += format!("{}\n{}\n", file_name, content).as_str();
    }

    let mut compressed_data = Vec::new();
    {
        let mut compressor = brotli::CompressorWriter::new(&mut compressed_data, 4096, 11, 22);
        compressor.write_all(collected_data.as_bytes()).ok()?;
        compressor.flush().ok()?;
    }

    Some(compressed_data)
}



pub fn export_file(file_path: &String) -> bool {
    match export_data() {
        Some(compressed_data) => {
            if let Ok(mut file) = std::fs::File::create(file_path) {
                if let Err(e) = file.write_all(&compressed_data) {
                    eprintln!("Failed to write compressed data to file: {}", e);
                    return false;
                }
                true
            } else {
                eprintln!("Failed to create file at path: {}", file_path);
                false
            }
        }
        None => {
            eprintln!("Failed to export data");
            false
        }
    }
}

fn vecu8_to_string(vec: Vec<u8>) -> String {
    let mut do_print_warning = false;
    let mut out: String = String::new();
    for ind in vec {
        if let Ok(a) = std::str::from_utf8(&[ind]) {
            out += a;
        } else {
            do_print_warning = true;
            eprintln!("[WARNING] malformed character {}",ind);
            let mut temp_vec: Vec<u8> = Vec::new();
            temp_vec.insert(0,ind%128);
            out += vecu8_to_string(temp_vec).as_str();
        }
    }
    if do_print_warning {
        println!("[WARNING] Output may be corrupt");
    }
    out
}

fn encrypt_pass(nonce_arg:String, pass: String,mpw: String) -> String {
    let mut nonce_argument = String::new();
    if nonce_arg.len() < 12 {
        nonce_argument = nonce_arg.clone() + &" ".repeat(12-nonce_arg.len());
    }
    if nonce_arg.len() > 12 {
        nonce_argument = nonce_arg[0..12].to_string();
    }

    let mut nonce_hasher = Sha256::new();
    nonce_hasher.update(nonce_argument.as_bytes());

    let nonce_final = &nonce_hasher.finalize()[0..12];


    let mut hasher = Sha256::new();
    hasher.update(mpw.as_bytes());

    let master_pw = &hasher.finalize();

    let cipher = Aes256Gcm::new(master_pw);
    let nonce = Nonce::from_slice(nonce_final); // 96-bits; unique per message
    let ciphertext = cipher.encrypt(nonce, pass.as_ref()).unwrap();
    hex::encode(ciphertext)
}



fn decrypt_pass(nonce_arg:String, pass: Vec<u8>,mpw: String) -> Result<String,String> {
    let mut nonce_argument = String::new();
    if nonce_arg.len() < 12 {
        nonce_argument = nonce_arg.clone() + &" ".repeat(12-nonce_arg.len());
    }
    if nonce_arg.len() > 12 {
        nonce_argument = nonce_arg[0..12].to_string();
    }

    let mut nonce_hasher = Sha256::new();
    nonce_hasher.update(nonce_argument.as_bytes());

    let nonce_final = &nonce_hasher.finalize()[0..12];

    let mut hasher = Sha256::new();
    hasher.update(mpw.as_bytes());

    let master_pw = &hasher.finalize();
    let cipher = Aes256Gcm::new(master_pw);
    let nonce = Nonce::from_slice(nonce_final); // 96-bits; unique per message

    let plaintext = cipher.decrypt(nonce, pass.as_ref());
    match plaintext {
        Ok(res) => {
            Ok(vecu8_to_string(res))
        }
        Err(_) => {
            Err("[ERROR] Error decrypting data, check your master password".to_string())
        }
    }
}

pub fn get_home_folder_str() -> String {
    match home::home_dir() {
        Some(path) => {
            let p = path.to_str();
            match p {
                Some(pa) => pa.to_owned(),
                None => "".to_owned(),
            }
        },
        None => "".to_owned(),
    }
}

pub fn get_ipass_folder() -> String {
    let path = get_home_folder_str()+"/.IPass/";
    std::fs::create_dir_all(&path).unwrap();
    path
}

pub fn create_entry(name: &String, pw: String, mpw: String) -> bool {
    if std::path::Path::new(&(get_ipass_folder()+name+".ipass")).exists() {
        return false;
    }
    // println!("{pw}");
    let pw = encrypt_pass(name.to_owned(), pw,mpw);
    let mut file = File::create(get_ipass_folder()+name+".ipass").unwrap();
    file.write_all(pw.as_bytes()).unwrap();
    true
}

fn read_entry(name:&String,mpw:String) -> Result<String,String> {
    let content = &mut read_to_string(get_ipass_folder()+name+".ipass").expect("Should have been able to read the file");
    decrypt_pass(name.to_owned(),hex::decode(content).unwrap(),mpw)
}

pub fn get_entry(name:&String, mpw: String) -> Result<String,String> {
    read_entry(name,mpw)
}

pub fn edit_password(name:&String, password:String, mpw: String) -> bool {
    let entry_result = read_entry(name, mpw.clone());
    if let Ok(entry) = entry_result {
        // println!("entry: {entry}");
        let mut parts = entry.split(';');
        let username = parts.next().unwrap().to_string();
        let _old_password = parts.next().unwrap();
        let data = encrypt_pass(name.to_owned(), username+";"+password.as_str(),mpw);
        let mut file = File::create(get_ipass_folder()+name+".ipass").unwrap();
        file.write_all(data.as_bytes()).unwrap();
        return true;
    }
    false
}

pub fn edit_username(name:&String, username: String, mpw: String) -> bool {
    let entry_result = read_entry(name, mpw.clone());
    if let Ok(entry) = entry_result {
        // println!("entry: {entry}");
        let mut parts = entry.split(';');
        let _old_username = parts.next().unwrap();
        let password = parts.next().unwrap();
        let data = encrypt_pass(name.to_owned(), username+";"+password,mpw);
        let mut file = File::create(get_ipass_folder()+name+".ipass").unwrap();
        file.write_all(data.as_bytes()).unwrap();
        return true;
    }
    false
}

pub fn prompt_answer(toprint: String) -> String {
    prompt_answer_nolower(toprint).to_lowercase()
}

pub fn prompt_answer_nolower(toprint: String) -> String {
    print!("{toprint}");
    std::io::stdout().flush().unwrap();
    let mut choice = String::new();
    std::io::stdin().read_line(&mut choice).expect("Failed to read choice");

    return choice.trim().to_string();
}

pub fn rename(name: &String, new_name: &String, mpw: String) -> bool {
    if !std::path::Path::new(&(get_ipass_folder()+name+".ipass")).exists() {
        return false;
    }
    if std::path::Path::new(&(get_ipass_folder()+new_name+".ipass")).exists() {
        return false;
    }
    let content = &mut read_to_string(get_ipass_folder()+name+".ipass").expect("Should have been able to read the file");
    if let Ok(mut data) = decrypt_pass(name.to_owned(),hex::decode(content).unwrap(),mpw.clone()) {
        data = encrypt_pass(new_name.to_owned(), data,mpw);
        let mut file = File::create(get_ipass_folder()+new_name+".ipass").unwrap();
        file.write_all(data.as_bytes()).unwrap();
        return true;
    }
    false
}

pub fn get_entries() -> std::fs::ReadDir {
    read_dir(get_ipass_folder()).unwrap()
}

pub fn random_password() -> String {
    let alphabet: &str = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890!\"$%&/()=?{[]}\\,.-;:_><|+*#'";
    let alph_len: usize = alphabet.chars().count();
    let char_set:Vec<char> = alphabet.chars().collect();
    let mut chars_index: Vec<u8> = vec![0;20];
    OsRng.fill_bytes(&mut chars_index);
    let mut chars: String = String::new();

    for index in chars_index {
        // println!("{} - {} - {}",index,(index as usize)%(alph_len-1),alph_len);
        chars += &char_set[(index as usize)%(alph_len-1)].to_string();
    }
    chars
}