Tree context menu

I'm building an IDEA plugin that will create a tree based on a model. Is there some API to have context menus attached to elements of the tree with filtering based on the node nature ?