Update: extract get key type function

sunrabbit123 · Aug 13, 2023 · 813d328 · 813d328
1 parent e6e4c63
commit 813d328
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 79 deletions.
diff --git a/crates/stc_ts_file_analyzer/src/analyzer/expr/bin.rs b/crates/stc_ts_file_analyzer/src/analyzer/expr/bin.rs
@@ -2091,32 +2091,7 @@ impl Analyzer<'_, '_> {
                     ty => {
                         if let Err(err) = self.assign_with_opts(
                             &mut Default::default(),
-                            &Type::Union(Union {
-                                span,
-                                types: vec![
-                                    Type::Keyword(KeywordType {
-                                        span,
-                                        kind: TsKeywordTypeKind::TsStringKeyword,
-                                        metadata: Default::default(),
-                                        tracker: Default::default(),
-                                    }),
-                                    Type::Keyword(KeywordType {
-                                        span,
-                                        kind: TsKeywordTypeKind::TsNumberKeyword,
-                                        metadata: Default::default(),
-                                        tracker: Default::default(),
-                                    }),
-                                    Type::Keyword(KeywordType {
-                                        span,
-                                        kind: TsKeywordTypeKind::TsSymbolKeyword,
-                                        metadata: Default::default(),
-                                        tracker: Default::default(),
-                                    }),
-                                ],
-                                metadata: Default::default(),
-                                tracker: Default::default(),
-                            })
-                            .freezed(),
+                            &Type::get_any_key_type(span),
                             ty,
                             AssignOpts {
                                 span: ls,

diff --git a/crates/stc_ts_file_analyzer/src/analyzer/types/keyof.rs b/crates/stc_ts_file_analyzer/src/analyzer/types/keyof.rs
@@ -89,32 +89,7 @@ impl Analyzer<'_, '_> {
                         .context("tried applying `keyof` to a literal by delegating to keyword type handler")
                 }
                 Type::Keyword(KeywordType { kind, .. }) => match kind {
-                    TsKeywordTypeKind::TsAnyKeyword => {
-                        let string = Type::Keyword(KeywordType {
-                            span,
-                            kind: TsKeywordTypeKind::TsStringKeyword,
-                            metadata: Default::default(),
-                            tracker: Default::default(),
-                        });
-                        let number = Type::Keyword(KeywordType {
-                            span,
-                            kind: TsKeywordTypeKind::TsNumberKeyword,
-                            metadata: Default::default(),
-                            tracker: Default::default(),
-                        });
-                        let symbol = Type::Keyword(KeywordType {
-                            span,
-                            kind: TsKeywordTypeKind::TsSymbolKeyword,
-                            metadata: Default::default(),
-                            tracker: Default::default(),
-                        });
-                        return Ok(Type::Union(Union {
-                            span,
-                            types: vec![string, number, symbol],
-                            metadata: Default::default(),
-                            tracker: Default::default(),
-                        }));
-                    }
+                    TsKeywordTypeKind::TsAnyKeyword | TsKeywordTypeKind::TsNeverKeyword => return Ok(Type::get_any_key_type(span)),
                     TsKeywordTypeKind::TsVoidKeyword
                     | TsKeywordTypeKind::TsUndefinedKeyword
                     | TsKeywordTypeKind::TsNullKeyword
@@ -156,33 +131,6 @@ impl Analyzer<'_, '_> {
 
                     TsKeywordTypeKind::TsBigIntKeyword => {}
                     TsKeywordTypeKind::TsSymbolKeyword => {}
-                    TsKeywordTypeKind::TsNeverKeyword => {
-                        return Ok(Type::Union(Union {
-                            span,
-                            types: vec![
-                                Type::Keyword(KeywordType {
-                                    span,
-                                    kind: TsKeywordTypeKind::TsStringKeyword,
-                                    metadata: Default::default(),
-                                    tracker: Default::default(),
-                                }),
-                                Type::Keyword(KeywordType {
-                                    span,
-                                    kind: TsKeywordTypeKind::TsNumberKeyword,
-                                    metadata: Default::default(),
-                                    tracker: Default::default(),
-                                }),
-                                Type::Keyword(KeywordType {
-                                    span,
-                                    kind: TsKeywordTypeKind::TsSymbolKeyword,
-                                    metadata: Default::default(),
-                                    tracker: Default::default(),
-                                }),
-                            ],
-                            metadata: Default::default(),
-                            tracker: Default::default(),
-                        }))
-                    }
                     TsKeywordTypeKind::TsIntrinsicKeyword => {}
                 },
 

diff --git a/crates/stc_ts_types/src/lib.rs b/crates/stc_ts_types/src/lib.rs
@@ -1900,6 +1900,35 @@ impl Type {
             _ => None,
         }
     }
+
+    pub fn get_any_key_type(span: Span) -> Type {
+        let string = Type::Keyword(KeywordType {
+            span,
+            kind: TsKeywordTypeKind::TsStringKeyword,
+            metadata: Default::default(),
+            tracker: Default::default(),
+        });
+        let number = Type::Keyword(KeywordType {
+            span,
+            kind: TsKeywordTypeKind::TsNumberKeyword,
+            metadata: Default::default(),
+            tracker: Default::default(),
+        });
+        let symbol = Type::Keyword(KeywordType {
+            span,
+            kind: TsKeywordTypeKind::TsSymbolKeyword,
+            metadata: Default::default(),
+            tracker: Default::default(),
+        });
+
+        Type::Union(Union {
+            span,
+            types: vec![string, number, symbol],
+            metadata: Default::default(),
+            tracker: Default::default(),
+        })
+        .freezed()
+    }
 }
 
 impl Type {