fix: await type check diag && better error tolerate

Author: Chronostasys

src/ast/node/function.rs

@@ -329,7 +329,7 @@ impl Node for FuncCallNode {
             ctx.save_if_comment_doc_hover(id_range, Some(fnvalue.doc.clone()));
             handle_ret(ret, rettp, fnvalue.is_declare && skip == 0, ctx, builder)
         });
-        ctx.set_if_refs_tp(pltype, id_range);
+        // ctx.set_if_refs_tp(pltype, id_range);
         ctx.emit_comment_highlight(&self.comments[0]);
         res
     }

src/ast/node/operator.rs

@@ -88,7 +88,13 @@ impl Node for UnaryOpNode {
                             .map(|(_, v)| v.clone())
                             .unwrap_or(Arc::new(RefCell::new(PLType::Unknown)))
                     }
-                    _ => Arc::new(RefCell::new(PLType::Unknown)),
+                    _ => {
+                        return Err(self
+                            .range
+                            .new_err(ErrorCode::ONLY_TASK_CAN_BE_AWAIT)
+                            .add_label(exp_range, ctx.get_file(), None)
+                            .add_to_ctx(ctx));
+                    }
                 };
                 let awaited = builder.await_task(ctx, rv.get_value());
                 return Ok(NodeOutput::new_value(NodeValue::new(awaited, poll_ty)));

src/nomparser/error.rs

@@ -77,6 +77,23 @@ where
     alt((parser, except(ex, msg, code)))
 }
 
+pub fn eat_any_err_block_before<'a, E, F, R>(
+    parser: F,
+    parser2: F,
+    until: &'static str,
+    code: ErrorCode,
+    msg: &'static str,
+) -> impl FnMut(Span<'a>) -> IResult<Span<'a>, R, E>
+where
+    E: ParseError<Span<'a>> + FromExternalError<Span<'a>, std::fmt::Error>,
+    F: FnMut(Span<'a>) -> IResult<Span<'a>, R, E>,
+{
+    alt((
+        parser,
+        preceded(match_paired_until(until, code, msg), parser2),
+    ))
+}
+
 /// match all token with paired `(` or `{` or `[` until find given token out of the paired token
 ///
 /// will report an error

src/nomparser/implement.rs

@@ -1,4 +1,5 @@
 use crate::ast::{node::implement::ImplNode, tokens::TokenType};
+use error::eat_any_err_block_before;
 use nom::{
     combinator::{map_res, opt},
     multi::many0,
@@ -70,9 +71,21 @@ pub fn impl_def(input: Span) -> IResult<Span, Box<TopLevel>> {
             opt(pair(type_name, tag_token_word(TokenType::FOR))),
             type_name,
             del_newline_or_space!(tag_token_symbol(TokenType::LBRACE)),
-            many0(del_newline_or_space!(function_def)),
+            many0(eat_any_err_block_before(
+                function_def,
+                function_def,
+                "\n",
+                crate::ast::diag::ErrorCode::SYNTAX_ERROR_TOP_STATEMENT,
+                "failed to parse",
+            )),
             many0(comment),
-            del_newline_or_space!(tag_token_symbol(TokenType::RBRACE)),
+            eat_any_err_block_before(
+                del_newline_or_space!(tag_token_symbol(TokenType::RBRACE)),
+                del_newline_or_space!(tag_token_symbol(TokenType::RBRACE)),
+                "}",
+                crate::ast::diag::ErrorCode::SYNTAX_ERROR_TOP_STATEMENT,
+                "failed to parse",
+            ),
         )),
         |(_, generics, impl_trait, structure, (_, start), func_def, comment0, (_, end))| {
             res_box(Box::new(TopLevel::ImplDef(ImplNode {

test/main.pi

@@ -77,6 +77,7 @@ async fn main() Task<()> {
     println!("async 1");
     await delay::delay(2000 as u64);
     println!("async 1 end");
+
     await std_test::test_nested_async_closure();
     await std_test::test_nested_async_closure_in_normal_f();
     await std_test::test_nested_async_closure_in_normal_f2();

test/test/std_test.pi

@@ -33,7 +33,6 @@ impl Task<i64> for CustomTask {
         }
         return 100 as Option<i64>;
     }
-    
 }