#!/bin/sh
#
# Copyright (c) 2005 Junio C Hamano.
#

USAGE='[--interactive | -i] [-v] [--onto <newbase>] <upstream> [<branch>]'
LONG_USAGE='git-rebase replaces <branch> with a new branch of the
same name.  When the --onto option is provided the new branch starts
out with a HEAD equal to <newbase>, otherwise it is equal to <upstream>
It then attempts to create a new commit for each commit from the original
<branch> that does not exist in the <upstream> branch.

It is possible that a merge failure will prevent this process from being
completely automatic.  You will have to resolve any such merge failure
and run git rebase --continue.  Another option is to bypass the commit
that caused the merge failure with git rebase --skip.  To restore the
original <branch> and remove the .dotest working files, use the command
git rebase --abort instead.

Note that if <branch> is not specified on the command line, the
currently checked out branch is used.  You must be in the top
directory of your project to start (or continue) a rebase.

Example:       git-rebase master~1 topic

        A---B---C topic                   A'\''--B'\''--C'\'' topic
       /                   -->           /
  D---E---F---G master          D---E---F---G master
'

SUBDIRECTORY_OK=Yes
. git-sh-setup
set_reflog_action rebase
require_work_tree
cd_to_toplevel

RESOLVEMSG="
When you have resolved this problem run \"git rebase --continue\".
If you would prefer to skip this patch, instead run \"git rebase --skip\".
To restore the original branch and stop rebasing run \"git rebase --abort\".
"
unset newbase
strategy=recursive
do_merge=
dotest=$GIT_DIR/.dotest-merge
prec=4
verbose=
git_am_opt=

continue_merge () {
        test -n "$prev_head" || die "prev_head must be defined"
        test -d "$dotest" || die "$dotest directory does not exist"

        unmerged=$(git ls-files -u)
        if test -n "$unmerged"
        then
                echo "You still have unmerged paths in your index"
                echo "did you forget to use git add?"
                die "$RESOLVEMSG"
        fi

        cmt=`cat $dotest/current`
        if ! git diff-index --quiet HEAD
        then
                if ! git-commit -C "$cmt"
                then
                        echo "Commit failed, please do not call \"git commit\""
                        echo "directly, but instead do one of the following: "
                        die "$RESOLVEMSG"
                fi
                printf "Committed: %0${prec}d " $msgnum
        else
                printf "Already applied: %0${prec}d " $msgnum
        fi
        git rev-list --pretty=oneline -1 "$cmt" | sed -e 's/^[^ ]* //'

        prev_head=`git rev-parse HEAD^0`
        # save the resulting commit so we can read-tree on it later
        echo "$prev_head" > "$dotest/prev_head"

        # onto the next patch:
        msgnum=$(($msgnum + 1))
        echo "$msgnum" >"$dotest/msgnum"
}

call_merge () {
        cmt="$(cat $dotest/cmt.$1)"
        echo "$cmt" > "$dotest/current"
        hd=$(git rev-parse --verify HEAD)
        cmt_name=$(git symbolic-ref HEAD)
        msgnum=$(cat $dotest/msgnum)
        end=$(cat $dotest/end)
        eval GITHEAD_$cmt='"${cmt_name##refs/heads/}~$(($end - $msgnum))"'
        eval GITHEAD_$hd='"$(cat $dotest/onto_name)"'
        export GITHEAD_$cmt GITHEAD_$hd
        git-merge-$strategy "$cmt^" -- "$hd" "$cmt"
        rv=$?
        case "$rv" in
        0)
                unset GITHEAD_$cmt GITHEAD_$hd
                return
                ;;
        1)
                git rerere
                die "$RESOLVEMSG"
                ;;
        2)
                echo "Strategy: $rv $strategy failed, try another" 1>&2
                die "$RESOLVEMSG"
                ;;
        *)
                die "Unknown exit code ($rv) from command:" \
                        "git-merge-$strategy $cmt^ -- HEAD $cmt"
                ;;
        esac
}

finish_rb_merge () {
        if test -f "$dotest/stash"
        then
                stash=$(cat "$dotest/stash")
                git stash apply --index "$stash"
        fi
        rm -r "$dotest"
        echo "All done."
}

