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

Merge pull request #3157 from raman-bt/accounts_relationships

Browse Source 
Milestone 1 UI/CommuncationsManager integration
This commit is contained in:
Richard Cordovano 2017-10-26 09:19:36 -04:00 committed by GitHub
commit 01e218d248
3 changed files with 61 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

69
Core/src/org/sleuthkit/autopsy/communications/AccountNode.java → Core/src/org/sleuthkit/autopsy/communications/AccountDeviceInstanceNode.java
View File

@ -18,26 +18,35 @@
*/
package org.sleuthkit.autopsy.communications;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.openide.nodes.AbstractNode;
import org.openide.nodes.Children;
import org.openide.nodes.Sheet;
import org.openide.util.Exceptions;
import org.openide.util.NbBundle;
import org.openide.util.lookup.Lookups;
import org.sleuthkit.autopsy.casemodule.Case;
import org.sleuthkit.autopsy.datamodel.NodeProperty;
import org.sleuthkit.datamodel.Account;
import org.sleuthkit.datamodel.AccountDeviceInstance;
import org.sleuthkit.datamodel.CommunicationsFilter;
import org.sleuthkit.datamodel.TskCoreException;
/**
* Node to represent an Account in the AccountsBrowser
*/
class AccountNode extends AbstractNode {
class AccountDeviceInstanceNode extends AbstractNode {
private final Account account;
private static final Logger LOGGER = Logger.getLogger(AbstractNode.class.getName());
AccountNode(Account account) {
super(Children.LEAF, Lookups.fixed(account));
this.account = account;
setName(account.getAccountUniqueID());
setIconBaseWithExtension("org/sleuthkit/autopsy/communications/images/" + getIconFileName(account.getAccountType()));
private final AccountDeviceInstance accountDeviceInstance;
AccountDeviceInstanceNode(AccountDeviceInstance accountDeviceInstance) {
super(Children.LEAF, Lookups.fixed(accountDeviceInstance));
this.accountDeviceInstance = accountDeviceInstance;
setName(accountDeviceInstance.getAccount().getAccountUniqueID());
setIconBaseWithExtension("org/sleuthkit/autopsy/communications/images/" + getIconFileName(accountDeviceInstance.getAccount().getAccountType()));
}
/**
@ -47,25 +56,25 @@ class AccountNode extends AbstractNode {
* @return The file name of the icon for the given Account Type.
*/
final String getIconFileName(Account.Type type) {
if (type == Account.Type.CREDIT_CARD) {
if (type.equals(Account.Type.CREDIT_CARD)) {
return "credit-card.png";
} else if (type == Account.Type.DEVICE) {
} else if (type.equals(Account.Type.DEVICE)) {
return "image.png";
} else if (type == Account.Type.EMAIL) {
} else if (type.equals(Account.Type.EMAIL)) {
return "email.png";
} else if (type == Account.Type.FACEBOOK) {
} else if (type.equals(Account.Type.FACEBOOK)) {
return "facebook.png";
} else if (type == Account.Type.INSTAGRAM) {
} else if (type.equals(Account.Type.INSTAGRAM)) {
return "instagram.png";
} else if (type == Account.Type.MESSAGING_APP) {
} else if (type.equals(Account.Type.MESSAGING_APP)) {
return "messaging.png";
} else if (type == Account.Type.PHONE) {
} else if (type.equals(Account.Type.PHONE)) {
return "phone.png";
} else if (type == Account.Type.TWITTER) {
} else if (type.equals(Account.Type.TWITTER)) {
return "twitter.png";
} else if (type == Account.Type.WEBSITE) {
} else if (type.equals(Account.Type.WEBSITE)) {
return "web-file.png";
} else if (type == Account.Type.WHATSAPP) {
} else if (type.equals(Account.Type.WHATSAPP)) {
return "WhatsApp.png";
} else {
//there could be a default icon instead...
@ -88,20 +97,32 @@ class AccountNode extends AbstractNode {
s.put(properties);
}
// RAMAN TBD: need to figure out how to get the right filters here
// We talked about either creating a wrapper class around AccountDeviceInstance to push the selected filters
// Or some kind of static access to pull the selected filters
long msgCount = 0;
try {
CommunicationsFilter filter = null;
msgCount = Case.getCurrentCase().getSleuthkitCase().getCommunicationsManager().getRelationshipsCount(filter, accountDeviceInstance);
}
catch (TskCoreException ex) {
LOGGER.log(Level.WARNING, "Failed to get message count for account", ex); //NON-NLS
}
properties.put(new NodeProperty<>("type",
Bundle.AccountNode_accountType(),
"type",
account.getAccountType().getDisplayName())); // NON-NLS
accountDeviceInstance.getAccount().getAccountType().getDisplayName())); // NON-NLS
properties.put(new NodeProperty<>("count",
Bundle.AccountNode_messageCount(),
"count",
1)); // NON-NLS //dummy value
msgCount)); // NON-NLS
//how do I get the device name
// properties.put(new NodeProperty<>("device",
// Bundle.AccountNode_device(),
// "device",
// account.)); // NON-NLS
properties.put(new NodeProperty<>("device",
Bundle.AccountNode_device(),
"device",
accountDeviceInstance.getDeviceId())); // NON-NLS
return s;
}
}

