/*
 Copyright (c) 2000-2006, The JAP-Team
 All rights reserved.
 Redistribution and use in source and binary forms, with or without modification,
 are permitted provided that the following conditions are met:

 - Redistributions of source code must retain the above copyright notice,
  this list of conditions and the following disclaimer.

 - Redistributions in binary form must reproduce the above copyright notice,
  this list of conditions and the following disclaimer in the documentation and/or
  other materials provided with the distribution.

 - Neither the name of the University of Technology Dresden, Germany nor the names of its contributors
  may be used to endorse or promote products derived from this software without specific
  prior written permission.


 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ``AS IS'' AND ANY EXPRESS
 OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
 AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS
 BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
 OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
 IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE
 */
package mixconfig;

import gui.ClipFrame;
import gui.GUIUtils;
import gui.dialog.JAPDialog;
import gui.help.JAPHelp;

import java.awt.Component;
import java.awt.Frame;
import java.awt.event.WindowAdapter;
import java.awt.event.WindowEvent;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.InetSocketAddress;
import java.net.Proxy;
import java.net.SocketAddress;
import java.net.Proxy.Type;
import java.security.SecureRandom;

import javax.swing.ImageIcon;
import javax.swing.JApplet;
import javax.swing.JFileChooser;
import javax.swing.JFrame;
import javax.swing.JPanel;
import javax.swing.WindowConstants;

import logging.LogHolder;
import logging.LogLevel;
import logging.LogType;
import logging.SystemErrLog;
import mixconfig.network.ProxyAdapter;
import mixconfig.tools.CertificateGenerator;

import org.w3c.dom.Document;

import anon.crypto.PKCS12;
import anon.crypto.X509DistinguishedName;
import anon.crypto.X509SubjectKeyIdentifier;
import anon.infoservice.Database;
import anon.infoservice.IDistributable;
import anon.infoservice.IDistributor;
import anon.infoservice.IMutableProxyInterface;
import anon.infoservice.InfoServiceDBEntry;
import anon.util.ClassUtil;
import anon.util.JAPMessages;
import anon.util.ResourceLoader;
import anon.util.XMLUtil;

public class MixConfig extends JApplet
{
	static
	{
		// Set log level here
		LogHolder.setLogInstance(new SystemErrLog(LogLevel.INFO, LogType.ALL));
	}

	public final static int SAVE_DIALOG = 1;
	public final static int OPEN_DIALOG = 2;
	public final static int CHOOSE_DIR_DIALOG = 3;
	public final static int OPEN_MULTIPLE_DIALOG = 4;
	public final static int FILTER_ALL = 0;
	public final static int FILTER_CER = 1;
	public final static int FILTER_XML = 2;
	public final static int FILTER_PFX = 4;
	public final static int FILTER_B64_PFX = 8;
	public final static int FILTER_B64_CER = 16;
	public final static int FILTER_P10 = 32;
	public final static int FILTER_B64_P10 = 64;

	public final static String VERSION = "00.05.015";  //NEVER change the layout of this line!!
	
	private static final String IMG_MAIN = MixConfig.class.getName() + "_icon.gif";

	private static final String MSG_COULD_NOT_INITIALISE =
		MixConfig.class.getName() + "_couldNotInitialise";
	private static final String MSG_CONFIG_FILE_NOT_FOUND =
		MixConfig.class.getName() + "_configFileNotFound";
	private static final String MSG_ERROR_OPEN_FILE =
		MixConfig.class.getName() + "_errorOpenFile";

    /** The configuration object edited by this <CODE>MixConfig</CODE> application. */
	private static MixConfiguration m_mixConfiguration;
	protected static JPanel m_mainPanel;
	private static ChoicePanel m_startPanel;
	private static Frame m_MainWindow;
	private static File m_fileCurrentDir = new File(ClassUtil.getUserDir());
	private static String m_currentFileName;

	private static Proxy proxy = null;
	
