Extensions Docs Server: Samples zip files

chrome/common/extensions/docs/server2/template_data_source.py

 # 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.
 
+import re
 from path_utils import FormatKey
 from third_party.handlebar import Handlebar
 
 EXTENSIONS_URL = '/chrome/extensions'
 
 class TemplateDataSource(object):
   """This class fetches and compiles templates using the fetcher passed in with
   |cache_builder|.
   """
   def __init__(self,
                branch,
                api_data_source,
                intro_data_source,
                cache_builder,
+               samples_path,
                base_paths):
     self._branch_info = self._MakeBranchDict(branch)
     self._static_resources = ((('/' + branch) if branch != 'local' else '') +
                               '/static')
     self._api_data_source = api_data_source
     self._intro_data_source = intro_data_source
     self._cache = cache_builder.build(self._LoadTemplate)
+    self._dir_cache = cache_builder.build(self._MakeSamplesList)
+    self._file_cache = cache_builder.build(lambda x: x)
     self._base_paths = base_paths
+    self._samples_path = samples_path
 
   def _MakeBranchDict(self, branch):
     return {
       'showWarning': branch != 'stable',
       'branches': [
         { 'name': 'Stable', 'path': EXTENSIONS_URL + '/stable' },
         { 'name': 'Dev', 'path': EXTENSIONS_URL + '/dev' },
         { 'name': 'Beta', 'path': EXTENSIONS_URL + '/beta' },
         { 'name': 'Trunk', 'path': EXTENSIONS_URL + '/trunk' }
       ],
       'current': branch
     }
 
+  def _InsertApiFunctions(self, api_functions, js_file):
+    funcs = re.findall('(chrome\.[a-zA-Z0-9\.]+)',
+                       self._file_cache.get(js_file))
+    return api_functions + list(set(funcs) - set(api_functions))
+
+  def _MakeSamplesList(self, files):
+    samples_list = []
+    for filename in files:
+      if filename.rsplit('/')[-1] == 'manifest.json':
+        # This is a little hacky, but it makes a sample page.
+        sample_path = filename.rsplit('/', 1)[-2]
+        sample_files = filter(lambda x: x.startswith(sample_path), files)
+        api_functions = []
+        for file_ in sample_files:
+          if file_.endswith('.js'):
+            api_functions = self._InsertApiFunctions(api_functions, file_)
+        samples_list.append({
+          'name': filename.rsplit('/')[-2],
+          'path': sample_path.split('/', 1)[1] + '.zip',
+          'files': [f.replace(sample_path + '/', '') for f in sample_files],
+          'api_functions': api_functions
+        })
+    return samples_list
+
   def _LoadTemplate(self, template):
     return Handlebar(template)
 
   def Render(self, template_name):
     """This method will render a template named |template_name|, fetching all
     the partial templates needed from |self._cache|. Partials are retrieved
     from the TemplateDataSource with the |get| method.
     """
     template = self.get(template_name)
     if not template:
       return ''
       # TODO error handling
     return template.render({
       'apis': self._api_data_source,
       'branchInfo': self._branch_info,
       'intros': self._intro_data_source,
       'partials': self,
+      'samples': self._dir_cache.get(self._samples_path, True),
       'static': self._static_resources
     }).text
 
   def __getitem__(self, key):
     return self.get(key)
 
   def get(self, key):
     real_path = FormatKey(key)
     for base_path in self._base_paths:
       try:
         return self._cache.get(base_path + '/' + real_path)
       except:
         pass
     return None