sway-core/src/language/call_path.rs - master - githubmirror/sway

sway-core/src/language/call_path.rs
656 строк25 KB
Igor Rončević
Remove cloning of parsed tree elements during type checking (#7667)
25 июн 2026, 04:30
25 июн 2026, 04:30794943d
use crate::{
    engine_threading::{
        DebugWithEngines, DisplayWithEngines, EqWithEngines, HashWithEngines, OrdWithEngines,
        OrdWithEnginesContext, PartialEqWithEngines, PartialEqWithEnginesContext,
    },
    parsed::QualifiedPathType,
    Engines, GenericArgument, Ident, Namespace,
};
use serde::{Deserialize, Serialize};
use std::{
    cmp::Ordering,
    fmt,
    hash::{Hash, Hasher},
    sync::Arc,
};
use sway_error::{
    error::CompileError,
    handler::{ErrorEmitted, Handler},
};
use sway_types::{span::Span, Spanned};

#[derive(Clone, Debug, Serialize, Deserialize)]
pub struct CallPathTree {
    pub qualified_call_path: QualifiedCallPath,
    pub children: Vec<CallPathTree>,
}

impl HashWithEngines for CallPathTree {
    fn hash<H: Hasher>(&self, state: &mut H, engines: &Engines) {
        let CallPathTree {
            qualified_call_path,
            children,
        } = self;
        qualified_call_path.hash(state, engines);
        children.hash(state, engines);
    }
}

impl EqWithEngines for CallPathTree {}
impl PartialEqWithEngines for CallPathTree {
    fn eq(&self, other: &Self, ctx: &PartialEqWithEnginesContext) -> bool {
        let CallPathTree {
            qualified_call_path,
            children,
        } = self;
        qualified_call_path.eq(&other.qualified_call_path, ctx) && children.eq(&other.children, ctx)
    }
}

impl<T: PartialEqWithEngines> EqWithEngines for Vec<T> {}
impl<T: PartialEqWithEngines> PartialEqWithEngines for Vec<T> {
    fn eq(&self, other: &Self, ctx: &PartialEqWithEnginesContext) -> bool {
        if self.len() != other.len() {
            return false;
        }
        self.iter().zip(other.iter()).all(|(a, b)| a.eq(b, ctx))
    }
}

impl OrdWithEngines for CallPathTree {
    fn cmp(&self, other: &Self, ctx: &OrdWithEnginesContext) -> Ordering {
        let CallPathTree {
            qualified_call_path: l_call_path,
            children: l_children,
        } = self;
        let CallPathTree {
            qualified_call_path: r_call_path,
            children: r_children,
        } = other;
        l_call_path
            .cmp(r_call_path, ctx)
            .then_with(|| l_children.cmp(r_children, ctx))
    }
}

#[derive(Clone, Debug, Serialize, Deserialize)]

pub struct QualifiedCallPath {
    pub call_path: CallPath,
    pub qualified_path_root: Option<Box<QualifiedPathType>>,
}

impl std::convert::From<Ident> for QualifiedCallPath {
    fn from(other: Ident) -> Self {
        QualifiedCallPath {
            call_path: CallPath {
                prefixes: vec![],
                suffix: other,
                callpath_type: CallPathType::Ambiguous,
            },
            qualified_path_root: None,
        }
    }
}

impl std::convert::From<CallPath> for QualifiedCallPath {
    fn from(other: CallPath) -> Self {
        QualifiedCallPath {
            call_path: other,
            qualified_path_root: None,
        }
    }
}

impl QualifiedCallPath {
    pub fn to_call_path(self, handler: &Handler) -> Result<CallPath, ErrorEmitted> {
        if let Some(qualified_path_root) = self.qualified_path_root {
            Err(handler.emit_err(CompileError::Internal(
                "Unexpected qualified path.",
                qualified_path_root.as_trait_span,
            )))
        } else {
            Ok(self.call_path)
        }
    }
}

impl HashWithEngines for QualifiedCallPath {
    fn hash<H: Hasher>(&self, state: &mut H, engines: &Engines) {
        let QualifiedCallPath {
            call_path,
            qualified_path_root,
        } = self;
        call_path.hash(state);
        qualified_path_root.hash(state, engines);
    }
}

impl EqWithEngines for QualifiedCallPath {}
impl PartialEqWithEngines for QualifiedCallPath {
    fn eq(&self, other: &Self, ctx: &PartialEqWithEnginesContext) -> bool {
        let QualifiedCallPath {
            call_path,
            qualified_path_root,
        } = self;
        PartialEqWithEngines::eq(call_path, &other.call_path, ctx)
            && qualified_path_root.eq(&other.qualified_path_root, ctx)
    }
}

impl OrdWithEngines for QualifiedCallPath {
    fn cmp(&self, other: &Self, ctx: &OrdWithEnginesContext) -> Ordering {
        let QualifiedCallPath {
            call_path: l_call_path,
            qualified_path_root: l_qualified_path_root,
        } = self;
        let QualifiedCallPath {
            call_path: r_call_path,
            qualified_path_root: r_qualified_path_root,
        } = other;
        l_call_path
            .cmp(r_call_path)
            .then_with(|| l_qualified_path_root.cmp(r_qualified_path_root, ctx))
    }
}

impl DisplayWithEngines for QualifiedCallPath {
    fn fmt(&self, f: &mut fmt::Formatter<'_>, engines: &Engines) -> fmt::Result {
        if let Some(qualified_path_root) = &self.qualified_path_root {
            write!(
                f,
                "{}::{}",
                engines.help_out(qualified_path_root),
                &self.call_path
            )
        } else {
            write!(f, "{}", &self.call_path)
        }
    }
}

impl DebugWithEngines for QualifiedCallPath {
    fn fmt(&self, f: &mut fmt::Formatter<'_>, engines: &Engines) -> fmt::Result {
        write!(f, "{}", engines.help_out(self))
    }
}

#[derive(Debug, Copy, Clone, Eq, PartialEq, Hash, Ord, PartialOrd, Serialize, Deserialize)]
pub enum CallPathType {
    /// An unresolved path on the form `::X::Y::Z`. The path must be resolved relative to the
    /// current package root module.
    /// The path can be converted to a full path by prepending the package name, so if the path
    /// `::X::Y::Z` occurs in package `A`, then the corresponding full path will be `A::X::Y::Z`.
    RelativeToPackageRoot,
    /// An unresolved path on the form `X::Y::Z`. The path must either be resolved relative to the
    /// current module, in which case `X` is either a submodule or a name bound in the current
    /// module, or as a full path, in which case `X` is the name of an external package.
    /// If the path is resolved relative to the current module, and the current module has a module
    /// path `A::B::C`, then the corresponding full path is `A::B::C::X::Y::Z`.
    /// If the path is resolved as a full path, then the full path is `X::Y::Z`.
    Ambiguous,
    /// A full path on the form `X::Y::Z`. The first identifier `X` is the name of either the
    /// current package or an external package.
    /// After that comes a (possibly empty) series of names of submodules. Then comes the name of an
    /// item (a type, a trait, a function, or something else declared in that module). Additionally,
    /// there may be additional names such as the name of an enum variant or associated types.
    Full,
}

/// In the expression `a::b::c()`, `a` and `b` are the prefixes and `c` is the suffix.
/// `c` can be any type `T`, but in practice `c` is either an `Ident` or a `TypeInfo`.
#[derive(Debug, Clone, Eq, PartialEq, Hash, Ord, PartialOrd, Serialize, Deserialize)]
pub struct CallPath<T = Ident> {
    pub prefixes: Vec<Ident>,
    pub suffix: T,
    pub callpath_type: CallPathType,
}

impl EqWithEngines for CallPath {}
impl PartialEqWithEngines for CallPath {
    fn eq(&self, other: &Self, _ctx: &PartialEqWithEnginesContext) -> bool {
        self.prefixes == other.prefixes
            && self.suffix == other.suffix
            && self.callpath_type == other.callpath_type
    }
}

impl<T: EqWithEngines> EqWithEngines for CallPath<T> {}
impl<T: PartialEqWithEngines> PartialEqWithEngines for CallPath<T> {
    fn eq(&self, other: &Self, ctx: &PartialEqWithEnginesContext) -> bool {
        self.prefixes == other.prefixes
            && self.suffix.eq(&other.suffix, ctx)
            && self.callpath_type == other.callpath_type
    }
}

impl<T: OrdWithEngines> OrdWithEngines for CallPath<T> {
    fn cmp(&self, other: &Self, ctx: &OrdWithEnginesContext) -> Ordering {
        self.prefixes
            .cmp(&other.prefixes)
            .then_with(|| self.suffix.cmp(&other.suffix, ctx))
            .then_with(|| self.callpath_type.cmp(&other.callpath_type))
    }
}

#[derive(Debug, Clone, Eq, PartialEq, Hash, Ord, PartialOrd)]
pub struct ResolvedCallPath<T, U = Ident> {
    pub decl: T,
    pub unresolved_call_path: CallPath<U>,
}

impl std::convert::From<Ident> for CallPath {
    fn from(other: Ident) -> Self {
        CallPath {
            prefixes: vec![],
            suffix: other,
            callpath_type: CallPathType::Ambiguous,
        }
    }
}

impl<T> fmt::Display for CallPath<T>
where
    T: fmt::Display,
{
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        // TODO: Remove this workaround once https://github.com/FuelLabs/sway/issues/7304 is fixed
        //       and uncomment the original code below.

        if let Some((first_prefix, rest_prefixes)) = self.prefixes.split_first() {
            let first_prefix = if !first_prefix.as_str().contains('-') {
                first_prefix.as_str()
            } else {
                &first_prefix.as_str().replace('-', "_")
            };
            write!(f, "{first_prefix}::")?;
            for prefix in rest_prefixes {
                write!(f, "{}::", prefix.as_str())?;
            }
        }
        write!(f, "{}", &self.suffix)

        // for prefix in self.prefixes.iter() {
        //     write!(f, "{}::", prefix.as_str())?;
        // }
        // write!(f, "{}", &self.suffix)
    }
}

