Fix lagdisplay

lagdisplay/src/com/dmdirc/addons/lagdisplay/ServerInfoDialog.java

@@ -26,6 +26,7 @@ import com.dmdirc.ServerState;
 import com.dmdirc.addons.ui_swing.MainFrame;
 import com.dmdirc.addons.ui_swing.components.statusbar.StatusbarPanel;
 import com.dmdirc.addons.ui_swing.components.statusbar.StatusbarPopupWindow;
+import com.dmdirc.addons.ui_swing.interfaces.ActiveFrameManager;
 import com.dmdirc.interfaces.Connection;
 import com.dmdirc.interfaces.ConnectionManager;
 
@@ -45,10 +46,10 @@ public class ServerInfoDialog extends StatusbarPopupWindow {
     private static final long serialVersionUID = 3;
     /** The lag display manager. */
     protected final LagDisplayManager manager;
-    /** Swing main frame. */
-    private final MainFrame mainFrame;
     /** Server manager to retrieve servers from. */
     private final ConnectionManager connectionManager;
+    /** The active window manager. */
+    private final ActiveFrameManager activeFrameManager;
 
     /**
      * Creates a new ServerInfoDialog.
@@ -62,12 +63,13 @@ public class ServerInfoDialog extends StatusbarPopupWindow {
             final LagDisplayManager manager,
             final StatusbarPanel<JLabel> parent,
             final MainFrame mainFrame,
+            final ActiveFrameManager activeFrameManager,
             final ConnectionManager connectionManager) {
         super(parent, mainFrame);
 
         this.manager = manager;
-        this.mainFrame = mainFrame;
         this.connectionManager = connectionManager;
+        this.activeFrameManager = activeFrameManager;
     }
 
     @Override
@@ -78,7 +80,7 @@ public class ServerInfoDialog extends StatusbarPopupWindow {
             panel.add(new JLabel("No open servers."));
         } else {
             if (manager.shouldShowGraph()) {
-                panel.add(new PingHistoryPanel(manager, mainFrame), "span, grow, wrap");
+                panel.add(new PingHistoryPanel(manager, activeFrameManager), "span, grow, wrap");
                 panel.add(new JSeparator(), "span, grow, wrap");
             }
 

lagdisplay/src/com/dmdirc/addons/lagdisplay/ServerInfoDialogFactory.java

@@ -24,6 +24,7 @@ package com.dmdirc.addons.lagdisplay;
 
 import com.dmdirc.addons.ui_swing.MainFrame;
 import com.dmdirc.addons.ui_swing.components.statusbar.StatusbarPanel;
+import com.dmdirc.addons.ui_swing.interfaces.ActiveFrameManager;
 import com.dmdirc.interfaces.ConnectionManager;
 
 import javax.inject.Inject;
@@ -39,18 +40,20 @@ public class ServerInfoDialogFactory {
     private final LagDisplayManager manager;
     private final MainFrame mainFrame;
     private final ConnectionManager connectionManager;
+    private final ActiveFrameManager activeFrameManager;
 
     @Inject
     public ServerInfoDialogFactory(final LagDisplayManager manager, final MainFrame mainFrame,
-            final ConnectionManager connectionManager) {
+            final ActiveFrameManager activeFrameManager, final ConnectionManager connectionManager) {
         this.manager = manager;
         this.mainFrame = mainFrame;
         this.connectionManager = connectionManager;
+        this.activeFrameManager = activeFrameManager;
     }
 
     public ServerInfoDialog getServerInfoDialog(
             final StatusbarPanel<JLabel> parent) {
-        return new ServerInfoDialog(manager, parent, mainFrame, connectionManager);
+        return new ServerInfoDialog(manager, parent, mainFrame, activeFrameManager, connectionManager);
     }
 
 }