Add App.enableComponent and App.updateComponent for Widevine support

rogerwang · rogerwang · commit ca51b02d4a19 · 2020-11-21T18:30:20.000-07:00
Ref nwjs#6425
diff --git a/src/api/nw_app.idl b/src/api/nw_app.idl
@@ -5,7 +5,12 @@
 // nw APP API
 [implemented_in="content/nw/src/api/nw_app_api.h"]
 namespace nw.App {
+  enum ComponentExtensions {
+    WIDEVINE
+  };
   callback EventCallback = void();
+  callback ErrorCallback = void(boolean result);
+  callback ComponentCallback = void(DOMString version);
   interface Functions {
     // crash the renderer for testing
     [nocompile] static void crashRenderer();
@@ -15,6 +20,8 @@ namespace nw.App {
     [nocompile] static void removeListener(DOMString event,
                                            EventCallback callback);
     [nocompile] static void removeAllListeners(DOMString event);
+    static void enableComponent(ComponentExtensions extension_id, ComponentCallback callback);
+    static void updateComponent(ComponentExtensions extension_id, ErrorCallback callback);
     static void quit();
     static void closeAllWindows();
     static void clearCache();
diff --git a/src/api/nw_app_api.cc b/src/api/nw_app_api.cc
@@ -11,6 +11,9 @@
 #include "content/public/browser/browser_task_traits.h"
 #include "components/browsing_data/content/appcache_helper.h"
 #include "components/browsing_data/content/browsing_data_helper.h"
+#include "components/component_updater/component_updater_service.h"
+#include "chrome/browser/browser_process.h"
+#include "chrome/browser/component_updater/widevine_cdm_component_installer.h"
 #include "chrome/browser/devtools/devtools_window.h"
 #include "chrome/browser/extensions/devtools_util.h"
 #include "chrome/browser/extensions/extension_service.h"
@@ -99,6 +102,49 @@ NwAppCloseAllWindowsFunction::Run() {
   return RespondNow(NoArguments());
 }
 
+ExtensionFunction::ResponseAction
+NwAppEnableComponentFunction::Run() {
+  component_updater::RegisterWidevineCdmComponent(g_browser_process->component_updater(),
+                                                  base::BindOnce(&NwAppEnableComponentFunction::OnRegistered,
+                                                                 this));
+  return RespondLater();
+}
+
+void NwAppEnableComponentFunction::OnRegistered() {
+  std::string ret;
+  const std::vector<component_updater::ComponentInfo> component_list =
+    g_browser_process->component_updater()->GetComponents();
+
+  for (const auto& component : component_list) {
+    if (component.id == "oimompecagnajdejgnnjijobebaeigek") {
+      ret = component.version.GetString();
+    }
+  }
+  auto result_value = std::make_unique<base::Value>(ret);
+
+  Respond(OneArgument(std::move(result_value)));
+}
+
+ExtensionFunction::ResponseAction
+  NwAppUpdateComponentFunction::Run() {
+  g_browser_process->component_updater()->GetOnDemandUpdater().OnDemandUpdate(
+   "oimompecagnajdejgnnjijobebaeigek", component_updater::OnDemandUpdater::Priority::FOREGROUND,
+   base::BindOnce(&NwAppUpdateComponentFunction::OnUpdated, this));
+
+  return RespondLater();
+}
+
+void NwAppUpdateComponentFunction::OnUpdated(update_client::Error error) {
+  bool ret = (error == update_client::Error::NONE);
+  auto result_value = std::make_unique<base::Value>(ret);
+  if (ret) {
+    Respond(OneArgument(std::move(result_value)));
+  } else {
+    VLOG(1) << "update component error: " << (int)error;
+    Respond(OneArgument(std::move(result_value)));
+  }
+}
+
 NwAppGetArgvSyncFunction::NwAppGetArgvSyncFunction() {
 }
 
diff --git a/src/api/nw_app_api.h b/src/api/nw_app_api.h
@@ -6,6 +6,7 @@
 #include "base/run_loop.h"
 #include "content/public/browser/browsing_data_remover.h"
 #include "extensions/browser/extension_function.h"
+#include "components/update_client/update_client_errors.h"
 
 namespace extensions {
 class AppWindowRegistry;
@@ -39,6 +40,32 @@ class NwAppCloseAllWindowsFunction : public ExtensionFunction {
   DECLARE_EXTENSION_FUNCTION("nw.App.closeAllWindows", UNKNOWN)
 };
 
+class NwAppEnableComponentFunction : public ExtensionFunction {
+ public:
+  NwAppEnableComponentFunction() {}
+
+ protected:
+  ~NwAppEnableComponentFunction() override {}
+
+  void OnRegistered();
+  // ExtensionFunction:
+  ResponseAction Run() override;
+  DECLARE_EXTENSION_FUNCTION("nw.App.enableComponent", UNKNOWN)
+};
+
+class NwAppUpdateComponentFunction : public ExtensionFunction {
+ public:
+  NwAppUpdateComponentFunction() {}
+
+ protected:
+  ~NwAppUpdateComponentFunction() override {}
+
+  void OnUpdated(update_client::Error error);
+  // ExtensionFunction:
+  ResponseAction Run() override;
+  DECLARE_EXTENSION_FUNCTION("nw.App.updateComponent", UNKNOWN)
+};
+
 class NwAppGetArgvSyncFunction : public NWSyncExtensionFunction {
  public:
   NwAppGetArgvSyncFunction();
