/*
* 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.action;
import java.awt.event.ActionEvent;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.HashSet;
import java.util.Set;
import javax.swing.JFileChooser;
import javax.swing.JTree;
import javax.swing.tree.TreePath;
import org.apache.jmeter.exceptions.IllegalUserActionException;
import org.apache.jmeter.gui.GuiPackage;
import org.apache.jmeter.gui.tree.JMeterTreeNode;
import org.apache.jmeter.gui.util.FileDialoger;
import org.apache.jmeter.gui.util.MenuFactory;
import org.apache.jmeter.save.SaveService;
import org.apache.jmeter.services.FileServer;
import org.apache.jmeter.testelement.TestElement;
import org.apache.jmeter.testelement.TestPlan;
import org.apache.jmeter.testelement.WorkBench;
import org.apache.jmeter.util.JMeterUtils;
import org.apache.jorphan.collections.HashTree;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.jorphan.util.JOrphanUtils;
import org.apache.log.Logger;
import com.thoughtworks.xstream.converters.ConversionException;
/**
* Handles the Open (load a new file) and Merge commands.
*
*/
public class Load implements Command {
private static final Logger log = LoggingManager.getLoggerForClass();
private static final boolean expandTree = JMeterUtils.getPropDefault("onload.expandtree", true); //$NON-NLS-1$
private static final Set commands = new HashSet();
static {
commands.add(ActionNames.OPEN);
commands.add(ActionNames.MERGE);
}
public Load() {
super();
}
public Set getActionNames() {
return commands;
}
public void doAction(ActionEvent e) {
JFileChooser chooser = FileDialoger.promptToOpenFile(new String[] { ".jmx" }); //$NON-NLS-1$
if (chooser == null) {
return;
}
File selectedFile = chooser.getSelectedFile();
if(selectedFile != null) {
boolean merging = e.getActionCommand().equals(ActionNames.MERGE);
// We must ask the user if it is ok to close current project
if(!merging) {
if (!Close.performAction(e)) {
return;
}
}
loadProjectFile(e, selectedFile, merging);
}
}
static void loadProjectFile(ActionEvent e, File f, boolean merging) {
GuiPackage guiPackage = GuiPackage.getInstance();
InputStream reader = null;
try {
if (f != null) {
boolean isTestPlan = false;
if (merging) {
log.info("Merging file: " + f);
} else {
log.info("Loading file: " + f);
FileServer.getFileServer().setBasedir(f.getAbsolutePath());
}
reader = new FileInputStream(f);
HashTree tree = SaveService.loadTree(reader);
isTestPlan = insertLoadedTree(e.getID(), tree, merging);
// don't change name if merging
if (!merging && isTestPlan) {
guiPackage.setTestPlanFile(f.getAbsolutePath());
}
}
} catch (NoClassDefFoundError ex) // Allow for missing optional jars
{
log.warn("Missing jar file", ex);
String msg = ex.getMessage();
if (msg == null) {
msg = "Missing jar file - see log for details";
}
JMeterUtils.reportErrorToUser(msg);
} catch (ConversionException ex) {
log.warn("Could not convert file "+ex);
JMeterUtils.reportErrorToUser(SaveService.CEtoString(ex));
} catch (IOException ex) {
log.warn("Error reading file: "+ex);
String msg = ex.getMessage();
if (msg == null) {
msg = "Unexpected error - see log for details";
}
JMeterUtils.reportErrorToUser(msg);
} catch (Exception ex) {
log.warn("Unexpected error", ex);
String msg = ex.getMessage();
if (msg == null) {
msg = "Unexpected error - see log for details";
}
JMeterUtils.reportErrorToUser(msg);
} finally {
JOrphanUtils.closeQuietly(reader);
guiPackage.updateCurrentGui();
guiPackage.getMainFrame().repaint();
}
}
/**
* Returns a boolean indicating whether the loaded tree was a full test plan
*/
public static boolean insertLoadedTree(int id, HashTree tree, boolean merging) throws Exception, IllegalUserActionException {
// convertTree(tree);
if (tree == null) {
throw new Exception("Error in TestPlan - see log file");
}
boolean isTestPlan = tree.getArray()[0] instanceof TestPlan;
// If we are loading a new test plan, initialize the tree with the testplan node we are loading
GuiPackage guiInstance = GuiPackage.getInstance();
if(isTestPlan && !merging) {
guiInstance.clearTestPlan((TestElement)tree.getArray()[0]);
}
if (merging){ // Check if target of merge is reasonable
TestElement te = (TestElement)tree.getArray()[0];
if (!(te instanceof WorkBench || te instanceof TestPlan)){// These are handled specially by addToTree
boolean ok = MenuFactory.canAddTo(guiInstance.getCurrentNode(), te);
if (!ok){
String name = te.getName();
String className = te.getClass().getName();
className = className.substring(className.lastIndexOf(".")+1);
throw new IllegalUserActionException("Can't merge "+name+" ("+className+") here");
}
}
}
HashTree newTree = guiInstance.addSubTree(tree);
guiInstance.updateCurrentGui();
guiInstance.getMainFrame().getTree().setSelectionPath(
new TreePath(((JMeterTreeNode) newTree.getArray()[0]).getPath()));
tree = guiInstance.getCurrentSubTree();
// Send different event wether we are merging a test plan into another test plan,
// or loading a testplan from scratch
ActionEvent actionEvent = null;
if(!merging) {
actionEvent = new ActionEvent(tree.get(tree.getArray()[tree.size() - 1]), id, ActionNames.SUB_TREE_LOADED);
}
else {
actionEvent = new ActionEvent(tree.get(tree.getArray()[tree.size() - 1]), id, ActionNames.SUB_TREE_MERGED);
}
ActionRouter.getInstance().actionPerformed(actionEvent);
if (expandTree && !merging) { // don't automatically expand when merging
JTree jTree = guiInstance.getMainFrame().getTree();
for(int i = 0; i < jTree.getRowCount(); i++) {
jTree.expandRow(i);
}
}
return isTestPlan;
}
public static boolean insertLoadedTree(int id, HashTree tree) throws Exception, IllegalUserActionException {
return insertLoadedTree(id, tree, false);
}
}