| Index: chrome/browser/importer.cc
|
| ===================================================================
|
| --- chrome/browser/importer.cc (revision 2150)
|
| +++ chrome/browser/importer.cc (working copy)
|
| @@ -1,541 +0,0 @@
|
| -// 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.h"
|
| -
|
| -#include <map>
|
| -
|
| -#include "base/file_util.h"
|
| -#include "base/gfx/image_operations.h"
|
| -#include "base/gfx/png_encoder.h"
|
| -#include "base/string_util.h"
|
| -#include "chrome/browser/bookmarks/bookmark_model.h"
|
| -#include "chrome/browser/browser_process.h"
|
| -#include "chrome/browser/firefox2_importer.h"
|
| -#include "chrome/browser/firefox3_importer.h"
|
| -#include "chrome/browser/firefox_importer_utils.h"
|
| -#include "chrome/browser/firefox_profile_lock.h"
|
| -#include "chrome/browser/ie_importer.h"
|
| -#include "chrome/browser/template_url_model.h"
|
| -#include "chrome/browser/shell_integration.h"
|
| -#include "chrome/browser/webdata/web_data_service.h"
|
| -#include "chrome/common/gfx/favicon_size.h"
|
| -#include "chrome/common/l10n_util.h"
|
| -#include "chrome/common/pref_names.h"
|
| -#include "chrome/common/pref_service.h"
|
| -#include "chrome/views/window.h"
|
| -#include "webkit/glue/image_decoder.h"
|
| -
|
| -#include "generated_resources.h"
|
| -
|
| -// ProfileWriter.
|
| -
|
| -bool ProfileWriter::BookmarkModelIsLoaded() const {
|
| - return profile_->GetBookmarkModel()->IsLoaded();
|
| -}
|
| -
|
| -void ProfileWriter::AddBookmarkModelObserver(BookmarkModelObserver* observer) {
|
| - profile_->GetBookmarkModel()->AddObserver(observer);
|
| -}
|
| -
|
| -bool ProfileWriter::TemplateURLModelIsLoaded() const {
|
| - return profile_->GetTemplateURLModel()->loaded();
|
| -}
|
| -
|
| -void ProfileWriter::AddTemplateURLModelObserver(
|
| - NotificationObserver* observer) {
|
| - TemplateURLModel* model = profile_->GetTemplateURLModel();
|
| - NotificationService::current()->AddObserver(
|
| - observer, TEMPLATE_URL_MODEL_LOADED,
|
| - Source<TemplateURLModel>(model));
|
| - model->Load();
|
| -}
|
| -
|
| -void ProfileWriter::AddPasswordForm(const PasswordForm& form) {
|
| - profile_->GetWebDataService(Profile::EXPLICIT_ACCESS)->AddLogin(form);
|
| -}
|
| -
|
| -void ProfileWriter::AddIE7PasswordInfo(const IE7PasswordInfo& info) {
|
| - profile_->GetWebDataService(Profile::EXPLICIT_ACCESS)->AddIE7Login(info);
|
| -}
|
| -
|
| -void ProfileWriter::AddHistoryPage(const std::vector<history::URLRow>& page) {
|
| - profile_->GetHistoryService(Profile::EXPLICIT_ACCESS)->
|
| - AddPagesWithDetails(page);
|
| -}
|
| -
|
| -void ProfileWriter::AddHomepage(const GURL& home_page) {
|
| - DCHECK(profile_);
|
| -
|
| - PrefService* prefs = profile_->GetPrefs();
|
| - // NOTE: We set the kHomePage value, but keep the NewTab page as the homepage.
|
| - prefs->SetString(prefs::kHomePage, ASCIIToWide(home_page.spec()));
|
| - prefs->ScheduleSavePersistentPrefs(g_browser_process->file_thread());
|
| -}
|
| -
|
| -void ProfileWriter::AddBookmarkEntry(
|
| - const std::vector<BookmarkEntry>& bookmark) {
|
| - BookmarkModel* model = profile_->GetBookmarkModel();
|
| - DCHECK(model->IsLoaded());
|
| -
|
| - bool show_bookmark_toolbar = false;
|
| - std::set<BookmarkNode*> groups_added_to;
|
| - for (std::vector<BookmarkEntry>::const_iterator it = bookmark.begin();
|
| - it != bookmark.end(); ++it) {
|
| - // Don't insert this url if it exists in model or url is not valid.
|
| - if (model->GetNodeByURL(it->url) != NULL || !it->url.is_valid())
|
| - continue;
|
| -
|
| - // Set up groups in BookmarkModel in such a way that path[i] is
|
| - // the subgroup of path[i-1]. Finally they construct a path in the
|
| - // model:
|
| - // path[0] \ path[1] \ ... \ path[size() - 1]
|
| - BookmarkNode* parent =
|
| - (it->in_toolbar ? model->GetBookmarkBarNode() : model->other_node());
|
| - for (std::vector<std::wstring>::const_iterator i = it->path.begin();
|
| - i != it->path.end(); ++i) {
|
| - BookmarkNode* child = NULL;
|
| - for (int index = 0; index < parent->GetChildCount(); ++index) {
|
| - BookmarkNode* node = parent->GetChild(index);
|
| - if ((node->GetType() == history::StarredEntry::BOOKMARK_BAR ||
|
| - node->GetType() == history::StarredEntry::USER_GROUP) &&
|
| - node->GetTitle() == *i) {
|
| - child = node;
|
| - break;
|
| - }
|
| - }
|
| - if (child == NULL)
|
| - child = model->AddGroup(parent, parent->GetChildCount(), *i);
|
| - parent = child;
|
| - }
|
| - groups_added_to.insert(parent);
|
| - model->AddURLWithCreationTime(parent, parent->GetChildCount(),
|
| - it->title, it->url, it->creation_time);
|
| -
|
| - // If some items are put into toolbar, it looks like the user was using
|
| - // it in their last browser. We turn on the bookmarks toolbar.
|
| - if (it->in_toolbar)
|
| - show_bookmark_toolbar = true;
|
| - }
|
| -
|
| - // Reset the date modified time of the groups we added to. We do this to
|
| - // make sure the 'recently added to' combobox in the bubble doesn't get random
|
| - // groups.
|
| - for (std::set<BookmarkNode*>::const_iterator i = groups_added_to.begin();
|
| - i != groups_added_to.end(); ++i) {
|
| - model->ResetDateGroupModified(*i);
|
| - }
|
| -
|
| - if (show_bookmark_toolbar)
|
| - ShowBookmarkBar();
|
| -}
|
| -
|
| -void ProfileWriter::AddFavicons(
|
| - const std::vector<history::ImportedFavIconUsage>& favicons) {
|
| - profile_->GetHistoryService(Profile::EXPLICIT_ACCESS)->
|
| - SetImportedFavicons(favicons);
|
| -}
|
| -
|
| -typedef std::map<std::string, const TemplateURL*> HostPathMap;
|
| -
|
| -// Builds the key to use in HostPathMap for the specified TemplateURL. Returns
|
| -// an empty string if a host+path can't be generated for the TemplateURL.
|
| -// If an empty string is returned, it should not be added to HostPathMap.
|
| -static std::string BuildHostPathKey(const TemplateURL* t_url) {
|
| - if (t_url->url() && t_url->url()->SupportsReplacement()) {
|
| - GURL search_url(t_url->url()->ReplaceSearchTerms(
|
| - *t_url, L"random string", TemplateURLRef::NO_SUGGESTIONS_AVAILABLE,
|
| - std::wstring()));
|
| - if (search_url.is_valid())
|
| - return search_url.host() + search_url.path();
|
| - }
|
| - return std::string();
|
| -}
|
| -
|
| -// Builds a set that contains an entry of the host+path for each TemplateURL in
|
| -// the TemplateURLModel that has a valid search url.
|
| -static void BuildHostPathMap(const TemplateURLModel& model,
|
| - HostPathMap* host_path_map) {
|
| - std::vector<const TemplateURL*> template_urls = model.GetTemplateURLs();
|
| - for (size_t i = 0; i < template_urls.size(); ++i) {
|
| - const std::string host_path = BuildHostPathKey(template_urls[i]);
|
| - if (!host_path.empty()) {
|
| - const TemplateURL* existing_turl = (*host_path_map)[host_path];
|
| - if (!existing_turl ||
|
| - (template_urls[i]->show_in_default_list() &&
|
| - !existing_turl->show_in_default_list())) {
|
| - // If there are multiple TemplateURLs with the same host+path, favor
|
| - // those shown in the default list. If there are multiple potential
|
| - // defaults, favor the first one, which should be the more commonly used
|
| - // one.
|
| - (*host_path_map)[host_path] = template_urls[i];
|
| - }
|
| - } // else case, TemplateURL doesn't have a search url, doesn't support
|
| - // replacement, or doesn't have valid GURL. Ignore it.
|
| - }
|
| -}
|
| -
|
| -void ProfileWriter::AddKeywords(const std::vector<TemplateURL*>& template_urls,
|
| - int default_keyword_index,
|
| - bool unique_on_host_and_path) {
|
| - TemplateURLModel* model = profile_->GetTemplateURLModel();
|
| - HostPathMap host_path_map;
|
| - if (unique_on_host_and_path)
|
| - BuildHostPathMap(*model, &host_path_map);
|
| -
|
| - for (std::vector<TemplateURL*>::const_iterator i = template_urls.begin();
|
| - i != template_urls.end(); ++i) {
|
| - TemplateURL* t_url = *i;
|
| - bool default_keyword =
|
| - default_keyword_index >= 0 &&
|
| - (i - template_urls.begin() == default_keyword_index);
|
| -
|
| - // TemplateURLModel requires keywords to be unique. If there is already a
|
| - // TemplateURL with this keyword, don't import it again.
|
| - const TemplateURL* turl_with_keyword =
|
| - model->GetTemplateURLForKeyword(t_url->keyword());
|
| - if (turl_with_keyword != NULL) {
|
| - if (default_keyword)
|
| - model->SetDefaultSearchProvider(turl_with_keyword);
|
| - delete t_url;
|
| - continue;
|
| - }
|
| -
|
| - // For search engines if there is already a keyword with the same
|
| - // host+path, we don't import it. This is done to avoid both duplicate
|
| - // search providers (such as two Googles, or two Yahoos) as well as making
|
| - // sure the search engines we provide aren't replaced by those from the
|
| - // imported browser.
|
| - if (unique_on_host_and_path &&
|
| - host_path_map.find(BuildHostPathKey(t_url)) != host_path_map.end()) {
|
| - if (default_keyword) {
|
| - const TemplateURL* turl_with_host_path =
|
| - host_path_map[BuildHostPathKey(t_url)];
|
| - if (turl_with_host_path)
|
| - model->SetDefaultSearchProvider(turl_with_host_path);
|
| - else
|
| - NOTREACHED(); // BuildHostPathMap should only insert non-null values.
|
| - }
|
| - delete t_url;
|
| - continue;
|
| - }
|
| - model->Add(t_url);
|
| - if (default_keyword)
|
| - model->SetDefaultSearchProvider(t_url);
|
| - }
|
| -}
|
| -
|
| -void ProfileWriter::ShowBookmarkBar() {
|
| - DCHECK(profile_);
|
| -
|
| - PrefService* prefs = profile_->GetPrefs();
|
| - // Check whether the bookmark bar is shown in current pref.
|
| - if (!prefs->GetBoolean(prefs::kShowBookmarkBar)) {
|
| - // Set the pref and notify the notification service.
|
| - prefs->SetBoolean(prefs::kShowBookmarkBar, true);
|
| - prefs->ScheduleSavePersistentPrefs(g_browser_process->file_thread());
|
| - Source<Profile> source(profile_);
|
| - NotificationService::current()->Notify(
|
| - NOTIFY_BOOKMARK_BAR_VISIBILITY_PREF_CHANGED, source,
|
| - NotificationService::NoDetails());
|
| - }
|
| -}
|
| -
|
| -// Importer.
|
| -
|
| -// static
|
| -bool Importer::ReencodeFavicon(const unsigned char* src_data, size_t src_len,
|
| - std::vector<unsigned char>* png_data) {
|
| - // Decode the favicon using WebKit's image decoder.
|
| - webkit_glue::ImageDecoder decoder(gfx::Size(kFavIconSize, kFavIconSize));
|
| - SkBitmap decoded = decoder.Decode(src_data, src_len);
|
| - if (decoded.empty())
|
| - return false; // Unable to decode.
|
| -
|
| - if (decoded.width() != kFavIconSize || decoded.height() != kFavIconSize) {
|
| - // The bitmap is not the correct size, re-sample.
|
| - int new_width = decoded.width();
|
| - int new_height = decoded.height();
|
| - calc_favicon_target_size(&new_width, &new_height);
|
| - decoded = gfx::ImageOperations::Resize(
|
| - decoded, gfx::ImageOperations::RESIZE_LANCZOS3,
|
| - gfx::Size(new_width, new_height));
|
| - }
|
| -
|
| - // Encode our bitmap as a PNG.
|
| - SkAutoLockPixels decoded_lock(decoded);
|
| - PNGEncoder::Encode(reinterpret_cast<unsigned char*>(decoded.getPixels()),
|
| - PNGEncoder::FORMAT_BGRA, decoded.width(),
|
| - decoded.height(), decoded.width() * 4, false, png_data);
|
| - return true;
|
| -}
|
| -
|
| -// ImporterHost.
|
| -
|
| -ImporterHost::ImporterHost()
|
| - : observer_(NULL),
|
| - task_(NULL),
|
| - importer_(NULL),
|
| - file_loop_(g_browser_process->file_thread()->message_loop()),
|
| - waiting_for_bookmarkbar_model_(false),
|
| - waiting_for_template_url_model_(false),
|
| - is_source_readable_(true) {
|
| - DetectSourceProfiles();
|
| -}
|
| -
|
| -ImporterHost::ImporterHost(MessageLoop* file_loop)
|
| - : observer_(NULL),
|
| - task_(NULL),
|
| - importer_(NULL),
|
| - file_loop_(file_loop),
|
| - waiting_for_bookmarkbar_model_(false),
|
| - waiting_for_template_url_model_(false),
|
| - is_source_readable_(true) {
|
| - DetectSourceProfiles();
|
| -}
|
| -
|
| -ImporterHost::~ImporterHost() {
|
| - STLDeleteContainerPointers(source_profiles_.begin(), source_profiles_.end());
|
| -}
|
| -
|
| -void ImporterHost::Loaded(BookmarkModel* model) {
|
| - model->RemoveObserver(this);
|
| - waiting_for_bookmarkbar_model_ = false;
|
| - InvokeTaskIfDone();
|
| -}
|
| -
|
| -void ImporterHost::Observe(NotificationType type,
|
| - const NotificationSource& source,
|
| - const NotificationDetails& details) {
|
| - DCHECK(type == TEMPLATE_URL_MODEL_LOADED);
|
| - TemplateURLModel* model = Source<TemplateURLModel>(source).ptr();
|
| - NotificationService::current()->RemoveObserver(
|
| - this, TEMPLATE_URL_MODEL_LOADED,
|
| - Source<TemplateURLModel>(model));
|
| - waiting_for_template_url_model_ = false;
|
| - InvokeTaskIfDone();
|
| -}
|
| -
|
| -void ImporterHost::ShowWarningDialog() {
|
| - ChromeViews::Window::CreateChromeWindow(GetActiveWindow(), gfx::Rect(),
|
| - new ImporterLockView(this))->Show();
|
| -}
|
| -
|
| -void ImporterHost::OnLockViewEnd(bool is_continue) {
|
| - if (is_continue) {
|
| - // User chose to continue, then we check the lock again to make
|
| - // sure that Firefox has been closed. Try to import the settings
|
| - // if successful. Otherwise, show a warning dialog.
|
| - firefox_lock_->Lock();
|
| - if (firefox_lock_->HasAcquired()) {
|
| - is_source_readable_ = true;
|
| - InvokeTaskIfDone();
|
| - } else {
|
| - ShowWarningDialog();
|
| - }
|
| - } else {
|
| - // User chose to skip the import process. We should delete
|
| - // the task and notify the ImporterHost to finish.
|
| - delete task_;
|
| - task_ = NULL;
|
| - importer_ = NULL;
|
| - ImportEnded();
|
| - }
|
| -}
|
| -
|
| -void ImporterHost::StartImportSettings(const ProfileInfo& profile_info,
|
| - uint16 items,
|
| - ProfileWriter* writer,
|
| - bool first_run) {
|
| - // Preserves the observer and creates a task, since we do async import
|
| - // so that it doesn't block the UI. When the import is complete, observer
|
| - // will be notified.
|
| - writer_ = writer;
|
| - importer_ = CreateImporterByType(profile_info.browser_type);
|
| - importer_->set_first_run(first_run);
|
| - task_ = NewRunnableMethod(importer_, &Importer::StartImport,
|
| - profile_info, items, writer_.get(), this);
|
| -
|
| - // We should lock the Firefox profile directory to prevent corruption.
|
| - if (profile_info.browser_type == FIREFOX2 ||
|
| - profile_info.browser_type == FIREFOX3) {
|
| - firefox_lock_.reset(new FirefoxProfileLock(profile_info.source_path));
|
| - if (!firefox_lock_->HasAcquired()) {
|
| - // If fail to acquire the lock, we set the source unreadable and
|
| - // show a warning dialog.
|
| - is_source_readable_ = false;
|
| - ShowWarningDialog();
|
| - }
|
| - }
|
| -
|
| - // BookmarkModel should be loaded before adding IE favorites. So we observe
|
| - // the BookmarkModel if needed, and start the task after it has been loaded.
|
| - if ((items & FAVORITES) && !writer_->BookmarkModelIsLoaded()) {
|
| - writer_->AddBookmarkModelObserver(this);
|
| - waiting_for_bookmarkbar_model_ = true;
|
| - }
|
| -
|
| - // Observes the TemplateURLModel if needed to import search engines from the
|
| - // other browser. We also check to see if we're importing bookmarks because
|
| - // we can import bookmark keywords from Firefox as search engines.
|
| - if ((items & SEARCH_ENGINES) || (items & FAVORITES)) {
|
| - if (!writer_->TemplateURLModelIsLoaded()) {
|
| - writer_->AddTemplateURLModelObserver(this);
|
| - waiting_for_template_url_model_ = true;
|
| - }
|
| - }
|
| -
|
| - AddRef();
|
| - InvokeTaskIfDone();
|
| -}
|
| -
|
| -void ImporterHost::Cancel() {
|
| - if (importer_)
|
| - importer_->Cancel();
|
| -}
|
| -
|
| -void ImporterHost::SetObserver(Observer* observer) {
|
| - observer_ = observer;
|
| -}
|
| -
|
| -void ImporterHost::InvokeTaskIfDone() {
|
| - if (waiting_for_bookmarkbar_model_ || waiting_for_template_url_model_ ||
|
| - !is_source_readable_)
|
| - return;
|
| - file_loop_->PostTask(FROM_HERE, task_);
|
| -}
|
| -
|
| -void ImporterHost::ImportItemStarted(ImportItem item) {
|
| - if (observer_)
|
| - observer_->ImportItemStarted(item);
|
| -}
|
| -
|
| -void ImporterHost::ImportItemEnded(ImportItem item) {
|
| - if (observer_)
|
| - observer_->ImportItemEnded(item);
|
| -}
|
| -
|
| -void ImporterHost::ImportStarted() {
|
| - if (observer_)
|
| - observer_->ImportStarted();
|
| -}
|
| -
|
| -void ImporterHost::ImportEnded() {
|
| - firefox_lock_.reset(); // Release the Firefox profile lock.
|
| - if (observer_)
|
| - observer_->ImportEnded();
|
| - Release();
|
| -}
|
| -
|
| -Importer* ImporterHost::CreateImporterByType(ProfileType type) {
|
| - switch (type) {
|
| - case MS_IE:
|
| - return new IEImporter();
|
| - case FIREFOX2:
|
| - return new Firefox2Importer();
|
| - case FIREFOX3:
|
| - return new Firefox3Importer();
|
| - }
|
| - NOTREACHED();
|
| - return NULL;
|
| -}
|
| -
|
| -int ImporterHost::GetAvailableProfileCount() {
|
| - return static_cast<int>(source_profiles_.size());
|
| -}
|
| -
|
| -std::wstring ImporterHost::GetSourceProfileNameAt(int index) const {
|
| - DCHECK(index < static_cast<int>(source_profiles_.size()));
|
| - return source_profiles_[index]->description;
|
| -}
|
| -
|
| -const ProfileInfo& ImporterHost::GetSourceProfileInfoAt(int index) const {
|
| - DCHECK(index < static_cast<int>(source_profiles_.size()));
|
| - return *source_profiles_[index];
|
| -}
|
| -
|
| -void ImporterHost::DetectSourceProfiles() {
|
| - if (ShellIntegration::IsFirefoxDefaultBrowser()) {
|
| - DetectFirefoxProfiles();
|
| - DetectIEProfiles();
|
| - } else {
|
| - DetectIEProfiles();
|
| - DetectFirefoxProfiles();
|
| - }
|
| -}
|
| -
|
| -void ImporterHost::DetectIEProfiles() {
|
| - // IE always exists and don't have multiple profiles.
|
| - ProfileInfo* ie = new ProfileInfo();
|
| - ie->description = l10n_util::GetString(IDS_IMPORT_FROM_IE);
|
| - ie->browser_type = MS_IE;
|
| - ie->source_path.clear();
|
| - ie->app_path.clear();
|
| - source_profiles_.push_back(ie);
|
| -}
|
| -
|
| -void ImporterHost::DetectFirefoxProfiles() {
|
| - // Detects which version of Firefox is installed.
|
| - int version = GetCurrentFirefoxMajorVersion();
|
| - ProfileType firefox_type;
|
| - if (version == 2) {
|
| - firefox_type = FIREFOX2;
|
| - } else if (version == 3) {
|
| - firefox_type = FIREFOX3;
|
| - } else {
|
| - // Ignores other versions of firefox.
|
| - return;
|
| - }
|
| -
|
| - std::wstring ini_file = GetProfilesINI();
|
| - DictionaryValue root;
|
| - ParseProfileINI(ini_file, &root);
|
| -
|
| - std::wstring source_path;
|
| - for (int i = 0; ; ++i) {
|
| - std::wstring current_profile = L"Profile" + IntToWString(i);
|
| - if (!root.HasKey(current_profile)) {
|
| - // Profiles are continuously numbered. So we exit when we can't
|
| - // find the i-th one.
|
| - break;
|
| - }
|
| - std::wstring is_relative, path, profile_path;
|
| - if (root.GetString(current_profile + L".IsRelative", &is_relative) &&
|
| - root.GetString(current_profile + L".Path", &path)) {
|
| - ReplaceSubstringsAfterOffset(&path, 0, L"/", L"\\");
|
| -
|
| - // IsRelative=1 means the folder path would be relative to the
|
| - // path of profiles.ini. IsRelative=0 refers to a custom profile
|
| - // location.
|
| - if (is_relative == L"1") {
|
| - profile_path = file_util::GetDirectoryFromPath(ini_file);
|
| - file_util::AppendToPath(&profile_path, path);
|
| - } else {
|
| - profile_path = path;
|
| - }
|
| -
|
| - // We only import the default profile when multiple profiles exist,
|
| - // since the other profiles are used mostly by developers for testing.
|
| - // Otherwise, Profile0 will be imported.
|
| - std::wstring is_default;
|
| - if ((root.GetString(current_profile + L".Default", &is_default) &&
|
| - is_default == L"1") || i == 0) {
|
| - source_path = profile_path;
|
| - // We break out of the loop when we have found the default profile.
|
| - if (is_default == L"1")
|
| - break;
|
| - }
|
| - }
|
| - }
|
| -
|
| - if (!source_path.empty()) {
|
| - ProfileInfo* firefox = new ProfileInfo();
|
| - firefox->description = l10n_util::GetString(IDS_IMPORT_FROM_FIREFOX);
|
| - firefox->browser_type = firefox_type;
|
| - firefox->source_path = source_path;
|
| - firefox->app_path = GetFirefoxInstallPath();
|
| - source_profiles_.push_back(firefox);
|
| - }
|
| -}
|
| -
|
|
|