[Contextual Search] Trigger the translation one-box.

chrome/android/java/src/org/chromium/chrome/browser/contextualsearch/ContextualSearchRequest.java

Index: chrome/android/java/src/org/chromium/chrome/browser/contextualsearch/ContextualSearchRequest.java
diff --git a/chrome/android/java/src/org/chromium/chrome/browser/contextualsearch/ContextualSearchRequest.java b/chrome/android/java/src/org/chromium/chrome/browser/contextualsearch/ContextualSearchRequest.java
index 0b9bb03e4a97cc05a4fa02d4237311d9f2216a24..614a818b2994e33939fc130c977ab04cfbd58e55 100644
--- a/chrome/android/java/src/org/chromium/chrome/browser/contextualsearch/ContextualSearchRequest.java
+++ b/chrome/android/java/src/org/chromium/chrome/browser/contextualsearch/ContextualSearchRequest.java
@@ -5,11 +5,14 @@
 package org.chromium.chrome.browser.contextualsearch;
 
 import android.net.Uri;
+import android.text.TextUtils;
 
+import org.chromium.base.VisibleForTesting;
 import org.chromium.chrome.browser.search_engines.TemplateUrlService;
 
 import java.net.MalformedURLException;
 import java.net.URL;
+import java.util.Locale;
 
 import javax.annotation.Nullable;
 
