harn_parser/builtin_signatures/
types.rs1use crate::ast::{ShapeField, TypeExpr};
11
12pub use harn_builtin_meta::{
13 BuiltinMetadata, BuiltinSignature, Param, ShapeFieldDescriptor, Ty, TY_ANY, TY_BOOL, TY_BYTES,
14 TY_BYTES_OR_NIL, TY_CLOSURE, TY_DECIMAL, TY_DICT, TY_DICT_OR_NIL, TY_DURATION, TY_FLOAT,
15 TY_INT, TY_INT_OR_NIL, TY_LIST, TY_NEVER, TY_NIL, TY_NUMBER, TY_STRING, TY_STRING_OR_NIL,
16};
17
18pub fn ty_to_type_expr(ty: &Ty) -> TypeExpr {
22 match ty {
23 Ty::Named(name) => TypeExpr::Named((*name).into()),
24 Ty::Generic(name) => TypeExpr::Named((*name).into()),
25 Ty::Any => TypeExpr::Named("any".into()),
26 Ty::Optional(inner) => {
27 TypeExpr::Union(vec![ty_to_type_expr(inner), TypeExpr::Named("nil".into())])
28 }
29 Ty::Apply(name, args) => TypeExpr::Applied {
30 name: (*name).into(),
31 args: args.iter().map(ty_to_type_expr).collect(),
32 },
33 Ty::Union(members) => TypeExpr::Union(members.iter().map(ty_to_type_expr).collect()),
34 Ty::Fn(params, return_type) => TypeExpr::FnType {
35 params: params.iter().map(ty_to_type_expr).collect(),
36 return_type: Box::new(ty_to_type_expr(return_type)),
37 },
38 Ty::Shape(fields) => TypeExpr::Shape(
39 fields
40 .iter()
41 .map(|f| ShapeField::synthetic(f.name, ty_to_type_expr(&f.ty), f.optional))
42 .collect(),
43 ),
44 Ty::SchemaOf(name) => TypeExpr::Applied {
45 name: "Schema".into(),
46 args: vec![TypeExpr::Named((*name).into())],
47 },
48 Ty::Never => TypeExpr::Never,
49 Ty::LitInt(v) => TypeExpr::LitInt(*v),
50 Ty::LitString(s) => TypeExpr::LitString((*s).into()),
51 }
52}
53
54pub trait TyExt {
58 fn to_type_expr(&self) -> TypeExpr;
60}
61
62impl TyExt for Ty {
63 fn to_type_expr(&self) -> TypeExpr {
64 ty_to_type_expr(self)
65 }
66}
67
68pub trait BuiltinSignatureExt {
69 fn param_type_exprs(&self) -> Vec<TypeExpr>;
72 fn return_type_expr(&self) -> TypeExpr;
74}
75
76impl BuiltinSignatureExt for BuiltinSignature {
77 fn param_type_exprs(&self) -> Vec<TypeExpr> {
78 self.params.iter().map(|p| ty_to_type_expr(&p.ty)).collect()
79 }
80
81 fn return_type_expr(&self) -> TypeExpr {
82 ty_to_type_expr(&self.returns)
83 }
84}