feat: add class member getter、setter

Author: engsr6982

generator/binder/WrapperClassGenerator.py

@@ -62,6 +62,11 @@ def jinja2_def_remove_tempate(data: str):
     return re.sub(r"<.*?>", "", data)
 
 
+def jinja2_def_is_empty(data):
+    # str、List、Dict
+    return len(data) == 0
+
+
 logger = create_console_logger(__name__)
 
 
@@ -95,6 +100,7 @@ def __init__(self, ast_dir: str, out_dir: str):
         self.env.globals["get_wrapper_class_name"] = jinja2_def_get_wrapper_class_name
         self.env.globals["can_render"] = jinja2_def_can_render
         self.env.globals["remove_template"] = jinja2_def_remove_tempate
+        self.env.globals["is_empty"] = jinja2_def_is_empty
         pass
 
     def load_ast(self, ast_file_path: str):
@@ -125,11 +131,12 @@ def process_class_method(
         while stack:
             cur_class_info = stack.pop()
 
-            rendered_scriptx_method.append(
-                class_method_template.render(
-                    class_info=cur_class_info, gened_method=gened_method
+            if not jinja2_def_is_empty(cur_class_info.methods):
+                rendered_scriptx_method.append(
+                    class_method_template.render(
+                        class_info=cur_class_info, gened_method=gened_method
+                    )
                 )
-            )
 
             for inheritances in cur_class_info.inheritances:
                 if inheritances.definition == "":
@@ -161,7 +168,7 @@ def process_class_method(
                         if class_info.name == jinja2_def_remove_tempate(
                             inheritances.name
                         ):
-                            if not class_info.methods:
+                            if class_info.methods == []:
                                 logger.warning(
                                     f"skip class: {class_info.name}, no methods"
                                 )
@@ -174,7 +181,58 @@ def process_class_member(self, class_info: ClassInfo):
         # TODO: 处理类 member
         # 静态成员注册 .prototype(Getter)
         # 实例成员注册 .instancePrototype(Getter、Setter)
-        pass
+
+        class_member_template = self.env.get_template("class_member.h.j2")
+        rendered_getter_setter: List[str] = []
+
+        # 递归处理继承类
+        stack: List[ClassInfo] = [class_info]
+        while stack:
+            cur_class_info = stack.pop()
+
+            if not jinja2_def_is_empty(cur_class_info.members):
+                rendered_getter_setter.append(
+                    class_member_template.render(class_info=cur_class_info)
+                )
+
+            for inheritances in cur_class_info.inheritances:
+                if inheritances.definition == "":
+                    logger.warning(f"skip empty definition class: {inheritances.name}")
+                    continue
+                if inheritances.name.startswith("std::"):
+                    logger.warning(f"skip std class: {inheritances.name}")
+                    continue
+
+                # 获取继承类所在文件
+                fixed_file = jinja2_def_replace_str(
+                    jinja2_def_remove_line_and_column(inheritances.definition),
+                    r".+endstone",
+                    self.ast_dir,  # 替换基路径
+                )
+                fixed_file = FileUtils.replace_file_extension(fixed_file, ".json")
+
+                # 加载继承类所在的ast文件
+                inherit_class_info = self.load_ast(fixed_file)
+                if inherit_class_info is None:
+                    logger.warning(f"skip not found class: {inheritances.name}")
+                    continue
+
+                # 查找继承类
+                for namespace in inherit_class_info.namespaces:
+                    if namespace.name != "endstone":
+                        continue
+                    for class_info in namespace.classes:
+                        if class_info.name == jinja2_def_remove_tempate(
+                            inheritances.name
+                        ):
+                            if class_info.members == []:
+                                logger.warning(
+                                    f"skip class: {class_info.name}, no methods"
+                                )
+                                continue
+                            stack.append(class_info)  # 加入继承类
+                            break
+        return rendered_getter_setter
 
     def process_class(self, class_info: ClassInfo, base_namespace: str):
         class_wrapper_template = self.env.get_template("class_wrapper.h.j2")
@@ -194,13 +252,15 @@ def process_class(self, class_info: ClassInfo, base_namespace: str):
 
             # 渲染方法
             rendered_class_method = self.process_class_method(cur_class_info)
+            rendered_scriptx_member = self.process_class_member(cur_class_info)
 
             # 生成类包装
             rendered_class_wrapper.append(
                 class_wrapper_template.render(
                     class_info=cur_class_info,
                     class_full_name=full_class_name,
                     rendered_scriptx_method=rendered_class_method,
+                    rendered_scriptx_member=rendered_scriptx_member,
                 )
             )
 

generator/binder/templates/class_member.h.j2

@@ -0,0 +1,19 @@
+// @{{class_info.isStruct and "struct" or "class"}}: {{class_info.name}}
+{% for member_info in class_info.members | only_public %}
+{% if not member_info.isStatic %}
+    /**
+     * @getter {{member_info.name}}
+     * @definition {{ replace_str(member_info.definition, ".+endstone\/", "") }}
+     */
+    Local<Value> Getter_for_{{member_info.name}}();
+
+{% if not member_info.isConst %}
+    /**
+     * @setter {{member_info.name}}
+     * @definition {{ replace_str(member_info.definition, ".+endstone\/", "") }}
+     */
+    void Setter_for_{{member_info.name}}(Local<Value> value);
+{% endif %}
+
+{% endif %}
+{% endfor %}

generator/binder/templates/class_method.h.j2

@@ -1,7 +1,7 @@
 // @{{class_info.isStruct and "struct" or "class"}}: {{class_info.name}}
 {% for method_info in class_info.methods | only_public | remove_operator | unique_by_field("name") %}
 {% if not is_constructor(class_info, method_info) and not is_destructor(class_info, method_info)%}
-{% if can_render(gened_method, method_info.name) %}
+{% if can_render(gened_method, method_info.name) and not method_info.isStatic %}
     /**
      * @definition {{ replace_str(method_info.definition, ".+endstone\/", "") }}
      * @parameters num: {{method_info.parameters | length}}

generator/binder/templates/class_wrapper.h.j2

@@ -35,10 +35,20 @@ public:
 
     [[nodiscard]] SafeResourceHolder<{{class_full_name}}>& getResourceHolder() { return mResourceHolder; }
 
-public:
-    {% for rendered_template in rendered_scriptx_method %}
-    {{ rendered_template }}
+{% if not is_empty(rendered_scriptx_member) %}
+public: /* auto generated member getter、setter */
+    {% for rendered_member in rendered_scriptx_member %}
+    {{ rendered_member }}
+    {% endfor %}
+{% endif %}
+
+{% if not is_empty(rendered_scriptx_method) %}
+public: /* auto generated methods */
+    {% for rendered_method in rendered_scriptx_method %}
+    {{ rendered_method }}
     {% endfor %}
+{% endif %}
+
 public:
     static ClassDefine<{{get_wrapper_class_name(class_info)}}> builder;
 };