Save all version control dirs.

tools/dom/scripts/idlsync.py

 #!/usr/bin/python
 # Copyright (c) 2012, the Dart project authors.  Please see the AUTHORS file
 # for details. All rights reserved. Use of this source code is governed by a
 # BSD-style license that can be found in the LICENSE file.
 
 import optparse
 import os
 import os.path
 import re
 import shutil
@@ skipping 159 matching lines @@
   readme = template % {
     'script': os.path.basename(__file__),
     'url': url,
     'revision': revision }
 
   readme_path = os.path.join(local_path, 'README')
   out = open(readme_path, 'w')
   out.write(readme)
   out.close()
 
+ZIP_ARCHIVE = 'version-control-dirs.zip'
+
 def SaveVersionControlDir(local_path):
-  version_control_dir = os.path.join(local_path, '.git')
-  if not os.path.isdir(version_control_dir):
-    version_control_dir = os.path.join(local_path, '.svn')
-    if not os.path.isdir(version_control_dir):
-      raise '%s is not a git or svn directory' % local_path
-  temp_dir = tempfile.mkdtemp()
-  shutil.move(version_control_dir, temp_dir)
-  return temp_dir
+  RunCommand([
+    'sh', '-c',
+    'find %s -name .svn -or -name .git | zip -r %s -@' % (
+        os.path.relpath(local_path), ZIP_ARCHIVE)
+  ])
 
 
-def RestoreVersionControlDir(local_path, saved_version_control_dir):
-  version_control_dir = os.path.join(saved_version_control_dir, '.git')
-  if not os.path.isdir(version_control_dir):
-    version_control_dir = os.path.join(saved_version_control_dir, '.svn')
-    if not os.path.isdir(version_control_dir):
-      raise 'Failed restoring version control directory'
-  shutil.move(version_control_dir, local_path)
-  shutil.rmtree(saved_version_control_dir)
-
+def RestoreVersionControlDir():
+  RunCommand(['unzip', ZIP_ARCHIVE, '-d', '.'])
+  RunCommand(['rm', ZIP_ARCHIVE])
 
 def ParseOptions():
   parser = optparse.OptionParser()
   parser.add_option('--webkit-revision', '-w', dest='webkit_revision',
                     help='WebKit IDL revision to install', default=None)
   parser.add_option('--chrome-revision', '-c', dest='chrome_revision',
                     help='Chrome IDL revision to install', default=None)
   parser.add_option('--update', '-u', dest='update',
                     help='IDL to update (webkit | chrome | all)',
                     default='webkit')
   args, _ = parser.parse_args()
   update = {}
   if args.update == 'all' or args.update == 'chrome':
     update['chrome'] = args.chrome_revision
   if args.update == 'all' or args.update == 'webkit':
     update['webkit'] = args.webkit_revision
   return update
 
 
 def main():
   deps = GetDeps()
   update = ParseOptions()
   for (component, remote_path, local_path, readme, depth) in UPDATE_LIST:
     if component in update.keys():
       revision = update[component]
       url, latest = GetSvnRevision(deps, component)
       if revision is None:
         revision = latest
-      saved_version_control_dir = SaveVersionControlDir(local_path);
+      SaveVersionControlDir(local_path);
       RefreshFiles(url, revision, remote_path, local_path, depth)
       PruneExtraFiles(local_path)
       GenerateReadme(local_path, readme, url, revision)
-      RestoreVersionControlDir(local_path, saved_version_control_dir);
+      RestoreVersionControlDir();
 
 if __name__ == '__main__':
   main()