:art: Improve return stmt

diff --git a/src/parse/check.c b/src/parse/check.c
index c141924c97ef00fbca249c98e3f979f053b0f818..75d7a03364890751750a7cc85bc1627582a55ebb 100644 (file)
--- a/src/parse/check.c
+++ b/src/parse/check.c
@@ -996,22 +996,18 @@ ANN static m_bool check_stmt_return(const Env env, const Stmt_Exp stmt) {
   if(!env->func)
     ERR_B(stmt_self(stmt)->pos, _("'return' statement found outside function definition"))
   DECL_OB(const Type, ret_type, = stmt->val ? check_exp(env, stmt->val) : t_void)
-  if(env->func->value_ref->type == t_lambda) {
-    if(env->func->def->base->ret_type &&
-     isa(ret_type, env->func->def->base->ret_type) < 0 &&
-     isa(env->func->def->base->ret_type, ret_type) < 0)
-          ERR_B(stmt_self(stmt)->pos, _("return types do not match for lambda expression"))
-    env->func->value_ref->type = ret_type;
+  if(!env->func->def->base->ret_type) {
+    assert(isa(env->func->value_ref->type, t_lambda) > 0);
+    env->func->def->base->ret_type = ret_type;
     return GW_OK;
   }
-  if(isa(ret_type, t_null) > 0 &&
-     isa(env->func->def->base->ret_type, t_object) > 0)
-    return GW_OK;
-  if(env->func->def->base->ret_type && isa(ret_type, env->func->def->base->ret_type) < 0)
+  const struct Implicit imp = { stmt->val, env->func->def->base->ret_type, stmt_self(stmt)->pos };
+  struct Op_Import opi = { .op=insert_symbol("@implicit"), .lhs=ret_type, .rhs=env->func->def->base->ret_type,
+                      .data=(m_uint)&imp, .pos=stmt_self(stmt)->pos };
+  const Type ret = op_check(env, &opi);
+  if(!ret)
     ERR_B(stmt_self(stmt)->pos, _("invalid return type '%s' -- expecting '%s'"),
           ret_type->name, env->func->def->base->ret_type->name)
-  else //! set default return type for lambdas
-    env->func->def->base->ret_type = ret_type;
   return GW_OK;
 }
 

diff --git a/src/parse/scan2.c b/src/parse/scan2.c
index fee643818f5e42e294c2b8cc360df2a3752ead21..1739e38b1d9846d801de6174803d6db66fb7f6a8 100644 (file)
--- a/src/parse/scan2.c
+++ b/src/parse/scan2.c
@@ -325,7 +325,7 @@ ANN static Func scan_new_func(const Env env, const Func_Def f, const m_str name)
 }
 
 ANN static Type func_type(const Env env, const Func func) {
-  const Type t = type_copy(env->gwion->mp, t_function);
+  const Type t = type_copy(env->gwion->mp, func->def->base->td ? t_function : t_lambda);
   t->name = func->name;
   t->e->owner = env->curr;
   if(GET_FLAG(func, member))