16e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)// Copyright 2014 The Chromium Authors. All rights reserved. 26e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 36e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)// found in the LICENSE file. 46e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 56e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)#include "components/enhanced_bookmarks/enhanced_bookmark_utils.h" 66e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 76e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)#include "base/i18n/string_compare.h" 86e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)#include "base/i18n/string_search.h" 96e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)#include "base/strings/utf_string_conversions.h" 106e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)#include "components/bookmarks/browser/bookmark_model.h" 116e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)#include "ui/base/models/tree_node_iterator.h" 126e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 131320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tuccinamespace enhanced_bookmarks { 146e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 156e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)std::vector<const BookmarkNode*> FindBookmarksWithQuery( 166e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) BookmarkModel* bookmark_model, 176e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) const std::string& query) { 186e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) std::vector<const BookmarkNode*> results; 196e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) base::string16 query16 = base::UTF8ToUTF16(query); 206e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) base::i18n::FixedPatternStringSearchIgnoringCaseAndAccents pattern16(query16); 216e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 226e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) ui::TreeNodeIterator<const BookmarkNode> iterator( 236e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) bookmark_model->root_node()); 246e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) while (iterator.has_next()) { 256e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) const BookmarkNode* node = iterator.Next(); 266e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) if (!node->is_url()) 276e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) continue; 286e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 296e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) // Search the title for the query. 306e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) size_t match_index; 316e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) size_t match_length; 326e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) bool found = 336e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) pattern16.Search(node->GetTitle(), &match_index, &match_length); 346e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) if (found) 356e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) results.push_back(node); 366e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) } 376e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) return results; 386e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)} 396e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 406e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)// Comparator used to sort bookmarks. No folders are allowed. 416e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)class BookmarkNameComparator : public std::binary_function<const BookmarkNode*, 426e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) const BookmarkNode*, 436e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) bool> { 446e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) public: 456e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) explicit BookmarkNameComparator(icu::Collator* collator) 466e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) : collator_(collator) {} 476e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 486e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) // Returns true if |n1| preceeds |n2|. 496e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) bool operator()(const BookmarkNode* n1, const BookmarkNode* n2) { 506e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) DCHECK(!n1->is_folder()); 516e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) DCHECK(!n2->is_folder()); 526e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) if (!collator_) 536e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) return n1->GetTitle() < n2->GetTitle(); 546e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) return base::i18n::CompareString16WithCollator( 556e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) collator_, n1->GetTitle(), n2->GetTitle()) == UCOL_LESS; 566e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) } 576e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 586e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) private: 596e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) icu::Collator* collator_; 606e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)}; 616e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 626e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)void SortBookmarksByName(std::vector<const BookmarkNode*>& nodes) { 636e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) UErrorCode error = U_ZERO_ERROR; 646e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) scoped_ptr<icu::Collator> collator(icu::Collator::createInstance(error)); 656e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) if (U_FAILURE(error)) 666e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) collator.reset(NULL); 676e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) std::sort(nodes.begin(), nodes.end(), BookmarkNameComparator(collator.get())); 686e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)} 696e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 706e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)std::vector<const BookmarkNode*> PrimaryPermanentNodes(BookmarkModel* model) { 716e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) DCHECK(model->loaded()); 726e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) std::vector<const BookmarkNode*> nodes; 736e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) nodes.push_back(model->other_node()); 746e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) nodes.push_back(model->mobile_node()); 756e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) return nodes; 766e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)} 776e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 786e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)std::vector<const BookmarkNode*> RootLevelFolders(BookmarkModel* model) { 796e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) std::vector<const BookmarkNode*> root_level_folders; 806e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 816e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) // Find the direct folder children of the primary permanent nodes. 826e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) std::vector<const BookmarkNode*> primary_permanent_nodes = 836e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) PrimaryPermanentNodes(model); 846e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) for (const BookmarkNode* parent : primary_permanent_nodes) { 856e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) int child_count = parent->child_count(); 866e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) for (int i = 0; i < child_count; ++i) { 876e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) const BookmarkNode* node = parent->GetChild(i); 886e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) if (node->is_folder() && node->IsVisible()) 896e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) root_level_folders.push_back(node); 906e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) } 916e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) } 926e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 936e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) // Add the bookmark bar if it has children. 946e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) const BookmarkNode* bb_node = model->bookmark_bar_node(); 956e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) if (bb_node->child_count() > 0) 966e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) root_level_folders.push_back(bb_node); 976e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 986e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) return root_level_folders; 996e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)} 1006e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 1016e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)bool IsPrimaryPermanentNode(const BookmarkNode* node, BookmarkModel* model) { 1026e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) std::vector<const BookmarkNode*> primary_nodes(PrimaryPermanentNodes(model)); 1036e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) if (std::find(primary_nodes.begin(), primary_nodes.end(), node) != 1046e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) primary_nodes.end()) { 1056e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) return true; 1066e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) } 1076e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) return false; 1086e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)} 1096e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 1106e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)const BookmarkNode* RootLevelFolderForNode(const BookmarkNode* node, 1116e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) BookmarkModel* model) { 1126e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) // This helper function doesn't work for managed bookmarks. This checks that 1136e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) // |node| is editable by the user, which currently covers all the other 1146e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) // bookmarks except the managed bookmarks. 1156e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) DCHECK(model->client()->CanBeEditedByUser(node)); 1166e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 1176e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) const std::vector<const BookmarkNode*> root_folders(RootLevelFolders(model)); 1186e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) const BookmarkNode* top = node; 1196e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) while (top && 1206e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) std::find(root_folders.begin(), root_folders.end(), top) == 1216e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) root_folders.end()) { 1226e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) top = top->parent(); 1236e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) } 1246e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) return top; 1256e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)} 1266e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 1271320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} // namespace enhanced_bookmarks 128