From d1a760e5e321d32ced2b78e009da6f544df4fa2f Mon Sep 17 00:00:00 2001
From: Doug Felt <dougfelt@google.com>
Date: Thu, 18 Aug 2016 16:03:32 -0700
Subject: [PATCH] Add ability to highlight some emoji when generating html

This adds an argument to generate_emoji_html that specifies
a file listing codepoint sequences.  Emoji matching one of these
codepoint sequences are highlighted in the output.
---
 generate_emoji_html.py | 45 ++++++++++++++++++++++++++++++++++--------
 1 file changed, 37 insertions(+), 8 deletions(-)

diff --git a/generate_emoji_html.py b/generate_emoji_html.py
index 55595efb3..4b4750a0a 100755
--- a/generate_emoji_html.py
+++ b/generate_emoji_html.py
@@ -107,8 +107,11 @@ def _get_desc(key_tuple, dir_infos, basepaths):
   return CELL_PREFIX + desc
 
 
-def _get_name(key_tuple):
-  CELL_PREFIX = '<td class="name">'
+def _get_name(key_tuple, annotated_tuples):
+  CELL_PREFIX = '<td class="%s">' % (
+      'name' if annotated_tuples is None or key_tuple not in annotated_tuples
+      else 'aname')
+
   if len(key_tuple) != 1:
     name = '(' + ' '.join('U+%04X' % cp for cp in key_tuple) + ')'
   else:
@@ -120,12 +123,13 @@ def _get_name(key_tuple):
   return CELL_PREFIX + name
 
 
-def _generate_content(basedir, font, dir_infos, limit):
+def _generate_content(basedir, font, dir_infos, limit, annotate):
   """Generate an html table for the infos.  basedir is the parent directory
   of the content, filenames will be made relative to this if underneath it,
   else absolute. If limit is true and there are multiple dirs, limit the set of
   sequences to those in the first dir.  If font is not none, generate columns
-  for the text rendered in the font before other columns."""
+  for the text rendered in the font before other columns.  if annotate is
+  not none, highlight sequences that appear in this set."""
 
   lines = ['<table>']
   header_row = ['']
@@ -153,7 +157,7 @@ def _generate_content(basedir, font, dir_infos, limit):
     row = []
     row.extend(_generate_row_cells(key, font, dir_infos, basepaths))
     row.append(_get_desc(key, dir_infos, basepaths))
-    row.append(_get_name(key))
+    row.append(_get_name(key, annotate))
     lines.append(''.join(row))
   return '\n  <tr>'.join(lines) + '\n</table>'
 
@@ -238,6 +242,20 @@ def _get_dir_infos(
   return infos
 
 
+def _parse_annotation_file(afile):
+  annotations = set()
+  line_re = re.compile(r'([0-9a-f ]+)')
+  with open(afile, 'r') as f:
+    for line in f:
+      line = line.strip()
+      if not line or line[0] == '#':
+        continue
+      m = line_re.match(line)
+      if m:
+        annotations.add(tuple([int(s, 16) for s in m.group(1).split()]))
+  return frozenset(annotations)
+
+
 def _instantiate_template(template, arg_dict):
   id_regex = re.compile('{{([a-zA-Z0-9_]+)}}')
   ids = set(m.group(1) for m in id_regex.finditer(template))
@@ -278,10 +296,12 @@ STYLE = """
       td.desc { font-size: 20pt; font-weight: bold; background-color: rgb(210, 210, 210) }
       td.desc img { vertical-align: middle; width: 32px; height: 32px }
       td.name { background-color: white }
+      td.aname { background-color: rgb(250, 65, 75) }
 """
 
-def write_html_page(filename, page_title, font, dir_infos, limit):
-  content = _generate_content(path.dirname(filename), font, dir_infos, limit)
+def write_html_page(filename, page_title, font, dir_infos, limit, annotate):
+  content = _generate_content(
+      path.dirname(filename), font, dir_infos, limit, annotate)
   N_STYLE = STYLE
   if font:
     FONT_FACE_STYLE = """
@@ -329,6 +349,9 @@ def main():
       default=_default_prefix)
   parser.add_argument(
       '-f', '--font', help='emoji font', metavar='font')
+  parser.add_argument(
+      '-a', '--annotate', help='file listing sequences to annotate',
+      metavar='file')
 
   args = parser.parse_args()
   file_parts = path.splitext(args.outfile)
@@ -336,12 +359,18 @@ def main():
     args.outfile = file_parts[0] + '.html'
     print 'added .html extension to filename:\n%s' % args.outfile
 
+  if args.annotate:
+    annotations = _parse_annotation_file(args.annotate)
+  else:
+    annotations = None
+
   dir_infos = _get_dir_infos(
       args.image_dirs, args.exts, args.prefixes, args.titles,
       args.default_ext, args.default_prefix)
 
   write_html_page(
-      args.outfile, args.page_title, args.font, dir_infos, args.limit)
+      args.outfile, args.page_title, args.font, dir_infos, args.limit,
+      annotations)
 
 
 if __name__ == "__main__":