impl<T: DisplayWithEngines> DisplayWithEngines for CallPath<T> {
    fn fmt(&self, f: &mut fmt::Formatter<'_>, engines: &Engines) -> fmt::Result {
        // TODO: Remove this workaround once https://github.com/FuelLabs/sway/issues/7304 is fixed
        //       and uncomment the original code below.

        if let Some((first_prefix, rest_prefixes)) = self.prefixes.split_first() {
            let first_prefix = if !first_prefix.as_str().contains('-') {
                first_prefix.as_str()
            } else {
                &first_prefix.as_str().replace('-', "_")
            };
            write!(f, "{first_prefix}::")?;
            for prefix in rest_prefixes {
                write!(f, "{}::", prefix.as_str())?;
            }
        }
        write!(f, "{}", engines.help_out(&self.suffix))

        // for prefix in self.prefixes.iter() {
        //     write!(f, "{}::", prefix.as_str())?;
        // }
        // write!(f, "{}", engines.help_out(&self.suffix))
    }
}

impl<T: DisplayWithEngines> DebugWithEngines for CallPath<T> {
    fn fmt(&self, f: &mut fmt::Formatter<'_>, engines: &Engines) -> fmt::Result {
        for prefix in self.prefixes.iter() {
            write!(f, "{}::", prefix.as_str())?;
        }
        write!(f, "{}", engines.help_out(&self.suffix))
    }
}

