EventsTree.java

Go to the documentation of this file.

/*
 * Autopsy Forensic Browser
 *
 * Copyright 2013-15 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.timeline.ui.detailview.tree;

import com.google.common.collect.ImmutableList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Objects;
import javafx.beans.InvalidationListener;
import javafx.beans.Observable;
import javafx.beans.property.SimpleBooleanProperty;
import javafx.collections.ListChangeListener;
import javafx.fxml.FXML;
import javafx.scene.control.ComboBox;
import javafx.scene.control.Label;
import javafx.scene.control.ListCell;
import javafx.scene.control.SelectionMode;
import javafx.scene.control.Tooltip;
import javafx.scene.control.TreeCell;
import javafx.scene.control.TreeItem;
import javafx.scene.control.TreeView;
import javafx.scene.image.ImageView;
import javafx.scene.input.MouseButton;
import javafx.scene.input.MouseEvent;
import javafx.scene.layout.BorderPane;
import javafx.scene.layout.StackPane;
import javafx.scene.paint.Color;
import javafx.scene.shape.Rectangle;
import org.apache.commons.lang3.StringUtils;
import org.controlsfx.control.action.ActionUtils;
import org.openide.util.NbBundle;
import org.sleuthkit.autopsy.coreutils.ThreadConfined;
import org.sleuthkit.autopsy.timeline.FXMLConstructor;
import org.sleuthkit.autopsy.timeline.TimeLineController;
import org.sleuthkit.autopsy.timeline.datamodel.EventBundle;
import org.sleuthkit.autopsy.timeline.filters.AbstractFilter;
import org.sleuthkit.autopsy.timeline.filters.DescriptionFilter;
import org.sleuthkit.autopsy.timeline.ui.detailview.DetailViewPane;

final public class EventsTree extends BorderPane {

    private final TimeLineController controller;

    private DetailViewPane detailViewPane;

    @FXML
    private TreeView<EventBundle<?>> eventsTree;

    @FXML
    private Label eventsTreeLabel;

    @FXML
    private ComboBox<TreeComparator> sortByBox;

    public EventsTree(TimeLineController controller) {
        this.controller = controller;

        FXMLConstructor.construct(this, "EventsTree.fxml"); // NON-NLS 
    }

    public void setDetailViewPane(DetailViewPane detailViewPane) {
        this.detailViewPane = detailViewPane;
        detailViewPane.setSelectionModel(eventsTree.getSelectionModel());

        detailViewPane.getEventStripes().addListener((ListChangeListener.Change<? extends EventBundle<?>> c) -> {
            //on jfx thread
            while (c.next()) {
                for (EventBundle<?> bundle : c.getAddedSubList()) {
                    getRoot().insert(bundle);
                }
                for (EventBundle<?> bundle : c.getRemoved()) {
                    getRoot().remove(bundle);
                }
            }
        });

        setRoot();

        detailViewPane.getSelectedNodes().addListener((Observable observable) -> {
            eventsTree.getSelectionModel().clearSelection();
            detailViewPane.getSelectedNodes().forEach(eventBundleNode -> {
                eventsTree.getSelectionModel().select(getRoot().findTreeItemForEvent(eventBundleNode.getEventBundle()));
            });
        });

    }

    private RootItem getRoot() {
        return (RootItem) eventsTree.getRoot();
    }

    @ThreadConfined(type = ThreadConfined.ThreadType.JFX)
    private void setRoot() {
        RootItem root = new RootItem(TreeComparator.Type.reversed().thenComparing(sortByBox.getSelectionModel().getSelectedItem()));
        for (EventBundle<?> bundle : detailViewPane.getEventStripes()) {
            root.insert(bundle);
        }
        eventsTree.setRoot(root);

    }

    @FXML
    @NbBundle.Messages("EventsTree.Label.text=Sort By:")
    void initialize() {
        assert sortByBox != null : "fx:id=\"sortByBox\" was not injected: check your FXML file 'NavPanel.fxml'."; // NON-NLS

        sortByBox.getItems().setAll(Arrays.asList(TreeComparator.Description, TreeComparator.Count));
        sortByBox.getSelectionModel().select(TreeComparator.Description);
        sortByBox.setCellFactory(listView -> new TreeComparatorCell());
        sortByBox.setButtonCell(new TreeComparatorCell());
        sortByBox.getSelectionModel().selectedItemProperty().addListener((Observable o) -> {
            getRoot().resort(TreeComparator.Type.reversed().thenComparing(sortByBox.getSelectionModel().getSelectedItem()), true);
        });
        eventsTree.setShowRoot(false);

        eventsTree.setCellFactory(treeView -> new EventBundleTreeCell());
        eventsTree.getSelectionModel().setSelectionMode(SelectionMode.MULTIPLE);

        eventsTreeLabel.setText(Bundle.EventsTree_Label_text());
    }

    private class EventBundleTreeCell extends TreeCell<EventBundle<?>> {

        private static final double HIDDEN_MULTIPLIER = .6;
        private final Rectangle rect = new Rectangle(24, 24);
        private final ImageView imageView = new ImageView();
        private InvalidationListener filterStateChangeListener;
        SimpleBooleanProperty hidden = new SimpleBooleanProperty(false);

        EventBundleTreeCell() {
            rect.setArcHeight(5);
            rect.setArcWidth(5);
            rect.setStrokeWidth(2);
        }

        @Override
        protected void updateItem(EventBundle<?> item, boolean empty) {
            super.updateItem(item, empty);
            if (item == null || empty) {
                setText(null);
                setTooltip(null);
                setGraphic(null);
                setContextMenu(null);
                deRegisterListeners(controller.getQuickHideFilters());
            } else {
                filterStateChangeListener = (filterState) -> updateHiddenState(item);
                controller.getQuickHideFilters().addListener((ListChangeListener.Change<? extends DescriptionFilter> listChange) -> {
                    while (listChange.next()) {
                        deRegisterListeners(listChange.getRemoved());
                        registerListeners(listChange.getAddedSubList(), item);
                    }
                    updateHiddenState(item);
                });
                registerListeners(controller.getQuickHideFilters(), item);
                String text;
                if (getTreeItem() instanceof EventTypeTreeItem) {
                    text = item.getEventType().getDisplayName();
                    setDisable(true);
                } else {
                    setDisable(false);
                    text = item.getDescription() + " (" + item.getCount() + ")"; // NON-NLS
                    TreeItem<EventBundle<?>> parent = getTreeItem().getParent();
                    if (parent != null && parent.getValue() != null && (parent instanceof EventDescriptionTreeItem)) {
                        text = StringUtils.substringAfter(text, parent.getValue().getDescription());
                    }
                }
                setText(text);
                setTooltip(new Tooltip(text));
                imageView.setImage(item.getEventType().getFXImage());
                setGraphic(new StackPane(rect, imageView));
                updateHiddenState(item);
                if (getTreeItem() instanceof EventDescriptionTreeItem) {
                    setOnMouseClicked((MouseEvent event) -> {
                        if (event.getButton() == MouseButton.SECONDARY) {
                            if (hidden.get()) {
                                ActionUtils.createContextMenu(ImmutableList.of(detailViewPane.newUnhideDescriptionAction(item.getDescription(), item.getDescriptionLoD())))
                                        .show(EventBundleTreeCell.this, event.getScreenX(), event.getScreenY());
                            } else {
                                ActionUtils.createContextMenu(ImmutableList.of(detailViewPane.newHideDescriptionAction(item.getDescription(), item.getDescriptionLoD())))
                                        .show(EventBundleTreeCell.this, event.getScreenX(), event.getScreenY());
                            }
                        }
                    });
                } else {
                    setOnMouseClicked(null);
                }
            }
        }

        private void registerListeners(Collection<? extends DescriptionFilter> filters, EventBundle<?> item) {
            for (DescriptionFilter filter : filters) {
                if (filter.getDescription().equals(item.getDescription())) {
                    filter.activeProperty().addListener(filterStateChangeListener);
                }
            }
        }

        private void deRegisterListeners(Collection<? extends DescriptionFilter> filters) {
            if (Objects.nonNull(filterStateChangeListener)) {
                for (DescriptionFilter filter : filters) {
                    filter.activeProperty().removeListener(filterStateChangeListener);
                }
            }
        }

        private void updateHiddenState(EventBundle<?> item) {
            TreeItem<EventBundle<?>> treeItem = getTreeItem();

            hidden.set(controller.getQuickHideFilters().stream().
                    filter(AbstractFilter::isActive)
                    .anyMatch(filter -> filter.getDescription().equals(item.getDescription())));
            if (hidden.get()) {
                if (treeItem != null) {
                    treeItem.setExpanded(false);
                }
                setTextFill(Color.gray(0, HIDDEN_MULTIPLIER));
                imageView.setOpacity(HIDDEN_MULTIPLIER);
                rect.setStroke(item.getEventType().getColor().deriveColor(0, HIDDEN_MULTIPLIER, 1, HIDDEN_MULTIPLIER));
                rect.setFill(item.getEventType().getColor().deriveColor(0, HIDDEN_MULTIPLIER, HIDDEN_MULTIPLIER, 0.1));
            } else {
                setTextFill(Color.BLACK);
                imageView.setOpacity(1);
                rect.setStroke(item.getEventType().getColor());
                rect.setFill(item.getEventType().getColor().deriveColor(0, 1, 1, 0.1));
            }
        }
    }

    static private class TreeComparatorCell extends ListCell<TreeComparator> {

        @Override
        protected void updateItem(TreeComparator item, boolean empty) {
            super.updateItem(item, empty);
            if (empty || item == null) {
                setText(null);
            } else {
                setText(item.getDisplayName());
            }
        }
    }
}