starkli 0.1.4

Starkli (/ˈstɑːrklaɪ/), a blazing fast CLI tool for Starknet powered by starknet-rs
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

use anyhow::Result;
use clap::Parser;
use colored::Colorize;
use starknet::core::types::{contract::SierraClass, FieldElement};

use crate::{
    compiler::{BuiltInCompiler, CompilerVersion},
    network::{Network, NetworkSource},
};

#[derive(Debug, Clone, Parser)]
pub struct CasmArgs {
    #[clap(long, help = "Statically-linked Sierra compiler version")]
    compiler_version: Option<CompilerVersion>,
    #[clap(long, help = "Override Sierra compilation and use CASM hash directly")]
    casm_hash: Option<String>,
}

#[derive(Debug)]
pub enum CasmHashSource {
    BuiltInCompiler(BuiltInCompiler),
    Hash(FieldElement),
}

impl CasmArgs {
    pub async fn into_casm_hash_source<N>(self, network_source: N) -> Result<CasmHashSource>
    where
        N: NetworkSource,
    {
        match (self.compiler_version, self.casm_hash) {
            (Some(compiler_version), None) => {
                Ok(CasmHashSource::BuiltInCompiler(compiler_version.into()))
            }
            (None, Some(casm_hash)) => Ok(CasmHashSource::Hash(casm_hash.parse()?)),
            // Tries to detect compiler version if nothing provided
            (None, None) => {
                eprintln!(
                    "Sierra compiler version not specified. \
                    Attempting to automatically decide version to use..."
                );

                let network = network_source.get_network().await?;
                match network {
                    Some(network) => {
                        let auto_version = match network {
                            Network::Mainnet
                            | Network::Goerli1
                            | Network::Goerli2
                            | Network::Integration => CompilerVersion::V2_0_1,
                        };

                        eprintln!(
                            "Network detected: {}. \
                            Using the default compiler version for this network: {}. \
                            Use the --compiler-version flag to choose a different version.",
                            format!("{}", network).bright_yellow(),
                            format!("{}", auto_version).bright_yellow()
                        );

                        Ok(CasmHashSource::BuiltInCompiler(auto_version.into()))
                    }
                    None => {
                        let default_version: CompilerVersion = Default::default();

                        eprintln!(
                            "Unknown network. Falling back to the default compiler version {}. \
                            Use the --compiler-version flag to choose a different version.",
                            format!("{}", default_version).bright_yellow()
                        );

                        Ok(CasmHashSource::BuiltInCompiler(default_version.into()))
                    }
                }
            }
            _ => Err(anyhow::anyhow!(
                "invalid casm hash options. \
                Use either --compiler-version or --casm-hash but not at the same time"
            )),
        }
    }
}

impl CasmHashSource {
    pub fn get_casm_hash(&self, sierra_class: &SierraClass) -> Result<FieldElement> {
        match self {
            Self::BuiltInCompiler(compiler) => compiler.compile(sierra_class),
            Self::Hash(hash) => Ok(*hash),
        }
    }
}