From 10cf933f70657eb0c413259319a103e15abd9f5c Mon Sep 17 00:00:00 2001
From: Sean McGivern <sean@gitlab.com>
Date: Mon, 1 Aug 2016 14:50:08 +0100
Subject: [PATCH] Highlight files based on merged file

---
 lib/gitlab/conflict/file.rb | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/lib/gitlab/conflict/file.rb b/lib/gitlab/conflict/file.rb
index bbabcf265f3..3d2b7272bad 100644
--- a/lib/gitlab/conflict/file.rb
+++ b/lib/gitlab/conflict/file.rb
@@ -59,8 +59,11 @@ module Gitlab
       end
 
       def highlight_lines!
-        their_highlight = Gitlab::Highlight.highlight_lines(repository, their_ref, their_path)
-        our_highlight = Gitlab::Highlight.highlight_lines(repository, our_ref, our_path)
+        their_file = lines.reject { |line| line.type == 'new' }.map(&:text).join("\n")
+        our_file = lines.reject { |line| line.type == 'old' }.map(&:text).join("\n")
+
+        their_highlight = Gitlab::Highlight.highlight(their_path, their_file, repository: repository).lines.map(&:html_safe)
+        our_highlight = Gitlab::Highlight.highlight(our_path, our_file, repository: repository).lines.map(&:html_safe)
 
         lines.each do |line|
           if line.type == 'old'
-- 
GitLab