Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat(semantic,syntax): add SymbolFlags::ArrowFunction #4946

Merged
merged 1 commit into from
Aug 18, 2024
Merged

feat(semantic,syntax): add SymbolFlags::ArrowFunction #4946

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 7 additions & 1 deletion crates/oxc_semantic/src/binder.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ pub(crate) trait Binder<'a> {

impl<'a> Binder<'a> for VariableDeclarator<'a> {
fn bind(&self, builder: &mut SemanticBuilder<'a>) {
let (includes, excludes) = match self.kind {
let (mut includes, excludes) = match self.kind {
VariableDeclarationKind::Const => (
SymbolFlags::BlockScopedVariable | SymbolFlags::ConstVariable,
SymbolFlags::BlockScopedVariableExcludes,
Expand All @@ -32,6 +32,12 @@ impl<'a> Binder<'a> for VariableDeclarator<'a> {
}
};

if self.init.as_ref().is_some_and(|init| {
matches!(init.get_inner_expression(), Expression::ArrowFunctionExpression(_))
}) {
includes |= SymbolFlags::ArrowFunction;
}

if self.kind.is_lexical() {
self.id.bound_names(&mut |ident| {
let symbol_id = builder.declare_symbol(ident.span, &ident.name, includes, excludes);
Expand Down
2 changes: 1 addition & 1 deletion crates/oxc_semantic/tests/fixtures/typescript-eslint/call-expression/call-expression.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/call-expression
"node": "Program",
"symbols": [
{
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable | ArrowFunction)",
"id": 0,
"name": "foo",
"node": "VariableDeclarator(foo)",
Expand Down
2 changes: 1 addition & 1 deletion ...s/fixtures/typescript-eslint/functions/arrow/default-params/readable-ref-body-shadow.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow
]
},
{
"flag": "SymbolFlags(BlockScopedVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ArrowFunction)",
"id": 1,
"name": "foo",
"node": "VariableDeclarator(foo)",
Expand Down
2 changes: 1 addition & 1 deletion ...c/tests/fixtures/typescript-eslint/functions/arrow/default-params/readable-ref-const.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow
]
},
{
"flag": "SymbolFlags(BlockScopedVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ArrowFunction)",
"id": 1,
"name": "foo",
"node": "VariableDeclarator(foo)",
Expand Down
2 changes: 1 addition & 1 deletion ...tic/tests/fixtures/typescript-eslint/functions/arrow/default-params/readable-ref-let.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow
]
},
{
"flag": "SymbolFlags(BlockScopedVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ArrowFunction)",
"id": 1,
"name": "foo",
"node": "VariableDeclarator(foo)",
Expand Down
2 changes: 1 addition & 1 deletion ...res/typescript-eslint/functions/arrow/default-params/readable-ref-nested-body-shadow.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow
]
},
{
"flag": "SymbolFlags(BlockScopedVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ArrowFunction)",
"id": 1,
"name": "foo",
"node": "VariableDeclarator(foo)",
Expand Down
2 changes: 1 addition & 1 deletion .../tests/fixtures/typescript-eslint/functions/arrow/default-params/readable-ref-nested.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow
]
},
{
"flag": "SymbolFlags(BlockScopedVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ArrowFunction)",
"id": 1,
"name": "foo",
"node": "VariableDeclarator(foo)",
Expand Down
2 changes: 1 addition & 1 deletion .../fixtures/typescript-eslint/functions/arrow/default-params/readable-ref-param-shadow.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow
"references": []
},
{
"flag": "SymbolFlags(BlockScopedVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ArrowFunction)",
"id": 1,
"name": "foo",
"node": "VariableDeclarator(foo)",
Expand Down
2 changes: 1 addition & 1 deletion ...tests/fixtures/typescript-eslint/functions/arrow/default-params/readable-ref-partial.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow
]
},
{
"flag": "SymbolFlags(BlockScopedVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ArrowFunction)",
"id": 1,
"name": "foo",
"node": "VariableDeclarator(foo)",
Expand Down
2 changes: 1 addition & 1 deletion ...emantic/tests/fixtures/typescript-eslint/functions/arrow/default-params/writable-ref.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow
"node": "Program",
"symbols": [
{
"flag": "SymbolFlags(BlockScopedVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ArrowFunction)",
"id": 0,
"name": "foo",
"node": "VariableDeclarator(foo)",
Expand Down
2 changes: 1 addition & 1 deletion crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow/scope.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow
"node": "Program",
"symbols": [
{
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable | ArrowFunction)",
"id": 0,
"name": "arrow",
"node": "VariableDeclarator(arrow)",
Expand Down
2 changes: 1 addition & 1 deletion ...ntic/tests/fixtures/typescript-eslint/functions/arrow/type-parameters/body-reference.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow
"node": "Program",
"symbols": [
{
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable | ArrowFunction)",
"id": 0,
"name": "foo",
"node": "VariableDeclarator(foo)",
Expand Down
2 changes: 1 addition & 1 deletion ...tic/tests/fixtures/typescript-eslint/functions/arrow/type-parameters/param-reference.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow
"node": "Program",
"symbols": [
{
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable | ArrowFunction)",
"id": 0,
"name": "foo",
"node": "VariableDeclarator(foo)",
Expand Down
2 changes: 1 addition & 1 deletion ...ts/fixtures/typescript-eslint/functions/arrow/type-parameters/return-value-reference.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow
"node": "Program",
"symbols": [
{
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable | ArrowFunction)",
"id": 0,
"name": "foo",
"node": "VariableDeclarator(foo)",
Expand Down
2 changes: 1 addition & 1 deletion ...ests/fixtures/typescript-eslint/functions/arrow/type-parameters/type-param-reference.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow
"node": "Program",
"symbols": [
{
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable | ArrowFunction)",
"id": 0,
"name": "foo",
"node": "VariableDeclarator(foo)",
Expand Down
2 changes: 1 addition & 1 deletion ...ixtures/typescript-eslint/functions/arrow/type-parameters/type-parameter-declaration.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow
"node": "Program",
"symbols": [
{
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable | ArrowFunction)",
"id": 0,
"name": "foo",
"node": "VariableDeclarator(foo)",
Expand Down
2 changes: 1 addition & 1 deletion ...xc_semantic/tests/fixtures/typescript-eslint/functions/arrow/type-predicate-asserts1.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow
"node": "Program",
"symbols": [
{
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable | ArrowFunction)",
"id": 0,
"name": "foo",
"node": "VariableDeclarator(foo)",
Expand Down
2 changes: 1 addition & 1 deletion ...xc_semantic/tests/fixtures/typescript-eslint/functions/arrow/type-predicate-asserts2.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow
]
},
{
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable | ArrowFunction)",
"id": 1,
"name": "foo",
"node": "VariableDeclarator(foo)",
Expand Down
2 changes: 1 addition & 1 deletion crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow/type-predicate1.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow
"node": "Program",
"symbols": [
{
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable | ArrowFunction)",
"id": 0,
"name": "foo",
"node": "VariableDeclarator(foo)",
Expand Down
2 changes: 1 addition & 1 deletion crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow/type-predicate2.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/functions/arrow
]
},
{
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable | ArrowFunction)",
"id": 1,
"name": "foo",
"node": "VariableDeclarator(foo)",
Expand Down
2 changes: 1 addition & 1 deletion ...mantic/tests/fixtures/typescript-eslint/global-resolution/module/variable-decl-const.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/global-resoluti
"node": "Program",
"symbols": [
{
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable | ArrowFunction)",
"id": 0,
"name": "top",
"node": "VariableDeclarator(top)",
Expand Down
2 changes: 1 addition & 1 deletion ...semantic/tests/fixtures/typescript-eslint/global-resolution/module/variable-decl-let.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/global-resoluti
"node": "Program",
"symbols": [
{
"flag": "SymbolFlags(BlockScopedVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ArrowFunction)",
"id": 0,
"name": "top",
"node": "VariableDeclarator(top)",
Expand Down
2 changes: 1 addition & 1 deletion ...semantic/tests/fixtures/typescript-eslint/global-resolution/module/variable-decl-var.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/global-resoluti
"node": "Program",
"symbols": [
{
"flag": "SymbolFlags(FunctionScopedVariable)",
"flag": "SymbolFlags(FunctionScopedVariable | ArrowFunction)",
"id": 0,
"name": "top",
"node": "VariableDeclarator(top)",
Expand Down
2 changes: 1 addition & 1 deletion ...mantic/tests/fixtures/typescript-eslint/global-resolution/script/variable-decl-const.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/global-resoluti
"node": "Program",
"symbols": [
{
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ConstVariable | ArrowFunction)",
"id": 0,
"name": "top",
"node": "VariableDeclarator(top)",
Expand Down
2 changes: 1 addition & 1 deletion ...semantic/tests/fixtures/typescript-eslint/global-resolution/script/variable-decl-let.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/global-resoluti
"node": "Program",
"symbols": [
{
"flag": "SymbolFlags(BlockScopedVariable)",
"flag": "SymbolFlags(BlockScopedVariable | ArrowFunction)",
"id": 0,
"name": "top",
"node": "VariableDeclarator(top)",
Expand Down
2 changes: 1 addition & 1 deletion ...semantic/tests/fixtures/typescript-eslint/global-resolution/script/variable-decl-var.snap
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ input_file: crates/oxc_semantic/tests/fixtures/typescript-eslint/global-resoluti
"node": "Program",
"symbols": [
{
"flag": "SymbolFlags(FunctionScopedVariable)",
"flag": "SymbolFlags(FunctionScopedVariable | ArrowFunction)",
"id": 0,
"name": "top",
"node": "VariableDeclarator(top)",
Expand Down
46 changes: 33 additions & 13 deletions crates/oxc_syntax/src/symbol.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -77,22 +77,28 @@ bitflags! {
/// Is this symbol inside an export declaration
const Export = 1 << 4;
const Class = 1 << 5;
const CatchVariable = 1 << 6; // try {} catch(catch_variable) {}
/// `try {} catch(catch_variable) {}`
const CatchVariable = 1 << 6;
/// A function declaration or expression
const Function = 1 << 7;
const Import = 1 << 8; // Imported ESM binding
const TypeImport = 1 << 9; // Imported ESM type-only binding
/// A function or block-scoped variable initialized to an arrow function
const ArrowFunction = 1 << 8;
/// Imported ESM binding
const Import = 1 << 9;
/// Imported ESM type-only binding
const TypeImport = 1 << 10;
// Type specific symbol flags
const TypeAlias = 1 << 10;
const Interface = 1 << 11;
const RegularEnum = 1 << 12;
const ConstEnum = 1 << 13;
const EnumMember = 1 << 14;
const TypeLiteral = 1 << 15;
const TypeParameter = 1 << 16;
const NameSpaceModule = 1 << 17;
const ValueModule = 1 << 18;
const TypeAlias = 1 << 11;
const Interface = 1 << 12;
const RegularEnum = 1 << 13;
const ConstEnum = 1 << 14;
const EnumMember = 1 << 15;
const TypeLiteral = 1 << 16;
const TypeParameter = 1 << 17;
const NameSpaceModule = 1 << 18;
const ValueModule = 1 << 19;
// In a dts file or there is a declare flag
const Ambient = 1 << 19;
const Ambient = 1 << 20;

const Enum = Self::ConstEnum.bits() | Self::RegularEnum.bits();

Expand Down Expand Up @@ -150,11 +156,25 @@ impl SymbolFlags {
self.contains(Self::ConstVariable)
}

/// Returns `true` if this symbol is a function declaration or expression.
///
/// Use [`SymbolFlags::is_function_like`] to check if this symbol is a function or an arrow function.
#[inline]
pub fn is_function(&self) -> bool {
self.contains(Self::Function)
}

#[inline]
pub fn is_arrow_function(&self) -> bool {
self.contains(Self::ArrowFunction)
}

/// Returns `true` if this symbol is an arrow function or a function declaration/expression.
#[inline]
pub fn is_function_like(&self) -> bool {
self.intersects(Self::Function | Self::ArrowFunction)
}

#[inline]
pub fn is_class(&self) -> bool {
self.contains(Self::Class)
Expand Down