base::Callback-ify policy::DeviceManagementService.

chrome/browser/policy/device_management_service.h

-// Copyright (c) 2011 The Chromium Authors. All rights reserved.
+// Copyright (c) 2012 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.
 
 #ifndef CHROME_BROWSER_POLICY_DEVICE_MANAGEMENT_SERVICE_H_
 #define CHROME_BROWSER_POLICY_DEVICE_MANAGEMENT_SERVICE_H_
 #pragma once
 
 #include <deque>
 #include <map>
 #include <string>
 #include <vector>
 
 #include "base/basictypes.h"
+#include "base/callback.h"
 #include "base/compiler_specific.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/weak_ptr.h"
+#include "chrome/browser/policy/cloud_policy_constants.h"
+#include "chrome/browser/policy/proto/device_management_backend.pb.h"
 #include "content/public/common/url_fetcher_delegate.h"
-#include "googleurl/src/gurl.h"
 
 namespace net {
 class URLRequestContextGetter;
-class URLRequestStatus;
-typedef std::vector<std::string> ResponseCookies;
 }
 
 namespace policy {
 
 class DeviceManagementBackend;
+class DeviceManagementRequestJobImpl;
+class DeviceManagementService;
+
+// DeviceManagementRequestJob describes a request to send to the device
+// management service. Jobs are created by DeviceManagementService. They can be
+// canceled by deleting the object.
+class DeviceManagementRequestJob {
+ public:
+  // Describes the job type.
+  enum JobType {
+    TYPE_AUTO_ENROLLMENT,
+    TYPE_REGISTRATION,
+    TYPE_POLICY_FETCH,
+    TYPE_UNREGISTRATION,
+  };
+
+  typedef base::Callback<
+      void(DeviceManagementStatus,
+           const enterprise_management::DeviceManagementResponse&)> Callback;
+
+  virtual ~DeviceManagementRequestJob();
+
+  // Functions for configuring the job. These should only be called before
+  // Start()ing the job, but never afterwards.
+  void SetGaiaToken(const std::string& gaia_token);
+  void SetOAuthToken(const std::string& oauth_token);
+  void SetUserAffiliation(UserAffiliation user_affiliation);
+  void SetDMToken(const std::string& dm_token);
+  void SetClientID(const std::string& client_id);
+  enterprise_management::DeviceManagementRequest* GetRequest();
+
+  // Starts the job. |callback| will be invoked on completion.
+  void Start(const DeviceManagementRequestJob::Callback& callback);
+
+ protected:
+  typedef std::vector<std::pair<std::string, std::string> > ParameterMap;
+
+  explicit DeviceManagementRequestJob(JobType type);
+
+  // Appends a parameter to |query_params|.
+  void AddParameter(const std::string& name, const std::string& value);
+
+  // Fires the job, to be filled in by implementations.
+  virtual void Run() = 0;
+
+  ParameterMap query_params_;
+  std::string gaia_token_;
+  std::string dm_token_;
+  enterprise_management::DeviceManagementRequest request_;
+
+  Callback callback_;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(DeviceManagementRequestJob);
+};
 
 // The device management service is responsible for everything related to
 // communication with the device management server. It creates the backends
 // objects that the device management policy provider and friends use to issue
 // requests.
 class DeviceManagementService : public content::URLFetcherDelegate {
  public:
-  // Describes a device management job handled by the service.
-  class DeviceManagementJob {
-   public:
-    virtual ~DeviceManagementJob() {}
-
-    // Handles the URL request response.
-    virtual void HandleResponse(const net::URLRequestStatus& status,
-                                int response_code,
-                                const net::ResponseCookies& cookies,
-                                const std::string& data) = 0;
-
-    // Gets the URL to contact.
-    virtual GURL GetURL(const std::string& server_url) = 0;
-
-    // Configures the fetcher, setting up payload and headers.
-    virtual void ConfigureRequest(content::URLFetcher* fetcher) = 0;
-  };
-
   explicit DeviceManagementService(const std::string& server_url);
   virtual ~DeviceManagementService();
 
   // Constructs a device management backend for use by client code. Ownership of
   // the returned backend object is transferred to the caller.
   // Marked virtual for the benefit of tests.
+  // TODO(mnissler): This is deprecated, remove. Use CreateJob() instead.
   virtual DeviceManagementBackend* CreateBackend();
 
+  // Creates a new device management request job. Ownership is transferred to
+  // the caller.
+  virtual DeviceManagementRequestJob* CreateJob(
+      DeviceManagementRequestJob::JobType type);
+
   // Schedules a task to run |Initialize| after |delay_milliseconds| had passed.
   void ScheduleInitialization(int64 delay_milliseconds);
 
   // Makes the service stop all requests and drop the reference to the request
   // context.
   void Shutdown();
 
-  // Adds a job. Caller must make sure the job pointer stays valid until the job
-  // completes or gets canceled via RemoveJob().
-  void AddJob(DeviceManagementJob* job);
+ private:
+  typedef std::map<const content::URLFetcher*,
+                   DeviceManagementRequestJobImpl*> JobFetcherMap;
+  typedef std::deque<DeviceManagementRequestJobImpl*> JobQueue;
 
-  // Removes a job. The job will be removed and won't receive a completion
-  // callback.
-  void RemoveJob(DeviceManagementJob* job);
-
- protected:
-  // Starts the given job.
-  virtual void StartJob(DeviceManagementJob* job, bool bypass_proxy);
-
- private:
-  typedef std::map<const content::URLFetcher*, DeviceManagementJob*>
-      JobFetcherMap;
-  typedef std::deque<DeviceManagementJob*> JobQueue;
+  friend class DeviceManagementRequestJobImpl;
 
   // content::URLFetcherDelegate override.
   virtual void OnURLFetchComplete(const content::URLFetcher* source) OVERRIDE;
 
   // Does the actual initialization using the request context specified for
   // |PrepareInitialization|. This will also fire any pending network requests.
   void Initialize();
 
+  // Starts a job.
+  void StartJob(DeviceManagementRequestJobImpl* job, bool bypass_proxy);
+
+  // Adds a job. Caller must make sure the job pointer stays valid until the job
+  // completes or gets canceled via RemoveJob().
+  void AddJob(DeviceManagementRequestJobImpl* job);
+
+  // Removes a job. The job will be removed and won't receive a completion
+  // callback.
+  void RemoveJob(DeviceManagementRequestJobImpl* job);
+
   // Server at which to contact the service.
   const std::string server_url_;
 
   // The request context we use.
   scoped_refptr<net::URLRequestContextGetter> request_context_getter_;
 
   // The jobs we currently have in flight.
   JobFetcherMap pending_jobs_;
 
   // Jobs that are registered, but not started yet.
   JobQueue queued_jobs_;
 
   // If this service is initialized, incoming requests get fired instantly.
   // If it is not initialized, incoming requests are queued.
   bool initialized_;
 
   // Used to create tasks to run |Initialize| delayed on the UI thread.
   base::WeakPtrFactory<DeviceManagementService> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(DeviceManagementService);
 };
 
 }  // namespace policy
 
 #endif  // CHROME_BROWSER_POLICY_DEVICE_MANAGEMENT_SERVICE_H_