collect_structs: visit function args and returns

Prevent compilation error in the generated code when dealing with
structs only referenced by public functions.

Fixes #2655