Pass unit needed for proper check highlighting

Signed-off-by: Michal Čihař <michal@cihar.com>

weblate/html/translate.html

@@ -109,7 +109,7 @@
     {% endif %}
     {% endif %}
     {% endwith %}
-    {% format_translation unit.source unit.translation.subproject.project.source_language search_match=search_query num_plurals=unit.translation.language.nplurals checks=unit.checks %}
+    {% format_translation unit.source unit.translation.subproject.project.source_language search_match=search_query num_plurals=unit.translation.language.nplurals unit=unit %}
     </div>
     {% endif %}
     {% endif %}

weblate/trans/templatetags/translations.py

@@ -82,14 +82,16 @@ def fmt_whitespace(value):
     return value
 
 
-def fmt_check_highlights(value, checks):
+def fmt_check_highlights(value, unit):
     """Formats check highlights"""
-    highlights = None
+    if unit is None:
+        return []
     # Find all checks highlight
-    if checks:
     highlights = []
-        for check in checks:
-            highlights += check.check_obj.check_highlight(value, None)
+    for check in CHECKS:
+        if not CHECKS[check].target:
+            continue
+        highlights += CHECKS[check].check_highlight(value, unit)
     #Sort by order in string
     if highlights:
         highlights.sort(key=lambda tup: tup[0])
@@ -107,13 +109,13 @@ def fmt_check_highlights(value, checks):
             elif eltest[0] > elref[1]:
                 break
     #then transform highlights to escaped html
-        highlights = [(h[0], escape(force_text(h[1]))) for h in highlights]
+    highlights = [(h[0], h[1], escape(force_text(h[2]))) for h in highlights]
     return highlights
 
 
 @register.inclusion_tag('format-translation.html')
 def format_translation(value, language, diff=None, search_match=None,
-                       simple=False, num_plurals=2, checks=None):
+                       simple=False, num_plurals=2, unit=None):
     """
     Nicely formats translation text possibly handling plurals or diff.
     """
@@ -138,7 +140,7 @@ def format_translation(value, language, diff=None, search_match=None,
     parts = []
 
     for idx, value in enumerate(plurals):
-        highlights = fmt_check_highlights(value, checks)
+        highlights = fmt_check_highlights(value, unit)
 
         # HTML escape
         value = escape(force_text(value))