DataSourceSummaryDialog.java

Go to the documentation of this file.

/*
 * Autopsy Forensic Browser
 *
 * Copyright 2019 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.datasourcesummary.ui;

import java.awt.Frame;
import java.beans.PropertyChangeEvent;
import java.beans.PropertyChangeListener;
import java.util.EnumSet;
import java.util.Map;
import java.util.Observable;
import java.util.Observer;
import java.util.Set;
import javax.swing.WindowConstants;
import javax.swing.event.ListSelectionEvent;
import org.openide.util.NbBundle.Messages;
import org.sleuthkit.autopsy.datasourcesummary.datamodel.CaseDataSourcesSummary;
import org.sleuthkit.autopsy.ingest.IngestManager;
import org.sleuthkit.autopsy.ingest.events.DataSourceAnalysisCompletedEvent;
import org.sleuthkit.autopsy.ingest.events.DataSourceAnalysisCompletedEvent.Reason;
import org.sleuthkit.datamodel.DataSource;
import org.sleuthkit.datamodel.IngestJobInfo;

final class DataSourceSummaryDialog extends javax.swing.JDialog implements Observer {

    private static final long serialVersionUID = 1L;
    private static final Set<IngestManager.IngestJobEvent> INGEST_JOB_EVENTS_OF_INTEREST = EnumSet.of(IngestManager.IngestJobEvent.DATA_SOURCE_ANALYSIS_COMPLETED);
    private final DataSourceBrowser dataSourcesPanel;
    private final DataSourceSummaryTabbedPane dataSourceSummaryTabbedPane;
    private final PropertyChangeListener ingestEventListener;

    @Messages({
        "DataSourceSummaryDialog.window.title=Data Sources Summary"
    })
    DataSourceSummaryDialog(Frame owner) {
        super(owner, Bundle.DataSourceSummaryDialog_window_title(), true);
        Map<Long, String> usageMap = CaseDataSourcesSummary.getDataSourceTypes();
        Map<Long, Long> fileCountsMap = CaseDataSourcesSummary.getCountsOfFiles();
        dataSourcesPanel = new DataSourceBrowser(usageMap, fileCountsMap);
        dataSourceSummaryTabbedPane = new DataSourceSummaryTabbedPane();
        dataSourceSummaryTabbedPane.setParentCloseListener(() -> DataSourceSummaryDialog.this.dispose());
        initComponents();
        dataSourceSummarySplitPane.setLeftComponent(dataSourcesPanel);
        dataSourcesPanel.addListSelectionListener((ListSelectionEvent e) -> {
            if (!e.getValueIsAdjusting()) {
                DataSource selectedDataSource = dataSourcesPanel.getSelectedDataSource();
                dataSourceSummaryTabbedPane.setDataSource(selectedDataSource);
                this.repaint();
            }
        });

        ingestEventListener = (PropertyChangeEvent evt) -> {
            if (evt instanceof DataSourceAnalysisCompletedEvent) {
                DataSourceAnalysisCompletedEvent dsEvent = (DataSourceAnalysisCompletedEvent) evt;
                if (dsEvent.getResult() == Reason.ANALYSIS_COMPLETED) {
                    dataSourcesPanel.refresh(dsEvent.getDataSource().getId(), IngestJobInfo.IngestJobStatusType.COMPLETED);
                } else if (dsEvent.getResult() == Reason.ANALYSIS_CANCELLED) {
                    dataSourcesPanel.refresh(dsEvent.getDataSource().getId(), null);
                }
            }
        };

        //add listener to refresh jobs with Started status when they complete
        IngestManager.getInstance().addIngestJobEventListener(INGEST_JOB_EVENTS_OF_INTEREST, ingestEventListener);

        // verify that dialog will call dispose on close:
        // https://docs.oracle.com/javase/tutorial/uiswing/components/frame.html#windowevents
        setDefaultCloseOperation(WindowConstants.DISPOSE_ON_CLOSE);
        this.pack();
    }

    @Override
    public void dispose() {
        IngestManager.getInstance().removeIngestJobEventListener(INGEST_JOB_EVENTS_OF_INTEREST, ingestEventListener);
        this.dataSourceSummaryTabbedPane.close();
        super.dispose();
    }

    void enableObserver() {
        dataSourcesPanel.addObserver(this);
    }

    @Override
    public void update(Observable o, Object arg) {
        this.dispose();
    }

    @SuppressWarnings("unchecked")
    // <editor-fold defaultstate="collapsed" desc="Generated Code">//GEN-BEGIN:initComponents
    private void initComponents() {

        closeButton = new javax.swing.JButton();
        dataSourceSummarySplitPane = new javax.swing.JSplitPane();
        javax.swing.JPanel dataSourceTabbedPane = dataSourceSummaryTabbedPane;

        org.openide.awt.Mnemonics.setLocalizedText(closeButton, org.openide.util.NbBundle.getMessage(DataSourceSummaryDialog.class, "DataSourceSummaryDialog.closeButton.text")); // NOI18N
        closeButton.addActionListener(new java.awt.event.ActionListener() {
            public void actionPerformed(java.awt.event.ActionEvent evt) {
                closeButtonActionPerformed(evt);
            }
        });

        dataSourceSummarySplitPane.setDividerLocation(130);
        dataSourceSummarySplitPane.setOrientation(javax.swing.JSplitPane.VERTICAL_SPLIT);
        dataSourceSummarySplitPane.setRightComponent(dataSourceTabbedPane);

        dataSourceSummarySplitPane.setLeftComponent(dataSourcesPanel);

        javax.swing.GroupLayout layout = new javax.swing.GroupLayout(getContentPane());
        getContentPane().setLayout(layout);
        layout.setHorizontalGroup(
            layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
            .addGroup(javax.swing.GroupLayout.Alignment.TRAILING, layout.createSequentialGroup()
                .addContainerGap()
                .addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.TRAILING)
                    .addComponent(dataSourceSummarySplitPane, javax.swing.GroupLayout.DEFAULT_SIZE, 860, Short.MAX_VALUE)
                    .addGroup(layout.createSequentialGroup()
                        .addGap(0, 0, Short.MAX_VALUE)
                        .addComponent(closeButton)))
                .addContainerGap())
        );
        layout.setVerticalGroup(
            layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
            .addGroup(layout.createSequentialGroup()
                .addContainerGap()
                .addComponent(dataSourceSummarySplitPane, javax.swing.GroupLayout.DEFAULT_SIZE, 540, Short.MAX_VALUE)
                .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
                .addComponent(closeButton)
                .addContainerGap())
        );
    }// </editor-fold>//GEN-END:initComponents

    private void closeButtonActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_closeButtonActionPerformed
        this.dispose();
    }//GEN-LAST:event_closeButtonActionPerformed

    void selectDataSource(Long dataSourceId) {
        dataSourcesPanel.selectDataSource(dataSourceId);
    }

    // Variables declaration - do not modify//GEN-BEGIN:variables
    private javax.swing.JButton closeButton;
    private javax.swing.JSplitPane dataSourceSummarySplitPane;
    // End of variables declaration//GEN-END:variables
}