feat: add more format support

bioxelnodes/customnodes/menus.py

@@ -0,0 +1,192 @@
+import shutil
+import bpy
+from pathlib import Path
+from .nodes import AddCustomNode
+
+
+class SaveAllNodes(bpy.types.Operator):
+    bl_idname = "customnodes.save_all_nodes"
+    bl_label = "Save All Custom Nodes"
+    bl_description = "Save All Custom Nodes to Directory."
+    bl_options = {'UNDO'}
+
+    def execute(self, context):
+        files = []
+        for classname in dir(bpy.types):
+            if "CUSTOMNODES_MT_NODES_" in classname:
+                cls = getattr(bpy.types, classname)
+                files.append(cls.nodes_file)
+        files = list(set(files))
+
+        for file in files:
+            file_name = Path(file).name
+            # "//"
+            custom_nodes_dir = bpy.path.abspath(context.scene.custom_nodes_dir)
+
+            output_path: Path = Path(custom_nodes_dir, file_name).resolve()
+            source_path: Path = Path(file).resolve()
+
+            if output_path != source_path:
+                shutil.copy(source_path, output_path)
+
+            for lib in bpy.data.libraries:
+                lib_path = str(Path(bpy.path.abspath(lib.filepath)).resolve())
+                if lib_path == file:
+                    blend_path = Path(bpy.path.abspath("//")).resolve()
+                    lib.filepath = bpy.path.relpath(
+                        str(output_path), start=str(blend_path))
+
+            self.report({"INFO"}, f"Successfully saved to {output_path}")
+
+        return {'FINISHED'}
+
+
+class CUSTOMNODES_PT_CustomNodes(bpy.types.Panel):
+    bl_idname = "CUSTOMNODES_PT_CustomNodes"
+    bl_label = "Custom Nodes"
+    bl_space_type = "PROPERTIES"
+    bl_region_type = "WINDOW"
+    bl_context = "scene"
+
+    def draw(self, context):
+        layout = self.layout
+        scene = context.scene
+        layout.prop(scene, 'custom_nodes_dir')
+        layout.operator(SaveAllNodes.bl_idname)
+
+
+class CustomNodes():
+    def __init__(
+        self,
+        menu_items,
+        nodes_file,
+        root_label='CustomNodes',
+        root_icon='NONE'
+    ) -> None:
+        if not Path(nodes_file).is_file():
+            raise FileNotFoundError(str(Path(nodes_file).resolve()))
+
+        self.menu_items = menu_items
+        self.nodes_file = str(Path(nodes_file).resolve())
+
+        self.root_label = root_label
+        self.root_icon = root_icon
+
+        menu_classes = []
+        self._create_menu_class(
+            items=menu_items,
+            label=root_label,
+            menu_classes=menu_classes
+        )
+        self.menu_classes = menu_classes
+
+        idname = f"CUSTOMNODES_MT_NODES_{root_label.replace(' ', '').upper()}"
+
+        def add_node_menu(self, context):
+            if ('GeometryNodeTree' == bpy.context.area.spaces[0].tree_type):
+                layout = self.layout
+                layout.separator()
+                layout.menu(idname, icon=root_icon)
+
+        self.add_node_menu = add_node_menu
+
+    def _create_menu_class(self, menu_classes, items, label='CustomNodes', icon=0, idname_namespace=None):
+        nodes_file = self.nodes_file
+        idname_namespace = idname_namespace or "CUSTOMNODES_MT_NODES"
+        idname = f"{idname_namespace}_{label.replace(' ', '').upper()}"
+
+        # create submenu class if item is menu.
+        for item in items:
+            item_items = item.get('items') if item != 'separator' else None
+            if item_items:
+                menu_class = self._create_menu_class(
+                    menu_classes=menu_classes,
+                    items=item_items,
+                    label=item.get('label') or 'CustomNodes',
+                    icon=item.get('icon') or 0,
+                    idname_namespace=idname
+                )
+                item['menu_class'] = menu_class
+
+        # create menu class
+        class Menu(bpy.types.Menu):
+            bl_idname = idname
+            bl_label = label
+            nodes_file = self.nodes_file
+
+            def draw(self, context):
+                layout = self.layout
+
+                for item in items:
+                    # print(item)
+                    if item == "separator":
+                        layout.separator()
+                    elif item.get('menu_class'):
+                        layout.menu(
+                            item.get('menu_class').bl_idname,
+                            icon=item.get('icon') or 'NONE'
+                        )
+                    else:
+                        op = layout.operator(
+                            'customnodes.add_custom_node',
+                            text=item.get('label'),
+                            icon=item.get('icon') or 'NONE'
+                        )
+                        op.nodes_file = nodes_file
+                        op.node_type = item['node_type']
+                        op.node_label = item.get('label') or ""
+                        op.node_description = item.get(
+                            'node_description') or "Add Custom Node."
+                        op.node_link = item.get('link') or True
+                        op.node_callback = item.get('node_callback') or ""
+
+        menu_classes.append(Menu)
+        return Menu
+
+    def _find_item(self, found_items, menu_items, node_type: str):
+
+        for item in menu_items:
+            if item.get("node_type") == node_type:
+                found_items.append(item)
+
+            item_items = item.get('items') if item != 'separator' else None
+            if item_items:
+                self._find_item(found_items, item_items, node_type)
+
+    def find_item(self, node_type: str):
+        found_items = []
+        self._find_item(found_items, self.menu_items, node_type)
+        return found_items[0] if len(found_items) > 0 else None
+
+    def add_node(self, node_tree, node_type: str):
+        item = self.find_item(node_type)
+        if item:
+            op = AddCustomNode()
+            op.nodes_file = self.nodes_file
+            op.node_type = item['node_type']
+            op.node_label = item.get('label') or ""
+            op.node_link = item.get('link') or True
+            op.node_callback = item.get('node_callback') or ""
+            return op.add_node(node_tree)
+        else:
+            raise RuntimeError("No custom node type found.")
+
+    def register(self):
+        for cls in self.menu_classes:
+            bpy.utils.register_class(cls)
+
+        bpy.types.NODE_MT_add.append(self.add_node_menu)
+        bpy.types.Scene.custom_nodes_dir = bpy.props.StringProperty(
+            name="Nodes Directory",
+            subtype='DIR_PATH',
+            default="//"
+        )
+
+    def unregister(self):
+        bpy.types.NODE_MT_add.remove(self.add_node_menu)
+        try:
+            for cls in reversed(self.menu_classes):
+                bpy.utils.unregister_class(cls)
+
+        except RuntimeError:
+            pass