New authorization framework for sync. ...

chrome/browser/net/gaia/token_service.h

 // Copyright (c) 2010 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.
 //
 // The TokenService will supply authentication tokens for any service that
 // needs it, such as sync. Whenever the user logs in, a controller watching
 // the token service is expected to call ClientLogin to derive a new SID and
 // LSID. Whenever such credentials are available, the TokenService should be
 // updated with new credentials. The controller should then start fetching
 // tokens, which will be written to the database after retrieval, as well as
@@ skipping 19 matching lines @@
 //     SetMyToken(notification.token())
 //   }
 // }
 
 #ifndef CHROME_BROWSER_NET_GAIA_TOKEN_SERVICE_H_
 #define CHROME_BROWSER_NET_GAIA_TOKEN_SERVICE_H_
 #pragma once
 
 #include <map>
 #include <string>
+
+#include "base/gtest_prod_util.h"
 #include "base/scoped_ptr.h"
 #include "chrome/browser/webdata/web_data_service.h"
 #include "chrome/common/net/gaia/gaia_auth_consumer.h"
 #include "chrome/common/net/gaia/gaia_authenticator2.h"
 #include "chrome/common/net/gaia/google_service_auth_error.h"
-#include "base/gtest_prod_util.h"
+#include "chrome/common/notification_observer.h"
+#include "chrome/common/notification_registrar.h"
 
 class URLRequestContextGetter;
 class Profile;
 
 // The TokenService is a Profile member, so all calls are expected
 // from the UI thread.
 class TokenService : public GaiaAuthConsumer,
-                     public WebDataServiceConsumer {
+                     public WebDataServiceConsumer,
+                     public NotificationObserver {
  public:
    TokenService();
    virtual ~TokenService();
 
   // Notification classes
   class TokenAvailableDetails {
    public:
     TokenAvailableDetails() {}
     TokenAvailableDetails(const std::string& service,
                           const std::string& token)
@@ skipping 52 matching lines @@
   // Did we get a proper LSID?
   const bool AreCredentialsValid() const;
 
   // Tokens will be fetched for all services(sync, talk) in the background.
   // Results come back via event channel. Services can also poll before events
   // are issued.
   void StartFetchingTokens();
   const bool HasTokenForService(const char* const service) const;
   const std::string& GetTokenForService(const char* const service) const;
 
+  // For tests only. Doesn't save to the WebDB.
+  void IssueAuthTokenForTest(const std::string& service,
+                             const std::string& auth_token);
+
   // GaiaAuthConsumer implementation.
   virtual void OnIssueAuthTokenSuccess(const std::string& service,
                                        const std::string& auth_token);
   virtual void OnIssueAuthTokenFailure(const std::string& service,
                                        const GoogleServiceAuthError& error);
 
   // WebDataServiceConsumer implementation.
   virtual void OnWebDataServiceRequestDone(WebDataService::Handle h,
                                            const WDTypedResult* result);
 
+  // NotificationObserver implementation.
+  virtual void Observe(NotificationType type,
+                       const NotificationSource& source,
+                       const NotificationDetails& details);
+
  private:
 
   void FireTokenAvailableNotification(const std::string& service,
                                       const std::string& auth_token);
 
   void FireTokenRequestFailedNotification(const std::string& service,
                                           const GoogleServiceAuthError& error);
 
   void LoadTokensIntoMemory(const std::map<std::string, std::string>& in_toks,
                             std::map<std::string, std::string>* out_toks);
@@ skipping 16 matching lines @@
   // Size of array of services (must be defined here).
   static const int kNumServices = 2;
   // List of services that we're performing operations for.
   static const char* kServices[kNumServices];
   // A bunch of fetchers suitable for token issuing. We don't care about
   // the ordering, nor do we care which is for which service.
   scoped_ptr<GaiaAuthenticator2> fetchers_[kNumServices];
   // Map from service to token.
   std::map<std::string, std::string> token_map_;
 
+  NotificationRegistrar registrar_;
+
   FRIEND_TEST_ALL_PREFIXES(TokenServiceTest, LoadTokensIntoMemoryBasic);
   FRIEND_TEST_ALL_PREFIXES(TokenServiceTest, LoadTokensIntoMemoryAdvanced);
 
   DISALLOW_COPY_AND_ASSIGN(TokenService);
 };
 
 #endif  // CHROME_BROWSER_NET_GAIA_TOKEN_SERVICE_H_