From d1e86141604c56fce2213750b58771c09c284e7b Mon Sep 17 00:00:00 2001
From: Pranav K <prkrishn@hotmail.com>
Date: Mon, 28 Dec 2015 15:42:30 -0800
Subject: [PATCH] More changes per Pr comments

---
 .../Visitors/CSharpDesignTimeCodeVisitor.cs           | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/Microsoft.AspNet.Razor/CodeGenerators/Visitors/CSharpDesignTimeCodeVisitor.cs b/src/Microsoft.AspNet.Razor/CodeGenerators/Visitors/CSharpDesignTimeCodeVisitor.cs
index 21df5fe4a..39d78e033 100644
--- a/src/Microsoft.AspNet.Razor/CodeGenerators/Visitors/CSharpDesignTimeCodeVisitor.cs
+++ b/src/Microsoft.AspNet.Razor/CodeGenerators/Visitors/CSharpDesignTimeCodeVisitor.cs
@@ -5,6 +5,7 @@
 using System.Diagnostics;
 using System.Globalization;
 using Microsoft.AspNet.Razor.Chunks;
+using Microsoft.AspNet.Razor.Parser.SyntaxTree;
 
 namespace Microsoft.AspNet.Razor.CodeGenerators.Visitors
 {
@@ -74,20 +75,20 @@ protected override void Visit(SetBaseTypeChunk chunk)
 
         protected override void Visit(TagHelperPrefixDirectiveChunk chunk)
         {
-            VisitTagHelperDirectiveChunk(chunk.Prefix, chunk);
+            VisitTagHelperDirectiveChunk(chunk);
         }
 
         protected override void Visit(AddTagHelperChunk chunk)
         {
-            VisitTagHelperDirectiveChunk(chunk.LookupText, chunk);
+            VisitTagHelperDirectiveChunk(chunk);
         }
 
         protected override void Visit(RemoveTagHelperChunk chunk)
         {
-            VisitTagHelperDirectiveChunk(chunk.LookupText, chunk);
+            VisitTagHelperDirectiveChunk(chunk);
         }
 
-        private void VisitTagHelperDirectiveChunk(string text, Chunk chunk)
+        private void VisitTagHelperDirectiveChunk(Chunk chunk)
         {
             // We should always be in design time mode because of the calling AcceptTree method verification.
             Debug.Assert(Context.Host.DesignTimeMode);
@@ -98,6 +99,8 @@ private void VisitTagHelperDirectiveChunk(string text, Chunk chunk)
                 Writer.WriteVariableDeclaration("string", TagHelperDirectiveSyntaxHelper, "null");
             }
 
+            var text = ((Span)chunk.Association).Content.Trim();
+
             Writer
                 .WriteStartAssignment(TagHelperDirectiveSyntaxHelper)
                 .Write("\"");