IngestJobInfoPanel.java

Go to the documentation of this file.

/*
 * Autopsy Forensic Browser
 *
 * Copyright 2016-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.casemodule;

import java.beans.PropertyChangeEvent;
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.EnumSet;
import java.util.List;
import java.util.Set;
import java.util.logging.Level;
import javax.swing.JOptionPane;
import javax.swing.event.ListSelectionEvent;
import javax.swing.table.AbstractTableModel;
import org.openide.util.NbBundle.Messages;
import static org.sleuthkit.autopsy.casemodule.Case.Events.CURRENT_CASE;
import org.sleuthkit.autopsy.coreutils.Logger;
import org.sleuthkit.autopsy.events.AutopsyEvent;
import org.sleuthkit.autopsy.ingest.IngestManager;
import org.sleuthkit.datamodel.IngestJobInfo;
import org.sleuthkit.datamodel.IngestModuleInfo;
import org.sleuthkit.datamodel.SleuthkitCase;
import org.sleuthkit.datamodel.TskCoreException;
import org.sleuthkit.datamodel.DataSource;

@SuppressWarnings("PMD.SingularField") // UI widgets cause lots of false positives
public final class IngestJobInfoPanel extends javax.swing.JPanel {

    private static final Logger logger = Logger.getLogger(IngestJobInfoPanel.class.getName());
    private static final Set<IngestManager.IngestJobEvent> INGEST_JOB_EVENTS_OF_INTEREST = EnumSet.of(IngestManager.IngestJobEvent.STARTED, IngestManager.IngestJobEvent.CANCELLED, IngestManager.IngestJobEvent.COMPLETED);
    private static final Set<Case.Events> CASE_EVENTS_OF_INTEREST = EnumSet.of(Case.Events.CURRENT_CASE);

    private List<IngestJobInfo> ingestJobs;
    private final List<IngestJobInfo> ingestJobsForSelectedDataSource = new ArrayList<>();
    private IngestJobTableModel ingestJobTableModel = new IngestJobTableModel();
    private IngestModuleTableModel ingestModuleTableModel = new IngestModuleTableModel(null);
    private final DateFormat datetimeFormat = new SimpleDateFormat("yyyy/MM/dd HH:mm:ss");
    private DataSource selectedDataSource;

    public IngestJobInfoPanel() {
        initComponents();
        customizeComponents();
    }

    @Messages({"IngestJobInfoPanel.loadIngestJob.error.text=Failed to load ingest jobs.",
        "IngestJobInfoPanel.loadIngestJob.error.title=Load Failure"})
    private void customizeComponents() {
        refresh();
        this.ingestJobTable.getSelectionModel().addListSelectionListener((ListSelectionEvent e) -> {
            IngestJobInfo currJob = (ingestJobTable.getSelectedRow() < 0 ? null : this.ingestJobsForSelectedDataSource.get(ingestJobTable.getSelectedRow()));
            this.ingestModuleTableModel = new IngestModuleTableModel(currJob);
            this.ingestModuleTable.setModel(this.ingestModuleTableModel);
        });

        IngestManager.getInstance().addIngestJobEventListener(INGEST_JOB_EVENTS_OF_INTEREST , (PropertyChangeEvent evt) -> {
            if (evt.getPropertyName().equals(IngestManager.IngestJobEvent.STARTED.toString())
                    || evt.getPropertyName().equals(IngestManager.IngestJobEvent.CANCELLED.toString())
                    || evt.getPropertyName().equals(IngestManager.IngestJobEvent.COMPLETED.toString())) {
                refresh();
            }
        });
        
        Case.addEventTypeSubscriber(CASE_EVENTS_OF_INTEREST, (PropertyChangeEvent evt) -> {
            if (!(evt instanceof AutopsyEvent) || (((AutopsyEvent) evt).getSourceType() != AutopsyEvent.SourceType.LOCAL)) {
                return;
            }
                    
            if (CURRENT_CASE == Case.Events.valueOf(evt.getPropertyName())) {
                refresh();
            }
        });
    }

    public void setDataSource(DataSource selectedDataSource) {
        this.selectedDataSource = selectedDataSource;
        ingestJobsForSelectedDataSource.clear();
        if (selectedDataSource != null) {
            for (IngestJobInfo jobInfo : ingestJobs) {
                if (selectedDataSource.getId() == jobInfo.getObjectId()) {
                    ingestJobsForSelectedDataSource.add(jobInfo);
                }
            }
        }
        this.ingestJobTableModel = new IngestJobTableModel();
        this.ingestJobTable.setModel(ingestJobTableModel);
        //if there were ingest jobs select the first one by default
        if (!ingestJobsForSelectedDataSource.isEmpty()) {
            ingestJobTable.setRowSelectionInterval(0, 0);
        }
        this.repaint();
    }

    private void refresh() {
        try {
            if (Case.isCaseOpen()) {
                SleuthkitCase skCase = Case.getCurrentCaseThrows().getSleuthkitCase();
                this.ingestJobs = skCase.getIngestJobs();
                setDataSource(selectedDataSource);
            } else {
                this.ingestJobs = new ArrayList<>();
                setDataSource(null);
            }
            
        } catch (TskCoreException | NoCurrentCaseException ex) {
            logger.log(Level.SEVERE, "Failed to load ingest jobs.", ex);
            JOptionPane.showMessageDialog(this, Bundle.IngestJobInfoPanel_loadIngestJob_error_text(), Bundle.IngestJobInfoPanel_loadIngestJob_error_title(), JOptionPane.ERROR_MESSAGE);
        }
    }

    @Messages({"IngestJobInfoPanel.IngestJobTableModel.StartTime.header=Start Time",
        "IngestJobInfoPanel.IngestJobTableModel.EndTime.header=End Time",
        "IngestJobInfoPanel.IngestJobTableModel.IngestStatus.header=Ingest Status"})
    private class IngestJobTableModel extends AbstractTableModel {

        private final List<String> columnHeaders = new ArrayList<>();

        IngestJobTableModel() {
            columnHeaders.add(Bundle.IngestJobInfoPanel_IngestJobTableModel_StartTime_header());
            columnHeaders.add(Bundle.IngestJobInfoPanel_IngestJobTableModel_EndTime_header());
            columnHeaders.add(Bundle.IngestJobInfoPanel_IngestJobTableModel_IngestStatus_header());
        }

        @Override
        public int getRowCount() {
            return ingestJobsForSelectedDataSource.size();
        }

        @Override
        public int getColumnCount() {
            return columnHeaders.size();
        }

        @Override
        public Object getValueAt(int rowIndex, int columnIndex) {
            IngestJobInfo currIngestJob = ingestJobsForSelectedDataSource.get(rowIndex);
            if (columnIndex == 0) {
                return datetimeFormat.format(currIngestJob.getStartDateTime());
            } else if (columnIndex == 1) {
                Date endDate = currIngestJob.getEndDateTime();
                if (endDate.getTime() == 0) {
                    return "N/A";
                }
                return datetimeFormat.format(currIngestJob.getEndDateTime());
            } else if (columnIndex == 2) {
                return currIngestJob.getStatus().getDisplayName();
            }
            return null;
        }

        @Override
        public String getColumnName(int column) {
            return columnHeaders.get(column);
        }

    }

    @Messages({"IngestJobInfoPanel.IngestModuleTableModel.ModuleName.header=Module Name",
        "IngestJobInfoPanel.IngestModuleTableModel.ModuleVersion.header=Module Version"})
    private class IngestModuleTableModel extends AbstractTableModel {

        private final List<String> columnHeaders = new ArrayList<>();
        private final IngestJobInfo currJob;

        IngestModuleTableModel(IngestJobInfo currJob) {
            columnHeaders.add(Bundle.IngestJobInfoPanel_IngestModuleTableModel_ModuleName_header());
            columnHeaders.add(Bundle.IngestJobInfoPanel_IngestModuleTableModel_ModuleVersion_header());
            this.currJob = currJob;
        }

        @Override
        public int getRowCount() {
            if (currJob == null) {
                return 0;
            }
            return currJob.getIngestModuleInfo().size();
        }

        @Override
        public int getColumnCount() {
            return columnHeaders.size();
        }

        @Override
        public Object getValueAt(int rowIndex, int columnIndex) {
            if (currJob != null) {
                IngestModuleInfo currIngestModule = currJob.getIngestModuleInfo().get(rowIndex);
                if (columnIndex == 0) {
                    return currIngestModule.getDisplayName();
                } else if (columnIndex == 1) {
                    return currIngestModule.getVersion();
                }
                return null;
            }
            return null;
        }

        @Override
        public String getColumnName(int column) {
            return columnHeaders.get(column);
        }

    }

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

        jScrollPane1 = new javax.swing.JScrollPane();
        ingestJobTable = new javax.swing.JTable();
        jLabel1 = new javax.swing.JLabel();
        jLabel2 = new javax.swing.JLabel();
        jScrollPane2 = new javax.swing.JScrollPane();
        ingestModuleTable = new javax.swing.JTable();

        jScrollPane1.setBorder(null);

        ingestJobTable.setModel(ingestJobTableModel);
        ingestJobTable.getTableHeader().setReorderingAllowed(false);
        jScrollPane1.setViewportView(ingestJobTable);
        ingestJobTable.getColumnModel().getSelectionModel().setSelectionMode(javax.swing.ListSelectionModel.SINGLE_SELECTION);

        org.openide.awt.Mnemonics.setLocalizedText(jLabel1, org.openide.util.NbBundle.getMessage(IngestJobInfoPanel.class, "IngestJobInfoPanel.jLabel1.text")); // NOI18N

        org.openide.awt.Mnemonics.setLocalizedText(jLabel2, org.openide.util.NbBundle.getMessage(IngestJobInfoPanel.class, "IngestJobInfoPanel.jLabel2.text")); // NOI18N

        ingestModuleTable.setModel(ingestModuleTableModel);
        jScrollPane2.setViewportView(ingestModuleTable);

        javax.swing.GroupLayout layout = new javax.swing.GroupLayout(this);
        this.setLayout(layout);
        layout.setHorizontalGroup(
            layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
            .addGroup(layout.createSequentialGroup()
                .addGap(15, 15, 15)
                .addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
                    .addComponent(jLabel2)
                    .addComponent(jScrollPane1, javax.swing.GroupLayout.DEFAULT_SIZE, 485, Short.MAX_VALUE))
                .addGap(8, 8, 8)
                .addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
                    .addComponent(jScrollPane2, javax.swing.GroupLayout.PREFERRED_SIZE, 254, javax.swing.GroupLayout.PREFERRED_SIZE)
                    .addComponent(jLabel1))
                .addContainerGap())
        );
        layout.setVerticalGroup(
            layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
            .addGroup(layout.createSequentialGroup()
                .addGap(8, 8, 8)
                .addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
                    .addComponent(jLabel1)
                    .addComponent(jLabel2))
                .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
                .addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
                    .addComponent(jScrollPane1, javax.swing.GroupLayout.DEFAULT_SIZE, 162, Short.MAX_VALUE)
                    .addComponent(jScrollPane2, javax.swing.GroupLayout.PREFERRED_SIZE, 0, Short.MAX_VALUE))
                .addGap(10, 10, 10))
        );
    }// </editor-fold>//GEN-END:initComponents


    // Variables declaration - do not modify//GEN-BEGIN:variables
    private javax.swing.JTable ingestJobTable;
    private javax.swing.JTable ingestModuleTable;
    private javax.swing.JLabel jLabel1;
    private javax.swing.JLabel jLabel2;
    private javax.swing.JScrollPane jScrollPane1;
    private javax.swing.JScrollPane jScrollPane2;
    // End of variables declaration//GEN-END:variables
}