mstoeck3/autopsy-flatpak
1
0
Fork 0
mirror of https://github.com/overcuriousity/autopsy-flatpak.git synced 2025-07-17 10:17:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

account actions support multiselect via actionsGlobalLookup

Browse Source
This commit is contained in:
millmanorama 2016-09-21 18:27:13 +02:00
parent 2e239ca601
commit 845aadb165
1 changed files with 34 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

73
Core/src/org/sleuthkit/autopsy/datamodel/Accounts.java
View File

@ -30,7 +30,6 @@ import java.sql.ResultSet;
import java.sql.SQLException; import java.sql.SQLException;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.Arrays; import java.util.Arrays;
import java.util.Collection;
import java.util.Collections; import java.util.Collections;
import java.util.HashSet; import java.util.HashSet;
import java.util.List; import java.util.List;
@ -57,6 +56,7 @@ import org.openide.nodes.Children;
import org.openide.nodes.Node; import org.openide.nodes.Node;
import org.openide.nodes.Sheet; import org.openide.nodes.Sheet;
import org.openide.util.NbBundle; import org.openide.util.NbBundle;
import org.openide.util.Utilities;
import org.openide.util.lookup.Lookups; import org.openide.util.lookup.Lookups;
import org.sleuthkit.autopsy.casemodule.Case; import org.sleuthkit.autopsy.casemodule.Case;
import org.sleuthkit.autopsy.coreutils.MessageNotifyUtil; import org.sleuthkit.autopsy.coreutils.MessageNotifyUtil;
@ -154,6 +154,9 @@ public class Accounts extends Observable implements AutopsyVisitableItem {
} }
} }
private final RejectAccounts rejectActionInstance;
private final ApproveAccounts approveActionInstance;
/** /**
* Constructor * Constructor
* *
@ -161,6 +164,9 @@ public class Accounts extends Observable implements AutopsyVisitableItem {
*/ */
Accounts(SleuthkitCase skCase) { Accounts(SleuthkitCase skCase) {
this.skCase = skCase; this.skCase = skCase;
this.rejectActionInstance = new RejectAccounts();
this.approveActionInstance = new ApproveAccounts();
} }
/** /**
@ -183,7 +189,6 @@ public class Accounts extends Observable implements AutopsyVisitableItem {
notifyObservers(); notifyObservers();
} }
/** /**
* Get an IINInfo object with details about the given IIN * Get an IINInfo object with details about the given IIN
* *
@ -888,9 +893,6 @@ public class Accounts extends Observable implements AutopsyVisitableItem {
return s; return s;
} }
@NbBundle.Messages({
"ApproveAccountsAction.name=Approve Accounts",
"RejectAccountsAction.name=Reject Accounts"})
@Override @Override
public Action[] getActions(boolean context) { public Action[] getActions(boolean context) {
Action[] actions = super.getActions(context); Action[] actions = super.getActions(context);
@ -901,8 +903,8 @@ public class Accounts extends Observable implements AutopsyVisitableItem {
} catch (TskCoreException ex) { } catch (TskCoreException ex) {
LOGGER.log(Level.SEVERE, "Error gettung content by id", ex); LOGGER.log(Level.SEVERE, "Error gettung content by id", ex);
} }
arrayList.add(new ApproveAccounts(getLookup().lookupAll(BlackboardArtifact.class))); arrayList.add(approveActionInstance);
arrayList.add(new RejectAccounts(getLookup().lookupAll(BlackboardArtifact.class))); arrayList.add(rejectActionInstance);
return arrayList.toArray(new Action[arrayList.size()]); return arrayList.toArray(new Action[arrayList.size()]);
} }
} }
@ -1107,7 +1109,6 @@ public class Accounts extends Observable implements AutopsyVisitableItem {
return iinEnd; return iinEnd;
} }
public long getCount() { public long getCount() {
return count; return count;
} }
@ -1226,8 +1227,8 @@ public class Accounts extends Observable implements AutopsyVisitableItem {
public Action[] getActions(boolean context) { public Action[] getActions(boolean context) {
List<Action> actionsList = new ArrayList<>(); List<Action> actionsList = new ArrayList<>();
actionsList.addAll(Arrays.asList(super.getActions(context))); actionsList.addAll(Arrays.asList(super.getActions(context)));
actionsList.add(new ApproveAccounts(Collections.singleton(artifact))); actionsList.add(approveActionInstance);
actionsList.add(new RejectAccounts(Collections.singleton(artifact))); actionsList.add(rejectActionInstance);
return actionsList.toArray(new Action[actionsList.size()]); return actionsList.toArray(new Action[actionsList.size()]);
} }
@ -1258,52 +1259,46 @@ public class Accounts extends Observable implements AutopsyVisitableItem {
@Override @Override
public void actionPerformed(ActionEvent e) { public void actionPerformed(ActionEvent e) {
showRejected = !showRejected; showRejected = !showRejected;
update(); Accounts.this.update();
} }
} }
private class ApproveAccounts extends AbstractAction { private abstract class ReviewStatusAction extends AbstractAction {
private final Collection<? extends BlackboardArtifact> artifacts; private final BlackboardArtifact.ReviewStatus newStatus;
private ReviewStatusAction(String displayName, BlackboardArtifact.ReviewStatus newStatus) {
super(displayName);
this.newStatus = newStatus;
ApproveAccounts(Collection<? extends BlackboardArtifact> artifacts) {
super(Bundle.ApproveAccountsAction_name());
this.artifacts = artifacts;
} }
@Override @Override
public void actionPerformed(ActionEvent e) { public void actionPerformed(ActionEvent e) {
try { Utilities.actionsGlobalContext().lookupAll(BlackboardArtifact.class).forEach(artifact -> {
for (BlackboardArtifact artifact : artifacts) { try {
skCase.setReviewStatus(artifact, BlackboardArtifact.ReviewStatus.APPROVED); skCase.setReviewStatus(artifact, newStatus);
} catch (TskCoreException ex) {
LOGGER.log(Level.SEVERE, "Error changing artifact review status.", ex); //NON-NLS
} }
Accounts.this.update(); });
} catch (TskCoreException ex) { Accounts.this.update();
LOGGER.log(Level.SEVERE, "Error approving artifacts.", ex); //NON-NLS
}
} }
} }
private class RejectAccounts extends AbstractAction { private class ApproveAccounts extends ReviewStatusAction {
private final Collection<? extends BlackboardArtifact> artifacts; @NbBundle.Messages({"ApproveAccountsAction.name=Approve Accounts"})
private ApproveAccounts() {
RejectAccounts(Collection<? extends BlackboardArtifact> artifacts) { super(Bundle.ApproveAccountsAction_name(), BlackboardArtifact.ReviewStatus.APPROVED);
super(Bundle.RejectAccountsAction_name());
this.artifacts = artifacts;
} }
}
@Override private class RejectAccounts extends ReviewStatusAction {
public void actionPerformed(ActionEvent e) {
try {
for (BlackboardArtifact artifact : artifacts) {
skCase.setReviewStatus(artifact, BlackboardArtifact.ReviewStatus.REJECTED);
}
Accounts.this.update();
} catch (TskCoreException ex) { @NbBundle.Messages({"RejectAccountsAction.name=Reject Accounts"})
LOGGER.log(Level.SEVERE, "Error approving artifacts.", ex); //NON-NLS private RejectAccounts() {
} super(Bundle.RejectAccountsAction_name(), BlackboardArtifact.ReviewStatus.REJECTED);
} }
} }
} }