Add chrome_html gatherer, which inlines html and automatically generates image set…

grit/node/structure.py

 #!/usr/bin/env python
 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 '''The <structure> element.
 '''
 
 import os
 
 from grit.node import base
 from grit.node import variant
 
 from grit import constants
 from grit import exception
 from grit import util
 
 import grit.gather.admin_template
+import grit.gather.chrome_html
 import grit.gather.igoogle_strings
 import grit.gather.muppet_strings
 import grit.gather.policy_json
 import grit.gather.rc
 import grit.gather.tr_html
 import grit.gather.txt
 
 import grit.format.rc
 import grit.format.rc_header
 
 # Type of the gatherer to use for each type attribute
 _GATHERERS = {
   'accelerators' : grit.gather.rc.Accelerators,
   'admin_template' : grit.gather.admin_template.AdmGatherer,
+  'chrome_html' : grit.gather.chrome_html.ChromeHtml,
   'dialog'  : grit.gather.rc.Dialog,
   'igoogle' : grit.gather.igoogle_strings.IgoogleStrings,
   'menu'    : grit.gather.rc.Menu,
   'muppet'  : grit.gather.muppet_strings.MuppetStrings,
   'rcdata'  : grit.gather.rc.RCData,
   'tr_html' : grit.gather.tr_html.TrHtml,
   'txt'     : grit.gather.txt.TxtFile,
   'version' : grit.gather.rc.Version,
   'policy_template_metafile' : grit.gather.policy_json.PolicyJson,
 }
 
 
 # Formatter instance to use for each type attribute
 # when formatting .rc files.
 _RC_FORMATTERS = {
   'accelerators' : grit.format.rc.RcSection(),
   'admin_template' : grit.format.rc.RcInclude('ADM'),
+  'chrome_html' : grit.format.rc.RcInclude('HTML'),
   'dialog'  : grit.format.rc.RcSection(),
   'igoogle' : grit.format.rc.RcInclude('XML'),
   'menu'    : grit.format.rc.RcSection(),
   'muppet'  : grit.format.rc.RcInclude('XML'),
   'rcdata'  : grit.format.rc.RcSection(),
   'tr_html' : grit.format.rc.RcInclude('HTML'),
   'txt'     : grit.format.rc.RcInclude('TXT'),
   'version' : grit.format.rc.RcSection(),
   'policy_template_metafile': None,
 }
@@ skipping 51 matching lines @@
     else:
       raise exception.UnexpectedAttribute(
         "Attribute 'line_end' must be one of 'linux' (default), 'windows' or 'mac'")
 
   def GetCliques(self):
     if self.gatherer:
       return self.gatherer.GetCliques()
     else:
       return []
 
+  def GetDataPackPair(self, lang, encoding):
+    """Returns a (id, string|None) pair that represents the resource id and raw
+    bytes of the data (or None if no resource is generated).  This is used to
+    generate the data pack data file.
+    """
+    from grit.format import rc_header
+    id_map = rc_header.Item.tids_
+    id = id_map[self.GetTextualIds()[0]]
+    data = ''
+    if self.gatherer:
+      data = self.gatherer.GetData(lang, encoding)
+    return id, data
+
   def GetTextualIds(self):
-    if self.gatherer and self.attrs['type'] not in ['tr_html', 'admin_template', 'txt']:
+    if self.gatherer and self.attrs['type'] not in [
+        'chrome_html', 'tr_html', 'admin_template', 'txt']:
       return self.gatherer.GetTextualIds()
     else:
       return [self.attrs['name']]
 
   def ItemFormatter(self, t):
     if t == 'rc_header':
       return grit.format.rc_header.Item()
     elif (t in ['rc_all', 'rc_translateable', 'rc_nontranslateable'] and
           self.SatisfiesOutputCondition()):
       return _RC_FORMATTERS[self.attrs['type']]
     else:
       return super(type(self), self).ItemFormatter(t)
 
   def RunGatherers(self, recursive=False, debug=False):
     if self.gatherer:
       return  # idempotent
 
     gathertype = _GATHERERS[self.attrs['type']]
 
     if debug:
       print 'Running gatherer %s for file %s' % (str(gathertype), self.FilenameToOpen())
 
     self.gatherer = gathertype.FromFile(self.FilenameToOpen(),
                                         self.attrs['name'],
                                         self.attrs['encoding'])
     self.gatherer.SetUberClique(self.UberClique())
+    if hasattr(self.GetRoot(), 'defines'):
+      self.gatherer.SetDefines(self.GetRoot().defines)
     self.gatherer.SetAttributes(self.attrs)
     self.gatherer.Parse()
 
     for child in self.children:
       assert isinstance(child, variant.SkeletonNode)
       skel = gathertype.FromFile(child.FilenameToOpen(),
                                  self.attrs['name'],
                                  child.GetEncodingToUse())
       skel.SetUberClique(self.UberClique())
       skel.SetSkeleton(True)
@@ skipping 124 matching lines @@
   def SubstituteMessages(self, substituter):
     '''Propagates substitution to gatherer.
 
     Args:
       substituter: a grit.util.Substituter object.
     '''
     assert self.gatherer
     if self.ExpandVariables():
       self.gatherer.SubstituteMessages(substituter)