Check bottom and invariants

[stgit] / stgit / commands / pull.py

diff --git a/stgit/commands/pull.py b/stgit/commands/pull.py
index 1e37cb25f99f56149336677c41922e72a509eb7c..bec7fd72a0a7312a0cd90e4c59099fa1fc21dbb2 100644 (file)
--- a/stgit/commands/pull.py
+++ b/stgit/commands/pull.py
@@ -20,60 +20,90 @@ from optparse import OptionParser, make_option
 
 from stgit.commands.common import *
 from stgit.utils import *
 
 from stgit.commands.common import *
 from stgit.utils import *

+from stgit.out import *
+from stgit.config import GitConfigException

 from stgit import stack, git
 
 
 help = 'pull the changes from the remote repository'
 from stgit import stack, git
 
 
 help = 'pull the changes from the remote repository'

-usage = '%prog [options]'
+usage = """%prog [options] [<repository>]

 
 

+Pull the latest changes from the given remote repository (defaulting
+to branch.<name>.remote, or 'origin' if not set). 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 fetch' documentation for the <repository> format."""
+
+directory = DirectoryGotoToplevel()

 options = [make_option('-n', '--nopush',
                        help = 'do not push the patches back after pulling',
                        action = 'store_true'),
 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
     """
 
 def func(parser, options, args):
     """Pull the changes from a remote repository
     """

-    if len(args) != 0:
-        parser.error('incorrect number of arguments')
+    policy = config.get('branch.%s.stgit.pull-policy' % crt_series.get_name()) or \
+             config.get('stgit.pull-policy')

 
 

-    check_local_changes()
-    check_conflicts()
-    check_head_top_equal()
+    if policy == 'rebase':
+        # parent is local
+        if len(args) == 1:
+            parser.error('specifying a repository is meaningless for policy="%s"' % policy)
+        if len(args) > 0:
+            parser.error('incorrect number of arguments')

 
 

-    branch = git.get_head_file()
-    location = read_string(os.path.join(git.base_dir, 'branches', branch))
+    else:
+        # parent is remote
+        if len(args) > 1:
+            parser.error('incorrect number of arguments')

 
 

-    print 'Pulling from "%s"...' % location
-    new_head = git.fetch(location, options.head, options.tag)
-    print 'done'
+        if len(args) >= 1:
+            repository = args[0]
+        else:
+            repository = crt_series.get_parent_remote()

 
 

-    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'
-
-    print_crt_patch()
+    if crt_series.get_protected():
+        raise CmdException, 'This branch is protected. Pulls are not permitted'
+
+    check_local_changes()
+    check_conflicts()
+    check_head_top_equal(crt_series)
+
+    if policy not in ['pull', 'fetch-rebase', 'rebase']:
+        raise GitConfigException, 'Unsupported pull-policy "%s"' % policy
+
+    applied = prepare_rebase(crt_series)
+
+    # pull the remote changes
+    if policy == 'pull':
+        out.info('Pulling from "%s"' % repository)
+        git.pull(repository)
+    elif policy == 'fetch-rebase':
+        out.info('Fetching from "%s"' % repository)
+        git.fetch(repository)
+        try:
+            target = git.fetch_head()
+        except git.GitException:
+            out.error('Could not find the remote head to rebase onto - fix branch.%s.merge in .git/config' % crt_series.get_name())
+            out.error('Pushing any patches back...')
+            post_rebase(crt_series, applied, False, False)
+            raise
+
+        rebase(crt_series, target)
+    elif policy == 'rebase':
+        rebase(crt_series, crt_series.get_parent_branch())
+
+    post_rebase(crt_series, applied, options.nopush, options.merged)
+
+    # maybe tidy up
+    if config.get('stgit.keepoptimized') == 'yes':
+        git.repack()
+
+    print_crt_patch(crt_series)