git-gui: Reworded 'Include' to 'Add' to match core Git.
[gitweb.git] / git-gui
diff --git a/git-gui b/git-gui
index 063d83c124deeab6823203a4c9cba64dc64eaa76..1891215a646b8715ac08dfb408e9ea31a12b1d74 100755 (executable)
--- a/git-gui
+++ b/git-gui
@@ -2,10 +2,14 @@
 # Tcl ignores the next line -*- tcl -*- \
 exec wish "$0" -- "$@"
 
-# Copyright (C) 2006 Shawn Pearce, Paul Mackerras.  All rights reserved.
-# This program is free software; it may be used, copied, modified
-# and distributed under the terms of the GNU General Public Licence,
-# either version 2, or (at your option) any later version.
+set copyright {
+Copyright © 2006 Shawn Pearce, Paul Mackerras.
+
+All rights reserved.
+
+This program is free software; it may be used, copied, modified
+and distributed under the terms of the GNU General Public Licence,
+either version 2, or (at your option) any later version.}
 
 set appname [lindex [file split $argv0] end]
 set gitdir {}
@@ -141,6 +145,28 @@ proc error_popup {msg} {
        eval $cmd
 }
 
+proc warn_popup {msg} {
+       global gitdir appname
+
+       set title $appname
+       if {$gitdir ne {}} {
+               append title { (}
+               append title [lindex \
+                       [file split [file normalize [file dirname $gitdir]]] \
+                       end]
+               append title {)}
+       }
+       set cmd [list tk_messageBox \
+               -icon warning \
+               -type ok \
+               -title "$title: warning" \
+               -message $msg]
+       if {[winfo ismapped .]} {
+               lappend cmd -parent .
+       }
+       eval $cmd
+}
+
 proc info_popup {msg} {
        global gitdir appname
 
@@ -154,7 +180,7 @@ proc info_popup {msg} {
        }
        tk_messageBox \
                -parent . \
-               -icon error \
+               -icon info \
                -type ok \
                -title $title \
                -message $msg
@@ -231,25 +257,47 @@ proc unlock_index {} {
 ##
 ## status
 
-proc repository_state {hdvar ctvar} {
-       global gitdir
-       upvar $hdvar hd $ctvar ct
+proc repository_state {ctvar hdvar mhvar} {
+       global gitdir current_branch
+       upvar $ctvar ct $hdvar hd $mhvar mh
+
+       set mh [list]
+
+       if {[catch {set current_branch [exec git symbolic-ref HEAD]}]} {
+               set current_branch {}
+       } else {
+               regsub ^refs/((heads|tags|remotes)/)? \
+                       $current_branch \
+                       {} \
+                       current_branch
+       }
 
        if {[catch {set hd [exec git rev-parse --verify HEAD]}]} {
                set hd {}
                set ct initial
-       } elseif {[file exists [file join $gitdir MERGE_HEAD]]} {
+               return
+       }
+
+       set merge_head [file join $gitdir MERGE_HEAD]
+       if {[file exists $merge_head]} {
                set ct merge
-       } else {
-               set ct normal
+               set fd_mh [open $merge_head r]
+               while {[gets $fd_mh line] >= 0} {
+                       lappend mh $line
+               }
+               close $fd_mh
+               return
        }
+
+       set ct normal
 }
 
 proc PARENT {} {
        global PARENT empty_tree
 
-       if {$PARENT ne {}} {
-               return $PARENT
+       set p [lindex $PARENT 0]
+       if {$p ne {}} {
+               return $p
        }
        if {$empty_tree eq {}} {
                set empty_tree [exec git mktree << {}]
@@ -258,21 +306,22 @@ proc PARENT {} {
 }
 
 proc rescan {after} {
-       global HEAD PARENT commit_type
+       global HEAD PARENT MERGE_HEAD commit_type
        global ui_index ui_other ui_status_value ui_comm
        global rescan_active file_states
        global repo_config
 
        if {$rescan_active > 0 || ![lock_index read]} return
 
-       repository_state new_HEAD new_type
+       repository_state newType newHEAD newMERGE_HEAD
        if {[string match amend* $commit_type]
-               && $new_type eq {normal}
-               && $new_HEAD eq $HEAD} {
+               && $newType eq {normal}
+               && $newHEAD eq $HEAD} {
        } else {
-               set HEAD $new_HEAD
-               set PARENT $new_HEAD
-               set commit_type $new_type
+               set HEAD $newHEAD
+               set PARENT $newHEAD
+               set MERGE_HEAD $newMERGE_HEAD
+               set commit_type $newType
        }
 
        array unset file_states
@@ -283,8 +332,8 @@ proc rescan {after} {
                } elseif {[load_message MERGE_MSG]} {
                } elseif {[load_message SQUASH_MSG]} {
                }
-               $ui_comm edit modified false
                $ui_comm edit reset
+               $ui_comm edit modified false
        }
 
        if {$repo_config(gui.trustmtime) eq {true}} {
@@ -369,11 +418,14 @@ proc read_diff_index {fd after} {
                set z2 [string first "\0" $buf_rdi $z1]
                if {$z2 == -1} break
 
+               incr c
+               set i [split [string range $buf_rdi $c [expr {$z1 - 2}]] { }]
+               merge_state \
+                       [string range $buf_rdi $z1 [expr {$z2 - 1}]] \
+                       [lindex $i 4]? \
+                       [list [lindex $i 0] [lindex $i 2]] \
+                       [list]
                set c $z2
-               incr z2 -1
-               display_file \
-                       [string range $buf_rdi $z1 $z2] \
-                       [string index $buf_rdi [expr {$z1 - 2}]]?
                incr c
        }
        if {$c < $n} {
@@ -398,11 +450,14 @@ proc read_diff_files {fd after} {
                set z2 [string first "\0" $buf_rdf $z1]
                if {$z2 == -1} break
 
+               incr c
+               set i [split [string range $buf_rdf $c [expr {$z1 - 2}]] { }]
+               merge_state \
+                       [string range $buf_rdf $z1 [expr {$z2 - 1}]] \
+                       ?[lindex $i 4] \
+                       [list] \
+                       [list [lindex $i 0] [lindex $i 2]]
                set c $z2
-               incr z2 -1
-               display_file \
-                       [string range $buf_rdf $z1 $z2] \
-                       ?[string index $buf_rdf [expr {$z1 - 2}]]
                incr c
        }
        if {$c < $n} {
@@ -421,7 +476,7 @@ proc read_ls_others {fd after} {
        set pck [split $buf_rlo "\0"]
        set buf_rlo [lindex $pck end]
        foreach p [lrange $pck 0 end-1] {
-               display_file $p ?O
+               merge_state $p ?O
        }
        rescan_done $fd buf_rlo $after
 }
@@ -444,8 +499,8 @@ proc rescan_done {fd buf after} {
                set pathList [list]
                foreach path [array names file_states] {
                        switch -- [lindex $file_states($path) 0] {
-                       AM -
-                       MM {lappend pathList $path}
+                       A? -
+                       M? {lappend pathList $path}
                        }
                }
                if {$pathList ne {}} {
@@ -539,7 +594,7 @@ files list, to prevent possible confusion.
 
 proc show_diff {path {w {}} {lno {}}} {
        global file_states file_lists
-       global diff_3way diff_active repo_config
+       global is_3way_diff diff_active repo_config
        global ui_diff current_diff ui_status_value
 
        if {$diff_active || ![lock_index read]} return
@@ -560,7 +615,7 @@ proc show_diff {path {w {}} {lno {}}} {
 
        set s $file_states($path)
        set m [lindex $s 0]
-       set diff_3way 0
+       set is_3way_diff 0
        set diff_active 1
        set current_diff $path
        set ui_status_value "Loading diff of [escape_path $path]..."
@@ -615,48 +670,52 @@ proc show_diff {path {w {}} {lno {}}} {
 }
 
 proc read_diff {fd} {
-       global ui_diff ui_status_value diff_3way diff_active
+       global ui_diff ui_status_value is_3way_diff diff_active
        global repo_config
 
+       $ui_diff conf -state normal
        while {[gets $fd line] >= 0} {
-               if {[string match {diff --git *} $line]} continue
+               # -- Cleanup uninteresting diff header lines.
+               #
+               if {[string match {diff --git *}      $line]} continue
                if {[string match {diff --combined *} $line]} continue
-               if {[string match {--- *} $line]} continue
-               if {[string match {+++ *} $line]} continue
-               if {[string match index* $line]} {
-                       if {[string first , $line] >= 0} {
-                               set diff_3way 1
-                       }
+               if {[string match {--- *}             $line]} continue
+               if {[string match {+++ *}             $line]} continue
+               if {$line eq {deleted file mode 120000}} {
+                       set line "deleted symlink"
                }
 
-               $ui_diff conf -state normal
-               if {!$diff_3way} {
-                       set x [string index $line 0]
-                       switch -- $x {
-                       "@" {set tags da}
-                       "+" {set tags dp}
-                       "-" {set tags dm}
+               # -- Automatically detect if this is a 3 way diff.
+               #
+               if {[string match {@@@ *} $line]} {set is_3way_diff 1}
+
+               # -- Reformat a 3 way diff, 'cause its too weird.
+               #
+               if {$is_3way_diff} {
+                       set op [string range $line 0 1]
+                       switch -- $op {
+                       {@@} {set tags d_@}
+                       {++} {set tags d_+ ; set op { +}}
+                       {--} {set tags d_- ; set op { -}}
+                       { +} {set tags d_++; set op {++}}
+                       { -} {set tags d_--; set op {--}}
+                       {+ } {set tags d_-+; set op {-+}}
+                       {- } {set tags d_+-; set op {+-}}
                        default {set tags {}}
                        }
+                       set line [string replace $line 0 1 $op]
                } else {
-                       set x [string range $line 0 1]
-                       switch -- $x {
-                       default {set tags {}}
-                       "@@" {set tags da}
-                       "++" {set tags dp; set x " +"}
-                       " +" {set tags {di bold}; set x "++"}
-                       "+ " {set tags dni; set x "-+"}
-                       "--" {set tags dm; set x " -"}
-                       " -" {set tags {dm bold}; set x "--"}
-                       "- " {set tags di; set x "+-"}
+                       switch -- [string index $line 0] {
+                       @ {set tags d_@}
+                       + {set tags d_+}
+                       - {set tags d_-}
                        default {set tags {}}
                        }
-                       set line [string replace $line 0 1 $x]
                }
                $ui_diff insert end $line $tags
-               $ui_diff insert end "\n"
-               $ui_diff conf -state disabled
+               $ui_diff insert end "\n" $tags
        }
+       $ui_diff conf -state disabled
 
        if {[eof $fd]} {
                close $fd
@@ -676,23 +735,36 @@ proc read_diff {fd} {
 ## commit
 
 proc load_last_commit {} {
-       global HEAD PARENT commit_type ui_comm
+       global HEAD PARENT MERGE_HEAD commit_type ui_comm
 
-       if {[string match amend* $commit_type]} return
-       if {$commit_type ne {normal}} {
-               error_popup "Can't amend a $commit_type commit."
+       if {[llength $PARENT] == 0} {
+               error_popup {There is nothing to amend.
+
+You are about to create the initial commit.
+There is no commit before this to amend.
+}
+               return
+       }
+
+       repository_state curType curHEAD curMERGE_HEAD
+       if {$curType eq {merge}} {
+               error_popup {Cannot amend while merging.
+
+You are currently in the middle of a merge that
+has not been fully completed.  You cannot amend
+the prior commit unless you first abort the
+current merge activity.
+}
                return
        }
 
        set msg {}
-       set parent {}
-       set parent_count 0
+       set parents [list]
        if {[catch {
-                       set fd [open "| git cat-file commit $HEAD" r]
+                       set fd [open "| git cat-file commit $curHEAD" r]
                        while {[gets $fd line] > 0} {
                                if {[string match {parent *} $line]} {
-                                       set parent [string range $line 7 end]
-                                       incr parent_count
+                                       lappend parents [string range $line 7 end]
                                }
                        }
                        set msg [string trim [read $fd]]
@@ -702,23 +774,19 @@ proc load_last_commit {} {
                return
        }
 
-       if {$parent_count > 1} {
-               error_popup {Can't amend a merge commit.}
-               return
-       }
-
-       if {$parent_count == 0} {
-               set commit_type amend-initial
-               set PARENT {}
-       } elseif {$parent_count == 1} {
-               set commit_type amend
-               set PARENT $parent
+       set HEAD $curHEAD
+       set PARENT $parents
+       set MERGE_HEAD [list]
+       switch -- [llength $parents] {
+       0       {set commit_type amend-initial}
+       1       {set commit_type amend}
+       default {set commit_type amend-merge}
        }
 
        $ui_comm delete 0.0 end
        $ui_comm insert end $msg
-       $ui_comm edit modified false
        $ui_comm edit reset
+       $ui_comm edit modified false
        rescan {set ui_status_value {Ready.}}
 }
 
@@ -727,8 +795,8 @@ proc create_new_commit {} {
 
        set commit_type normal
        $ui_comm delete 0.0 end
-       $ui_comm edit modified false
        $ui_comm edit reset
+       $ui_comm edit modified false
        rescan {set ui_status_value {Ready.}}
 }
 
@@ -760,11 +828,11 @@ proc commit_tree {} {
 
        # -- Our in memory state should match the repository.
        #
-       repository_state curHEAD cur_type
+       repository_state curType curHEAD curMERGE_HEAD
        if {[string match amend* $commit_type]
-               && $cur_type eq {normal}
+               && $curType eq {normal}
                && $curHEAD eq $HEAD} {
-       } elseif {$commit_type ne $cur_type || $HEAD ne $curHEAD} {
+       } elseif {$commit_type ne $curType || $HEAD ne $curHEAD} {
                info_popup {Last scanned state does not match repository state.
 
 Another Git program has modified this repository
@@ -854,14 +922,14 @@ A good commit message has the following format:
 }
 
 proc commit_prehook {curHEAD msg} {
-       global tcl_platform gitdir ui_status_value pch_error
+       global gitdir ui_status_value pch_error
+
+       set pchook [file join $gitdir hooks pre-commit]
 
        # On Cygwin [file executable] might lie so we need to ask
        # the shell if the hook is executable.  Yes that's annoying.
-
-       set pchook [file join $gitdir hooks pre-commit]
-       if {$tcl_platform(platform) eq {windows}
-               && [file isfile $pchook]} {
+       #
+       if {[is_Windows] && [file isfile $pchook]} {
                set pchook [list sh -c [concat \
                        "if test -x \"$pchook\";" \
                        "then exec \"$pchook\" 2>&1;" \
@@ -910,9 +978,10 @@ proc commit_writetree {curHEAD msg} {
 }
 
 proc commit_committree {fd_wt curHEAD msg} {
-       global single_commit gitdir HEAD PARENT commit_type tcl_platform
+       global HEAD PARENT MERGE_HEAD commit_type
+       global single_commit gitdir
        global ui_status_value ui_comm selected_commit_type
-       global file_states selected_paths
+       global file_states selected_paths rescan_active
 
        gets $fd_wt tree_id
        if {$tree_id eq {} || [catch {close $fd_wt} err]} {
@@ -925,24 +994,12 @@ proc commit_committree {fd_wt curHEAD msg} {
        # -- Create the commit.
        #
        set cmd [list git commit-tree $tree_id]
-       if {$PARENT ne {}} {
-               lappend cmd -p $PARENT
-       }
-       if {$commit_type eq {merge}} {
-               if {[catch {
-                               set fd_mh [open [file join $gitdir MERGE_HEAD] r]
-                               while {[gets $fd_mh merge_head] >= 0} {
-                                       lappend cmd -p $merge_head
-                               }
-                               close $fd_mh
-                       } err]} {
-                       error_popup "Loading MERGE_HEAD failed:\n\n$err"
-                       set ui_status_value {Commit failed.}
-                       unlock_index
-                       return
+       set parents [concat $PARENT $MERGE_HEAD]
+       if {[llength $parents] > 0} {
+               foreach p $parents {
+                       lappend cmd -p $p
                }
-       }
-       if {$PARENT eq {}} {
+       } else {
                # git commit-tree writes to stderr during initial commit.
                lappend cmd 2>/dev/null
        }
@@ -990,7 +1047,7 @@ proc commit_committree {fd_wt curHEAD msg} {
        # -- Run the post-commit hook.
        #
        set pchook [file join $gitdir hooks post-commit]
-       if {$tcl_platform(platform) eq {windows} && [file isfile $pchook]} {
+       if {[is_Windows] && [file isfile $pchook]} {
                set pchook [list sh -c [concat \
                        "if test -x \"$pchook\";" \
                        "then exec \"$pchook\";" \
@@ -1003,32 +1060,47 @@ proc commit_committree {fd_wt curHEAD msg} {
        }
 
        $ui_comm delete 0.0 end
-       $ui_comm edit modified false
        $ui_comm edit reset
+       $ui_comm edit modified false
 
        if {$single_commit} do_quit
 
-       # -- Update status without invoking any git commands.
+       # -- Update in memory status
        #
-       set commit_type normal
        set selected_commit_type new
+       set commit_type normal
        set HEAD $cmt_id
        set PARENT $cmt_id
+       set MERGE_HEAD [list]
 
        foreach path [array names file_states] {
                set s $file_states($path)
                set m [lindex $s 0]
                switch -glob -- $m {
-               A? -
-               M? -
-               D? {set m _[string index $m 1]}
-               }
-
-               if {$m eq {__}} {
+               _O -
+               _M -
+               _D {continue}
+               __ -
+               A_ -
+               M_ -
+               DD {
                        unset file_states($path)
                        catch {unset selected_paths($path)}
-               } else {
-                       lset file_states($path) 0 $m
+               }
+               DO {
+                       set file_states($path) [list _O [lindex $s 1] {} {}]
+               }
+               AM -
+               AD -
+               MM -
+               MD -
+               DM {
+                       set file_states($path) [list \
+                               _[string index $m 1] \
+                               [lindex $s 1] \
+                               [lindex $s 3] \
+                               {}]
+               }
                }
        }
 
@@ -1058,13 +1130,15 @@ proc pull_remote {remote branch} {
 
        # -- Our in memory state should match the repository.
        #
-       repository_state curHEAD cur_type
-       if {$commit_type ne $cur_type || $HEAD ne $curHEAD} {
-               error_popup {Last scanned state does not match repository state.
+       repository_state curType curHEAD curMERGE_HEAD
+       if {$commit_type ne $curType || $HEAD ne $curHEAD} {
+               info_popup {Last scanned state does not match repository state.
+
+Another Git program has modified this repository
+since the last scan.  A rescan must be performed
+before a pull operation can be started.
 
-Its highly likely that another Git program modified the
-repository since our last scan.  A rescan is required
-before a pull can be started.
+The rescan will be automatically started now.
 }
                unlock_index
                rescan {set ui_status_value {Ready.}}
@@ -1076,10 +1150,12 @@ before a pull can be started.
        if {[array size file_states] != 0} {
                error_popup {Uncommitted but modified files are present.
 
-You should not perform a pull with unmodified files in your working
-directory as Git would be unable to recover from an incorrect merge.
+You should not perform a pull with unmodified
+files in your working directory as Git will be
+unable to recover from an incorrect merge.
 
-Commit or throw away all changes before starting a pull operation.
+You should commit or revert all changes before
+starting a pull operation.
 }
                unlock_index
                return
@@ -1097,18 +1173,18 @@ Commit or throw away all changes before starting a pull operation.
 }
 
 proc post_pull_remote {remote branch success} {
-       global HEAD PARENT commit_type selected_commit_type
+       global HEAD PARENT MERGE_HEAD commit_type selected_commit_type
        global ui_status_value
 
        unlock_index
        if {$success} {
-               repository_state HEAD commit_type
+               repository_state commit_type HEAD MERGE_HEAD
                set PARENT $HEAD
                set selected_commit_type new
-               set $ui_status_value "Pulling $branch from $remote complete."
+               set ui_status_value "Pulling $branch from $remote complete."
        } else {
-               set m "Conflicts detected while pulling $branch from $remote."
-               rescan "set ui_status_value {$m}"
+               rescan [list set ui_status_value \
+                       "Conflicts detected while pulling $branch from $remote."]
        }
 }
 
@@ -1164,9 +1240,10 @@ proc short_path {path} {
 }
 
 set next_icon_id 0
+set null_sha1 [string repeat 0 40]
 
-proc merge_state {path new_state} {
-       global file_states next_icon_id
+proc merge_state {path new_state {head_info {}} {index_info {}}} {
+       global file_states next_icon_id null_sha1
 
        set s0 [string index $new_state 0]
        set s1 [string index $new_state 1]
@@ -1177,39 +1254,59 @@ proc merge_state {path new_state} {
        } else {
                set state [lindex $info 0]
                set icon [lindex $info 1]
+               if {$head_info eq {}}  {set head_info  [lindex $info 2]}
+               if {$index_info eq {}} {set index_info [lindex $info 3]}
        }
 
-       if {$s0 eq {?}} {
-               set s0 [string index $state 0]
-       } elseif {$s0 eq {_}} {
-               set s0 _
-       }
+       if     {$s0 eq {?}} {set s0 [string index $state 0]} \
+       elseif {$s0 eq {_}} {set s0 _}
 
-       if {$s1 eq {?}} {
-               set s1 [string index $state 1]
-       } elseif {$s1 eq {_}} {
-               set s1 _
+       if     {$s1 eq {?}} {set s1 [string index $state 1]} \
+       elseif {$s1 eq {_}} {set s1 _}
+
+       if {$s0 eq {A} && $s1 eq {_} && $head_info eq {}} {
+               set head_info [list 0 $null_sha1]
+       } elseif {$s0 ne {_} && [string index $state 0] eq {_}
+               && $head_info eq {}} {
+               set head_info $index_info
        }
 
-       set file_states($path) [list $s0$s1 $icon]
+       set file_states($path) [list $s0$s1 $icon \
+               $head_info $index_info \
+               ]
        return $state
 }
 
 proc display_file {path state} {
-       global file_states file_lists selected_paths rescan_active
+       global file_states file_lists selected_paths
 
        set old_m [merge_state $path $state]
-       if {$rescan_active > 0} return
-
        set s $file_states($path)
        set new_m [lindex $s 0]
        set new_w [mapcol $new_m $path] 
        set old_w [mapcol $old_m $path]
        set new_icon [mapicon $new_m $path]
 
+       if {$new_m eq {__}} {
+               set lno [lsearch -sorted $file_lists($old_w) $path]
+               if {$lno >= 0} {
+                       set file_lists($old_w) \
+                               [lreplace $file_lists($old_w) $lno $lno]
+                       incr lno
+                       $old_w conf -state normal
+                       $old_w delete $lno.0 [expr {$lno + 1}].0
+                       $old_w conf -state disabled
+               }
+               unset file_states($path)
+               catch {unset selected_paths($path)}
+               return
+       }
+
        if {$new_w ne $old_w} {
                set lno [lsearch -sorted $file_lists($old_w) $path]
                if {$lno >= 0} {
+                       set file_lists($old_w) \
+                               [lreplace $file_lists($old_w) $lno $lno]
                        incr lno
                        $old_w conf -state normal
                        $old_w delete $lno.0 [expr {$lno + 1}].0
@@ -1278,6 +1375,80 @@ proc display_all_files {} {
        $ui_other conf -state disabled
 }
 
+proc update_indexinfo {msg pathList after} {
+       global update_index_cp ui_status_value
+
+       if {![lock_index update]} return
+
+       set update_index_cp 0
+       set pathList [lsort $pathList]
+       set totalCnt [llength $pathList]
+       set batch [expr {int($totalCnt * .01) + 1}]
+       if {$batch > 25} {set batch 25}
+
+       set ui_status_value [format \
+               "$msg... %i/%i files (%.2f%%)" \
+               $update_index_cp \
+               $totalCnt \
+               0.0]
+       set fd [open "| git update-index -z --index-info" w]
+       fconfigure $fd \
+               -blocking 0 \
+               -buffering full \
+               -buffersize 512 \
+               -translation binary
+       fileevent $fd writable [list \
+               write_update_indexinfo \
+               $fd \
+               $pathList \
+               $totalCnt \
+               $batch \
+               $msg \
+               $after \
+               ]
+}
+
+proc write_update_indexinfo {fd pathList totalCnt batch msg after} {
+       global update_index_cp ui_status_value
+       global file_states current_diff
+
+       if {$update_index_cp >= $totalCnt} {
+               close $fd
+               unlock_index
+               uplevel #0 $after
+               return
+       }
+
+       for {set i $batch} \
+               {$update_index_cp < $totalCnt && $i > 0} \
+               {incr i -1} {
+               set path [lindex $pathList $update_index_cp]
+               incr update_index_cp
+
+               set s $file_states($path)
+               switch -glob -- [lindex $s 0] {
+               A? {set new _O}
+               M? {set new _M}
+               D? {set new _?}
+               ?? {continue}
+               }
+               set info [lindex $s 2]
+               if {$info eq {}} continue
+
+               puts -nonewline $fd $info
+               puts -nonewline $fd "\t"
+               puts -nonewline $fd $path
+               puts -nonewline $fd "\0"
+               display_file $path $new
+       }
+
+       set ui_status_value [format \
+               "$msg... %i/%i files (%.2f%%)" \
+               $update_index_cp \
+               $totalCnt \
+               [expr {100.0 * $update_index_cp / $totalCnt}]]
+}
+
 proc update_index {msg pathList after} {
        global update_index_cp ui_status_value
 
@@ -1331,29 +1502,201 @@ proc write_update_index {fd pathList totalCnt batch msg after} {
                switch -glob -- [lindex $file_states($path) 0] {
                AD -
                MD -
-               _D {set new D_}
+               UD -
+               _D {set new DD}
 
                _M -
                MM -
+               UM -
+               U_ -
                M_ {set new M_}
 
                _O -
                AM -
                A_ {set new A_}
 
-               ?? {continue}
-               }
+               ?? {continue}
+               }
+
+               puts -nonewline $fd $path
+               puts -nonewline $fd "\0"
+               display_file $path $new
+       }
+
+       set ui_status_value [format \
+               "$msg... %i/%i files (%.2f%%)" \
+               $update_index_cp \
+               $totalCnt \
+               [expr {100.0 * $update_index_cp / $totalCnt}]]
+}
+
+proc checkout_index {msg pathList after} {
+       global update_index_cp ui_status_value
+
+       if {![lock_index update]} return
+
+       set update_index_cp 0
+       set pathList [lsort $pathList]
+       set totalCnt [llength $pathList]
+       set batch [expr {int($totalCnt * .01) + 1}]
+       if {$batch > 25} {set batch 25}
+
+       set ui_status_value [format \
+               "$msg... %i/%i files (%.2f%%)" \
+               $update_index_cp \
+               $totalCnt \
+               0.0]
+       set cmd [list git checkout-index]
+       lappend cmd --index
+       lappend cmd --quiet
+       lappend cmd --force
+       lappend cmd -z
+       lappend cmd --stdin
+       set fd [open "| $cmd " w]
+       fconfigure $fd \
+               -blocking 0 \
+               -buffering full \
+               -buffersize 512 \
+               -translation binary
+       fileevent $fd writable [list \
+               write_checkout_index \
+               $fd \
+               $pathList \
+               $totalCnt \
+               $batch \
+               $msg \
+               $after \
+               ]
+}
+
+proc write_checkout_index {fd pathList totalCnt batch msg after} {
+       global update_index_cp ui_status_value
+       global file_states current_diff
+
+       if {$update_index_cp >= $totalCnt} {
+               close $fd
+               unlock_index
+               uplevel #0 $after
+               return
+       }
+
+       for {set i $batch} \
+               {$update_index_cp < $totalCnt && $i > 0} \
+               {incr i -1} {
+               set path [lindex $pathList $update_index_cp]
+               incr update_index_cp
+
+               switch -glob -- [lindex $file_states($path) 0] {
+               AM -
+               AD {set new A_}
+               MM -
+               MD {set new M_}
+               _M -
+               _D {set new __}
+               ?? {continue}
+               }
+
+               puts -nonewline $fd $path
+               puts -nonewline $fd "\0"
+               display_file $path $new
+       }
+
+       set ui_status_value [format \
+               "$msg... %i/%i files (%.2f%%)" \
+               $update_index_cp \
+               $totalCnt \
+               [expr {100.0 * $update_index_cp / $totalCnt}]]
+}
+
+######################################################################
+##
+## branch management
+
+proc load_all_heads {} {
+       global all_heads tracking_branches
+
+       set all_heads [list]
+       set cmd [list git for-each-ref]
+       lappend cmd --format=%(refname)
+       lappend cmd refs/heads
+       set fd [open "| $cmd" r]
+       while {[gets $fd line] > 0} {
+               if {![catch {set info $tracking_branches($line)}]} continue
+               if {![regsub ^refs/heads/ $line {} name]} continue
+               lappend all_heads $name
+       }
+       close $fd
+
+       set all_heads [lsort $all_heads]
+}
+
+proc populate_branch_menu {m} {
+       global all_heads disable_on_lock
+
+       $m add separator
+       foreach b $all_heads {
+               $m add radiobutton \
+                       -label $b \
+                       -command [list switch_branch $b] \
+                       -variable current_branch \
+                       -value $b \
+                       -font font_ui
+               lappend disable_on_lock \
+                       [list $m entryconf [$m index last] -state]
+       }
+}
+
+proc do_create_branch {} {
+       error "NOT IMPLEMENTED"
+}
+
+proc do_delete_branch {} {
+       error "NOT IMPLEMENTED"
+}
+
+proc switch_branch {b} {
+       global HEAD commit_type file_states current_branch
+       global selected_commit_type ui_comm
+
+       if {![lock_index switch]} return
+
+       # -- Backup the selected branch (repository_state resets it)
+       #
+       set new_branch $current_branch
+
+       # -- Our in memory state should match the repository.
+       #
+       repository_state curType curHEAD curMERGE_HEAD
+       if {[string match amend* $commit_type]
+               && $curType eq {normal}
+               && $curHEAD eq $HEAD} {
+       } elseif {$commit_type ne $curType || $HEAD ne $curHEAD} {
+               info_popup {Last scanned state does not match repository state.
+
+Another Git program has modified this repository
+since the last scan.  A rescan must be performed
+before the current branch can be changed.
+
+The rescan will be automatically started now.
+}
+               unlock_index
+               rescan {set ui_status_value {Ready.}}
+               return
+       }
 
-               puts -nonewline $fd $path
-               puts -nonewline $fd "\0"
-               display_file $path $new
+       # -- Toss the message buffer if we are in amend mode.
+       #
+       if {[string match amend* $curType]} {
+               $ui_comm delete 0.0 end
+               $ui_comm edit reset
+               $ui_comm edit modified false
        }
 
-       set ui_status_value [format \
-               "$msg... %i/%i files (%.2f%%)" \
-               $update_index_cp \
-               $totalCnt \
-               [expr {100.0 * $update_index_cp / $totalCnt}]]
+       set selected_commit_type new
+       set current_branch $new_branch
+
+       unlock_index
+       error "NOT FINISHED"
 }
 
 ######################################################################
@@ -1361,21 +1704,49 @@ proc write_update_index {fd pathList totalCnt batch msg after} {
 ## remote management
 
 proc load_all_remotes {} {
-       global gitdir all_remotes repo_config
+       global gitdir repo_config
+       global all_remotes tracking_branches
 
        set all_remotes [list]
+       array unset tracking_branches
+
        set rm_dir [file join $gitdir remotes]
        if {[file isdirectory $rm_dir]} {
-               set all_remotes [concat $all_remotes [glob \
+               set all_remotes [glob \
                        -types f \
                        -tails \
                        -nocomplain \
-                       -directory $rm_dir *]]
+                       -directory $rm_dir *]
+
+               foreach name $all_remotes {
+                       catch {
+                               set fd [open [file join $rm_dir $name] r]
+                               while {[gets $fd line] >= 0} {
+                                       if {![regexp {^Pull:[   ]*([^:]+):(.+)$} \
+                                               $line line src dst]} continue
+                                       if {![regexp ^refs/ $dst]} {
+                                               set dst "refs/heads/$dst"
+                                       }
+                                       set tracking_branches($dst) [list $name $src]
+                               }
+                               close $fd
+                       }
+               }
        }
 
        foreach line [array names repo_config remote.*.url] {
-               if {[regexp ^remote\.(.*)\.url\$ $line line name]} {
-                       lappend all_remotes $name
+               if {![regexp ^remote\.(.*)\.url\$ $line line name]} continue
+               lappend all_remotes $name
+
+               if {[catch {set fl $repo_config(remote.$name.fetch)}]} {
+                       set fl {}
+               }
+               foreach line $fl {
+                       if {![regexp {^([^:]+):(.+)$} $line line src dst]} continue
+                       if {![regexp ^refs/ $dst]} {
+                               set dst "refs/heads/$dst"
+                       }
+                       set tracking_branches($dst) [list $name $src]
                }
        }
 
@@ -1564,17 +1935,19 @@ foreach i {
                {_M i mod      "Modified"}
                {M_ i fulltick "Included in commit"}
                {MM i parttick "Partially included"}
+               {MD i question "Included (but gone)"}
 
                {_O o plain    "Untracked"}
                {A_ o fulltick "Added by commit"}
                {AM o parttick "Partially added"}
-               {AD o question "Added (but now gone)"}
+               {AD o question "Added (but gone)"}
 
                {_D i question "Missing"}
-               {D_ i removed  "Removed by commit"}
                {DD i removed  "Removed by commit"}
                {DO i removed  "Removed (still exists)"}
+               {DM i removed  "Removed (but modified)"}
 
+               {UD i merge    "Merge conflicts"}
                {UM i merge    "Merge conflicts"}
                {U_ i merge    "Merge conflicts"}
        } {
@@ -1597,9 +1970,15 @@ unset filemask i
 
 proc is_MacOSX {} {
        global tcl_platform tk_library
-       if {$tcl_platform(platform) eq {unix}
-               && $tcl_platform(os) eq {Darwin}
-               && [string match /Library/Frameworks/* $tk_library]} {
+       if {[tk windowingsystem] eq {aqua}} {
+               return 1
+       }
+       return 0
+}
+
+proc is_Windows {} {
+       global tcl_platform
+       if {$tcl_platform(platform) eq {windows}} {
                return 1
        }
        return 0
@@ -1735,12 +2114,10 @@ proc console_init {w} {
 }
 
 proc console_exec {w cmd {after {}}} {
-       global tcl_platform
-
        # -- Windows tosses the enviroment when we exec our child.
        #    But most users need that so we have to relogin. :-(
        #
-       if {$tcl_platform(platform) eq {windows}} {
+       if {[is_Windows]} {
                set cmd [list sh --login -c "cd \"[pwd]\" && [join $cmd { }]"]
        }
 
@@ -1814,35 +2191,54 @@ proc console_read {w fd after} {
 
 set starting_gitk_msg {Please wait... Starting gitk...}
 
-proc do_gitk {} {
-       global tcl_platform ui_status_value starting_gitk_msg
+proc do_gitk {revs} {
+       global ui_status_value starting_gitk_msg
 
-       set ui_status_value $starting_gitk_msg
-       after 10000 {
-               if {$ui_status_value eq $starting_gitk_msg} {
-                       set ui_status_value {Ready.}
-               }
+       set cmd gitk
+       if {$revs ne {}} {
+               append cmd { }
+               append cmd $revs
+       }
+       if {[is_Windows]} {
+               set cmd "sh -c \"exec $cmd\""
        }
+       append cmd { &}
 
-       if {$tcl_platform(platform) eq {windows}} {
-               exec sh -c gitk &
+       if {[catch {eval exec $cmd} err]} {
+               error_popup "Failed to start gitk:\n\n$err"
        } else {
-               exec gitk &
+               set ui_status_value $starting_gitk_msg
+               after 10000 {
+                       if {$ui_status_value eq $starting_gitk_msg} {
+                               set ui_status_value {Ready.}
+                       }
+               }
        }
 }
 
 proc do_repack {} {
-       set w [new_console "repack" "Repacking the object database"]
+       set w [new_console {repack} \
+               {Repacking the object database}]
        set cmd [list git repack]
        lappend cmd -a
        lappend cmd -d
        console_exec $w $cmd
 }
 
+proc do_fsck_objects {} {
+       set w [new_console {fsck-objects} \
+               {Verifying the object database with fsck-objects}]
+       set cmd [list git fsck-objects]
+       lappend cmd --full
+       lappend cmd --cache
+       lappend cmd --strict
+       console_exec $w $cmd
+}
+
 set is_quitting 0
 
 proc do_quit {} {
-       global gitdir ui_comm is_quitting repo_config
+       global gitdir ui_comm is_quitting repo_config commit_type
 
        if {$is_quitting} return
        set is_quitting 1
@@ -1851,14 +2247,16 @@ proc do_quit {} {
        #
        set save [file join $gitdir GITGUI_MSG]
        set msg [string trim [$ui_comm get 0.0 end]]
-       if {[$ui_comm edit modified] && $msg ne {}} {
+       if {![string match amend* $commit_type]
+               && [$ui_comm edit modified]
+               && $msg ne {}} {
                catch {
                        set fd [open $save w]
                        puts $fd [string trim [$ui_comm get 0.0 end]]
                        close $fd
                }
-       } elseif {$msg eq {} && [file exists $save]} {
-               file delete $save
+       } else {
+               catch {file delete $save}
        }
 
        # -- Stash our current window geometry into this repository.
@@ -1881,6 +2279,49 @@ proc do_rescan {} {
        rescan {set ui_status_value {Ready.}}
 }
 
+proc remove_helper {txt paths} {
+       global file_states current_diff
+
+       if {![lock_index begin-update]} return
+
+       set pathList [list]
+       set after {}
+       foreach path $paths {
+               switch -glob -- [lindex $file_states($path) 0] {
+               A? -
+               M? -
+               D? {
+                       lappend pathList $path
+                       if {$path eq $current_diff} {
+                               set after {reshow_diff;}
+                       }
+               }
+               }
+       }
+       if {$pathList eq {}} {
+               unlock_index
+       } else {
+               update_indexinfo \
+                       $txt \
+                       $pathList \
+                       [concat $after {set ui_status_value {Ready.}}]
+       }
+}
+
+proc do_remove_selection {} {
+       global current_diff selected_paths
+
+       if {[array size selected_paths] > 0} {
+               remove_helper \
+                       {Removing selected files from commit} \
+                       [array names selected_paths]
+       } elseif {$current_diff ne {}} {
+               remove_helper \
+                       "Removing [short_path $current_diff] from commit" \
+                       [list $current_diff]
+       }
+}
+
 proc include_helper {txt paths} {
        global file_states current_diff
 
@@ -1889,12 +2330,12 @@ proc include_helper {txt paths} {
        set pathList [list]
        set after {}
        foreach path $paths {
-               switch -- [lindex $file_states($path) 0] {
+               switch -glob -- [lindex $file_states($path) 0] {
                AM -
                AD -
                MM -
-               UM -
-               U_ -
+               MD -
+               U? -
                _M -
                _D -
                _O {
@@ -1938,6 +2379,7 @@ proc do_include_all {} {
                AM -
                AD -
                MM -
+               MD -
                _M -
                _D {lappend paths $path}
                }
@@ -1947,6 +2389,79 @@ proc do_include_all {} {
                $paths
 }
 
+proc revert_helper {txt paths} {
+       global gitdir appname
+       global file_states current_diff
+
+       if {![lock_index begin-update]} return
+
+       set pathList [list]
+       set after {}
+       foreach path $paths {
+               switch -glob -- [lindex $file_states($path) 0] {
+               AM -
+               AD -
+               MM -
+               MD -
+               _M -
+               _D {
+                       lappend pathList $path
+                       if {$path eq $current_diff} {
+                               set after {reshow_diff;}
+                       }
+               }
+               }
+       }
+
+       set n [llength $pathList]
+       if {$n == 0} {
+               unlock_index
+               return
+       } elseif {$n == 1} {
+               set s "[short_path [lindex $pathList]]"
+       } else {
+               set s "these $n files"
+       }
+
+       set reponame [lindex [file split \
+               [file normalize [file dirname $gitdir]]] \
+               end]
+
+       set reply [tk_dialog \
+               .confirm_revert \
+               "$appname ($reponame)" \
+               "Revert unincluded changes in $s?
+
+Any unincluded changes will be permanently lost by the revert." \
+               question \
+               1 \
+               {Do Nothing} \
+               {Revert Changes} \
+               ]
+       if {$reply == 1} {
+               checkout_index \
+                       $txt \
+                       $pathList \
+                       [concat $after {set ui_status_value {Ready.}}]
+       } else {
+               unlock_index
+       }
+}
+
+proc do_revert_selection {} {
+       global current_diff selected_paths
+
+       if {[array size selected_paths] > 0} {
+               revert_helper \
+                       {Reverting selected files} \
+                       [array names selected_paths]
+       } elseif {$current_diff ne {}} {
+               revert_helper \
+                       "Reverting [short_path $current_diff]" \
+                       [list $current_diff]
+       }
+}
+
 proc do_signoff {} {
        global ui_comm
 
@@ -1989,6 +2504,61 @@ proc do_commit {} {
        commit_tree
 }
 
+proc do_about {} {
+       global appname copyright
+       global tcl_patchLevel tk_patchLevel
+
+       set w .about_dialog
+       toplevel $w
+       wm geometry $w "+[winfo rootx .]+[winfo rooty .]"
+
+       label $w.header -text "About $appname" \
+               -font font_uibold
+       pack $w.header -side top -fill x
+
+       frame $w.buttons
+       button $w.buttons.close -text {Close} \
+               -font font_ui \
+               -command [list destroy $w]
+       pack $w.buttons.close -side right
+       pack $w.buttons -side bottom -fill x -pady 10 -padx 10
+
+       label $w.desc \
+               -text "$appname - a commit creation tool for Git.
+$copyright" \
+               -padx 5 -pady 5 \
+               -justify left \
+               -anchor w \
+               -borderwidth 1 \
+               -relief solid \
+               -font font_ui
+       pack $w.desc -side top -fill x -padx 5 -pady 5
+
+       set v [exec git --version]
+       append v "\n\n"
+       if {$tcl_patchLevel eq $tk_patchLevel} {
+               append v "Tcl/Tk version $tcl_patchLevel"
+       } else {
+               append v "Tcl version $tcl_patchLevel"
+               append v ", Tk version $tk_patchLevel"
+       }
+
+       label $w.vers \
+               -text $v \
+               -padx 5 -pady 5 \
+               -justify left \
+               -anchor w \
+               -borderwidth 1 \
+               -relief solid \
+               -font font_ui
+       pack $w.vers -side top -fill x -padx 5 -pady 5
+
+       bind $w <Visibility> "grab $w; focus $w"
+       bind $w <Key-Escape> "destroy $w"
+       wm title $w "About $appname"
+       tkwait window $w
+}
+
 proc do_options {} {
        global appname gitdir font_descs
        global repo_config global_config
@@ -2266,7 +2836,7 @@ proc do_macosx_app {} {
 }
 
 proc toggle_or_diff {w x y} {
-       global file_lists current_diff ui_index ui_other
+       global file_states file_lists current_diff ui_index ui_other
        global last_clicked selected_paths
 
        set pos [split [$w index @$x,$y] .]
@@ -2289,10 +2859,24 @@ proc toggle_or_diff {w x y} {
                } else {
                        set after {}
                }
-               update_index \
-                       "Including [short_path $path]" \
-                       [list $path] \
-                       [concat $after {set ui_status_value {Ready.}}]
+               switch -glob -- [lindex $file_states($path) 0] {
+               A_ -
+               M_ -
+               DD -
+               DO -
+               DM {
+                       update_indexinfo \
+                               "Removing [short_path $path] from commit" \
+                               [list $path] \
+                               [concat $after {set ui_status_value {Ready.}}]
+               }
+               ?? {
+                       update_index \
+                               "Including [short_path $path]" \
+                               [list $path] \
+                               [concat $after {set ui_status_value {Ready.}}]
+               }
+               }
        } else {
                show_diff $path $w $lno
        }
@@ -2368,14 +2952,15 @@ catch {
 font create font_uibold
 font create font_diffbold
 
-set M1B M1
-set M1T M1
-if {$tcl_platform(platform) eq {windows}} {
+if {[is_Windows]} {
        set M1B Control
        set M1T Ctrl
 } elseif {[is_MacOSX]} {
        set M1B M1
        set M1T Cmd
+} else {
+       set M1B M1
+       set M1T M1
 }
 
 proc apply_config {} {
@@ -2418,8 +3003,11 @@ apply_config
 # -- Menu Bar
 #
 menu .mbar -tearoff 0
-.mbar add cascade -label Project -menu .mbar.project
+.mbar add cascade -label Repository -menu .mbar.repository
 .mbar add cascade -label Edit -menu .mbar.edit
+if {!$single_commit} {
+       .mbar add cascade -label Branch -menu .mbar.branch
+}
 .mbar add cascade -label Commit -menu .mbar.commit
 if {!$single_commit} {
        .mbar add cascade -label Fetch -menu .mbar.fetch
@@ -2428,30 +3016,46 @@ if {!$single_commit} {
 }
 . configure -menu .mbar
 
-# -- Project Menu
+# -- Repository Menu
 #
-menu .mbar.project
-.mbar.project add command -label Visualize \
-       -command do_gitk \
+menu .mbar.repository
+.mbar.repository add command \
+       -label {Visualize Current Branch} \
+       -command {do_gitk {}} \
        -font font_ui
+if {![is_MacOSX]} {
+       .mbar.repository add command \
+               -label {Visualize All Branches} \
+               -command {do_gitk {--all}} \
+               -font font_ui
+}
+.mbar.repository add separator
+
 if {!$single_commit} {
-       .mbar.project add command -label {Repack Database} \
+       .mbar.repository add command -label {Repack Database} \
                -command do_repack \
                -font font_ui
 
-       if {$tcl_platform(platform) eq {windows}} {
-               .mbar.project add command \
+       .mbar.repository add command -label {Verify Database} \
+               -command do_fsck_objects \
+               -font font_ui
+
+       .mbar.repository add separator
+
+       if {[is_Windows]} {
+               .mbar.repository add command \
                        -label {Create Desktop Icon} \
                        -command do_windows_shortcut \
                        -font font_ui
        } elseif {[is_MacOSX]} {
-               .mbar.project add command \
+               .mbar.repository add command \
                        -label {Create Desktop Icon} \
                        -command do_macosx_app \
                        -font font_ui
        }
 }
-.mbar.project add command -label Quit \
+
+.mbar.repository add command -label Quit \
        -command do_quit \
        -accelerator $M1T-Q \
        -font font_ui
@@ -2489,10 +3093,24 @@ menu .mbar.edit
        -command {catch {[focus] tag add sel 0.0 end}} \
        -accelerator $M1T-A \
        -font font_ui
-.mbar.edit add separator
-.mbar.edit add command -label {Options...} \
-       -command do_options \
-       -font font_ui
+
+# -- Branch Menu
+#
+if {!$single_commit} {
+       menu .mbar.branch
+
+       .mbar.branch add command -label {Create...} \
+               -command do_create_branch \
+               -font font_ui
+       lappend disable_on_lock [list .mbar.branch entryconf \
+               [.mbar.branch index last] -state]
+
+       .mbar.branch add command -label {Delete...} \
+               -command do_delete_branch \
+               -font font_ui
+       lappend disable_on_lock [list .mbar.branch entryconf \
+               [.mbar.branch index last] -state]
+}
 
 # -- Commit Menu
 #
@@ -2525,19 +3143,33 @@ lappend disable_on_lock \
 lappend disable_on_lock \
        [list .mbar.commit entryconf [.mbar.commit index last] -state]
 
-.mbar.commit add command -label {Include Selected Files} \
+.mbar.commit add command -label {Add To Commit} \
        -command do_include_selection \
        -font font_ui
 lappend disable_on_lock \
        [list .mbar.commit entryconf [.mbar.commit index last] -state]
 
-.mbar.commit add command -label {Include All Files} \
+.mbar.commit add command -label {Add All To Commit} \
        -command do_include_all \
        -accelerator $M1T-I \
        -font font_ui
 lappend disable_on_lock \
        [list .mbar.commit entryconf [.mbar.commit index last] -state]
 
+.mbar.commit add command -label {Remove From Commit} \
+       -command do_remove_selection \
+       -font font_ui
+lappend disable_on_lock \
+       [list .mbar.commit entryconf [.mbar.commit index last] -state]
+
+.mbar.commit add command -label {Revert Changes} \
+       -command do_revert_selection \
+       -font font_ui
+lappend disable_on_lock \
+       [list .mbar.commit entryconf [.mbar.commit index last] -state]
+
+.mbar.commit add separator
+
 .mbar.commit add command -label {Sign Off} \
        -command do_signoff \
        -accelerator $M1T-S \
@@ -2558,6 +3190,56 @@ if {!$single_commit} {
        menu .mbar.push
 }
 
+if {[is_MacOSX]} {
+       # -- Apple Menu (Mac OS X only)
+       #
+       .mbar add cascade -label Apple -menu .mbar.apple
+       menu .mbar.apple
+
+       .mbar.apple add command -label "About $appname" \
+               -command do_about \
+               -font font_ui
+       .mbar.apple add command -label "$appname Options..." \
+               -command do_options \
+               -font font_ui
+} else {
+       # -- Edit Menu
+       #
+       .mbar.edit add separator
+       .mbar.edit add command -label {Options...} \
+               -command do_options \
+               -font font_ui
+
+       # -- Help Menu
+       #
+       .mbar add cascade -label Help -menu .mbar.help
+       menu .mbar.help
+
+       .mbar.help add command -label "About $appname" \
+               -command do_about \
+               -font font_ui
+}
+
+
+# -- Branch Control
+#
+frame .branch \
+       -borderwidth 1 \
+       -relief sunken
+label .branch.l1 \
+       -text {Current Branch:} \
+       -anchor w \
+       -justify left \
+       -font font_ui
+label .branch.cb \
+       -textvariable current_branch \
+       -anchor w \
+       -justify left \
+       -font font_ui
+pack .branch.l1 -side left
+pack .branch.cb -side left -fill x
+pack .branch -side top -fill x
+
 # -- Main Window Layout
 #
 panedwindow .vpane -orient vertical
@@ -2635,7 +3317,7 @@ pack .vpane.lower.commarea.buttons.rescan -side top -fill x
 lappend disable_on_lock \
        {.vpane.lower.commarea.buttons.rescan conf -state}
 
-button .vpane.lower.commarea.buttons.incall -text {Include All} \
+button .vpane.lower.commarea.buttons.incall -text {Add All} \
        -command do_include_all \
        -font font_ui
 pack .vpane.lower.commarea.buttons.incall -side top -fill x
@@ -2686,6 +3368,7 @@ proc trace_commit_type {varname args} {
        initial       {set txt {Initial Commit Message:}}
        amend         {set txt {Amended Commit Message:}}
        amend-initial {set txt {Amended Initial Commit Message:}}
+       amend-merge   {set txt {Amended Merge Commit Message:}}
        merge         {set txt {Merge Commit Message:}}
        *             {set txt {Commit Message:}}
        }
@@ -2834,12 +3517,17 @@ pack $ui_diff -side left -fill both -expand 1
 pack .vpane.lower.diff.header -side top -fill x
 pack .vpane.lower.diff.body -side bottom -fill both -expand 1
 
-$ui_diff tag conf dm -foreground red
-$ui_diff tag conf dp -foreground blue
-$ui_diff tag conf di -foreground {#00a000}
-$ui_diff tag conf dni -foreground {#a000a0}
-$ui_diff tag conf da -font font_diffbold
-$ui_diff tag conf bold -font font_diffbold
+$ui_diff tag conf d_@ -font font_diffbold
+$ui_diff tag conf d_+  -foreground blue
+$ui_diff tag conf d_-  -foreground red
+$ui_diff tag conf d_++ -foreground {#00a000}
+$ui_diff tag conf d_-- -foreground {#a000a0}
+$ui_diff tag conf d_+- \
+       -foreground red \
+       -background {light goldenrod yellow}
+$ui_diff tag conf d_-+ \
+       -foreground blue \
+       -background azure2
 
 # -- Diff Body Context Menu
 #
@@ -2975,18 +3663,82 @@ set file_lists($ui_other) [list]
 
 set HEAD {}
 set PARENT {}
+set MERGE_HEAD [list]
 set commit_type {}
 set empty_tree {}
+set current_branch {}
 set current_diff {}
 set selected_commit_type new
 
 wm title . "$appname ([file normalize [file dirname $gitdir]])"
 focus -force $ui_comm
+
+# -- Warn the user about environmental problems.  Cygwin's Tcl
+#    does *not* pass its env array onto any processes it spawns.
+#    This means that git processes get none of our environment.
+#
+if {[is_Windows]} {
+       set ignored_env 0
+       set suggest_user {}
+       set msg "Possible environment issues exist.
+
+The following environment variables are probably
+going to be ignored by any Git subprocess run
+by $appname:
+
+"
+       foreach name [array names env] {
+               switch -regexp -- $name {
+               {^GIT_INDEX_FILE$} -
+               {^GIT_OBJECT_DIRECTORY$} -
+               {^GIT_ALTERNATE_OBJECT_DIRECTORIES$} -
+               {^GIT_DIFF_OPTS$} -
+               {^GIT_EXTERNAL_DIFF$} -
+               {^GIT_PAGER$} -
+               {^GIT_TRACE$} -
+               {^GIT_CONFIG$} -
+               {^GIT_CONFIG_LOCAL$} -
+               {^GIT_(AUTHOR|COMMITTER)_DATE$} {
+                       append msg " - $name\n"
+                       incr ignored_env
+               }
+               {^GIT_(AUTHOR|COMMITTER)_(NAME|EMAIL)$} {
+                       append msg " - $name\n"
+                       incr ignored_env
+                       set suggest_user $name
+               }
+               }
+       }
+       if {$ignored_env > 0} {
+               append msg "
+This is due to a known issue with the
+Tcl binary distributed by Cygwin."
+
+               if {$suggest_user ne {}} {
+                       append msg "
+
+A good replacement for $suggest_user
+is placing values for the user.name and
+user.email settings into your personal
+~/.gitconfig file.
+"
+               }
+               warn_popup $msg
+       }
+       unset ignored_env msg suggest_user name
+}
+
+# -- Only initialize complex UI if we are going to stay running.
+#
 if {!$single_commit} {
        load_all_remotes
+       load_all_heads
+
+       populate_branch_menu .mbar.branch
        populate_fetch_menu .mbar.fetch
        populate_pull_menu .mbar.pull
        populate_push_menu .mbar.push
 }
+
 lock_index begin-read
 after 1 do_rescan