Adds import/export of bookmarks to bookmarks.html file....

chrome/browser/importer/firefox2_importer.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 "chrome/browser/importer/firefox2_importer.h"
 
 #include "base/file_util.h"
 #include "base/path_service.h"
 #include "base/registry.h"
 #include "base/string_util.h"
 #include "base/values.h"
 #include "chrome/browser/importer/firefox_importer_utils.h"
 #include "chrome/browser/importer/mork_reader.h"
 #include "chrome/browser/template_url.h"
 #include "chrome/browser/template_url_parser.h"
 #include "chrome/common/l10n_util.h"
 #include "chrome/common/time_format.h"
 #include "generated_resources.h"
 #include "net/base/data_url.h"
 
 using base::Time;
 
 // Firefox2Importer.
 
-Firefox2Importer::Firefox2Importer() {
+Firefox2Importer::Firefox2Importer() : parsing_bookmarks_html_file_(false) {
 }
 
 Firefox2Importer::~Firefox2Importer() {
 }
 
 void Firefox2Importer::StartImport(ProfileInfo profile_info,
                                    uint16 items, ProfileWriter* writer,
                                    MessageLoop* delagate_loop,
                                    ImporterHost* host) {
   writer_ = writer;
   source_path_ = profile_info.source_path;
   app_path_ = profile_info.app_path;
   importer_host_ = host;
 
+  parsing_bookmarks_html_file_ = (profile_info.browser_type == BOOKMARKS_HTML);
+
   // The order here is important!
   NotifyStarted();
   if ((items & HOME_PAGE) && !cancelled())
     ImportHomepage();  // Doesn't have a UI item.
   if ((items & FAVORITES) && !cancelled()) {
     NotifyItemStarted(FAVORITES);
     ImportBookmarks();
     NotifyItemEnded(FAVORITES);
   }
   if ((items & SEARCH_ENGINES) && !cancelled()) {
@@ skipping 63 matching lines @@
   TemplateURL* t_url = new TemplateURL();
   // We set short name by using the title if it exists.
   // Otherwise, we use the shortcut.
   t_url->set_short_name(!title.empty() ? title : keyword);
   t_url->set_keyword(keyword);
   t_url->SetURL(TemplateURLRef::DisplayURLToURLRef(UTF8ToWide(url.spec())),
                 0, 0);
   return t_url;
 }
 
-void Firefox2Importer::ImportBookmarks() {
-  // Read the whole file.
-  std::wstring file = source_path_;
-  file_util::AppendToPath(&file, L"bookmarks.html");
+// static
+void Firefox2Importer::ImportBookmarksFile(
+    const std::wstring& file_path,
+    const std::set<GURL>& default_urls,
+    bool first_run,
+    const std::wstring& first_folder_name,
+    Importer* importer,
+    std::vector<ProfileWriter::BookmarkEntry>* bookmarks,
+    std::vector<TemplateURL*>* template_urls,
+    std::vector<history::ImportedFavIconUsage>* favicons) {
   std::string content;
-  file_util::ReadFileToString(file, &content);
+  file_util::ReadFileToString(file_path, &content);
   std::vector<std::string> lines;
   SplitString(content, '\n', &lines);
-
-  // Load the default bookmarks.
-  std::set<GURL> default_urls;
-  LoadDefaultBookmarks(app_path_, &default_urls);
-
-  // Parse the bookmarks.html file.
-  std::vector<ProfileWriter::BookmarkEntry> bookmarks, toolbar_bookmarks;
-  std::vector<TemplateURL*> template_urls;
-  std::vector<history::ImportedFavIconUsage> favicons;
-  std::wstring last_folder
-      = l10n_util::GetString(IDS_BOOKMARK_GROUP_FROM_FIREFOX);
+  
+  std::vector<ProfileWriter::BookmarkEntry> toolbar_bookmarks;
+  std::wstring last_folder = first_folder_name;
   bool last_folder_on_toolbar = false;
   std::vector<std::wstring> path;
   size_t toolbar_folder = 0;
   std::string charset;
-  for (size_t i = 0; i < lines.size() && !cancelled(); ++i) {
+  for (size_t i = 0; i < lines.size() && (!importer || !importer->cancelled());
+       ++i) {
     std::string line;
     TrimString(lines[i], " ", &line);
 
     // Get the encoding of the bookmark file.
     if (ParseCharsetFromLine(line, &charset))
       continue;
 
     // Get the folder name.
     if (ParseFolderNameFromLine(line, charset, &last_folder,
                                 &last_folder_on_toolbar))
@@ skipping 15 matching lines @@
       if (toolbar_folder > path.size() && path.size() > 0) {
         NOTREACHED();  // error in parsing.
         break;
       }
 
       ProfileWriter::BookmarkEntry entry;
       entry.creation_time = add_date;
       entry.url = url;
       entry.title = title;
 
-      if (first_run() && toolbar_folder) {
+      if (first_run && toolbar_folder) {
         // Flatten the items in toolbar.
         entry.in_toolbar = true;
         entry.path.assign(path.begin() + toolbar_folder, path.end());
         toolbar_bookmarks.push_back(entry);
       } else {
         // Insert the item into the "Imported from Firefox" folder after
         // the first run.
         entry.path.assign(path.begin(), path.end());
-        if (first_run())
+        if (first_run)
           entry.path.erase(entry.path.begin());
-        bookmarks.push_back(entry);
+        bookmarks->push_back(entry);
       }
 
       // Save the favicon. DataURLToFaviconUsage will handle the case where
       // there is no favicon.
-      DataURLToFaviconUsage(url, favicon, &favicons);
+      if (favicons)
+        DataURLToFaviconUsage(url, favicon, favicons);
 
-      // If there is a SHORTCUT attribute for this bookmark, we
-      // add it as our keywords.
-      TemplateURL* t_url = CreateTemplateURL(title, shortcut, url);
-      if (t_url)
-        template_urls.push_back(t_url);
+      if (template_urls) {
+        // If there is a SHORTCUT attribute for this bookmark, we
+        // add it as our keywords.
+        TemplateURL* t_url = CreateTemplateURL(title, shortcut, url);
+        if (t_url)
+          template_urls->push_back(t_url);
+      }
 
       continue;
     }
 
     // Bookmarks in sub-folder are encapsulated with <DL> tag.
     if (StartsWithASCII(line, "<DL>", true)) {
       path.push_back(last_folder);
       last_folder.clear();
       if (last_folder_on_toolbar && !toolbar_folder)
         toolbar_folder = path.size();
     } else if (StartsWithASCII(line, "</DL>", true)) {
       if (path.empty())
         break;  // Mismatch <DL>.
       path.pop_back();
       if (toolbar_folder > path.size())
         toolbar_folder = 0;
     }
   }
 
+  bookmarks->insert(bookmarks->begin(), toolbar_bookmarks.begin(),
+                    toolbar_bookmarks.end());
+}
+
+void Firefox2Importer::ImportBookmarks() {
+  // Load the default bookmarks.
+  std::set<GURL> default_urls;
+  if (!parsing_bookmarks_html_file_)
+    LoadDefaultBookmarks(app_path_, &default_urls);
+
+  // Parse the bookmarks.html file.
+  std::vector<ProfileWriter::BookmarkEntry> bookmarks, toolbar_bookmarks;
+  std::vector<TemplateURL*> template_urls;
+  std::vector<history::ImportedFavIconUsage> favicons;
+  std::wstring file = source_path_;
+  if (!parsing_bookmarks_html_file_)
+    file_util::AppendToPath(&file, L"bookmarks.html");
+  std::wstring first_folder_name;
+  if (parsing_bookmarks_html_file_)
+    first_folder_name = l10n_util::GetString(IDS_BOOKMARK_GROUP);
+  else
+    first_folder_name = l10n_util::GetString(IDS_BOOKMARK_GROUP_FROM_FIREFOX);
+
+  ImportBookmarksFile(file, default_urls, first_run(),
+                      first_folder_name, this, &bookmarks, &template_urls,
+                      &favicons);
+
   // Write data into profile.
-  bookmarks.insert(bookmarks.begin(), toolbar_bookmarks.begin(),
-                   toolbar_bookmarks.end());
   if (!bookmarks.empty() && !cancelled()) {
     main_loop_->PostTask(FROM_HERE, NewRunnableMethod(writer_,
-        &ProfileWriter::AddBookmarkEntry, bookmarks, false));
+        &ProfileWriter::AddBookmarkEntry, bookmarks,
+        first_folder_name,
+        first_run() ? ProfileWriter::FIRST_RUN : 0));
   }
-  if (!template_urls.empty() && !cancelled()) {
+  if (!parsing_bookmarks_html_file_ && !template_urls.empty() &&
+      !cancelled()) {
     main_loop_->PostTask(FROM_HERE, NewRunnableMethod(writer_,
         &ProfileWriter::AddKeywords, template_urls, -1, false));
   } else {
     STLDeleteContainerPointers(template_urls.begin(), template_urls.end());
   }
   if (!favicons.empty()) {
     main_loop_->PostTask(FROM_HERE, NewRunnableMethod(writer_,
         &ProfileWriter::AddFavicons, favicons));
   }
 }
@@ skipping 266 matching lines @@
 
   // We need to make up a URL for the favicon. We use a version of the page's
   // URL so that we can be sure it will not collide.
   usage.favicon_url = GURL(std::string("made-up-favicon:") + link_url.spec());
 
   // We only have one URL per favicon for Firefox 2 bookmarks.
   usage.urls.insert(link_url);
 
   favicons->push_back(usage);
 }