impl<T: Spanned> Spanned for CallPath<T> {
    fn span(&self) -> Span {
        if self.prefixes.is_empty() {
            self.suffix.span()
        } else {
            let suffix_span = self.suffix.span();
            let mut prefixes_spans = self
                .prefixes
                .iter()
                .map(|x| x.span())
                // Depending on how the call path is constructed, we
                // might have a situation that the parts do not belong
                // to the same source and do not have the same source id.
                // In that case, we will take only the suffix' span, as
                // the span for the whole call path. Otherwise, we join
                // the spans of all the parts.
                .filter(|x| {
                    Arc::ptr_eq(&x.src().text, &suffix_span.src().text)
                        && x.source_id() == suffix_span.source_id()
                })
                .peekable();
            if prefixes_spans.peek().is_some() {
                Span::join(Span::join_all(prefixes_spans), &suffix_span)
            } else {
                suffix_span
            }
        }
    }
}

/// This controls the type of display type for call path display string conversions.
pub enum CallPathDisplayType {
    /// Prints the regular call path as exists internally.
    Regular,
    /// Strips the current root package if it exists as prefix.
    StripPackagePrefix,
}

impl CallPath {
    pub fn fullpath(path: &[&str]) -> Self {
        assert!(!path.is_empty());

        CallPath {
            prefixes: path
                .iter()
                .take(path.len() - 1)
                .map(|&x| Ident::new_no_span(x.into()))
                .collect(),
            suffix: path.last().map(|&x| Ident::new_no_span(x.into())).unwrap(),
            callpath_type: CallPathType::Full,
        }
    }

