Refactor to minimize dependence on codegen trait

bytecodealliance · Oct 22, 2024 · e1425ad · e1425ad
1 parent 4a7f0ad
commit e1425ad
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 24 deletions.
diff --git a/crates/cli/src/codegen/builder.rs b/crates/cli/src/codegen/builder.rs
@@ -83,11 +83,8 @@ impl CodeGenBuilder {
     }
 
     /// Build a [`CodeGenerator`].
-    pub fn build<T>(self, js_runtime_config: Config) -> Result<Box<dyn CodeGen>>
-    where
-        T: CodeGen,
-    {
-        match T::classify() {
+    pub fn build(self, ty: CodeGenType, js_runtime_config: Config) -> Result<Box<dyn CodeGen>> {
+        match ty {
             CodeGenType::Static => self.build_static(js_runtime_config),
             CodeGenType::Dynamic => {
                 if js_runtime_config != Config::default() {

diff --git a/crates/cli/src/codegen/dynamic.rs b/crates/cli/src/codegen/dynamic.rs
@@ -1,6 +1,6 @@
 use super::transform::{self, SourceCodeSection};
 use crate::{
-    codegen::{exports, CodeGen, CodeGenType, Exports, WitOptions},
+    codegen::{exports, CodeGen, Exports, WitOptions},
     js::JS,
     providers::Provider,
 };
@@ -280,10 +280,6 @@ impl CodeGen for DynamicGenerator {
         print_wat(&wasm)?;
         Ok(wasm)
     }
-
-    fn classify() -> CodeGenType {
-        CodeGenType::Dynamic
-    }
 }
 
 #[cfg(feature = "dump_wat")]

diff --git a/crates/cli/src/codegen/mod.rs b/crates/cli/src/codegen/mod.rs
@@ -61,8 +61,4 @@ pub(crate) enum CodeGenType {
 pub(crate) trait CodeGen {
     /// Generate Wasm from a given JS source.
     fn generate(&mut self, source: &JS) -> anyhow::Result<Vec<u8>>;
-    /// Classify the [`CodeGen`] type.
-    fn classify() -> CodeGenType
-    where
-        Self: Sized;
 }
diff --git a/crates/cli/src/codegen/static.rs b/crates/cli/src/codegen/static.rs
@@ -12,7 +12,7 @@ use crate::{
     codegen::{
         exports,
         transform::{self, SourceCodeSection},
-        CodeGen, CodeGenType, Exports, WitOptions,
+        CodeGen, Exports, WitOptions,
     },
     js::JS,
 };
@@ -148,10 +148,6 @@ impl CodeGen for StaticGenerator {
         transform::add_producers_section(&mut module.producers);
         Ok(module.emit_wasm())
     }
-
-    fn classify() -> CodeGenType {
-        CodeGenType::Static
-    }
 }
 
 fn export_exported_js_functions(

diff --git a/crates/cli/src/main.rs b/crates/cli/src/main.rs
@@ -10,7 +10,7 @@ use crate::codegen::WitOptions;
 use crate::commands::{Cli, Command, EmitProviderCommandOpts};
 use anyhow::Result;
 use clap::Parser;
-use codegen::{CodeGenBuilder, DynamicGenerator, StaticGenerator};
+use codegen::{CodeGenBuilder, CodeGenType};
 use commands::{CodegenOptionGroup, JsOptionGroup};
 use javy_config::Config;
 use js::JS;
@@ -49,9 +49,9 @@ fn main() -> Result<()> {
 
             let config = Config::default();
             let mut gen = if opts.dynamic {
-                builder.build::<DynamicGenerator>(config)?
+                builder.build(CodeGenType::Dynamic, config)?
             } else {
-                builder.build::<StaticGenerator>(config)?
+                builder.build(CodeGenType::Static, config)?
             };
 
             let wasm = gen.generate(&js)?;
@@ -70,9 +70,9 @@ fn main() -> Result<()> {
 
             let js_opts: JsOptionGroup = opts.js.clone().into();
             let mut gen = if codegen.dynamic {
-                builder.build::<DynamicGenerator>(js_opts.into())?
+                builder.build(CodeGenType::Dynamic, js_opts.into())?
             } else {
-                builder.build::<StaticGenerator>(js_opts.into())?
+                builder.build(CodeGenType::Static, js_opts.into())?
             };
 
             let wasm = gen.generate(&js)?;