content/browser/download/drag_download_file.h

   1 // Copyright (c) 2012 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 CONTENT_BROWSER_DOWNLOAD_DRAG_DOWNLOAD_FILE_H_
   6 #define CONTENT_BROWSER_DOWNLOAD_DRAG_DOWNLOAD_FILE_H_
   7
   8 #include "base/compiler_specific.h"
   9 #include "base/files/file_path.h"
  10 #include "base/memory/scoped_ptr.h"
  11 #include "base/memory/weak_ptr.h"
  12 #include "base/run_loop.h"
  13 #include "content/browser/download/download_file.h"
  14 #include "content/common/content_export.h"
  15 #include "content/public/browser/download_id.h"
  16 #include "content/public/browser/download_item.h"
  17 #include "content/public/browser/download_manager.h"
  18 #include "content/public/common/referrer.h"
  19 #include "googleurl/src/gurl.h"
  20 #include "net/base/file_stream.h"
  21 #include "ui/base/dragdrop/download_file_interface.h"
  22 #include "ui/base/ui_export.h"
  23
  24 namespace net {
  25 class FileStream;
  26 }
  27
  28 namespace content {
  29
  30 class DownloadManager;
  31 class WebContents;
  32
  33 // This class implements downloading a file via dragging virtual files out of
  34 // the browser:
  35 // http://lists.whatwg.org/htdig.cgi/whatwg-whatwg.org/2009-August/022118.html
  36 // http://www.html5rocks.com/en/tutorials/casestudies/box_dnd_download/
  37 class CONTENT_EXPORT DragDownloadFile : public ui::DownloadFileProvider {
  38  public:
  39   // On Windows, we need to download into a temporary file. On posix, we need to
  40   // download into a file stream that has already been created, so only the UI
  41   // thread is involved. |file_stream| must be null on windows but non-null on
  42   // posix systems. |file_path| is an absolute path on all systems.
  43   DragDownloadFile(const base::FilePath& file_path,
  44                    scoped_ptr<net::FileStream> file_stream,
  45                    const GURL& url,
  46                    const Referrer& referrer,
  47                    const std::string& referrer_encoding,
  48                    WebContents* web_contents);
  49
  50   // DownloadFileProvider methods.
  51   virtual void Start(ui::DownloadFileObserver* observer) OVERRIDE;
  52   virtual bool Wait() OVERRIDE;
  53   virtual void Stop() OVERRIDE;
  54
  55  private:
  56   class DragDownloadFileUI;
  57   enum State {INITIALIZED, STARTED, SUCCESS, FAILURE};
  58
  59   virtual ~DragDownloadFile();
  60
  61   void DownloadCompleted(bool is_successful);
  62   void CheckThread();
  63
  64   base::FilePath file_path_;
  65   scoped_ptr<net::FileStream> file_stream_;
  66   base::MessageLoop* drag_message_loop_;
  67   State state_;
  68   scoped_refptr<ui::DownloadFileObserver> observer_;
  69   base::RunLoop nested_loop_;
  70   DragDownloadFileUI* drag_ui_;
  71   base::WeakPtrFactory<DragDownloadFile> weak_ptr_factory_;
  72
  73   DISALLOW_COPY_AND_ASSIGN(DragDownloadFile);
  74 };
  75
  76 }  // namespace content
  77
  78 #endif  // CONTENT_BROWSER_DOWNLOAD_DRAG_DOWNLOAD_FILE_H_