    /// Shifts the last prefix into the suffix, and removes the old suffix.
    /// Does nothing if prefixes are empty, or if the path is a full path and there is only a single prefix (which must be the package name, which is obligatory for full paths)
    pub fn rshift(&self) -> CallPath {
        if self.prefixes.is_empty()
            || (matches!(self.callpath_type, CallPathType::Full) && self.prefixes.len() == 1)
        {
            self.clone()
        } else {
            CallPath {
                prefixes: self.prefixes[0..self.prefixes.len() - 1].to_vec(),
                suffix: self.prefixes.last().unwrap().clone(),
                callpath_type: self.callpath_type,
            }
        }
    }

    /// Removes the first prefix. Does nothing if prefixes are empty.
    pub fn lshift(&self) -> CallPath {
        if self.prefixes.is_empty() {
            self.clone()
        } else {
            let new_callpath_type = match self.callpath_type {
                CallPathType::RelativeToPackageRoot | CallPathType::Ambiguous => {
                    CallPathType::Ambiguous
                }
                CallPathType::Full => CallPathType::RelativeToPackageRoot,
            };
            CallPath {
                prefixes: self.prefixes[1..self.prefixes.len()].to_vec(),
                suffix: self.suffix.clone(),
                callpath_type: new_callpath_type,
            }
        }
    }

    pub fn as_vec_string(&self) -> Vec<String> {
        self.prefixes
            .iter()
            .map(|p| p.to_string())
            .chain(std::iter::once(self.suffix.to_string()))
            .collect::<Vec<_>>()
    }

    pub fn as_vec_ident(&self) -> Vec<Ident> {
        self.as_vec_string()
            .iter()
            .map(|s| Ident::new_no_span(s.clone()))
            .collect::<Vec<_>>()
    }

    /// Create a full [CallPath] from a given [Ident] and the [Namespace] in which the [Ident] is
    /// declared.
    ///
    /// This function is intended to be used while typechecking the identifier declaration, i.e.,
    /// before the identifier is added to the environment.
    pub fn ident_to_fullpath(suffix: Ident, namespace: &Namespace) -> CallPath {
        let mut res: Self = suffix.into();
        for mod_path in namespace.current_mod_path() {
            res.prefixes.push(mod_path.clone())
        }
        res.callpath_type = CallPathType::Full;
        res
    }

    /// Convert a given [CallPath] into a call path suitable for a `use` statement.
    ///
    /// For example, given a path `pkga::SOME_CONST` where `pkga` is an _internal_ library of a package named
    /// `my_project`, the corresponding call path is `pkga::SOME_CONST`.
    ///
    /// Paths to _external_ libraries such `std::lib1::lib2::my_obj` are left unchanged.
    pub fn to_import_path(&self, engines: &Engines, namespace: &Namespace) -> CallPath {
        let converted = self.to_fullpath(engines, namespace);

        if let Some(first) = converted.prefixes.first() {
            if namespace.current_package_name() == first {
                return converted.lshift();
            }
        }
        converted
    }

