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 CHROME_BROWSER_DOWNLOAD_DOWNLOAD_COMPLETION_BLOCKER_H_
6#define CHROME_BROWSER_DOWNLOAD_DOWNLOAD_COMPLETION_BLOCKER_H_
7
8#include "base/callback.h"
9#include "base/supports_user_data.h"
10#include "content/public/browser/download_item.h"
11
12// A subsystem may use a DownloadCompletionBlocker in conjunction with
13// DownloadManagerDelegate::ShouldCompleteDownload() in order to block the
14// completion of a DownloadItem. CompleteDownload() will run the most recent
15// callback set.
16class DownloadCompletionBlocker : public base::SupportsUserData::Data {
17 public:
18  DownloadCompletionBlocker();
19  virtual ~DownloadCompletionBlocker();
20
21  bool is_complete() const { return is_complete_; }
22
23  void set_callback(const base::Closure& callback) {
24    if (!is_complete())
25      callback_ = callback;
26  }
27
28  // Mark this download item as complete with respect to this blocker. (Other
29  // blockers may continue to block the item.) Run |callback_|. This method may
30  // only be called once, so |callback_| will only be called once.  Subclasses
31  // must call the base implementation if they override this method.
32  virtual void CompleteDownload();
33
34 private:
35  bool is_complete_;
36  base::Closure callback_;
37
38  DISALLOW_COPY_AND_ASSIGN(DownloadCompletionBlocker);
39};
40
41#endif  // CHROME_BROWSER_DOWNLOAD_DOWNLOAD_COMPLETION_BLOCKER_H_
42