AddEditCentralRepoCommentAction.java

Go to the documentation of this file.

/*
 * Autopsy Forensic Browser
 *
 * Copyright 2018-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.centralrepository;

import java.awt.event.ActionEvent;
import java.util.logging.Level;
import javax.swing.AbstractAction;
import javax.swing.Action;
import org.apache.commons.lang.StringUtils;
import org.openide.DialogDisplayer;
import org.openide.NotifyDescriptor;
import org.openide.util.NbBundle.Messages;
import org.sleuthkit.autopsy.casemodule.Case;
import org.sleuthkit.autopsy.casemodule.NoCurrentCaseException;
import org.sleuthkit.autopsy.centralrepository.datamodel.CorrelationAttributeInstance;
import org.sleuthkit.autopsy.centralrepository.datamodel.CorrelationAttributeUtil;
import org.sleuthkit.autopsy.centralrepository.datamodel.CentralRepoException;
import org.sleuthkit.autopsy.coreutils.Logger;
import org.sleuthkit.datamodel.AbstractFile;
import org.sleuthkit.autopsy.centralrepository.datamodel.CentralRepository;

@Messages({"AddEditCentralRepoCommentAction.menuItemText.addEditCentralRepoCommentEmptyFile=Add/Edit Central Repository Comment (Empty File)",
    "AddEditCentralRepoCommentAction.menuItemText.addEditCentralRepoCommentNoMD5=Add/Edit Central Repository Comment (No MD5 Hash)",
    "AddEditCentralRepoCommentAction.menuItemText.addEditCentralRepoComment=Add/Edit Central Repository Comment"})
public final class AddEditCentralRepoCommentAction extends AbstractAction {

    private static final Logger logger = Logger.getLogger(AddEditCentralRepoCommentAction.class.getName());
    private static final long serialVersionUID = 1L;

    private boolean addToDatabase;
    private CorrelationAttributeInstance correlationAttributeInstance;
    private String comment;
    private final Long fileId;

    public AddEditCentralRepoCommentAction(AbstractFile file) {
        fileId = file.getId();
        correlationAttributeInstance = CorrelationAttributeUtil.getCorrAttrForFile(file);
        if (correlationAttributeInstance == null) {
            addToDatabase = true;
            correlationAttributeInstance = CorrelationAttributeUtil.makeCorrAttrFromFile(file);
        }
        if (file.getSize() == 0) {
            putValue(Action.NAME, Bundle.AddEditCentralRepoCommentAction_menuItemText_addEditCentralRepoCommentEmptyFile());
        } else if (StringUtils.isBlank(file.getMd5Hash())) {
            putValue(Action.NAME, Bundle.AddEditCentralRepoCommentAction_menuItemText_addEditCentralRepoCommentNoMD5());
        } else {
            putValue(Action.NAME, Bundle.AddEditCentralRepoCommentAction_menuItemText_addEditCentralRepoComment());
        }
    }

    @Override
    public void actionPerformed(ActionEvent event) {
        CentralRepoCommentDialog centralRepoCommentDialog = new CentralRepoCommentDialog(correlationAttributeInstance);
        centralRepoCommentDialog.display();

        comment = null;

        if (centralRepoCommentDialog.isCommentUpdated()) {
            CentralRepository dbManager;

            try {
                dbManager = CentralRepository.getInstance();

                if (addToDatabase) {
                    dbManager.addArtifactInstance(correlationAttributeInstance);
                } else {
                    dbManager.updateAttributeInstanceComment(correlationAttributeInstance);
                }

                comment = centralRepoCommentDialog.getComment();
                try {
                    Case.getCurrentCaseThrows().notifyCentralRepoCommentChanged(fileId, comment);
                } catch (NoCurrentCaseException ex) {
                    logger.log(Level.WARNING, "Case not open after changing central repository comment", ex);
                }
            } catch (CentralRepoException ex) {
                logger.log(Level.SEVERE, "Error adding comment", ex);
                NotifyDescriptor notifyDescriptor = new NotifyDescriptor.Message(
                        "An error occurred while trying to save the comment to the central repository.",
                        NotifyDescriptor.ERROR_MESSAGE);
                DialogDisplayer.getDefault().notify(notifyDescriptor);
            }
        }
    }

    public String getComment() {
        return comment;
    }

    public CorrelationAttributeInstance getCorrelationAttribute() {
        return correlationAttributeInstance;
    }
}