read_stash () {
        if test -f "$1"
        then
                cat "$1"
        fi
}
unstash_and_exit () {
        err=$?
        if test -f "$1" && test $err = 0
        then
                git stash apply --index "$1"
        fi
        exit $err
}

is_interactive () {
        test -f "$dotest"/interactive ||
        while case $#,"$1" in 0,|*,-i|*,--interactive) break ;; esac
        do
                shift
        done && test -n "$1"
}

is_interactive "$@" && exec git-rebase--interactive "$@"

while case "$#" in 0) break ;; esac
do
        case "$1" in
        --continue)
                git diff-files --quiet || {
                        echo "You must edit all merge conflicts and then"
                        echo "mark them as resolved using git add"
                        exit 1
                }
                if test -d "$dotest"
                then
                        prev_head="`cat $dotest/prev_head`"
                        end="`cat $dotest/end`"
                        msgnum="`cat $dotest/msgnum`"
                        onto="`cat $dotest/onto`"
                        continue_merge
                        while test "$msgnum" -le "$end"
                        do
                                call_merge "$msgnum"
                                continue_merge
                        done
                        finish_rb_merge
                        exit
                fi
                stash=$(read_stash ".dotest/stash")
                git am --resolved --3way --resolvemsg="$RESOLVEMSG"
                unstash_and_exit "$stash"
                ;;
        --skip)
                if test -d "$dotest"
                then
                        git rerere clear
                        prev_head="`cat $dotest/prev_head`"
                        end="`cat $dotest/end`"
                        msgnum="`cat $dotest/msgnum`"
                        msgnum=$(($msgnum + 1))
                        onto="`cat $dotest/onto`"
                        while test "$msgnum" -le "$end"
                        do
                                call_merge "$msgnum"
                                continue_merge
                        done
                        finish_rb_merge
                        exit
                fi
                stash=$(read_stash ".dotest/stash")
                git am -3 --skip --resolvemsg="$RESOLVEMSG"
                unstash_and_exit "$stash"
                ;;
        --abort)
                git rerere clear
                if test -d "$dotest"
                then
                        if test -f "$dotest/stash"
                        then
                                stash=$(cat "$dotest/stash")
                        fi
                        rm -r "$dotest"
                elif test -d .dotest
                then
                        if test -f ".dotest/stash"
                        then
                                stash=$(cat ".dotest/stash")
                        fi
                        rm -r .dotest
                else
                        die "No rebase in progress?"
                fi
                git reset --hard ORIG_HEAD
                test -z "$stash" || git stash apply --index "$stash"
                exit
                ;;
        --onto)
                test 2 -le "$#" || usage
                newbase="$2"
                shift
                ;;
        -M|-m|--m|--me|--mer|--merg|--merge)
                do_merge=t
                ;;
        -s=*|--s=*|--st=*|--str=*|--stra=*|--strat=*|--strate=*|\
                --strateg=*|--strategy=*|\
        -s|--s|--st|--str|--stra|--strat|--strate|--strateg|--strategy)
                case "$#,$1" in
                *,*=*)
                        strategy=`expr "z$1" : 'z-[^=]*=\(.*\)'` ;;
                1,*)
                        usage ;;
                *)
                        strategy="$2"
                        shift ;;
                esac
                do_merge=t
                ;;
        -v|--verbose)
                verbose=t
                ;;
        --whitespace=*)
                git_am_opt="$git_am_opt $1"
                ;;
        -C*)
                git_am_opt="$git_am_opt $1"
                ;;
        -*)
                usage
                ;;
        *)
                break
                ;;
        esac
        shift
done

# Make sure we do not have .dotest
if test -z "$do_merge"
then
        if mkdir .dotest
        then
                rmdir .dotest
        else
                echo >&2 '
It seems that I cannot create a .dotest directory, and I wonder if you
are in the middle of patch application or another rebase.  If that is not
the case, please rm -fr .dotest and run me again.  I am stopping in case
you still have something valuable there.'
                exit 1
        fi
