blob: 42d6dc5059df92278604464cedb5e19b3ce24c52 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
// Copyright 2014 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.
#include "components/suggestions/suggestions_store.h"
#include <string>
#include "base/base64.h"
#include "base/prefs/pref_service.h"
#include "base/time/default_clock.h"
#include "base/time/time.h"
#include "components/pref_registry/pref_registry_syncable.h"
#include "components/suggestions/suggestions_pref_names.h"
namespace suggestions {
SuggestionsStore::SuggestionsStore(PrefService* profile_prefs)
: pref_service_(profile_prefs), clock_(new base::DefaultClock()) {
DCHECK(profile_prefs);
}
SuggestionsStore::SuggestionsStore() {
}
SuggestionsStore::~SuggestionsStore() {}
void SuggestionsStore::SetClockForTesting(scoped_ptr<base::Clock> test_clock) {
this->clock_ = test_clock.Pass();
}
bool SuggestionsStore::LoadSuggestions(SuggestionsProfile* suggestions) {
DCHECK(suggestions);
const std::string base64_suggestions_data =
pref_service_->GetString(prefs::kSuggestionsData);
if (base64_suggestions_data.empty()) {
suggestions->Clear();
return false;
}
// If the decode process fails, assume the pref value is corrupt and clear it.
std::string suggestions_data;
if (!base::Base64Decode(base64_suggestions_data, &suggestions_data) ||
!suggestions->ParseFromString(suggestions_data)) {
VLOG(1) << "Suggestions data in profile pref is corrupt, clearing it.";
suggestions->Clear();
ClearSuggestions();
return false;
}
// Filter expired suggestions and update the stored suggestions if at least
// one was filtered. Return false if all suggestions are filtered.
int unfiltered_size = suggestions->suggestions_size();
FilterExpiredSuggestions(suggestions);
if (suggestions->suggestions_size() != unfiltered_size) {
if (!suggestions->suggestions_size()) {
suggestions->Clear();
ClearSuggestions();
return false;
} else {
StoreSuggestions(*suggestions);
}
}
return true;
}
void SuggestionsStore::FilterExpiredSuggestions(
SuggestionsProfile* suggestions) {
SuggestionsProfile filtered_suggestions;
int64 now_usec =
(this->clock_->Now() - base::Time::UnixEpoch()).ToInternalValue();
for (int i = 0; i < suggestions->suggestions_size(); ++i) {
ChromeSuggestion* suggestion = suggestions->mutable_suggestions(i);
if (!suggestion->has_expiry_ts() || suggestion->expiry_ts() > now_usec) {
filtered_suggestions.add_suggestions()->Swap(suggestion);
}
}
suggestions->Swap(&filtered_suggestions);
}
bool SuggestionsStore::StoreSuggestions(const SuggestionsProfile& suggestions) {
std::string suggestions_data;
if (!suggestions.SerializeToString(&suggestions_data)) return false;
std::string base64_suggestions_data;
base::Base64Encode(suggestions_data, &base64_suggestions_data);
pref_service_->SetString(prefs::kSuggestionsData, base64_suggestions_data);
return true;
}
void SuggestionsStore::ClearSuggestions() {
pref_service_->ClearPref(prefs::kSuggestionsData);
}
// static
void SuggestionsStore::RegisterProfilePrefs(
user_prefs::PrefRegistrySyncable* registry) {
registry->RegisterStringPref(prefs::kSuggestionsData, std::string());
}
} // namespace suggestions
|
