Don't crash when file can't be highlighted

1 files changed, 3 insertions, 9 deletions

diff --git a/lib/gitlab/highlight.rb b/lib/gitlab/highlight.rb
index 28cfebef968..4ddb4fea977 100644
--- a/lib/gitlab/highlight.rb
+++ b/lib/gitlab/highlight.rb
@@ -17,19 +17,13 @@ module Gitlab
     end
 
     def highlight(text, continue: true)
-      @formatter.format(lex(text, continue: continue)).html_safe
+      @formatter.format(@lexer.lex(text, continue: continue)).html_safe
+    rescue
+      @formatter.format(Rouge::Lexers::PlainText.lex(text)).html_safe
     end
 
     private
 
-    def lex(text, continue: true)
-      if @lexer == Rouge::Lexers::PlainText
-        @lexer.lex(text)
-      else
-        @lexer.lex(text, continue: continue)
-      end
-    end
-
     def rouge_formatter(options = {})
       options = options.reverse_merge(
         nowrap: true,