jmetermenubar.java

测试工具

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * 
 */

package org.apache.jmeter.gui.util;

import java.awt.Component;
import java.util.Collection;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.Locale;

import javax.swing.JComponent;
import javax.swing.JMenu;
import javax.swing.JMenuBar;
import javax.swing.JMenuItem;
import javax.swing.JPopupMenu;
import javax.swing.MenuElement;
import javax.swing.UIManager;

import org.apache.jmeter.gui.action.ActionNames;
import org.apache.jmeter.gui.action.ActionRouter;
import org.apache.jmeter.gui.action.KeyStrokes;
import org.apache.jmeter.gui.action.LoadRecentProject;
import org.apache.jmeter.util.JMeterUtils;
import org.apache.jmeter.util.LocaleChangeEvent;
import org.apache.jmeter.util.LocaleChangeListener;
import org.apache.jmeter.util.SSLManager;
import org.apache.jorphan.util.JOrphanUtils;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.log.Logger;

public class JMeterMenuBar extends JMenuBar implements LocaleChangeListener {
	private static final Logger log = LoggingManager.getLoggerForClass();

    private JMenu fileMenu;

    private JMenuItem file_save_as;

    private JMenuItem file_selection_as;

    private JMenuItem file_revert;

    private JMenuItem file_load;

    private List file_load_recent_files;

    private JMenuItem file_merge;

    private JMenuItem file_exit;

    private JMenuItem file_close;

    private JMenu editMenu;

    private JMenu edit_add;

	// JMenu edit_add_submenu;
    private JMenuItem edit_remove; // TODO - should this be created?

    private JMenu runMenu;

    private JMenuItem run_start;

    private JMenu remote_start;

    private JMenuItem remote_start_all;

    private Collection remote_engine_start;

    private JMenuItem run_stop;

	private JMenuItem run_shut;
    
    private JMenu remote_stop;

    private JMenuItem remote_stop_all;

    private Collection remote_engine_stop;

    private JMenuItem run_clear;

    private JMenuItem run_clearAll;

	// JMenu reportMenu;
	// JMenuItem analyze;
    private JMenu optionsMenu;

    private JMenu lafMenu;

    private JMenuItem sslManager;

    private JMenu helpMenu;

    private JMenuItem help_about;

    private String[] remoteHosts;

	private JMenu remote_exit;

	private JMenuItem remote_exit_all;

	private Collection remote_engine_exit;

	public JMeterMenuBar() {
		// List for recent files menu items
        file_load_recent_files = new LinkedList();
		// Lists for remote engines menu items
		remote_engine_start = new LinkedList();
		remote_engine_stop = new LinkedList();
		remote_engine_exit = new LinkedList();
		remoteHosts = JOrphanUtils.split(JMeterUtils.getPropDefault("remote_hosts", ""), ","); //$NON-NLS-1$
		if (remoteHosts.length == 1 && remoteHosts[0].equals("")) {
			remoteHosts = new String[0];
		}
		this.getRemoteItems();
		createMenuBar();
	}

	public void setFileSaveEnabled(boolean enabled) {
        if(file_save_as != null) {
            file_save_as.setEnabled(enabled);
        }
	}

	public void setFileLoadEnabled(boolean enabled) {
		if (file_load != null) {
			file_load.setEnabled(enabled);
		}
		if (file_merge != null) {
			file_merge.setEnabled(enabled);
		}
	}

    public void setFileRevertEnabled(boolean enabled) {
        if(file_revert != null) {
            file_revert.setEnabled(enabled);
        }
    }
    
    public void setProjectFileLoaded(String file) {
        if(file_load_recent_files != null && file != null) {
            LoadRecentProject.updateRecentFileMenuItems(file_load_recent_files, file);
        }
    }

	public void setEditEnabled(boolean enabled) {
		if (editMenu != null) {
			editMenu.setEnabled(enabled);
		}
	}

	public void setEditAddMenu(JMenu menu) {
		// If the Add menu already exists, remove it.
		if (edit_add != null) {
			editMenu.remove(edit_add);
		}
		// Insert the Add menu as the first menu item in the Edit menu.
		edit_add = menu;
		editMenu.insert(edit_add, 0);
	}

	public void setEditMenu(JPopupMenu menu) {
		if (menu != null) {
			editMenu.removeAll();
			Component[] comps = menu.getComponents();
			for (int i = 0; i < comps.length; i++) {
				editMenu.add(comps[i]);
			}
			editMenu.setEnabled(true);
		} else {
			editMenu.setEnabled(false);
		}
	}

	public void setEditAddEnabled(boolean enabled) {
		// There was a NPE being thrown without the null check here.. JKB
		if (edit_add != null) {
			edit_add.setEnabled(enabled);
		}
		// If we are enabling the Edit-->Add menu item, then we also need to
		// enable the Edit menu. The Edit menu may already be enabled, but
		// there's no harm it trying to enable it again.
		if (enabled) {
			setEditEnabled(true);
		} else {
			// If we are disabling the Edit-->Add menu item and the
			// Edit-->Remove menu item is disabled, then we also need to
			// disable the Edit menu.
			// The Java Look and Feel Guidelines say to disable a menu if all
			// menu items are disabled.
			if (!edit_remove.isEnabled()) {
				editMenu.setEnabled(false);
			}
		}
	}

	public void setEditRemoveEnabled(boolean enabled) {
		edit_remove.setEnabled(enabled);
		// If we are enabling the Edit-->Remove menu item, then we also need to
		// enable the Edit menu. The Edit menu may already be enabled, but
		// there's no harm it trying to enable it again.
		if (enabled) {
			setEditEnabled(true);
		} else {
			// If we are disabling the Edit-->Remove menu item and the
			// Edit-->Add menu item is disabled, then we also need to disable
			// the Edit menu.
			// The Java Look and Feel Guidelines say to disable a menu if all
			// menu items are disabled.
			if (!edit_add.isEnabled()) {
				editMenu.setEnabled(false);
			}
		}
	}

	/**
	 * Creates the MenuBar for this application. I believe in my heart that this
	 * should be defined in a file somewhere, but that is for later.
	 */
	public void createMenuBar() {
		makeFileMenu();
		makeEditMenu();
		makeRunMenu();
		makeOptionsMenu();
		makeHelpMenu();
		this.add(fileMenu);
		this.add(editMenu);
		this.add(runMenu);
		this.add(optionsMenu);
		this.add(helpMenu);
	}

	private void makeHelpMenu() {
		// HELP MENU
		helpMenu = new JMenu(JMeterUtils.getResString("help")); //$NON-NLS-1$
		helpMenu.setMnemonic('H');
		JMenuItem contextHelp = new JMenuItem(JMeterUtils.getResString("help"), 'H'); //$NON-NLS-1$
		contextHelp.setActionCommand(ActionNames.HELP);
		contextHelp.setAccelerator(KeyStrokes.HELP);
		contextHelp.addActionListener(ActionRouter.getInstance());

        JMenuItem whatClass = new JMenuItem(JMeterUtils.getResString("help_node"), 'W');//$NON-NLS-1$
        whatClass.setActionCommand(ActionNames.WHAT_CLASS);
        whatClass.setAccelerator(KeyStrokes.WHAT_CLASS);
        whatClass.addActionListener(ActionRouter.getInstance());

        JMenuItem setDebug = new JMenuItem(JMeterUtils.getResString("debug_on"));//$NON-NLS-1$
        setDebug.setActionCommand(ActionNames.DEBUG_ON);
        setDebug.setAccelerator(KeyStrokes.DEBUG_ON);
        setDebug.addActionListener(ActionRouter.getInstance());

        JMenuItem resetDebug = new JMenuItem(JMeterUtils.getResString("debug_off"));//$NON-NLS-1$
        resetDebug.setActionCommand(ActionNames.DEBUG_OFF);
        resetDebug.setAccelerator(KeyStrokes.DEBUG_OFF);
        resetDebug.addActionListener(ActionRouter.getInstance());

        help_about = new JMenuItem(JMeterUtils.getResString("about"), 'A'); //$NON-NLS-1$
		help_about.setActionCommand(ActionNames.ABOUT);
		help_about.addActionListener(ActionRouter.getInstance());
		helpMenu.add(contextHelp);
        helpMenu.addSeparator();
        helpMenu.add(whatClass);
        helpMenu.add(setDebug);
        helpMenu.add(resetDebug);
        helpMenu.addSeparator();
		helpMenu.add(help_about);
	}

	private void makeOptionsMenu() {
		// OPTIONS MENU
		optionsMenu = new JMenu(JMeterUtils.getResString("option")); //$NON-NLS-1$
		JMenuItem functionHelper = new JMenuItem(JMeterUtils.getResString("function_dialog_menu_item"), 'F'); //$NON-NLS-1$
		functionHelper.addActionListener(ActionRouter.getInstance());
		functionHelper.setActionCommand(ActionNames.FUNCTIONS);
		functionHelper.setAccelerator(KeyStrokes.FUNCTIONS);
		lafMenu = new JMenu(JMeterUtils.getResString("appearance")); //$NON-NLS-1$
		UIManager.LookAndFeelInfo lafs[] = UIManager.getInstalledLookAndFeels();
		for (int i = 0; i < lafs.length; ++i) {
			JMenuItem laf = new JMenuItem(lafs[i].getName());
			laf.addActionListener(ActionRouter.getInstance());
			laf.setActionCommand(ActionNames.LAF_PREFIX + lafs[i].getClassName());
			lafMenu.setMnemonic('L');
			lafMenu.add(laf);
		}
		optionsMenu.setMnemonic('O');
		optionsMenu.add(functionHelper);
		optionsMenu.add(lafMenu);
		if (SSLManager.isSSLSupported()) {
			sslManager = new JMenuItem(JMeterUtils.getResString("sslmanager")); //$NON-NLS-1$
			sslManager.addActionListener(ActionRouter.getInstance());
			sslManager.setActionCommand(ActionNames.SSL_MANAGER);
			sslManager.setMnemonic('S');
			sslManager.setAccelerator(KeyStrokes.SSL_MANAGER);
			optionsMenu.add(sslManager);
		}
		optionsMenu.add(makeLanguageMenu());
		
		JMenuItem collapse = new JMenuItem(JMeterUtils.getResString("menu_collapse_all")); //$NON-NLS-1$
		collapse.addActionListener(ActionRouter.getInstance());
		collapse.setActionCommand(ActionNames.COLLAPSE_ALL);