tsz-solver 0.1.8

TypeScript type solver for the tsz compiler
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75

use crate::{TypeDatabase, TypeId, type_queries};
use tsz_scanner::SyntaxKind;

pub const fn is_compound_assignment_operator(operator_token: u16) -> bool {
    matches!(
        operator_token,
        k if k == SyntaxKind::PlusEqualsToken as u16
            || k == SyntaxKind::MinusEqualsToken as u16
            || k == SyntaxKind::AsteriskEqualsToken as u16
            || k == SyntaxKind::AsteriskAsteriskEqualsToken as u16
            || k == SyntaxKind::SlashEqualsToken as u16
            || k == SyntaxKind::PercentEqualsToken as u16
            || k == SyntaxKind::LessThanLessThanEqualsToken as u16
            || k == SyntaxKind::GreaterThanGreaterThanEqualsToken as u16
            || k == SyntaxKind::GreaterThanGreaterThanGreaterThanEqualsToken as u16
            || k == SyntaxKind::AmpersandEqualsToken as u16
            || k == SyntaxKind::BarEqualsToken as u16
            || k == SyntaxKind::CaretEqualsToken as u16
            || k == SyntaxKind::AmpersandAmpersandEqualsToken as u16
            || k == SyntaxKind::BarBarEqualsToken as u16
            || k == SyntaxKind::QuestionQuestionEqualsToken as u16
    )
}

pub const fn map_compound_assignment_to_binary(operator_token: u16) -> Option<&'static str> {
    match operator_token {
        k if k == SyntaxKind::PlusEqualsToken as u16 => Some("+"),
        k if k == SyntaxKind::MinusEqualsToken as u16 => Some("-"),
        k if k == SyntaxKind::AsteriskEqualsToken as u16 => Some("*"),
        k if k == SyntaxKind::AsteriskAsteriskEqualsToken as u16 => Some("**"),
        k if k == SyntaxKind::SlashEqualsToken as u16 => Some("/"),
        k if k == SyntaxKind::PercentEqualsToken as u16 => Some("%"),
        k if k == SyntaxKind::LessThanLessThanEqualsToken as u16 => Some("<<"),
        k if k == SyntaxKind::GreaterThanGreaterThanEqualsToken as u16 => Some(">>"),
        k if k == SyntaxKind::GreaterThanGreaterThanGreaterThanEqualsToken as u16 => Some(">>>"),
        k if k == SyntaxKind::AmpersandEqualsToken as u16 => Some("&"),
        k if k == SyntaxKind::BarEqualsToken as u16 => Some("|"),
        k if k == SyntaxKind::CaretEqualsToken as u16 => Some("^"),
        k if k == SyntaxKind::AmpersandAmpersandEqualsToken as u16 => Some("&&"),
        k if k == SyntaxKind::BarBarEqualsToken as u16 => Some("||"),
        k if k == SyntaxKind::QuestionQuestionEqualsToken as u16 => Some("??"),
        _ => None,
    }
}

pub fn fallback_compound_assignment_result(
    db: &dyn TypeDatabase,
    operator_token: u16,
    rhs_literal_type: Option<TypeId>,
) -> Option<TypeId> {
    match operator_token {
        k if k == SyntaxKind::MinusEqualsToken as u16
            || k == SyntaxKind::AsteriskEqualsToken as u16
            || k == SyntaxKind::AsteriskAsteriskEqualsToken as u16
            || k == SyntaxKind::SlashEqualsToken as u16
            || k == SyntaxKind::PercentEqualsToken as u16
            || k == SyntaxKind::LessThanLessThanEqualsToken as u16
            || k == SyntaxKind::GreaterThanGreaterThanEqualsToken as u16
            || k == SyntaxKind::GreaterThanGreaterThanGreaterThanEqualsToken as u16
            || k == SyntaxKind::AmpersandEqualsToken as u16
            || k == SyntaxKind::BarEqualsToken as u16
            || k == SyntaxKind::CaretEqualsToken as u16 =>
        {
            Some(TypeId::NUMBER)
        }
        k if k == SyntaxKind::PlusEqualsToken as u16 => rhs_literal_type.and_then(|literal| {
            if literal == TypeId::NUMBER || type_queries::is_number_literal(db, literal) {
                Some(TypeId::NUMBER)
            } else {
                None
            }
        }),
        _ => None,
    }
}