GoogleTranslatorSettings.java

Go to the documentation of this file.

/*
 * Autopsy
 *
 * Copyright 2019 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.texttranslation.translators;

import com.google.cloud.translate.TranslateOptions;
import org.apache.commons.lang3.StringUtils;
import org.sleuthkit.autopsy.coreutils.ModuleSettings;

public final class GoogleTranslatorSettings {

    private static final String DEFAULT_TARGET_LANGUAGE = TranslateOptions.getDefaultInstance().getTargetLanguage();
    private static final String CREDENTIAL_PATH_KEY = "CredentialPath";
    private static final String TARGET_LANGUAGE_CODE_KEY = "TargetLanguageCode";
    private static final String GOOGLE_TRANSLATE_NAME = "GoogleTranslate";
    private static final String DEFAULT_CREDENTIAL_PATH = "";
    private String targetLanguageCode;
    private String credentialPath;

    GoogleTranslatorSettings() {
        loadSettings();
    }

    String getTargetLanguageCode() {
        return targetLanguageCode;
    }

    String getCredentialPath() {
        return credentialPath;
    }

    void setCredentialPath(String path) {
        credentialPath = path;
    }

    void setTargetLanguageCode(String code) {
        if (StringUtils.isBlank(code)) {
            targetLanguageCode = DEFAULT_TARGET_LANGUAGE;
        } else {
            targetLanguageCode = code;
        }
    }

    void loadSettings() {
        if (!ModuleSettings.configExists(GOOGLE_TRANSLATE_NAME)) {
            ModuleSettings.makeConfigFile(GOOGLE_TRANSLATE_NAME);
        }
        if (ModuleSettings.settingExists(GOOGLE_TRANSLATE_NAME, TARGET_LANGUAGE_CODE_KEY)) {
            targetLanguageCode = ModuleSettings.getConfigSetting(GOOGLE_TRANSLATE_NAME, TARGET_LANGUAGE_CODE_KEY);
        }
        if (targetLanguageCode == null || StringUtils.isBlank(targetLanguageCode)) {
            targetLanguageCode = DEFAULT_TARGET_LANGUAGE;
        }
        if (ModuleSettings.settingExists(GOOGLE_TRANSLATE_NAME, CREDENTIAL_PATH_KEY)) {
            credentialPath = ModuleSettings.getConfigSetting(GOOGLE_TRANSLATE_NAME, CREDENTIAL_PATH_KEY);
        }
        if (credentialPath == null) {
            credentialPath = DEFAULT_CREDENTIAL_PATH;
        }
    }

    void saveSettings() {
        ModuleSettings.setConfigSetting(GOOGLE_TRANSLATE_NAME, TARGET_LANGUAGE_CODE_KEY, targetLanguageCode);
        ModuleSettings.setConfigSetting(GOOGLE_TRANSLATE_NAME, CREDENTIAL_PATH_KEY, credentialPath);
    }
}