From 9361423335c20c41e37f2a028e883434e041aaea Mon Sep 17 00:00:00 2001
From: Yusarina <hello@yusarina.xyz>
Date: Fri, 20 Sep 2024 12:18:08 +0100
Subject: [PATCH] Remove Bone Constraints

- Adds Remove Bone Constraints, this uses pose mode instead of edit mode as this is supposedly a safer method.
---
 functions/additional_tools.py     | 30 ++++++++++++++++++++++++++++++
 resources/translations/en_US.json |  4 ++++
 ui/tools.py                       |  4 +++-
 3 files changed, 37 insertions(+), 1 deletion(-)

diff --git a/functions/additional_tools.py b/functions/additional_tools.py
index 9745945..241cac1 100644
--- a/functions/additional_tools.py
+++ b/functions/additional_tools.py
@@ -89,3 +89,33 @@ class AvatarToolKit_OT_ConnectBones(Operator):
     def draw(self, context):
         layout = self.layout
         layout.prop(self, "min_distance")
+
+@register_wrap
+class AvatarToolKit_OT_DeleteBoneConstraints(Operator):
+    bl_idname = "avatar_toolkit.delete_bone_constraints"
+    bl_label = t("Tools.delete_bone_constraints.label")
+    bl_description = t("Tools.delete_bone_constraints.desc")
+    bl_options = {'REGISTER', 'UNDO'}
+
+    @classmethod
+    def poll(cls, context: Context) -> bool:
+        return get_selected_armature(context) is not None
+
+    def execute(self, context: Context) -> set[str]:
+        armature = get_selected_armature(context)
+        if not is_valid_armature(armature):
+            self.report({'ERROR'}, t("Tools.delete_bone_constraints.invalid_armature"))
+            return {'CANCELLED'}
+
+        bpy.ops.object.mode_set(mode='POSE')
+        
+        constraints_removed = 0
+        for bone in armature.pose.bones:
+            while bone.constraints:
+                bone.constraints.remove(bone.constraints[0])
+                constraints_removed += 1
+
+        bpy.ops.object.mode_set(mode='OBJECT')
+
+        self.report({'INFO'}, t("Tools.delete_bone_constraints.success").format(constraints_removed=constraints_removed))
+        return {'FINISHED'}
diff --git a/resources/translations/en_US.json b/resources/translations/en_US.json
index e4c242e..e5e5897 100644
--- a/resources/translations/en_US.json
+++ b/resources/translations/en_US.json
@@ -171,6 +171,10 @@
     "Tools.connect_bones.min_distance.label": "Minimum Distance",
     "Tools.connect_bones.min_distance.desc": "Minimum distance between bones to connect them",
     "Tools.connect_bones.success": "Connected {bones_connected} bones successfully",
+    "Tools.delete_bone_constraints.label": "Delete Bone Constraints",
+    "Tools.delete_bone_constraints.desc": "Remove all constraints from bones in the armature",
+    "Tools.delete_bone_constraints.invalid_armature": "Invalid armature selected",
+    "Tools.delete_bone_constraints.success": "Removed {constraints_removed} constraints from bones",
     "MergeArmatures.select_armature": "Please select an armature",
     "MergeArmatures.title.label": "Merge Armatures:",
     "MergeArmatures.label": "Merge Armatures",
diff --git a/ui/tools.py b/ui/tools.py
index 21cd476..9c958d0 100644
--- a/ui/tools.py
+++ b/ui/tools.py
@@ -8,7 +8,7 @@ from ..functions.translations import t
 from ..core.common import get_selected_armature
 from ..functions.mesh_tools import AvatarToolkit_OT_RemoveUnusedShapekeys
 from ..functions.seperate_by import AvatarToolKit_OT_SeparateByMaterials, AvatarToolKit_OT_SeparateByLooseParts
-from ..functions.additional_tools import AvatarToolKit_OT_ApplyTransforms, AvatarToolKit_OT_ConnectBones
+from ..functions.additional_tools import AvatarToolKit_OT_ApplyTransforms, AvatarToolKit_OT_ConnectBones, AvatarToolKit_OT_DeleteBoneConstraints
 from ..functions.armature_modifying import AvatarToolkit_OT_RemoveZeroWeightBones, AvatarToolkit_OT_MergeBonesToActive, AvatarToolkit_OT_MergeBonesToParents
 
 @register_wrap
@@ -49,5 +49,7 @@ class AvatarToolkit_PT_ToolsPanel(bpy.types.Panel):
             row.operator(AvatarToolkit_OT_MergeBonesToParents.bl_idname, text=t("Tools.merge_bones_to_parents.label"), icon='BONE_DATA')
             row = layout.row(align=True)
             row.operator(AvatarToolKit_OT_ConnectBones.bl_idname, text=t("Tools.connect_bones.label"), icon='BONE_DATA')
+            row = layout.row(align=True)
+            row.operator(AvatarToolKit_OT_DeleteBoneConstraints.bl_idname, text=t("Tools.delete_bone_constraints.label"), icon='CONSTRAINT_BONE')
         else:
             layout.label(text=t("Tools.select_armature"), icon='ERROR')