TextTranslationService.java

Go to the documentation of this file.

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

import java.util.Collection;
import java.util.Collections;
import java.util.Optional;
import org.openide.util.Lookup;
import org.sleuthkit.autopsy.core.UserPreferences;

public final class TextTranslationService {

    private final static TextTranslationService tts = new TextTranslationService();

    private final Collection<? extends TextTranslator> translators;
    private Optional<TextTranslator> selectedTranslator;

    private TextTranslationService() {
        //Perform look up for Text Translation implementations ONLY ONCE during 
        //class loading.
        translators = Lookup.getDefault().lookupAll(TextTranslator.class);
        updateSelectedTranslator();
    }

    public static TextTranslationService getInstance() {
        return tts;
    }

    public void updateSelectedTranslator() {
        String translatorName = UserPreferences.getTextTranslatorName();
        for (TextTranslator translator : translators) {
            if (translator.getName().equals(translatorName)) {
                selectedTranslator = Optional.ofNullable(translator);
                return;
            }
        }
        selectedTranslator = Optional.empty();
    }

    public String translate(String input) throws NoServiceProviderException, TranslationException {
        if (hasProvider()) {
            return selectedTranslator.get().translate(input);
        }
        throw new NoServiceProviderException(
                "Could not find a TextTranslator service provider");
    }

    public TextTranslator getTranslatorByName(String translatorName) throws NoServiceProviderException {
        for (TextTranslator translator : translators) {
            if (translator.getName().equals(translatorName)) {
                return translator;
            }
        }
        throw new NoServiceProviderException(
                "Could not find the specified TextTranslator service provider: " + translatorName);
    }

    public Collection<? extends TextTranslator> getTranslators() {
        return Collections.unmodifiableCollection(translators);
    }

    public boolean hasProvider() {
        return selectedTranslator.isPresent();
    }
    
    public int getMaxPayloadSize() {
        return selectedTranslator.get().getMaxPayloadSize();
    }
}