Allow passing in a file path to a file that will be used for the new new tab...

chrome/browser/dom_ui/new_tab_ui.cc

 // Copyright (c) 2006-2008 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 "build/build_config.h"
 
 #include "chrome/browser/dom_ui/new_tab_ui.h"
 
 #include "app/l10n_util.h"
 #include "app/resource_bundle.h"
 #include "base/command_line.h"
+#include "base/file_util.h"
 #include "base/histogram.h"
 #include "base/string_piece.h"
 #include "chrome/browser/bookmarks/bookmark_model.h"
 #include "chrome/browser/bookmarks/bookmark_utils.h"
 #include "chrome/browser/browser.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/dom_ui/dom_ui_favicon_source.h"
 #include "chrome/browser/dom_ui/dom_ui_thumbnail_source.h"
 #include "chrome/browser/dom_ui/dom_ui_theme_source.h"
 #include "chrome/browser/dom_ui/downloads_dom_handler.h"
@@ skipping 170 matching lines @@
 
   virtual std::string GetMimeType(const std::string&) const {
     return "text/html";
   }
 
   // Setters and getters for first_view.
   static void set_first_view(bool first_view) { first_view_ = first_view; }
   static bool first_view() { return first_view_; }
 
  private:
+  // In case a file path to the new new tab page was provided this tries to load
+  // the file and returns the file content if successful. This returns an empty
+  // string in case of failure.
+  static std::string GetNewNewTabFromCommandLine();
+
   // Whether this is the is the first viewing of the new tab page and
   // we think it is the user's startup page.
   static bool first_view_;
 
   DISALLOW_COPY_AND_ASSIGN(NewTabHTMLSource);
 };
 
 bool NewTabHTMLSource::first_view_ = true;
 
 NewTabHTMLSource::NewTabHTMLSource()
@@ skipping 64 matching lines @@
   SetFontAndTextDirection(&localized_strings);
 
   // Let the tab know whether it's the first tab being viewed.
   localized_strings.SetString(L"firstview",
                               first_view_ ? L"true" : std::wstring());
   first_view_ = false;
 
 #ifdef CHROME_PERSONALIZATION
   localized_strings.SetString(L"p13nsrc", Personalization::GetNewTabSource());
 #endif
-  static const StringPiece new_tab_html(
-      ResourceBundle::GetSharedInstance().GetRawDataResource(
-          NewTabUI::EnableNewNewTabPage() ?
-              IDR_NEW_NEW_TAB_HTML : IDR_NEW_TAB_HTML));
+
+  // In case we have the new new tab page enabled we first try to read the file
+  // provided on the command line. If that fails we just get the resource from
+  // the resource bundle.
+  StringPiece new_tab_html;
+  std::string new_tab_html_str;
+  if (NewTabUI::EnableNewNewTabPage()) {
+    new_tab_html_str = GetNewNewTabFromCommandLine();
+
+    if (!new_tab_html_str.empty()) {
+      new_tab_html = StringPiece(new_tab_html_str);
+    } else {
+      // Use the new new tab page from the resource bundle.
+      new_tab_html = ResourceBundle::GetSharedInstance().GetRawDataResource(
+              IDR_NEW_NEW_TAB_HTML);
+    }
+  } else {
+    // Use the default new tab page resource.
+    new_tab_html = ResourceBundle::GetSharedInstance().GetRawDataResource(
+        IDR_NEW_TAB_HTML);
+  }
 
   const std::string full_html = jstemplate_builder::GetTemplateHtml(
       new_tab_html, &localized_strings, "t" /* template root node id */);
 
   scoped_refptr<RefCountedBytes> html_bytes(new RefCountedBytes);
   html_bytes->data.resize(full_html.size());
   std::copy(full_html.begin(), full_html.end(), html_bytes->data.begin());
 
   SendResponse(request_id, html_bytes);
 }
 
+// static
+std::string NewTabHTMLSource::GetNewNewTabFromCommandLine() {
+  const CommandLine* command_line = CommandLine::ForCurrentProcess();
+  const std::wstring file_path_wstring = command_line->GetSwitchValue(
+      switches::kNewNewTabPage);
+
+#if defined(OS_WIN)
+  const FilePath::StringType file_path = file_path_wstring;
+#else
+  const FilePath::StringType file_path = WideToASCII(file_path_wstring);
+#endif
+
+  if (!file_path.empty()) {
+    std::string file_contents;
+    if (file_util::ReadFileToString(FilePath(file_path), &file_contents))
+      return file_contents;
+  }
+
+  return std::string();
+}
+
 ///////////////////////////////////////////////////////////////////////////////
 // IncognitoTabHTMLSource
 
 class IncognitoTabHTMLSource : public ChromeURLDataManager::DataSource {
  public:
   // Creates our datasource and sets our user message to a specific message
   // from our string bundle.
   IncognitoTabHTMLSource();
 
   // Called when the network layer has requested a resource underneath
@@ skipping 899 matching lines @@
 // static
 void NewTabUI::RegisterUserPrefs(PrefService* prefs) {
   MostVisitedHandler::RegisterUserPrefs(prefs);
 }
 
 // static
 bool NewTabUI::EnableNewNewTabPage() {
   const CommandLine* command_line = CommandLine::ForCurrentProcess();
   return command_line->HasSwitch(switches::kNewNewTabPage);
 }