Allow the info panel's navButton to be removed on shutDown

This commit is contained in:
Max Weber
2018-02-20 17:03:10 -07:00
parent 397cb2f99f
commit 41ce72122a

View File

@@ -40,13 +40,15 @@ public class InfoPlugin extends Plugin
@Inject @Inject
private ClientUI ui; private ClientUI ui;
private NavigationButton navButton;
@Override @Override
protected void startUp() throws Exception protected void startUp() throws Exception
{ {
final InfoPanel panel = injector.getInstance(InfoPanel.class); final InfoPanel panel = injector.getInstance(InfoPanel.class);
panel.init(); panel.init();
final NavigationButton navButton = new NavigationButton( navButton = new NavigationButton(
"Info", "Info",
ImageIO.read(getClass().getResourceAsStream("info_icon.png")), ImageIO.read(getClass().getResourceAsStream("info_icon.png")),
() -> panel () -> panel
@@ -54,4 +56,10 @@ public class InfoPlugin extends Plugin
ui.getPluginToolbar().addNavigation(navButton); ui.getPluginToolbar().addNavigation(navButton);
} }
@Override
protected void shutDown()
{
ui.getPluginToolbar().removeNavigation(navButton);
}
} }