DomainDetailsPanel.java

Go to the documentation of this file.

/*
 * Autopsy
 *
 * Copyright 2020 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.discovery.ui;

import com.google.common.eventbus.Subscribe;
import java.awt.Component;
import javax.swing.JPanel;
import javax.swing.SwingUtilities;
import javax.swing.event.ChangeEvent;
import javax.swing.event.ChangeListener;
import org.apache.commons.lang.StringUtils;
import org.openide.util.NbBundle;
import org.sleuthkit.autopsy.coreutils.ThreadConfined;
import org.sleuthkit.autopsy.discovery.search.DiscoveryEventUtils;
import org.sleuthkit.datamodel.BlackboardArtifact;
import org.sleuthkit.autopsy.discovery.search.SearchData;

final class DomainDetailsPanel extends JPanel {

    private static final long serialVersionUID = 1L;
    private ArtifactsWorker singleArtifactDomainWorker;
    private String domain;
    private String selectedTabName = null;

    @ThreadConfined(type = ThreadConfined.ThreadType.AWT)
    DomainDetailsPanel() {
        initComponents();
        MiniTimelinePanel timelinePanel = new MiniTimelinePanel();
        DiscoveryEventUtils.getDiscoveryEventBus().register(timelinePanel);
        jTabbedPane1.add(Bundle.DomainDetailsPanel_miniTimelineTitle_text(), timelinePanel);
        for (BlackboardArtifact.ARTIFACT_TYPE type : SearchData.Type.DOMAIN.getArtifactTypes()) {
            jTabbedPane1.add(type.getDisplayName(), new DomainArtifactsTabPanel(type));
        }
    }

    @NbBundle.Messages({"DomainDetailsPanel.miniTimelineTitle.text=Timeline"})
    @ThreadConfined(type = ThreadConfined.ThreadType.AWT)
    void configureArtifactTabs(String tabName) {
        selectedTabName = tabName;
        if (StringUtils.isBlank(selectedTabName)) {
            selectedTabName = Bundle.DomainDetailsPanel_miniTimelineTitle_text();
        }
        selectTab();
        jTabbedPane1.addChangeListener(new ChangeListener() {
            @Override
            public void stateChanged(ChangeEvent e) {
                if (jTabbedPane1.getSelectedIndex() >= 0) {
                    String newTabTitle = jTabbedPane1.getTitleAt(jTabbedPane1.getSelectedIndex());
                    if (selectedTabName == null || !selectedTabName.equals(newTabTitle)) {
                        selectedTabName = newTabTitle;
                        Component selectedComponent = jTabbedPane1.getSelectedComponent();
                        if (selectedComponent instanceof DomainArtifactsTabPanel) {
                            runDomainWorker((DomainArtifactsTabPanel) selectedComponent);
                        } else if (selectedComponent instanceof MiniTimelinePanel) {
                            runMiniTimelineWorker((MiniTimelinePanel) selectedComponent);
                        }
                    }
                }
            }
        });
    }

    @ThreadConfined(type = ThreadConfined.ThreadType.AWT)
    private void selectTab() {
        for (int i = 0; i < jTabbedPane1.getTabCount(); i++) {
            if (!StringUtils.isBlank(selectedTabName) && selectedTabName.equals(jTabbedPane1.getTitleAt(i))) {
                jTabbedPane1.setSelectedIndex(i);
                return;
            }
        }
    }

    DomainArtifactsTabPanel.ArtifactRetrievalStatus getCurrentTabStatus() {
        if (jTabbedPane1.getSelectedComponent() instanceof MiniTimelinePanel) {
            return ((MiniTimelinePanel) jTabbedPane1.getSelectedComponent()).getStatus();
        } else if (jTabbedPane1.getSelectedComponent() instanceof DomainArtifactsTabPanel) {
            return ((DomainArtifactsTabPanel) jTabbedPane1.getSelectedComponent()).getStatus();
        }
        return null;
    }

    @ThreadConfined(type = ThreadConfined.ThreadType.AWT)
    private void runDomainWorker(DomainArtifactsTabPanel domainArtifactsTabPanel) {
        if (singleArtifactDomainWorker != null && !singleArtifactDomainWorker.isDone()) {
            singleArtifactDomainWorker.cancel(true);
        }
        if (domainArtifactsTabPanel.getStatus() == DomainArtifactsTabPanel.ArtifactRetrievalStatus.UNPOPULATED) {
            DiscoveryEventUtils.getDiscoveryEventBus().register(domainArtifactsTabPanel);
            domainArtifactsTabPanel.setStatus(DomainArtifactsTabPanel.ArtifactRetrievalStatus.POPULATING);
            singleArtifactDomainWorker = new ArtifactsWorker(domainArtifactsTabPanel.getArtifactType(), domain);
            singleArtifactDomainWorker.execute();
        }

    }

    private void runMiniTimelineWorker(MiniTimelinePanel miniTimelinePanel) {
        if (miniTimelinePanel.getStatus() == DomainArtifactsTabPanel.ArtifactRetrievalStatus.UNPOPULATED) {
            miniTimelinePanel.setStatus(DomainArtifactsTabPanel.ArtifactRetrievalStatus.POPULATING, domain);
            new MiniTimelineWorker(domain).execute();
        }
    }

    @Subscribe
    void handlePopulateDomainTabsEvent(DiscoveryEventUtils.PopulateDomainTabsEvent populateEvent) {
        SwingUtilities.invokeLater(() -> {
            if (StringUtils.isBlank(populateEvent.getDomain())) {
                resetTabsStatus();
                //send fade out event
                DiscoveryEventUtils.getDiscoveryEventBus().post(new DiscoveryEventUtils.DetailsVisibleEvent(false));
            } else {
                domain = populateEvent.getDomain();
                Component selectedComponent = jTabbedPane1.getSelectedComponent();
                if (selectedComponent instanceof DomainArtifactsTabPanel) {
                    runDomainWorker((DomainArtifactsTabPanel) selectedComponent);
                } else if (selectedComponent instanceof MiniTimelinePanel) {
                    runMiniTimelineWorker((MiniTimelinePanel) selectedComponent);
                }
                //send fade in event
                DiscoveryEventUtils.getDiscoveryEventBus().post(new DiscoveryEventUtils.DetailsVisibleEvent(true));
            }
        });
    }

    @ThreadConfined(type = ThreadConfined.ThreadType.AWT)
    private void resetTabsStatus() {
        for (Component comp : jTabbedPane1.getComponents()) {
            if (comp instanceof DomainArtifactsTabPanel) {
                ((DomainArtifactsTabPanel) comp).setStatus(DomainArtifactsTabPanel.ArtifactRetrievalStatus.UNPOPULATED);
            } else if (comp instanceof MiniTimelinePanel) {
                ((MiniTimelinePanel) comp).setStatus(DomainArtifactsTabPanel.ArtifactRetrievalStatus.UNPOPULATED, domain);
            }
        }
    }

    @ThreadConfined(type = ThreadConfined.ThreadType.AWT)
    String getSelectedTabName() {
        return selectedTabName;
    }

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

        jTabbedPane1 = new javax.swing.JTabbedPane();

        setEnabled(false);
        setMinimumSize(new java.awt.Dimension(0, 0));
        setPreferredSize(new java.awt.Dimension(0, 0));
        setLayout(new java.awt.BorderLayout());

        jTabbedPane1.setMinimumSize(new java.awt.Dimension(0, 0));
        jTabbedPane1.setPreferredSize(new java.awt.Dimension(0, 0));
        add(jTabbedPane1, java.awt.BorderLayout.CENTER);
    }// </editor-fold>//GEN-END:initComponents

    // Variables declaration - do not modify//GEN-BEGIN:variables
    private javax.swing.JTabbedPane jTabbedPane1;
    // End of variables declaration//GEN-END:variables

    /*
     * Unregister the MiniTimelinePanel from the event bus.
     */
    void unregister() {
        for (Component comp : jTabbedPane1.getComponents()) {
            if (comp instanceof MiniTimelinePanel) {
                DiscoveryEventUtils.getDiscoveryEventBus().unregister(comp);
            }
        }
    }
}