19 package org.sleuthkit.autopsy.directorytree;
21 import java.util.ArrayList;
22 import java.util.Collections;
23 import java.util.List;
24 import java.util.logging.Level;
25 import javax.swing.Action;
26 import org.openide.nodes.FilterNode;
27 import org.openide.nodes.Node;
28 import org.openide.util.NbBundle;
29 import org.openide.util.lookup.Lookups;
30 import org.openide.util.lookup.ProxyLookup;
48 class DirectoryTreeFilterNode
extends FilterNode {
50 private static final Logger logger = Logger.getLogger(DirectoryTreeFilterNode.class.getName());
51 private static final Action collapseAllAction =
new CollapseAction(NbBundle.getMessage(DirectoryTreeFilterNode.class,
"DirectoryTreeFilterNode.action.collapseAll.text"));
62 DirectoryTreeFilterNode(Node nodeToWrap,
boolean createChildren) {
64 DirectoryTreeFilterChildren.createInstance(nodeToWrap, createChildren),
65 new ProxyLookup(Lookups.singleton(
new OriginalNode(nodeToWrap)), nodeToWrap.getLookup()));
75 public String getDisplayName() {
76 final Node orig = getOriginal();
77 String name = orig.getDisplayName();
78 if (orig instanceof AbstractContentNode) {
79 AbstractFile file = getLookup().lookup(AbstractFile.class);
82 int numVisibleChildren = getVisibleChildCount(file);
89 name = name +
" \u200E(\u200E" + numVisibleChildren +
")\u200E";
91 }
catch (TskCoreException ex) {
92 logger.log(Level.SEVERE,
"Error getting children count to display for file: " + file, ex);
108 private int getVisibleChildCount(AbstractFile file)
throws TskCoreException {
109 List<Content> childList = file.getChildren();
111 int numVisibleChildren = file.getChildrenCount();
112 boolean purgeKnownFiles = UserPreferences.hideKnownFilesInDataSourcesTree();
113 boolean purgeSlackFiles = UserPreferences.hideSlackFilesInDataSourcesTree();
115 if(purgeKnownFiles || purgeSlackFiles) {
117 for(
int i=0; i < childList.size(); i++) {
118 AbstractFile childFile = (AbstractFile)childList.get(i);
119 if((purgeKnownFiles && childFile.getKnown() == TskData.FileKnown.KNOWN)
120 || (purgeSlackFiles && childFile.getType() == TskData.TSK_DB_FILES_TYPE_ENUM.SLACK)) {
121 numVisibleChildren--;
126 return numVisibleChildren;
138 public Action[] getActions(
boolean context) {
139 List<Action> actions =
new ArrayList<>();
140 final Content content = this.getLookup().lookup(Content.class);
141 if (content != null) {
142 actions.addAll(ExplorerNodeActionVisitor.getActions(content));
144 Directory dir = this.getLookup().lookup(Directory.class);
146 actions.add(ExtractAction.getInstance());
147 actions.add(
new RunIngestModulesAction(dir));
150 final Image img = this.getLookup().lookup(Image.class);
151 final VirtualDirectory virtualDirectory = this.getLookup().lookup(VirtualDirectory.class);
152 boolean isRootVD =
false;
153 if (virtualDirectory != null) {
155 if (virtualDirectory.getParent() == null) {
158 }
catch (TskCoreException ex) {
159 logger.log(Level.WARNING,
"Error determining the parent of the virtual directory", ex);
162 if (img != null || isRootVD) {
163 actions.add(
new FileSearchAction(NbBundle.getMessage(
this.getClass(),
"DirectoryTreeFilterNode.action.openFileSrcByAttr.text")));
164 actions.add(
new RunIngestModulesAction(Collections.<Content>singletonList(content)));
167 actions.add(collapseAllAction);
168 return actions.toArray(
new Action[actions.size()]);