Home | History | Annotate | Download | only in drive 
      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_
     94