hohn/codeql
1
0
Fork 0
mirror of https://github.com/github/codeql.git synced 2026-05-03 20:58:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

C#: Add interpolated string handler attributes to generated stubs

Browse Source
This commit is contained in:
Tamas Vajk
2023-12-05 17:11:44 +01:00
parent faa63dda8b
commit 50b7ab8448
3 changed files with 32 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
csharp/extractor/Semmle.Extraction.CSharp.StubGenerator/StubVisitor.cs
View File

@@ -203,10 +203,12 @@ internal sealed class StubVisitor : SymbolVisitor
private static readonly HashSet<string> attributeAllowList = new() {
"System.FlagsAttribute",
"System.AttributeUsageAttribute"
"System.AttributeUsageAttribute",
"System.Runtime.CompilerServices.InterpolatedStringHandlerAttribute",
"System.Runtime.CompilerServices.InterpolatedStringHandlerArgumentAttribute",
};
private void StubAttribute(AttributeData a, string prefix)
private void StubAttribute(AttributeData a, string prefix, bool addNewLine)
{
if (a.AttributeClass is not INamedTypeSymbol @class)
return;
@@ -232,14 +234,18 @@ internal sealed class StubVisitor : SymbolVisitor
});
stubWriter.Write(")");
}
stubWriter.WriteLine("]");
stubWriter.Write("]");
if (addNewLine)
{
stubWriter.WriteLine();
}
}
public void StubAttributes(IEnumerable<AttributeData> a, string prefix = "")
public void StubAttributes(IEnumerable<AttributeData> a, string prefix = "", bool addNewLine = true)
{
foreach (var attribute in a)
{
StubAttribute(attribute, prefix);
StubAttribute(attribute, prefix, addNewLine);
}
}
@@ -513,6 +519,8 @@ internal sealed class StubVisitor : SymbolVisitor
{
WriteCommaSep(parameters, parameter =>
{
StubAttributes(parameter.GetAttributes(), addNewLine: false);
switch (parameter.RefKind)
{
case RefKind.None: