adding include tracking callback

SQFvm · Sep 27, 2023 · ea17a9f · ea17a9f
1 parent 4eebf1d
commit ea17a9f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/parser/preprocessor/default.cpp b/src/parser/preprocessor/default.cpp
@@ -942,6 +942,9 @@ std::string sqf::parser::preprocessor::impl_default::instance::parse_ppinstructi
             output.append(" \"");
             output.append(file_context.path.physical);
             output.append("\"\n");
+            if (m_file_included_callback) {
+                m_file_included_callback(otherfinfo, file_context);
+            }
             return output;
         }
         catch (const std::runtime_error &ex) {
@@ -1477,6 +1480,7 @@ std::optional<std::string> sqf::parser::preprocessor::impl_default::preprocess(
     fileinfo.content = view;
     instance i(this, get_logger(), m_macros);
     i.m_macro_resolved_callback = m_macro_resolved_callback;
+    i.m_file_included_callback = m_file_included_callback;
     auto res = i.parse_file(runtime, fileinfo);
     if (out_included) {
         for (const auto &entry: i.m_visited) {

diff --git a/src/parser/preprocessor/default.h b/src/parser/preprocessor/default.h
@@ -38,6 +38,7 @@ namespace sqf::parser::preprocessor {
                 &m,
                 const std::unordered_map<std::string, std::string> &param_map
         )> m_macro_resolved_callback;
+        std::function<void(context & included_fileinfo, context & source_fileinfo)> m_file_included_callback;
         struct condition_scope {
             bool allow_write{};
             ::sqf::runtime::diagnostics::diag_info info_if;
@@ -62,6 +63,7 @@ namespace sqf::parser::preprocessor {
                     &m,
                     const std::unordered_map<std::string, std::string> &param_map
             )> m_macro_resolved_callback;
+            std::function<void(context & included_fileinfo, context & source_fileinfo)> m_file_included_callback;
 
             instance(
                     impl_default *owner,
@@ -177,6 +179,10 @@ namespace sqf::parser::preprocessor {
             assert(m_pragmas.find(std::string(name.begin(), name.end())) != m_pragmas.end());
         };
 
+        void file_included(std::function<void(context & included_fileinfo, context & source_fileinfo)> callback) {
+            m_file_included_callback = std::move(callback);
+        }
+
         void macro_resolved(std::function<void(
                 macro_resolved_data orig_start,
                 macro_resolved_data orig_end,