feat(validator): move return type and arity check to infer

crates/aiken-project/src/blueprint/error.rs

@@ -1,44 +1,11 @@
 use super::schema;
-use crate::module::CheckedModule;
-use aiken_lang::{
-    ast::{Span, TypedFunction},
-    tipo::Type,
-};
+use aiken_lang::ast::Span;
 use miette::{Diagnostic, NamedSource};
 use owo_colors::OwoColorize;
-use std::{fmt::Debug, sync::Arc};
+use std::fmt::Debug;
 
 #[derive(Debug, thiserror::Error, Diagnostic)]
 pub enum Error {
-    #[error("A validator must return {}", "Bool".bright_blue().bold())]
-    #[diagnostic(code("aiken::blueprint::invalid::return_type"))]
-    #[diagnostic(help(r#"While analyzing the return type of your validator, I found it to be:
-
-╰─▶ {signature}
-
-...but I expected this to be a {type_Bool}. If I am inferring the wrong type, you may want to add a type annotation to the function."#
-        , type_Bool = "Bool".bright_blue().bold()
-        , signature = return_type.to_pretty(0).red()
-    ))]
-    ValidatorMustReturnBool {
-        #[label("invalid return type")]
-        location: Span,
-        #[source_code]
-        source_code: NamedSource,
-        return_type: Arc<Type>,
-    },
-
-    #[error("A {} validator requires at least {} arguments.", name.purple().bold(), at_least.to_string().purple().bold())]
-    #[diagnostic(code("aiken::blueprint::invalid::arity"))]
-    WrongValidatorArity {
-        name: String,
-        at_least: u8,
-        #[label("not enough arguments")]
-        location: Span,
-        #[source_code]
-        source_code: NamedSource,
-    },
-
     #[error("{}", error)]
     #[diagnostic(help("{}", error.help()))]
     #[diagnostic(code("aiken::blueprint::interface"))]
@@ -64,38 +31,3 @@ pub enum Error {
     #[diagnostic(help("I can only compute addresses of validators that are fully applied. For example, a {keyword_spend} validator must have exactly 3 arguments: a datum, a redeemer and a context. If it has more, they need to be provided beforehand and applied directly in the validator. Applying parameters change the validator's compiled code, and thus the address.\n\nThis is why I need you to apply parmeters first.", keyword_spend = "spend".purple()))]
     ParameterizedValidator { n: usize },
 }
-
-pub fn assert_return_bool(module: &CheckedModule, def: &TypedFunction) -> Result<(), Error> {
-    if !def.return_type.is_bool() {
-        Err(Error::ValidatorMustReturnBool {
-            return_type: def.return_type.clone(),
-            location: def.location,
-            source_code: NamedSource::new(
-                module.input_path.display().to_string(),
-                module.code.clone(),
-            ),
-        })
-    } else {
-        Ok(())
-    }
-}
-
-pub fn assert_min_arity(
-    module: &CheckedModule,
-    def: &TypedFunction,
-    at_least: u8,
-) -> Result<(), Error> {
-    if def.arguments.len() < at_least as usize {
-        Err(Error::WrongValidatorArity {
-            name: def.name.clone(),
-            at_least,
-            location: def.location,
-            source_code: NamedSource::new(
-                module.input_path.display().to_string(),
-                module.code.clone(),
-            ),
-        })
-    } else {
-        Ok(())
-    }
-}