Generate ADMX template for Chrome OS policies

components/policy/tools/template_writers/writers/adm_writer.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.
 
 
 from writers import template_writer
 import re
 
 NEWLINE = '\r\n'
@@ skipping 150 matching lines @@
 
     builder.AddLine('END POLICY', -1)
     builder.AddLine()
 
   def WriteComment(self, comment):
     self.lines.AddLine('; ' + comment)
 
   def WritePolicy(self, policy):
     if self.CanBeMandatory(policy):
       self._WritePolicy(policy,
-                        self.config['win_reg_mandatory_key_name'],
+                        self.winconfig['reg_mandatory_key_name'],
                         self.policies)
 
   def WriteRecommendedPolicy(self, policy):
     self._WritePolicy(policy,
-                      self.config['win_reg_recommended_key_name'],
+                      self.winconfig['reg_recommended_key_name'],
                       self.recommended_policies)
 
   def BeginPolicyGroup(self, group):
     category_name = self._Escape(group['name'] + '_Category')
     self._AddGuiString(category_name, group['caption'])
     self.policies.AddLine('CATEGORY !!' + category_name, 1)
 
   def EndPolicyGroup(self):
     self.policies.AddLine('END CATEGORY', -1)
     self.policies.AddLine('')
@@ skipping 29 matching lines @@
     lines.AddLine()
 
     return lines
 
   def BeginTemplate(self):
     if self._GetChromiumVersionString() is not None:
       self.WriteComment(self.config['build'] + ' version: ' + \
           self._GetChromiumVersionString())
     self._AddGuiString(self.config['win_supported_os'],
                        self.messages['win_supported_winxpsp2']['text'])
-    categories = self.config['win_mandatory_category_path'] + \
-                  self.config['win_recommended_category_path']
-    strings = self.config['win_category_path_strings'].copy()
+    categories = self.winconfig['mandatory_category_path'] + \
+                 self.winconfig['recommended_category_path']
+    strings = self.winconfig['category_path_strings'].copy()
     if 'adm_category_path_strings' in self.config:
       strings.update(self.config['adm_category_path_strings'])
     for category in categories:
       if (category in strings):
         # Replace {...} by localized messages.
         string = re.sub(r"\{(\w+)\}", \
                         lambda m: self.messages[m.group(1)]['text'], \
                         strings[category])
         self._AddGuiString(category, string)
     # All the policies will be written into self.policies.
     # The final template text will be assembled into self.lines by
     # self.EndTemplate().
 
   def EndTemplate(self):
     # Copy policies into self.lines.
-    policy_class = self.config['win_group_policy_class'].upper()
+    policy_class = self.GetClass().upper();
     for class_name in ['MACHINE', 'USER']:
       if policy_class != 'BOTH' and policy_class != class_name:
         continue
       self.lines.AddLine('CLASS ' + class_name, 1)
       self.lines.AddLines(self._CreateTemplate(
-          self.config['win_mandatory_category_path'],
-          self.config['win_reg_mandatory_key_name'],
+          self.winconfig['mandatory_category_path'],
+          self.winconfig['reg_mandatory_key_name'],
           self.policies))
       self.lines.AddLines(self._CreateTemplate(
-          self.config['win_recommended_category_path'],
-          self.config['win_reg_recommended_key_name'],
+          self.winconfig['recommended_category_path'],
+          self.winconfig['reg_recommended_key_name'],
           self.recommended_policies))
       self.lines.AddLine('', -1)
     # Copy user strings into self.lines.
     self.lines.AddLine('[Strings]')
     self.lines.AddLines(self.strings)
 
   def Init(self):
     # String buffer for building the whole ADM file.
     self.lines = IndentedStringBuilder()
     # String buffer for building the strings section of the ADM file.
     self.strings = IndentedStringBuilder()
     # Map of strings seen, to avoid duplicates.
     self.strings_seen = {}
     # String buffer for building the policies of the ADM file.
     self.policies = IndentedStringBuilder()
     # String buffer for building the recommended policies of the ADM file.
     self.recommended_policies = IndentedStringBuilder()
+    # Shortcut to platform-specific ADMX/ADM specific configuration.
+    assert len(self.platforms) == 1
+    self.winconfig = self.config['win_config'][self.platforms[0]]
 
   def GetTemplateText(self):
     return self.lines.ToString()
+
+  def GetClass(self):
+    return 'Both';