fix @as

Author: CppCXY

crates/emmylua_code_analysis/src/compilation/analyzer/doc/type_ref_tags.rs

@@ -1,5 +1,5 @@
 use emmylua_parser::{
-    LuaAst, LuaAstNode, LuaAstToken, LuaDocDescriptionOwner, LuaDocTagAs, LuaDocTagCast,
+    LuaAst, LuaAstNode, LuaAstToken, LuaBlock, LuaDocDescriptionOwner, LuaDocTagAs, LuaDocTagCast,
     LuaDocTagModule, LuaDocTagOther, LuaDocTagOverload, LuaDocTagParam, LuaDocTagReturn,
     LuaDocTagSee, LuaDocTagType, LuaExpr, LuaLocalName, LuaTokenKind, LuaVarExpr,
 };
@@ -278,7 +278,16 @@ pub fn analyze_as(analyzer: &mut DocAnalyzer, tag: LuaDocTagAs) -> Option<()> {
         left_token = left_token.prev_token()?;
     }
 
-    let expr = LuaExpr::cast(left_token.parent()?)?;
+    let mut ast_node = left_token.parent()?;
+    loop {
+        if LuaExpr::can_cast(ast_node.kind().into()) {
+            break;
+        } else if LuaBlock::can_cast(ast_node.kind().into()) {
+            return None;
+        }
+        ast_node = ast_node.parent()?;
+    }
+    let expr = LuaExpr::cast(ast_node)?;
 
     let file_id = analyzer.file_id;
     let in_filed_syntax_id = InFiled::new(file_id, expr.get_syntax_id());

crates/emmylua_code_analysis/src/diagnostic/test/return_type_mismatch_test.rs

@@ -205,4 +205,24 @@ mod tests {
             "#
         ));
     }
+
+    #[test]
+    fn test_as_return_type() {
+        let mut ws = VirtualWorkspace::new();
+
+        assert!(ws.check_code_for(
+            DiagnosticCode::ReturnTypeMismatch,
+            r#"
+            local function dd()
+                return "11231"
+            end
+            
+            ---@return integer
+            local function f()
+            
+                return dd() ---@as integer
+            end
+        "#
+        ));
+    }
 }