1 files changed, 90 insertions, 0 deletions
diff --git a/devtools/client/accessibility/components/AuditController.js b/devtools/client/accessibility/components/AuditController.js
new file mode 100644
index 0000000000..e6ada9a0a9
--- /dev/null
+++ b/devtools/client/accessibility/components/AuditController.js
@@ -0,0 +1,90 @@
+/* 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 React = require("resource://devtools/client/shared/vendor/react.js");
+const PropTypes = require("resource://devtools/client/shared/vendor/react-prop-types.js");
+
+class AuditController extends React.Component {
+  static get propTypes() {
+    return {
+      accessibleFront: PropTypes.object.isRequired,
+      children: PropTypes.any,
+    };
+  }
+
+  constructor(props) {
+    super(props);
+
+    const {
+      accessibleFront: { checks },
+    } = props;
+    this.state = {
+      checks,
+    };
+
+    this.onAudited = this.onAudited.bind(this);
+  }
+
+  // FIXME: https://bugzilla.mozilla.org/show_bug.cgi?id=1774507
+  UNSAFE_componentWillMount() {
+    const { accessibleFront } = this.props;
+    accessibleFront.on("audited", this.onAudited);
+  }
+
+  componentDidMount() {
+    this.maybeRequestAudit();
+  }
+
+  componentDidUpdate() {
+    this.maybeRequestAudit();
+  }
+
+  componentWillUnmount() {
+    const { accessibleFront } = this.props;
+    accessibleFront.off("audited", this.onAudited);
+  }
+
+  onAudited() {
+    const { accessibleFront } = this.props;
+    if (accessibleFront.isDestroyed()) {
+      // Accessible front is being removed, stop listening for 'audited' events.
+      accessibleFront.off("audited", this.onAudited);
+      return;
+    }
+
+    this.setState({ checks: accessibleFront.checks });
+  }
+
+  maybeRequestAudit() {
+    const { accessibleFront } = this.props;
+    if (accessibleFront.isDestroyed()) {
+      // Accessible front is being removed, stop listening for 'audited' events.
+      accessibleFront.off("audited", this.onAudited);
+      return;
+    }
+
+    if (accessibleFront.checks) {
+      return;
+    }
+
+    accessibleFront.audit().catch(error => {
+      // If the actor was destroyed (due to a connection closed for instance) do
+      // nothing, otherwise log a warning
+      if (!accessibleFront.isDestroyed()) {
+        console.warn(error);
+      }
+    });
+  }
+
+  render() {
+    const { children } = this.props;
+    const { checks } = this.state;
+
+    return React.Children.only(React.cloneElement(children, { checks }));
+  }
+}
+
+module.exports = AuditController;