-
-
Notifications
You must be signed in to change notification settings - Fork 1.4k
/
ThisVariable.ts
102 lines (92 loc) · 2.92 KB
/
ThisVariable.ts
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
import type { AstContext } from '../../Module';
import type { HasEffectsContext } from '../ExecutionContext';
import type { NodeEvent } from '../NodeEvents';
import { type ExpressionEntity, UNKNOWN_EXPRESSION } from '../nodes/shared/Expression';
import {
DiscriminatedPathTracker,
type ObjectPath,
SHARED_RECURSION_TRACKER
} from '../utils/PathTracker';
import LocalVariable from './LocalVariable';
interface ThisDeoptimizationEvent {
event: NodeEvent;
path: ObjectPath;
thisParameter: ExpressionEntity;
}
export default class ThisVariable extends LocalVariable {
private readonly deoptimizedPaths: ObjectPath[] = [];
private readonly entitiesToBeDeoptimized = new Set<ExpressionEntity>();
private readonly thisDeoptimizationList: ThisDeoptimizationEvent[] = [];
private readonly thisDeoptimizations = new DiscriminatedPathTracker();
constructor(context: AstContext) {
super('this', null, null, context);
}
addEntityToBeDeoptimized(entity: ExpressionEntity): void {
for (const path of this.deoptimizedPaths) {
entity.deoptimizePath(path);
}
for (const thisDeoptimization of this.thisDeoptimizationList) {
this.applyThisDeoptimizationEvent(entity, thisDeoptimization);
}
this.entitiesToBeDeoptimized.add(entity);
}
deoptimizePath(path: ObjectPath): void {
if (
path.length === 0 ||
this.deoptimizationTracker.trackEntityAtPathAndGetIfTracked(path, this)
) {
return;
}
this.deoptimizedPaths.push(path);
for (const entity of this.entitiesToBeDeoptimized) {
entity.deoptimizePath(path);
}
}
deoptimizeThisOnEventAtPath(
event: NodeEvent,
path: ObjectPath,
thisParameter: ExpressionEntity
): void {
const thisDeoptimization: ThisDeoptimizationEvent = {
event,
path,
thisParameter
};
if (!this.thisDeoptimizations.trackEntityAtPathAndGetIfTracked(path, event, thisParameter)) {
for (const entity of this.entitiesToBeDeoptimized) {
this.applyThisDeoptimizationEvent(entity, thisDeoptimization);
}
this.thisDeoptimizationList.push(thisDeoptimization);
}
}
hasEffectsWhenAccessedAtPath(path: ObjectPath, context: HasEffectsContext): boolean {
return (
this.getInit(context).hasEffectsWhenAccessedAtPath(path, context) ||
super.hasEffectsWhenAccessedAtPath(path, context)
);
}
hasEffectsWhenAssignedAtPath(
path: ObjectPath,
context: HasEffectsContext,
ignoreAccessors: boolean
): boolean {
return (
this.getInit(context).hasEffectsWhenAssignedAtPath(path, context) ||
super.hasEffectsWhenAssignedAtPath(path, context, ignoreAccessors)
);
}
private applyThisDeoptimizationEvent(
entity: ExpressionEntity,
{ event, path, thisParameter }: ThisDeoptimizationEvent
) {
entity.deoptimizeThisOnEventAtPath(
event,
path,
thisParameter === this ? entity : thisParameter,
SHARED_RECURSION_TRACKER
);
}
private getInit(context: HasEffectsContext): ExpressionEntity {
return context.replacedVariableInits.get(this) || UNKNOWN_EXPRESSION;
}
}