Die build.py, Die: Part 2 (LocalFetcher, Handlebar support, build script)

chrome/common/extensions/docs/server2/echo_handler.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.
 
+import sys
+import os
+
+# Add the original server location to sys.path so we are able to import
+# modules from there.
+SERVER_PATH = 'chrome/common/extensions/docs/server2/'
+if os.path.abspath(SERVER_PATH) not in sys.path:
+  sys.path.append(os.path.abspath(SERVER_PATH))
+
 import logging
 import urlfetch
 
 from google.appengine.ext import webapp
 from google.appengine.ext.webapp.util import run_wsgi_app
+
 from branch_utility import BranchUtility
-from resource_fetcher import SubversionFetcher
+from local_fetcher import LocalFetcher
+from subversion_fetcher import SubversionFetcher
+from template_utility import TemplateUtility
 
-EXTENSIONS_PATH = 'src/chrome/common/extensions/docs/'
-
+EXTENSIONS_PATH = 'chrome/common/extensions/'
+DOCS_PATH = 'docs/'
+STATIC_PATH = DOCS_PATH + 'static/'
+PUBLIC_TEMPLATE_PATH = DOCS_PATH + 'template2/public/'
+PRIVATE_TEMPLATE_PATH = DOCS_PATH + 'template2/private/'
 
 class MainPage(webapp.RequestHandler):
+  def _FetchResource(self, fetcher, branch, path):
+    result = fetcher.FetchResource(branch, path)
+    for key in result.headers:
+      self.response.headers[key] = result.headers[key]
+    return result.content
+
   def get(self):
-    path = self.request.path.replace('/chrome/extensions/', '')
+    path = self.request.path
 
+    # Hack for getting local files. Still researching how to use a command line
+    # argument to toggle this.
+    if path.startswith('/local'):
+      logging.info(path)
+      path = path.replace('/local', '')
+      fetcher = LocalFetcher(EXTENSIONS_PATH)
+    else:
+      fetcher = SubversionFetcher(EXTENSIONS_PATH, urlfetch)
+    branch_util = BranchUtility(urlfetch)
+    template_util = TemplateUtility()
+
+    path = path.replace('/chrome/extensions/', '')
     parts = path.split('/')
     if len(parts) > 1:
       filename = parts[1]
     else:
       filename = parts[0]
 
     if len(path) > 0 and path[0] == '/':
       path = path.strip('/')
 
-    fetcher = SubversionFetcher(urlfetch)
-    b_util = BranchUtility(urlfetch)
-    channel_name = b_util.GetChannelNameFromPath(path)
-    branch = b_util.GetBranchNumberForChannelName(channel_name)
+    channel_name = branch_util.GetChannelNameFromPath(path)
+    branch = branch_util.GetBranchNumberForChannelName(channel_name)
 
-    logging.info(channel_name + ' ' + branch)
     try:
-      result = fetcher.FetchResource(branch, EXTENSIONS_PATH + filename)
-      content = result.content
-      for key in result.headers:
-        self.response.headers[key] = result.headers[key]
+      content = self._FetchResource(fetcher, branch,
+          PUBLIC_TEMPLATE_PATH + filename)
+      content = template_util.RenderTemplate(content, '{"test": "Hello"}')
     except:
-      content = 'File not found.'
+      logging.info('Template for %s not found.' % filename)
+
+      try:
+        content = self._FetchResource(fetcher, branch, STATIC_PATH + filename)
+      except:
+        content = 'File not found.'
 
     self.response.out.write(content)
 
 application = webapp.WSGIApplication([
   ('/.*', MainPage),
 ], debug=False)
 
-
 def main():
   run_wsgi_app(application)
 
 
 if __name__ == '__main__':
   main()