summaryrefslogtreecommitdiffstats
path: root/chrome/browser/extensions/token_cache/token_cache_service.cc
blob: 6ee5422ad323a942d4cfaf827d8c064a31f7c5c6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
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/browser/signin/signin_manager_factory.h"
#include "components/signin/core/browser/signin_manager.h"

using base::Time;
using base::TimeDelta;

namespace extensions {

TokenCacheService::TokenCacheService(Profile* profile) : profile_(profile) {
  SigninManagerFactory::GetForProfile(profile)->AddObserver(this);
}

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();
}

void TokenCacheService::Shutdown() {
  SigninManagerFactory::GetForProfile(const_cast<Profile*>(profile_))
      ->RemoveObserver(this);
}

void TokenCacheService::GoogleSignedOut(const std::string& account_id,
                                        const std::string& username) {
  token_cache_.clear();
}

}  // namespace extensions