Add USER_INCLUDES region for custom header includes in generated code

Author: StackAndPointer

core/code_generator.py

@@ -77,6 +77,10 @@ def generate_header_for_interface(self, iface: Interface, project: Project) -> s
 #include <string>
 #include <vector>
 
+// [[[USER_INCLUDES]]]
+// Add custom header includes here
+{iface.user_code.header_includes}// [[[END_USER_INCLUDES]]]
+
 class LawnApp;
 
 namespace Sexy
@@ -225,6 +229,10 @@ def generate_cpp_for_interface(self, iface: Interface, project: Project) -> str:
 {interface_includes}
 {action_includes}
 
+// [[[USER_INCLUDES]]]
+// Add custom header includes here
+{iface.user_code.cpp_includes}// [[[END_USER_INCLUDES]]]
+
 using namespace Sexy;
 
 """

core/code_parser.py

@@ -65,13 +65,15 @@ def extract_from_files(cls, header_path: str, cpp_path: str) -> ProjectCode:
             with open(header_path, 'r', encoding='utf-8') as f:
                 header = f.read()
             blocks = cls.extract_blocks(header)
+            code.header_includes = blocks.get('USER_INCLUDES', '')
             code.declarations = blocks.get('USER_DECLARATIONS', '')
 
         if cpp_path and os.path.exists(cpp_path):
             with open(cpp_path, 'r', encoding='utf-8') as f:
                 cpp = f.read()
             blocks = cls.extract_blocks(cpp)
 
+            code.cpp_includes = blocks.get('USER_INCLUDES', '')
             code.init_code = blocks.get('USER_INIT', '')
             code.destroy_code = blocks.get('USER_DESTROY', '')
             code.draw_code = blocks.get('USER_DRAW', '')
@@ -98,6 +100,7 @@ def extract_from_cs_file(cls, cs_path: str) -> ProjectCode:
                 cs_content = f.read()
             blocks = cls.extract_blocks(cs_content)
 
+            code.cpp_includes = blocks.get('USER_INCLUDES', '')
             code.declarations = blocks.get('USER_DECLARATIONS', '')
             code.init_code = blocks.get('USER_INIT', '')
             code.destroy_code = blocks.get('USER_DESTROY', '')

core/project.py

@@ -30,6 +30,8 @@ def __post_init__(self):
 
 @dataclass
 class ProjectCode:
+    header_includes: str = ""
+    cpp_includes: str = ""
     declarations: str = ""
     init_code: str = ""
     destroy_code: str = ""
@@ -288,6 +290,8 @@ def to_dict(self) -> dict:
                     },
                     "root_widget_ids": iface.root_widget_ids,
                     "user_code": {
+                        "header_includes": iface.user_code.header_includes,
+                        "cpp_includes": iface.user_code.cpp_includes,
                         "declarations": iface.user_code.declarations,
                         "init_code": iface.user_code.init_code,
                         "destroy_code": iface.user_code.destroy_code,
@@ -365,6 +369,8 @@ def from_dict(self, data: dict):
                 iface.root_widget_ids = idata.get("root_widget_ids", [])
                 uc = idata.get("user_code", {})
                 iface.user_code = ProjectCode(
+                    header_includes=uc.get("header_includes", ""),
+                    cpp_includes=uc.get("cpp_includes", ""),
                     declarations=uc.get("declarations", ""),
                     init_code=uc.get("init_code", ""),
                     destroy_code=uc.get("destroy_code", ""),
@@ -409,6 +415,8 @@ def from_dict(self, data: dict):
             iface.root_widget_ids = data.get("root_widget_ids", [])
             uc = data.get("user_code", {})
             iface.user_code = ProjectCode(
+                header_includes=uc.get("header_includes", ""),
+                cpp_includes=uc.get("cpp_includes", ""),
                 declarations=uc.get("declarations", ""),
                 init_code=uc.get("init_code", ""),
                 destroy_code=uc.get("destroy_code", ""),