src/chrome/browser/chromeos/drive/drive_url_request_job.h

   1 // Copyright 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 #ifndef CHROME_BROWSER_CHROMEOS_DRIVE_DRIVE_URL_REQUEST_JOB_H_
   6 #define CHROME_BROWSER_CHROMEOS_DRIVE_DRIVE_URL_REQUEST_JOB_H_
   7
   8 #include <string>
   9
  10 #include "base/basictypes.h"
  11 #include "base/callback.h"
  12 #include "base/memory/ref_counted.h"
  13 #include "base/memory/scoped_ptr.h"
  14 #include "base/memory/weak_ptr.h"
  15 #include "chrome/browser/chromeos/drive/file_errors.h"
  16 #include "net/http/http_byte_range.h"
  17 #include "net/url_request/url_request_job.h"
  18
  19 namespace base {
  20 class SequencedTaskRunner;
  21 }  // namespace
  22
  23 namespace net {
  24 class IOBuffer;
  25 class NetworkDelegate;
  26 class URLRequest;
  27 }  // namespace net
  28
  29 namespace drive {
  30
  31 class DriveFileStreamReader;
  32 class FileSystemInterface;
  33 class ResourceEntry;
  34
  35 // DriveURLRequestJob is the gateway between network-level drive:...
  36 // requests for drive resources and FileSystem.  It exposes content URLs
  37 // formatted as drive:<drive-file-path>.
  38 // The methods should be run on IO thread, and the operations to communicate
  39 // with a locally cached file will run on |file_task_runner|.
  40 class DriveURLRequestJob : public net::URLRequestJob {
  41  public:
  42
  43   // Callback to return the FileSystemInterface instance. This is an
  44   // injecting point for testing.
  45   // Note that the callback will be copied between threads (IO and UI), and
  46   // will be called on UI thread.
  47   typedef base::Callback<FileSystemInterface*()> FileSystemGetter;
  48
  49   DriveURLRequestJob(const FileSystemGetter& file_system_getter,
  50                      base::SequencedTaskRunner* file_task_runner,
  51                      net::URLRequest* request,
  52                      net::NetworkDelegate* network_delegate);
  53
  54   // net::URLRequestJob overrides:
  55   virtual void SetExtraRequestHeaders(const net::HttpRequestHeaders& headers)
  56       OVERRIDE;
  57   virtual void Start() OVERRIDE;
  58   virtual void Kill() OVERRIDE;
  59   virtual bool GetMimeType(std::string* mime_type) const OVERRIDE;
  60   virtual bool IsRedirectResponse(
  61       GURL* location, int* http_status_code) OVERRIDE;
  62   virtual bool ReadRawData(
  63       net::IOBuffer* buf, int buf_size, int* bytes_read) OVERRIDE;
  64
  65  protected:
  66   virtual ~DriveURLRequestJob();
  67
  68  private:
  69   // Called when the initialization of DriveFileStreamReader is completed.
  70   void OnDriveFileStreamReaderInitialized(
  71       int error, scoped_ptr<ResourceEntry> entry);
  72
  73   // Called when DriveFileStreamReader::Read is completed.
  74   void OnReadCompleted(int read_result);
  75
  76   const FileSystemGetter file_system_getter_;
  77   scoped_refptr<base::SequencedTaskRunner> file_task_runner_;
  78
  79   // The range of the file to be returned.
  80   net::HttpByteRange byte_range_;
  81
  82   scoped_ptr<DriveFileStreamReader> stream_reader_;
  83   scoped_ptr<ResourceEntry> entry_;
  84
  85   // This should remain the last member so it'll be destroyed first and
  86   // invalidate its weak pointers before other members are destroyed.
  87   base::WeakPtrFactory<DriveURLRequestJob> weak_ptr_factory_;
  88   DISALLOW_COPY_AND_ASSIGN(DriveURLRequestJob);
  89 };
  90
  91 }  // namespace drive
  92
  93 #endif  // CHROME_BROWSER_CHROMEOS_DRIVE_DRIVE_URL_REQUEST_JOB_H_