chrome.experimental.downloads

chrome/browser/extensions/extension_downloads.cc

Index: chrome/browser/extensions/extension_downloads.cc
diff --git a/chrome/browser/extensions/extension_downloads.cc b/chrome/browser/extensions/extension_downloads.cc
new file mode 100644
index 0000000000000000000000000000000000000000..dbd0907437d67fb591f5a57c2640db51f7853277
--- /dev/null
+++ b/chrome/browser/extensions/extension_downloads.cc
@@ -0,0 +1,273 @@
+// Copyright (c) 2011 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/browser/extensions/extension_downloads.h"
+
+#include <algorithm>
+#include <string>
+#include <vector>
+
+#include "base/bind.h"
+#include "base/logging.h"
+#include "base/stl_util-inl.h"
+#include "base/values.h"
+#include "chrome/browser/browser_process.h"
+#include "chrome/browser/download/download_create_info.h"
+#include "chrome/browser/download/download_file_manager.h"
+#include "chrome/browser/download/download_item.h"
+#include "chrome/browser/download/download_query.h"
+#include "chrome/browser/download/download_manager.h"
+#include "content/browser/renderer_host/render_view_host.h"
+#include "content/browser/renderer_host/resource_dispatcher_host.h"
+#include "chrome/browser/ui/browser_list.h"
+#include "chrome/browser/icon_loader.h"
+#include "chrome/browser/download/download_util.h"
+#include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
+#include "chrome/browser/icon_manager.h"
+
+bool DownloadsDownloadFunction::RunImpl() {
+  EXTENSION_FUNCTION_VALIDATE(args_->GetDictionary(0, &options_));
+  EXTENSION_FUNCTION_VALIDATE(options_->GetString("url", &url_));
+  if (url_.empty()) {
+    error_ = "'url' cannot be empty.";
+    return false;
+  }
+  rdh_ = g_browser_process->resource_dispatcher_host();
+  if (rdh_ == NULL) {
+    error_ = "I'm afraid I can't do that.";
+    return false;
+  }
+  VLOG(1) << __FUNCTION__ << " " << url_;
+  products_ = new DictionaryValue();
+  result_.reset(products_);
+  options_->GetString("filename", &filename_);
+  options_->GetBoolean("save_as", &save_as_);
+  options_->GetString("method", &method_);
+  options_->GetDictionary("headers", &extra_headers_);
+  options_->GetString("body", &post_body_);
+  // TODO sanity check method_, extra_headers_, filename_
+  dl_man_ = profile()->GetDownloadManager();
+  tab_contents_ = BrowserList::GetLastActive()->GetSelectedTabContentsWrapper()
+    ->tab_contents();
+  resource_context_ = &profile()->GetResourceContext();
+  render_process_host_id_ = tab_contents_->GetRenderProcessHost()->id();
+  render_view_host_routing_id_ = tab_contents_->render_view_host()
+    ->routing_id();
+  VLOG(1) << __FUNCTION__ << " " << url_;
+  if (!BrowserThread::PostTask(BrowserThread::IO, FROM_HERE, NewRunnableMethod(
+          this, &DownloadsDownloadFunction::BeginDownloadOnIOThread))) {
+    error_ = "I'm afraid I can't do that.";
+    return false;
+  }
+  return true;
+}
+
+void DownloadsDownloadFunction::BeginDownloadOnIOThread() {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
+  VLOG(1) << __FUNCTION__ << " " << url_;
+  DownloadSaveInfo save_info;
+  save_info.file_path = FilePath(filename_);
+  net::URLRequest* request = new net::URLRequest(GURL(url_), rdh_);
+  if (method_.empty()) {
+    method_ = "GET";
+  }
+  request->set_method(method_);
+  if (extra_headers_ != NULL) {
+    DictionaryValue::key_iterator headers_end = extra_headers_->end_keys();
+    for (DictionaryValue::key_iterator headers_iter =
+         extra_headers_->begin_keys();
+         headers_iter != headers_end; ++headers_iter) {
+      std::string value;
+      if (extra_headers_->GetStringWithoutPathExpansion(
+            *headers_iter, &value)) {
+        request->SetExtraRequestHeaderByName(
+            *headers_iter, value, false/*overwrite*/);
+      }
+    }
+  }
+  if (!post_body_.empty()) {
+    request->AppendBytesToUpload(post_body_.data(), post_body_.size());
+  }
+  rdh_->BeginDownload(
+      request,
+      save_info,
+      save_as_,
+      base::Bind(&DownloadsDownloadFunction::OnStarted, this),
+      render_process_host_id_,
+      render_view_host_routing_id_,
+      *resource_context_);
+}
+
+void DownloadsDownloadFunction::OnStarted(int dl_id, int error) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
+  VLOG(1) << __FUNCTION__ << " " << url_ << " " << dl_id << " " << error;
+  dl_id_ = dl_id;
+  dl_error_ = error;
+  BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, NewRunnableMethod(
+      this, &DownloadsDownloadFunction::RespondOnUIThread));
+}
+
+void DownloadsDownloadFunction::RespondOnUIThread() {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+  VLOG(1) << __FUNCTION__ << " " << url_ << " " << dl_man_ << " " << filename_
+          << " " << dl_id_ << " " << dl_error_ << " " << method_;
+  if (dl_id_ >= 0) {
+    products_->Set("id", Value::CreateIntegerValue(dl_id_));
+  } else {
+    products_->Set("error", Value::CreateIntegerValue(dl_error_));
+  }
+  SendResponse(true);
+}
+
+bool DownloadsSearchFunction::RunImpl() {
+  DictionaryValue* query_json = NULL;
+  EXTENSION_FUNCTION_VALIDATE(args_->GetDictionary(0, &query_json));
+  CHECK(query_json);
+  download_util::DownloadQuery query(*query_json);
+  DownloadManager* dlman = profile()->GetDownloadManager();
+  if (dlman == NULL) {
+    error_ = "I'm afraid I can't do that.";
+    return false;
+  }
+  ListValue* results = new ListValue();
+  if (!dlman->Search(
+        download_util::DownloadQuery(*query_json),
+        NULL/*DownloadItem results*/,
+        &error_,
+        results)) {
+    VLOG(1) << __PRETTY_FUNCTION__ << " " << error_;
+    return false;
+  }
+  result_.reset(results);
+  return true;
+}
+
+bool DownloadsPauseFunction::RunImpl() {
+  int dl_id = 0;
+  EXTENSION_FUNCTION_VALIDATE(args_->GetInteger(0, &dl_id));
+  DownloadManager* dlman = profile()->GetDownloadManager();
+  if (dlman == NULL) {
+    error_ = "I'm afraid I can't do that.";
+    return false;
+  }
+  DownloadItem* item = dlman->GetDownloadItem(dl_id);
+  if (item == NULL) {
+    error_ = "Non-existent download";
+    return false;
+  }
+  VLOG(1) << __FUNCTION__ << " " << item;
+  if (!item->is_paused()) {
+    item->TogglePause();
+  }
+  return true;
+}
+
+bool DownloadsResumeFunction::RunImpl() {
+  int dl_id = 0;
+  EXTENSION_FUNCTION_VALIDATE(args_->GetInteger(0, &dl_id));
+  DownloadManager* dlman = profile()->GetDownloadManager();
+  if (dlman == NULL) {
+    error_ = "I'm afraid I can't do that.";
+    return false;
+  }
+  DownloadItem* item = dlman->GetDownloadItem(dl_id);
+  if (item == NULL) {
+    error_ = "Non-existent download";
+    return false;
+  }
+  VLOG(1) << __FUNCTION__ << " " << item;
+  if (item->is_paused()) {
+    item->TogglePause();
+  }
+  return true;
+}
+
+bool DownloadsCancelFunction::RunImpl() {
+  int dl_id = 0;
+  EXTENSION_FUNCTION_VALIDATE(args_->GetInteger(0, &dl_id));
+  DownloadManager* dlman = profile()->GetDownloadManager();
+  if (dlman == NULL) {
+    error_ = "I'm afraid I can't do that.";
+    return false;
+  }
+  DownloadItem* item = dlman->GetDownloadItem(dl_id);
+  if (item == NULL) {
+    error_ = "Non-existent download";
+    return false;
+  }
+  VLOG(1) << __FUNCTION__ << " " << item;
+  item->Cancel(true);
+  return true;
+}
+
+bool DownloadsEraseFunction::RunImpl() {
+  return true;
+}
+
+bool DownloadsSetDestinationFunction::RunImpl() {
+  int dl_id = 0;
+  std::string rel_dest_path;
+  if ((args_.get() == NULL) ||
+      (args_->GetSize() < 2) ||
+      !args_->GetInteger(0, &dl_id) ||
+      (dl_id == 0) ||
+      !args_->GetString(1, &rel_dest_path) ||
+      rel_dest_path.empty()) return false;
+  DownloadManager* dlman = profile()->GetDownloadManager();
+  DownloadItem* item = dlman->GetDownloadItem(dl_id);
+  if (item == NULL) return false;
+  VLOG(1) << __FUNCTION__ << " " << dl_id << " " << item << " "
+          << rel_dest_path;
+  return true;
+}
+
+bool DownloadsAcceptDangerFunction::RunImpl() {
+  int dl_id = 0;
+  EXTENSION_FUNCTION_VALIDATE(args_->GetInteger(0, &dl_id));
+  DownloadManager* dlman = profile()->GetDownloadManager();
+  if (dlman == NULL) {
+    error_ = "I'm afraid I can't do that.";
+    return false;
+  }
+  DownloadItem* item = dlman->GetDownloadItem(dl_id);
+  if (item == NULL) {
+    error_ = "Non-existent download";
+    return false;
+  }
+  VLOG(1) << __FUNCTION__ << " " << item;
+  item->Cancel(true);
+  return true;
+}
+
+bool DownloadsShowFunction::RunImpl() {
+  int dl_id = 0;
+  if ((args_.get() == NULL) ||
+      (args_->GetSize() < 1) ||
+      !args_->GetInteger(0, &dl_id) ||
+      (dl_id == 0)) return false;
+  DownloadManager* dlman = profile()->GetDownloadManager();
+  DownloadItem* item = dlman->GetDownloadItem(dl_id);
+  if (item == NULL) return false;
+  VLOG(1) << __FUNCTION__ << " " << item;
+  return true;
+}
+
+bool DownloadsDragFunction::RunImpl() {
+  int dl_id = 0;
+  if ((args_.get() == NULL) ||
+      (args_->GetSize() < 1) ||
+      !args_->GetInteger(0, &dl_id) ||
+      (dl_id == 0)) return false;
+  DownloadManager* dlman = profile()->GetDownloadManager();
+  DownloadItem* item = dlman->GetDownloadItem(dl_id);
+  if (item == NULL) return false;
+  IconManager* im = g_browser_process->icon_manager();
+  gfx::Image* icon = im->LookupIcon(item->GetUserVerifiedFilePath(),
+                                    IconLoader::NORMAL);
+  gfx::NativeView view = BrowserList::GetLastActive()
+    ->GetSelectedTabContentsWrapper()->tab_contents()->GetNativeView();
+  download_util::DragDownload(item, icon, view);
+  VLOG(1) << __FUNCTION__ << " " << dl_id;
+  return true;
+}