Emit nested functions

trans/transpiler/phases/emit_cpp/block.py

@@ -46,7 +46,9 @@ class BlockVisitor(NodeVisitor):
                 yield ast.Return()
 
             from transpiler.phases.emit_cpp.function import FunctionVisitor
-            yield from FunctionVisitor(self.scope, CoroutineMode.TASK).emit_block(node.scope, block())
+            yield "{"
+            yield from self.visit_func_decls(block(), node.scope, CoroutineMode.TASK)
+            yield "}"
             return
 
         if emission == FunctionEmissionKind.DECLARATION:
@@ -55,6 +57,17 @@ class BlockVisitor(NodeVisitor):
         if emission == FunctionEmissionKind.DECLARATION:
             yield f"}} {node.name};"
 
+    def visit_func_decls(self, body: list[ast.stmt], inner_scope: Scope, mode = CoroutineMode.ASYNC) -> Iterable[str]:
+        for child in body:
+            from transpiler.phases.emit_cpp.function import FunctionVisitor
+            child_visitor = FunctionVisitor(inner_scope, mode)
+
+            for name, decl in getattr(child, "decls", {}).items():
+                #yield f"decltype({' '.join(self.expr().visit(decl.type))}) {name};"
+                yield from self.visit(decl.type)
+                yield f" {name};"
+            yield from child_visitor.visit(child)
+
     def visit_func_new(self, node: ast.FunctionDef, emission: FunctionEmissionKind, skip_first_arg: bool = False) -> Iterable[str]:
         if emission == FunctionEmissionKind.LAMBDA:
             yield "[&]"
@@ -117,11 +130,7 @@ class BlockVisitor(NodeVisitor):
             from transpiler.phases.emit_cpp.function import FunctionVisitor
             child_visitor = FunctionVisitor(inner_scope, CoroutineMode.ASYNC)
 
-            for name, decl in getattr(child, "decls", {}).items():
-                #yield f"decltype({' '.join(self.expr().visit(decl.type))}) {name};"
-                yield from self.visit(decl.type)
-                yield f" {name};"
-            yield from child_visitor.visit(child)
+        yield from self.visit_func_decls(node.body, inner_scope)
 
         if not has_return and isinstance(node.type.return_type, Promise):
             yield "co_return;"