com.googleapis.ajax.services.impl.TranslateLanguageQueryImpl.java Source code

Introduction

Here is the source code for com.googleapis.ajax.services.impl.TranslateLanguageQueryImpl.java

Source

/*
 * Copyright 2010 Nabeel Mukhtar 
 * 
 * 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 com.googleapis.ajax.services.impl;

import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import com.google.gson.Gson;
import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.googleapis.ajax.common.PagedArrayList;
import com.googleapis.ajax.common.PagedList;
import com.googleapis.ajax.schema.Language;
import com.googleapis.ajax.schema.TranslateLanguageResult;
import com.googleapis.ajax.services.GoogleSearchException;
import com.googleapis.ajax.services.TranslateLanguageQuery;
import com.googleapis.ajax.services.constant.GoogleSearchApiUrls;
import com.googleapis.ajax.services.constant.ParameterNames;
import com.googleapis.ajax.services.enumeration.TranslationFormat;

/**
 * The Class TranslateLanguageQueryImpl.
 */
public class TranslateLanguageQueryImpl extends BaseGoogleSearchApiQuery<TranslateLanguageResult>
        implements TranslateLanguageQuery {
    private Map<String, List<String>> parameters = new HashMap<String, List<String>>();

    /**
     * Instantiates a new translate language query impl.
     * 
     * @param applicationId the application id
     */
    public TranslateLanguageQueryImpl(String applicationId) {
        super(applicationId);
    }

    /* (non-Javadoc)
     * @see com.google.code.googlesearch.client.GoogleSearchQuery#reset()
     */
    @Override
    public void reset() {
        apiUrlBuilder = createGoogleSearchApiUrlBuilder(GoogleSearchApiUrls.LANGUAGE_TRANSLATE_URL);
    }

    /* (non-Javadoc)
     * @see com.google.code.googlesearch.client.TranslateLanguageQuery#withFormat(com.google.code.googlesearch.client.enumeration.TranslationFormat)
     */
    @Override
    public TranslateLanguageQuery withFormat(TranslationFormat format) {
        //      apiUrlBuilder.withParameterEnum(ParameterNames.FORMAT, format);
        putParameters(ParameterNames.FORMAT, format.value());
        return this;
    }

    /* (non-Javadoc)
     * @see com.google.code.googlesearch.client.TranslateLanguageQuery#withLanguagePair(java.lang.String, java.lang.String)
     */
    @Override
    public TranslateLanguageQuery withLanguagePair(Language sourceLanguage, Language targetLanguage) {
        String languagePair = "|" + targetLanguage.value();
        if (sourceLanguage != null) {
            languagePair = sourceLanguage.value() + languagePair;
        }
        //      apiUrlBuilder.withParameter(ParameterNames.LANGUAGE_PAIR, languagePair);
        putParameters(ParameterNames.LANGUAGE_PAIR, languagePair);
        return this;
    }

    @Override
    public TranslateLanguageQuery withQuery(String query) {
        //      apiUrlBuilder.withParameter(ParameterNames.QUERY, query);
        putParameters(ParameterNames.QUERY, query);
        return this;
    }

    @Override
    public PagedList<TranslateLanguageResult> list() {
        InputStream jsonContent = null;
        try {
            jsonContent = callApiPost(apiUrlBuilder.buildUrl(), parameters);
            JsonElement response = parser.parse(new InputStreamReader(jsonContent, UTF_8_CHAR_SET));
            if (response.isJsonObject()) {
                PagedList<TranslateLanguageResult> responseList = unmarshallList(response.getAsJsonObject());
                notifyObservers(responseList);
                return responseList;
            }
            throw new GoogleSearchException("Unknown content found in response:" + response.toString());
        } catch (Exception e) {
            throw new GoogleSearchException(e);
        } finally {
            closeStream(jsonContent);
        }
    }

    @Override
    public TranslateLanguageResult singleResult() {
        InputStream jsonContent = null;
        try {
            jsonContent = callApiPost(apiUrlBuilder.buildUrl(), parameters);
            JsonElement response = parser.parse(new InputStreamReader(jsonContent, UTF_8_CHAR_SET));
            if (response.isJsonObject()) {
                JsonObject json = response.getAsJsonObject();
                int status = json.get("responseStatus").getAsInt();
                if (status != 200) {
                    throw new GoogleSearchException(json.get("responseDetails").getAsString());
                }
                JsonElement data = json.get("responseData");
                if (data != null) {
                    return unmarshall(data);
                }
            }
            throw new GoogleSearchException("Unknown content found in response:" + response.toString());
        } catch (Exception e) {
            throw new GoogleSearchException(e);
        } finally {
            closeStream(jsonContent);
        }
    }

    /**
     * Unmarshall list.
     * 
     * @param response the response
     * 
     * @return the paged list< t>
     */
    protected PagedList<TranslateLanguageResult> unmarshallList(JsonObject response) {
        int status = response.get("responseStatus").getAsInt();
        if (status != 200) {
            throw new GoogleSearchException(String.valueOf(response.get("responseDetails").getAsString()));
        }
        JsonArray dataArray = response.get("responseData").getAsJsonArray();
        PagedArrayList<TranslateLanguageResult> list = new PagedArrayList<TranslateLanguageResult>();
        if (dataArray != null) {
            for (JsonElement element : dataArray) {
                if (element.isJsonObject()) {
                    JsonObject json = element.getAsJsonObject();
                    status = json.get("responseStatus").getAsInt();
                    if (status != 200) {
                        throw new GoogleSearchException(json.get("responseDetails").getAsString());
                    }
                    JsonElement data = json.get("responseData");
                    if (data != null) {
                        list.add(unmarshall(data));
                    }
                }
            }
        }
        return list;
    }

    /* (non-Javadoc)
     * @see com.google.code.googlesearch.client.impl.BaseGoogleSearchApiQuery#unmarshall(com.google.gson.JsonElement)
     */
    @Override
    protected TranslateLanguageResult unmarshall(JsonElement object) {
        Gson gson = getGsonBuilder().create();
        return gson.fromJson(object, TranslateLanguageResult.class);
    }

    protected void putParameters(String name, String value) {
        List<String> values = parameters.get(name);
        if (values == null) {
            values = new ArrayList<String>();
            parameters.put(name, values);
        }
        values.add(value);
    }
}