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

feat: add identity, always, & flip

This commit is contained in:
rvcas 2023-01-14 23:21:42 -05:00 committed by Lucas
parent 9daf86e984
commit 00f2150eed
11 changed files with 314 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

280
crates/aiken-lang/src/builtins.rs
View File

@ -5,7 +5,7 @@ use strum::IntoEnumIterator;
use uplc::builtins::DefaultFunction;
use crate::{
ast::{Arg, ArgName, Function, ModuleKind, Span, TypedFunction, UnOp},
ast::{Arg, ArgName, CallArg, Function, ModuleKind, Span, TypedFunction, UnOp},
builder::FunctionAccessKey,
expr::TypedExpr,
tipo::{
@ -121,6 +121,7 @@ pub fn prelude(id_gen: &IdGenerator) -> TypeInfo {
},
);
// not
prelude.values.insert(
"not".to_string(),
ValueConstructor::public(
@ -136,6 +137,67 @@ pub fn prelude(id_gen: &IdGenerator) -> TypeInfo {
),
);
// identity
let identity_var = generic_var(id_gen.next());
prelude.values.insert(
"identity".to_string(),
ValueConstructor::public(
function(vec![identity_var.clone()], identity_var),
ValueConstructorVariant::ModuleFn {
name: "identity".to_string(),
field_map: None,
module: "".to_string(),
arity: 1,
location: Span::empty(),
builtin: None,
},
),
);
// always
let always_a_var = generic_var(id_gen.next());
let always_b_var = generic_var(id_gen.next());
prelude.values.insert(
"always".to_string(),
ValueConstructor::public(
function(vec![always_a_var.clone(), always_b_var], always_a_var),
ValueConstructorVariant::ModuleFn {
name: "always".to_string(),
field_map: None,
module: "".to_string(),
arity: 2,
location: Span::empty(),
builtin: None,
},
),
);
// flip
let flip_a_var = generic_var(id_gen.next());
let flip_b_var = generic_var(id_gen.next());
let flip_c_var = generic_var(id_gen.next());
let input_type = function(
vec![flip_a_var.clone(), flip_b_var.clone()],
flip_c_var.clone(),
);
let return_type = function(vec![flip_b_var, flip_a_var], flip_c_var);
prelude.values.insert(
"flip".to_string(),
ValueConstructor::public(
function(vec![input_type], return_type),
ValueConstructorVariant::ModuleFn {
name: "flip".to_string(),
field_map: None,
module: "".to_string(),
arity: 1,
location: Span::empty(),
builtin: None,
},
),
);
// List(a)
let list_parameter = generic_var(id_gen.next());
prelude.types.insert(
@ -471,7 +533,7 @@ pub fn from_default_function(
})
}
pub fn prelude_functions() -> HashMap<FunctionAccessKey, TypedFunction> {
pub fn prelude_functions(id_gen: &IdGenerator) -> HashMap<FunctionAccessKey, TypedFunction> {
let mut functions = HashMap::new();
// /// Negate the argument. Useful for map/fold and pipelines.
@ -521,6 +583,220 @@ pub fn prelude_functions() -> HashMap<FunctionAccessKey, TypedFunction> {
},
);
// /// A function that returns its argument. Handy as a default behavior sometimes.
// pub fn identity(a: a) -> a {
// a
// }
let a_var = generic_var(id_gen.next());
functions.insert(
FunctionAccessKey {
module_name: "".to_string(),
function_name: "identity".to_string(),
variant_name: "".to_string(),
},
Function {
arguments: vec![Arg {
arg_name: ArgName::Named {
name: "a".to_string(),
label: "a".to_string(),
location: Span::empty(),
},
location: Span::empty(),
annotation: None,
tipo: a_var.clone(),
}],
body: TypedExpr::Var {
location: Span::empty(),
constructor: ValueConstructor {
public: true,
tipo: a_var.clone(),
variant: ValueConstructorVariant::LocalVariable {
location: Span::empty(),
},
},
name: "a".to_string(),
},
doc: None,
location: Span::empty(),
name: "identity".to_string(),
public: true,
return_annotation: None,
return_type: a_var,
end_position: 0,
},
);
// /// A function that always return its first argument. Handy in folds and maps.
// pub fn always(a: a, b _b: b) -> a {
// a
// }
let a_var = generic_var(id_gen.next());
let b_var = generic_var(id_gen.next());
functions.insert(
FunctionAccessKey {
module_name: "".to_string(),
function_name: "always".to_string(),
variant_name: "".to_string(),
},
Function {
arguments: vec![
Arg {
arg_name: ArgName::Named {
name: "a".to_string(),
label: "a".to_string(),
location: Span::empty(),
},
location: Span::empty(),
annotation: None,
tipo: a_var.clone(),
},
Arg {
arg_name: ArgName::Discarded {
name: "_b".to_string(),
label: "_b".to_string(),
location: Span::empty(),
},
location: Span::empty(),
annotation: None,
tipo: b_var,
},
],
body: TypedExpr::Var {
location: Span::empty(),
constructor: ValueConstructor {
public: true,
tipo: a_var.clone(),
variant: ValueConstructorVariant::LocalVariable {
location: Span::empty(),
},
},
name: "a".to_string(),
},
doc: None,
location: Span::empty(),
name: "always".to_string(),
public: true,
return_annotation: None,
return_type: a_var,
end_position: 0,
},
);
// /// A function that flips the arguments of a function.
// pub fn flip(f: fn(a, b) -> c) -> fn(b, a) -> c {
// fn(b, a) { f(a, b) }
// }
let a_var = generic_var(id_gen.next());
let b_var = generic_var(id_gen.next());
let c_var = generic_var(id_gen.next());
let input_type = function(vec![a_var.clone(), b_var.clone()], c_var.clone());
let return_type = function(vec![b_var.clone(), a_var.clone()], c_var.clone());
functions.insert(
FunctionAccessKey {
module_name: "".to_string(),
function_name: "flip".to_string(),
variant_name: "".to_string(),
},
Function {
arguments: vec![Arg {
arg_name: ArgName::Named {
name: "f".to_string(),
label: "f".to_string(),
location: Span::empty(),
},
location: Span::empty(),
annotation: None,
tipo: input_type.clone(),
}],
body: TypedExpr::Fn {
location: Span::empty(),
tipo: return_type.clone(),
is_capture: false,
args: vec![
Arg {
arg_name: ArgName::Named {
name: "b".to_string(),
label: "b".to_string(),
location: Span::empty(),
},
location: Span::empty(),
annotation: None,
tipo: b_var.clone(),
},
Arg {
arg_name: ArgName::Named {
name: "a".to_string(),
label: "a".to_string(),
location: Span::empty(),
},
location: Span::empty(),
annotation: None,
tipo: a_var.clone(),
},
],
body: Box::new(TypedExpr::Call {
location: Span::empty(),
tipo: c_var,
fun: Box::new(TypedExpr::Var {
location: Span::empty(),
constructor: ValueConstructor {
public: true,
tipo: input_type,
variant: ValueConstructorVariant::LocalVariable {
location: Span::empty(),
},
},
name: "f".to_string(),
}),
args: vec![
CallArg {
label: None,
location: Span::empty(),
value: TypedExpr::Var {
location: Span::empty(),
constructor: ValueConstructor {
public: true,
tipo: a_var,
variant: ValueConstructorVariant::LocalVariable {
location: Span::empty(),
},
},
name: "a".to_string(),
},
},
CallArg {
label: None,
location: Span::empty(),
value: TypedExpr::Var {
location: Span::empty(),
constructor: ValueConstructor {
public: true,
tipo: b_var,
variant: ValueConstructorVariant::LocalVariable {
location: Span::empty(),
},
},
name: "b".to_string(),
},
},
],
}),
return_annotation: None,
},
doc: None,
location: Span::empty(),
name: "flip".to_string(),
public: true,
return_annotation: None,
return_type,
end_position: 0,
},
);
functions
}

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

@ -490,7 +490,7 @@ where
let mut type_aliases = HashMap::new();
let mut data_types = HashMap::new();
let prelude_functions = builtins::prelude_functions();
let prelude_functions = builtins::prelude_functions(&self.id_gen);
for (access_key, func) in prelude_functions.iter() {
functions.insert(access_key.clone(), func);
}
@ -581,7 +581,7 @@ where
let mut type_aliases = HashMap::new();
let mut data_types = HashMap::new();
let prelude_functions = builtins::prelude_functions();
let prelude_functions = builtins::prelude_functions(&self.id_gen);
for (access_key, func) in prelude_functions.iter() {
functions.insert(access_key.clone(), func);
}

5
examples/acceptance_tests/041/aiken.lock Normal file
View File

@ -0,0 +1,5 @@
# This file was generated by Aiken
# You typically do not need to edit this file
requirements = []
packages = []

2
examples/acceptance_tests/041/aiken.toml Normal file
View File

@ -0,0 +1,2 @@
name = "aiken-lang/acceptance_test_041"
version = "0.0.0"

3
examples/acceptance_tests/041/lib/tests.ak Normal file
View File

@ -0,0 +1,3 @@
test identity_1() {
identity(1) == 1
}

5
examples/acceptance_tests/042/aiken.lock Normal file
View File

@ -0,0 +1,5 @@
# This file was generated by Aiken
# You typically do not need to edit this file
requirements = []
packages = []

2
examples/acceptance_tests/042/aiken.toml Normal file
View File

@ -0,0 +1,2 @@
name = "aiken-lang/acceptance_test_042"
version = "0.0.0"

3
examples/acceptance_tests/042/lib/tests.ak Normal file
View File

@ -0,0 +1,3 @@
test always_1() {
always(1, 2) == 1
}

5
examples/acceptance_tests/043/aiken.lock Normal file
View File

@ -0,0 +1,5 @@
# This file was generated by Aiken
# You typically do not need to edit this file
requirements = []
packages = []

2
examples/acceptance_tests/043/aiken.toml Normal file
View File

@ -0,0 +1,2 @@
name = "aiken-lang/acceptance_test_043"
version = "0.0.0"

7
examples/acceptance_tests/043/lib/tests.ak Normal file
View File

@ -0,0 +1,7 @@
fn substract(x, y) {
x - y
}
test flip_1() {
flip(substract)(3, 2) == -1
}