feat(semantic): add to indicate referenced by

Dunqing · Dunqing · commit 610a705e0eb5 · 2024-07-17T17:22:22.000+08:00
diff --git a/crates/oxc_linter/src/rules/typescript/consistent_type_imports.rs b/crates/oxc_linter/src/rules/typescript/consistent_type_imports.rs
@@ -10,7 +10,7 @@ use oxc_ast::{
 };
 use oxc_diagnostics::OxcDiagnostic;
 use oxc_macros::declare_oxc_lint;
-use oxc_semantic::SymbolId;
+use oxc_semantic::{Reference, SymbolId};
 use oxc_span::{CompactStr, GetSpan, Span};
 
 use crate::{
@@ -309,23 +309,7 @@ fn is_only_has_type_references(symbol_id: SymbolId, ctx: &LintContext) -> bool {
     if peekable_iter.peek().is_none() {
         return false;
     }
-    peekable_iter.all(|reference| {
-        if reference.is_type() {
-            return true;
-        } else if reference.is_read() {
-            for node in ctx.nodes().iter_parents(reference.node_id()).skip(1) {
-                return match node.kind() {
-                    // CASE 1:
-                    // `type T = typeof foo` will create a value reference because "foo" must be a value type
-                    // however this value reference is safe to use with type-only imports
-                    AstKind::TSTypeQuery(_) => true,
-                    AstKind::TSTypeName(_) | AstKind::TSQualifiedName(_) => continue,
-                    _ => false,
-                };
-            }
-        }
-        false
-    })
+    peekable_iter.all(Reference::is_type)
 }
 
 struct FixOptions<'a, 'b> {
diff --git a/crates/oxc_semantic/src/builder.rs b/crates/oxc_semantic/src/builder.rs
@@ -1715,15 +1715,17 @@ impl<'a> SemanticBuilder<'a> {
             AstKind::TSInterfaceHeritage(_) => {
                 self.current_reference_flag = ReferenceFlag::Type;
             }
+            AstKind::TSTypeQuery(_) => {
+                // type A = typeof a;
+                //          ^^^^^^^^
+                self.current_reference_flag = ReferenceFlag::Read | ReferenceFlag::TSTypeQuery;
+            }
             AstKind::TSTypeName(_) => {
                 match self.nodes.parent_kind(self.current_node_id) {
                     Some(
-                        // type A = typeof a;
-                        //          ^^^^^^^^
-                        AstKind::TSTypeQuery(_)
                         // import A = a;
                         //            ^
-                        | AstKind::TSModuleReference(_)
+                        AstKind::TSModuleReference(_),
                     ) => {
                         self.current_reference_flag = ReferenceFlag::Read;
                     }
@@ -1732,7 +1734,9 @@ impl<'a> SemanticBuilder<'a> {
                         //            ^^^ Keep the current reference flag
                     }
                     _ => {
-                        self.current_reference_flag = ReferenceFlag::Type;
+                        if !self.current_reference_flag.is_ts_type_query() {
+                            self.current_reference_flag = ReferenceFlag::Type;
+                        }
                     }
                 }
             }
@@ -1847,7 +1851,9 @@ impl<'a> SemanticBuilder<'a> {
                     self.current_reference_flag -= ReferenceFlag::Read;
                 }
             }
-            AstKind::MemberExpression(_) => self.current_reference_flag = ReferenceFlag::empty(),
+            AstKind::MemberExpression(_) | AstKind::TSTypeQuery(_) => {
+                self.current_reference_flag = ReferenceFlag::empty();
+            }
             AstKind::AssignmentTarget(_) => self.current_reference_flag -= ReferenceFlag::Write,
             _ => {}
         }
@@ -1873,10 +1879,10 @@ impl<'a> SemanticBuilder<'a> {
 
     /// Resolve reference flags for the current ast node.
     fn resolve_reference_usages(&self) -> ReferenceFlag {
-        if self.current_reference_flag.is_write() || self.current_reference_flag.is_type() {
-            self.current_reference_flag
-        } else {
+        if self.current_reference_flag.is_empty() {
             ReferenceFlag::Read
+        } else {
+            self.current_reference_flag
         }
     }
 
diff --git a/crates/oxc_semantic/src/reference.rs b/crates/oxc_semantic/src/reference.rs
@@ -80,6 +80,6 @@ impl Reference {
     }
 
     pub fn is_type(&self) -> bool {
-        self.flag.is_type()
+        self.flag.is_type() || self.flag.is_ts_type_query()
     }
 }
diff --git a/crates/oxc_semantic/tests/fixtures/typescript-eslint/class/declaration/type-reference.snap b/crates/oxc_semantic/tests/fixtures/typescript-eslint/class/declaration/type-reference.snap
@@ -44,7 +44,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/class/declarati
             "node_id": 8
           },
           {
-            "flag": "ReferenceFlag(Read)",
+            "flag": "ReferenceFlag(Read | TSTypeQuery)",
             "id": 1,
             "name": "A",
             "node_id": 13
diff --git a/crates/oxc_semantic/tests/fixtures/typescript-eslint/instantiation-expressions/type-arguments2.snap b/crates/oxc_semantic/tests/fixtures/typescript-eslint/instantiation-expressions/type-arguments2.snap
@@ -52,14 +52,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/instantiation-e
             "id": 4,
             "name": "T",
             "node": "TSTypeParameter",
-            "references": [
-              {
-                "flag": "ReferenceFlag(Type)",
-                "id": 0,
-                "name": "T",
-                "node_id": 31
-              }
-            ]
+            "references": []
           }
         ]
       }
