refactor: move module specific code to separate module in frontend-wasm

frontend-wasm/src/code_translator/mod.rs

@@ -17,10 +17,10 @@ use std::collections::{hash_map, HashMap};
 use std::u64;
 
 use crate::error::{WasmError, WasmResult};
-use crate::func_translation_state::ControlStackFrame;
-use crate::func_translation_state::{ElseData, FuncTranslationState};
-use crate::function_builder_ext::FunctionBuilderExt;
-use crate::module_env::ModuleInfo;
+use crate::module::environ::ModuleInfo;
+use crate::module::func_translation_state::ControlStackFrame;
+use crate::module::func_translation_state::{ElseData, FuncTranslationState};
+use crate::module::function_builder_ext::FunctionBuilderExt;
 use crate::ssa::Variable;
 use crate::unsupported_diag;
 use crate::wasm_types::{BlockType, GlobalIndex};

frontend-wasm/src/code_translator/tests_unsupported.rs

@@ -8,10 +8,10 @@ use wasmparser::MemArg;
 use wasmparser::Operator;
 use wasmparser::Operator::*;
 
-use crate::func_translation_state::FuncTranslationState;
-use crate::function_builder_ext::FunctionBuilderContext;
-use crate::function_builder_ext::FunctionBuilderExt;
-use crate::module_env::ModuleInfo;
+use crate::module::environ::ModuleInfo;
+use crate::module::func_translation_state::FuncTranslationState;
+use crate::module::function_builder_ext::FunctionBuilderContext;
+use crate::module::function_builder_ext::FunctionBuilderExt;
 use crate::test_utils::test_diagnostics;
 
 use super::translate_operator;

frontend-wasm/src/lib.rs

@@ -5,15 +5,12 @@
 #![deny(missing_docs)]
 #![deny(rustdoc::broken_intra_doc_links)]
 
+mod component;
+mod module;
+
 mod code_translator;
 mod config;
 mod error;
-mod func_translation_state;
-mod func_translator;
-mod function_builder_ext;
-mod module_env;
-mod module_translator;
-mod sections_translator;
 mod ssa;
 mod translation_utils;
 mod wasm_types;
@@ -23,4 +20,4 @@ mod test_utils;
 
 pub use self::config::WasmTranslationConfig;
 pub use self::error::WasmError;
-pub use self::module_translator::translate_module;
+pub use self::module::translate_module;

frontend-wasm/src/module_env.rs → frontend-wasm/src/module/environ.rs

@@ -1,7 +1,7 @@
 //! Wasm to Miden IR translation environment
 
 use crate::error::{WasmError, WasmResult};
-use crate::func_translator::FuncTranslator;
+use crate::module::func_translator::FuncTranslator;
 use crate::translation_utils::sig_from_funct_type;
 use crate::wasm_types::{
     DataSegment, DataSegmentIndex, DefinedFuncIndex, FuncIndex, Global, GlobalIndex, Memory,

frontend-wasm/src/func_translation_state.rs → frontend-wasm/src/module/func_translation_state.rs

@@ -7,8 +7,8 @@
 
 use crate::{
     error::{WasmError, WasmResult},
-    function_builder_ext::FunctionBuilderExt,
-    module_env::ModuleInfo,
+    module::environ::ModuleInfo,
+    module::function_builder_ext::FunctionBuilderExt,
     translation_utils::sig_from_funct_type,
     wasm_types::{BlockType, FuncIndex},
 };

frontend-wasm/src/func_translator.rs → frontend-wasm/src/module/func_translator.rs

@@ -8,9 +8,9 @@
 
 use crate::code_translator::translate_operator;
 use crate::error::WasmResult;
-use crate::func_translation_state::FuncTranslationState;
-use crate::function_builder_ext::{FunctionBuilderContext, FunctionBuilderExt};
-use crate::module_env::ModuleInfo;
+use crate::module::environ::ModuleInfo;
+use crate::module::func_translation_state::FuncTranslationState;
+use crate::module::function_builder_ext::{FunctionBuilderContext, FunctionBuilderExt};
 use crate::ssa::Variable;
 use crate::translation_utils::emit_zero;
 use crate::wasm_types::valtype_to_type;

frontend-wasm/src/module_translator.rs → frontend-wasm/src/module/mod.rs

@@ -1,9 +1,12 @@
-//! Translation skeleton that traverses the whole WebAssembly module and call helper functions
-//! to deal with each part of it.
+pub(crate) mod environ;
+pub(crate) mod func_translation_state;
+mod func_translator;
+pub(crate) mod function_builder_ext;
+mod sections_translator;
 
 use crate::error::WasmResult;
-use crate::module_env::ModuleEnvironment;
-use crate::sections_translator::{
+use crate::module::environ::ModuleEnvironment;
+use crate::module::sections_translator::{
     parse_data_section, parse_element_section, parse_function_section, parse_global_section,
     parse_import_section, parse_memory_section, parse_name_section, parse_type_section,
 };

frontend-wasm/src/sections_translator.rs → frontend-wasm/src/module/sections_translator.rs

@@ -3,7 +3,7 @@
 
 use crate::{
     error::{WasmError, WasmResult},
-    module_env::ModuleEnvironment,
+    module::environ::ModuleEnvironment,
     unsupported_diag,
     wasm_types::{
         convert_func_type, convert_global_type, DataSegment, DataSegmentIndex, DataSegmentOffset,

frontend-wasm/src/translation_utils.rs

@@ -4,7 +4,7 @@ use miden_diagnostics::SourceSpan;
 use miden_hir::{AbiParam, CallConv, InstBuilder, Linkage, Signature, Value};
 use miden_hir_type::{FunctionType, Type};
 
-use crate::function_builder_ext::FunctionBuilderExt;
+use crate::module::function_builder_ext::FunctionBuilderExt;
 
 /// Emit instructions to produce a zero value in the given type.
 pub fn emit_zero(ty: &Type, builder: &mut FunctionBuilderExt) -> Value {

frontend-wasm/src/wasm_types.rs

@@ -8,7 +8,7 @@ use miden_hir_type::Type;
 
 use crate::error::WasmError;
 use crate::error::WasmResult;
-use crate::module_env::ModuleInfo;
+use crate::module::environ::ModuleInfo;
 use crate::unsupported_diag;
 
 /// Index type of a function (imported or defined) inside the WebAssembly module.