use super::ParseModule;
use sway_types::Ident;
#[derive(Debug, Clone)]
pub struct ParseProgram {
pub kind: TreeType,
pub root: ParseModule,
}
#[derive(Clone, Debug, PartialEq, Eq)]
pub enum TreeType {
Predicate,
Script,
Contract,
Library { name: Ident },
}
impl std::fmt::Display for TreeType {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(
f,
"{}",
match self {
Self::Predicate => "predicate",
Self::Script => "script",
Self::Contract => "contract",
Self::Library { .. } => "library",
}
)
}
}
impl ParseProgram {
pub(crate) fn exclude_tests(&mut self) {
self.root.tree.exclude_tests()
}
}