Add ENABLE_MEDIA_ROUTER define to builds other than Android and iOS.
[chromium-blink-merge.git] / chrome / utility / importer / firefox_importer.cc
blob97c530a96da892423de68a73d72c03b67e3ec3e3
1 // Copyright 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
5 #include "chrome/utility/importer/firefox_importer.h"
7 #include <set>
9 #include "base/files/file_enumerator.h"
10 #include "base/files/file_util.h"
11 #include "base/json/json_file_value_serializer.h"
12 #include "base/memory/scoped_ptr.h"
13 #include "base/message_loop/message_loop.h"
14 #include "base/stl_util.h"
15 #include "base/strings/string_util.h"
16 #include "base/strings/utf_string_conversions.h"
17 #include "chrome/common/importer/firefox_importer_utils.h"
18 #include "chrome/common/importer/imported_bookmark_entry.h"
19 #include "chrome/common/importer/importer_autofill_form_data_entry.h"
20 #include "chrome/common/importer/importer_bridge.h"
21 #include "chrome/common/importer/importer_url_row.h"
22 #include "chrome/grit/generated_resources.h"
23 #include "chrome/utility/importer/bookmark_html_reader.h"
24 #include "chrome/utility/importer/favicon_reencode.h"
25 #include "chrome/utility/importer/nss_decryptor.h"
26 #include "components/autofill/core/common/password_form.h"
27 #include "sql/connection.h"
28 #include "sql/statement.h"
29 #include "url/gurl.h"
31 namespace {
33 // Original definition is in http://mxr.mozilla.org/firefox/source/toolkit/
34 // components/places/public/nsINavBookmarksService.idl
35 enum BookmarkItemType {
36 TYPE_BOOKMARK = 1,
37 TYPE_FOLDER = 2,
38 TYPE_SEPARATOR = 3,
39 TYPE_DYNAMIC_CONTAINER = 4
42 // Loads the default bookmarks in the Firefox installed at |app_path|,
43 // and stores their locations in |urls|.
44 void LoadDefaultBookmarks(const base::FilePath& app_path,
45 std::set<GURL>* urls) {
46 base::FilePath file = app_path.AppendASCII("defaults")
47 .AppendASCII("profile")
48 .AppendASCII("bookmarks.html");
49 urls->clear();
51 std::vector<ImportedBookmarkEntry> bookmarks;
52 std::vector<importer::SearchEngineInfo> search_engines;
53 bookmark_html_reader::ImportBookmarksFile(base::Callback<bool(void)>(),
54 base::Callback<bool(const GURL&)>(),
55 file,
56 &bookmarks,
57 &search_engines,
58 NULL);
59 for (size_t i = 0; i < bookmarks.size(); ++i)
60 urls->insert(bookmarks[i].url);
63 // Returns true if |url| has a valid scheme that we allow to import. We
64 // filter out the URL with a unsupported scheme.
65 bool CanImportURL(const GURL& url) {
66 // The URL is not valid.
67 if (!url.is_valid())
68 return false;
70 // Filter out the URLs with unsupported schemes.
71 const char* const kInvalidSchemes[] = {"wyciwyg", "place", "about", "chrome"};
72 for (size_t i = 0; i < arraysize(kInvalidSchemes); ++i) {
73 if (url.SchemeIs(kInvalidSchemes[i]))
74 return false;
77 return true;
80 } // namespace
82 struct FirefoxImporter::BookmarkItem {
83 int parent;
84 int id;
85 GURL url;
86 base::string16 title;
87 BookmarkItemType type;
88 std::string keyword;
89 base::Time date_added;
90 int64 favicon;
91 bool empty_folder;
94 FirefoxImporter::FirefoxImporter() {
97 FirefoxImporter::~FirefoxImporter() {
100 void FirefoxImporter::StartImport(
101 const importer::SourceProfile& source_profile,
102 uint16 items,
103 ImporterBridge* bridge) {
104 bridge_ = bridge;
105 source_path_ = source_profile.source_path;
106 app_path_ = source_profile.app_path;
108 #if defined(OS_POSIX)
109 locale_ = source_profile.locale;
110 #endif
112 // The order here is important!
113 bridge_->NotifyStarted();
114 if ((items & importer::HOME_PAGE) && !cancelled()) {
115 bridge_->NotifyItemStarted(importer::HOME_PAGE);
116 ImportHomepage(); // Doesn't have a UI item.
117 bridge_->NotifyItemEnded(importer::HOME_PAGE);
120 // Note history should be imported before bookmarks because bookmark import
121 // will also import favicons and we store favicon for a URL only if the URL
122 // exist in history or bookmarks.
123 if ((items & importer::HISTORY) && !cancelled()) {
124 bridge_->NotifyItemStarted(importer::HISTORY);
125 ImportHistory();
126 bridge_->NotifyItemEnded(importer::HISTORY);
129 if ((items & importer::FAVORITES) && !cancelled()) {
130 bridge_->NotifyItemStarted(importer::FAVORITES);
131 ImportBookmarks();
132 bridge_->NotifyItemEnded(importer::FAVORITES);
134 if ((items & importer::SEARCH_ENGINES) && !cancelled()) {
135 bridge_->NotifyItemStarted(importer::SEARCH_ENGINES);
136 ImportSearchEngines();
137 bridge_->NotifyItemEnded(importer::SEARCH_ENGINES);
139 if ((items & importer::PASSWORDS) && !cancelled()) {
140 bridge_->NotifyItemStarted(importer::PASSWORDS);
141 ImportPasswords();
142 bridge_->NotifyItemEnded(importer::PASSWORDS);
144 if ((items & importer::AUTOFILL_FORM_DATA) && !cancelled()) {
145 bridge_->NotifyItemStarted(importer::AUTOFILL_FORM_DATA);
146 ImportAutofillFormData();
147 bridge_->NotifyItemEnded(importer::AUTOFILL_FORM_DATA);
149 bridge_->NotifyEnded();
152 void FirefoxImporter::ImportHistory() {
153 base::FilePath file = source_path_.AppendASCII("places.sqlite");
154 if (!base::PathExists(file))
155 return;
157 sql::Connection db;
158 if (!db.Open(file))
159 return;
161 // |visit_type| represent the transition type of URLs (typed, click,
162 // redirect, bookmark, etc.) We eliminate some URLs like sub-frames and
163 // redirects, since we don't want them to appear in history.
164 // Firefox transition types are defined in:
165 // toolkit/components/places/public/nsINavHistoryService.idl
166 const char query[] =
167 "SELECT h.url, h.title, h.visit_count, "
168 "h.hidden, h.typed, v.visit_date "
169 "FROM moz_places h JOIN moz_historyvisits v "
170 "ON h.id = v.place_id "
171 "WHERE v.visit_type <= 3";
173 sql::Statement s(db.GetUniqueStatement(query));
175 std::vector<ImporterURLRow> rows;
176 while (s.Step() && !cancelled()) {
177 GURL url(s.ColumnString(0));
179 // Filter out unwanted URLs.
180 if (!CanImportURL(url))
181 continue;
183 ImporterURLRow row(url);
184 row.title = s.ColumnString16(1);
185 row.visit_count = s.ColumnInt(2);
186 row.hidden = s.ColumnInt(3) == 1;
187 row.typed_count = s.ColumnInt(4);
188 row.last_visit = base::Time::FromTimeT(s.ColumnInt64(5)/1000000);
190 rows.push_back(row);
193 if (!rows.empty() && !cancelled())
194 bridge_->SetHistoryItems(rows, importer::VISIT_SOURCE_FIREFOX_IMPORTED);
197 void FirefoxImporter::ImportBookmarks() {
198 base::FilePath file = source_path_.AppendASCII("places.sqlite");
199 if (!base::PathExists(file))
200 return;
202 sql::Connection db;
203 if (!db.Open(file))
204 return;
206 // Get the bookmark folders that we are interested in.
207 int toolbar_folder_id = -1;
208 int menu_folder_id = -1;
209 int unsorted_folder_id = -1;
210 LoadRootNodeID(&db, &toolbar_folder_id, &menu_folder_id, &unsorted_folder_id);
212 // Load livemark IDs.
213 std::set<int> livemark_id;
214 LoadLivemarkIDs(&db, &livemark_id);
216 // Load the default bookmarks.
217 std::set<GURL> default_urls;
218 LoadDefaultBookmarks(app_path_, &default_urls);
220 BookmarkList list;
221 GetTopBookmarkFolder(&db, toolbar_folder_id, &list);
222 GetTopBookmarkFolder(&db, menu_folder_id, &list);
223 GetTopBookmarkFolder(&db, unsorted_folder_id, &list);
224 size_t count = list.size();
225 for (size_t i = 0; i < count; ++i)
226 GetWholeBookmarkFolder(&db, &list, i, NULL);
228 std::vector<ImportedBookmarkEntry> bookmarks;
229 std::vector<importer::SearchEngineInfo> search_engines;
230 FaviconMap favicon_map;
232 // TODO(jcampan): http://b/issue?id=1196285 we do not support POST based
233 // keywords yet. We won't include them in the list.
234 std::set<int> post_keyword_ids;
235 const char query[] =
236 "SELECT b.id FROM moz_bookmarks b "
237 "INNER JOIN moz_items_annos ia ON ia.item_id = b.id "
238 "INNER JOIN moz_anno_attributes aa ON ia.anno_attribute_id = aa.id "
239 "WHERE aa.name = 'bookmarkProperties/POSTData'";
240 sql::Statement s(db.GetUniqueStatement(query));
242 if (!s.is_valid())
243 return;
245 while (s.Step() && !cancelled())
246 post_keyword_ids.insert(s.ColumnInt(0));
248 for (size_t i = 0; i < list.size(); ++i) {
249 BookmarkItem* item = list[i];
251 // Folders are added implicitly on adding children, so we only explicitly
252 // add empty folders.
253 if (item->type != TYPE_BOOKMARK &&
254 ((item->type != TYPE_FOLDER) || !item->empty_folder))
255 continue;
257 if (CanImportURL(item->url)) {
258 // Skip the default bookmarks and unwanted URLs.
259 if (default_urls.find(item->url) != default_urls.end() ||
260 post_keyword_ids.find(item->id) != post_keyword_ids.end())
261 continue;
263 // Find the bookmark path by tracing their links to parent folders.
264 std::vector<base::string16> path;
265 BookmarkItem* child = item;
266 bool found_path = false;
267 bool is_in_toolbar = false;
268 while (child->parent >= 0) {
269 BookmarkItem* parent = list[child->parent];
270 if (livemark_id.find(parent->id) != livemark_id.end()) {
271 // Don't import live bookmarks.
272 break;
275 if (parent->id != menu_folder_id) {
276 // To avoid excessive nesting, omit the name for the bookmarks menu
277 // folder.
278 path.insert(path.begin(), parent->title);
281 if (parent->id == toolbar_folder_id)
282 is_in_toolbar = true;
284 if (parent->id == toolbar_folder_id ||
285 parent->id == menu_folder_id ||
286 parent->id == unsorted_folder_id) {
287 // We've reached a root node, hooray!
288 found_path = true;
289 break;
292 child = parent;
295 if (!found_path)
296 continue;
298 ImportedBookmarkEntry entry;
299 entry.creation_time = item->date_added;
300 entry.title = item->title;
301 entry.url = item->url;
302 entry.path = path;
303 entry.in_toolbar = is_in_toolbar;
304 entry.is_folder = item->type == TYPE_FOLDER;
306 bookmarks.push_back(entry);
309 if (item->type == TYPE_BOOKMARK) {
310 if (item->favicon)
311 favicon_map[item->favicon].insert(item->url);
313 // Import this bookmark as a search engine if it has a keyword and its URL
314 // is usable as a search engine URL. (Even if the URL doesn't allow
315 // substitution, importing as a "search engine" allows users to trigger
316 // the bookmark by entering its keyword in the omnibox.)
317 if (item->keyword.empty())
318 continue;
319 importer::SearchEngineInfo search_engine_info;
320 std::string search_engine_url;
321 if (item->url.is_valid())
322 search_engine_info.url = base::UTF8ToUTF16(item->url.spec());
323 else if (bookmark_html_reader::CanImportURLAsSearchEngine(
324 item->url,
325 &search_engine_url))
326 search_engine_info.url = base::UTF8ToUTF16(search_engine_url);
327 else
328 continue;
329 search_engine_info.keyword = base::UTF8ToUTF16(item->keyword);
330 search_engine_info.display_name = item->title;
331 search_engines.push_back(search_engine_info);
335 STLDeleteElements(&list);
337 // Write into profile.
338 if (!bookmarks.empty() && !cancelled()) {
339 const base::string16& first_folder_name =
340 bridge_->GetLocalizedString(IDS_BOOKMARK_GROUP_FROM_FIREFOX);
341 bridge_->AddBookmarks(bookmarks, first_folder_name);
343 if (!search_engines.empty() && !cancelled()) {
344 bridge_->SetKeywords(search_engines, false);
346 if (!favicon_map.empty() && !cancelled()) {
347 favicon_base::FaviconUsageDataList favicons;
348 LoadFavicons(&db, favicon_map, &favicons);
349 bridge_->SetFavicons(favicons);
353 void FirefoxImporter::ImportPasswords() {
354 // Initializes NSS3.
355 NSSDecryptor decryptor;
356 if (!decryptor.Init(source_path_, source_path_) &&
357 !decryptor.Init(app_path_, source_path_)) {
358 return;
361 std::vector<autofill::PasswordForm> forms;
362 base::FilePath source_path = source_path_;
363 base::FilePath file = source_path.AppendASCII("signons.sqlite");
364 if (base::PathExists(file)) {
365 // Since Firefox 3.1, passwords are in signons.sqlite db.
366 decryptor.ReadAndParseSignons(file, &forms);
367 } else {
368 // Firefox 3.0 uses signons3.txt to store the passwords.
369 file = source_path.AppendASCII("signons3.txt");
370 if (!base::PathExists(file))
371 file = source_path.AppendASCII("signons2.txt");
373 std::string content;
374 base::ReadFileToString(file, &content);
375 decryptor.ParseSignons(content, &forms);
378 if (!cancelled()) {
379 for (size_t i = 0; i < forms.size(); ++i) {
380 if (!forms[i].username_value.empty() ||
381 !forms[i].password_value.empty() ||
382 forms[i].blacklisted_by_user) {
383 bridge_->SetPasswordForm(forms[i]);
389 void FirefoxImporter::ImportSearchEngines() {
390 std::vector<std::string> search_engine_data;
391 GetSearchEnginesXMLData(&search_engine_data);
393 bridge_->SetFirefoxSearchEnginesXMLData(search_engine_data);
396 void FirefoxImporter::ImportHomepage() {
397 GURL home_page = GetHomepage(source_path_);
398 if (home_page.is_valid() && !IsDefaultHomepage(home_page, app_path_)) {
399 bridge_->AddHomePage(home_page);
403 void FirefoxImporter::ImportAutofillFormData() {
404 base::FilePath file = source_path_.AppendASCII("formhistory.sqlite");
405 if (!base::PathExists(file))
406 return;
408 sql::Connection db;
409 if (!db.Open(file))
410 return;
412 const char query[] =
413 "SELECT fieldname, value, timesUsed, firstUsed, lastUsed FROM "
414 "moz_formhistory";
416 sql::Statement s(db.GetUniqueStatement(query));
418 std::vector<ImporterAutofillFormDataEntry> form_entries;
419 while (s.Step() && !cancelled()) {
420 ImporterAutofillFormDataEntry form_entry;
421 form_entry.name = s.ColumnString16(0);
422 form_entry.value = s.ColumnString16(1);
423 form_entry.times_used = s.ColumnInt(2);
424 form_entry.first_used = base::Time::FromTimeT(s.ColumnInt64(3) / 1000000);
425 form_entry.last_used = base::Time::FromTimeT(s.ColumnInt64(4) / 1000000);
427 // Don't import search bar history.
428 if (base::UTF16ToUTF8(form_entry.name) == "searchbar-history")
429 continue;
431 form_entries.push_back(form_entry);
434 if (!form_entries.empty() && !cancelled())
435 bridge_->SetAutofillFormData(form_entries);
438 void FirefoxImporter::GetSearchEnginesXMLData(
439 std::vector<std::string>* search_engine_data) {
440 base::FilePath file = source_path_.AppendASCII("search.sqlite");
441 if (!base::PathExists(file)) {
442 // Since Firefox 3.5, search engines are no longer stored in search.sqlite.
443 // Instead, search.json is used for storing search engines.
444 GetSearchEnginesXMLDataFromJSON(search_engine_data);
445 return;
448 sql::Connection db;
449 if (!db.Open(file))
450 return;
452 const char query[] =
453 "SELECT engineid FROM engine_data "
454 "WHERE engineid NOT IN "
455 "(SELECT engineid FROM engine_data "
456 "WHERE name='hidden') "
457 "ORDER BY value ASC";
459 sql::Statement s(db.GetUniqueStatement(query));
460 if (!s.is_valid())
461 return;
463 const base::FilePath searchplugins_path(FILE_PATH_LITERAL("searchplugins"));
464 // Search engine definitions are XMLs stored in two directories. Default
465 // engines are in the app directory (app_path_) and custom engines are
466 // in the profile directory (source_path_).
468 // Since Firefox 21, app_path_ engines are in 'browser' subdirectory:
469 base::FilePath app_path =
470 app_path_.AppendASCII("browser").Append(searchplugins_path);
471 if (!base::PathExists(app_path)) {
472 // This might be an older Firefox, try old location without the 'browser'
473 // path component:
474 app_path = app_path_.Append(searchplugins_path);
477 base::FilePath profile_path = source_path_.Append(searchplugins_path);
479 // Firefox doesn't store a search engine in its sqlite database unless the
480 // user has added a engine. So we get search engines from sqlite db as well
481 // as from the file system.
482 if (s.Step()) {
483 const std::string kAppPrefix("[app]/");
484 const std::string kProfilePrefix("[profile]/");
485 do {
486 base::FilePath file;
487 std::string engine(s.ColumnString(0));
489 // The string contains [app]/<name>.xml or [profile]/<name>.xml where
490 // the [app] and [profile] need to be replaced with the actual app or
491 // profile path.
492 size_t index = engine.find(kAppPrefix);
493 if (index != std::string::npos) {
494 // Remove '[app]/'.
495 file = app_path.AppendASCII(engine.substr(index + kAppPrefix.length()));
496 } else if ((index = engine.find(kProfilePrefix)) != std::string::npos) {
497 // Remove '[profile]/'.
498 file = profile_path.AppendASCII(
499 engine.substr(index + kProfilePrefix.length()));
500 } else {
501 // Looks like absolute path to the file.
502 file = base::FilePath::FromUTF8Unsafe(engine);
504 std::string file_data;
505 base::ReadFileToString(file, &file_data);
506 search_engine_data->push_back(file_data);
507 } while (s.Step() && !cancelled());
510 #if defined(OS_POSIX)
511 // Ubuntu-flavored Firefox supports locale-specific search engines via
512 // locale-named subdirectories. They fall back to en-US.
513 // See http://crbug.com/53899
514 // TODO(jshin): we need to make sure our locale code matches that of
515 // Firefox.
516 DCHECK(!locale_.empty());
517 base::FilePath locale_app_path = app_path.AppendASCII(locale_);
518 base::FilePath default_locale_app_path = app_path.AppendASCII("en-US");
519 if (base::DirectoryExists(locale_app_path))
520 app_path = locale_app_path;
521 else if (base::DirectoryExists(default_locale_app_path))
522 app_path = default_locale_app_path;
523 #endif
525 // Get search engine definition from file system.
526 base::FileEnumerator engines(app_path, false, base::FileEnumerator::FILES);
527 for (base::FilePath engine_path = engines.Next();
528 !engine_path.value().empty(); engine_path = engines.Next()) {
529 std::string file_data;
530 base::ReadFileToString(file, &file_data);
531 search_engine_data->push_back(file_data);
535 void FirefoxImporter::GetSearchEnginesXMLDataFromJSON(
536 std::vector<std::string>* search_engine_data) {
537 // search-metadata.json contains keywords for search engines. This
538 // file exists only if the user has set keywords for search engines.
539 base::FilePath search_metadata_json_file =
540 source_path_.AppendASCII("search-metadata.json");
541 JSONFileValueDeserializer metadata_deserializer(search_metadata_json_file);
542 scoped_ptr<base::Value> metadata_root(
543 metadata_deserializer.Deserialize(NULL, NULL));
544 const base::DictionaryValue* search_metadata_root = NULL;
545 if (metadata_root)
546 metadata_root->GetAsDictionary(&search_metadata_root);
548 // search.json contains information about search engines to import.
549 base::FilePath search_json_file = source_path_.AppendASCII("search.json");
550 if (!base::PathExists(search_json_file))
551 return;
553 JSONFileValueDeserializer deserializer(search_json_file);
554 scoped_ptr<base::Value> root(deserializer.Deserialize(NULL, NULL));
555 const base::DictionaryValue* search_root = NULL;
556 if (!root || !root->GetAsDictionary(&search_root))
557 return;
559 const std::string kDirectories("directories");
560 const base::DictionaryValue* search_directories = NULL;
561 if (!search_root->GetDictionary(kDirectories, &search_directories))
562 return;
564 // Dictionary |search_directories| contains a list of search engines
565 // (default and installed). The list can be found from key <engines>
566 // of the dictionary. Key <engines> is a grandchild of key <directories>.
567 // However, key <engines> parent's key is dynamic which depends on
568 // operating systems. For example,
569 // Ubuntu (for default search engine):
570 // /usr/lib/firefox/distribution/searchplugins/locale/en-US
571 // Ubuntu (for installed search engines):
572 // /home/<username>/.mozilla/firefox/lcd50n4n.default/searchplugins
573 // Windows (for default search engine):
574 // C:\\Program Files (x86)\\Mozilla Firefox\\browser\\searchplugins
575 // Therefore, it needs to be retrieved by searching.
577 for (base::DictionaryValue::Iterator it(*search_directories); !it.IsAtEnd();
578 it.Advance()) {
579 // The key of |it| may contains dot (.) which cannot be used as <key>
580 // for retrieving <engines>. Hence, it is needed to get |it| as dictionary.
581 // The resulted dictionary can be used for retrieving <engines>.
582 const std::string kEngines("engines");
583 const base::DictionaryValue* search_directory = NULL;
584 if (!it.value().GetAsDictionary(&search_directory))
585 continue;
587 const base::ListValue* search_engines = NULL;
588 if (!search_directory->GetList(kEngines, &search_engines))
589 continue;
591 const std::string kFilePath("filePath");
592 const std::string kHidden("_hidden");
593 for (size_t i = 0; i < search_engines->GetSize(); ++i) {
594 const base::DictionaryValue* engine_info = NULL;
595 if (!search_engines->GetDictionary(i, &engine_info))
596 continue;
598 bool is_hidden = false;
599 std::string file_path;
600 if (!engine_info->GetBoolean(kHidden, &is_hidden) ||
601 !engine_info->GetString(kFilePath, &file_path))
602 continue;
604 if (!is_hidden) {
605 const std::string kAppPrefix("[app]/");
606 const std::string kProfilePrefix("[profile]/");
607 base::FilePath xml_file = base::FilePath::FromUTF8Unsafe(file_path);
609 // If |file_path| contains [app] or [profile] then they need to be
610 // replaced with the actual app or profile path.
611 size_t index = file_path.find(kAppPrefix);
612 if (index != std::string::npos) {
613 // Replace '[app]/' with actual app path.
614 xml_file = app_path_.AppendASCII("searchplugins").AppendASCII(
615 file_path.substr(index + kAppPrefix.length()));
616 } else if ((index = file_path.find(kProfilePrefix)) !=
617 std::string::npos) {
618 // Replace '[profile]/' with actual profile path.
619 xml_file = source_path_.AppendASCII("searchplugins").AppendASCII(
620 file_path.substr(index + kProfilePrefix.length()));
623 std::string file_data;
624 base::ReadFileToString(xml_file, &file_data);
626 // If a keyword is mentioned for this search engine, then add
627 // it to the XML string as an <Alias> element and use this updated
628 // string.
629 const base::DictionaryValue* search_xml_path = NULL;
630 if (search_metadata_root && search_metadata_root->HasKey(file_path) &&
631 search_metadata_root->GetDictionaryWithoutPathExpansion(
632 file_path, &search_xml_path)) {
633 std::string alias;
634 search_xml_path->GetString("alias", &alias);
636 // Add <Alias> element as the last child element.
637 size_t end_of_parent = file_data.find("</SearchPlugin>");
638 if (end_of_parent != std::string::npos && !alias.empty())
639 file_data.insert(end_of_parent, "<Alias>" + alias + "</Alias> \n");
641 search_engine_data->push_back(file_data);
647 void FirefoxImporter::LoadRootNodeID(sql::Connection* db,
648 int* toolbar_folder_id,
649 int* menu_folder_id,
650 int* unsorted_folder_id) {
651 static const char kToolbarFolderName[] = "toolbar";
652 static const char kMenuFolderName[] = "menu";
653 static const char kUnsortedFolderName[] = "unfiled";
655 const char query[] = "SELECT root_name, folder_id FROM moz_bookmarks_roots";
656 sql::Statement s(db->GetUniqueStatement(query));
658 while (s.Step()) {
659 std::string folder = s.ColumnString(0);
660 int id = s.ColumnInt(1);
661 if (folder == kToolbarFolderName)
662 *toolbar_folder_id = id;
663 else if (folder == kMenuFolderName)
664 *menu_folder_id = id;
665 else if (folder == kUnsortedFolderName)
666 *unsorted_folder_id = id;
670 void FirefoxImporter::LoadLivemarkIDs(sql::Connection* db,
671 std::set<int>* livemark) {
672 static const char kFeedAnnotation[] = "livemark/feedURI";
673 livemark->clear();
675 const char query[] =
676 "SELECT b.item_id "
677 "FROM moz_anno_attributes a "
678 "JOIN moz_items_annos b ON a.id = b.anno_attribute_id "
679 "WHERE a.name = ? ";
680 sql::Statement s(db->GetUniqueStatement(query));
681 s.BindString(0, kFeedAnnotation);
683 while (s.Step() && !cancelled())
684 livemark->insert(s.ColumnInt(0));
687 void FirefoxImporter::GetTopBookmarkFolder(sql::Connection* db,
688 int folder_id,
689 BookmarkList* list) {
690 const char query[] =
691 "SELECT b.title "
692 "FROM moz_bookmarks b "
693 "WHERE b.type = 2 AND b.id = ? "
694 "ORDER BY b.position";
695 sql::Statement s(db->GetUniqueStatement(query));
696 s.BindInt(0, folder_id);
698 if (s.Step()) {
699 BookmarkItem* item = new BookmarkItem;
700 item->parent = -1; // The top level folder has no parent.
701 item->id = folder_id;
702 item->title = s.ColumnString16(0);
703 item->type = TYPE_FOLDER;
704 item->favicon = 0;
705 item->empty_folder = true;
706 list->push_back(item);
710 void FirefoxImporter::GetWholeBookmarkFolder(sql::Connection* db,
711 BookmarkList* list,
712 size_t position,
713 bool* empty_folder) {
714 if (position >= list->size()) {
715 NOTREACHED();
716 return;
719 const char query[] =
720 "SELECT b.id, h.url, COALESCE(b.title, h.title), "
721 "b.type, k.keyword, b.dateAdded, h.favicon_id "
722 "FROM moz_bookmarks b "
723 "LEFT JOIN moz_places h ON b.fk = h.id "
724 "LEFT JOIN moz_keywords k ON k.id = b.keyword_id "
725 "WHERE b.type IN (1,2) AND b.parent = ? "
726 "ORDER BY b.position";
727 sql::Statement s(db->GetUniqueStatement(query));
728 s.BindInt(0, (*list)[position]->id);
730 BookmarkList temp_list;
731 while (s.Step()) {
732 BookmarkItem* item = new BookmarkItem;
733 item->parent = static_cast<int>(position);
734 item->id = s.ColumnInt(0);
735 item->url = GURL(s.ColumnString(1));
736 item->title = s.ColumnString16(2);
737 item->type = static_cast<BookmarkItemType>(s.ColumnInt(3));
738 item->keyword = s.ColumnString(4);
739 item->date_added = base::Time::FromTimeT(s.ColumnInt64(5)/1000000);
740 item->favicon = s.ColumnInt64(6);
741 item->empty_folder = true;
743 temp_list.push_back(item);
744 if (empty_folder != NULL)
745 *empty_folder = false;
748 // Appends all items to the list.
749 for (BookmarkList::iterator i = temp_list.begin();
750 i != temp_list.end(); ++i) {
751 list->push_back(*i);
752 // Recursive add bookmarks in sub-folders.
753 if ((*i)->type == TYPE_FOLDER)
754 GetWholeBookmarkFolder(db, list, list->size() - 1, &(*i)->empty_folder);
758 void FirefoxImporter::LoadFavicons(
759 sql::Connection* db,
760 const FaviconMap& favicon_map,
761 favicon_base::FaviconUsageDataList* favicons) {
762 const char query[] = "SELECT url, data FROM moz_favicons WHERE id=?";
763 sql::Statement s(db->GetUniqueStatement(query));
765 if (!s.is_valid())
766 return;
768 for (FaviconMap::const_iterator i = favicon_map.begin();
769 i != favicon_map.end(); ++i) {
770 s.BindInt64(0, i->first);
771 if (s.Step()) {
772 favicon_base::FaviconUsageData usage;
774 usage.favicon_url = GURL(s.ColumnString(0));
775 if (!usage.favicon_url.is_valid())
776 continue; // Don't bother importing favicons with invalid URLs.
778 std::vector<unsigned char> data;
779 s.ColumnBlobAsVector(1, &data);
780 if (data.empty())
781 continue; // Data definitely invalid.
783 if (!importer::ReencodeFavicon(&data[0], data.size(), &usage.png_data))
784 continue; // Unable to decode.
786 usage.urls = i->second;
787 favicons->push_back(usage);
789 s.Reset(true);