/
LabeledStatement.ts
54 lines (50 loc) · 1.63 KB
/
LabeledStatement.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
import MagicString from 'magic-string';
import {
findFirstOccurrenceOutsideComment,
findNonWhiteSpace,
RenderOptions
} from '../../utils/renderHelpers';
import { HasEffectsContext, InclusionContext } from '../ExecutionContext';
import Identifier from './Identifier';
import * as NodeType from './NodeType';
import { IncludeChildren, StatementBase, StatementNode } from './shared/Node';
export default class LabeledStatement extends StatementBase {
body!: StatementNode;
label!: Identifier;
type!: NodeType.tLabeledStatement;
hasEffects(context: HasEffectsContext) {
const brokenFlow = context.brokenFlow;
context.ignore.labels.add(this.label.name);
if (this.body.hasEffects(context)) return true;
context.ignore.labels.delete(this.label.name);
if (context.includedLabels.has(this.label.name)) {
context.includedLabels.delete(this.label.name);
context.brokenFlow = brokenFlow;
}
return false;
}
include(context: InclusionContext, includeChildrenRecursively: IncludeChildren) {
this.included = true;
const brokenFlow = context.brokenFlow;
this.body.include(context, includeChildrenRecursively);
if (includeChildrenRecursively || context.includedLabels.has(this.label.name)) {
this.label.include();
context.includedLabels.delete(this.label.name);
context.brokenFlow = brokenFlow;
}
}
render(code: MagicString, options: RenderOptions) {
if (this.label.included) {
this.label.render(code, options);
} else {
code.remove(
this.start,
findNonWhiteSpace(
code.original,
findFirstOccurrenceOutsideComment(code.original, ':', this.label.end) + 1
)
);
}
this.body.render(code, options);
}
}