@@ -75,7 +68,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/instantiation-e
         "node": "Function(makeBox)",
         "references": [
           {
-            "flag": "ReferenceFlag(Read)",
+            "flag": "ReferenceFlag(Read | TSTypeQuery)",
             "id": 0,
             "name": "makeBox",
             "node_id": 27
diff --git a/crates/oxc_semantic/tests/fixtures/typescript-eslint/type-declaration/function/function2.snap b/crates/oxc_semantic/tests/fixtures/typescript-eslint/type-declaration/function/function2.snap
@@ -24,7 +24,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/type-declaratio
         "node": "VariableDeclarator",
         "references": [
           {
-            "flag": "ReferenceFlag(Read)",
+            "flag": "ReferenceFlag(Read | TSTypeQuery)",
             "id": 0,
             "name": "arg",
             "node_id": 15
diff --git a/crates/oxc_semantic/tests/fixtures/typescript-eslint/type-declaration/index-access3.snap b/crates/oxc_semantic/tests/fixtures/typescript-eslint/type-declaration/index-access3.snap
@@ -45,7 +45,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/type-declaratio
         "node": "VariableDeclarator",
         "references": [
           {
-            "flag": "ReferenceFlag(Read)",
+            "flag": "ReferenceFlag(Read | TSTypeQuery)",
             "id": 0,
             "name": "k",
             "node_id": 21
diff --git a/crates/oxc_semantic/tests/fixtures/typescript-eslint/type-declaration/type-query-qualified.snap b/crates/oxc_semantic/tests/fixtures/typescript-eslint/type-declaration/type-query-qualified.snap
@@ -31,7 +31,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/type-declaratio
         "node": "VariableDeclarator",
         "references": [
           {
-            "flag": "ReferenceFlag(Read)",
+            "flag": "ReferenceFlag(Read | TSTypeQuery)",
             "id": 0,
             "name": "x",
             "node_id": 21
diff --git a/crates/oxc_semantic/tests/fixtures/typescript-eslint/type-declaration/type-query-with-parameters.snap b/crates/oxc_semantic/tests/fixtures/typescript-eslint/type-declaration/type-query-with-parameters.snap
@@ -52,14 +52,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/type-declaratio
             "id": 4,
             "name": "T",
             "node": "TSTypeParameter",
-            "references": [
-              {
-                "flag": "ReferenceFlag(Type)",
-                "id": 0,
-                "name": "T",
-                "node_id": 33
-              }
-            ]
+            "references": []
           }
         ]
       }
@@ -75,7 +68,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/type-declaratio
         "node": "Function(foo)",
         "references": [
           {
-            "flag": "ReferenceFlag(Read)",
+            "flag": "ReferenceFlag(Read | TSTypeQuery)",
             "id": 0,
             "name": "foo",
             "node_id": 29
diff --git a/crates/oxc_semantic/tests/fixtures/typescript-eslint/type-declaration/type-query.snap b/crates/oxc_semantic/tests/fixtures/typescript-eslint/type-declaration/type-query.snap
@@ -31,7 +31,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/type-declaratio
         "node": "VariableDeclarator",
         "references": [
           {
-            "flag": "ReferenceFlag(Read)",
+            "flag": "ReferenceFlag(Read | TSTypeQuery)",
             "id": 0,
             "name": "x",
             "node_id": 9
diff --git a/crates/oxc_syntax/src/reference.rs b/crates/oxc_syntax/src/reference.rs
@@ -45,6 +45,8 @@ bitflags! {
         const Write = 1 << 1;
         // Used in type definitions.
         const Type = 1 << 2;
+        // Used in `typeof xx`
+        const TSTypeQuery = 1 << 3;
         const Value = Self::Read.bits() | Self::Write.bits();
     }
 }
@@ -87,6 +89,11 @@ impl ReferenceFlag {
         self.contains(Self::Read | Self::Write)
     }
 
+    /// The identifier is used in a type referenced
+    pub fn is_ts_type_query(&self) -> bool {
+        self.contains(Self::TSTypeQuery)
+    }
+
     /// The identifier is used in a type definition.
     pub const fn is_type(&self) -> bool {
         self.contains(Self::Type)
diff --git a/tasks/coverage/transformer_typescript.snap b/tasks/coverage/transformer_typescript.snap
@@ -2,10 +2,8 @@ commit: d8086f14
 
 transformer_typescript Summary:
 AST Parsed     : 6456/6456 (100.00%)
-Positive Passed: 6450/6456 (99.91%)
+Positive Passed: 6452/6456 (99.94%)
 Mismatch: "compiler/constEnumNamespaceReferenceCausesNoImport2.ts"
 Mismatch: "compiler/elidedEmbeddedStatementsReplacedWithSemicolon.ts"
-Mismatch: "conformance/dynamicImport/importCallExpressionReturnPromiseOfAny.ts"
 Mismatch: "conformance/externalModules/typeOnly/exportDeclaration.ts"
-Mismatch: "conformance/externalModules/typeOnly/namespaceImportTypeQuery2.ts"
 Mismatch: "conformance/jsx/inline/inlineJsxAndJsxFragPragmaOverridesCompilerOptions.tsx"
diff --git a/tasks/transform_conformance/babel.snap.md b/tasks/transform_conformance/babel.snap.md
@@ -1,6 +1,6 @@
 commit: 12619ffe
 
-Passed: 473/927
+Passed: 474/927
 
 # All Passed:
 * babel-preset-react
@@ -445,14 +445,13 @@ Passed: 473/927
 * opts/optimizeConstEnums/input.ts
 * opts/rewriteImportExtensions/input.ts
 
-# babel-plugin-transform-typescript (129/151)
+# babel-plugin-transform-typescript (130/151)
 * class/accessor-allowDeclareFields-false/input.ts
 * class/accessor-allowDeclareFields-true/input.ts
 * enum/mix-references/input.ts
 * enum/ts5.0-const-foldable/input.ts
 * exports/declared-types/input.ts
 * exports/interface/input.ts
-* imports/elide-typeof/input.ts
 * imports/only-remove-type-imports/input.ts
 * imports/type-only-export-specifier-2/input.ts
 * imports/type-only-import-specifier-4/input.ts

Original file line number	Diff line number	Diff line change
`@@ -1715,15 +1715,17 @@ impl<'a> SemanticBuilder<'a> {`
`1715`	`1715`	`AstKind::TSInterfaceHeritage(_) => {`
`1716`	`1716`	`self.current_reference_flag = ReferenceFlag::Type;`
`1717`	`1717`	`}`
	`1718`	`+ AstKind::TSTypeQuery(_) => {`
	`1719`	`+ // type A = typeof a;`
	`1720`	`+ // ^^^^^^^^`
	`1721`	`+ self.current_reference_flag = ReferenceFlag::Read \| ReferenceFlag::TSTypeQuery;`
	`1722`	`+ }`
`1718`	`1723`	`AstKind::TSTypeName(_) => {`
`1719`	`1724`	`match self.nodes.parent_kind(self.current_node_id) {`
`1720`	`1725`	`Some(`
`1721`		`- // type A = typeof a;`
`1722`		`- // ^^^^^^^^`
`1723`		`- AstKind::TSTypeQuery(_)`
`1724`	`1726`	`// import A = a;`
`1725`	`1727`	`// ^`
`1726`		`- \| AstKind::TSModuleReference(_)`
	`1728`	`+ AstKind::TSModuleReference(_),`
`1727`	`1729`	`) => {`
`1728`	`1730`	`self.current_reference_flag = ReferenceFlag::Read;`
`1729`	`1731`	`}`
`@@ -1732,7 +1734,9 @@ impl<'a> SemanticBuilder<'a> {`
`1732`	`1734`	`// ^^^ Keep the current reference flag`
`1733`	`1735`	`}`
`1734`	`1736`	`_ => {`
`1735`		`- self.current_reference_flag = ReferenceFlag::Type;`
	`1737`	`+ if !self.current_reference_flag.is_ts_type_query() {`
	`1738`	`+ self.current_reference_flag = ReferenceFlag::Type;`
	`1739`	`+ }`
`1736`	`1740`	`}`
`1737`	`1741`	`}`
`1738`	`1742`	`}`
`@@ -1847,7 +1851,9 @@ impl<'a> SemanticBuilder<'a> {`
`1847`	`1851`	`self.current_reference_flag -= ReferenceFlag::Read;`
`1848`	`1852`	`}`
`1849`	`1853`	`}`
`1850`		`- AstKind::MemberExpression(_) => self.current_reference_flag = ReferenceFlag::empty(),`
	`1854`	`+ AstKind::MemberExpression(_) \| AstKind::TSTypeQuery(_) => {`
	`1855`	`+ self.current_reference_flag = ReferenceFlag::empty();`
	`1856`	`+ }`
`1851`	`1857`	`AstKind::AssignmentTarget(_) => self.current_reference_flag -= ReferenceFlag::Write,`
`1852`	`1858`	`_ => {}`
`1853`	`1859`	`}`
`@@ -1873,10 +1879,10 @@ impl<'a> SemanticBuilder<'a> {`
`1873`	`1879`
`1874`	`1880`	`/// Resolve reference flags for the current ast node.`
`1875`	`1881`	`fn resolve_reference_usages(&self) -> ReferenceFlag {`
`1876`		`- if self.current_reference_flag.is_write() \|\| self.current_reference_flag.is_type() {`
`1877`		`- self.current_reference_flag`
`1878`		`- } else {`
	`1882`	`+ if self.current_reference_flag.is_empty() {`
`1879`	`1883`	`ReferenceFlag::Read`
	`1884`	`+ } else {`
	`1885`	`+ self.current_reference_flag`
`1880`	`1886`	`}`
`1881`	`1887`	`}`
`1882`	`1888`
Original file line number	Diff line number	Diff line change
`@@ -80,6 +80,6 @@ impl Reference {`
`80`	`80`	`}`
`81`	`81`
`82`	`82`	`pub fn is_type(&self) -> bool {`
`83`		`- self.flag.is_type()`
	`83`	`+ self.flag.is_type() \|\| self.flag.is_ts_type_query()`
`84`	`84`	`}`
`85`	`85`	`}`
Original file line number	Diff line number	Diff line change
`@@ -44,7 +44,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/class/declarati`
`44`	`44`	`"node_id": 8`
`45`	`45`	`},`
`46`	`46`	`{`
`47`		`- "flag": "ReferenceFlag(Read)",`
	`47`	`+ "flag": "ReferenceFlag(Read \| TSTypeQuery)",`
`48`	`48`	`"id": 1,`
`49`	`49`	`"name": "A",`
`50`	`50`	`"node_id": 13`
Original file line number	Diff line number	Diff line change
`@@ -52,14 +52,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/instantiation-e`
`52`	`52`	`"id": 4,`
`53`	`53`	`"name": "T",`
`54`	`54`	`"node": "TSTypeParameter",`
`55`		`- "references": [`
`56`		`- {`
`57`		`- "flag": "ReferenceFlag(Type)",`
`58`		`- "id": 0,`
`59`		`- "name": "T",`
`60`		`- "node_id": 31`
`61`		`- }`
`62`		`- ]`
	`55`	`+ "references": []`
`63`	`56`	`}`
`64`	`57`	`]`
`65`	`58`	`}`
`@@ -75,7 +68,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/instantiation-e`
`75`	`68`	`"node": "Function(makeBox)",`
`76`	`69`	`"references": [`
`77`	`70`	`{`
`78`		`- "flag": "ReferenceFlag(Read)",`
	`71`	`+ "flag": "ReferenceFlag(Read \| TSTypeQuery)",`
`79`	`72`	`"id": 0,`
`80`	`73`	`"name": "makeBox",`
`81`	`74`	`"node_id": 27`
Original file line number	Diff line number	Diff line change
`@@ -24,7 +24,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/type-declaratio`
`24`	`24`	`"node": "VariableDeclarator",`
`25`	`25`	`"references": [`
`26`	`26`	`{`
`27`		`- "flag": "ReferenceFlag(Read)",`
	`27`	`+ "flag": "ReferenceFlag(Read \| TSTypeQuery)",`
`28`	`28`	`"id": 0,`
`29`	`29`	`"name": "arg",`
`30`	`30`	`"node_id": 15`
Original file line number	Diff line number	Diff line change
`@@ -45,7 +45,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/type-declaratio`
`45`	`45`	`"node": "VariableDeclarator",`
`46`	`46`	`"references": [`
`47`	`47`	`{`
`48`		`- "flag": "ReferenceFlag(Read)",`
	`48`	`+ "flag": "ReferenceFlag(Read \| TSTypeQuery)",`
`49`	`49`	`"id": 0,`
`50`	`50`	`"name": "k",`
`51`	`51`	`"node_id": 21`
Original file line number	Diff line number	Diff line change
`@@ -31,7 +31,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/type-declaratio`
`31`	`31`	`"node": "VariableDeclarator",`
`32`	`32`	`"references": [`
`33`	`33`	`{`
`34`		`- "flag": "ReferenceFlag(Read)",`
	`34`	`+ "flag": "ReferenceFlag(Read \| TSTypeQuery)",`
`35`	`35`	`"id": 0,`
`36`	`36`	`"name": "x",`
`37`	`37`	`"node_id": 21`