language-servers/tinymist
1
0
Fork 0
mirror of https://github.com/Myriad-Dreamin/tinymist.git synced 2025-07-24 21:23:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

feat: check union type annotation (#842)

Browse source
This commit is contained in:
Myriad-Dreamin 2024-11-18 11:44:28 +08:00 • committed by GitHub
parent a2e7e4c88c
commit c551950a85
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 28 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

29
crates/tinymist-query/src/syntax/docs.rs
View file

@ -13,7 +13,10 @@ use crate::{
docs::{convert_docs, identify_pat_docs, identify_tidy_module_docs, UntypedDefDocs, VarDocsT},
prelude::*,
syntax::{Decl, DefKind},
ty::{BuiltinTy, Interned, PackageId, SigTy, StrRef, Ty, TypeBounds, TypeVar, TypeVarBounds},
ty::{
BuiltinTy, InsTy, Interned, PackageId, SigTy, StrRef, Ty, TypeBounds, TypeVar,
TypeVarBounds,
},
};
use super::DeclExpr;
@ -298,6 +301,30 @@ impl<'a> DocsChecker<'a> {
log::debug!("check doc type expr: {s:?}");
match s {
ast::Expr::Ident(i) => self.check_type_ident(m, i.get().as_str()),
ast::Expr::None(_)
| ast::Expr::Auto(_)
| ast::Expr::Bool(..)
| ast::Expr::Int(..)
| ast::Expr::Float(..)
| ast::Expr::Numeric(..)
| ast::Expr::Str(..) => SharedContext::const_eval(s).map(|v| Ty::Value(InsTy::new(v))),
ast::Expr::Binary(b) => {
let mut components = Vec::with_capacity(2);
components.push(self.check_type_expr(m, b.lhs())?);
let mut expr = b.rhs();
while let ast::Expr::Binary(b) = expr {
if b.op() != ast::BinOp::Or {
break;
}
components.push(self.check_type_expr(m, b.lhs())?);
expr = b.rhs();
}
components.push(self.check_type_expr(m, expr)?);
Some(Ty::from_types(components.into_iter()))
}
ast::Expr::FuncCall(c) => match c.callee() {
ast::Expr::Ident(i) => {
let name = i.get().as_str();