Chromium Code Reviews| Index: chrome/browser/extensions/token_cache/token_cache_service.cc |
| diff --git a/chrome/browser/extensions/token_cache/token_cache_service.cc b/chrome/browser/extensions/token_cache/token_cache_service.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..15271542f6d63e0b18f3f5dcf3fe4aa1b79fd464 |
| --- /dev/null |
| +++ b/chrome/browser/extensions/token_cache/token_cache_service.cc |
| @@ -0,0 +1,78 @@ |
| +// Copyright (c) 2013 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 "chrome/browser/extensions/token_cache/token_cache_service.h" |
| + |
| +#include "base/logging.h" |
| +#include "chrome/common/chrome_notification_types.h" |
| +#include "content/public/browser/notification_source.h" |
| + |
| +using base::Time; |
| +using base::TimeDelta; |
| + |
| +namespace extensions { |
| + |
| +TokenCacheService::TokenCacheService(Profile* profile) : profile_(profile) { |
| + registrar_.Add(this, |
| + chrome::NOTIFICATION_GOOGLE_SIGNED_OUT, |
| + content::Source<Profile>(profile_)); |
| +} |
| + |
| +TokenCacheService::~TokenCacheService() { |
| +} |
| + |
| +void TokenCacheService::StoreToken(const std::string& token_name, |
| + const std::string& token_value, |
| + base::TimeDelta time_to_live) { |
| + TokenCacheData token_data; |
| + |
| + // Get the current time, and make sure that the token has not already expired. |
| + Time expiration_time; |
| + TimeDelta zero_delta; |
| + |
| + // Negative time deltas are meaningless to this function. |
| + DCHECK(time_to_live >= zero_delta); |
| + |
| + if (zero_delta < time_to_live) { |
| + expiration_time = Time::Now(); |
| + expiration_time += time_to_live; |
| + } |
| + |
| + token_data.token = token_value; |
| + token_data.expiration_time = expiration_time; |
| + |
| + // Add the token to our cache, overwriting any existing token with this name. |
| + token_cache_[token_name] = token_data; |
| +} |
| + |
| +// Retrieve a token for the currently logged in user. This returns an empty |
| +// string if the token was not found or timed out. |
| +std::string TokenCacheService::RetrieveToken(const std::string& token_name) { |
| + std::map<std::string, TokenCacheData>::iterator it = |
| + token_cache_.find(token_name); |
| + |
| + if (it != token_cache_.end()) { |
| + Time now = Time::Now(); |
| + if (it->second.expiration_time.is_null() || |
| + now < it->second.expiration_time) { |
| + return it->second.token; |
| + } else { |
| + // Remove this entry if it is expired. |
| + token_cache_.erase(it); |
| + return std::string(); |
| + } |
| + } |
| + |
| + return std::string(); |
| +} |
| + |
| +// Inherited from NotificationObserver. |
| +void TokenCacheService::Observe(int type, |
| + const content::NotificationSource& source, |
| + const content::NotificationDetails& details) { |
| + if (type == chrome::NOTIFICATION_GOOGLE_SIGNED_OUT) |
|
Andrew T Wilson (Slow)
2013/03/07 11:48:09
I'd make this:
DCHECK_EQ(type, chrome::NOTIFICATI
Pete Williamson
2013/03/07 20:02:46
Done.
|
| + token_cache_.clear(); |
| +} |
| + |
| +} // namespace extensions |