compiler: add a flag denoting module functions

Introduce a further uc_function_t structure member indicating whether the
underlying function is a module constructor.

Signed-off-by: Jo-Philipp Wich <jo@mein.io>

2 files changed, 5 insertions, 1 deletions

diff --git a/compiler.c b/compiler.c
index 4878be1..4eae895 100644
--- a/compiler.c
+++ b/compiler.c
@@ -3484,6 +3484,10 @@ uc_compile_from_source(uc_parse_config_t *config, uc_source_t *source, uc_progra
 	uc_compiler_init(&compiler, name, source, 0, progptr,
 		config && config->strict_declarations);
 
+	if (progptr == prog) {
+		compiler.function->module = true;
+	}
+
 	uc_compiler_parse_advance(&compiler);
 
 	while (!uc_compiler_parse_match(&compiler, TK_EOF))
diff --git a/include/ucode/types.h b/include/ucode/types.h
index 0b63501..64ae7cb 100644
--- a/include/ucode/types.h
+++ b/include/ucode/types.h
@@ -112,7 +112,7 @@ typedef struct uc_weakref {
 
 typedef struct uc_function {
 	uc_weakref_t progref;
-	bool arrow, vararg, strict;
+	bool arrow, vararg, strict, module;
 	size_t nargs;
 	size_t nupvals;
 	size_t srcidx;