mirror of
https://github.com/swc-project/swc.git
synced 2024-12-20 12:12:16 +03:00
575 lines
25 KiB
TypeScript
575 lines
25 KiB
TypeScript
|
// Copyright (c) Microsoft. All rights reserved. Licensed under the Apache License, Version 2.0.
|
||
|
|
// See LICENSE.txt in the project root for complete license information.
|
||
|
|
|
||
|
|
///<reference path='typescript.ts' />
|
||
|
|
|
||
|
|
module TypeScript {
|
||
|
|
export function lastOf(items: any[]): any {
|
||
|
|
return (items === null || items.length === 0) ? null : items[items.length - 1];
|
||
|
|
}
|
||
|
|
|
||
|
|
export function max(a: number, b: number): number {
|
||
|
|
return a >= b ? a : b;
|
||
|
|
}
|
||
|
|
|
||
|
|
export function min(a: number, b: number): number {
|
||
|
|
return a <= b ? a : b;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Helper class representing a path from a root ast node to a (grand)child ast node.
|
||
|
|
// This is helpful as our tree don't have parents.
|
||
|
|
//
|
||
|
|
export class AstPath {
|
||
|
|
public asts: TypeScript.AST[] = [];
|
||
|
|
public top: number = -1;
|
||
|
|
|
||
|
|
static reverseIndexOf(items: any[], index: number): any {
|
||
|
|
return (items === null || items.length <= index) ? null : items[items.length - index - 1];
|
||
|
|
}
|
||
|
|
|
||
|
|
public clone(): AstPath {
|
||
|
|
var clone = new AstPath();
|
||
|
|
clone.asts = this.asts.map((value) => { return value; });
|
||
|
|
clone.top = this.top;
|
||
|
|
return clone;
|
||
|
|
}
|
||
|
|
|
||
|
|
public pop(): TypeScript.AST {
|
||
|
|
var head = this.ast();
|
||
|
|
this.up();
|
||
|
|
|
||
|
|
while (this.asts.length > this.count()) {
|
||
|
|
this.asts.pop();
|
||
|
|
}
|
||
|
|
return head;
|
||
|
|
}
|
||
|
|
|
||
|
|
public push(ast: TypeScript.AST) {
|
||
|
|
while (this.asts.length > this.count()) {
|
||
|
|
this.asts.pop();
|
||
|
|
}
|
||
|
|
this.top = this.asts.length;
|
||
|
|
this.asts.push(ast);
|
||
|
|
}
|
||
|
|
|
||
|
|
public up() {
|
||
|
|
if (this.top <= -1)
|
||
|
|
throw new Error("Invalid call to 'up'");
|
||
|
|
this.top--;
|
||
|
|
}
|
||
|
|
|
||
|
|
public down() {
|
||
|
|
if (this.top == this.ast.length - 1)
|
||
|
|
throw new Error("Invalid call to 'down'");
|
||
|
|
this.top++;
|
||
|
|
}
|
||
|
|
|
||
|
|
public nodeType(): TypeScript.NodeType {
|
||
|
|
if (this.ast() == null)
|
||
|
|
return TypeScript.NodeType.None;
|
||
|
|
return this.ast().nodeType;
|
||
|
|
}
|
||
|
|
|
||
|
|
public ast() {
|
||
|
|
return <TypeScript.AST>AstPath.reverseIndexOf(this.asts, this.asts.length - (this.top + 1));
|
||
|
|
}
|
||
|
|
|
||
|
|
public parent() {
|
||
|
|
return <TypeScript.AST>AstPath.reverseIndexOf(this.asts, this.asts.length - this.top);
|
||
|
|
}
|
||
|
|
|
||
|
|
public count() {
|
||
|
|
return this.top + 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
public get(index: number): TypeScript.AST {
|
||
|
|
return this.asts[index];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isNameOfClass(): boolean {
|
||
|
|
if (this.ast() === null || this.parent() === null)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
return (this.ast().nodeType === TypeScript.NodeType.Name) &&
|
||
|
|
(this.parent().nodeType === TypeScript.NodeType.ClassDeclaration) &&
|
||
|
|
((<TypeScript.InterfaceDeclaration>this.parent()).name === this.ast());
|
||
|
|
}
|
||
|
|
|
||
|
|
public isNameOfInterface(): boolean {
|
||
|
|
if (this.ast() === null || this.parent() === null)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
return (this.ast().nodeType === TypeScript.NodeType.Name) &&
|
||
|
|
(this.parent().nodeType === TypeScript.NodeType.InterfaceDeclaration) &&
|
||
|
|
((<TypeScript.InterfaceDeclaration>this.parent()).name === this.ast());
|
||
|
|
}
|
||
|
|
|
||
|
|
public isNameOfArgument(): boolean {
|
||
|
|
if (this.ast() === null || this.parent() === null)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
return (this.ast().nodeType === TypeScript.NodeType.Name) &&
|
||
|
|
(this.parent().nodeType === TypeScript.NodeType.ArgDecl) &&
|
||
|
|
((<TypeScript.ArgDecl>this.parent()).id === this.ast());
|
||
|
|
}
|
||
|
|
|
||
|
|
public isNameOfVariable(): boolean {
|
||
|
|
if (this.ast() === null || this.parent() === null)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
return (this.ast().nodeType === TypeScript.NodeType.Name) &&
|
||
|
|
(this.parent().nodeType === TypeScript.NodeType.VarDecl) &&
|
||
|
|
((<TypeScript.VarDecl>this.parent()).id === this.ast());
|
||
|
|
}
|
||
|
|
|
||
|
|
public isNameOfModule(): boolean {
|
||
|
|
if (this.ast() === null || this.parent() === null)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
return (this.ast().nodeType === TypeScript.NodeType.Name) &&
|
||
|
|
(this.parent().nodeType === TypeScript.NodeType.ModuleDeclaration) &&
|
||
|
|
((<TypeScript.ModuleDeclaration>this.parent()).name === this.ast());
|
||
|
|
}
|
||
|
|
|
||
|
|
public isNameOfFunction(): boolean {
|
||
|
|
if (this.ast() === null || this.parent() === null)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
return (this.ast().nodeType === TypeScript.NodeType.Name) &&
|
||
|
|
(this.parent().nodeType === TypeScript.NodeType.FuncDecl) &&
|
||
|
|
((<TypeScript.FuncDecl>this.parent()).name === this.ast());
|
||
|
|
}
|
||
|
|
|
||
|
|
public isChildOfScript(): boolean {
|
||
|
|
var ast = lastOf(this.asts);
|
||
|
|
return this.count() >= 3 &&
|
||
|
|
this.asts[this.top] === ast &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.List &&
|
||
|
|
this.asts[this.top - 2].nodeType === TypeScript.NodeType.Script;
|
||
|
|
}
|
||
|
|
|
||
|
|
public isChildOfModule(): boolean {
|
||
|
|
var ast = lastOf(this.asts);
|
||
|
|
return this.count() >= 3 &&
|
||
|
|
this.asts[this.top] === ast &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.List &&
|
||
|
|
this.asts[this.top - 2].nodeType === TypeScript.NodeType.ModuleDeclaration;
|
||
|
|
}
|
||
|
|
|
||
|
|
public isChildOfClass(): boolean {
|
||
|
|
var ast = lastOf(this.asts);
|
||
|
|
return this.count() >= 3 &&
|
||
|
|
this.asts[this.top] === ast &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.List &&
|
||
|
|
this.asts[this.top - 2].nodeType === TypeScript.NodeType.ClassDeclaration;
|
||
|
|
}
|
||
|
|
|
||
|
|
public isArgumentOfClassConstructor(): boolean {
|
||
|
|
var ast = lastOf(this.asts);
|
||
|
|
return this.count() >= 5 &&
|
||
|
|
this.asts[this.top] === ast &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.List &&
|
||
|
|
this.asts[this.top - 2].nodeType === TypeScript.NodeType.FuncDecl &&
|
||
|
|
this.asts[this.top - 3].nodeType === TypeScript.NodeType.List &&
|
||
|
|
this.asts[this.top - 4].nodeType === TypeScript.NodeType.ClassDeclaration &&
|
||
|
|
((<TypeScript.FuncDecl>this.asts[this.top - 2]).isConstructor) &&
|
||
|
|
((<TypeScript.FuncDecl>this.asts[this.top - 2]).arguments === this.asts[this.top - 1]) &&
|
||
|
|
((<TypeScript.ClassDeclaration>this.asts[this.top - 4]).constructorDecl === this.asts[this.top - 2]);
|
||
|
|
}
|
||
|
|
|
||
|
|
public isChildOfInterface(): boolean {
|
||
|
|
var ast = lastOf(this.asts);
|
||
|
|
return this.count() >= 3 &&
|
||
|
|
this.asts[this.top] === ast &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.List &&
|
||
|
|
this.asts[this.top - 2].nodeType === TypeScript.NodeType.InterfaceDeclaration;
|
||
|
|
}
|
||
|
|
|
||
|
|
public isTopLevelImplicitModule() {
|
||
|
|
return this.count() >= 1 &&
|
||
|
|
this.asts[this.top].nodeType === TypeScript.NodeType.ModuleDeclaration &&
|
||
|
|
TypeScript.hasFlag((<TypeScript.ModuleDeclaration>this.asts[this.top]).modFlags, TypeScript.ModuleFlags.IsWholeFile);
|
||
|
|
}
|
||
|
|
|
||
|
|
public isBodyOfTopLevelImplicitModule() {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 0].nodeType === TypeScript.NodeType.List &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.ModuleDeclaration &&
|
||
|
|
(<TypeScript.ModuleDeclaration>this.asts[this.top - 1]).members == this.asts[this.top - 0] &&
|
||
|
|
TypeScript.hasFlag((<TypeScript.ModuleDeclaration>this.asts[this.top - 1]).modFlags, TypeScript.ModuleFlags.IsWholeFile);
|
||
|
|
}
|
||
|
|
|
||
|
|
public isBodyOfScript(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.Script &&
|
||
|
|
(<TypeScript.Script>this.asts[this.top - 1]).bod == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isBodyOfSwitch(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.Switch &&
|
||
|
|
(<TypeScript.SwitchStatement>this.asts[this.top - 1]).caseList == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isBodyOfModule(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.ModuleDeclaration &&
|
||
|
|
(<TypeScript.ModuleDeclaration>this.asts[this.top - 1]).members == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isBodyOfClass(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.ClassDeclaration &&
|
||
|
|
(<TypeScript.ClassDeclaration>this.asts[this.top - 1]).members == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isBodyOfFunction(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.FuncDecl &&
|
||
|
|
(<TypeScript.FuncDecl>this.asts[this.top - 1]).bod == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isBodyOfInterface(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.InterfaceDeclaration &&
|
||
|
|
(<TypeScript.InterfaceDeclaration>this.asts[this.top - 1]).members == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isBodyOfBlock(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.Block &&
|
||
|
|
(<TypeScript.Block>this.asts[this.top - 1]).statements == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isBodyOfFor(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.For &&
|
||
|
|
(<TypeScript.ForStatement>this.asts[this.top - 1]).body == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isBodyOfCase(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.Case &&
|
||
|
|
(<TypeScript.CaseStatement>this.asts[this.top - 1]).body == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isBodyOfTry(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.Try &&
|
||
|
|
(<TypeScript.Try>this.asts[this.top - 1]).body == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isBodyOfCatch(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.Catch &&
|
||
|
|
(<TypeScript.Catch>this.asts[this.top - 1]).body == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isBodyOfDoWhile(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.DoWhile &&
|
||
|
|
(<TypeScript.DoWhileStatement>this.asts[this.top - 1]).body == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isBodyOfWhile(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.While &&
|
||
|
|
(<TypeScript.WhileStatement>this.asts[this.top - 1]).body == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isBodyOfForIn(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.ForIn &&
|
||
|
|
(<TypeScript.ForInStatement>this.asts[this.top - 1]).body == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isBodyOfWith(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.With &&
|
||
|
|
(<TypeScript.WithStatement>this.asts[this.top - 1]).body == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isBodyOfFinally(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.Finally &&
|
||
|
|
(<TypeScript.Finally>this.asts[this.top - 1]).body == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isCaseOfSwitch(): boolean {
|
||
|
|
return this.count() >= 3 &&
|
||
|
|
this.asts[this.top - 2].nodeType === TypeScript.NodeType.Switch &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.List &&
|
||
|
|
(<TypeScript.SwitchStatement>this.asts[this.top - 2]).caseList == this.asts[this.top - 1];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isDefaultCaseOfSwitch(): boolean {
|
||
|
|
return this.count() >= 3 &&
|
||
|
|
this.asts[this.top - 2].nodeType === TypeScript.NodeType.Switch &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.List &&
|
||
|
|
(<TypeScript.SwitchStatement>this.asts[this.top - 2]).caseList == this.asts[this.top - 1] &&
|
||
|
|
(<TypeScript.SwitchStatement>this.asts[this.top - 2]).defaultCase == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isListOfObjectLit(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.ObjectLit &&
|
||
|
|
this.asts[this.top - 0].nodeType === TypeScript.NodeType.List &&
|
||
|
|
(<TypeScript.UnaryExpression>this.asts[this.top - 1]).operand == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isBodyOfObjectLit(): boolean {
|
||
|
|
return this.isListOfObjectLit();
|
||
|
|
}
|
||
|
|
|
||
|
|
public isEmptyListOfObjectLit(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.ObjectLit &&
|
||
|
|
this.asts[this.top - 0].nodeType === TypeScript.NodeType.List &&
|
||
|
|
(<TypeScript.UnaryExpression>this.asts[this.top - 1]).operand == this.asts[this.top - 0] &&
|
||
|
|
(<TypeScript.ASTList>this.asts[this.top - 0]).members.length == 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
public isMemberOfObjectLit(): boolean {
|
||
|
|
return this.count() >= 3 &&
|
||
|
|
this.asts[this.top - 2].nodeType === TypeScript.NodeType.ObjectLit &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.List &&
|
||
|
|
this.asts[this.top - 0].nodeType === TypeScript.NodeType.Member &&
|
||
|
|
(<TypeScript.UnaryExpression>this.asts[this.top - 2]).operand == this.asts[this.top - 1];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isNameOfMemberOfObjectLit(): boolean {
|
||
|
|
return this.count() >= 4 &&
|
||
|
|
this.asts[this.top - 3].nodeType === TypeScript.NodeType.ObjectLit &&
|
||
|
|
this.asts[this.top - 2].nodeType === TypeScript.NodeType.List &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.Member &&
|
||
|
|
this.asts[this.top - 0].nodeType === TypeScript.NodeType.Name &&
|
||
|
|
(<TypeScript.UnaryExpression>this.asts[this.top - 3]).operand == this.asts[this.top - 2];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isListOfArrayLit(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.ArrayLit &&
|
||
|
|
this.asts[this.top - 0].nodeType === TypeScript.NodeType.List &&
|
||
|
|
(<TypeScript.UnaryExpression>this.asts[this.top - 1]).operand == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isTargetOfMember(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.Member &&
|
||
|
|
(<TypeScript.BinaryExpression>this.asts[this.top - 1]).operand1 === this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isMemberOfMember(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.Member &&
|
||
|
|
(<TypeScript.BinaryExpression>this.asts[this.top - 1]).operand2 === this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isItemOfList(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.List;
|
||
|
|
//(<Tools.ASTList>this.asts[this.top - 1]).operand2 === this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isThenOfIf(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.If &&
|
||
|
|
(<TypeScript.IfStatement>this.asts[this.top - 1]).thenBod == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isElseOfIf(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.If &&
|
||
|
|
(<TypeScript.IfStatement>this.asts[this.top - 1]).elseBod == this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isBodyOfDefaultCase(): boolean {
|
||
|
|
return this.isBodyOfCase();
|
||
|
|
}
|
||
|
|
|
||
|
|
public isSingleStatementList(): boolean {
|
||
|
|
return this.count() >= 1 &&
|
||
|
|
this.asts[this.top].nodeType === TypeScript.NodeType.List &&
|
||
|
|
(<TypeScript.ASTList>this.asts[this.top]).members.length === 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
public isArgumentListOfFunction(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 0].nodeType === TypeScript.NodeType.List &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.FuncDecl &&
|
||
|
|
(<TypeScript.FuncDecl>this.asts[this.top - 1]).arguments === this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isArgumentOfFunction(): boolean {
|
||
|
|
return this.count() >= 3 &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.List &&
|
||
|
|
this.asts[this.top - 2].nodeType === TypeScript.NodeType.FuncDecl &&
|
||
|
|
(<TypeScript.FuncDecl>this.asts[this.top - 2]).arguments === this.asts[this.top - 1];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isArgumentListOfCall(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 0].nodeType === TypeScript.NodeType.List &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.Call &&
|
||
|
|
(<TypeScript.CallExpression>this.asts[this.top - 1]).arguments === this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isArgumentListOfNew(): boolean {
|
||
|
|
return this.count() >= 2 &&
|
||
|
|
this.asts[this.top - 0].nodeType === TypeScript.NodeType.List &&
|
||
|
|
this.asts[this.top - 1].nodeType === TypeScript.NodeType.New &&
|
||
|
|
(<TypeScript.CallExpression>this.asts[this.top - 1]).arguments === this.asts[this.top - 0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public isSynthesizedBlock(): boolean {
|
||
|
|
return this.count() >= 1 &&
|
||
|
|
this.asts[this.top - 0].nodeType === TypeScript.NodeType.Block &&
|
||
|
|
(<TypeScript.Block>this.asts[this.top - 0]).isStatementBlock === false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
export function isValidAstNode(ast: TypeScript.ASTSpan): boolean {
|
||
|
|
if (ast === null)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
if (ast.minChar === -1 || ast.limChar === -1)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
export class AstPathContext {
|
||
|
|
public path = new TypeScript.AstPath();
|
||
|
|
}
|
||
|
|
|
||
|
|
export enum GetAstPathOptions {
|
||
|
|
Default = 0,
|
||
|
|
EdgeInclusive = 1,
|
||
|
|
//We need this options dealing with an AST coming from an incomplete AST. For example:
|
||
|
|
// class foo { // r
|
||
|
|
// If we ask for the AST at the position after the "r" character, we won't see we are
|
||
|
|
// inside a comment, because the "class" AST node has a limChar corresponding to the position of
|
||
|
|
// the "{" character, meaning we don't traverse the tree down to the stmt list of the class, meaning
|
||
|
|
// we don't find the "precomment" attached to the errorneous empty stmt.
|
||
|
|
//TODO: It would be nice to be able to get rid of this.
|
||
|
|
DontPruneSearchBasedOnPosition = 1 << 1,
|
||
|
|
}
|
||
|
|
|
||
|
|
///
|
||
|
|
/// Return the stack of AST nodes containing "position"
|
||
|
|
///
|
||
|
|
export function getAstPathToPosition(script: TypeScript.AST, pos: number, options = GetAstPathOptions.Default): TypeScript.AstPath {
|
||
|
|
var lookInComments = (comments: TypeScript.Comment[]) => {
|
||
|
|
if (comments && comments.length > 0) {
|
||
|
|
for (var i = 0; i < comments.length; i++) {
|
||
|
|
var minChar = comments[i].minChar;
|
||
|
|
var limChar = comments[i].limChar;
|
||
|
|
if (!comments[i].isBlockComment) {
|
||
|
|
limChar++; // For single line comments, include 1 more character (for the newline)
|
||
|
|
}
|
||
|
|
if (pos >= minChar && pos < limChar) {
|
||
|
|
ctx.path.push(comments[i]);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
var pre = function (cur: TypeScript.AST, parent: TypeScript.AST, walker: IAstWalker) {
|
||
|
|
if (isValidAstNode(cur)) {
|
||
|
|
|
||
|
|
// Add "cur" to the stack if it contains our position
|
||
|
|
// For "identifier" nodes, we need a special case: A position equal to "limChar" is
|
||
|
|
// valid, since the position corresponds to a caret position (in between characters)
|
||
|
|
// For example:
|
||
|
|
// bar
|
||
|
|
// 0123
|
||
|
|
// If "position == 3", the caret is at the "right" of the "r" character, which should be considered valid
|
||
|
|
var inclusive =
|
||
|
|
hasFlag(options, GetAstPathOptions.EdgeInclusive) ||
|
||
|
|
cur.nodeType === TypeScript.NodeType.Name ||
|
||
|
|
pos === script.limChar; // Special "EOF" case
|
||
|
|
|
||
|
|
var minChar = cur.minChar;
|
||
|
|
var limChar = cur.limChar + (inclusive ? 1 : 0)
|
||
|
|
if (pos >= minChar && pos < limChar) {
|
||
|
|
|
||
|
|
// TODO: Since AST is sometimes not correct wrt to position, only add "cur" if it's better
|
||
|
|
// than top of the stack.
|
||
|
|
var previous = ctx.path.ast();
|
||
|
|
if (previous == null || (cur.minChar >= previous.minChar && cur.limChar <= previous.limChar)) {
|
||
|
|
ctx.path.push(cur);
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
//logger.log("TODO: Ignoring node because minChar, limChar not better than previous node in stack");
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// The AST walker skips comments, but we might be in one, so check the pre/post comments for this node manually
|
||
|
|
if (pos < limChar) {
|
||
|
|
lookInComments(cur.preComments);
|
||
|
|
}
|
||
|
|
if (pos >= minChar) {
|
||
|
|
lookInComments(cur.postComments);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!hasFlag(options, GetAstPathOptions.DontPruneSearchBasedOnPosition)) {
|
||
|
|
// Don't go further down the tree if pos is outside of [minChar, limChar]
|
||
|
|
walker.options.goChildren = (minChar <= pos && pos <= limChar);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return cur;
|
||
|
|
}
|
||
|
|
|
||
|
|
var ctx = new AstPathContext();
|
||
|
|
TypeScript.getAstWalkerFactory().walk(script, pre, null, null, ctx);
|
||
|
|
return ctx.path;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Find a source text offset that is safe for lexing tokens at the given position.
|
||
|
|
// This is used when "position" might be inside a comment or string, etc.
|
||
|
|
//
|
||
|
|
export function getTokenizationOffset(script: TypeScript.Script, position: number): number {
|
||
|
|
var bestOffset = 0;
|
||
|
|
var pre = (cur: TypeScript.AST, parent: TypeScript.AST, walker: TypeScript.IAstWalker): TypeScript.AST => {
|
||
|
|
if (TypeScript.isValidAstNode(cur)) {
|
||
|
|
// Did we find a closer offset?
|
||
|
|
if (cur.minChar <= position) {
|
||
|
|
bestOffset = max(bestOffset, cur.minChar);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Stop the walk if this node is not related to "minChar"
|
||
|
|
if (cur.minChar > position || cur.limChar < bestOffset) {
|
||
|
|
walker.options.goChildren = false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return cur;
|
||
|
|
}
|
||
|
|
|
||
|
|
TypeScript.getAstWalkerFactory().walk(script, pre);
|
||
|
|
return bestOffset;
|
||
|
|
}
|
||
|
|
|
||
|
|
///
|
||
|
|
/// Simple function to Walk an AST using a simple callback function.
|
||
|
|
///
|
||
|
|
export function walkAST(ast: TypeScript.AST, callback: (path: AstPath, walker: TypeScript.IAstWalker) => void ): void {
|
||
|
|
var pre = function (cur: TypeScript.AST, parent: TypeScript.AST, walker: TypeScript.IAstWalker) {
|
||
|
|
var path: TypeScript.AstPath = walker.state;
|
||
|
|
path.push(cur);
|
||
|
|
callback(path, walker);
|
||
|
|
return cur;
|
||
|
|
}
|
||
|
|
var post = function (cur: TypeScript.AST, parent: TypeScript.AST, walker: TypeScript.IAstWalker) {
|
||
|
|
var path: TypeScript.AstPath = walker.state;
|
||
|
|
path.pop();
|
||
|
|
return cur;
|
||
|
|
}
|
||
|
|
|
||
|
|
var path = new AstPath();
|
||
|
|
TypeScript.getAstWalkerFactory().walk(ast, pre, post, null, path);
|
||
|
|
}
|
||
|
|
}
|