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;
38 import org.
sleuthkit.datamodel.BlackboardArtifact.ARTIFACT_TYPE;
51 class DirectoryTreeFilterNode
extends FilterNode {
53 private static final Logger logger = Logger.getLogger(DirectoryTreeFilterNode.class.getName());
54 private static final Action collapseAllAction =
new CollapseAction(NbBundle.getMessage(DirectoryTreeFilterNode.class,
"DirectoryTreeFilterNode.action.collapseAll.text"));
65 DirectoryTreeFilterNode(Node nodeToWrap,
boolean createChildren) {
67 DirectoryTreeFilterChildren.createInstance(nodeToWrap, createChildren),
68 new ProxyLookup(Lookups.singleton(nodeToWrap), nodeToWrap.getLookup()));
78 public String getDisplayName() {
79 final Node orig = getOriginal();
80 String name = orig.getDisplayName();
81 if (orig instanceof AbstractContentNode) {
82 AbstractFile file = getLookup().lookup(AbstractFile.class);
83 if ((file != null) && (
false == (orig instanceof BlackboardArtifactNode)) ){
85 int numVisibleChildren = getVisibleChildCount(file);
92 name = name +
" \u200E(\u200E" + numVisibleChildren +
")\u200E";
94 }
catch (TskCoreException ex) {
95 logger.log(Level.SEVERE,
"Error getting children count to display for file: " + file, ex);
98 else if (orig instanceof BlackboardArtifactNode) {
99 BlackboardArtifact artifact = ((BlackboardArtifactNode) orig).getArtifact();
101 int numAttachments = artifact.getChildrenCount();
102 name = name +
" \u200E(\u200E" + numAttachments +
")\u200E";
103 }
catch (TskCoreException ex) {
104 logger.log(Level.SEVERE,
"Error getting chidlren count for atifact: " + artifact, ex);
120 private int getVisibleChildCount(AbstractFile file)
throws TskCoreException {
121 List<Content> childList = file.getChildren();
123 int numVisibleChildren = childList.size();
124 boolean purgeKnownFiles = UserPreferences.hideKnownFilesInDataSourcesTree();
125 boolean purgeSlackFiles = UserPreferences.hideSlackFilesInDataSourcesTree();
127 if (purgeKnownFiles || purgeSlackFiles) {
129 for (
int i = 0; i < childList.size(); i++) {
130 Content child = childList.get(i);
131 if (child instanceof AbstractFile) {
132 AbstractFile childFile = (AbstractFile) child;
133 if ((purgeKnownFiles && childFile.getKnown() == TskData.FileKnown.KNOWN)
134 || (purgeSlackFiles && childFile.getType() == TskData.TSK_DB_FILES_TYPE_ENUM.SLACK)) {
135 numVisibleChildren--;
137 }
else if(child instanceof BlackboardArtifact){
138 BlackboardArtifact bba = (BlackboardArtifact) child;
141 if((bba.getArtifactTypeID() != ARTIFACT_TYPE.TSK_EMAIL_MSG.getTypeID())
142 && (bba.getArtifactTypeID() != ARTIFACT_TYPE.TSK_MESSAGE.getTypeID())){
143 numVisibleChildren--;
150 return numVisibleChildren;
162 public Action[] getActions(
boolean context) {
163 List<Action> actions =
new ArrayList<>();
164 final Content content = this.getLookup().lookup(Content.class);
165 if (content != null) {
166 actions.addAll(ExplorerNodeActionVisitor.getActions(content));
168 Directory dir = this.getLookup().lookup(Directory.class);
170 actions.add(ExtractAction.getInstance());
171 actions.add(
new RunIngestModulesAction(dir));
174 final Image img = this.getLookup().lookup(Image.class);
175 final VirtualDirectory virtualDirectory = this.getLookup().lookup(VirtualDirectory.class);
176 boolean isRootVD =
false;
177 if (virtualDirectory != null) {
179 if (virtualDirectory.getParent() == null) {
182 }
catch (TskCoreException ex) {
183 logger.log(Level.WARNING,
"Error determining the parent of the virtual directory", ex);
186 if (img != null || isRootVD) {
187 actions.add(
new FileSearchAction(NbBundle.getMessage(
this.getClass(),
"DirectoryTreeFilterNode.action.openFileSrcByAttr.text")));
188 actions.add(
new RunIngestModulesAction(Collections.<Content>singletonList(content)));
191 actions.add(collapseAllAction);
192 return actions.toArray(
new Action[actions.size()]);
201 public Node getOriginal() {
202 return super.getOriginal();