waalge
/
aiken
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat(Project::export): use Export::from_function and transpose 

Co-authored-by: Kasey White <kwhitemsg@gmail.com>
This commit is contained in:
rvcas 2024-04-03 19:17:42 -04:00 committed by Lucas
parent 8ed930ac5a
commit dac3308620
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
crates/aiken-project/src/lib.rs
View File

@ -42,6 +42,7 @@ use aiken_lang::{
tipo::{Type, TypeInfo},
IdGenerator,
};
use export::Export;
use indexmap::IndexMap;
use miette::NamedSource;
use options::{CodeGenMode, Options};
@ -465,22 +466,21 @@ where
})
}
pub fn export(&self, module: &str, name: &str) -> Result<String, Error> {
let mut generator = self.new_generator(Tracing::silent());
pub fn export(&self, module: &str, name: &str) -> Result<Export, Error> {
self.checked_modules
.get(module)
.and_then(|checked_module| {
checked_module.ast.definitions().find_map(|def| match def {
Definition::Fn(func) if func.name == name => Some(func),
Definition::Fn(func) if func.name == name => Some((checked_module, func)),
_ => None,
})
})
.map(|func| {
let program = generator.generate_raw(&func.body, &func.arguments, module);
.map(|(checked_module, func)| {
let mut generator = self.new_generator(Tracing::silent());
program.to_debruijn().unwrap().to_hex().unwrap()
Export::from_function(func, checked_module, &mut generator, &self.checked_modules)
})
.transpose()?
.ok_or_else(|| Error::ExportNotFound {
module: module.to_string(),
name: name.to_string(),