Insert branches into DataTree at index

Hi,

Given an input tree, i want to add items at a location which match with a key. (See 1st image below)

I made the first step : to find which branches which match with the key (See 2nd image below)

But the second step is difficult to me : build the output tree with the correct paths...

Thanks for reading !

JL