else
        if test -d "$dotest"
        then
                die "previous dotest directory $dotest still exists." \
                        'try git-rebase < --continue | --abort >'
        fi
fi

# The upstream head must be given.  Make sure it is valid.
upstream_name="$1"
upstream=`git rev-parse --verify "${upstream_name}^0"` ||
    die "invalid upstream $upstream_name"

# Make sure the branch to rebase onto is valid.
onto_name=${newbase-"$upstream_name"}
onto=$(git rev-parse --verify "${onto_name}^0") || exit

# The tree must be clean enough for us to create a stash
stash=$(git stash create) || exit
if test -n "$stash"
then
        echo >&2 "Stashed away your working tree changes"
fi

# If a hook exists, give it a chance to interrupt
if test -x "$GIT_DIR/hooks/pre-rebase"
then
        "$GIT_DIR/hooks/pre-rebase" ${1+"$@"} || {
                echo >&2 "The pre-rebase hook refused to rebase."
                test -z "$stash" || git stash apply --index "$stash"
                exit 1
        }
fi

# If the branch to rebase is given, first switch to it.
case "$#" in
2)
        branch_name="$2"
        git-checkout "$2" || {
                test -z "$stash" || git stash apply --index "$stash"
                usage
        }
        ;;
*)
        if branch_name=`git symbolic-ref -q HEAD`
        then
                branch_name=`expr "z$branch_name" : 'zrefs/heads/\(.*\)'`
        else
                branch_name=HEAD ;# detached
        fi
        ;;
esac
branch=$(git rev-parse --verify "${branch_name}^0") || exit

# Now we are rebasing commits $upstream..$branch on top of $onto

# Check if we are already based on $onto with linear history,
# but this should be done only when upstream and onto are the same.
mb=$(git merge-base "$onto" "$branch")
if test "$upstream" = "$onto" && test "$mb" = "$onto" &&
        # linear history?
        ! git rev-list --parents "$onto".."$branch" | grep " .* " > /dev/null
then
        echo >&2 "Current branch $branch_name is up to date."
        test -z "$stash" || git stash apply --index "$stash"
        exit 0
fi

if test -n "$verbose"
then
        echo "Changes from $mb to $onto:"
        # We want color (if set), but no pager
        GIT_PAGER='' git diff --stat --summary "$mb" "$onto"
fi

# Rewind the head to "$onto"; this saves our current head in ORIG_HEAD.
echo "First, rewinding head to replay your work on top of it..."
git-reset --hard "$onto"

# If the $onto is a proper descendant of the tip of the branch, then
# we just fast forwarded.
if test "$mb" = "$branch"
then
        echo >&2 "Fast-forwarded $branch_name to $onto_name."
        test -z "$stash" || git stash apply --index "$stash"
        exit 0
fi

if test -z "$do_merge"
then
        git format-patch -k --stdout --full-index --ignore-if-in-upstream "$upstream"..ORIG_HEAD |
        git am $git_am_opt --binary -3 -k --resolvemsg="$RESOLVEMSG"
        err=$?

        if test $err = 0
        then
                test -z "$stash" || git stash apply --index "$stash"
                exit
        else
                test -z "$stash" || echo "$stash" >.dotest/stash
                exit $err
        fi
fi

# start doing a rebase with git-merge
# this is rename-aware if the recursive (default) strategy is used

mkdir -p "$dotest"
echo "$onto" > "$dotest/onto"
echo "$onto_name" > "$dotest/onto_name"
prev_head=`git rev-parse HEAD^0`
echo "$prev_head" > "$dotest/prev_head"
test -z "$stash" || echo "$stash" >"$dotest/stash"

msgnum=0
for cmt in `git rev-list --reverse --no-merges "$upstream"..ORIG_HEAD`
do
        msgnum=$(($msgnum + 1))
        echo "$cmt" > "$dotest/cmt.$msgnum"
done

echo 1 >"$dotest/msgnum"
echo $msgnum >"$dotest/end"

end=$msgnum
msgnum=1

while test "$msgnum" -le "$end"
do
        call_merge "$msgnum"
        continue_merge
done

finish_rb_merge