component-store | manage react dumb component 's state , then you can use | State Container library

I needed a dynamically loading nodes in the angular material tree, so I created my own GeoDataSource which internally uses @ngrx/component-store to handle all the interaction between data source, TreeControl and the tree itself.

Now, the problem is the tree doesn't keep its expansion state because the abstract class BaseTreeControl uses SelectionModel class and this in turn uses Set to keep track of items that are selected. So if node is expanded it just does this: