org.jasig.i18n.translate.GoogleAutoTranslationService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of maven-translate-plugin Show documentation
Message File Translation Maven Plugin
The newest version!
/**
 * Licensed to Jasig under one or more contributor license
 * agreements. See the NOTICE file distributed with this work
 * for additional information regarding copyright ownership.
 * Jasig licenses this file to you 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.jasig.i18n.translate;

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.ListIterator;
import java.util.Map;
import java.util.Set;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.jasig.i18n.translate.GoogleTranslationResponse.Translation;
import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.http.converter.json.MappingJacksonHttpMessageConverter;
import org.springframework.web.client.RestTemplate;

public class GoogleAutoTranslationService {

    protected final Log log = LogFactory.getLog(getClass());

    private final String urlTemplate = "https://www.googleapis.com/language/translate/v2?key={key}&source={source}&target={target}&q={query}";

    private String apiKey;

    private String defaultLanguageKey = "en";

    private final RestTemplate restTemplate;

    public GoogleAutoTranslationService() {
        this.restTemplate = new RestTemplate();
        final List> converters = Collections.>singletonList(new MappingJacksonHttpMessageConverter());
        this.restTemplate.setMessageConverters(converters);
    }

    public Map getAutoUpdatedTranslationMap(final Map mainMap, final Map targetMap, final String languageKey) {

        // assemble a set of keys represented in the main map but not in the
        // target language map
        final Set missing = mainMap.keySet();
        missing.removeAll(targetMap.keySet());

        // put the keys in a list so that we have a consistent ordering
        final List keys = new ArrayList();
        keys.addAll(missing);

        // assemble a list of untranslated messages in the same order as our
        // missing key list
        final List untranslatedMessages = new ArrayList();
        for (final String key : keys) {
            untranslatedMessages.add(mainMap.get(key));
        }

        final Map parameters = new HashMap();
        parameters.put("key", this.apiKey);
        parameters.put("source", this.defaultLanguageKey);
        parameters.put("target", languageKey);
        parameters.put("query", untranslatedMessages);

        final GoogleTranslationResponse response = this.restTemplate.getForObject(this.urlTemplate, GoogleTranslationResponse.class, parameters);
        final List translations = response.getTranslations();

        // iterate through the auto-translations, adding each to the target
        // map
        final ListIterator iter = translations.listIterator();
        while (iter.hasNext()) {
            final Translation translation = iter.next();
            final String key = keys.get(iter.previousIndex());
            targetMap.put(key, translation.getTranslatedText());
        }

        return targetMap;
    }

    public void setApiKey(final String apiKey) {
        this.apiKey = apiKey;
    }

    public void setDefaultLanguageKey(final String defaultLanguageKey) {
        this.defaultLanguageKey = defaultLanguageKey;
    }

}