/*
*
* Paros and its related class files.
*
* Paros is an HTTP/HTTPS proxy for assessing web application security.
* Copyright (C) 2003-2004 Chinotec Technologies Company
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the Clarified Artistic License
* as published by the Free Software Foundation.
*
* This program 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
* Clarified Artistic License for more details.
*
* You should have received a copy of the Clarified Artistic License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*/
package org.parosproxy.paros.control;
import java.io.File;
import javax.swing.JFileChooser;
import javax.swing.JOptionPane;
import javax.swing.filechooser.FileFilter;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.parosproxy.paros.Constant;
import org.parosproxy.paros.model.Model;
import org.parosproxy.paros.model.Session;
import org.parosproxy.paros.model.SessionListener;
import org.parosproxy.paros.view.SessionDialog;
import org.parosproxy.paros.view.View;
import org.parosproxy.paros.view.WaitMessageDialog;
/**
*
* To change the template for this generated type comment go to Window -
* Preferences - Java - Code Generation - Code and Comments
*/
public class MenuFileControl implements SessionListener {
private static Log log = LogFactory.getLog(MenuFileControl.class);
private View view = null;
private Model model = null;
private Control control = null;
private WaitMessageDialog waitMessageDialog = null;
public MenuFileControl(Model model, View view, Control control) {
this.view = view;
this.model = model;
this.control = control;
}
public void exit() {
boolean isNewState = model.getSession().isNewState();
if (isNewState) {
if (view.showConfirmDialog("The current session is not saved. Discard?") != JOptionPane.OK_OPTION) {
return;
}
model.getSession().discard();
}
WaitMessageDialog dialog = view.getWaitMessageDialog("Shutting down server and database...");
Thread t = new Thread(new Runnable() {
public void run() {
control.shutdown(false);
log.info(Constant.PROGRAM_TITLE + " terminated.");
System.exit(0);
}
});
t.start();
dialog.setVisible(true);
}
public void newSession(boolean isPromptNewSession) throws ClassNotFoundException, Exception {
if (isPromptNewSession) {
if (model.getSession().isNewState()) {
if (view.showConfirmDialog("The current session is not saved. Discard and create new session?") != JOptionPane.OK_OPTION) {
return;
}
model.getSession().discard();
} else if (view.showConfirmDialog("The current session will be closed. Create new session?") != JOptionPane.OK_OPTION) {
return;
}
model.createAndOpenUntitledDb();
}
Session session = new Session(model);
log.info("new session file created");
model.setSession(session);
view.getSiteTreePanel().getTreeSite().setModel(session.getSiteTree());
control.getExtensionLoader().sessionChangedAllPlugin(session);
// refresh display
view.getMainFrame().setTitle(session.getSessionName() + " - " + Constant.PROGRAM_NAME + " " + Constant.PROGRAM_VERSION);
view.getOutputPanel().clear();
}
public void openSession() {
JFileChooser chooser = new JFileChooser(model.getOptionsParam().getUserDirectory());
File file = null;
chooser.setFileFilter(new FileFilter() {
public boolean accept(File file) {
if (file.isDirectory()) {
return true;
} else if (file.isFile() && file.getName().endsWith(".session")) {
return true;
}
return false;
}
public String getDescription() {
return "Andiparos session";
}
});
int rc = chooser.showOpenDialog(view.getMainFrame());
if (rc == JFileChooser.APPROVE_OPTION) {
try {
file = chooser.getSelectedFile();
if (file == null) {
return;
}
model.getOptionsParam().setUserDirectory(chooser.getCurrentDirectory());
Session session = model.getSession();
log.info("opening session file " + file.getAbsolutePath());
waitMessageDialog = view.getWaitMessageDialog("Loading session file. Please wait ...");
session.open(file, this);
waitMessageDialog.setVisible(true);
} catch (Exception e) {
e.printStackTrace();
}
}
}
public void saveSession() {
Session session = model.getSession();
if (session.isNewState()) {
view.showWarningDialog("Please use \"Save As...\"");
return;
}
try {
waitMessageDialog = view.getWaitMessageDialog("Saving session file. Please wait ...");
session.save(session.getFileName(), this);
log.info("saving session file " + session.getFileName());
// ZAP: If the save is quick the dialog can already be null here
if (waitMessageDialog != null) {
waitMessageDialog.setVisible(true);
}
} catch (Exception e) {
view.showWarningDialog("Error saving session file.");
log.error("error saving session file " + session.getFileName());
log.error(e.getMessage());
}
}
public void saveAsSession() {
Session session = model.getSession();
JFileChooser chooser = new JFileChooser(model.getOptionsParam().getUserDirectory());
chooser.setFileFilter(new FileFilter() {
public boolean accept(File file) {
if (file.isDirectory()) {
return true;
} else if (file.isFile() && file.getName().endsWith(".session")) {
return true;
}
return false;
}
public String getDescription() {
return "Andiparos session";
}
});
File file = null;
int rc = chooser.showSaveDialog(view.getMainFrame());
if (rc == JFileChooser.APPROVE_OPTION) {
file = chooser.getSelectedFile();
if (file == null) {
return;
}
model.getOptionsParam().setUserDirectory(chooser.getCurrentDirectory());
String fileName = file.getAbsolutePath();
if (!fileName.endsWith(".session")) {
fileName += ".session";
}
try {
waitMessageDialog = view.getWaitMessageDialog("Saving new session file. Please wait ...");
session.save(fileName, this);
log.info("save as session file " + session.getFileName());
waitMessageDialog.setVisible(true);
} catch (Exception e) {
e.printStackTrace();
}
}
}
private void setTitle() {
if (model.getSession().isNewState()) {
// No file name
view.getMainFrame().setTitle(model.getSession().getSessionName() + " - " + Constant.PROGRAM_NAME);
} else {
File file = new File(model.getSession().getFileName());
view.getMainFrame().setTitle(
model.getSession().getSessionName() + " - " +
file.getName().replaceAll(".session\\z","") + " - " +
Constant.PROGRAM_NAME);
}
}
public void properties() {
SessionDialog dialog = view.getSessionDialog("Session Properties");
dialog.initParam(model.getSession());
dialog.showDialog(false);
// ZAP: Set the title consistently
setTitle();
}
/*
* (non-Javadoc)
*
* @see
* org.parosproxy.paros.model.SessionListener#sessionOpened(java.lang.Exception
* )
*/
public void sessionOpened(File file, Exception e) {
if (e == null) {
control.getExtensionLoader().sessionChangedAllPlugin(model.getSession());
// ZAP: Set the title consistently
setTitle();
} else {
view.showWarningDialog("Error opening session file");
log.error("error opening session file " + file.getAbsolutePath());
log.error(e.getMessage());
}
if (waitMessageDialog != null) {
waitMessageDialog.setVisible(false);
waitMessageDialog = null;
}
}
/*
* (non-Javadoc)
*
* @see
* org.parosproxy.paros.model.SessionListener#sessionSaved(java.lang.Exception
* )
*/
public void sessionSaved(Exception e) {
if (e == null) {
view.getMainFrame().getMainMenuBar().getMenuFileSave().setEnabled(true);
// ZAP: Set the title consistently
setTitle();
} else {
view.showWarningDialog("Error saving session file.");
e.printStackTrace();
log.error("error saving session file " + model.getSession().getFileName());
log.error(e.getStackTrace());
}
if (waitMessageDialog != null) {
waitMessageDialog.setVisible(false);
waitMessageDialog = null;
}
}
}