Implement Chrome Extension TabCapture API.

chrome/browser/extensions/api/tab_capture/tab_capture_api.cc

+// Copyright (c) 2012 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.
+
+// Implements the Chrome Extensions Tab Capture API.
+
+#include "chrome/browser/extensions/api/tab_capture/tab_capture_api.h"
+
+#include "base/stringprintf.h"
+#include "base/values.h"
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/extensions/api/tab_capture/tab_capture_registry.h"
+#include "chrome/browser/extensions/api/tab_capture/tab_capture_registry_factory.h"
+#include "chrome/browser/extensions/browser_event_router.h"
+#include "chrome/browser/extensions/event_names.h"
+#include "chrome/browser/extensions/extension_tab_id_map.h"
+#include "chrome/common/extensions/feature_switch.h"
+
+namespace TabCapture = extensions::api::tab_capture;
+namespace GetCapturedTabs = TabCapture::GetCapturedTabs;
+
+namespace extensions {
+
+namespace {
+
+const char kPermissionError[] =
+    "Extension does not have permission for tab capture.";
+const char kErrorTabIdNotFound[] = "Could not find the specified tab.";
+const char kCapturingSameTab[] = "Cannot capture a tab with an active stream.";
+
+// Keys/values for media stream constraints.
+const char kMediaStreamSource[] = "chromeMediaSource";
+const char kMediaStreamSourceId[] = "chromeMediaSourceId";
+const char kMediaStreamSourceTab[] = "tab";
+
+}  // namespace
+
+TabCaptureCaptureFunction::TabCaptureCaptureFunction() { }
+TabCaptureCaptureFunction::~TabCaptureCaptureFunction() { }
+
+bool TabCaptureCaptureFunction::Prepare() {
+  if (!GetExtension()->HasAPIPermission(APIPermission::kTabCapture) ||
+      !FeatureSwitch::tab_capture()->IsEnabled()) {
+    error_ = kPermissionError;
+    return false;
+  }
+
+  params_ = TabCapture::Capture::Params::Create(*args_);
+  EXTENSION_FUNCTION_VALIDATE(params_.get());
+  return true;
+}
+
+void TabCaptureCaptureFunction::Work() {
+  ExtensionTabIdMap* tab_id_map = ExtensionTabIdMap::GetInstance();
+  int render_process_id = -1;
+  int routing_id = -1;
+  int tab_id = *params_->tab_id.get();
+  if (!tab_id_map ||
+      !tab_id_map->GetProcessAndRoutingId(tab_id,
+                                          &render_process_id,
+                                          &routing_id)) {
+    error_ = kErrorTabIdNotFound;
+    SetResult(base::Value::CreateIntegerValue(0));
+    return;
+  }
+
+  std::string device_id = base::StringPrintf("%i:%i", render_process_id,
+      routing_id);
+
+  request_key_ = device_id;
+  request_ = TabCaptureRegistry::TabCaptureRequest(GetExtension()->id(), tab_id,
+      tab_capture::TAB_CAPTURE_TAB_CAPTURE_STATE_NONE);
+
+
+  bool audio = false;
+  bool video = false;
+  if (params_->options.get()) {
+    if (params_->options->audio.get())
+      audio = *params_->options->audio.get();
+    if (params_->options->video.get())
+      video = *params_->options->video.get();
+  }
+
+  base::DictionaryValue* result = new base::DictionaryValue();
+  if (video) {
+    result->SetString(std::string("videoConstraints.mandatory.") +
+        kMediaStreamSource, kMediaStreamSourceTab);
+    result->SetString(std::string("videoConstraints.mandatory.") +
+        kMediaStreamSourceId, device_id);
+  } else {
+    result->SetBoolean(std::string("videoConstraints"), false);
+  }
+
+  if (audio) {
+    result->SetString(std::string("audioConstraints.mandatory.") +
+        kMediaStreamSource, kMediaStreamSourceTab);
+    result->SetString(std::string("audioConstraints.mandatory.") +
+        kMediaStreamSourceId, device_id);
+  } else {
+    result->SetBoolean(std::string("audioConstraints"), false);
+  }
+
+  SetResult(result);
+}
+
+bool TabCaptureCaptureFunction::Respond() {
+  extensions::TabCaptureRegistry* registry =
+      extensions::TabCaptureRegistryFactory::GetForProfile(profile());
+  if (!request_key_.empty() && !registry->AddRequest(request_key_, request_)) {
+    error_ = kCapturingSameTab;
+    SetResult(base::Value::CreateIntegerValue(0));
+  }
+
+  return error_.empty();
+}
+
+bool TabCaptureGetCapturedTabsFunction::RunImpl() {
+  if (!GetExtension()->HasAPIPermission(APIPermission::kTabCapture) ||
+      !FeatureSwitch::tab_capture()->IsEnabled()) {
+    error_ = kPermissionError;
+    return false;
+  }
+
+  extensions::TabCaptureRegistry* registry =
+      extensions::TabCaptureRegistryFactory::GetForProfile(profile());
+
+  const TabCaptureRegistry::CaptureRequestList& captured_tabs =
+      registry->GetCapturedTabs(GetExtension()->id());
+
+  base::ListValue *list = new base::ListValue();
+  for (TabCaptureRegistry::CaptureRequestList::const_iterator it =
+       captured_tabs.begin(); it != captured_tabs.end(); it++) {
+    scoped_ptr<tab_capture::CaptureInfo> info(new tab_capture::CaptureInfo());
+    info->tab_id = it->tab_id;
+    info->status = it->status;
+    list->Append(info->ToValue().release());
+  }
+
+  SetResult(list);
+  return true;
+}
+
+}  // namespace extensions