ServerTunnelRootNode.java :  » SSH » ssh-tunneler » SSHTunneler » server_tunnel » tree » Java Open Source

/*
 *    This file is part of SSHTunneler.
 *
 *    Copyright 2009 Jean-Sebastien Gelinas <calestar@gmail.com>
 *
 *    SSHTunneler is free software: you can redistribute it and/or modify
 *    it under the terms of the GNU General Public License as published by
 *    the Free Software Foundation, either version 3 of the License, or
 *    (at your option) any later version.
 *
 *    SSHTunneler is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU General Public License for more details.
 *
 *    You should have received a copy of the GNU General Public License
 *    along with SSHTunneler.  If not, see <http://www.gnu.org/licenses/>.
 *
 */
package sshtunneler.server_tunnel.tree;

import javax.swing.JPopupMenu;

import sshtunneler.library.Server;
import sshtunneler.library.ServerContainer;

public class ServerTunnelRootNode extends MyServerTunnelTreeNode {
  private static final long serialVersionUID = -9175805529722030731L;
  private GenericServerTunnelTree tree;
  private ServerContainer<? extends Server> container;

  public ServerTunnelRootNode(GenericServerTunnelTree tree,
      ServerContainer<? extends Server> container) {
    this.container = container;
    this.tree = tree;
    this.setAllowsChildren(true);
    this.setUserObject("Servers");
  }

  public ServerNode addServerNode(Server server) {
    ServerNode newNode = new ServerNode(server, this.tree);

    this.add(newNode);

    return newNode;
  }

  @Override
  public void refresh() {
    this.removeAllChildren();
    for (Server server : this.container.getServers()) {
      this.addServerNode(server).refresh();
    }
  }

  @Override
  public JPopupMenu getPopup() {
    return this.tree.get_root_menu(this);
  }
}