"""Functions used to generate source files during build time""" import re import methods RE_PATH_SPLIT = re.compile(r"[^/\\]+?(?=\.)") TEMPLATE = """\ #ifndef _WIN32 #define TEST_DLL_PRIVATE __attribute__((visibility("hidden"))) #else #define TEST_DLL_PRIVATE #endif // _WIN32 namespace ForceLink {{ TEST_DLL_PRIVATE void force_link_tests(); {TESTS_DECLARE} }} // namespace ForceLink void ForceLink::force_link_tests() {{ {TESTS_CALL} }} """ def force_link_builder(target, source, env): names = [RE_PATH_SPLIT.search(str(path)).group() for path in source[0].read()] declares = [f"TEST_DLL_PRIVATE void force_link_{name}();" for name in names] calls = [f"force_link_{name}();" for name in names] with methods.generated_wrapper(str(target[0])) as file: file.write( TEMPLATE.format( TESTS_DECLARE="\n\t".join(declares), TESTS_CALL="\n\t".join(calls), ) )