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

commenting

Browse Source
This commit is contained in:
Greg DiCristofaro 2022-02-16 14:40:41 -05:00
parent 99eb6a9499
commit a06fec7fe9
1 changed files with 61 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

65
Core/src/org/sleuthkit/autopsy/mainui/datamodel/TagsDAO.java
View File

@ -414,10 +414,15 @@ public class TagsDAO extends AbstractDAO {
return null; return null;
} }
private SleuthkitCase getCase() throws NoCurrentCaseException, TskCoreException { /**
return Case.getCurrentCaseThrows().getSleuthkitCase(); * Returns the counts of each tag name.
} *
* @param dataSourceId The data source object id to filter on.
*
* @return The tree item results.
*
* @throws ExecutionException
*/
public TreeResultsDTO<? extends TagNameSearchParams> getNameCounts(Long dataSourceId) throws ExecutionException { public TreeResultsDTO<? extends TagNameSearchParams> getNameCounts(Long dataSourceId) throws ExecutionException {
Set<TagName> indeterminateTagNameIds = this.treeCounts.getEnqueued().stream() Set<TagName> indeterminateTagNameIds = this.treeCounts.getEnqueued().stream()
.filter(evt -> dataSourceId == null || evt.getDataSourceId() == dataSourceId) .filter(evt -> dataSourceId == null || evt.getDataSourceId() == dataSourceId)
@ -481,6 +486,15 @@ public class TagsDAO extends AbstractDAO {
return new TreeResultsDTO<>(tagNameParams); return new TreeResultsDTO<>(tagNameParams);
} }
/**
* Creates a tag name tree item.
*
* @param tagName The tag name.
* @param dataSourceId The data source object id or null if not present.
* @param treeDisplayCount The tree display count.
*
* @return The tree item dto.
*/
public TreeItemDTO<TagNameSearchParams> createTagNameTreeItem(TagName tagName, Long dataSourceId, TreeResultsDTO.TreeDisplayCount treeDisplayCount) { public TreeItemDTO<TagNameSearchParams> createTagNameTreeItem(TagName tagName, Long dataSourceId, TreeResultsDTO.TreeDisplayCount treeDisplayCount) {
return new TreeItemDTO<>( return new TreeItemDTO<>(
TagNameSearchParams.getTypeId(), TagNameSearchParams.getTypeId(),
@ -491,6 +505,18 @@ public class TagsDAO extends AbstractDAO {
); );
} }
/**
* The count of content tags.
*
* @param dataSourceId The data source id where the content tag should
* appear or null.
* @param tagName The tag name.
*
* @return The count.
*
* @throws NoCurrentCaseException
* @throws TskCoreException
*/
private long getContentTagCount(Long dataSourceId, TagName tagName) throws NoCurrentCaseException, TskCoreException { private long getContentTagCount(Long dataSourceId, TagName tagName) throws NoCurrentCaseException, TskCoreException {
if (UserPreferences.showOnlyCurrentUserTags()) { if (UserPreferences.showOnlyCurrentUserTags()) {
@ -505,6 +531,18 @@ public class TagsDAO extends AbstractDAO {
} }
} }
/**
* The count of result tags.
*
* @param dataSourceId The data source id where the result tag should appear
* or null.
* @param tagName The tag name.
*
* @return The count.
*
* @throws NoCurrentCaseException
* @throws TskCoreException
*/
private long getArtifactTagCount(Long dataSourceId, TagName tagName) throws NoCurrentCaseException, TskCoreException { private long getArtifactTagCount(Long dataSourceId, TagName tagName) throws NoCurrentCaseException, TskCoreException {
if (UserPreferences.showOnlyCurrentUserTags()) { if (UserPreferences.showOnlyCurrentUserTags()) {
String userName = System.getProperty(USER_NAME_PROPERTY); String userName = System.getProperty(USER_NAME_PROPERTY);
@ -518,6 +556,15 @@ public class TagsDAO extends AbstractDAO {
} }
} }
/**
* Returns the counts of file and result type given the search params.
*
* @param searchParams The tag name search params.
*
* @return The tree item results.
*
* @throws ExecutionException
*/
public TreeResultsDTO<? extends TagsSearchParams> getTypeCounts(TagNameSearchParams searchParams) throws ExecutionException { public TreeResultsDTO<? extends TagsSearchParams> getTypeCounts(TagNameSearchParams searchParams) throws ExecutionException {
Long dataSourceId = searchParams.getDataSourceId(); Long dataSourceId = searchParams.getDataSourceId();
TagName tagName = searchParams.getTagName(); TagName tagName = searchParams.getTagName();
@ -549,6 +596,16 @@ public class TagsDAO extends AbstractDAO {
} }
} }
/**
* Creates a tag type tree item.
*
* @param tagName The tag name.
* @param tagType The tag type.
* @param dataSourceId The data source object id or null if not present.
* @param treeDisplayCount The tree display count.
*
* @return The tree item dto.
*/
public TreeItemDTO<TagsSearchParams> createTagTypeTreeItem(TagName tagName, TagType tagType, Long dataSourceId, TreeResultsDTO.TreeDisplayCount treeDisplayCount) { public TreeItemDTO<TagsSearchParams> createTagTypeTreeItem(TagName tagName, TagType tagType, Long dataSourceId, TreeResultsDTO.TreeDisplayCount treeDisplayCount) {
return new TreeItemDTO<>( return new TreeItemDTO<>(
TagsSearchParams.getTypeId(), TagsSearchParams.getTypeId(),