| // Copyright 2016 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. |
| |
| #ifndef COMPONENTS_READING_LIST_CORE_READING_LIST_MODEL_IMPL_H_ |
| #define COMPONENTS_READING_LIST_CORE_READING_LIST_MODEL_IMPL_H_ |
| |
| #include <map> |
| #include <memory> |
| |
| #include "components/keyed_service/core/keyed_service.h" |
| #include "components/reading_list/core/reading_list_entry.h" |
| #include "components/reading_list/core/reading_list_model.h" |
| #include "components/reading_list/core/reading_list_model_storage.h" |
| #include "components/reading_list/core/reading_list_store_delegate.h" |
| |
| namespace base { |
| class Clock; |
| } |
| |
| class PrefService; |
| |
| // Concrete implementation of a reading list model using in memory lists. |
| class ReadingListModelImpl : public ReadingListModel, |
| public ReadingListStoreDelegate, |
| public KeyedService { |
| public: |
| using ReadingListEntries = std::map<GURL, ReadingListEntry>; |
| |
| // Initialize a ReadingListModelImpl to load and save data in |
| // |storage_layer|. Passing null to |storage_layer| will create a |
| // ReadingListModelImpl without persistence. Data will not be persistent |
| // across sessions. |
| // |clock| will be used to timestamp all the operations. |
| ReadingListModelImpl(std::unique_ptr<ReadingListModelStorage> storage_layer, |
| PrefService* pref_service, |
| base::Clock* clock_); |
| |
| ReadingListModelImpl(); |
| |
| syncer::ModelTypeSyncBridge* GetModelTypeSyncBridge() override; |
| |
| ~ReadingListModelImpl() override; |
| |
| void StoreLoaded(std::unique_ptr<ReadingListEntries> entries) override; |
| |
| // KeyedService implementation. |
| void Shutdown() override; |
| |
| // ReadingListModel implementation. |
| bool loaded() const override; |
| |
| size_t size() const override; |
| size_t unread_size() const override; |
| size_t unseen_size() const override; |
| |
| void MarkAllSeen() override; |
| bool GetLocalUnseenFlag() const override; |
| void ResetLocalUnseenFlag() override; |
| |
| const std::vector<GURL> Keys() const override; |
| |
| const ReadingListEntry* GetEntryByURL(const GURL& gurl) const override; |
| const ReadingListEntry* GetFirstUnreadEntry(bool distilled) const override; |
| |
| void RemoveEntryByURL(const GURL& url) override; |
| |
| const ReadingListEntry& AddEntry(const GURL& url, |
| const std::string& title, |
| reading_list::EntrySource source) override; |
| |
| void SetReadStatus(const GURL& url, bool read) override; |
| |
| void SetEntryTitle(const GURL& url, const std::string& title) override; |
| void SetEntryDistilledState( |
| const GURL& url, |
| ReadingListEntry::DistillationState state) override; |
| void SetEntryDistilledInfo(const GURL& url, |
| const base::FilePath& distilled_path, |
| const GURL& distilled_url, |
| int64_t distillation_size, |
| const base::Time& distillation_date) override; |
| void SetContentSuggestionsExtra( |
| const GURL& url, |
| const reading_list::ContentSuggestionsExtra& extra) override; |
| |
| void SyncAddEntry(std::unique_ptr<ReadingListEntry> entry) override; |
| ReadingListEntry* SyncMergeEntry( |
| std::unique_ptr<ReadingListEntry> entry) override; |
| void SyncRemoveEntry(const GURL& url) override; |
| bool DeleteAllEntries() override; |
| |
| std::unique_ptr<ReadingListModel::ScopedReadingListBatchUpdate> |
| CreateBatchToken() override; |
| |
| // Helper class that is used to scope batch updates. |
| class ScopedReadingListBatchUpdate |
| : public ReadingListModel::ScopedReadingListBatchUpdate { |
| public: |
| explicit ScopedReadingListBatchUpdate(ReadingListModelImpl* model); |
| |
| ~ScopedReadingListBatchUpdate() override; |
| |
| private: |
| std::unique_ptr<ReadingListModelStorage::ScopedBatchUpdate> storage_token_; |
| |
| DISALLOW_COPY_AND_ASSIGN(ScopedReadingListBatchUpdate); |
| }; |
| |
| protected: |
| void EnteringBatchUpdates() override; |
| void LeavingBatchUpdates() override; |
| |
| private: |
| // Sets/Loads the pref flag that indicate if some entries have never been seen |
| // since being added to the store. |
| void SetPersistentHasUnseen(bool has_unseen); |
| bool GetPersistentHasUnseen(); |
| |
| // Returns a mutable pointer to the entry with URL |url|. Return nullptr if |
| // no entry is found. |
| ReadingListEntry* GetMutableEntryFromURL(const GURL& url) const; |
| |
| // Returns the |storage_layer_| of the model. |
| ReadingListModelStorage* StorageLayer(); |
| |
| // Remove entry |url| and propagate to store if |from_sync| is false. |
| void RemoveEntryByURLImpl(const GURL& url, bool from_sync); |
| |
| void RebuildIndex() const; |
| |
| std::unique_ptr<ReadingListEntries> entries_; |
| size_t unread_entry_count_; |
| size_t read_entry_count_; |
| size_t unseen_entry_count_; |
| |
| // Update the 3 counts above considering addition/removal of |entry|. |
| void UpdateEntryStateCountersOnEntryRemoval(const ReadingListEntry& entry); |
| void UpdateEntryStateCountersOnEntryInsertion(const ReadingListEntry& entry); |
| |
| // Set the unseen flag to true. |
| void SetUnseenFlag(); |
| |
| base::Clock* clock_; |
| std::unique_ptr<ReadingListModelStorage> storage_layer_; |
| PrefService* pref_service_; |
| bool has_unseen_; |
| bool loaded_; |
| |
| base::WeakPtrFactory<ReadingListModelImpl> weak_ptr_factory_; |
| |
| DISALLOW_COPY_AND_ASSIGN(ReadingListModelImpl); |
| }; |
| |
| #endif // COMPONENTS_READING_LIST_CORE_READING_LIST_MODEL_IMPL_H_ |