From b4b4a195491020e3786e84884fc9420883cfbaa6 Mon Sep 17 00:00:00 2001
From: Ryan Nowak <nowakra@gmail.com>
Date: Mon, 10 Apr 2017 18:14:57 -0700
Subject: [PATCH] Fix #1194 - DirectiveIRNode doesn't have source

Now with line mappings!
---
 .../InjectDirective.cs                                          | 1 -
 .../DefaultRazorIRLoweringPhase.cs                              | 2 ++
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/Microsoft.AspNetCore.Mvc.Razor.Extensions/InjectDirective.cs b/src/Microsoft.AspNetCore.Mvc.Razor.Extensions/InjectDirective.cs
index 1244ac8192..80ffefb0ff 100644
--- a/src/Microsoft.AspNetCore.Mvc.Razor.Extensions/InjectDirective.cs
+++ b/src/Microsoft.AspNetCore.Mvc.Razor.Extensions/InjectDirective.cs
@@ -53,7 +53,6 @@ namespace Microsoft.AspNetCore.Mvc.Razor.Extensions
                     {
                         TypeName = typeName,
                         MemberName = memberName,
-                        Source = directive.Source,
                         Parent = visitor.Class,
                     };
 
diff --git a/src/Microsoft.AspNetCore.Razor.Language/DefaultRazorIRLoweringPhase.cs b/src/Microsoft.AspNetCore.Razor.Language/DefaultRazorIRLoweringPhase.cs
index c0764a8813..215d0ea86d 100644
--- a/src/Microsoft.AspNetCore.Razor.Language/DefaultRazorIRLoweringPhase.cs
+++ b/src/Microsoft.AspNetCore.Razor.Language/DefaultRazorIRLoweringPhase.cs
@@ -204,6 +204,7 @@ namespace Microsoft.AspNetCore.Razor.Language
                     {
                         Name = chunkGenerator.Descriptor.Name,
                         Descriptor = chunkGenerator.Descriptor,
+                        Source = BuildSourceSpanFromNode(block),
                     });
 
                     base.VisitDirectiveBlock(chunkGenerator, block);
@@ -242,6 +243,7 @@ namespace Microsoft.AspNetCore.Razor.Language
                 {
                     Name = chunkGenerator.Descriptor.Name,
                     Descriptor = chunkGenerator.Descriptor,
+                    Source = BuildSourceSpanFromNode(block),
                 });
 
                 VisitDefault(block);