public IEnumerable <IMiniButton> CreateForDiagramShape(IDiagramShapeUi diagramShapeUi)
{
if (!(diagramShapeUi is IDiagramNodeUi diagramNodeUi))
{
yield break;
}
yield return(new CloseMiniButtonViewModel(_modelEventSource, _diagramEventSource));
foreach (var entityRelationType in _relatedNodeTypeProvider.GetRelatedNodeTypes(diagramNodeUi.DiagramNode.ModelNode.Stereotype))
{
yield return(new RelatedNodeMiniButtonViewModel(_modelEventSource, _diagramEventSource, entityRelationType));
}
}
private IEnumerable <RelatedNodeType> GetRelatedNodeTypes() => _relatedNodeTypeProvider.GetRelatedNodeTypes(ModelNode.Stereotype);