	public static void main(String[] argv)
	{
		File f = null;
		long startTime = System.currentTimeMillis();

		if (!JAPMessages.init("MixConfigMessages"))
		{
			GUIUtils.exitWithNoMessagesError("MixConfigMessages");
		}

		try
		{
	        for (int i = 0; i < argv.length; i++)
			{
				if (argv[i].equals("--help"))
				{
					usage();
					System.exit(0);
				}
				else if (argv[i].equals("--createConf"))
				{
					createMixOnCDConfiguration();
					System.exit(0);
				}
				else if (argv[i].equals("--logdetail"))
				{
					try
					{
						i++;
						LogHolder.setDetailLevel(Integer.parseInt(argv[i]));
					}
					catch (Exception a_e)
					{
						usage();
						System.exit(1);
					}
				}
				else if (argv[i].equals("--loglevel"))
				{
					try
					{
						i++;
						int j;
						for (j = 0; j < LogLevel.getLevelCount(); j++)
						{
							if (argv[i].trim().equalsIgnoreCase(LogLevel.getLevelName(j).trim()))
							{
								LogHolder.setLogInstance(new SystemErrLog(j, LogType.ALL));
								break;
							}
						}

						if (j >= LogLevel.getLevelCount())
						{
							throw new Exception();
						}

					}
					catch (Exception a_e)
					{
						a_e.printStackTrace();
						usage();
						System.exit(1);
					}
				}
				else
				{
					setCurrentFilename(argv[i]);
				}
			}

	        //read proxy settings which must be specified as system properties
	        String httpProxyHost = System.getProperty("http.proxyHost");
	        String httpProxyPort = System.getProperty("http.proxyPort");
	        //port and host must be specified.
	        if( (httpProxyHost != null) && (httpProxyPort != null) )
	        {
	        	configureProxy(httpProxyHost, httpProxyPort);
	        }
	        
			m_MainWindow = new JFrame();
			Database.registerDistributor(new IDistributor()
			{
				public void addJob(IDistributable a_newJob)
				{
					
				}
			});
			
			((JFrame)m_MainWindow).setDefaultCloseOperation(WindowConstants.DO_NOTHING_ON_CLOSE);
			
			// Initially do not allow resizing
			m_MainWindow.setResizable(false);
			JAPHelp.init(m_MainWindow, null);

			boolean fileLoaded = false;
			if (m_currentFileName != null && (f = new File(m_currentFileName)).exists())
			{
				LogHolder.log(LogLevel.DEBUG, LogType.MISC, "Load a configuration file...");
				//Force UTF-8 as standard input charset. 
				InputStreamReader reader = new InputStreamReader(new FileInputStream(f), "UTF-8");
				m_mixConfiguration = new MixConfiguration(reader);
				//m_mixConfiguration = new MixConfiguration(new FileReader(f));
				//If a file could be loaded, start with the expert view.
				fileLoaded = true;
			}
			else //no existing file is given
			{
				if (m_currentFileName != null)
				{
					JAPDialog.showErrorDialog(getMainWindow(),
											  JAPMessages.getString(MSG_CONFIG_FILE_NOT_FOUND,
						new File(m_currentFileName).toString()),
											  LogType.MISC);
				}

				m_mixConfiguration = new MixConfiguration();
			}
			// start with the expert view if a file was loaded.
			if(fileLoaded)
			{
				m_startPanel = new ChoicePanel((JFrame)m_MainWindow, null, ChoicePanel.CARD_EXPERT);
				m_startPanel.setExpertVisible();
			}
			else
			{
				m_startPanel = new ChoicePanel((JFrame)m_MainWindow, null,ChoicePanel.CARD_CHOICE);
			}
			((JFrame)m_MainWindow).setContentPane(m_startPanel);
			m_MainWindow.pack();
			GUIUtils.centerOnScreen(m_MainWindow);
			LogHolder.log(LogLevel.DEBUG, LogType.GUI, "Show the GUI's main window");
			m_MainWindow.setVisible(true);

			ImageIcon icon = GUIUtils.loadImageIcon(IMG_MAIN);
			if (icon != null)
			{
				m_MainWindow.setIconImage(icon.getImage());
			}

			m_MainWindow.addWindowListener(new WindowAdapter()
			{
				public void windowClosed(WindowEvent e)
				{
					windowClosing(e);
				}

				public void windowClosing(WindowEvent e)
				{
					m_startPanel.getMenu().exit();
				}
			});

			m_startPanel.setMessageTitle(); //set Message Title
			m_mixConfiguration.setSavedToFile(); // tell the GUI that this configuration has not changed

			LogHolder.log(LogLevel.INFO, LogType.MISC, "Startup time: "+(System.currentTimeMillis()-startTime)+" ms");
/*
			JAPDialog dialog = new JAPDialog(getMainWindow(), "Test", true);
			//javax.swing.JDialog dialog = new javax.swing.JDialog(getMainWindow(), "Test", true);
			dialog.setDefaultCloseOperation(JAPDialog.DO_NOTHING_ON_CLOSE);

			DialogContentPane pane =
				new DialogContentPane(dialog, new DialogContentPane.Layout("Title", JAPDialog.MESSAGE_TYPE_QUESTION),
									  new DialogContentPane.Options(JAPDialog.OPTION_TYPE_CANCEL_OK, "index"));
			pane.setDefaultButtonOperation(DialogContentPane.ON_CLICK_DISPOSE_DIALOG);
			pane.updateDialog();
			dialog.pack();

			dialog.setVisible(true);
			System.out.println("hello");
*/
			LogHolder.log(LogLevel.INFO, LogType.CRYPTO, "Initialising secure random generator ..");
			new SecureRandom().nextDouble();
			LogHolder.log(LogLevel.INFO, LogType.CRYPTO, "Secure random generator is initialised!");
		}
		catch (Exception e)
		{
			JAPDialog.showErrorDialog(getMainWindow(), JAPMessages.getString(MSG_COULD_NOT_INITIALISE), LogType.MISC, e);
			System.exit(1);
		}
	}