@@ -19,16 +22,24 @@ import javax.annotation.Nullable;
  * fall-back when the low-priority version fails, and tracks which one is in use.
  */
 class ContextualSearchRequest {
-
-    private final Uri mLowPriorityUri;
-    private final Uri mNormalPriorityUri;
     private final boolean mWasPrefetch;
 
+    private Uri mLowPriorityUri;
+    private Uri mNormalPriorityUri;
+
     private boolean mIsLowPriority;
     private boolean mHasFailedLowPriorityLoad;
+    private boolean mIsTranslationForced;
 
+    private static final String GWS_LOW_PRIORITY_SEARCH_PATH = "s";
+    private static final String GWS_SEARCH_NO_SUGGESTIONS_PARAM = "sns";
+    private static final String GWS_SEARCH_NO_SUGGESTIONS_PARAM_VALUE = "1";
+    private static final String GWS_QUERY_PARAM = "q";
     private static final String CTXS_PARAM_PATTERN = "(ctxs=[^&]+)";
     private static final String CTXR_PARAM = "ctxr";
+    @VisibleForTesting static final String TLITE_SOURCE_LANGUAGE_PARAM = "tlitesl";
+    private static final String TLITE_TARGET_LANGUAGE_PARAM = "tlitetl";
+    private static final String TLITE_QUERY_PARAM = "tlitetxt";
 
     /**
      * Creates a search request for the given search term without any alternate term and
@@ -53,10 +64,7 @@ class ContextualSearchRequest {
         if (isLowPriorityEnabled) {
             // TODO(donnd): Call TemplateURL once we have an API for 3rd-party providers.
             Uri baseLowPriorityUri = getUriTemplate(searchTerm, alternateTerm, true);
-            mLowPriorityUri = baseLowPriorityUri.buildUpon()
-                .path("s")
-                .appendQueryParameter("sns", "1")
-                .build();
+            mLowPriorityUri = makeLowPriorityUri(baseLowPriorityUri);
             mIsLowPriority = true;
         } else {
             mIsLowPriority = false;
@@ -140,18 +148,94 @@ class ContextualSearchRequest {
     }
 
     /**
+     * Adds translation parameters.
+     * @param sourceLanguage The language of the original search term.
+     * @param targetLanguage The language the that the user prefers.
+     */
+    void forceTranslation(String sourceLanguage, String targetLanguage) {
+        mIsTranslationForced = true;
+        if (mLowPriorityUri != null) {
+            mLowPriorityUri = makeTranslateUri(mLowPriorityUri, sourceLanguage, targetLanguage);
+        }
+        mNormalPriorityUri = makeTranslateUri(mNormalPriorityUri, sourceLanguage, targetLanguage);
+    }
+
+    /**
+     * @return Whether translation was forced for this request.
+     */
+    @VisibleForTesting
+    boolean isTranslationForced() {
+        return mIsTranslationForced;
+    }
+
+    /**
      * Uses TemplateUrlService to generate the url for the given query
      * {@link String} for {@code query} with the contextual search version param set.
      * @param query The search term to use as the main query in the returned search url.
      * @param alternateTerm The alternate search term to use as an alternate suggestion.
      * @param shouldPrefetch Whether the returned url should include a prefetch parameter.
-     * @return      A {@link String} that contains the url of the default search engine with
-     *              {@code query} and {@code alternateTerm} inserted as parameters and contextual
-     *              search and prefetch parameters conditionally set.
+     * @return A {@link Uri} that contains the url of the default search engine with
+     *         {@code query} and {@code alternateTerm} inserted as parameters and contextual
+     *         search and prefetch parameters conditionally set.
      */
     private Uri getUriTemplate(String query, @Nullable String alternateTerm,
             boolean shouldPrefetch) {
         return Uri.parse(TemplateUrlService.getInstance().getUrlForContextualSearchQuery(
                 query, alternateTerm, shouldPrefetch));
     }
+
+    /**
+     * @return a low-priority {@code Uri} from the given base {@code Uri}.
+     */
+    private Uri makeLowPriorityUri(Uri baseUri) {
+        return baseUri.buildUpon()
+                .path(GWS_LOW_PRIORITY_SEARCH_PATH)
+                .appendQueryParameter(
+                        GWS_SEARCH_NO_SUGGESTIONS_PARAM, GWS_SEARCH_NO_SUGGESTIONS_PARAM_VALUE)
+                .build();
+    }
+
+    /**
+     * Makes the given {@code Uri} into a similar Uri that triggers a Translate one-box.
+     * @param baseUri The base Uri to build off of.
+     * @param sourceLanguage The language of the original search term.
+     * @param targetLanguage The language the that the user prefers.
+     * @return A {@link Uri} that has additional parameters for Translate appropriately set.
+     */
+    private Uri makeTranslateUri(Uri baseUri, String sourceLanguage, String targetLanguage) {
+        // TODO(donnd): update to work for non-English.  See also getTranslateQuery.
+        if (!TextUtils.equals(targetLanguage, Locale.ENGLISH.getLanguage())) return baseUri;
+        Uri resultUri = baseUri;
+        if (!sourceLanguage.isEmpty() || !targetLanguage.isEmpty()) {
+            // We must replace the q= param, and there seems to be no good way other than clearing
+            // all the query params and adding them all back in, changing q=.
+            Uri.Builder builder = baseUri.buildUpon().clearQuery();
+            String query = null;
+            for (String param : baseUri.getQueryParameterNames()) {
+                String value = baseUri.getQueryParameter(param);
+                if (TextUtils.equals(param, GWS_QUERY_PARAM)) {
+                    query = value;
+                    value = getTranslateQuery();
+                }
+                builder.appendQueryParameter(param, value);
+            }
+            if (!sourceLanguage.isEmpty()) {
+                builder.appendQueryParameter(TLITE_SOURCE_LANGUAGE_PARAM, sourceLanguage);
+            }
+            if (!targetLanguage.isEmpty()) {
+                builder.appendQueryParameter(TLITE_TARGET_LANGUAGE_PARAM, targetLanguage);
+            }
+            builder.appendQueryParameter(TLITE_QUERY_PARAM, query);
+            resultUri = builder.build();
+        }
+        return resultUri;
+    }
+
+    /**
+     * TODO(donnd): This translate API is evolving.  Update this code!
+     * TODO(donnd): As of Oct 2015 this will only work on production GWS to translate into English.
+     */
+    private String getTranslateQuery() {
+        return "Translate";
+    }
 }