[ostree] ostbuild: Make --fetch a separate option

[Colin Walters <walters src gnome org>]

commit 610afbd2f1745ff800ec39ab474ef199e186ff46
Author: Colin Walters <walters verbum org>
Date:   Sat Jan 14 16:56:31 2012 -0500

    ostbuild: Make --fetch a separate option

 src/ostbuild/pyostbuild/builtin_resolve.py |   24 ++++++++++++++++++++----
 1 files changed, 20 insertions(+), 4 deletions(-)
---
diff --git a/src/ostbuild/pyostbuild/builtin_resolve.py b/src/ostbuild/pyostbuild/builtin_resolve.py
index 38379e3..d680094 100755
--- a/src/ostbuild/pyostbuild/builtin_resolve.py
+++ b/src/ostbuild/pyostbuild/builtin_resolve.py
@@ -44,9 +44,6 @@ class OstbuildResolve(builtins.Builtin):
         if not os.path.isdir(mirror):
             run_sync(['git', 'clone', '--mirror', uri, tmp_mirror])
             os.rename(tmp_mirror, mirror)
-        elif self.args.fetch:
-            log("Running git fetch for %s" % (name, ))
-            run_sync(['git', 'fetch'], cwd=mirror, log_initiation=False)
         return mirror
 
     def _parse_src_key(self, srckey):
@@ -94,6 +91,7 @@ class OstbuildResolve(builtins.Builtin):
     def execute(self, argv):
         parser = argparse.ArgumentParser(description=self.short_description)
         parser.add_argument('--fetch', action='store_true')
+        parser.add_argument('components', nargs='*')
 
         args = parser.parse_args(argv)
         self.args = args
@@ -104,7 +102,6 @@ class OstbuildResolve(builtins.Builtin):
         self.manifest = json.load(open(manifest_path))
 
         self.resolved_components = map(self._resolve_component_meta, self.manifest['components'])
-
         for component in self.resolved_components:
             (keytype, uri) = self._parse_src_key(component['src'])
             mirrordir = self._ensure_vcs_mirror(component['name'],
@@ -114,6 +111,25 @@ class OstbuildResolve(builtins.Builtin):
                                                           component['branch'])
             component['revision'] = revision
 
+        if args.fetch:
+            if len(args.components) == 0:
+                fetch_components = map(lambda x: x['name'], self.resolved_components)
+            else:
+                fetch_components = args.components
+            for component_name in fetch_components:
+                found = False
+                for component in self.resolved_components:
+                    if component['name'] == component_name:
+                        found = True
+                        break
+                if not found:
+                    fatal("Unknown component %r" % (component_name, ))
+                mirrordir = self._ensure_vcs_mirror(component['name'],
+                                                    keytype, uri,
+                                                    component['branch'])
+                log("Running git fetch for %s" % (component['name'], ))
+                run_sync(['git', 'fetch'], cwd=mirrordir, log_initiation=False)
+
         self.manifest['components'] = self.resolved_components
 
         out_manifest = os.path.join(self.workdir, 'manifest.json')