Translate: make URL handling logic more readable

chrome/browser/translate/translate_manager.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/translate/translate_manager.h"
 
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/compiler_specific.h"
 #include "base/json/json_reader.h"
@@ skipping 32 matching lines @@
 #include "content/public/browser/notification_source.h"
 #include "content/public/browser/notification_types.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/render_view_host.h"
 #include "content/public/browser/web_contents.h"
 #include "google_apis/google_api_keys.h"
 #include "grit/browser_resources.h"
 #include "net/base/escape.h"
 #include "net/base/load_flags.h"
 #include "net/base/url_util.h"
+#include "net/http/http_status_code.h"
 #include "net/url_request/url_fetcher.h"
 #include "net/url_request/url_request_status.h"
 #include "ui/base/resource/resource_bundle.h"
 
 #ifdef FILE_MANAGER_EXTENSION
 #include "chrome/browser/chromeos/extensions/file_manager/file_manager_util.h"
 #include "extensions/common/constants.h"
 #endif
 
 using content::NavigationController;
@@ skipping 66 matching lines @@
     "th",     // Thai
     "tr",     // Turkish
     "uk",     // Ukrainian
     "vi",     // Vietnamese
     "cy",     // Welsh
     "yi",     // Yiddish
 };
 
 const char* const kTranslateScriptURL =
     "https://translate.google.com/translate_a/element.js";
-const char* const kTranslateScriptQuery =
-    "?cb=cr.googleTranslate.onTranslateElementLoad&hl=%s";
 const char* const kTranslateScriptHeader =
     "Google-Translate-Element-Mode: library";
 const char* const kReportLanguageDetectionErrorURL =
-    "https://translate.google.com/translate_error";
+    "https://translate.google.com/translate_error?client=cr&action=langidc";
 const char* const kLanguageListFetchURL =
     "https://translate.googleapis.com/translate_a/l?client=chrome&cb=sl";
 
-const char* const kLanguageListFetchLocaleQueryName = "hl";
-const char* const kLanguageListFetchAlphaLanguageQueryName = "alpha";
-const char* const kLanguageListFetchAlphaLanguageQueryValue = "1";
+// Used in kTranslateScriptURL to request supporting languages list including
+// "alpha languages".
+const char* const kAlphaLanguageQueryName = "alpha";
+const char* const kAlphaLanguageQueryValue = "1";
+
+// Used in all translate URLs to specify API Key.
+const char* const kApiKeyName = "key";
+
+// Used in kTranslateScriptURL to specify a callback function name.
+const char* const kCallbackQueryName = "cb";
+const char* const kCallbackQueryValue =
+    "cr.googleTranslate.onTranslateElementLoad";
+
+// Used in kTranslateScriptURL and kLanguageListFetchURL to specify the
+// application locale.
+const char* const kHostLocaleQueryName = "hl";
+
+// Used in kReportLanguageDetectionErrorURL to specify the original page
+// language.
+const char* const kSourceLanguageQueryName = "sl";
+
+// Used in kReportLanguageDetectionErrorURL to specify the page URL.
+const char* const kUrlQueryName = "u";
 
 const int kMaxRetryLanguageListFetch = 5;
 const int kTranslateScriptExpirationDelayDays = 1;
 
-void AddApiKeyToUrl(GURL* url) {
-  std::string api_key = google_apis::GetAPIKey();
-  std::string query(url->query());
-  if (!query.empty())
-    query += "&";
-  query += "key=" + net::EscapeQueryParamValue(api_key, true);
-  GURL::Replacements replacements;
-  replacements.SetQueryStr(query);
-  *url = url->ReplaceComponents(replacements);
+GURL AddApiKeyToUrl(const GURL& url) {
+  return net::AppendQueryParameter(url, kApiKeyName, google_apis::GetAPIKey());
+}
+
+GURL AddHostLocaleToUrl(const GURL& url) {
+  return net::AppendQueryParameter(
+      url,
+      kHostLocaleQueryName,
+      TranslateManager::GetLanguageCode(
+          g_browser_process->GetApplicationLocale()));
 }
 
 }  // namespace
 
 // This must be kept in sync with the &cb= value in the kLanguageListFetchURL.
 const char* const TranslateManager::kLanguageListCallbackName = "sl(";
 const char* const TranslateManager::kTargetLanguagesKey = "tl";
 
 // static
 base::LazyInstance<std::set<std::string> >
