1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72
   73
   74
   75
   76
   77
   78
   79
   80
   81
   82
   83
   84
   85
   86
   87
   88
   89
   90
   91
   92
   93
   94
   95
   96
   97
   98
   99
  100
  101
  102
  103
  104
  105
  106
  107


content / browser / devtools / protocol / devtools_download_manager_delegate.h [blame]

// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CONTENT_BROWSER_DEVTOOLS_PROTOCOL_DEVTOOLS_DOWNLOAD_MANAGER_DELEGATE_H_
#define CONTENT_BROWSER_DEVTOOLS_PROTOCOL_DEVTOOLS_DOWNLOAD_MANAGER_DELEGATE_H_

#include <stdint.h>

#include <string>

#include "base/functional/callback_forward.h"
#include "base/memory/raw_ptr.h"
#include "base/memory/ref_counted.h"
#include "content/public/browser/download_manager_delegate.h"

namespace base {
class FilePath;
}

namespace content {

class DownloadManager;

namespace protocol {

class DevToolsDownloadManagerDelegate
    : public base::SupportsUserData::Data,
      public content::DownloadManagerDelegate {
 public:
  enum class DownloadBehavior {
    // All downloads are denied.
    DENY,

    // All downloads are accepted.
    ALLOW,

    // All downloads are accepted and named using Guids.
    ALLOW_AND_NAME,

    // Use default download behavior if available, otherwise deny.
    DEFAULT
  };

  // Takes over the |browser_Context|'s download manager.
  // When existing delegate is set, this proxy will use the original's
  // |GetNextId| function to ensure compatibility. It will also call its
  // |Shutdown| method when shutting down and it will fallback to the original
  // delegate if it cannot find any DevToolsDownloadManagerHelper associated
  // with the download.
  static DevToolsDownloadManagerDelegate* GetOrCreateInstance(
      content::BrowserContext* browser_Context);
  static DevToolsDownloadManagerDelegate* GetInstance(
      content::BrowserContext* browser_Context);

  DevToolsDownloadManagerDelegate(const DevToolsDownloadManagerDelegate&) =
      delete;
  DevToolsDownloadManagerDelegate& operator=(
      const DevToolsDownloadManagerDelegate&) = delete;

  ~DevToolsDownloadManagerDelegate() override = default;

  void set_download_behavior(DownloadBehavior behavior) {
    download_behavior_ = behavior;
  }
  void set_download_path(const std::string& path) { download_path_ = path; }

  // DownloadManagerDelegate overrides.
  void Shutdown() override;
  bool DetermineDownloadTarget(
      download::DownloadItem* download,
      download::DownloadTargetCallback* callback) override;
  bool ShouldOpenDownload(
      download::DownloadItem* item,
      content::DownloadOpenDelayedCallback callback) override;
  void GetNextId(content::DownloadIdCallback callback) override;
  download::DownloadItem* GetDownloadByGuid(const std::string& guid) override;

 private:
  friend class base::RefCounted<DevToolsDownloadManagerDelegate>;

  explicit DevToolsDownloadManagerDelegate(BrowserContext* browser_context);

  using FilenameDeterminedCallback =
      base::OnceCallback<void(const base::FilePath&)>;

  static void GenerateFilename(const GURL& url,
                               const std::string& content_disposition,
                               const std::string& suggested_filename,
                               const std::string& mime_type,
                               const base::FilePath& suggested_directory,
                               FilenameDeterminedCallback callback);

  void OnDownloadPathGenerated(uint32_t download_id,
                               download::DownloadTargetCallback callback,
                               const base::FilePath& suggested_path);

  raw_ptr<content::DownloadManager> download_manager_;
  raw_ptr<content::DownloadManagerDelegate> original_download_delegate_;
  DownloadBehavior download_behavior_ = DownloadBehavior::DEFAULT;
  std::string download_path_;
};

}  // namespace protocol
}  // namespace content

#endif  // CONTENT_BROWSER_DEVTOOLS_PROTOCOL_DEVTOOLS_DOWNLOAD_MANAGER_DELEGATE_H_