sway-core 0.37.3

use crate::{language::parsed::TreeType, BuildTarget};

#[derive(Default)]
pub struct Context {
    /// Indicates whether the module being parsed has a `configurable` block
    module_has_configurable_block: bool,

    /// Unique suffix used to generate unique names for destructured structs
    destructured_struct_unique_suffix: usize,

    /// Unique suffix used to generate unique names for destructured tuples
    destructured_tuple_unique_suffix: usize,

    /// Unique suffix used to generate unique names for vars returned from `match` expressions
    match_expression_return_var_unique_suffix: usize,

    /// The build target
    build_target: BuildTarget,

    /// The build target
    program_type: Option<TreeType>,
}

impl Context {
    /// Create a new context
    pub fn new(build_target: BuildTarget) -> Self {
        Self {
            build_target,
            ..Default::default()
        }
    }

    /// Update the value of `module_has_configurable_block`
    pub fn set_module_has_configurable_block(&mut self, val: bool) {
        self.module_has_configurable_block = val;
    }

    /// Returns whether the module being parsed has a `configurable` block
    pub fn module_has_configurable_block(&self) -> bool {
        self.module_has_configurable_block
    }

    /// Returns a suffix used to generate a unique name for a destructured struct
    pub fn next_destructured_struct_unique_suffix(&mut self) -> usize {
        self.destructured_struct_unique_suffix += 1;
        self.destructured_struct_unique_suffix
    }

    /// Returns a unique suffix used to generate a unique name for a destructured tuple
    pub fn next_destructured_tuple_unique_suffix(&mut self) -> usize {
        self.destructured_tuple_unique_suffix += 1;
        self.destructured_tuple_unique_suffix
    }

    /// Returns a unique suffix used to generate a unique name for a var returned from a `match`
    /// expressions
    pub fn next_match_expression_return_var_unique_suffix(&mut self) -> usize {
        self.match_expression_return_var_unique_suffix += 1;
        self.match_expression_return_var_unique_suffix
    }

    /// Returns the build target
    pub fn build_target(&self) -> BuildTarget {
        self.build_target
    }

    /// Returns the build target
    pub fn program_type(&self) -> Option<TreeType> {
        self.program_type.clone()
    }

    /// Update the value of `program_type`
    pub fn set_program_type(&mut self, program_type: TreeType) {
        self.program_type = Some(program_type);
    }
}