use super::errors::ITParserError;
use super::custom::IT_SECTION_NAME;
use walrus::ModuleConfig;
use std::path::PathBuf;
pub fn delete_it_section_from_file(
in_wasm_path: PathBuf,
out_wasm_path: PathBuf,
) -> Result<(), ITParserError> {
let module = ModuleConfig::new()
.parse_file(&in_wasm_path)
.map_err(ITParserError::CorruptedWasmFile)?;
let mut module = delete_it_section(module);
module
.emit_wasm_file(&out_wasm_path)
.map_err(ITParserError::WasmEmitError)?;
Ok(())
}
pub fn delete_it_section(mut wasm_module: walrus::Module) -> walrus::Module {
let wit_section_ids = wasm_module
.customs
.iter()
.filter_map(|(id, section)| {
if section.name() == IT_SECTION_NAME {
Some(id)
} else {
None
}
})
.collect::<Vec<_>>();
for id in wit_section_ids {
wasm_module.customs.delete(id);
}
wasm_module
}