	/**
	 * Displays a short help message about available command line options for this class.
	 */
	public static void usage()
	{
		String logLevels = "";

		for (int i= 0; i < LogLevel.getLevelCount(); i++)
		{
			logLevels += LogLevel.getLevelName(i).trim() + ",";
		}
		if (logLevels.length() > 0)
		{
			logLevels = logLevels.substring(0, logLevels.length() - 1);
		}

		String message[] =
			{
			"Usage: java -cp <classpath> mixconfig.MixConfig [options] [configfilename]",
			"where options is one of",
			"--help       Show this message and exit.",
			"--logDetail  Sets the detail level of log messages. " +
			"(lowest is " + LogHolder.DETAIL_LEVEL_LOWEST +  ", highest is " +
			LogHolder.DETAIL_LEVEL_HIGHEST  + ")",
			"--logLevel   Sets the level of the log messages. (" + logLevels + ")",
			"--createConf Creates a generic configuration for MixOnCD."
		};
		for (int i = 0; i < message.length; i++)
		{
			System.out.println(message[i]);
		}
	}

	public void init() // For the applet.
	{
		try
		{
			if (!JAPMessages.init("MixConfigMessages"))
			{
				GUIUtils.exitWithNoMessagesError("MixConfigMessages");
			}
			Database.registerDistributor(new IDistributor()
			{
				public void addJob(IDistributable a_newJob)
				{
					
				}
			});
			m_MainWindow = (Frame)GUIUtils.getParentWindow(this);
			m_mixConfiguration = new MixConfiguration();
			JAPHelp.init(m_MainWindow, null);

			m_mainPanel = new ConfigFrame(null);
			m_startPanel = new ChoicePanel(null, getRootPane());

			setContentPane(m_startPanel);
		}
		catch (Exception e)
		{
			JAPDialog.showErrorDialog(getMainWindow(), JAPMessages.getString(MSG_COULD_NOT_INITIALISE),
									  LogType.MISC, e);
			System.exit(1);
		}
	}

	/** Gets the Mix configuration that is currently being edited
	 * @return The current Mix configuration
	 */
	public static MixConfiguration getMixConfiguration()
	{
		return m_mixConfiguration;
	}

	/** Sets the Mix configuration to be edited
	 * @param a_mixConfiguration The new Mix configuration
	 */
	public static void setMixConfiguration(MixConfiguration a_mixConfiguration)
	{
		m_mixConfiguration = a_mixConfiguration;
	}

	/** Sets the filname of a saved or loaded file to a variable.
	 * You can read it out, using getCurrentFileName()
	 * eg. for setting the MessageTitle
	 * @param filename a file name
	 */
	public static void setCurrentFilename(String filename)
	{
		m_currentFileName = filename;
	}

	/**
	 * Returns the path name of the currently edited config file.
	 * @return the file name of the currently edited configuration
	 */
	public static String getCurrentFileName()
	{
		return m_currentFileName;
	}

