Попробуйте это:
Свяжите свойство IsFocused
текстового поля с IsSelected
свойством treeviewitems
<TextBox IsFocused = {Binding Path=IsSelected, RelativeSource={RelativeSource AncestorType={x:Type TreeViewItem}}, Mode="OneWayToSource"} />
Иначе Добавьте триггер к текстовому полю, чтобы всякий раз, когда для свойства IsFocused
было установлено значение true, для соответствующего свойства IsSelected
соответствующего элемента дерева также устанавливалось значение true.