X-Git-Url: https://git.distorted.org.uk/~mdw/stgit/blobdiff_plain/f338c3c088774601c6cd422bfac635c4187266c4..c73e63b7d7733f1308c2c0c504144e93062bb489:/stgit/commands/pull.py

diff --git a/stgit/commands/pull.py b/stgit/commands/pull.py
index 1e37cb2..e47858a 100644
--- a/stgit/commands/pull.py
+++ b/stgit/commands/pull.py
@@ -24,56 +24,67 @@ from stgit import stack, git
 
 
 help = 'pull the changes from the remote repository'
-usage = '%prog [options]'
+usage = """%prog [options] [<repository>] [<refspec>]
+
+Pull the latest changes from the given repository (defaulting to
+'origin'). This command works by popping all the patches from the
+stack, pulling the changes in the parent repository, setting the base
+of the stack to the latest parent HEAD and pushing the patches back
+(unless '--nopush' is specified). The 'push' operation can fail if
+there are conflicts. They need to be resolved and the patch pushed
+again.
+
+Check the 'git pull' documentation for the <repository> and <refspec>
+format."""
 
 options = [make_option('-n', '--nopush',
                        help = 'do not push the patches back after pulling',
                        action = 'store_true'),
-           make_option('--head', metavar='OTHER_HEAD',
-                       help = 'pull OTHER_HEAD instead of HEAD'),
-           make_option('--tag',
-                       help = 'pull TAG')]
-
+           make_option('-m', '--merged',
+                       help = 'check for patches merged upstream',
+                       action = 'store_true')]
 
 def func(parser, options, args):
     """Pull the changes from a remote repository
     """
-    if len(args) != 0:
+    if len(args) > 2:
         parser.error('incorrect number of arguments')
 
+    if len(args) >= 1:
+        repository = args[0]
+    else:
+        repository = config.get('branch.%s.remote' % git.get_head_file()) or 'origin'
+
+    refspec = None
+    if len(args) == 2:
+        refspec = args[1]
+
+    if crt_series.get_protected():
+        raise CmdException, 'This branch is protected. Pulls are not permitted'
+
     check_local_changes()
     check_conflicts()
     check_head_top_equal()
 
-    branch = git.get_head_file()
-    location = read_string(os.path.join(git.base_dir, 'branches', branch))
-
-    print 'Pulling from "%s"...' % location
-    new_head = git.fetch(location, options.head, options.tag)
+    # pop all patches
+    applied = crt_series.get_applied()
+    if len(applied) > 0:
+        print 'Popping all applied patches...',
+        sys.stdout.flush()
+        crt_series.pop_patch(applied[0])
+        print 'done'
+
+    # pull the remote changes
+    print 'Pulling from "%s"...' % repository
+    git.pull(repository, refspec)
     print 'done'
 
-    if new_head == git_id('base'):
-        print 'Branch already up-to-date'
-    else:
-        applied = crt_series.get_applied()
-
-        if len(applied) > 0:
-            print 'Popping all patches...',
-            sys.stdout.flush()
-            crt_series.pop_patch(applied[0])
-            print 'done'
-
-        git.switch(new_head)
-
-        if options.nopush:
-            applied = []
-        for p in applied:
-            print 'Pushing patch "%s"...' % p,
-            sys.stdout.flush()
-            crt_series.push_patch(p)
-            if crt_series.empty_patch(p):
-                print 'done (empty patch)'
-            else:
-                print 'done'
+    # push the patches back
+    if not options.nopush:
+        push_patches(applied, options.merged)
+
+    # maybe tidy up
+    if config.get('stgit.keepoptimized') == 'yes':
+        git.repack()
 
     print_crt_patch()