	public static void configureProxy(String host, String port)
	{
		try
    	{
    		SocketAddress proxyAddress = new InetSocketAddress(
    				host, Integer.parseInt(port));
    		configureProxy(new Proxy(Proxy.Type.HTTP, proxyAddress));
    	}
    	catch(NumberFormatException nfe)
    	{
    		throw new IllegalArgumentException(
    			"Bad proxy port: "+port+". Port must be a number between 0 and 65535");
    	}
	}
	
	public static void configureProxy(Proxy proxy)
	{
		if(proxy.type().equals(Type.DIRECT))
		{
			MixConfig.proxy = null;
			 System.clearProperty("http.proxyHost");
		     System.clearProperty("http.proxyPort");
			InfoServiceDBEntry.setMutableProxyInterface(
					new IMutableProxyInterface.DummyMutableProxyInterface());
		}
		else
		{
			MixConfig.proxy = proxy;
			System.setProperty("http.proxyHost", ((InetSocketAddress) proxy.address()).getHostName());
		     System.setProperty("http.proxyPort", ""+((InetSocketAddress) proxy.address()).getPort());
			//also set the proxy interface for the infoservice. 
			InfoServiceDBEntry.setMutableProxyInterface(
				new ProxyAdapter(proxy));
		}
	}
	
	public static Proxy getProxy()
	{
		return proxy;
	}
	
	/** 
	 * Display an info message dialog
	 * @param a_title The title of the message dialog
	 * @param a_message The messages to be displayed. Each message will be displayed in a single
	 * line.
	 */
	public static void info(String a_title, String[] a_message)
	{
		info(a_title, a_message, false);
	}
	
	public static void info(String a_title, String[] a_message, boolean a_bBlocking)
	{
		String message = "";
		if (a_message == null || a_message.length == 0)
		{
			JAPDialog.showMessageDialog(getMainWindow(), "", a_title);

		}
		else if (a_message.length == 1)
		{
			JAPDialog.showMessageDialog(getMainWindow(), a_message[0], a_title);
		}
		else
		{
			message += "<UL>";
			for (int i = 0; i < a_message.length && i < 20; i++)
			{
				if (a_message[i] == null || a_message[i].trim().length() == 0)
				{
					continue;
				}
				message +=  "<LI>" + a_message[i] + "</LI>";
			}
			message += "</UL>";
			if (a_bBlocking)
			{
				JAPDialog.showMessageDialog(getMainWindow(), message, a_title);
			}
			else
			{
				JAPDialog.showMessageDialog(getMainWindow(), message, a_title, new JAPDialog.LinkedInformationAdapter()
				{
					public boolean isModal()
					{
						return false;
					}
				});
			}
		}
	}

	public static void about()
	{
		JAPDialog.showMessageDialog(
			getMainWindow(),
			"Mix Configuration Tool<P>Version: " + VERSION + "</P>",
			"About",
			GUIUtils.loadImageIcon(IMG_MAIN));
	}


	public static Frame getMainWindow()
	{
		return m_MainWindow;
	}

	/**
	 *
	 * @param type int
	 * @param filter_type int
	 * @return the JFileChooser if an action was taken or null if the user clicked 'cancel'
	 */
	public static JFileChooser showFileDialog(Component a_component, int type, int filter_type)
	{
		int returnValue;
		SimpleFileFilter active = null;
		JFileChooser fd2;
		try
		{
			fd2 = new JFileChooser(m_fileCurrentDir);
		}
		catch (SecurityException a_e)
		{
			JAPDialog.showErrorDialog(a_component,
									  "Access to file system is not allowed when running as applet!",
									  LogType.MISC);
			return null;
		}
		if(type==MixConfig.CHOOSE_DIR_DIALOG)
			fd2.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY);
		else if (type == OPEN_MULTIPLE_DIALOG)
		{
			fd2.setFileSelectionMode(JFileChooser.FILES_ONLY);
			fd2.setMultiSelectionEnabled(true);
		}
		else
		{
			fd2.setFileSelectionMode(JFileChooser.FILES_ONLY);
		}
		if ( (filter_type & FILTER_CER) != 0)
		{
			fd2.addChoosableFileFilter(active = new SimpleFileFilter(FILTER_CER));
		}
		if ( (filter_type & FILTER_B64_CER) != 0)
		{
			fd2.addChoosableFileFilter(active = new SimpleFileFilter(FILTER_B64_CER));
		}
		if ( (filter_type & FILTER_XML) != 0)
		{
			fd2.addChoosableFileFilter(active = new SimpleFileFilter(FILTER_XML));
		}
		if ( (filter_type & FILTER_P10) != 0)
		{
			fd2.addChoosableFileFilter(new SimpleFileFilter(FILTER_P10));
		}
		if ( (filter_type & FILTER_B64_P10) != 0)
		{
			fd2.addChoosableFileFilter(new SimpleFileFilter(FILTER_B64_P10));
		}
		if ( (filter_type & FILTER_PFX) != 0)
		{
			fd2.addChoosableFileFilter(new SimpleFileFilter(FILTER_PFX));
		}
		if ( (filter_type & FILTER_B64_PFX) != 0)
		{
			fd2.addChoosableFileFilter(new SimpleFileFilter(FILTER_B64_PFX));
		}

