OLD | NEW |
| (Empty) |
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 | |
3 // found in the LICENSE file. | |
4 | |
5 #include "chrome/browser/webdata/token_web_data.h" | |
6 | |
7 #include "base/bind.h" | |
8 #include "base/stl_util.h" | |
9 #include "chrome/browser/webdata/token_service_table.h" | |
10 #include "components/webdata/common/web_database_service.h" | |
11 #include "content/public/browser/browser_thread.h" | |
12 | |
13 using base::Bind; | |
14 using base::Time; | |
15 using content::BrowserThread; | |
16 | |
17 class TokenWebDataBackend | |
18 : public base::RefCountedThreadSafe<TokenWebDataBackend, | |
19 BrowserThread::DeleteOnDBThread> { | |
20 | |
21 public: | |
22 TokenWebDataBackend() { | |
23 } | |
24 | |
25 WebDatabase::State RemoveAllTokens(WebDatabase* db) { | |
26 if (TokenServiceTable::FromWebDatabase(db)->RemoveAllTokens()) { | |
27 return WebDatabase::COMMIT_NEEDED; | |
28 } | |
29 return WebDatabase::COMMIT_NOT_NEEDED; | |
30 } | |
31 | |
32 WebDatabase::State RemoveTokenForService( | |
33 const std::string& service, WebDatabase* db) { | |
34 if (TokenServiceTable::FromWebDatabase(db) | |
35 ->RemoveTokenForService(service)) { | |
36 return WebDatabase::COMMIT_NEEDED; | |
37 } | |
38 return WebDatabase::COMMIT_NOT_NEEDED; | |
39 } | |
40 | |
41 WebDatabase::State SetTokenForService( | |
42 const std::string& service, const std::string& token, WebDatabase* db) { | |
43 if (TokenServiceTable::FromWebDatabase(db)->SetTokenForService(service, | |
44 token)) { | |
45 return WebDatabase::COMMIT_NEEDED; | |
46 } | |
47 return WebDatabase::COMMIT_NOT_NEEDED; | |
48 } | |
49 | |
50 scoped_ptr<WDTypedResult> GetAllTokens(WebDatabase* db) { | |
51 std::map<std::string, std::string> map; | |
52 TokenServiceTable::FromWebDatabase(db)->GetAllTokens(&map); | |
53 return scoped_ptr<WDTypedResult>( | |
54 new WDResult<std::map<std::string, std::string> >(TOKEN_RESULT, map)); | |
55 } | |
56 | |
57 protected: | |
58 virtual ~TokenWebDataBackend() { | |
59 } | |
60 | |
61 private: | |
62 friend struct BrowserThread::DeleteOnThread<BrowserThread::DB>; | |
63 friend class base::DeleteHelper<TokenWebDataBackend>; | |
64 // We have to friend RCTS<> so WIN shared-lib build is happy | |
65 // (http://crbug/112250). | |
66 friend class base::RefCountedThreadSafe<TokenWebDataBackend, | |
67 BrowserThread::DeleteOnDBThread>; | |
68 | |
69 }; | |
70 | |
71 TokenWebData::TokenWebData(scoped_refptr<WebDatabaseService> wdbs, | |
72 const ProfileErrorCallback& callback) | |
73 : WebDataServiceBase(wdbs, callback, | |
74 BrowserThread::GetMessageLoopProxyForThread(BrowserThread::UI)), | |
75 token_backend_(new TokenWebDataBackend()) { | |
76 } | |
77 | |
78 void TokenWebData::SetTokenForService(const std::string& service, | |
79 const std::string& token) { | |
80 wdbs_->ScheduleDBTask(FROM_HERE, | |
81 Bind(&TokenWebDataBackend::SetTokenForService, token_backend_, | |
82 service, token)); | |
83 } | |
84 | |
85 void TokenWebData::RemoveAllTokens() { | |
86 wdbs_->ScheduleDBTask(FROM_HERE, | |
87 Bind(&TokenWebDataBackend::RemoveAllTokens, token_backend_)); | |
88 } | |
89 | |
90 void TokenWebData::RemoveTokenForService(const std::string& service) { | |
91 wdbs_->ScheduleDBTask(FROM_HERE, | |
92 Bind(&TokenWebDataBackend::RemoveTokenForService, token_backend_, | |
93 service)); | |
94 } | |
95 | |
96 // Null on failure. Success is WDResult<std::string> | |
97 WebDataServiceBase::Handle TokenWebData::GetAllTokens( | |
98 WebDataServiceConsumer* consumer) { | |
99 return wdbs_->ScheduleDBTaskWithResult(FROM_HERE, | |
100 Bind(&TokenWebDataBackend::GetAllTokens, token_backend_), consumer); | |
101 } | |
102 | |
103 TokenWebData::TokenWebData() | |
104 : WebDataServiceBase(NULL, ProfileErrorCallback(), | |
105 BrowserThread::GetMessageLoopProxyForThread(BrowserThread::UI)), | |
106 token_backend_(new TokenWebDataBackend()) { | |
107 } | |
108 | |
109 TokenWebData::~TokenWebData() { | |
110 } | |
OLD | NEW |