PerCaseTimelineProperties.java

Go to the documentation of this file.

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

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Objects;
import java.util.Properties;
import org.apache.commons.lang3.StringUtils;
import org.sleuthkit.autopsy.casemodule.Case;

class PerCaseTimelineProperties {

    private static final String STALE_KEY = "stale"; //NON-NLS
    private static final String WAS_INGEST_RUNNING_KEY = "was_ingest_running"; // NON-NLS

    private final Case autoCase;
    private final Path propertiesPath;

    PerCaseTimelineProperties(Case c) {
        Objects.requireNonNull(c, "Case must not be null");
        this.autoCase = c;
        propertiesPath = Paths.get(autoCase.getModuleDirectory(), "Timeline", "timeline.properties"); //NON-NLS
    }

    public synchronized boolean isDBStale() throws IOException {

        String stale = getProperty(STALE_KEY);
        return StringUtils.isBlank(stale) ? true : Boolean.valueOf(stale);

    }

    public synchronized void setDbStale(Boolean stale) throws IOException {
        setProperty(STALE_KEY, stale.toString());
    }

    public synchronized boolean wasIngestRunning() throws IOException {
        String stale = getProperty(WAS_INGEST_RUNNING_KEY);
        return StringUtils.isBlank(stale) ? true : Boolean.valueOf(stale);
    }

    public synchronized void setIngestRunning(Boolean ingestRunning) throws IOException {
        setProperty(WAS_INGEST_RUNNING_KEY, ingestRunning.toString());
    }

    private synchronized Path getPropertiesPath() throws IOException {

        if (!Files.exists(propertiesPath)) {
            Path parent = propertiesPath.getParent();
            Files.createDirectories(parent);
            Files.createFile(propertiesPath);
        }
        return propertiesPath;
    }

    private synchronized String getProperty(String propertyKey) throws IOException {
        return getProperties().getProperty(propertyKey);
    }

    private synchronized void setProperty(String propertyKey, String propertyValue) throws IOException {
        Path propertiesFile = getPropertiesPath();
        Properties props = getProperties(propertiesFile);
        props.setProperty(propertyKey, propertyValue);

        try (OutputStream fos = Files.newOutputStream(propertiesFile)) {
            props.store(fos, ""); //NON-NLS
        }
    }

    private synchronized Properties getProperties() throws IOException {
        return getProperties(getPropertiesPath());
    }

    private synchronized Properties getProperties(final Path propertiesFile) throws IOException {
        try (InputStream inputStream = Files.newInputStream(propertiesFile)) {
            Properties props = new Properties();
            props.load(inputStream);
            return props;
        }
    }
}