AuthPanel.java

/******************************************************************************
 *                                                                            *
 *                    Copyright 2016 Subterranean Security                    *
 *                                                                            *
 *  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 com.subterranean_security.crimson.viewer.ui.screen.netman.auth;

import java.awt.BorderLayout;
import java.awt.Insets;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.File;

import javax.swing.JButton;
import javax.swing.JFileChooser;
import javax.swing.JMenuBar;
import javax.swing.JPanel;
import javax.swing.filechooser.FileNameExtensionFilter;

import com.subterranean_security.crimson.core.proto.Misc.AuthMethod;
import com.subterranean_security.crimson.core.util.CryptoUtil;
import com.subterranean_security.crimson.viewer.net.ViewerCommands;
import com.subterranean_security.crimson.viewer.ui.UIStore;
import com.subterranean_security.crimson.viewer.ui.UIUtil;
import com.subterranean_security.crimson.viewer.ui.common.panels.sl.epanel.EPanel;

public class AuthPanel extends JPanel {

	private static final long serialVersionUID = 1L;

	public EPanel ep;

	public JButton btnExport;

	private AuthTable authTable;

	public AuthPanel() {
		setLayout(new BorderLayout(0, 0));

		JPanel panel = new JPanel();
		panel.setLayout(new BorderLayout(0, 0));
		ep = new EPanel(panel);
		add(ep, BorderLayout.CENTER);

		authTable = new AuthTable(this);
		panel.add(authTable, BorderLayout.CENTER);

		JMenuBar menuBar = new JMenuBar();
		panel.add(menuBar, BorderLayout.NORTH);

		JButton btnAddGroup = new JButton(UIUtil.getIcon("icons16/general/group_add.png"));
		btnAddGroup.addActionListener(new ActionListener() {
			public void actionPerformed(ActionEvent e) {
				if (UIStore.ECreateGroup == null) {
					new Thread(new Runnable() {
						public void run() {
							resetEPanels();
							UIStore.ECreateGroup = new CreateGroup(ep);
							ep.raise(UIStore.ECreateGroup, 240);
						}
					}).start();
				}
			}
		});
		btnAddGroup.setToolTipText("Create new group");

		btnAddGroup.setMargin(new Insets(2, 2, 2, 2));
		menuBar.add(btnAddGroup);

		JButton btnImport = new JButton(UIUtil.getIcon("icons16/general/group_import.png"));
		btnImport.addActionListener(new ActionListener() {
			public void actionPerformed(ActionEvent arg0) {

				new Thread(new Runnable() {
					public void run() {
						btnImport.setEnabled(false);
						JFileChooser jfc = new JFileChooser();
						jfc.setDialogTitle("Select group to import");
						jfc.addChoosableFileFilter(new FileNameExtensionFilter("Crimson Group File", "cg"));
						jfc.setAcceptAllFileFilterUsed(false);
						jfc.setFileSelectionMode(JFileChooser.FILES_ONLY);

						if (jfc.showDialog(UIStore.netMan, "Import") == JFileChooser.APPROVE_OPTION) {
							File file = jfc.getSelectedFile();
							AuthMethod am = CryptoUtil.importGroup(file);
							ViewerCommands.createAuthMethod(am);
						}

						btnImport.setEnabled(true);
					}
				}).start();

			}
		});
		btnImport.setToolTipText("Import group from file");
		btnImport.setMargin(new Insets(2, 4, 2, 4));
		menuBar.add(btnImport);

		btnExport = new JButton(UIUtil.getIcon("icons16/general/group_export.png"));
		btnExport.addActionListener(new ActionListener() {
			public void actionPerformed(ActionEvent e) {

				new Thread(new Runnable() {
					public void run() {
						btnExport.setEnabled(false);
						JFileChooser jfc = new JFileChooser();
						jfc.setDialogTitle("Export selected group to file");
						jfc.addChoosableFileFilter(new FileNameExtensionFilter("Crimson Group File", "cg"));
						jfc.setAcceptAllFileFilterUsed(false);
						jfc.setFileSelectionMode(JFileChooser.FILES_ONLY);

						if (jfc.showDialog(UIStore.netMan, "Export") == JFileChooser.APPROVE_OPTION) {
							File file = jfc.getSelectedFile();
							if (!file.getAbsolutePath().endsWith(".cg")) {
								file = new File(file.getAbsolutePath() + ".cg");
							}
							CryptoUtil.exportGroup(authTable.getSelected(), file);
						}
						btnExport.setEnabled(true);
					}
				}).start();
			}

		});
		btnExport.setEnabled(false);
		btnExport.setToolTipText("Export group to file");
		btnExport.setMargin(new Insets(2, 2, 2, 2));
		menuBar.add(btnExport);

		JButton btnNewButton_1 = new JButton(UIUtil.getIcon("icons16/general/textfield_password_add.png"));
		btnNewButton_1.addActionListener(new ActionListener() {
			public void actionPerformed(ActionEvent arg0) {
				resetEPanels();

				if (UIStore.ECreatePassword == null) {
					new Thread(new Runnable() {
						public void run() {
							resetEPanels();
							UIStore.ECreatePassword = new CreatePassword(ep);
							ep.raise(UIStore.ECreatePassword, 100);
						}
					}).start();
				}
			}
		});
		btnNewButton_1.setToolTipText("Create new password");
		btnNewButton_1.setMargin(new Insets(2, 2, 2, 2));
		menuBar.add(btnNewButton_1);

	}

	public void resetEPanels() {
		UIStore.ECreateGroup = null;
		UIStore.ECreatePassword = null;
	}

}