From d478d0597a7f9e1ea3ff76f0f9536cff2f756d3e Mon Sep 17 00:00:00 2001
From: Campbell Barton <campbell@blender.org>
Date: Thu, 31 Aug 2023 17:44:32 +1000
Subject: [PATCH] Fix bone collection selecting not flushing to the tip/root

Also correct the notifier in edit-mode.
---
 .../blender/editors/armature/bone_collections.cc  | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/source/blender/editors/armature/bone_collections.cc b/source/blender/editors/armature/bone_collections.cc
index 8a4e4444572..33406df8def 100644
--- a/source/blender/editors/armature/bone_collections.cc
+++ b/source/blender/editors/armature/bone_collections.cc
@@ -465,13 +465,7 @@ static void bone_collection_select(bContext *C,
       if (!editbone_is_member(ebone, bcoll)) {
         continue;
       }
-
-      if (select) {
-        ebone->flag |= BONE_SELECTED;
-      }
-      else {
-        ebone->flag &= ~BONE_SELECTED;
-      }
+      ED_armature_ebone_select_set(ebone, select);
     }
   }
   else {
@@ -494,7 +488,12 @@ static void bone_collection_select(bContext *C,
   }
 
   DEG_id_tag_update(&armature->id, ID_RECALC_SELECT);
-  WM_event_add_notifier(C, NC_OBJECT | ND_POSE, ob);
+  if (is_editmode) {
+    WM_event_add_notifier(C, NC_OBJECT | ND_BONE_SELECT, ob);
+  }
+  else {
+    WM_event_add_notifier(C, NC_OBJECT | ND_POSE, ob);
+  }
 
   if (is_editmode) {
     ED_outliner_select_sync_from_edit_bone_tag(C);