| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "components/enhanced_bookmarks/enhanced_bookmark_model.h" | 5 #include "components/enhanced_bookmarks/enhanced_bookmark_model.h" |
| 6 | 6 |
| 7 #include <iomanip> | 7 #include <iomanip> |
| 8 #include <sstream> | 8 #include <sstream> |
| 9 | 9 |
| 10 #include "base/base64.h" | 10 #include "base/base64.h" |
| 11 #include "base/logging.h" | 11 #include "base/logging.h" |
| 12 #include "base/message_loop/message_loop_proxy.h" | 12 #include "base/message_loop/message_loop_proxy.h" |
| 13 #include "base/rand_util.h" | 13 #include "base/rand_util.h" |
| 14 #include "base/strings/string_number_conversions.h" | 14 #include "base/strings/string_number_conversions.h" |
| 15 #include "components/bookmarks/browser/bookmark_model.h" | 15 #include "components/bookmarks/browser/bookmark_model.h" |
| 16 #include "components/bookmarks/browser/bookmark_node.h" | 16 #include "components/bookmarks/browser/bookmark_node.h" |
| 17 #include "components/enhanced_bookmarks/enhanced_bookmark_model_observer.h" | 17 #include "components/enhanced_bookmarks/enhanced_bookmark_model_observer.h" |
| 18 #include "components/enhanced_bookmarks/proto/metadata.pb.h" | 18 #include "components/enhanced_bookmarks/proto/metadata.pb.h" |
| 19 #include "ui/base/models/tree_node_iterator.h" | 19 #include "ui/base/models/tree_node_iterator.h" |
| 20 #include "url/gurl.h" | 20 #include "url/gurl.h" |
| 21 | 21 |
| 22 using bookmarks::BookmarkModel; | 22 using bookmarks::BookmarkModel; |
| 23 using bookmarks::BookmarkNode; | 23 using bookmarks::BookmarkNode; |
| 24 | 24 |
| 25 namespace { | 25 namespace { |
| 26 const char* kBookmarkBarId = "f_bookmarks_bar"; | 26 const char* kBookmarkBarId = "f_bookmarks_bar"; |
| 27 | 27 |
| 28 const char* kFlagsKey = "stars.flags"; | |
| 29 const char* kIdKey = "stars.id"; | 28 const char* kIdKey = "stars.id"; |
| 30 const char* kImageDataKey = "stars.imageData"; | 29 const char* kImageDataKey = "stars.imageData"; |
| 31 const char* kNoteKey = "stars.note"; | 30 const char* kNoteKey = "stars.note"; |
| 32 const char* kOldIdKey = "stars.oldId"; | 31 const char* kOldIdKey = "stars.oldId"; |
| 33 const char* kPageDataKey = "stars.pageData"; | 32 const char* kPageDataKey = "stars.pageData"; |
| 34 const char* kVersionKey = "stars.version"; | 33 const char* kVersionKey = "stars.version"; |
| 35 | 34 |
| 36 const char* kBookmarkPrefix = "ebc_"; | 35 const char* kBookmarkPrefix = "ebc_"; |
| 37 | 36 |
| 38 enum Flags { | |
| 39 // When set the server will attempt to fill in image and snippet information. | |
| 40 NEEDS_OFFLINE_PROCESSING = 0x1, | |
| 41 }; | |
| 42 | |
| 43 // Helper method for working with bookmark metainfo. | 37 // Helper method for working with bookmark metainfo. |
| 44 std::string DataForMetaInfoField(const BookmarkNode* node, | 38 std::string DataForMetaInfoField(const BookmarkNode* node, |
| 45 const std::string& field) { | 39 const std::string& field) { |
| 46 std::string value; | 40 std::string value; |
| 47 if (!node->GetMetaInfo(field, &value)) | 41 if (!node->GetMetaInfo(field, &value)) |
| 48 return std::string(); | 42 return std::string(); |
| 49 | 43 |
| 50 std::string decoded; | 44 std::string decoded; |
| 51 if (!base::Base64Decode(value, &decoded)) | 45 if (!base::Base64Decode(value, &decoded)) |
| 52 return std::string(); | 46 return std::string(); |
| (...skipping 252 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 305 } | 299 } |
| 306 | 300 |
| 307 void EnhancedBookmarkModel::BookmarkNodeAdded(BookmarkModel* model, | 301 void EnhancedBookmarkModel::BookmarkNodeAdded(BookmarkModel* model, |
| 308 const BookmarkNode* parent, | 302 const BookmarkNode* parent, |
| 309 int index) { | 303 int index) { |
| 310 const BookmarkNode* node = parent->GetChild(index); | 304 const BookmarkNode* node = parent->GetChild(index); |
| 311 std::string remote_id; | 305 std::string remote_id; |
| 312 if (node->GetMetaInfo(kIdKey, &remote_id)) { | 306 if (node->GetMetaInfo(kIdKey, &remote_id)) { |
| 313 AddToIdMap(node); | 307 AddToIdMap(node); |
| 314 ScheduleResetDuplicateRemoteIds(); | 308 ScheduleResetDuplicateRemoteIds(); |
| 315 } else if (node->is_url()) { | |
| 316 set_needs_offline_processing_tasks_[node] = | |
| 317 make_linked_ptr(new base::CancelableClosure( | |
| 318 base::Bind(&EnhancedBookmarkModel::SetNeedsOfflineProcessing, | |
| 319 weak_ptr_factory_.GetWeakPtr(), | |
| 320 base::Unretained(node)))); | |
| 321 base::MessageLoopProxy::current()->PostTask( | |
| 322 FROM_HERE, set_needs_offline_processing_tasks_[node]->callback()); | |
| 323 } | 309 } |
| 324 FOR_EACH_OBSERVER( | 310 FOR_EACH_OBSERVER( |
| 325 EnhancedBookmarkModelObserver, observers_, EnhancedBookmarkAdded(node)); | 311 EnhancedBookmarkModelObserver, observers_, EnhancedBookmarkAdded(node)); |
| 326 } | 312 } |
| 327 | 313 |
| 328 void EnhancedBookmarkModel::BookmarkNodeRemoved( | 314 void EnhancedBookmarkModel::BookmarkNodeRemoved( |
| 329 BookmarkModel* model, | 315 BookmarkModel* model, |
| 330 const BookmarkNode* parent, | 316 const BookmarkNode* parent, |
| 331 int old_index, | 317 int old_index, |
| 332 const BookmarkNode* node, | 318 const BookmarkNode* node, |
| (...skipping 68 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 401 } | 387 } |
| 402 } | 388 } |
| 403 | 389 |
| 404 void EnhancedBookmarkModel::RemoveNodeFromMaps(const BookmarkNode* node) { | 390 void EnhancedBookmarkModel::RemoveNodeFromMaps(const BookmarkNode* node) { |
| 405 for (int i = 0; i < node->child_count(); i++) { | 391 for (int i = 0; i < node->child_count(); i++) { |
| 406 RemoveNodeFromMaps(node->GetChild(i)); | 392 RemoveNodeFromMaps(node->GetChild(i)); |
| 407 } | 393 } |
| 408 std::string remote_id = GetRemoteId(node); | 394 std::string remote_id = GetRemoteId(node); |
| 409 id_map_.erase(remote_id); | 395 id_map_.erase(remote_id); |
| 410 nodes_to_reset_.erase(node); | 396 nodes_to_reset_.erase(node); |
| 411 set_needs_offline_processing_tasks_.erase(node); | |
| 412 } | 397 } |
| 413 | 398 |
| 414 void EnhancedBookmarkModel::ScheduleResetDuplicateRemoteIds() { | 399 void EnhancedBookmarkModel::ScheduleResetDuplicateRemoteIds() { |
| 415 if (!nodes_to_reset_.empty()) { | 400 if (!nodes_to_reset_.empty()) { |
| 416 base::MessageLoopProxy::current()->PostTask( | 401 base::MessageLoopProxy::current()->PostTask( |
| 417 FROM_HERE, | 402 FROM_HERE, |
| 418 base::Bind(&EnhancedBookmarkModel::ResetDuplicateRemoteIds, | 403 base::Bind(&EnhancedBookmarkModel::ResetDuplicateRemoteIds, |
| 419 weak_ptr_factory_.GetWeakPtr())); | 404 weak_ptr_factory_.GetWeakPtr())); |
| 420 } | 405 } |
| 421 } | 406 } |
| 422 | 407 |
| 423 void EnhancedBookmarkModel::ResetDuplicateRemoteIds() { | 408 void EnhancedBookmarkModel::ResetDuplicateRemoteIds() { |
| 424 for (NodeToIdMap::iterator it = nodes_to_reset_.begin(); | 409 for (NodeToIdMap::iterator it = nodes_to_reset_.begin(); |
| 425 it != nodes_to_reset_.end(); | 410 it != nodes_to_reset_.end(); |
| 426 ++it) { | 411 ++it) { |
| 427 BookmarkNode::MetaInfoMap meta_info; | 412 BookmarkNode::MetaInfoMap meta_info; |
| 428 meta_info[kIdKey] = ""; | 413 meta_info[kIdKey] = ""; |
| 429 meta_info[kOldIdKey] = it->second; | 414 meta_info[kOldIdKey] = it->second; |
| 430 SetMultipleMetaInfo(it->first, meta_info); | 415 SetMultipleMetaInfo(it->first, meta_info); |
| 431 } | 416 } |
| 432 nodes_to_reset_.clear(); | 417 nodes_to_reset_.clear(); |
| 433 } | 418 } |
| 434 | 419 |
| 435 void EnhancedBookmarkModel::SetNeedsOfflineProcessing( | |
| 436 const BookmarkNode* node) { | |
| 437 set_needs_offline_processing_tasks_.erase(node); | |
| 438 int flags = 0; | |
| 439 std::string flags_str; | |
| 440 if (node->GetMetaInfo(kFlagsKey, &flags_str)) { | |
| 441 if (!base::StringToInt(flags_str, &flags)) | |
| 442 flags = 0; | |
| 443 } | |
| 444 flags |= NEEDS_OFFLINE_PROCESSING; | |
| 445 SetMetaInfo(node, kFlagsKey, base::IntToString(flags)); | |
| 446 } | |
| 447 | |
| 448 void EnhancedBookmarkModel::SetMetaInfo(const BookmarkNode* node, | 420 void EnhancedBookmarkModel::SetMetaInfo(const BookmarkNode* node, |
| 449 const std::string& field, | 421 const std::string& field, |
| 450 const std::string& value) { | 422 const std::string& value) { |
| 451 DCHECK(!bookmark_model_->is_permanent_node(node)); | 423 DCHECK(!bookmark_model_->is_permanent_node(node)); |
| 452 | 424 |
| 453 BookmarkNode::MetaInfoMap meta_info; | 425 BookmarkNode::MetaInfoMap meta_info; |
| 454 const BookmarkNode::MetaInfoMap* old_meta_info = node->GetMetaInfoMap(); | 426 const BookmarkNode::MetaInfoMap* old_meta_info = node->GetMetaInfoMap(); |
| 455 if (old_meta_info) | 427 if (old_meta_info) |
| 456 meta_info.insert(old_meta_info->begin(), old_meta_info->end()); | 428 meta_info.insert(old_meta_info->begin(), old_meta_info->end()); |
| 457 | 429 |
| (...skipping 97 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 555 if (!result) | 527 if (!result) |
| 556 return false; | 528 return false; |
| 557 | 529 |
| 558 std::string encoded; | 530 std::string encoded; |
| 559 base::Base64Encode(output, &encoded); | 531 base::Base64Encode(output, &encoded); |
| 560 bookmark_model_->SetNodeMetaInfo(node, kImageDataKey, encoded); | 532 bookmark_model_->SetNodeMetaInfo(node, kImageDataKey, encoded); |
| 561 return true; | 533 return true; |
| 562 } | 534 } |
| 563 | 535 |
| 564 } // namespace enhanced_bookmarks | 536 } // namespace enhanced_bookmarks |
| OLD | NEW |