1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

use ethers::{
    middleware::SignerMiddleware,
    prelude::k256::ecdsa::SigningKey,
    providers::{ Http, Middleware, Provider },
    signers::{ LocalWallet, Signer, Wallet },
    types::U256,
};

use std::{ fs::File, io::BufReader, path::PathBuf, sync::Arc };

use validator::{ validate_email, validate_url };

use eyre::{ eyre, ContextCompat, WrapErr };

use serde::de::DeserializeOwned;

use pdf::file::FileOptions as PdfFile;

use crate::constants::CLI_PATH;

pub type WalletSignerMiddleware = SignerMiddleware<Provider<Http>, Wallet<SigningKey>>;

pub trait Pdf {
    fn pdf_file_check(&self) -> eyre::Result<&PathBuf>;
}

impl Pdf for PathBuf {
    fn pdf_file_check(&self) -> eyre::Result<&Self> {
        let _ = PdfFile::uncached().open(self)?;

        Ok(self)
    }
}

pub fn parse_json<T: DeserializeOwned>(path: &PathBuf) -> eyre::Result<T> {
    let file = File::open(path).wrap_err_with(|| format!("Failed to open file {path:?}"))?;
    let reader = BufReader::new(file);
    let data = serde_json::from_reader(reader)?;

    Ok(data)
}

pub fn validate_links(link: &str) -> eyre::Result<String> {
    if !validate_url(link) {
        return Err(eyre!("Invalid URL"));
    }

    Ok(link.to_string())
}

pub fn validate_emails(email: &str) -> eyre::Result<String> {
    if !validate_email(email) {
        return Err(eyre!("Invalid Contact Email"));
    }

    Ok(email.to_string())
}

pub fn max_length_string(val: &str) -> eyre::Result<String> {
    if val.len() > 28 {
        return Err(eyre!("Name should be less than 28 characters"));
    }

    Ok(val.to_string())
}

pub fn validate_pdf(val: &str) -> eyre::Result<PathBuf> {
    let path_buf = PathBuf::from(val);

    path_buf
        .pdf_file_check()
        .map_err(|e| eyre!("Invalid PDF file. Please upload a valid PDF file: {e}"))?;

    Ok(path_buf)
}

pub fn apply_dotenv() -> eyre::Result<()> {
    let home_dir = dirs::home_dir().wrap_err("Could not find home directory")?;

    let env_path = home_dir.join(format!("{CLI_PATH}/.env"));

    Ok(dotenv::from_path(env_path.as_path())?)
}

pub async fn get_client(
    rpc_url: &str,
    private_key: Option<String>
) -> eyre::Result<(Arc<WalletSignerMiddleware>, U256, Wallet<SigningKey>)> {
    let home_dir = dirs::home_dir().wrap_err("Could not find home directory")?;

    let env_path = home_dir.join(format!("{CLI_PATH}/.env"));

    dotenv::from_path(env_path.as_path())?;

    let wallet_key = match private_key {
        Some(key) => key,
        None => std::env::var("WALLET_KEY")?,
    };

    let provider = Provider::<Http>::try_from(rpc_url)?;

    let chain_id = provider.get_chainid().await?;

    // this wallet's private key
    let wallet = wallet_key.parse::<LocalWallet>()?.with_chain_id(chain_id.as_u64());

    let provider = SignerMiddleware::new(provider, wallet.clone());

    let client = Arc::new(provider);
    Ok((client, chain_id, wallet))
}