feat: use new generate_raw function

This commit is contained in:
rvcas 2024-03-04 12:57:15 -05:00 committed by Lucas
parent 7d67f1497c
commit 1d462314c4
2 changed files with 6 additions and 11 deletions

View File

@ -31,11 +31,11 @@ use crate::{
}; };
use aiken_lang::{ use aiken_lang::{
ast::{ ast::{
DataTypeKey, Definition, FunctionAccessKey, ModuleKind, Tracing, TypedDataType, DataTypeKey, Definition, FunctionAccessKey, ModuleKind, Span, Tracing, TypedDataType,
TypedFunction, TypedFunction,
}, },
builtins, builtins::{self, function},
expr::UntypedExpr, expr::{TypedExpr, UntypedExpr},
gen_uplc::CodeGenerator, gen_uplc::CodeGenerator,
line_numbers::LineNumbers, line_numbers::LineNumbers,
tipo::{Type, TypeInfo}, tipo::{Type, TypeInfo},
@ -465,12 +465,7 @@ where
} }
pub fn export(&self, module: &str, name: &str) -> Result<String, Error> { pub fn export(&self, module: &str, name: &str) -> Result<String, Error> {
let mut generator = self.checked_modules.new_generator( let mut generator = self.new_generator(Tracing::silent());
&self.functions,
&self.data_types,
&self.module_types,
Tracing::silent(),
);
self.checked_modules self.checked_modules
.get(module) .get(module)
@ -495,7 +490,7 @@ where
}) })
}) })
.map(|body| { .map(|body| {
let program = generator.generate_test(&body, &name.to_string()); let program = generator.generate_raw(&body, &[], module);
program.to_pretty() program.to_pretty()
}) })

View File

@ -23,7 +23,7 @@ pub fn exec(
name, name,
}: Args, }: Args,
) -> miette::Result<()> { ) -> miette::Result<()> {
with_project(directory.as_deref(), false, |p| { with_project(directory.as_deref(), 0, false, |p| {
p.compile(Options::default())?; p.compile(Options::default())?;
let raw_uplc = p.export(&module, &name)?; let raw_uplc = p.export(&module, &name)?;