1// Copyright (c) 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/extensions/token_cache/token_cache_service.h"
6
7#include "base/logging.h"
8#include "chrome/browser/signin/signin_manager_factory.h"
9#include "components/signin/core/browser/signin_manager.h"
10
11using base::Time;
12using base::TimeDelta;
13
14namespace extensions {
15
16TokenCacheService::TokenCacheService(Profile* profile) : profile_(profile) {
17  SigninManagerFactory::GetForProfile(profile)->AddObserver(this);
18}
19
20TokenCacheService::~TokenCacheService() {
21}
22
23void TokenCacheService::StoreToken(const std::string& token_name,
24                                   const std::string& token_value,
25                                   base::TimeDelta time_to_live) {
26  TokenCacheData token_data;
27
28  // Get the current time, and make sure that the token has not already expired.
29  Time expiration_time;
30  TimeDelta zero_delta;
31
32  // Negative time deltas are meaningless to this function.
33  DCHECK(time_to_live >= zero_delta);
34
35  if (zero_delta < time_to_live) {
36    expiration_time = Time::Now();
37    expiration_time += time_to_live;
38  }
39
40  token_data.token = token_value;
41  token_data.expiration_time = expiration_time;
42
43  // Add the token to our cache, overwriting any existing token with this name.
44  token_cache_[token_name] = token_data;
45}
46
47// Retrieve a token for the currently logged in user.  This returns an empty
48// string if the token was not found or timed out.
49std::string TokenCacheService::RetrieveToken(const std::string& token_name) {
50  std::map<std::string, TokenCacheData>::iterator it =
51      token_cache_.find(token_name);
52
53  if (it != token_cache_.end()) {
54    Time now = Time::Now();
55    if (it->second.expiration_time.is_null() ||
56        now < it->second.expiration_time) {
57      return it->second.token;
58    } else {
59      // Remove this entry if it is expired.
60      token_cache_.erase(it);
61      return std::string();
62    }
63  }
64
65  return std::string();
66}
67
68void TokenCacheService::Shutdown() {
69  SigninManagerFactory::GetForProfile(const_cast<Profile*>(profile_))
70      ->RemoveObserver(this);
71}
72
73void TokenCacheService::GoogleSignedOut(const std::string& account_id,
74                                        const std::string& username) {
75  token_cache_.clear();
76}
77
78}  // namespace extensions
79