OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012 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. | |
4 | |
5 #ifndef CHROME_BROWSER_HISTORY_SHORTCUTS_BACKEND_H_ | |
6 #define CHROME_BROWSER_HISTORY_SHORTCUTS_BACKEND_H_ | |
7 | |
8 #include <map> | |
9 #include <string> | |
10 #include <vector> | |
11 | |
12 #include "base/files/file_path.h" | |
13 #include "base/gtest_prod_util.h" | |
14 #include "base/memory/ref_counted.h" | |
15 #include "base/memory/scoped_ptr.h" | |
16 #include "base/observer_list.h" | |
17 #include "base/strings/string16.h" | |
18 #include "base/synchronization/lock.h" | |
19 #include "base/time/time.h" | |
20 #include "chrome/browser/autocomplete/autocomplete_match.h" | |
21 #include "components/keyed_service/content/refcounted_browser_context_keyed_serv
ice.h" | |
22 #include "content/public/browser/notification_observer.h" | |
23 #include "content/public/browser/notification_registrar.h" | |
24 #include "url/gurl.h" | |
25 | |
26 class Profile; | |
27 | |
28 namespace history { | |
29 | |
30 class ShortcutsDatabase; | |
31 | |
32 // This class manages the shortcut provider backend - access to database on the | |
33 // db thread, etc. | |
34 class ShortcutsBackend : public RefcountedBrowserContextKeyedService, | |
35 public content::NotificationObserver { | |
36 public: | |
37 // The following struct encapsulates one previously selected omnibox shortcut. | |
38 struct Shortcut { | |
39 // The pieces of an AutocompleteMatch that we preserve in a shortcut. | |
40 struct MatchCore { | |
41 explicit MatchCore(const AutocompleteMatch& match); | |
42 MatchCore(const base::string16& fill_into_edit, | |
43 const GURL& destination_url, | |
44 const base::string16& contents, | |
45 const ACMatchClassifications& contents_class, | |
46 const base::string16& description, | |
47 const ACMatchClassifications& description_class, | |
48 content::PageTransition transition, | |
49 AutocompleteMatch::Type type, | |
50 const base::string16& keyword); | |
51 ~MatchCore(); | |
52 | |
53 AutocompleteMatch ToMatch() const; | |
54 | |
55 base::string16 fill_into_edit; | |
56 GURL destination_url; | |
57 base::string16 contents; | |
58 // For both contents_class and description_class, we strip MATCH | |
59 // classifications; the ShortcutsProvider will re-mark MATCH regions based | |
60 // on the user's current typing. | |
61 ACMatchClassifications contents_class; | |
62 base::string16 description; | |
63 ACMatchClassifications description_class; | |
64 content::PageTransition transition; | |
65 AutocompleteMatch::Type type; | |
66 base::string16 keyword; | |
67 }; | |
68 | |
69 Shortcut(const std::string& id, | |
70 const base::string16& text, | |
71 const MatchCore& match_core, | |
72 const base::Time& last_access_time, | |
73 int number_of_hits); | |
74 // Required for STL, we don't use this directly. | |
75 Shortcut(); | |
76 ~Shortcut(); | |
77 | |
78 std::string id; // Unique guid for the shortcut. | |
79 base::string16 text; // The user's original input string. | |
80 MatchCore match_core; | |
81 base::Time last_access_time; // Last time shortcut was selected. | |
82 int number_of_hits; // How many times shortcut was selected. | |
83 }; | |
84 | |
85 typedef std::multimap<base::string16, const Shortcut> ShortcutMap; | |
86 | |
87 // |profile| is necessary for profile notifications only and can be NULL in | |
88 // unit-tests. For unit testing, set |suppress_db| to true to prevent creation | |
89 // of the database, in which case all operations are performed in memory only. | |
90 ShortcutsBackend(Profile* profile, bool suppress_db); | |
91 | |
92 // The interface is guaranteed to be called on the thread AddObserver() | |
93 // was called. | |
94 class ShortcutsBackendObserver { | |
95 public: | |
96 // Called after the database is loaded and Init() completed. | |
97 virtual void OnShortcutsLoaded() = 0; | |
98 // Called when shortcuts changed (added/updated/removed) in the database. | |
99 virtual void OnShortcutsChanged() {} | |
100 | |
101 protected: | |
102 virtual ~ShortcutsBackendObserver() {} | |
103 }; | |
104 | |
105 // Asynchronously initializes the ShortcutsBackend, it is safe to call | |
106 // multiple times - only the first call will be processed. | |
107 bool Init(); | |
108 | |
109 // All of the public functions *must* be called on UI thread only! | |
110 | |
111 bool initialized() const { return current_state_ == INITIALIZED; } | |
112 const ShortcutMap& shortcuts_map() const { return shortcuts_map_; } | |
113 | |
114 // Deletes the Shortcuts with the url. | |
115 bool DeleteShortcutsWithUrl(const GURL& shortcut_url); | |
116 | |
117 void AddObserver(ShortcutsBackendObserver* obs); | |
118 void RemoveObserver(ShortcutsBackendObserver* obs); | |
119 | |
120 // Looks for an existing shortcut to match.destination_url that starts with | |
121 // |text|. Updates that shortcut if found, otherwise adds a new shortcut. | |
122 void AddOrUpdateShortcut(const base::string16& text, | |
123 const AutocompleteMatch& match); | |
124 | |
125 private: | |
126 friend class base::RefCountedThreadSafe<ShortcutsBackend>; | |
127 friend class ShortcutsProviderTest; | |
128 FRIEND_TEST_ALL_PREFIXES(ShortcutsBackendTest, AddAndUpdateShortcut); | |
129 FRIEND_TEST_ALL_PREFIXES(ShortcutsBackendTest, DeleteShortcuts); | |
130 | |
131 enum CurrentState { | |
132 NOT_INITIALIZED, // Backend created but not initialized. | |
133 INITIALIZING, // Init() called, but not completed yet. | |
134 INITIALIZED, // Initialization completed, all accessors can be safely | |
135 // called. | |
136 }; | |
137 | |
138 typedef std::map<std::string, ShortcutMap::iterator> GuidMap; | |
139 | |
140 virtual ~ShortcutsBackend(); | |
141 | |
142 // RefcountedBrowserContextKeyedService: | |
143 virtual void ShutdownOnUIThread() OVERRIDE; | |
144 | |
145 // content::NotificationObserver: | |
146 virtual void Observe(int type, | |
147 const content::NotificationSource& source, | |
148 const content::NotificationDetails& details) OVERRIDE; | |
149 | |
150 // Internal initialization of the back-end. Posted by Init() to the DB thread. | |
151 // On completion posts InitCompleted() back to UI thread. | |
152 void InitInternal(); | |
153 | |
154 // Finishes initialization on UI thread, notifies all observers. | |
155 void InitCompleted(); | |
156 | |
157 // Adds the Shortcut to the database. | |
158 bool AddShortcut(const Shortcut& shortcut); | |
159 | |
160 // Updates timing and selection count for the Shortcut. | |
161 bool UpdateShortcut(const Shortcut& shortcut); | |
162 | |
163 // Deletes the Shortcuts with the id. | |
164 bool DeleteShortcutsWithIds(const std::vector<std::string>& shortcut_ids); | |
165 | |
166 // Deletes all shortcuts whose URLs begin with |url|. If |exact_match| is | |
167 // true, only shortcuts from exactly |url| are deleted. | |
168 bool DeleteShortcutsWithUrl(const GURL& url, bool exact_match); | |
169 | |
170 // Deletes all of the shortcuts. | |
171 bool DeleteAllShortcuts(); | |
172 | |
173 CurrentState current_state_; | |
174 ObserverList<ShortcutsBackendObserver> observer_list_; | |
175 scoped_refptr<ShortcutsDatabase> db_; | |
176 | |
177 // The |temp_shortcuts_map_| and |temp_guid_map_| used for temporary storage | |
178 // between InitInternal() and InitComplete() to avoid doing a potentially huge | |
179 // copy. | |
180 scoped_ptr<ShortcutMap> temp_shortcuts_map_; | |
181 scoped_ptr<GuidMap> temp_guid_map_; | |
182 | |
183 ShortcutMap shortcuts_map_; | |
184 // This is a helper map for quick access to a shortcut by guid. | |
185 GuidMap guid_map_; | |
186 | |
187 content::NotificationRegistrar notification_registrar_; | |
188 | |
189 // For some unit-test only. | |
190 bool no_db_access_; | |
191 | |
192 DISALLOW_COPY_AND_ASSIGN(ShortcutsBackend); | |
193 }; | |
194 | |
195 } // namespace history | |
196 | |
197 #endif // CHROME_BROWSER_HISTORY_SHORTCUTS_BACKEND_H_ | |
OLD | NEW |