/*
 *
 * 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.model;

import java.io.File;
import java.io.FilenameFilter;
import java.io.IOException;


import org.parosproxy.paros.Constant;
import org.parosproxy.paros.db.Database;
import org.xml.sax.SAXException;


/**
 *
 * To change the template for this generated type comment go to
 * Window - Preferences - Java - Code Generation - Code and Comments
 */
public class Model {

    private static Model model = null;

    private static final String DBNAME_TEMPLATE = Constant.DBNAME_TEMPLATE;
    private static final String DBNAME_UNTITLED = Constant.DBNAME_UNTITLED;
	private static int DBNAME_COPY = 1;
	
    private Session session = null;
	private OptionsParam optionsParam = null;
	private Database db = null;
	private String currentDBNameUntitled = "";
	
	public Model() {
	    // make sure the variable here will not refer back to model itself.
	    // DO it in init or respective getter.
	    
	    session = new Session(this);
	    optionsParam = new OptionsParam();

	}
	
	/**
	 * @return Returns the optionsParam.
	 */
	public OptionsParam getOptionsParam() {
	    if (optionsParam == null) {
	        optionsParam = new OptionsParam();
	    }
		return optionsParam;
	}
	/**
	 * @param optionsParam The optionsParam to set.
	 */
	public void setOptionsParam(OptionsParam param) {
		optionsParam = param;
	}
	/**
	 * @return Returns the session.
	 */
	public Session getSession() {
	    if (session == null) {
	        session = new Session(this);
	    }
		return session;
	}
	/**
	 * @param session The session to set.
	 */
	public void setSession(Session paramSession) {
		session = paramSession;
	}
	
	public void init() throws SAXException, IOException, Exception {
	    db = Database.getSingleton();
	    
	    createAndOpenUntitledDb();
		HistoryReference.setTableHistory(getDb().getTableHistory());
		getOptionsParam().readAndParseFile(Constant.FILE_OPTIONS);
	}

	
	public static Model getSingleton() {
	    if (model == null) {
	        model = new Model();
	    }
	    return model;
	}
	
    /**
     * @return Returns the db.
     */
    public Database getDb() {
        return db;
    }
    
    public void moveSessionDb(String destFile) throws Exception {

        // 	always use copySession because moving file does not work in Debian,
        //	and for Windows renaming file acrossing different drives does not work.

        copySessionDb(currentDBNameUntitled, destFile);

        
//        getDb().close();
//        
//        boolean result = false;
//        File fileIn1 = new File(currentDBNameUntitled + ".data");
//        File fileIn2 = new File(currentDBNameUntitled + ".script");
//        File fileIn3 = new File(currentDBNameUntitled + ".properties");
//        File fileIn4 = new File(currentDBNameUntitled + ".backup");
//
//        File fileOut1 = new File(destFile + ".data");
//        File fileOut2 = new File(destFile + ".script");
//        File fileOut3 = new File(destFile + ".properties");
//        File fileOut4 = new File(destFile + ".backup");
//        
//        if (fileOut1.exists()) fileOut1.delete();
//        if (fileOut2.exists()) fileOut2.delete();
//        if (fileOut3.exists()) fileOut3.delete();
//        if (fileOut4.exists()) fileOut4.delete();
//
//        result = fileIn1.renameTo(fileOut1);
//        result = fileIn2.renameTo(fileOut2);
//        result = fileIn3.renameTo(fileOut3);
//        if (fileIn4.exists()) {
//            result = fileIn4.renameTo(fileOut4);
//        }
//
//        getDb().open(destFile);
        
    }
    
    public void copySessionDb(String currentFile, String destFile) throws Exception {
        
        getDb().close(false);
        
        // copy session related files to the path specified
        FileCopier copier = new FileCopier();        
        
        File fileIn1 = new File(currentFile + ".data");
        File fileOut1 = new File(destFile + ".data");
        copier.copy(fileIn1, fileOut1);
        
        File fileIn2 = new File(currentFile + ".script");
        File fileOut2 = new File(destFile + ".script");
        copier.copy(fileIn2, fileOut2);
        
        File fileIn3 = new File(currentFile + ".properties");
        File fileOut3 = new File(destFile + ".properties");
        copier.copy(fileIn3, fileOut3);

        File fileIn4 = new File(currentFile + ".backup");
        if (fileIn4.exists()) {
            File fileOut4 = new File(destFile + ".backup");
            copier.copy(fileIn4, fileOut4);
        }

        getDb().open(destFile);
        
    }

    
    public void createAndOpenUntitledDb() throws ClassNotFoundException, Exception {

        getDb().close(false);

        // delete all untitled session db in "session" directory
        File dir = new File("session");
        File[] listFile = dir.listFiles(new FilenameFilter() {
            public boolean accept(File dir1, String fileName) {
                if (fileName.startsWith("untitled")) {
                    return true;
                }
                return false;
            }
        });
        for (int i=0; i<listFile.length;i++) {
            listFile[i].delete();
        }
        
        // copy and create new template db
        currentDBNameUntitled = DBNAME_UNTITLED + DBNAME_COPY;
        FileCopier copier = new FileCopier();
	    File fileIn = new File(DBNAME_TEMPLATE + ".data");
	    File fileOut = new File(currentDBNameUntitled + ".data");
	    fileOut.delete();	    
	    
	    copier.copy(fileIn, fileOut);

	    fileIn = new File(DBNAME_TEMPLATE + ".properties");
	    fileOut = new File(currentDBNameUntitled  + ".properties"); 
	    fileOut.delete();

	    copier.copy(fileIn, fileOut);
	    
	    fileIn = new File(DBNAME_TEMPLATE + ".script");
	    fileOut = new File(currentDBNameUntitled + ".script"); 
	    fileOut.delete();
	    copier.copy(fileIn, fileOut);
	    
	    fileIn = new File(currentDBNameUntitled + ".backup");
	    if (fileIn.exists()) {
	        fileIn.delete();
	    }
	    
	    getDb().open(currentDBNameUntitled);
	    DBNAME_COPY++;
    }
}
