1 files changed, 39 insertions, 0 deletions
diff --git a/toolkit/components/uniffi-bindgen-gecko-js/src/render/shared.rs b/toolkit/components/uniffi-bindgen-gecko-js/src/render/shared.rs
new file mode 100644
index 0000000000..c356158715
--- /dev/null
+++ b/toolkit/components/uniffi-bindgen-gecko-js/src/render/shared.rs
@@ -0,0 +1,39 @@
+/* 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/. */
+
+/// Extension traits that are shared across multiple render targets
+use crate::Config;
+use extend::ext;
+use uniffi_bindgen::interface::{Function, Method, Object};
+
+fn is_async(config: &Config, spec: &str) -> bool {
+    if config.receiver_thread.main.contains(spec) {
+        false
+    } else if config.receiver_thread.worker.contains(spec) {
+        true
+    } else {
+        match &config.receiver_thread.default {
+            Some(t) => t != "main",
+            _ => true,
+        }
+    }
+}
+
+#[ext]
+pub impl Function {
+    fn is_async(&self, config: &Config) -> bool {
+        is_async(config, self.name())
+    }
+}
+
+#[ext]
+pub impl Object {
+    fn is_constructor_async(&self, config: &Config) -> bool {
+        is_async(config, self.name())
+    }
+
+    fn is_method_async(&self, method: &Method, config: &Config) -> bool {
+        is_async(config, &format!("{}.{}", self.name(), method.name()))
+    }
+}