private void VisitBranch(
SyntaxTreeMapBranchInfo branch,
IStNode node,
TContext context,
Func <IStNode, TContext, bool> onNodeEntry,
Action <IStNode, TContext> onNodeExit,
NodePath pathToFollow = null,
int pathToFollowStartLevelIndex = 0)
{
var propValue = branch.Property.GetValue(node);
if (propValue != null)
{
if (branch.IsCollection)
{
var nodeCollection = propValue as IReadOnlyList <IStNode>;
for (var i = 0; i < nodeCollection.Count; i++)
{
context.CurrentPath.Push(new NodePathLevel(branch.Property.Name, i));
this.Visit(nodeCollection[i], context, onNodeEntry, onNodeExit, pathToFollow, pathToFollowStartLevelIndex + 1);
context.CurrentPath.Pop();
}
}
else
{
context.CurrentPath.Push(new NodePathLevel(branch.Property.Name, null));
this.Visit((StNode)propValue, context, onNodeEntry, onNodeExit, pathToFollow, pathToFollowStartLevelIndex + 1);
context.CurrentPath.Pop();
}
}
}
protected override NodePath GetInitialRootPath(IStNode astNode, bool annotate = true)
{
this.annotationId = Guid.NewGuid().ToString();
NodePath nodePath = null;
this.mappedVisitor.Start(
this.codeFileTs.Model,
new SyntaxTreeMappedVisitorContext(new[] { new NodePathLevel(this.codeFileTs.Id, null) }),
(n, c) =>
{
if (n == astNode)
{
nodePath = new NodePath(Languages.TypeScript, c.CurrentPath.Reverse());
}
return(nodePath == null);
},
(n, c) =>
{
if (annotate && n == astNode)
{
n.SetAnnotation(this.GetSnapshotRootAnnotationKey(), this.annotationId);
n.SetAnnotation(this.GetOriginalPathAnnotationKey(), NodePath.PartsToString(Languages.TypeScript, c.CurrentPath.Reverse()));
}
});
return(nodePath);
}
public void Initialize(IStNode astNode)
{
this.annotationId = Guid.NewGuid().ToString();
this.originalRootPath = this.GetInitialRootPath(astNode, !this.codeFileTs.IsReadOnly);
this.treeIsAnnotated = !this.codeFileTs.IsReadOnly;
if (this.codeFileTs.IsReadOnly)
{
this.roots = new[] { new MetadataRoot(this.originalRootPath, this.originalRootPath) };
}
}
public void Start(
IStNode syntaxNode,
TContext context,
Func <IStNode, TContext, bool> onNodeEntry = null,
Action <IStNode, TContext> onNodeExit = null,
NodePath pathToFollow = null,
int pathToFollowStartLevelIndex = 0
)
{
this.Visit(syntaxNode, context, onNodeEntry, onNodeExit, pathToFollow, pathToFollowStartLevelIndex);
}
private void Visit(
IStNode node,
TContext context,
Func <IStNode, TContext, bool> onNodeEntry = null,
Action <IStNode, TContext> onNodeExit = null,
NodePath pathToFollow = null,
int pathToFollowStartLevelIndex = 0)
{
if (onNodeEntry == null || onNodeEntry(node, context))
{
if (node != null)
{
var nodeType = node.GetType();
if (context.Map.ContainsKey(nodeType))
{
var nodeTypeMap = context.Map[nodeType];
if (pathToFollow != null)
{
if (pathToFollow.Levels.Count > pathToFollowStartLevelIndex)
{
var nextLevel = pathToFollow.Levels[pathToFollowStartLevelIndex];
if (nodeTypeMap.ContainsKey(nextLevel.Name))
{
var branch = nodeTypeMap[nextLevel.Name];
this.VisitBranch(branch, node, context, onNodeEntry, onNodeExit, pathToFollow, pathToFollowStartLevelIndex);
}
}
}
else
{
foreach (var branch in nodeTypeMap.Values)
{
this.VisitBranch(branch, node, context, onNodeEntry, onNodeExit, pathToFollow, pathToFollowStartLevelIndex);
}
}
}
}
}
onNodeExit?.Invoke(node, context);
}
protected abstract NodePath GetInitialRootPath(IStNode astNode, bool annotate = true);
