sway-core 0.71.0

use strum::EnumString;

use crate::Engines;

use super::ParseModule;

/// A parsed, but not yet type-checked, Sway program.
///
/// Includes all modules in the form of a `ParseModule` tree accessed via the `root`.
#[derive(Debug, Clone)]
pub struct ParseProgram {
    pub kind: TreeType,
    pub root: ParseModule,
}

/// A Sway program can be either a contract, script, predicate, or a library.
///
/// A submodule declared with `mod` can be only a [TreeType::Library].
#[derive(Copy, Clone, Debug, PartialEq, Eq, EnumString)]
pub enum TreeType {
    #[strum(serialize = "predicate")]
    Predicate,
    #[strum(serialize = "script")]
    Script,
    #[strum(serialize = "contract")]
    Contract,
    #[strum(serialize = "library")]
    Library,
}

impl TreeType {
    pub const CFG: &'static [&'static str] = &["contract", "library", "predicate", "script"];
}

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 {
    /// Excludes all test functions from the parse tree.
    pub(crate) fn exclude_tests(&mut self, engines: &Engines) {
        self.root.tree.exclude_tests(engines)
    }
}