18
Core/src/org/sleuthkit/autopsy/communications/AccountsNodeChildren.java → Core/src/org/sleuthkit/autopsy/communications/AccountDeviceInstancesNodeChildren.java
View File

@ -22,15 +22,15 @@ import java.util.Collections;
import java.util.List;
import org.openide.nodes.Children;
import org.openide.nodes.Node;
import org.sleuthkit.datamodel.Account;
import org.sleuthkit.datamodel.AccountDeviceInstance;
class AccountsNodeChildren extends Children.Keys<Account> {
class AccountDeviceInstancesNodeChildren extends Children.Keys<AccountDeviceInstance> {
private final List<Account> accounts;
private final List<AccountDeviceInstance> accountDeviceInstances;
AccountsNodeChildren(List<Account> accounts) {
AccountDeviceInstancesNodeChildren(List<AccountDeviceInstance> accountDeviceInstances) {
super(true);
this.accounts = accounts;
this.accountDeviceInstances = accountDeviceInstances;
}
@Override
@ -42,7 +42,7 @@ class AccountsNodeChildren extends Children.Keys<Account> {
@Override
protected void addNotify() {
super.addNotify();
setKeys(accounts);
setKeys(accountDeviceInstances);
}
//These are the methods for ChildFactory. I am going to keep them around but commented until we make a final descision.
@ -54,10 +54,10 @@ class AccountsNodeChildren extends Children.Keys<Account> {
//
// @Override
// protected Node createNodeForKey(Account key) {
// return new AccountNode(key);
// return new AccountDeviceInstanceNode(key);
// }
@Override
protected Node[] createNodes(Account key) {
return new Node[]{new AccountNode(key)};
protected Node[] createNodes(AccountDeviceInstance key) {
return new Node[]{new AccountDeviceInstanceNode(key)};
}
}

12
Core/src/org/sleuthkit/autopsy/communications/CVTTopComponent.java
View File

@ -31,7 +31,7 @@ import org.openide.windows.TopComponent;
import org.openide.windows.WindowManager;
import org.sleuthkit.autopsy.casemodule.Case;
import org.sleuthkit.autopsy.coreutils.ThreadConfined;
import org.sleuthkit.datamodel.Account;
import org.sleuthkit.datamodel.AccountDeviceInstance;
import org.sleuthkit.datamodel.CommunicationsManager;
import org.sleuthkit.datamodel.TskCoreException;
@ -162,12 +162,12 @@ public final class CVTTopComponent extends TopComponent implements ExplorerManag
* ExplorerManager
*/
try {
List<Account> accounts = new ArrayList<>();
final CommunicationsManager communicationsManager = Case.getCurrentCase().getSleuthkitCase().getCommunicationsManager();
accounts.addAll(communicationsManager.getAccounts(Account.Type.EMAIL));
accounts.addAll(communicationsManager.getAccounts(Account.Type.DEVICE));
List<AccountDeviceInstance> accountDeviceInstances = new ArrayList<>();
em.setRootContext(new AbstractNode(new AccountsNodeChildren(accounts)));
final CommunicationsManager communicationsManager = Case.getCurrentCase().getSleuthkitCase().getCommunicationsManager();
accountDeviceInstances.addAll(communicationsManager.getAccountDeviceInstancesWithRelationships(null));
em.setRootContext(new AbstractNode(new AccountDeviceInstancesNodeChildren(accountDeviceInstances)));
} catch (TskCoreException ex) {
Exceptions.printStackTrace(ex);
}