Docserver: Add support for viewing docs with a codereview patch applied

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

Index: chrome/common/extensions/docs/server2/chained_compiled_file_system.py
diff --git a/chrome/common/extensions/docs/server2/chained_compiled_file_system.py b/chrome/common/extensions/docs/server2/chained_compiled_file_system.py
new file mode 100644
index 0000000000000000000000000000000000000000..f592ed6cf2a706fb174b363281f25755f2e58ed2
--- /dev/null
+++ b/chrome/common/extensions/docs/server2/chained_compiled_file_system.py
@@ -0,0 +1,70 @@
+# Copyright 2013 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 compiled_file_system import CompiledFileSystem
+from file_system import FileNotFoundError
+
+class ChainedCompiledFileSystem(object):
+  ''' A CompiledFileSystem implementation that fetches data from a chain of
+  CompiledFileSystems that have different file systems and separate cache
+  namespaces.
+
+  The rules for the compiled file system chain are:
+    - Versions are fetched from the first compiled file system's underlying
+      file system.
+    - Each compiled file system is read in the reverse order (the last one is
+      read first). If the version matches, return the data. Otherwise, read
+      from the previous compiled file system until the first one is read.
+
+  It is used to chain compiled file systems whose underlying file systems are
+  slightly different. This makes it possible to reuse cached compiled data in
+  one of them without recompiling everything that is shared by them.
+  '''
+  class Factory(CompiledFileSystem.Factory):
+    def __init__(self,
+                 factory_and_fs_chain):
+      self._factory_and_fs_chain = factory_and_fs_chain
+
+    def Create(self, populate_function, cls, category=None):
+      return ChainedCompiledFileSystem(
+          [(factory.Create(populate_function, cls, category), fs)
+           for factory, fs in self._factory_and_fs_chain])
+
+  def __init__(self, compiled_fs_chain):
+    assert len(compiled_fs_chain) > 0
+    self._compiled_fs_chain = compiled_fs_chain
+
+  def GetFromFile(self, path, binary=False):
+    # It's possible that a new file is added in the first compiled file system
+    # and it doesn't exist in other compiled file systems.
+    try:
+      first_compiled_fs, first_file_system = self._compiled_fs_chain[0]
+      # The first file system contains both files of a newer version and files
+      # shared with other compiled file systems. We are going to try each
+      # compiled file system in the reverse order and return the data when
+      # version matches. Data cached in other compiled file system will be
+      # reused whenever possible so that we don't need to recompile things that
+      # are not changed across these file systems.
+      version = first_file_system.Stat(path).version
+      for compiled_fs, _ in reversed(self._compiled_fs_chain):
+        if compiled_fs.StatFile(path) == version:
+          return compiled_fs.GetFromFile(path, binary)
+    except FileNotFoundError:
+      pass
+    # Try first operation again to generate the correct stack trace
+    return first_compiled_fs.GetFromFile(path, binary)
+
+  def GetFromFileListing(self, path):
+    if not path.endswith('/'):
+      path += '/'
+    try:
+      first_compiled_fs, first_file_system = self._compiled_fs_chain[0]
+      version = first_file_system.Stat(path).version
+      for compiled_fs, _ in reversed(self._compiled_fs_chain):
+        if compiled_fs.StatFileListing(path) == version:
+          return compiled_fs.GetFromFileListing(path)
+    except FileNotFoundError:
+      pass
+    # Try first operation again to generate the correct stack trace
+    return first_compiled_fs.GetFromFileListing(path)