OLD | NEW |
(Empty) | |
| 1 // Copyright 2013 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 |