fix[ux]: include imported structs in interface output

tests/functional/codegen/test_interfaces.py

@@ -774,3 +774,67 @@ def foo(s: MyStruct) -> MyStruct:
     assert "b: uint256" in out
     assert "struct Voter:" in out
     assert "voted: bool" in out
+
+
+def test_interface_with_imported_structures(make_input_bundle):
+    a = """
+import b
+
+struct Foo:
+    val:uint256
+        """
+    b = """
+import c
+
+struct Bar:
+    val:uint256
+        """
+    c = """
+struct Baz:
+    val:uint256
+        """
+    input_bundle = make_input_bundle({"a.vy": a, "b.vy": b, "c.vy": c})
+    out = compile_code(
+        a, input_bundle=input_bundle, contract_path="a.vy", output_formats=["interface"]
+    )["interface"]
+
+    assert "# Structs" in out
+    assert "struct Foo:" in out
+    assert "struct Bar:" in out
+    assert "struct Baz" in out
+
+
+def test_interface_with_doubly_imported_interface(make_input_bundle):
+    a = """
+import b
+import c
+
+struct Foo:
+    val:uint256
+        """
+    b = """
+import d
+
+struct Bar:
+    val:uint256
+        """
+    c = """
+import d
+struct Baz:
+    val:uint256
+        """
+    d = """
+struct Boo:
+    val:uint256
+        """
+
+    input_bundle = make_input_bundle({"a.vy": a, "b.vy": b, "c.vy": c, "d.vy": d})
+    out = compile_code(
+        a, input_bundle=input_bundle, contract_path="a.vy", output_formats=["interface"]
+    )["interface"]
+
+    assert "# Structs" in out
+    assert "struct Foo:" in out
+    assert "struct Bar:" in out
+    assert "struct Baz" in out
+    assert out.count("struct Boo") == 1

vyper/compiler/output.py

@@ -14,7 +14,7 @@
 from vyper.ir import compile_ir
 from vyper.semantics.analysis.base import ModuleInfo
 from vyper.semantics.types.function import FunctionVisibility, StateMutability
-from vyper.semantics.types.module import InterfaceT
+from vyper.semantics.types.module import InterfaceT, ModuleT, StructT
 from vyper.typing import StorageLayout
 from vyper.utils import vyper_warn
 from vyper.warnings import ContractSizeLimitWarning
@@ -125,12 +125,14 @@ def build_external_interface_output(compiler_data: CompilerData) -> str:
 
 
 def build_interface_output(compiler_data: CompilerData) -> str:
-    interface = compiler_data.annotated_vyper_module._metadata["type"].interface
+    module_t = compiler_data.annotated_vyper_module._metadata["type"]
+    interface = module_t.interface
     out = ""
 
-    if len(interface.structs) > 0:
+    structs = _get_structs(module_t)
+    if len(structs) > 0:
         out += "# Structs\n\n"
-        for struct in interface.structs.values():
+        for struct in structs:
             out += f"struct {struct.name}:\n"
             for member_name, member_type in struct.members.items():
                 out += f"    {member_name}: {member_type}\n"
@@ -159,6 +161,19 @@ def build_interface_output(compiler_data: CompilerData) -> str:
     return out
 
 
+def _get_structs(m: ModuleT, visited: set[ModuleT] = None) -> [StructT]:
+    visited = visited or set()
+    if m in visited:
+        return []
+    visited.add(m)
+    structs = list(m.interface.structs.values())
+
+    for val in m.imported_modules.values():
+        structs += _get_structs(val.module_node._metadata["type"], visited)
+
+    return structs
+
+
 def build_bb_output(compiler_data: CompilerData) -> IRnode:
     return compiler_data.venom_functions[0]