use std::str::FromStr;
use serde::{Deserialize, Serialize};
pub const MESSAGE_SELECTOR_INVALID: &str = "Invalid option to --combined-json";
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Serialize, Deserialize)]
pub enum Selector {
#[serde(rename = "abi")]
ABI,
#[serde(rename = "hashes")]
Hashes,
#[serde(rename = "metadata")]
Metadata,
#[serde(rename = "devdoc")]
Devdoc,
#[serde(rename = "userdoc")]
Userdoc,
#[serde(rename = "storage-layout")]
StorageLayout,
#[serde(rename = "transient-storage-layout")]
TransientStorageLayout,
#[serde(rename = "ast")]
AST,
#[serde(rename = "asm")]
ASM,
#[serde(rename = "assembly", skip_serializing)]
Assembly,
#[serde(rename = "bin", skip_serializing)]
Bytecode,
#[serde(rename = "bin-runtime", skip_serializing)]
BytecodeRuntime,
}
impl Selector {
pub fn from_cli(format: &str) -> Vec<anyhow::Result<Self>> {
format
.split(',')
.map(|flag| Self::from_str(flag.trim()))
.collect()
}
pub fn is_source_solc(&self) -> bool {
!matches!(
self,
Self::Assembly | Self::Bytecode | Self::BytecodeRuntime
)
}
}
impl FromStr for Selector {
type Err = anyhow::Error;
fn from_str(string: &str) -> Result<Self, Self::Err> {
match string {
"abi" => Ok(Self::ABI),
"hashes" => Ok(Self::Hashes),
"metadata" => Ok(Self::Metadata),
"devdoc" => Ok(Self::Devdoc),
"userdoc" => Ok(Self::Userdoc),
"storage-layout" => Ok(Self::StorageLayout),
"transient-storage-layout" => Ok(Self::TransientStorageLayout),
"ast" => Ok(Self::AST),
"asm" => Ok(Self::ASM),
"bin" => Ok(Self::Bytecode),
"bin-runtime" => Ok(Self::BytecodeRuntime),
"assembly" => Ok(Self::Assembly),
selector => anyhow::bail!("{MESSAGE_SELECTOR_INVALID}: {selector}"),
}
}
}
impl std::fmt::Display for Selector {
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
match self {
Self::ABI => write!(f, "abi"),
Self::Hashes => write!(f, "hashes"),
Self::Metadata => write!(f, "metadata"),
Self::Devdoc => write!(f, "devdoc"),
Self::Userdoc => write!(f, "userdoc"),
Self::StorageLayout => write!(f, "storage-layout"),
Self::TransientStorageLayout => write!(f, "transient-storage-layout"),
Self::AST => write!(f, "ast"),
Self::ASM => write!(f, "asm"),
Self::Bytecode => write!(f, "bin"),
Self::BytecodeRuntime => write!(f, "bin-runtime"),
Self::Assembly => write!(f, "assembly"),
}
}
}