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 |