From 1d462314c47267e29c0a4e70fcb5f6cc18acc965 Mon Sep 17 00:00:00 2001
From: rvcas <x@rvcas.dev>
Date: Mon, 4 Mar 2024 12:57:15 -0500
Subject: [PATCH] feat: use new generate_raw function

---
 crates/aiken-project/src/lib.rs | 15 +++++----------
 crates/aiken/src/cmd/export.rs  |  2 +-
 2 files changed, 6 insertions(+), 11 deletions(-)

diff --git a/crates/aiken-project/src/lib.rs b/crates/aiken-project/src/lib.rs
index b988b7a9..5e00bd43 100644
--- a/crates/aiken-project/src/lib.rs
+++ b/crates/aiken-project/src/lib.rs
@@ -31,11 +31,11 @@ use crate::{
 };
 use aiken_lang::{
     ast::{
-        DataTypeKey, Definition, FunctionAccessKey, ModuleKind, Tracing, TypedDataType,
+        DataTypeKey, Definition, FunctionAccessKey, ModuleKind, Span, Tracing, TypedDataType,
         TypedFunction,
     },
-    builtins,
-    expr::UntypedExpr,
+    builtins::{self, function},
+    expr::{TypedExpr, UntypedExpr},
     gen_uplc::CodeGenerator,
     line_numbers::LineNumbers,
     tipo::{Type, TypeInfo},
@@ -465,12 +465,7 @@ where
     }
 
     pub fn export(&self, module: &str, name: &str) -> Result<String, Error> {
-        let mut generator = self.checked_modules.new_generator(
-            &self.functions,
-            &self.data_types,
-            &self.module_types,
-            Tracing::silent(),
-        );
+        let mut generator = self.new_generator(Tracing::silent());
 
         self.checked_modules
             .get(module)
@@ -495,7 +490,7 @@ where
                 })
             })
             .map(|body| {
-                let program = generator.generate_test(&body, &name.to_string());
+                let program = generator.generate_raw(&body, &[], module);
 
                 program.to_pretty()
             })
diff --git a/crates/aiken/src/cmd/export.rs b/crates/aiken/src/cmd/export.rs
index 7eec0c38..b97184eb 100644
--- a/crates/aiken/src/cmd/export.rs
+++ b/crates/aiken/src/cmd/export.rs
@@ -23,7 +23,7 @@ pub fn exec(
         name,
     }: Args,
 ) -> miette::Result<()> {
-    with_project(directory.as_deref(), false, |p| {
+    with_project(directory.as_deref(), 0, false, |p| {
         p.compile(Options::default())?;
 
         let raw_uplc = p.export(&module, &name)?;