1 files changed, 84 insertions, 0 deletions
diff --git a/devtools/client/aboutdebugging/src/middleware/worker-component-data.js b/devtools/client/aboutdebugging/src/middleware/worker-component-data.js
new file mode 100644
index 0000000000..8507c50160
--- /dev/null
+++ b/devtools/client/aboutdebugging/src/middleware/worker-component-data.js
@@ -0,0 +1,84 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+"use strict";
+
+const { Ci } = require("chrome");
+
+const {
+  DEBUG_TARGETS,
+  REQUEST_WORKERS_SUCCESS,
+  SERVICE_WORKER_FETCH_STATES,
+  SERVICE_WORKER_STATUSES,
+} = require("devtools/client/aboutdebugging/src/constants");
+
+/**
+ * This middleware converts workers object that get from DevToolsClient.listAllWorkers()
+ * to data which is used in DebugTargetItem.
+ */
+const workerComponentDataMiddleware = store => next => action => {
+  switch (action.type) {
+    case REQUEST_WORKERS_SUCCESS: {
+      action.otherWorkers = toComponentData(action.otherWorkers);
+      action.serviceWorkers = toComponentData(action.serviceWorkers, true);
+      action.sharedWorkers = toComponentData(action.sharedWorkers);
+      break;
+    }
+  }
+
+  return next(action);
+};
+
+function getServiceWorkerStatus(worker) {
+  const isActive = worker.state === Ci.nsIServiceWorkerInfo.STATE_ACTIVATED;
+  const isRunning = !!worker.workerDescriptorFront;
+
+  if (isActive && isRunning) {
+    return SERVICE_WORKER_STATUSES.RUNNING;
+  } else if (isActive) {
+    return SERVICE_WORKER_STATUSES.STOPPED;
+  }
+
+  // We cannot get service worker registrations unless the registration is in
+  // ACTIVE state. Unable to know the actual state ("installing", "waiting"), we
+  // display a custom state "registering" for now. See Bug 1153292.
+  return SERVICE_WORKER_STATUSES.REGISTERING;
+}
+
+function toComponentData(workers, isServiceWorker) {
+  return workers.map(worker => {
+    // Here `worker` is the worker object created by RootFront.listAllWorkers
+    const type = DEBUG_TARGETS.WORKER;
+    const icon = "chrome://devtools/skin/images/debugging-workers.svg";
+    let { fetch } = worker;
+    const { id, name, registrationFront, scope, subscription } = worker;
+
+    let pushServiceEndpoint = null;
+    let status = null;
+
+    if (isServiceWorker) {
+      fetch = fetch
+        ? SERVICE_WORKER_FETCH_STATES.LISTENING
+        : SERVICE_WORKER_FETCH_STATES.NOT_LISTENING;
+      status = getServiceWorkerStatus(worker);
+      pushServiceEndpoint = subscription ? subscription.endpoint : null;
+    }
+
+    return {
+      details: {
+        fetch,
+        pushServiceEndpoint,
+        registrationFront,
+        scope,
+        status,
+      },
+      icon,
+      id,
+      name,
+      type,
+    };
+  });
+}
+
+module.exports = workerComponentDataMiddleware;