diff --git a/src/EntityFrameworkCore.Projectables.Generator/ReturnTypeSyntaxRewriter.cs b/src/EntityFrameworkCore.Projectables.Generator/ReturnTypeSyntaxRewriter.cs index eca178f..a36cceb 100644 --- a/src/EntityFrameworkCore.Projectables.Generator/ReturnTypeSyntaxRewriter.cs +++ b/src/EntityFrameworkCore.Projectables.Generator/ReturnTypeSyntaxRewriter.cs @@ -18,6 +18,20 @@ namespace EntityFrameworkCore.Projectables.Generator _semanticModel = semanticModel; } + public override SyntaxNode? VisitIdentifierName(IdentifierNameSyntax node) + { + var visitedNode = base.VisitIdentifierName(node); + + var symbolInfo = _semanticModel.GetSymbolInfo(visitedNode); + + if (symbolInfo.Symbol is not null) + { + return SyntaxFactory.IdentifierName(symbolInfo.Symbol.ToDisplayString(SymbolDisplayFormat.FullyQualifiedFormat)); + } + + return visitedNode; + } + public override SyntaxNode? VisitNullableType(NullableTypeSyntax node) { var typeInfo = _semanticModel.GetTypeInfo(node);