/*
* 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.functions.gui;
import java.awt.BorderLayout;
import java.awt.FlowLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.IOException;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import javax.swing.JButton;
import javax.swing.JDialog;
import javax.swing.JFrame;
import javax.swing.JPanel;
import javax.swing.event.ChangeEvent;
import javax.swing.event.ChangeListener;
import org.apache.jmeter.config.Argument;
import org.apache.jmeter.config.Arguments;
import org.apache.jmeter.config.gui.ArgumentsPanel;
import org.apache.jmeter.functions.Function;
import org.apache.jmeter.gui.action.ActionRouter;
import org.apache.jmeter.gui.action.Help;
import org.apache.jmeter.testelement.property.PropertyIterator;
import org.apache.jmeter.util.JMeterUtils;
import org.apache.jorphan.gui.ComponentUtil;
import org.apache.jorphan.gui.JLabeledChoice;
import org.apache.jorphan.gui.JLabeledTextField;
import org.apache.jorphan.reflect.ClassFinder;
public class FunctionHelper extends JDialog implements ActionListener, ChangeListener {
JLabeledChoice functionList;
ArgumentsPanel parameterPanel;
JLabeledTextField cutPasteFunction;
private Map functionMap = new HashMap();
JButton generateButton;
public FunctionHelper() {
super((JFrame) null, JMeterUtils.getResString("function_helper_title"), false); //$NON-NLS-1$
init();
}
private void init() {
parameterPanel = new ArgumentsPanel(JMeterUtils.getResString("function_params")); //$NON-NLS-1$
initializeFunctionList();
this.getContentPane().setLayout(new BorderLayout(10, 10));
JPanel comboPanel = new JPanel(new FlowLayout(FlowLayout.CENTER));
comboPanel.add(functionList);
JButton helpButton = new JButton(JMeterUtils.getResString("help")); //$NON-NLS-1$
helpButton.addActionListener(new HelpListener());
comboPanel.add(helpButton);
this.getContentPane().add(comboPanel, BorderLayout.NORTH);
this.getContentPane().add(parameterPanel, BorderLayout.CENTER);
JPanel resultsPanel = new JPanel(new FlowLayout(FlowLayout.CENTER));
cutPasteFunction = new JLabeledTextField(JMeterUtils.getResString("cut_paste_function"), 35); //$NON-NLS-1$
resultsPanel.add(cutPasteFunction);
generateButton = new JButton(JMeterUtils.getResString("generate")); //$NON-NLS-1$
generateButton.addActionListener(this);
resultsPanel.add(generateButton);
this.getContentPane().add(resultsPanel, BorderLayout.SOUTH);
this.pack();
ComponentUtil.centerComponentInWindow(this);
}
private void initializeFunctionList() {
try {
List functionClasses = ClassFinder.findClassesThatExtend(JMeterUtils.getSearchPaths(),
new Class[] { Function.class }, true);
Iterator iter = functionClasses.iterator();
String[] functionNames = new String[functionClasses.size()];
int count = 0;
while (iter.hasNext()) {
Class cl = Class.forName((String) iter.next());
functionNames[count] = ((Function) cl.newInstance()).getReferenceKey();
functionMap.put(functionNames[count], cl);
count++;
}
functionList = new JLabeledChoice(JMeterUtils.getResString("choose_function"), functionNames); //$NON-NLS-1$
functionList.addChangeListener(this);
} catch (IOException e) {
} catch (ClassNotFoundException e) {
} catch (InstantiationException e) {
} catch (IllegalAccessException e) {
}
}
public void stateChanged(ChangeEvent event) {
try {
Arguments args = new Arguments();
Function function = (Function) ((Class) functionMap.get(functionList.getText())).newInstance();
List argumentDesc = function.getArgumentDesc();
Iterator iter = argumentDesc.iterator();
while (iter.hasNext()) {
String help = (String) iter.next();
args.addArgument(help, ""); //$NON-NLS-1$
}
parameterPanel.configure(args);
parameterPanel.revalidate();
getContentPane().remove(parameterPanel);
this.pack();
getContentPane().add(parameterPanel, BorderLayout.CENTER);
this.pack();
this.validate();
this.repaint();
} catch (InstantiationException e) {
} catch (IllegalAccessException e) {
}
}
public void actionPerformed(ActionEvent e) {
StringBuffer functionCall = new StringBuffer("${");
functionCall.append(functionList.getText());
Arguments args = (Arguments) parameterPanel.createTestElement();
if (args.getArguments().size() > 0) {
functionCall.append("(");
PropertyIterator iter = args.iterator();
boolean first = true;
while (iter.hasNext()) {
Argument arg = (Argument) iter.next().getObjectValue();
if (!first) {
functionCall.append(",");
}
functionCall.append(arg.getValue());
first = false;
}
functionCall.append(")");
}
functionCall.append("}");
cutPasteFunction.setText(functionCall.toString());
}
private class HelpListener implements ActionListener {
public void actionPerformed(ActionEvent e) {
String[] source = new String[] { Help.HELP_FUNCTIONS, functionList.getText() };
ActionEvent helpEvent = new ActionEvent(source, e.getID(), "help"); //$NON-NLS-1$
ActionRouter.getInstance().actionPerformed(helpEvent);
}
}
}