Fix generate_profile and allow it to be built on Linux as well as Windows.

Review URL: https://chromiumcodereview.appspot.com/11573018

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@174024 0039d316-1c4b-4281-b951-d872f2087c98

1 files changed, 42 insertions, 19 deletions

diff --git a/chrome/tools/profiles/generate_profile.cc b/chrome/tools/profiles/generate_profile.cc
index ca4ee3e..2b6a6e6 100644
--- a/chrome/tools/profiles/generate_profile.cc
+++ b/chrome/tools/profiles/generate_profile.cc
@@ -12,6 +12,7 @@
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/i18n/icu_util.h"
+#include "base/logging.h"
 #include "base/message_loop.h"
 #include "base/path_service.h"
 #include "base/process_util.h"
@@ -23,6 +24,7 @@
 #include "chrome/browser/history/top_sites.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/thumbnail_score.h"
+#include "chrome/test/base/testing_browser_process.h"
 #include "chrome/test/base/testing_profile.h"
 #include "content/browser/browser_thread_impl.h"
 #include "content/public/browser/browser_thread.h"
@@ -32,6 +34,10 @@
 #include "ui/base/ui_base_paths.h"
 #include "ui/gfx/codec/jpeg_codec.h"
 
+#if defined(TOOLKIT_GTK)
+#include <gtk/gtk.h>
+#endif
+
 using base::Time;
 using content::BrowserThread;
 
@@ -42,6 +48,21 @@ enum Types {
   FULL_TEXT = 1 << 1
 };
 
+// RAII for initializing and shutting down the TestBrowserProcess
+class InitBrowserProcess {
+ public:
+  InitBrowserProcess() {
+    DCHECK(!g_browser_process);
+    g_browser_process = new TestingBrowserProcess;
+  }
+
+  ~InitBrowserProcess() {
+    DCHECK(g_browser_process);
+    delete g_browser_process;
+    g_browser_process = NULL;
+  }
+};
+
 // Probabilities of different word lengths, as measured from Darin's profile.
 //   kWordLengthProbabilities[n-1] = P(word of length n)
 const float kWordLengthProbabilities[] = { 0.069f, 0.132f, 0.199f,
@@ -60,20 +81,20 @@ int RandomInt(int min, int max) {
 }
 
 // Return a string of |count| lowercase random characters.
-std::wstring RandomChars(int count) {
-  std::wstring str;
+string16 RandomChars(int count) {
+  string16 str;
   for (int i = 0; i < count; ++i)
     str += L'a' + rand() % 26;
   return str;
 }
 
-std::wstring RandomWord() {
+string16 RandomWord() {
   // TODO(evanm): should we instead use the markov chain based
   // version of this that I already wrote?
 
   // Sample a word length from kWordLengthProbabilities.
   float sample = RandomFloat();
-  int i;
+  size_t i;
   for (i = 0; i < arraysize(kWordLengthProbabilities); ++i) {
     sample -= kWordLengthProbabilities[i];
     if (sample < 0) break;
@@ -83,8 +104,8 @@ std::wstring RandomWord() {
 }
 
 // Return a string of |count| random words.
-std::wstring RandomWords(int count) {
-  std::wstring str;
+string16 RandomWords(int count) {
+  string16 str;
   for (int i = 0; i < count; ++i) {
     if (!str.empty())
       str += L' ';
@@ -95,17 +116,17 @@ std::wstring RandomWords(int count) {
 
 // Return a random URL-looking string.
 GURL ConstructRandomURL() {
-  return GURL(std::wstring(L"http://") + RandomChars(3) + L".com/" +
+  return GURL(ASCIIToUTF16("http://") + RandomChars(3) + ASCIIToUTF16(".com/") +
       RandomChars(RandomInt(5, 20)));
 }
 
 // Return a random page title-looking string.
-std::wstring ConstructRandomTitle() {
+string16 ConstructRandomTitle() {
   return RandomWords(RandomInt(3, 15));
 }
 
 // Return a random string that could function as page contents.
-std::wstring ConstructRandomPage() {
+string16 ConstructRandomPage() {
   return RandomWords(RandomInt(10, 4000));
 }
 
@@ -195,7 +216,7 @@ void InsertURLBatch(Profile* profile,
   }
 }
 
-int main(int argc, const char* argv[]) {
+int main(int argc, char* argv[]) {
   CommandLine::Init(argc, argv);
   base::EnableTerminationOnHeapCorruption();
   base::AtExitManager exit_manager;
@@ -218,7 +239,7 @@ int main(int argc, const char* argv[]) {
   }
 
   int url_count = 0;
-  base::StringToInt(WideToUTF8(args[0]), &url_count);
+  base::StringToInt(args[0], &url_count);
   FilePath dst_dir(args[1]);
   if (!dst_dir.IsAbsolute()) {
     FilePath current_dir;
@@ -226,21 +247,22 @@ int main(int argc, const char* argv[]) {
     dst_dir = current_dir.Append(dst_dir);
   }
   if (!file_util::CreateDirectory(dst_dir)) {
-    printf("Unable to create directory %ls: %d\n",
-           dst_dir.value().c_str(),
-           ::GetLastError());
+    PLOG(ERROR) << "Unable to create directory " << dst_dir.value().c_str();
   }
 
   icu_util::Initialize();
+  // Copied from base/test/test_suite.cc.
+#if defined(TOOLKIT_GTK)
+  gtk_init_check(&argc, &argv);
+#endif
 
+  InitBrowserProcess initialize_browser_process;
   chrome::RegisterPathProvider();
   ui::RegisterPathProvider();
-  ResourceBundle::InitSharedInstanceWithLocale("en-US", NULL);
-  scoped_ptr<content::NotificationService> notification_service(
-      content::NotificationService::Create());
   MessageLoopForUI message_loop;
   content::BrowserThreadImpl ui_thread(BrowserThread::UI, &message_loop);
   content::BrowserThreadImpl db_thread(BrowserThread::DB, &message_loop);
+  ResourceBundle::InitSharedInstanceWithLocale("en-US", NULL);
   TestingProfile profile;
   profile.CreateHistoryService(false, false);
   if (types & TOP_SITES) {
@@ -274,10 +296,11 @@ int main(int argc, const char* argv[]) {
   while (!path.empty()) {
     FilePath dst_file = dst_dir.Append(path.BaseName());
     file_util::Delete(dst_file, false);
-    printf("Copying file %ls to %ls\n", path.value().c_str(),
+    printf("Copying file %" PRFilePath " to "
+           "%" PRFilePath "\n", path.value().c_str(),
            dst_file.value().c_str());
     if (!file_util::CopyFile(path, dst_file)) {
-      printf("Copying file failed: %d\n", ::GetLastError());
+      PLOG(ERROR) << "Copying file failed";
       return -1;
     }
     path = file_iterator.Next();