MediaPerceptionPrivate API impl and testing.

extensions/browser/api/media_perception_private/media_perception_private_api.cc

 // Copyright 2017 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 "extensions/browser/api/media_perception_private/media_perception_private_api.h"
 
 #include "extensions/browser/extension_function.h"
 
 namespace media_perception = extensions::api::media_perception_private;
 
 namespace extensions {
 
+using CallbackStatus = MediaPerceptionAPIManager::CallbackStatus;
+
+namespace {
+
+const char kErrorStringStatusDbusError[] =
+    "API call failed because the D-Bus communication with the analytics "
+    "process failed.";
+
+const char kErrorStringStatusIdle[] =
+    "API call failed because the analytics process is not running.";
+
+const char kErrorStringStatusLaunching[] =
+    "API call failed because the analytics process is still in the process "
+    "of launching.";
+
+std::string CallbackStatusToErrorMessage(const CallbackStatus& status) {
+  switch (status) {
+    case CallbackStatus::DBUS_ERROR:
+      return kErrorStringStatusDbusError;
+    case CallbackStatus::PROCESS_IDLE_ERROR:
+      return kErrorStringStatusIdle;
+    case CallbackStatus::PROCESS_LAUNCHING_ERROR:
+      return kErrorStringStatusLaunching;
+    default:
+      return "API call failed.";
+  }
+}
+
+}  // namespace
+
 MediaPerceptionPrivateGetStateFunction ::
     MediaPerceptionPrivateGetStateFunction() {}
 
 MediaPerceptionPrivateGetStateFunction ::
     ~MediaPerceptionPrivateGetStateFunction() {}
 
 ExtensionFunction::ResponseAction
 MediaPerceptionPrivateGetStateFunction::Run() {
-  return RespondNow(Error("Not implemented."));
+  MediaPerceptionAPIManager* manager =
+      MediaPerceptionAPIManager::Get(browser_context());
+  DCHECK(manager) << "Can't get manager.";
+  manager->GetState(base::Bind(
+      &MediaPerceptionPrivateGetStateFunction::GetStateCallback, this));
+  return RespondLater();
+}
+
+void MediaPerceptionPrivateGetStateFunction::GetStateCallback(
+    CallbackStatus status,
+    media_perception::State state) {
+  if (status != CallbackStatus::SUCCESS) {
+    Respond(Error(CallbackStatusToErrorMessage(status)));
+    return;
+  }
+  Respond(OneArgument(state.ToValue()));
 }
 
 MediaPerceptionPrivateSetStateFunction ::
     MediaPerceptionPrivateSetStateFunction() {}
 
 MediaPerceptionPrivateSetStateFunction ::
     ~MediaPerceptionPrivateSetStateFunction() {}
 
 ExtensionFunction::ResponseAction
 MediaPerceptionPrivateSetStateFunction::Run() {
-  return RespondNow(Error("Not implemented."));
+  std::unique_ptr<media_perception::SetState::Params> params =
+      media_perception::SetState::Params::Create(*args_);
+  EXTENSION_FUNCTION_VALIDATE(params.get());
+  if (params->state.status != media_perception::STATUS_RUNNING &&
+      params->state.status != media_perception::STATUS_SUSPENDED) {
+    return RespondNow(
+        Error("Status can only be set to RUNNING and SUSPENDED."));
+  }
+
+  // Check that device context is only provided with SetState RUNNING.
+  if (params->state.status != media_perception::STATUS_RUNNING &&
+      params->state.device_context.get() != nullptr) {
+    return RespondNow(
+        Error("Only provide deviceContext with SetState RUNNING."));
+  }
+  MediaPerceptionAPIManager* manager =
+      MediaPerceptionAPIManager::Get(browser_context());
+  DCHECK(manager) << "Can't get manager.";
+  manager->SetState(
+      params->state,
+      base::Bind(&MediaPerceptionPrivateSetStateFunction::SetStateCallback,
+                 this));
+  return RespondLater();
+}
+
+void MediaPerceptionPrivateSetStateFunction::SetStateCallback(
+    CallbackStatus status,
+    media_perception::State state) {
+  if (status != CallbackStatus::SUCCESS) {
+    Respond(Error(CallbackStatusToErrorMessage(status)));
+    return;
+  }
+  Respond(OneArgument(state.ToValue()));
 }
 
 MediaPerceptionPrivateGetDiagnosticsFunction ::
     MediaPerceptionPrivateGetDiagnosticsFunction() {}
 
 MediaPerceptionPrivateGetDiagnosticsFunction ::
     ~MediaPerceptionPrivateGetDiagnosticsFunction() {}
 
 ExtensionFunction::ResponseAction
 MediaPerceptionPrivateGetDiagnosticsFunction::Run() {
-  return RespondNow(Error("Not implemented."));
+  MediaPerceptionAPIManager* manager =
+      MediaPerceptionAPIManager::Get(browser_context());
+  DCHECK(manager) << "Can't get manager.";
+  manager->GetDiagnostics(base::Bind(
+      &MediaPerceptionPrivateGetDiagnosticsFunction::GetDiagnosticsCallback,
+      this));
+  return RespondLater();
+}
+
+void MediaPerceptionPrivateGetDiagnosticsFunction::GetDiagnosticsCallback(
+    CallbackStatus status,
+    media_perception::Diagnostics diagnostics) {
+  if (status != CallbackStatus::SUCCESS) {
+    Respond(Error(CallbackStatusToErrorMessage(status)));
+    return;
+  }
+  Respond(OneArgument(diagnostics.ToValue()));
 }
 
 }  // namespace extensions