/*
* Copyright 2003,2004,2005,2006,2007 Colin Crist
*
* Licensed 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 hermes.swing.actions;
import hermes.Hermes;
import hermes.HermesException;
import hermes.browser.HermesBrowser;
import hermes.browser.IconCache;
import hermes.browser.model.tree.DestinationConfigTreeNode;
import hermes.browser.model.tree.HermesTreeNode;
import hermes.browser.model.tree.MessageStoreTreeNode;
import hermes.browser.tasks.BrowseDestinationTask;
import hermes.browser.tasks.BrowseMessageStoreTask;
import hermes.browser.tasks.MessageTaskListener;
import hermes.browser.tasks.Task;
import hermes.browser.tasks.TaskSupport;
import hermes.config.DestinationConfig;
import hermes.store.MessageStore;
import hermes.util.TextUtils;
import java.awt.event.ActionEvent;
import java.io.File;
import java.io.FileOutputStream;
import java.util.ArrayList;
import java.util.List;
import javax.jms.Message;
import javax.naming.NamingException;
import javax.swing.Action;
import javax.swing.JFileChooser;
import javax.swing.ProgressMonitor;
import javax.swing.event.TreeSelectionEvent;
import javax.swing.event.TreeSelectionListener;
import javax.swing.tree.TreePath;
import org.apache.log4j.Logger;
/**
* Save all messages on the selected queue into a file in jms2xml.xsd format.
*
* @author colincrist@hermesjms.com
* @version $Id$
*/
public class SaveAllMessagesAsXMLAction extends BrowseActionListenerAdapter {
/**
*
*/
private static final long serialVersionUID = 2695865313851844473L;
private static final Logger log = Logger.getLogger(SaveAllMessagesAsXMLAction.class);
public SaveAllMessagesAsXMLAction() {
putValue(Action.NAME, "Save all as XML...");
putValue(Action.SHORT_DESCRIPTION, "Save all messages encoded a XML.");
putValue(Action.SMALL_ICON, IconCache.getIcon("hermes.messages.save.xml.many"));
setEnabled(false);
HermesBrowser.getBrowser().getBrowserTree().getSelectionModel().addTreeSelectionListener(new TreeSelectionListener() {
public void valueChanged(TreeSelectionEvent e) {
if (e.getNewLeadSelectionPath() != null) {
setEnabled(e.getNewLeadSelectionPath().getLastPathComponent() instanceof DestinationConfigTreeNode
|| e.getNewLeadSelectionPath().getLastPathComponent() instanceof MessageStoreTreeNode);
} else {
setEnabled(false);
}
}
});
}
private Hermes getMyHermes() throws HermesException, NamingException {
if (HermesBrowser.getBrowser().getConfig().getMessageStoreMessageFactory() != null) {
return (Hermes) HermesBrowser.getBrowser().getContext().lookup(HermesBrowser.getBrowser().getConfig().getMessageStoreMessageFactory());
} else {
return HermesBrowser.getBrowser().getHermes();
}
}
public void actionPerformed(ActionEvent arg0) {
try {
TaskSupport task = null;
String f = null;
Hermes h = null;
TreePath currentPath = HermesBrowser.getBrowser().getBrowserTree().getSelectionModel().getSelectionPath();
if (currentPath.getLastPathComponent() instanceof DestinationConfigTreeNode) {
final DestinationConfigTreeNode destinationNode = HermesBrowser.getBrowser().getBrowserTree().getSelectedDestinationNodes().get(0);
final DestinationConfig destinationConfig = destinationNode.getConfig();
h = ((HermesTreeNode) destinationNode.getParent()).getHermes();
task = new BrowseDestinationTask(h, destinationConfig);
f = destinationConfig.getName();
} else {
MessageStore messageStore = HermesBrowser.getBrowser().getBrowserTree().getSelectedMessageStore();
h = getMyHermes();
task = new BrowseMessageStoreTask(h, messageStore, null, null);
f = messageStore.getId();
}
final Hermes hermes = h ;
final String from = f ;
JFileChooser chooser;
if (DirectoryCache.lastSaveAsDirectory == null) {
chooser = new JFileChooser(new File(HermesBrowser.getBrowser().getRepositoryManager().getDirectory()));
} else {
chooser = new JFileChooser(DirectoryCache.lastSaveAsDirectory);
}
if (chooser.showDialog(HermesBrowser.getBrowser(), "Save All Messages...") == JFileChooser.APPROVE_OPTION) {
DirectoryCache.lastSaveAsDirectory = chooser.getCurrentDirectory();
final File file = chooser.getSelectedFile();
final List<Message> messages = new ArrayList<Message>();
final ProgressMonitor progress = new ProgressMonitor(HermesBrowser.getBrowser(), "Saving from " + from + " to " + file.getName(),
"Browsing...", 0, 3);
progress.setMillisToPopup(0);
progress.setMillisToDecideToPopup(0);
task.addTaskListener(new MessageTaskListener() {
public void onMessage(Task task, Message message) {
if (progress.isCanceled()) {
task.stop();
}
messages.add(message);
progress.setNote(messages.size() + " message" + TextUtils.plural(messages.size()) + " read");
}
public void onThrowable(Task task, Throwable t) {
HermesBrowser.getBrowser().showErrorDialog(t);
}
public void onStopped(Task task) {
if (!progress.isCanceled()) {
doSave();
} else {
Hermes.ui.getDefaultMessageSink().add("Cancelled");
}
}
public void onStatus(Task task, String status) {
// NOP
}
public void onStarted(Task task) {
Hermes.ui.getDefaultMessageSink().add("Browsing " + from + "...");
progress.setProgress(1);
}
private void doSave() {
FileOutputStream ostream = null;
try {
progress.setProgress(2);
progress.setNote("Saving to " + file.getName());
Hermes.ui.getDefaultMessageSink().add("Saving to " + file.getName());
ostream = new FileOutputStream(file);
hermes.toXML(messages, ostream);
ostream.close();
Hermes.ui.getDefaultMessageSink().add(
messages.size() + " message" + TextUtils.plural(messages.size()) + " saved to " + file.getName());
} catch (Exception ex) {
HermesBrowser.getBrowser().showErrorDialog(ex);
} finally {
progress.setProgress(3);
progress.close();
if (ostream != null) {
try {
ostream.close();
} catch (Exception ex) {
log.error(ex.getMessage(), ex);
}
}
}
}
});
task.start();
}
} catch (Exception e) {
HermesBrowser.getBrowser().showErrorDialog("Unable to save:", e);
}
}
}