DataSourceSummaryNode.java

Go to the documentation of this file.

/*
 * Autopsy Forensic Browser
 *
 * Copyright 2019-2021 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.event.ActionEvent;
import java.util.ArrayList;
import java.util.List;
import java.util.Observable;
import java.util.Observer;
import javax.swing.Action;
import org.openide.nodes.AbstractNode;
import org.openide.nodes.ChildFactory;
import org.openide.nodes.Children;
import org.openide.nodes.Node;
import org.openide.nodes.Sheet;
import org.openide.util.NbBundle.Messages;
import org.sleuthkit.autopsy.datamodel.NodeProperty;
import org.sleuthkit.autopsy.directorytree.ViewContextAction;
import org.sleuthkit.datamodel.DataSource;

final class DataSourceSummaryNode extends AbstractNode {

    private final static Observable closeDialogObservable = new Observable() {
        @Override
        public void notifyObservers() {
            //set changed before notify observers
            //we want this observerable to always cause the observer to update when notified
            this.setChanged();
            super.notifyObservers();
        }
    };

    DataSourceSummaryNode(List<DataSourceSummary> dataSourceList) {
        super(Children.create(new DataSourceSummaryChildren(dataSourceList), true));
    }

    void addObserver(Observer observer) {
        closeDialogObservable.addObserver(observer);
    }

    static final class DataSourceSummaryChildren extends ChildFactory<DataSourceSummary> {

        private final List<DataSourceSummary> dataSourceList;

        DataSourceSummaryChildren(List<DataSourceSummary> dataSourceList) {
            this.dataSourceList = dataSourceList;
        }

        @Override
        protected boolean createKeys(List<DataSourceSummary> list) {
            if (dataSourceList != null && !dataSourceList.isEmpty()) {
                list.addAll(dataSourceList);
            }
            return true;
        }

        @Override
        protected Node createNodeForKey(DataSourceSummary key) {
            return new DataSourceSummaryEntryNode(key);
        }
    }

    static final class DataSourceSummaryEntryNode extends AbstractNode {

        private final DataSource dataSource;
        private final String status;
        private final String type;
        private final long filesCount;
        private final long resultsCount;
        private final long tagsCount;

        DataSourceSummaryEntryNode(DataSourceSummary dataSourceSummary) {
            super(Children.LEAF);
            dataSource = dataSourceSummary.getDataSource();
            status = dataSourceSummary.getIngestStatus() == null ? "" : dataSourceSummary.getIngestStatus().getDisplayName();
            type = dataSourceSummary.getType();
            filesCount = dataSourceSummary.getFilesCount();
            resultsCount = dataSourceSummary.getResultsCount();
            tagsCount = dataSourceSummary.getTagsCount();
            super.setName(dataSource.getName());
            setName(dataSource.getName());
            setDisplayName(dataSource.getName());
        }

        DataSource getDataSource() {
            return dataSource;
        }

        @Messages({"DataSourceSummaryNode.column.dataSourceName.header=Data Source Name",
            "DataSourceSummaryNode.column.status.header=Ingest Status",
            "DataSourceSummaryNode.column.type.header=Type",
            "DataSourceSummaryNode.column.files.header=Files",
            "DataSourceSummaryNode.column.results.header=Artifacts",
            "DataSourceSummaryNode.column.tags.header=Tags"})
        @Override
        protected Sheet createSheet() {
            Sheet sheet = super.createSheet();
            Sheet.Set sheetSet = sheet.get(Sheet.PROPERTIES);
            if (sheetSet == null) {
                sheetSet = Sheet.createPropertiesSet();
                sheet.put(sheetSet);
            }
            sheetSet.put(new NodeProperty<>(Bundle.DataSourceSummaryNode_column_dataSourceName_header(), Bundle.DataSourceSummaryNode_column_dataSourceName_header(), Bundle.DataSourceSummaryNode_column_dataSourceName_header(),
                    dataSource));
            sheetSet.put(new NodeProperty<>(Bundle.DataSourceSummaryNode_column_status_header(), Bundle.DataSourceSummaryNode_column_status_header(), Bundle.DataSourceSummaryNode_column_status_header(), status));
            sheetSet.put(new NodeProperty<>(Bundle.DataSourceSummaryNode_column_type_header(), Bundle.DataSourceSummaryNode_column_type_header(), Bundle.DataSourceSummaryNode_column_type_header(),
                    type));
            sheetSet.put(new NodeProperty<>(Bundle.DataSourceSummaryNode_column_files_header(), Bundle.DataSourceSummaryNode_column_files_header(), Bundle.DataSourceSummaryNode_column_files_header(),
                    filesCount));
            sheetSet.put(new NodeProperty<>(Bundle.DataSourceSummaryNode_column_results_header(), Bundle.DataSourceSummaryNode_column_results_header(), Bundle.DataSourceSummaryNode_column_results_header(),
                    resultsCount));
            sheetSet.put(new NodeProperty<>(Bundle.DataSourceSummaryNode_column_tags_header(), Bundle.DataSourceSummaryNode_column_tags_header(), Bundle.DataSourceSummaryNode_column_tags_header(),
                    tagsCount));
            return sheet;
        }

        @Override
        public Action getPreferredAction() {
            return new ViewDataSourceInContextAction();
        }

        @Override
        public Action[] getActions(boolean context) {
            List<Action> actions = new ArrayList<>();
            actions.add(new ViewDataSourceInContextAction());
            return actions.toArray(new Action[actions.size()]);
        }

        private class ViewDataSourceInContextAction extends ViewContextAction {

            private static final long serialVersionUID = 1L;

            @Messages({"DataSourceSummaryNode.viewDataSourceAction.text=Go to Data Source"})
            ViewDataSourceInContextAction() {
                super(Bundle.DataSourceSummaryNode_viewDataSourceAction_text(), dataSource);
            }

            @Override
            public void actionPerformed(ActionEvent e) {
                closeDialogObservable.notifyObservers();
                super.actionPerformed(e);
            }
        }
    }

}