@@ skipping 131 matching lines @@
       }
 
       TranslateTabHelper* translate_tab_helper =
           TranslateTabHelper::FromWebContents(controller->GetWebContents());
       if (!translate_tab_helper)
         return;
 
       // If the navigation happened while offline don't show the translate
       // bar since there will be nothing to translate.
       if (load_details->http_status_code == 0 ||
-          load_details->http_status_code == 500) {
+          load_details->http_status_code == net::HTTP_INTERNAL_SERVER_ERROR) {
         return;
       }
 
       if (!load_details->is_main_frame &&
           translate_tab_helper->language_state().translation_declined()) {
         // Some sites (such as Google map) may trigger sub-frame navigations
         // when the user interacts with the page.  We don't want to show a new
         // infobar if the user already dismissed one in that case.
         return;
       }
@@ skipping 74 matching lines @@
   if (translate_script_request_pending_.get() != source &&
       language_list_request_pending_.get() != source) {
     // Looks like crash on Mac is possibly caused with callback entering here
     // with unknown fetcher when network is refreshed.
     scoped_ptr<const net::URLFetcher> delete_ptr(source);
     return;
   }
 
   bool error =
       (source->GetStatus().status() != net::URLRequestStatus::SUCCESS ||
-      source->GetResponseCode() != 200);
+      source->GetResponseCode() != net::HTTP_OK);
   if (translate_script_request_pending_.get() == source) {
     scoped_ptr<const net::URLFetcher> delete_ptr(
         translate_script_request_pending_.release());
     if (!error) {
       base::StringPiece str = ResourceBundle::GetSharedInstance().
           GetRawDataResource(IDR_TRANSLATE_JS);
       DCHECK(translate_script_.empty());
       str.CopyToString(&translate_script_);
       std::string argument = "('";
       std::string api_key = google_apis::GetAPIKey();
@@ skipping 210 matching lines @@
 
 void TranslateManager::ReportLanguageDetectionError(WebContents* web_contents) {
   UMA_HISTOGRAM_COUNTS("Translate.ReportLanguageDetectionError", 1);
   // We'll open the URL in a new tab so that the user can tell us more.
   Browser* browser = chrome::FindBrowserWithWebContents(web_contents);
   if (!browser) {
     NOTREACHED();
     return;
   }
 
+  GURL report_error_url = GURL(kReportLanguageDetectionErrorURL);
+
   GURL page_url = web_contents->GetController().GetActiveEntry()->GetURL();
-  // Report option should be disabled for secure URLs.
-  DCHECK(!page_url.SchemeIsSecure());
-  std::string report_error_url_str(kReportLanguageDetectionErrorURL);
-  report_error_url_str += "?client=cr&action=langidc&u=";
-  report_error_url_str += net::EscapeUrlEncodedData(page_url.spec(), true);
-  report_error_url_str += "&sl=";
+  report_error_url = net::AppendQueryParameter(
+      report_error_url,
+      kUrlQueryName,
+      page_url.spec());
 
   TranslateTabHelper* translate_tab_helper =
       TranslateTabHelper::FromWebContents(web_contents);
-  report_error_url_str +=
-      translate_tab_helper->language_state().original_language();
-  report_error_url_str += "&hl=";
-  report_error_url_str +=
-      GetLanguageCode(g_browser_process->GetApplicationLocale());
+  report_error_url = net::AppendQueryParameter(
+      report_error_url,
+      kSourceLanguageQueryName,
+      translate_tab_helper->language_state().original_language());
 
-  GURL report_error_url(report_error_url_str);
-  AddApiKeyToUrl(&report_error_url);
+  report_error_url = AddHostLocaleToUrl(report_error_url);
+  report_error_url = AddApiKeyToUrl(report_error_url);
+
   chrome::AddSelectedTabWithURL(browser, report_error_url,
                                 content::PAGE_TRANSITION_AUTO_BOOKMARK);
 }
 
 void TranslateManager::DoTranslatePage(WebContents* web_contents,
                                        const std::string& translate_script,
                                        const std::string& source_lang,
                                        const std::string& target_lang) {
   NavigationEntry* entry = web_contents->GetController().GetActiveEntry();
   if (!entry) {
@@ skipping 104 matching lines @@
 
   // We don't want to do this when translate is disabled.
   DCHECK(prefs != NULL);
   if (CommandLine::ForCurrentProcess()->HasSwitch(
       switches::kDisableTranslate) ||
       (prefs != NULL && !prefs->GetBoolean(prefs::kEnableTranslate))) {
     return;
   }
 
   GURL language_list_fetch_url = GURL(kLanguageListFetchURL);
-  language_list_fetch_url = net::AppendQueryParameter(
-      language_list_fetch_url,
-      kLanguageListFetchLocaleQueryName,
-      GetLanguageCode(g_browser_process->GetApplicationLocale()).c_str());
+  language_list_fetch_url = AddHostLocaleToUrl(language_list_fetch_url);
+  language_list_fetch_url = AddApiKeyToUrl(language_list_fetch_url);
 
   const CommandLine& command_line = *CommandLine::ForCurrentProcess();
   if (command_line.HasSwitch(switches::kEnableTranslateAlphaLanguages)) {
     language_list_fetch_url = net::AppendQueryParameter(
         language_list_fetch_url,
-        kLanguageListFetchAlphaLanguageQueryName,
-        kLanguageListFetchAlphaLanguageQueryValue);
+        kAlphaLanguageQueryName,
+        kAlphaLanguageQueryValue);
   }
-  AddApiKeyToUrl(&language_list_fetch_url);
+
   VLOG(9) << "Fetch supporting language list from: "
           << language_list_fetch_url.spec().c_str();
 
   language_list_request_pending_.reset(net::URLFetcher::Create(
       1, language_list_fetch_url, net::URLFetcher::GET, this));
   language_list_request_pending_->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES |
                                                net::LOAD_DO_NOT_SAVE_COOKIES);
   language_list_request_pending_->SetRequestContext(
       g_browser_process->system_request_context());
   language_list_request_pending_->SetMaxRetriesOn5xx(
       kMaxRetryLanguageListFetch);
   language_list_request_pending_->Start();
 }
 
 void TranslateManager::CleanupPendingUlrFetcher() {
   language_list_request_pending_.reset();
   translate_script_request_pending_.reset();
 }
 
 void TranslateManager::RequestTranslateScript() {
   if (translate_script_request_pending_.get() != NULL)
     return;
 
   GURL translate_script_url;
-  std::string translate_script;
   // Check if command-line contains an alternative URL for translate service.
   const CommandLine& command_line = *CommandLine::ForCurrentProcess();
   if (command_line.HasSwitch(switches::kTranslateScriptURL)) {
-    translate_script = std::string(
+    translate_script_url = GURL(
         command_line.GetSwitchValueASCII(switches::kTranslateScriptURL));
-    translate_script_url = GURL(translate_script.c_str());
     if (!translate_script_url.is_valid() ||
         !translate_script_url.query().empty()) {
       LOG(WARNING) << "The following translate URL specified at the "
-                   << "command-line is invalid: " << translate_script;
-      translate_script.clear();
+                   << "command-line is invalid: "
+                   << translate_script_url.spec();
+      translate_script_url = GURL();
     }
   }
   // Use default URL when command-line argument is not specified, or specified
   // URL is invalid.
-  if (translate_script.empty())
-    translate_script = std::string(kTranslateScriptURL);
+  if (translate_script_url.is_empty())
+    translate_script_url = GURL(kTranslateScriptURL);
 
-  translate_script += std::string(kTranslateScriptQuery);
-  translate_script_url = GURL(base::StringPrintf(
-      translate_script.c_str(),
-      GetLanguageCode(g_browser_process->GetApplicationLocale()).c_str()));
-  AddApiKeyToUrl(&translate_script_url);
+  translate_script_url = net::AppendQueryParameter(
+      translate_script_url,
+      kCallbackQueryName,
+      kCallbackQueryValue);
+  translate_script_url = AddHostLocaleToUrl(translate_script_url);
+  translate_script_url = AddApiKeyToUrl(translate_script_url);
+
   translate_script_request_pending_.reset(net::URLFetcher::Create(
       0, translate_script_url, net::URLFetcher::GET, this));
   translate_script_request_pending_->SetLoadFlags(
       net::LOAD_DO_NOT_SEND_COOKIES | net::LOAD_DO_NOT_SAVE_COOKIES);
   translate_script_request_pending_->SetRequestContext(
       g_browser_process->system_request_context());
   translate_script_request_pending_->SetExtraRequestHeaders(
       kTranslateScriptHeader);
   translate_script_request_pending_->Start();
 }
@@ skipping 15 matching lines @@
   // list or not at all if no such candidate exists
   std::vector<std::string>::iterator iter;
   for (iter = accept_langs_list.begin();
        iter != accept_langs_list.end(); ++iter) {
     std::string lang_code = GetLanguageCode(*iter);
     if (IsSupportedLanguage(lang_code))
       return lang_code;
   }
   return std::string();
 }