reenable highlighting of (the first line of) comments

llvm-svn: 49816

2 files changed, 8 insertions, 1 deletions

diff --git a/clang/include/clang/Lex/Lexer.h b/clang/include/clang/Lex/Lexer.h
index 2e8267d794b..0ba51a60378 100644
--- a/clang/include/clang/Lex/Lexer.h
+++ b/clang/include/clang/Lex/Lexer.h
@@ -154,6 +154,13 @@ public:
     return BufferPtr == BufferEnd; 
   }
   
+  /// SetCommentRetentionMode - Change the comment retention mode of the lexer
+  /// to the specified mode.  This is really only useful when lexing in raw
+  /// mode, because otherwise the lexer needs to manage this.
+  void SetCommentRetentionState(bool Mode) { 
+    KeepCommentMode = Mode;
+  }
+  
   /// ReadToEndOfLine - Read the rest of the current preprocessor line as an
   /// uninterpreted string.  This switches the lexer out of directive mode.
   std::string ReadToEndOfLine();
diff --git a/clang/lib/Rewrite/HTMLRewrite.cpp b/clang/lib/Rewrite/HTMLRewrite.cpp
index 682fffd82ae..fc50eb1151c 100644
--- a/clang/lib/Rewrite/HTMLRewrite.cpp
+++ b/clang/lib/Rewrite/HTMLRewrite.cpp
@@ -237,7 +237,7 @@ void html::SyntaxHighlight(Rewriter &R, unsigned FileID, Preprocessor &PP) {
   
   // Inform the preprocessor that we want to retain comments as tokens, so we 
   // can highlight them.
-  //PP.SetCommentRetentionState(true, false);
+  L.SetCommentRetentionState(true);
  
   // Lex all the tokens in raw mode, to avoid entering #includes or expanding
   // macros.