Gather trait impls from the argument type module when handling trait operator calls.

Fixes #7330.

Author: tritao

sway-core/src/language/call_path.rs

@@ -375,6 +375,13 @@ impl CallPath {
             .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.
     ///

sway-core/src/semantic_analysis/type_check_context.rs

@@ -2,6 +2,7 @@
 use std::collections::{BTreeMap, HashSet};
 
 use crate::{
+    ast_elements::type_argument::GenericTypeArgument,
     decl_engine::{DeclEngineGet, DeclRefFunction, MaterializeConstGenerics},
     engine_threading::*,
     language::{
@@ -11,7 +12,7 @@ use crate::{
     },
     monomorphization::{monomorphize_with_modpath, MonomorphizeHelper},
     namespace::{
-        IsExtendingExistingImpl, IsImplSelf, ModulePath, ResolvedDeclaration,
+        IsExtendingExistingImpl, IsImplSelf, Module, ModulePath, ResolvedDeclaration,
         ResolvedTraitImplItem, TraitMap,
     },
     semantic_analysis::{
@@ -796,6 +797,19 @@ impl<'a> TypeCheckContext<'a> {
             &mut filter_item,
         );
 
+        // grab the items from where the argument type is declared
+        if let Some(MethodName::FromTrait { .. }) = method_name {
+            let type_module = self.get_namespace_module_from_type_id(type_id);
+            if let Ok(type_module) = type_module {
+                TraitMap::find_items_and_trait_key_for_type(
+                    type_module,
+                    self.engines,
+                    type_id,
+                    &mut filter_item,
+                );
+            }
+        }
+
         if item_prefix != self.namespace().current_mod_path.as_slice() {
             // grab the module where the type itself is declared
             let type_module = self
@@ -814,6 +828,33 @@ impl<'a> TypeCheckContext<'a> {
         Ok(matching_item_decl_refs)
     }
 
+    fn get_namespace_module_from_type_id(&self, type_id: TypeId) -> Result<&Module, ErrorEmitted> {
+        let type_info = self.engines().te().get(type_id);
+        if type_info.is_alias() {
+            if let TypeInfo::Alias { ty, .. } = &*type_info {
+                if let Some(GenericTypeArgument { type_id, .. }) = ty.as_type_argument() {
+                    return self.get_namespace_module_from_type_id(*type_id);
+                }
+            }
+        }
+
+        let handler = Handler::default();
+        let call_path = match *type_info {
+            TypeInfo::Enum(decl_id) => self.engines().de().get_enum(&decl_id).call_path.clone(),
+            TypeInfo::Struct(decl_id) => self.engines().de().get_struct(&decl_id).call_path.clone(),
+            _ => {
+                return Err(handler.emit_err(CompileError::Internal(
+                    "No call path for type id",
+                    Span::dummy(),
+                )))
+            }
+        };
+
+        let call_path = call_path.rshift();
+        self.namespace()
+            .require_module_from_absolute_path(&handler, &call_path.as_vec_ident())
+    }
+
     #[inline]
     fn default_numeric_if_needed(
         &self,

sway-core/src/type_system/info.rs

@@ -1342,6 +1342,10 @@ impl TypeInfo {
         matches!(self, TypeInfo::Array(_, _))
     }
 
+    pub fn is_alias(&self) -> bool {
+        matches!(self, TypeInfo::Alias { .. })
+    }
+
     pub fn is_contract(&self) -> bool {
         matches!(self, TypeInfo::Contract)
     }

test/src/e2e_vm_tests/test_programs/should_pass/trait_map_use_impl_in_scope/Forc.lock

@@ -0,0 +1,8 @@
+[[package]]
+name = "std"
+source = "path+from-root-AA6134FC2B337767"
+
+[[package]]
+name = "trait_map_use_impl_in_scope"
+source = "member"
+dependencies = ["std"]

test/src/e2e_vm_tests/test_programs/should_pass/trait_map_use_impl_in_scope/Forc.toml

@@ -0,0 +1,8 @@
+[project]
+name = "trait_map_use_impl_in_scope"
+authors = ["Fuel Labs <[email protected]>"]
+entry = "main.sw"
+license = "Apache-2.0"
+
+[dependencies]
+std = { path = "../../../../../../sway-lib-std" }

test/src/e2e_vm_tests/test_programs/should_pass/trait_map_use_impl_in_scope/src/main.sw

@@ -0,0 +1,9 @@
+script;
+
+use std::time::Time;
+// use std::time::*;
+
+pub fn main() {
+    let duration = Time::now().duration_since(Time::from(0)).unwrap();
+    let _ = duration + duration;
+}

test/src/e2e_vm_tests/test_programs/should_pass/trait_map_use_impl_in_scope/test.toml

@@ -0,0 +1,2 @@
+category = "compile"
+expected_warnings = 0