    pub fn to_display_path(
        &self,
        display_type: CallPathDisplayType,
        namespace: &Namespace,
    ) -> CallPath {
        let mut display_path = self.clone();

        match display_type {
            CallPathDisplayType::Regular => {}
            CallPathDisplayType::StripPackagePrefix => {
                if let Some(first) = self.prefixes.first() {
                    if namespace.current_package_name() == first {
                        display_path = display_path.lshift();
                    }
                }
            }
        };

        display_path
    }

    /// Create a string form of the given [CallPath] and zero or more [TypeArgument]s.
    /// The returned string is convenient for displaying full names, including generic arguments, in help messages.
    /// E.g.:
    /// - `some::module::SomeType`
    /// - `some::module::SomeGenericType<T, u64>`
    ///
    /// Note that the trailing arguments are never separated by `::` from the suffix.
    pub(crate) fn to_string_with_args(
        &self,
        engines: &Engines,
        args: &[GenericArgument],
    ) -> String {
        let args = args
            .iter()
            .map(|type_arg| engines.help_out(type_arg).to_string())
            .collect::<Vec<_>>()
            .join(", ");

        format!(
            "{}{}",
            // TODO: Replace with a context aware string representation of the path
            //       once https://github.com/FuelLabs/sway/issues/6873 is fixed.
            &self,
            if args.is_empty() {
                String::new()
            } else {
                format!("<{args}>")
            }
        )
    }
}

impl<T: Clone> CallPath<T> {
    /// Convert a given [CallPath] to a symbol to a full [CallPath] to a program point in which the
    /// symbol can be resolved (assuming the given [CallPath] is a legal Sway path).
    ///
    /// The resulting [CallPath] is not guaranteed to be located in the package where the symbol is
    /// declared. To obtain the path to the declaration, use [to_canonical_path].
    ///
    /// The [CallPath] is converted within the current module of the supplied namespace.
    ///
    /// For example, given a path `pkga::SOME_CONST` where `pkga` is an _internal_ module of a
    /// package named `my_project`, the corresponding call path is
    /// `my_project::pkga::SOME_CONST`. This does not imply that `SOME_CONST` is declared in the
    /// `my_project::pkga`, but only that the name `SOME_CONST` is bound in `my_project::pkga`.
    ///
    /// Paths to _external_ libraries such `std::lib1::lib2::my_obj` are considered full already
    /// and are left unchanged since `std` is a root of the package `std`.
    pub fn to_fullpath(&self, engines: &Engines, namespace: &Namespace) -> CallPath<T> {
        self.to_fullpath_from_mod_path(engines, namespace, namespace.current_mod_path())
    }