		if (active != null)
		{
			fd2.setFileFilter(active);
		}
		fd2.setFileHidingEnabled(false);
		if (type == SAVE_DIALOG)
		{
			returnValue = fd2.showSaveDialog(a_component);
		}
		else
		{
			returnValue = GUIUtils.showMonitoredFileChooser(fd2, a_component);
		}
		if (returnValue == JFileChooser.CANCEL_OPTION)
		{
			return null;
		}

		m_fileCurrentDir = fd2.getCurrentDirectory();
		if(type==MixConfig.CHOOSE_DIR_DIALOG) //seems that the JFileChosser always sets the parent of the selected item
			                                  //that means if we select are directory, the parent of that directory become the current directory...
											//so we change this back here...
		{
			m_fileCurrentDir=fd2.getSelectedFile();
		}
		return fd2;
	}

	public static byte[] openFile(Component a_component, int type)
	{
		try
		{
			JFileChooser fileChooser = showFileDialog(a_component, MixConfig.OPEN_DIALOG, type);
			if (fileChooser == null)
			{
				return null;
			}

			File file = fileChooser.getSelectedFile();
	
			if (file != null)
			{
				try
				{
					return ResourceLoader.getStreamAsBytes(new FileInputStream(file));
				}
				catch (IOException e)
				{
					JAPDialog.showErrorDialog(getMainWindow(),
											  JAPMessages.getString(MSG_ERROR_OPEN_FILE, file.toString()),
											  LogType.MISC, e);
				}
			}
		}
		catch (SecurityException a_e)
		{
			ClipFrame open = new ClipFrame(a_component, "Paste the data to be imported in " +
					   "the area provided.", true);
			open.setVisible(true);
			return open.getText().getBytes();
		}
		return null;
	}

	private static void createMixOnCDConfiguration() throws Exception
	{

		MixConfiguration configuration = new MixConfiguration();

		// create mix certificate
		X509DistinguishedName dn = new X509DistinguishedName("CN=");
		CertificateGenerator certificateGenerator = new CertificateGenerator(dn, null, true);
		certificateGenerator.run();
		PKCS12 privateCert = certificateGenerator.getCertificate();
		configuration.setValue("Certificates/OwnCertificate/X509PKCS12", privateCert.toByteArray());
		configuration.setValue("Certificates/OwnCertificate/X509Certificate",
							   privateCert.getX509Certificate().toByteArray());

		// create other entries
		configuration.setValue("Network/ListenerInterfaces/ListenerInterface/Port", 6544);
		configuration.setValue("Network/ListenerInterfaces/ListenerInterface/NetworkProtocol",
								   "RAW/TCP");
		configuration.setValue("General/CascadeName", "Dynamic Cascade");
		configuration.setValue("General/MixName", "Dynamic Mix");
		configuration.setValue("General/MixID",
								   ((X509SubjectKeyIdentifier)privateCert.getExtensions().getExtension(
									X509SubjectKeyIdentifier.IDENTIFIER)).getValueWithoutColon());
		configuration.setValue("General/UserID", "mix");
		configuration.setValue("General/Logging/SysLog", true);
		configuration.setValue("Network/InfoService/AllowAutoConfiguration", true);

		Document document = configuration.getDocument();
		XMLUtil.formatHumanReadable(document);
		System.out.println(XMLUtil.toString(document));
	}
}