fix: error on statement after return

While it would have been better to specify the expected keyword to ends that particular block like actual Lua does, I do not see that being possible in the given structure.

fix #1163

Author: PennyJim

crates/emmylua_parser/locales/app.yml

@@ -255,6 +255,8 @@ expected 'end' to close while statement:
   en: expected 'end' to close while statement
   zh_CN: 期望 'end' 来关闭while语句
   zh_HK: 期望 'end' 來關閉while語句
+expected end of block after return:
+  en: expected end of block after return
 expected 'function', variable name, or attribute after 'local':
   en: expected 'function', variable name, or attribute after 'local'
   zh_CN: 期望在 'local' 后有 'function'、变量名或属性

crates/emmylua_parser/src/grammar/doc/test.rs

@@ -3724,4 +3724,26 @@ Syntax(Chunk)@0..47
         assert!(errors[0].range.is_empty());
         assert_eq!(u32::from(errors[0].range.start()) as usize, code.len());
     }
+
+    #[test]
+    fn test_statement_after_return() {
+        let code = "return 1, 2 x = 1";
+        let tree = LuaParser::parse(code, ParserConfig::default());
+        let errors = tree.get_errors();
+
+        assert_eq!(errors.len(), 1);
+        assert_eq!(errors[0].kind, LuaParseErrorKind::SyntaxError);
+        assert_eq!(errors[0].message, "expected end of block after return")
+    }
+
+    #[test]
+    fn test_statement_after_blank_return() {
+        let code = "return; x = 1";
+        let tree = LuaParser::parse(code, ParserConfig::default());
+        let errors = tree.get_errors();
+
+        assert_eq!(errors.len(), 1);
+        assert_eq!(errors[0].kind, LuaParseErrorKind::SyntaxError);
+        assert_eq!(errors[0].message, "expected end of block after return")
+    }
 }

crates/emmylua_parser/src/grammar/lua/stat.rs

@@ -752,6 +752,7 @@ fn parse_attrib(p: &mut LuaParser) -> ParseResult {
 
 fn parse_return(p: &mut LuaParser) -> ParseResult {
     let m = p.mark(LuaSyntaxKind::ReturnStat);
+    let return_start_range = p.current_token_range();
     p.bump();
     if !block_follow(p)
         && p.current_token() != LuaTokenKind::TkSemicolon
@@ -761,6 +762,14 @@ fn parse_return(p: &mut LuaParser) -> ParseResult {
     }
 
     if_token_bump(p, LuaTokenKind::TkSemicolon);
+
+    if !block_follow(p) {
+        p.push_error(LuaParseError::syntax_error_from(
+            &t!("expected end of block after return"),
+            return_start_range,
+        ));
+    };
+
     Ok(m.complete(p))
 }