    /// Convert a given [CallPath] to a symbol to a full [CallPath] to a program point in which the
    /// symbol can be resolved (assuming the given [CallPath] is a legal Sway path).
    ///
    /// The resulting [CallPath] is not guaranteed to be located in the package where the symbol is
    /// declared. To obtain the path to the declaration, use [to_canonical_path].
    ///
    /// The [CallPath] is converted within the module given by `mod_path`, which must be a legal
    /// path to a module.
    ///
    /// For example, given a path `pkga::SOME_CONST` where `pkga` is an _internal_ module of a
    /// package named `my_project`, the corresponding call path is
    /// `my_project::pkga::SOME_CONST`. This does not imply that `SOME_CONST` is declared in the
    /// `my_project::pkga`, but only that the name `SOME_CONST` is bound in `my_project::pkga`.
    ///
    /// Paths to _external_ libraries such `std::lib1::lib2::my_obj` are considered full already
    /// and are left unchanged since `std` is a root of the package `std`.
    pub fn to_fullpath_from_mod_path(
        &self,
        engines: &Engines,
        namespace: &Namespace,
        mod_path: &Vec<Ident>,
    ) -> CallPath<T> {
        let mod_path_module = namespace.module_from_absolute_path(mod_path);

        match self.callpath_type {
            CallPathType::Full => self.clone(),
            CallPathType::RelativeToPackageRoot => {
                let mut prefixes = vec![mod_path[0].clone()];
                for ident in self.prefixes.iter() {
                    prefixes.push(ident.clone());
                }
                Self {
                    prefixes,
                    suffix: self.suffix.clone(),
                    callpath_type: CallPathType::Full,
                }
            }
            CallPathType::Ambiguous => {
                if self.prefixes.is_empty() {
                    // Given a path to a symbol that has no prefixes, discover the path to the symbol as a
                    // combination of the package name in which the symbol is defined and the path to the
                    // current submodule.
                    CallPath {
                        prefixes: mod_path.clone(),
                        suffix: self.suffix.clone(),
                        callpath_type: CallPathType::Full,
                    }
                } else if mod_path_module.is_some()
                    && (mod_path_module.unwrap().has_submodule(&self.prefixes[0])
                        || namespace.module_has_binding(engines, mod_path, &self.prefixes[0]))
                {
                    // The first identifier in the prefix is a submodule of the current
                    // module.
                    //
                    // The path is a qualified path relative to the current module
                    //
                    // Complete the path by prepending the package name and the path to the current module.
                    CallPath {
                        prefixes: mod_path.iter().chain(&self.prefixes).cloned().collect(),
                        suffix: self.suffix.clone(),
                        callpath_type: CallPathType::Full,
                    }
                } else if namespace.package_exists(&self.prefixes[0])
                    && namespace.module_is_external(&self.prefixes)
                {
                    // The first identifier refers to an external package. The path is already fully qualified.
                    CallPath {
                        prefixes: self.prefixes.clone(),
                        suffix: self.suffix.clone(),
                        callpath_type: CallPathType::Full,
                    }
                } else {
                    // The first identifier in the prefix is neither a submodule of the current module nor the name of an external package.
                    // This is probably an illegal path, so let it fail by assuming it is bound in the current module.
                    CallPath {
                        prefixes: mod_path.iter().chain(&self.prefixes).cloned().collect(),
                        suffix: self.suffix.clone(),
                        callpath_type: CallPathType::Full,
                    }
                }
            }
        }
    }
}

impl CallPath {
    /// Convert a given [CallPath] to a symbol to a full [CallPath] to where the symbol is declared
    /// (assuming the given [CallPath] is a legal Sway path).
    ///
    /// The [CallPath] is converted within the current module of the supplied namespace.
    ///
    /// For example, given a path `pkga::SOME_CONST` where `pkga` is an _internal_ module of a
    /// package named `my_project`, and `SOME_CONST` is bound in the module `my_project::pkga`, then
    /// the corresponding call path is the full callpath to the declaration that `SOME_CONST` is
    /// bound to. This does not imply that `SOME_CONST` is declared in the `my_project::pkga`, since
    /// the binding may be the result of an import.
    ///
    /// Paths to _external_ libraries such `std::lib1::lib2::my_obj` are considered full already
    /// and are left unchanged since `std` is a root of the package `std`.
    pub fn to_canonical_path(&self, engines: &Engines, namespace: &Namespace) -> CallPath {
        // Generate a full path to a module where the suffix can be resolved
        let full_path = self.to_fullpath(engines, namespace);

        match namespace.module_from_absolute_path(&full_path.prefixes) {
            Some(module) => {
                // Resolve the path suffix in the found module
                match module.resolve_symbol(&Handler::default(), engines, &full_path.suffix) {
                    Ok((decl, decl_path)) => {
                        let name = decl.expect_typed().get_name(engines);
                        let suffix = if name.as_str() != full_path.suffix.as_str() {
                            name
                        } else {
                            full_path.suffix
                        };
                        // Replace the resolvable path with the declaration's path
                        CallPath {
                            prefixes: decl_path,
                            suffix,
                            callpath_type: full_path.callpath_type,
                        }
                    }
                    Err(_) => {
                        // The symbol does not resolve. The symbol isn't bound, so the best bet is
                        // the full path.
                        full_path
                    }
                }
            }
            None => {
                // The resolvable module doesn't exist. The symbol probably doesn't exist, so
                // the best bet is the full path.
                full_path
            }
        }
    }
}