AddFileTypeDialog.java

Go to the documentation of this file.

/*
 * Autopsy Forensic Browser
 *
 * Copyright 2011-2017 Basis Technology Corp.
 * Contact: carrier <at> sleuthkit <dot> org
 *
 * 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 org.sleuthkit.autopsy.modules.filetypeid;

import java.awt.BorderLayout;
import java.awt.Dimension;
import java.awt.Toolkit;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.WindowAdapter;
import java.awt.event.WindowEvent;
import java.beans.PropertyChangeEvent;
import java.beans.PropertyChangeListener;
import javax.swing.BoxLayout;
import javax.swing.JButton;
import javax.swing.JDialog;
import javax.swing.JFrame;
import javax.swing.JPanel;
import org.openide.util.NbBundle;
import org.openide.util.NbBundle.Messages;
import org.openide.windows.WindowManager;

class AddFileTypeDialog extends JDialog {

    enum BUTTON_PRESSED {

        OK, CANCEL;
    }

    private static final long serialVersionUID = 1L;
    private FileType fileType;
    private AddFileTypePanel addMimeTypePanel;
    private BUTTON_PRESSED result;
    private JButton okButton;
    private JButton closeButton;

    @Messages({"AddMimeTypedialog.title=File Type"})
    public AddFileTypeDialog() {
        super(new JFrame(Bundle.AddMimeTypedialog_title()), Bundle.AddMimeTypedialog_title(), true);
        addMimeTypePanel = new AddFileTypePanel();
        this.display(true);
    }

    public AddFileTypeDialog(FileType fileType) {
        super(new JFrame(Bundle.AddMimeTypedialog_title()), Bundle.AddMimeTypedialog_title(), true);
        addMimeTypePanel = new AddFileTypePanel(fileType);
        this.display(false);
    }

    @NbBundle.Messages({
        "AddMimeTypeDialog.addButton.title=Add",
        "AddMimeTypeDialog.addButton.title2=Done",
        "AddMimeTypeDialog.cancelButton.title=Cancel"})
    void display(boolean add) {
        setLayout(new BorderLayout());

        setLocationRelativeTo(WindowManager.getDefault().getMainWindow());

        add(this.addMimeTypePanel, BorderLayout.PAGE_START);

        // Add the add/done button.
        if (add) {
            okButton = new JButton(Bundle.AddMimeTypeDialog_addButton_title());
        } else {
            okButton = new JButton(Bundle.AddMimeTypeDialog_addButton_title2());
        }
        okButton.addActionListener(new ActionListener() {
            @Override
            public void actionPerformed(ActionEvent e) {
                doButtonAction(true);
            }
        });

        // Add a close button.
        closeButton = new JButton(Bundle.AddMimeTypeDialog_cancelButton_title());
        closeButton.addActionListener(new ActionListener() {
            @Override
            public void actionPerformed(ActionEvent e) {
                doButtonAction(false);
            }
        });

        // Put the buttons in their own panel, under the settings panel.
        JPanel buttonPanel = new JPanel();
        buttonPanel.setLayout(new BoxLayout(buttonPanel, BoxLayout.LINE_AXIS));
        buttonPanel.add(new javax.swing.Box.Filler(new Dimension(10, 10), new Dimension(10, 10), new Dimension(10, 10)));
        buttonPanel.add(okButton);
        buttonPanel.add(new javax.swing.Box.Filler(new Dimension(10, 10), new Dimension(10, 10), new Dimension(10, 10)));
        buttonPanel.add(closeButton);
        add(buttonPanel, BorderLayout.LINE_START);

        this.addWindowListener(new WindowAdapter() {
            @Override
            public void windowClosing(WindowEvent e) {
                doButtonAction(false);
            }
        });
        this.addMimeTypePanel.addPropertyChangeListener(new PropertyChangeListener() {
            @Override
            public void propertyChange(PropertyChangeEvent evt) {
                if (evt.getPropertyName().equals(AddFileTypePanel.EVENT.SIG_LIST_CHANGED.toString())) {
                    enableOkButton();
                }
            }
        });
        enableOkButton();
        pack();
        setResizable(false);
        setVisible(true);

    }

    private void doButtonAction(boolean okPressed) {
        if (okPressed) {
            FileType fType = addMimeTypePanel.getFileType();
            if (fType != null) {
                this.fileType = fType;
                this.result = BUTTON_PRESSED.OK;
                setVisible(false);
            }
        } else {
            this.fileType = null;
            this.result = BUTTON_PRESSED.CANCEL;
            setVisible(false);
        }
    }

    public FileType getFileType() {
        return fileType;
    }

    public BUTTON_PRESSED getResult() {
        return result;
    }

    private void enableOkButton() {
        this.okButton.setEnabled(addMimeTypePanel.hasSignature());
    }

}