hohn/codeql
1
0
Fork 0
mirror of https://github.com/github/codeql.git synced 2026-04-29 10:45:15 +02:00
Code Issues Packages Projects Releases Wiki Activity

Move AstLineCounter to top level class

Browse Source
This commit is contained in:
Tamas Vajk
2021-02-15 09:19:18 +01:00
parent 4f693be33b
commit 6cc858b9ef
3 changed files with 49 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
csharp/extractor/Semmle.Extraction.CSharp/Entities/Method.AstLineCounter.cs
View File

@@ -1,51 +0,0 @@
using Microsoft.CodeAnalysis;
using Microsoft.CodeAnalysis.CSharp;
using Microsoft.CodeAnalysis.CSharp.Syntax;
using Semmle.Util;
using System.IO;
namespace Semmle.Extraction.CSharp.Entities
{
public abstract partial class Method
{
private class AstLineCounter : CSharpSyntaxVisitor<LineCounts>
{
public override LineCounts DefaultVisit(SyntaxNode node)
{
var text = node.SyntaxTree.GetText().GetSubText(node.GetLocation().SourceSpan).ToString();
return LineCounter.ComputeLineCounts(text);
}
public override LineCounts VisitMethodDeclaration(MethodDeclarationSyntax method)
{
return Visit(method.Identifier, method.Body ?? (SyntaxNode)method.ExpressionBody);
}
public static LineCounts Visit(SyntaxToken identifier, SyntaxNode body)
{
var start = identifier.GetLocation().SourceSpan.Start;
var end = body.GetLocation().SourceSpan.End - 1;
var textSpan = new Microsoft.CodeAnalysis.Text.TextSpan(start, end - start);
var text = body.SyntaxTree.GetText().GetSubText(textSpan) + "\r\n";
return LineCounter.ComputeLineCounts(text);
}
public override LineCounts VisitConstructorDeclaration(ConstructorDeclarationSyntax method)
{
return Visit(method.Identifier, (SyntaxNode)method.Body ?? method.ExpressionBody);
}
public override LineCounts VisitDestructorDeclaration(DestructorDeclarationSyntax method)
{
return Visit(method.Identifier, (SyntaxNode)method.Body ?? method.ExpressionBody);
}
public override LineCounts VisitOperatorDeclaration(OperatorDeclarationSyntax node)
{
return Visit(node.OperatorToken, node.Body ?? (SyntaxNode)node.ExpressionBody);
}
}
}
}

2
csharp/extractor/Semmle.Extraction.CSharp/Entities/Method.cs
View File

@@ -8,7 +8,7 @@ using System.Linq;
namespace Semmle.Extraction.CSharp.Entities
{
public abstract partial class Method : CachedSymbol<IMethodSymbol>, IExpressionParentEntity, IStatementParentEntity
public abstract class Method : CachedSymbol<IMethodSymbol>, IExpressionParentEntity, IStatementParentEntity
{
protected Method(Context cx, IMethodSymbol init)
: base(cx, init) { }

48
csharp/extractor/Semmle.Extraction.CSharp/Populators/AstLineCounter.cs Normal file
View File

@@ -0,0 +1,48 @@
using Microsoft.CodeAnalysis;
using Microsoft.CodeAnalysis.CSharp;
using Microsoft.CodeAnalysis.CSharp.Syntax;
using Semmle.Util;
namespace Semmle.Extraction.CSharp.Populators
{
internal class AstLineCounter : CSharpSyntaxVisitor<LineCounts>
{
public override LineCounts DefaultVisit(SyntaxNode node)
{
var text = node.SyntaxTree.GetText().GetSubText(node.GetLocation().SourceSpan).ToString();
return LineCounter.ComputeLineCounts(text);
}
public override LineCounts VisitMethodDeclaration(MethodDeclarationSyntax method)
{
return Visit(method.Identifier, method.Body ?? (SyntaxNode)method.ExpressionBody);
}
public static LineCounts Visit(SyntaxToken identifier, SyntaxNode body)
{
var start = identifier.GetLocation().SourceSpan.Start;
var end = body.GetLocation().SourceSpan.End - 1;
var textSpan = new Microsoft.CodeAnalysis.Text.TextSpan(start, end - start);
var text = body.SyntaxTree.GetText().GetSubText(textSpan) + "\r\n";
return LineCounter.ComputeLineCounts(text);
}
public override LineCounts VisitConstructorDeclaration(ConstructorDeclarationSyntax method)
{
return Visit(method.Identifier, (SyntaxNode)method.Body ?? method.ExpressionBody);
}
public override LineCounts VisitDestructorDeclaration(DestructorDeclarationSyntax method)
{
return Visit(method.Identifier, (SyntaxNode)method.Body ?? method.ExpressionBody);
}
public override LineCounts VisitOperatorDeclaration(OperatorDeclarationSyntax node)
{
return Visit(node.OperatorToken, node.Body ?? (SyntaxNode)node.ExpressionBody);
}
}
}