93289c050a49c0d5074659fc770683fd01bfbcba
1#!/bin/sh
2#
3# Copyright (c) 2006 Johannes E. Schindelin
4
5# SHORT DESCRIPTION
6#
7# This script makes it easy to fix up commits in the middle of a series,
8# and rearrange commits.
9#
10# The original idea comes from Eric W. Biederman, in
11# http://article.gmane.org/gmane.comp.version-control.git/22407
12
13USAGE='(--continue | --abort | --skip | [--preserve-merges] [--verbose]
14 [--onto <branch>] <upstream> [<branch>])'
15
16. git-sh-setup
17require_work_tree
18
19DOTEST="$GIT_DIR/.dotest-merge"
20TODO="$DOTEST"/todo
21DONE="$DOTEST"/done
22MSG="$DOTEST"/message
23SQUASH_MSG="$DOTEST"/message-squash
24REWRITTEN="$DOTEST"/rewritten
25PRESERVE_MERGES=
26STRATEGY=
27VERBOSE=
28test -d "$REWRITTEN" && PRESERVE_MERGES=t
29test -f "$DOTEST"/strategy && STRATEGY="$(cat "$DOTEST"/strategy)"
30test -f "$DOTEST"/verbose && VERBOSE=t
31
32warn () {
33 echo "$*" >&2
34}
35
36output () {
37 case "$VERBOSE" in
38 '')
39 "$@" > "$DOTEST"/output 2>&1
40 status=$?
41 test $status != 0 &&
42 cat "$DOTEST"/output
43 return $status
44 ;;
45 *)
46 "$@"
47 esac
48}
49
50require_clean_work_tree () {
51 # test if working tree is dirty
52 git rev-parse --verify HEAD > /dev/null &&
53 git update-index --refresh &&
54 git diff-files --quiet &&
55 git diff-index --cached --quiet HEAD ||
56 die "Working tree is dirty"
57}
58
59ORIG_REFLOG_ACTION="$GIT_REFLOG_ACTION"
60
61comment_for_reflog () {
62 case "$ORIG_REFLOG_ACTION" in
63 ''|rebase*)
64 GIT_REFLOG_ACTION="rebase -i ($1)"
65 export GIT_REFLOG_ACTION
66 esac
67}
68
69mark_action_done () {
70 sed -e 1q < "$TODO" >> "$DONE"
71 sed -e 1d < "$TODO" >> "$TODO".new
72 mv -f "$TODO".new "$TODO"
73 count=$(($(wc -l < "$DONE")))
74 total=$(($count+$(wc -l < "$TODO")))
75 printf "Rebasing (%d/%d)\r" $count $total
76 test -z "$VERBOSE" || echo
77}
78
79make_patch () {
80 parent_sha1=$(git rev-parse --verify "$1"^ 2> /dev/null)
81 git diff "$parent_sha1".."$1" > "$DOTEST"/patch
82}
83
84die_with_patch () {
85 test -f "$DOTEST"/message ||
86 git cat-file commit $sha1 | sed "1,/^$/d" > "$DOTEST"/message
87 test -f "$DOTEST"/author-script ||
88 get_author_ident_from_commit $sha1 > "$DOTEST"/author-script
89 make_patch "$1"
90 die "$2"
91}
92
93die_abort () {
94 rm -rf "$DOTEST"
95 die "$1"
96}
97
98pick_one () {
99 case "$1" in -n) sha1=$2 ;; *) sha1=$1 ;; esac
100 output git rev-parse --verify $sha1 || die "Invalid commit name: $sha1"
101 test -d "$REWRITTEN" &&
102 pick_one_preserving_merges "$@" && return
103 parent_sha1=$(git rev-parse --verify $sha1^ 2>/dev/null)
104 current_sha1=$(git rev-parse --verify HEAD)
105 if test $current_sha1 = $parent_sha1; then
106 output git reset --hard $sha1
107 test "a$1" = a-n && output git reset --soft $current_sha1
108 sha1=$(git rev-parse --short $sha1)
109 output warn Fast forward to $sha1
110 else
111 output git cherry-pick $STRATEGY "$@"
112 fi
113}
114
115pick_one_preserving_merges () {
116 case "$1" in -n) sha1=$2 ;; *) sha1=$1 ;; esac
117 sha1=$(git rev-parse $sha1)
118
119 if test -f "$DOTEST"/current-commit
120 then
121 current_commit=$(cat "$DOTEST"/current-commit) &&
122 git rev-parse HEAD > "$REWRITTEN"/$current_commit &&
123 rm "$DOTEST"/current-commit ||
124 die "Cannot write current commit's replacement sha1"
125 fi
126
127 # rewrite parents; if none were rewritten, we can fast-forward.
128 fast_forward=t
129 preserve=t
130 new_parents=
131 for p in $(git rev-list --parents -1 $sha1 | cut -d\ -f2-)
132 do
133 if test -f "$REWRITTEN"/$p
134 then
135 preserve=f
136 new_p=$(cat "$REWRITTEN"/$p)
137 test $p != $new_p && fast_forward=f
138 case "$new_parents" in
139 *$new_p*)
140 ;; # do nothing; that parent is already there
141 *)
142 new_parents="$new_parents $new_p"
143 esac
144 fi
145 done
146 case $fast_forward in
147 t)
148 output warn "Fast forward to $sha1"
149 test $preserve=f && echo $sha1 > "$REWRITTEN"/$sha1
150 ;;
151 f)
152 test "a$1" = a-n && die "Refusing to squash a merge: $sha1"
153
154 first_parent=$(expr "$new_parents" : " \([^ ]*\)")
155 # detach HEAD to current parent
156 output git checkout $first_parent 2> /dev/null ||
157 die "Cannot move HEAD to $first_parent"
158
159 echo $sha1 > "$DOTEST"/current-commit
160 case "$new_parents" in
161 \ *\ *)
162 # redo merge
163 author_script=$(get_author_ident_from_commit $sha1)
164 eval "$author_script"
165 msg="$(git cat-file commit $sha1 | \
166 sed -e '1,/^$/d' -e "s/[\"\\]/\\\\&/g")"
167 # NEEDSWORK: give rerere a chance
168 if ! output git merge $STRATEGY -m "$msg" $new_parents
169 then
170 echo "$msg" > "$GIT_DIR"/MERGE_MSG
171 die Error redoing merge $sha1
172 fi
173 ;;
174 *)
175 output git cherry-pick $STRATEGY "$@" ||
176 die_with_patch $sha1 "Could not pick $sha1"
177 esac
178 esac
179}
180
181nth_string () {
182 case "$1" in
183 *1[0-9]|*[04-9]) echo "$1"th;;
184 *1) echo "$1"st;;
185 *2) echo "$1"nd;;
186 *3) echo "$1"rd;;
187 esac
188}
189
190make_squash_message () {
191 if test -f "$SQUASH_MSG"; then
192 COUNT=$(($(sed -n "s/^# This is [^0-9]*\([0-9]\+\).*/\1/p" \
193 < "$SQUASH_MSG" | tail -n 1)+1))
194 echo "# This is a combination of $COUNT commits."
195 sed -n "2,\$p" < "$SQUASH_MSG"
196 else
197 COUNT=2
198 echo "# This is a combination of two commits."
199 echo "# The first commit's message is:"
200 echo
201 git cat-file commit HEAD | sed -e '1,/^$/d'
202 echo
203 fi
204 echo "# This is the $(nth_string $COUNT) commit message:"
205 echo
206 git cat-file commit $1 | sed -e '1,/^$/d'
207}
208
209peek_next_command () {
210 sed -n "1s/ .*$//p" < "$TODO"
211}
212
213do_next () {
214 test -f "$DOTEST"/message && rm "$DOTEST"/message
215 test -f "$DOTEST"/author-script && rm "$DOTEST"/author-script
216 read command sha1 rest < "$TODO"
217 case "$command" in
218 \#|'')
219 mark_action_done
220 ;;
221 pick)
222 comment_for_reflog pick
223
224 mark_action_done
225 pick_one $sha1 ||
226 die_with_patch $sha1 "Could not apply $sha1... $rest"
227 ;;
228 edit)
229 comment_for_reflog edit
230
231 mark_action_done
232 pick_one $sha1 ||
233 die_with_patch $sha1 "Could not apply $sha1... $rest"
234 make_patch $sha1
235 warn
236 warn "You can amend the commit now, with"
237 warn
238 warn " git commit --amend"
239 warn
240 exit 0
241 ;;
242 squash)
243 comment_for_reflog squash
244
245 test -z "$(grep -ve '^$' -e '^#' < $DONE)" &&
246 die "Cannot 'squash' without a previous commit"
247
248 mark_action_done
249 make_squash_message $sha1 > "$MSG"
250 case "$(peek_next_command)" in
251 squash)
252 EDIT_COMMIT=
253 cp "$MSG" "$SQUASH_MSG"
254 ;;
255 *)
256 EDIT_COMMIT=-e
257 test -f "$SQUASH_MSG" && rm "$SQUASH_MSG"
258 esac
259
260 failed=f
261 pick_one -n $sha1 || failed=t
262 output git reset --soft HEAD^
263 author_script=$(get_author_ident_from_commit $sha1)
264 echo "$author_script" > "$DOTEST"/author-script
265 case $failed in
266 f)
267 # This is like --amend, but with a different message
268 eval "$author_script"
269 export GIT_AUTHOR_NAME GIT_AUTHOR_EMAIL GIT_AUTHOR_DATE
270 output git commit -F "$MSG" $EDIT_COMMIT
271 ;;
272 t)
273 cp "$MSG" "$GIT_DIR"/MERGE_MSG
274 warn
275 warn "Could not apply $sha1... $rest"
276 die_with_patch $sha1 ""
277 esac
278 ;;
279 *)
280 warn "Unknown command: $command $sha1 $rest"
281 die_with_patch $sha1 "Please fix this in the file $TODO."
282 esac
283 test -s "$TODO" && return
284
285 comment_for_reflog finish &&
286 HEADNAME=$(cat "$DOTEST"/head-name) &&
287 OLDHEAD=$(cat "$DOTEST"/head) &&
288 SHORTONTO=$(git rev-parse --short $(cat "$DOTEST"/onto)) &&
289 if test -d "$REWRITTEN"
290 then
291 test -f "$DOTEST"/current-commit &&
292 current_commit=$(cat "$DOTEST"/current-commit) &&
293 git rev-parse HEAD > "$REWRITTEN"/$current_commit
294 NEWHEAD=$(cat "$REWRITTEN"/$OLDHEAD)
295 else
296 NEWHEAD=$(git rev-parse HEAD)
297 fi &&
298 message="$GIT_REFLOG_ACTION: $HEADNAME onto $SHORTONTO)" &&
299 git update-ref -m "$message" $HEADNAME $NEWHEAD $OLDHEAD &&
300 git symbolic-ref HEAD $HEADNAME && {
301 test ! -f "$DOTEST"/verbose ||
302 git diff --stat $(cat "$DOTEST"/head)..HEAD
303 } &&
304 rm -rf "$DOTEST" &&
305 warn "Successfully rebased and updated $HEADNAME."
306
307 exit
308}
309
310do_rest () {
311 while :
312 do
313 do_next
314 done
315}
316
317while case $# in 0) break ;; esac
318do
319 case "$1" in
320 --continue)
321 comment_for_reflog continue
322
323 test -d "$DOTEST" || die "No interactive rebase running"
324
325 # commit if necessary
326 git rev-parse --verify HEAD > /dev/null &&
327 git update-index --refresh &&
328 git diff-files --quiet &&
329 ! git diff-index --cached --quiet HEAD &&
330 . "$DOTEST"/author-script &&
331 export GIT_AUTHOR_NAME GIT_AUTHOR_NAME GIT_AUTHOR_DATE &&
332 git commit -F "$DOTEST"/message -e
333
334 require_clean_work_tree
335 do_rest
336 ;;
337 --abort)
338 comment_for_reflog abort
339
340 test -d "$DOTEST" || die "No interactive rebase running"
341
342 HEADNAME=$(cat "$DOTEST"/head-name)
343 HEAD=$(cat "$DOTEST"/head)
344 git symbolic-ref HEAD $HEADNAME &&
345 output git reset --hard $HEAD &&
346 rm -rf "$DOTEST"
347 exit
348 ;;
349 --skip)
350 comment_for_reflog skip
351
352 test -d "$DOTEST" || die "No interactive rebase running"
353
354 output git reset --hard && do_rest
355 ;;
356 -s|--strategy)
357 shift
358 case "$#,$1" in
359 *,*=*)
360 STRATEGY="-s `expr "z$1" : 'z-[^=]*=\(.*\)'`" ;;
361 1,*)
362 usage ;;
363 *)
364 STRATEGY="-s $2"
365 shift ;;
366 esac
367 ;;
368 --merge)
369 # we use merge anyway
370 ;;
371 -C*)
372 die "Interactive rebase uses merge, so $1 does not make sense"
373 ;;
374 -v|--verbose)
375 VERBOSE=t
376 ;;
377 -p|--preserve-merges)
378 PRESERVE_MERGES=t
379 ;;
380 -i|--interactive)
381 # yeah, we know
382 ;;
383 ''|-h)
384 usage
385 ;;
386 *)
387 test -d "$DOTEST" &&
388 die "Interactive rebase already started"
389
390 git var GIT_COMMITTER_IDENT >/dev/null ||
391 die "You need to set your committer info first"
392
393 comment_for_reflog start
394
395 ONTO=
396 case "$1" in
397 --onto)
398 ONTO=$(git rev-parse --verify "$2") ||
399 die "Does not point to a valid commit: $2"
400 shift; shift
401 ;;
402 esac
403
404 require_clean_work_tree
405
406 if test ! -z "$2"
407 then
408 output git show-ref --verify --quiet "refs/heads/$2" ||
409 die "Invalid branchname: $2"
410 output git checkout "$2" ||
411 die "Could not checkout $2"
412 fi
413
414 HEAD=$(git rev-parse --verify HEAD) || die "No HEAD?"
415 UPSTREAM=$(git rev-parse --verify "$1") || die "Invalid base"
416
417 test -z "$ONTO" && ONTO=$UPSTREAM
418
419 mkdir "$DOTEST" || die "Could not create temporary $DOTEST"
420 : > "$DOTEST"/interactive || die "Could not mark as interactive"
421 git symbolic-ref HEAD > "$DOTEST"/head-name ||
422 die "Could not get HEAD"
423
424 echo $HEAD > "$DOTEST"/head
425 echo $UPSTREAM > "$DOTEST"/upstream
426 echo $ONTO > "$DOTEST"/onto
427 test -z "$STRATEGY" || echo "$STRATEGY" > "$DOTEST"/strategy
428 test t = "$VERBOSE" && : > "$DOTEST"/verbose
429 if test t = "$PRESERVE_MERGES"
430 then
431 # $REWRITTEN contains files for each commit that is
432 # reachable by at least one merge base of $HEAD and
433 # $UPSTREAM. They are not necessarily rewritten, but
434 # their children might be.
435 # This ensures that commits on merged, but otherwise
436 # unrelated side branches are left alone. (Think "X"
437 # in the man page's example.)
438 mkdir "$REWRITTEN" &&
439 for c in $(git merge-base --all $HEAD $UPSTREAM)
440 do
441 echo $ONTO > "$REWRITTEN"/$c ||
442 die "Could not init rewritten commits"
443 done
444 MERGES_OPTION=
445 else
446 MERGES_OPTION=--no-merges
447 fi
448
449 SHORTUPSTREAM=$(git rev-parse --short $UPSTREAM)
450 SHORTHEAD=$(git rev-parse --short $HEAD)
451 SHORTONTO=$(git rev-parse --short $ONTO)
452 cat > "$TODO" << EOF
453# Rebasing $SHORTUPSTREAM..$SHORTHEAD onto $SHORTONTO
454#
455# Commands:
456# pick = use commit
457# edit = use commit, but stop for amending
458# squash = use commit, but meld into previous commit
459#
460# If you remove a line here THAT COMMIT WILL BE LOST.
461#
462EOF
463 git rev-list $MERGES_OPTION --pretty=oneline --abbrev-commit \
464 --abbrev=7 --reverse $UPSTREAM..$HEAD | \
465 sed "s/^/pick /" >> "$TODO"
466
467 test -z "$(grep -ve '^$' -e '^#' < $TODO)" &&
468 die_abort "Nothing to do"
469
470 cp "$TODO" "$TODO".backup
471 git_editor "$TODO" ||
472 die "Could not execute editor"
473
474 test -z "$(grep -ve '^$' -e '^#' < $TODO)" &&
475 die_abort "Nothing to do"
476
477 output git checkout $ONTO && do_rest
478 esac
479 shift
480done