ResultDomain.java

Go to the documentation of this file.

/*
 * Autopsy
 *
 * Copyright 2020 Basis Technology Corp.
 * Contact: carrier <at> sleuthkit <dot> org
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.sleuthkit.autopsy.discovery.search;

import java.util.Set;
import java.util.HashSet;
import org.apache.commons.lang3.StringUtils;
import org.openide.util.NbBundle;
import org.sleuthkit.datamodel.Content;
import org.sleuthkit.datamodel.TskData;

public class ResultDomain extends Result {

    private final String domain;
    private final Long activityStart;
    private final Long activityEnd;
    private final Long totalPageViews;
    private final Long pageViewsInLast60;
    private final Long filesDownloaded;
    private final Long countOfKnownAccountTypes;
    private final String accountTypes;
    private final Set<String> webCategories = new HashSet<>();

    private final Content dataSource;
    private final long dataSourceId;

    ResultDomain(String domain, Long activityStart, Long activityEnd, Long totalPageViews,
            Long pageViewsInLast60, Long filesDownloaded, Long countOfKnownAccountTypes, String accountTypes, Content dataSource) {
        this.domain = domain;
        this.dataSource = dataSource;
        this.dataSourceId = dataSource.getId();
        this.activityStart = activityStart;
        this.activityEnd = activityEnd;
        this.totalPageViews = totalPageViews;
        this.pageViewsInLast60 = pageViewsInLast60;
        this.filesDownloaded = filesDownloaded;
        this.countOfKnownAccountTypes = countOfKnownAccountTypes;
        this.accountTypes = accountTypes;
    }

    ResultDomain(ResultDomain resultDomain) {
        this.domain = resultDomain.getDomain();
        this.dataSource = resultDomain.getDataSource();
        this.dataSourceId = resultDomain.getDataSourceObjectId();
        this.activityStart = resultDomain.getActivityStart();
        this.activityEnd = resultDomain.getActivityEnd();
        this.totalPageViews = resultDomain.getTotalPageViews();
        this.pageViewsInLast60 = resultDomain.getPageViewsInLast60Days();
        this.filesDownloaded = resultDomain.getFilesDownloaded();
        this.countOfKnownAccountTypes = resultDomain.getCountOfKnownAccountTypes();
        this.accountTypes = resultDomain.getAccountTypes();
    }

    public String getDomain() {
        return this.domain;
    }

    public Long getActivityStart() {
        return activityStart;
    }

    public Long getActivityEnd() {
        return activityEnd;
    }

    public Long getTotalPageViews() {
        return totalPageViews;
    }

    public Long getPageViewsInLast60Days() {
        return pageViewsInLast60;
    }

    public Long getFilesDownloaded() {
        return filesDownloaded;
    }

    @NbBundle.Messages({
        "ResultDomain_getDefaultCategory=Uncategorized"
    })
    public Set<String> getWebCategories() {
        Set<String> returnList = new HashSet<>();
        if (webCategories.isEmpty()) {
            returnList.add(Bundle.ResultDomain_getDefaultCategory());
        } else {
            returnList.addAll(webCategories);
        }
        return returnList;
    }

    public void addWebCategories(Set<String> categories) {
        if (categories != null && !categories.isEmpty()) {
            this.webCategories.addAll(categories);
        }
    }

    public boolean hasKnownAccountType() {
        return getCountOfKnownAccountTypes() != null
                && getCountOfKnownAccountTypes() > 0;
    }

    @NbBundle.Messages({
        "ResultDomain_noAccountTypes=Unknown"
    })
    public String getAccountTypes() {
        if (StringUtils.isBlank(accountTypes)) {
            return Bundle.ResultDomain_noAccountTypes();
        }
        return accountTypes;
    }

    @Override
    public long getDataSourceObjectId() {
        return this.dataSourceId;
    }

    @Override
    public Content getDataSource() {
        return this.dataSource;
    }

    @Override
    public TskData.FileKnown getKnown() {
        return TskData.FileKnown.UNKNOWN;
    }

    @Override
    public SearchData.Type getType() {
        return SearchData.Type.DOMAIN;
    }

    @Override
    public String toString() {
        return "[domain=" + this.domain + ", data_source=" + this.dataSourceId + ", start="
                + this.activityStart + ", end=" + this.activityEnd + ", totalVisits=" + this.totalPageViews + ", visitsLast60="
                + this.pageViewsInLast60 + ", downloads=" + this.filesDownloaded + ", frequency="
                + this.getFrequency() + "]";
    }

    Long getCountOfKnownAccountTypes() {
        return countOfKnownAccountTypes;
    }
}