git-gui: Pad the database statistics dialog window.
[gitweb.git] / git-gui.sh
index 37757cfb6522a69b2e3150b8a81eab284b9f874d..9dbe1156c51ca2a9752871bb102dde240c8cb497 100755 (executable)
@@ -410,9 +410,9 @@ proc rescan_stage2 {fd after} {
        set fd_df [open "| git diff-files -z" r]
        set fd_lo [open $ls_others r]
 
-       fconfigure $fd_di -blocking 0 -translation binary
-       fconfigure $fd_df -blocking 0 -translation binary
-       fconfigure $fd_lo -blocking 0 -translation binary
+       fconfigure $fd_di -blocking 0 -translation binary -encoding binary
+       fconfigure $fd_df -blocking 0 -translation binary -encoding binary
+       fconfigure $fd_lo -blocking 0 -translation binary -encoding binary
        fileevent $fd_di readable [list read_diff_index $fd_di $after]
        fileevent $fd_df readable [list read_diff_files $fd_df $after]
        fileevent $fd_lo readable [list read_ls_others $fd_lo $after]
@@ -428,6 +428,7 @@ proc load_message {file} {
                }
                set content [string trim [read $fd]]
                close $fd
+               regsub -all -line {[ \r\t]+$} $content {} content
                $ui_comm delete 0.0 end
                $ui_comm insert end $content
                return 1
@@ -450,8 +451,9 @@ proc read_diff_index {fd after} {
 
                incr c
                set i [split [string range $buf_rdi $c [expr {$z1 - 2}]] { }]
+               set p [string range $buf_rdi $z1 [expr {$z2 - 1}]]
                merge_state \
-                       [string range $buf_rdi $z1 [expr {$z2 - 1}]] \
+                       [encoding convertfrom $p] \
                        [lindex $i 4]? \
                        [list [lindex $i 0] [lindex $i 2]] \
                        [list]
@@ -482,8 +484,9 @@ proc read_diff_files {fd after} {
 
                incr c
                set i [split [string range $buf_rdf $c [expr {$z1 - 2}]] { }]
+               set p [string range $buf_rdf $z1 [expr {$z2 - 1}]]
                merge_state \
-                       [string range $buf_rdf $z1 [expr {$z2 - 1}]] \
+                       [encoding convertfrom $p] \
                        ?[lindex $i 4] \
                        [list] \
                        [list [lindex $i 0] [lindex $i 2]]
@@ -506,7 +509,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] {
-               merge_state $p ?O
+               merge_state [encoding convertfrom $p] ?O
        }
        rescan_done $fd buf_rlo $after
 }
@@ -543,13 +546,15 @@ proc prune_selection {} {
 ## diff
 
 proc clear_diff {} {
-       global ui_diff current_diff_path ui_index ui_workdir
+       global ui_diff current_diff_path current_diff_header
+       global ui_index ui_workdir
 
        $ui_diff conf -state normal
        $ui_diff delete 0.0 end
        $ui_diff conf -state disabled
 
        set current_diff_path {}
+       set current_diff_header {}
 
        $ui_index tag remove in_diff 0.0 end
        $ui_workdir tag remove in_diff 0.0 end
@@ -597,7 +602,7 @@ proc show_diff {path w {lno {}}} {
        global file_states file_lists
        global is_3way_diff diff_active repo_config
        global ui_diff ui_status_value ui_index ui_workdir
-       global current_diff_path current_diff_side
+       global current_diff_path current_diff_side current_diff_header
 
        if {$diff_active || ![lock_index read]} return
 
@@ -621,6 +626,7 @@ proc show_diff {path w {lno {}}} {
        set diff_active 1
        set current_diff_path $path
        set current_diff_side $w
+       set current_diff_header {}
        set ui_status_value "Loading diff of [escape_path $path]..."
 
        # - Git won't give us the diff, there's nothing to compare to!
@@ -657,7 +663,6 @@ proc show_diff {path w {lno {}}} {
                                $ui_diff insert end \
 "* Untracked file is $sz bytes.
 * Showing only first $max_sz bytes.
-
 " d_@
                        }
                        $ui_diff insert end $content
@@ -706,22 +711,30 @@ proc show_diff {path w {lno {}}} {
                return
        }
 
-       fconfigure $fd -blocking 0 -translation auto
+       fconfigure $fd \
+               -blocking 0 \
+               -encoding binary \
+               -translation binary
        fileevent $fd readable [list read_diff $fd]
 }
 
 proc read_diff {fd} {
-       global ui_diff ui_status_value is_3way_diff diff_active
+       global ui_diff ui_status_value diff_active
+       global is_3way_diff current_diff_header
 
        $ui_diff conf -state normal
        while {[gets $fd line] >= 0} {
                # -- Cleanup uninteresting diff header lines.
                #
-               if {[string match {diff --git *}      $line]} continue
-               if {[string match {diff --cc *}       $line]} continue
-               if {[string match {diff --combined *} $line]} continue
-               if {[string match {--- *}             $line]} continue
-               if {[string match {+++ *}             $line]} continue
+               if {   [string match {diff --git *}      $line]
+                       || [string match {diff --cc *}       $line]
+                       || [string match {diff --combined *} $line]
+                       || [string match {--- *}             $line]
+                       || [string match {+++ *}             $line]} {
+                       append current_diff_header $line "\n"
+                       continue
+               }
+               if {[string match {index *} $line]} continue
                if {$line eq {deleted file mode 120000}} {
                        set line "deleted symlink"
                }
@@ -730,10 +743,10 @@ proc read_diff {fd} {
                #
                if {[string match {@@@ *} $line]} {set is_3way_diff 1}
 
-               if {[string match {index *} $line]
-                       || [string match {mode *} $line]
+               if {[string match {mode *} $line]
                        || [string match {new file *} $line]
                        || [string match {deleted file *} $line]
+                       || [string match {Binary files * and * differ} $line]
                        || $line eq {\ No newline at end of file}
                        || [regexp {^\* Unmerged path } $line]} {
                        set tags {}
@@ -781,6 +794,9 @@ proc read_diff {fd} {
                        }
                }
                $ui_diff insert end $line $tags
+               if {[string index $line end] eq "\r"} {
+                       $ui_diff tag add d_cr {end - 2c}
+               }
                $ui_diff insert end "\n" $tags
        }
        $ui_diff conf -state disabled
@@ -797,12 +813,84 @@ proc read_diff {fd} {
        }
 }
 
+proc apply_hunk {x y} {
+       global current_diff_path current_diff_header current_diff_side
+       global ui_diff ui_index file_states
+
+       if {$current_diff_path eq {} || $current_diff_header eq {}} return
+       if {![lock_index apply_hunk]} return
+
+       set apply_cmd {git apply --cached --whitespace=nowarn}
+       set mi [lindex $file_states($current_diff_path) 0]
+       if {$current_diff_side eq $ui_index} {
+               set mode unstage
+               lappend apply_cmd --reverse
+               if {[string index $mi 0] ne {M}} {
+                       unlock_index
+                       return
+               }
+       } else {
+               set mode stage
+               if {[string index $mi 1] ne {M}} {
+                       unlock_index
+                       return
+               }
+       }
+
+       set s_lno [lindex [split [$ui_diff index @$x,$y] .] 0]
+       set s_lno [$ui_diff search -backwards -regexp ^@@ $s_lno.0 0.0]
+       if {$s_lno eq {}} {
+               unlock_index
+               return
+       }
+
+       set e_lno [$ui_diff search -forwards -regexp ^@@ "$s_lno + 1 lines" end]
+       if {$e_lno eq {}} {
+               set e_lno end
+       }
+
+       if {[catch {
+               set p [open "| $apply_cmd" w]
+               fconfigure $p -translation binary -encoding binary
+               puts -nonewline $p $current_diff_header
+               puts -nonewline $p [$ui_diff get $s_lno $e_lno]
+               close $p} err]} {
+               error_popup "Failed to $mode selected hunk.\n\n$err"
+               unlock_index
+               return
+       }
+
+       $ui_diff conf -state normal
+       $ui_diff delete $s_lno $e_lno
+       $ui_diff conf -state disabled
+
+       if {[$ui_diff get 1.0 end] eq "\n"} {
+               set o _
+       } else {
+               set o ?
+       }
+
+       if {$current_diff_side eq $ui_index} {
+               set mi ${o}M
+       } elseif {[string index $mi 0] eq {_}} {
+               set mi M$o
+       } else {
+               set mi ?$o
+       }
+       unlock_index
+       display_file $current_diff_path $mi
+       if {$o eq {_}} {
+               clear_diff
+       }
+}
+
 ######################################################################
 ##
 ## commit
 
 proc load_last_commit {} {
        global HEAD PARENT MERGE_HEAD commit_type ui_comm
+       global repo_config
 
        if {[llength $PARENT] == 0} {
                error_popup {There is nothing to amend.
@@ -829,11 +917,18 @@ current merge activity.
        set parents [list]
        if {[catch {
                        set fd [open "| git cat-file commit $curHEAD" r]
+                       fconfigure $fd -encoding binary -translation lf
+                       if {[catch {set enc $repo_config(i18n.commitencoding)}]} {
+                               set enc utf-8
+                       }
                        while {[gets $fd line] > 0} {
                                if {[string match {parent *} $line]} {
                                        lappend parents [string range $line 7 end]
+                               } elseif {[string match {encoding *} $line]} {
+                                       set enc [string tolower [string range $line 9 end]]
                                }
                        }
+                       fconfigure $fd -encoding $enc
                        set msg [string trim [read $fd]]
                        close $fd
                } err]} {
@@ -952,6 +1047,7 @@ You must add at least 1 file before you can commit.
        # -- A message is required.
        #
        set msg [string trim [$ui_comm get 1.0 end]]
+       regsub -all -line {[ \t\r]+$} $msg {} msg
        if {$msg eq {}} {
                error_popup {Please supply a commit message.
 
@@ -1025,6 +1121,7 @@ proc commit_committree {fd_wt curHEAD msg} {
        global single_commit all_heads current_branch
        global ui_status_value ui_comm selected_commit_type
        global file_states selected_paths rescan_active
+       global repo_config
 
        gets $fd_wt tree_id
        if {$tree_id eq {} || [catch {close $fd_wt} err]} {
@@ -1034,6 +1131,17 @@ proc commit_committree {fd_wt curHEAD msg} {
                return
        }
 
+       # -- Build the message.
+       #
+       set msg_p [gitdir COMMIT_EDITMSG]
+       set msg_wt [open $msg_p w]
+       if {[catch {set enc $repo_config(i18n.commitencoding)}]} {
+               set enc utf-8
+       }
+       fconfigure $msg_wt -encoding $enc -translation binary
+       puts -nonewline $msg_wt $msg
+       close $msg_wt
+
        # -- Create the commit.
        #
        set cmd [list git commit-tree $tree_id]
@@ -1046,7 +1154,7 @@ proc commit_committree {fd_wt curHEAD msg} {
                # git commit-tree writes to stderr during initial commit.
                lappend cmd 2>/dev/null
        }
-       lappend cmd << $msg
+       lappend cmd <$msg_p
        if {[catch {set cmt_id [eval exec $cmd]} err]} {
                error_popup "commit-tree failed:\n\n$err"
                set ui_status_value {Commit failed.}
@@ -1084,6 +1192,7 @@ proc commit_committree {fd_wt curHEAD msg} {
 
        # -- Cleanup after ourselves.
        #
+       catch {file delete $msg_p}
        catch {file delete [gitdir MERGE_HEAD]}
        catch {file delete [gitdir MERGE_MSG]}
        catch {file delete [gitdir SQUASH_MSG]}
@@ -1459,6 +1568,7 @@ proc update_indexinfo {msg pathList after} {
                -blocking 0 \
                -buffering full \
                -buffersize 512 \
+               -encoding binary \
                -translation binary
        fileevent $fd writable [list \
                write_update_indexinfo \
@@ -1499,7 +1609,7 @@ proc write_update_indexinfo {fd pathList totalCnt batch msg after} {
                set info [lindex $s 2]
                if {$info eq {}} continue
 
-               puts -nonewline $fd "$info\t$path\0"
+               puts -nonewline $fd "$info\t[encoding convertto $path]\0"
                display_file $path $new
        }
 
@@ -1531,6 +1641,7 @@ proc update_index {msg pathList after} {
                -blocking 0 \
                -buffering full \
                -buffersize 512 \
+               -encoding binary \
                -translation binary
        fileevent $fd writable [list \
                write_update_index \
@@ -1575,7 +1686,7 @@ proc write_update_index {fd pathList totalCnt batch msg after} {
                ?M {set new M_}
                ?? {continue}
                }
-               puts -nonewline $fd "$path\0"
+               puts -nonewline $fd "[encoding convertto $path]\0"
                display_file $path $new
        }
 
@@ -1613,6 +1724,7 @@ proc checkout_index {msg pathList after} {
                -blocking 0 \
                -buffering full \
                -buffersize 512 \
+               -encoding binary \
                -translation binary
        fileevent $fd writable [list \
                write_checkout_index \
@@ -1645,7 +1757,7 @@ proc write_checkout_index {fd pathList totalCnt batch msg after} {
                U? {continue}
                ?M -
                ?D {
-                       puts -nonewline $fd "$path\0"
+                       puts -nonewline $fd "[encoding convertto $path]\0"
                        display_file $path ?_
                }
                }
@@ -1878,7 +1990,7 @@ proc do_create_branch {} {
                -width 40 \
                -font font_ui
        $w.desc.name_t insert 0.0 $repo_config(gui.newbranchtemplate)
-       grid $w.desc.name_l $w.desc.name_t -stick we -padx {0 5}
+       grid $w.desc.name_l $w.desc.name_t -sticky we -padx {0 5}
        bind $w.desc.name_t <Shift-Key-Tab> {focus [tk_focusPrev %W];break}
        bind $w.desc.name_t <Key-Tab> {focus [tk_focusNext %W];break}
        bind $w.desc.name_t <Key-Return> "do_create_branch_action $w;break"
@@ -1928,7 +2040,7 @@ proc do_create_branch {} {
                -height 1 \
                -width 50 \
                -font font_ui
-       grid $w.from.exp_r $w.from.exp_t -stick we -padx {0 5}
+       grid $w.from.exp_r $w.from.exp_t -sticky we -padx {0 5}
        bind $w.from.exp_t <Shift-Key-Tab> {focus [tk_focusPrev %W];break}
        bind $w.from.exp_t <Key-Tab> {focus [tk_focusNext %W];break}
        bind $w.from.exp_t <Key-Return> "do_create_branch_action $w;break"
@@ -2129,16 +2241,11 @@ proc do_delete_branch {} {
        tkwait window $w
 }
 
-proc switch_branch {b} {
-       global HEAD commit_type file_states current_branch
-       global selected_commit_type ui_comm
+proc switch_branch {new_branch} {
+       global HEAD commit_type current_branch repo_config
 
        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
@@ -2159,19 +2266,110 @@ The rescan will be automatically started now.
                return
        }
 
-       # -- Toss the message buffer if we are in amend mode.
+       if {$repo_config(gui.trustmtime) eq {true}} {
+               switch_branch_stage2 {} $new_branch
+       } else {
+               set ui_status_value {Refreshing file status...}
+               set cmd [list git update-index]
+               lappend cmd -q
+               lappend cmd --unmerged
+               lappend cmd --ignore-missing
+               lappend cmd --refresh
+               set fd_rf [open "| $cmd" r]
+               fconfigure $fd_rf -blocking 0 -translation binary
+               fileevent $fd_rf readable \
+                       [list switch_branch_stage2 $fd_rf $new_branch]
+       }
+}
+
+proc switch_branch_stage2 {fd_rf new_branch} {
+       global ui_status_value HEAD
+
+       if {$fd_rf ne {}} {
+               read $fd_rf
+               if {![eof $fd_rf]} return
+               close $fd_rf
+       }
+
+       set ui_status_value "Updating working directory to '$new_branch'..."
+       set cmd [list git read-tree]
+       lappend cmd -m
+       lappend cmd -u
+       lappend cmd --exclude-per-directory=.gitignore
+       lappend cmd $HEAD
+       lappend cmd $new_branch
+       set fd_rt [open "| $cmd" r]
+       fconfigure $fd_rt -blocking 0 -translation binary
+       fileevent $fd_rt readable \
+               [list switch_branch_readtree_wait $fd_rt $new_branch]
+}
+
+proc switch_branch_readtree_wait {fd_rt new_branch} {
+       global selected_commit_type commit_type HEAD MERGE_HEAD PARENT
+       global current_branch
+       global ui_comm ui_status_value
+
+       # -- We never get interesting output on stdout; only stderr.
        #
-       if {[string match amend* $curType]} {
-               $ui_comm delete 0.0 end
-               $ui_comm edit reset
-               $ui_comm edit modified false
+       read $fd_rt
+       fconfigure $fd_rt -blocking 1
+       if {![eof $fd_rt]} {
+               fconfigure $fd_rt -blocking 0
+               return
        }
 
-       set selected_commit_type new
-       set current_branch $new_branch
+       # -- The working directory wasn't in sync with the index and
+       #    we'd have to overwrite something to make the switch. A
+       #    merge is required.
+       #
+       if {[catch {close $fd_rt} err]} {
+               regsub {^fatal: } $err {} err
+               warn_popup "File level merge required.
+
+$err
+
+Staying on branch '$current_branch'."
+               set ui_status_value "Aborted checkout of '$new_branch' (file level merging is required)."
+               unlock_index
+               return
+       }
+
+       # -- Update the symbolic ref.  Core git doesn't even check for failure
+       #    here, it Just Works(tm).  If it doesn't we are in some really ugly
+       #    state that is difficult to recover from within git-gui.
+       #
+       if {[catch {exec git symbolic-ref HEAD "refs/heads/$new_branch"} err]} {
+               error_popup "Failed to set current branch.
+
+This working directory is only partially switched.
+We successfully updated your files, but failed to
+update an internal Git file.
+
+This should not have occurred.  [appname] will now
+close and give up.
 
+$err"
+               do_quit
+               return
+       }
+
+       # -- Update our repository state.  If we were previously in amend mode
+       #    we need to toss the current buffer and do a full rescan to update
+       #    our file lists.  If we weren't in amend mode our file lists are
+       #    accurate and we can avoid the rescan.
+       #
        unlock_index
-       error "NOT FINISHED"
+       set selected_commit_type new
+       if {[string match amend* $commit_type]} {
+               $ui_comm delete 0.0 end
+               $ui_comm edit reset
+               $ui_comm edit modified false
+               rescan {set ui_status_value "Checked out branch '$current_branch'."}
+       } else {
+               repository_state commit_type HEAD MERGE_HEAD
+               set PARENT $HEAD
+               set ui_status_value "Checked out branch '$current_branch'."
+       }
 }
 
 ######################################################################
@@ -2693,6 +2891,74 @@ proc do_gitk {revs} {
        }
 }
 
+proc do_stats {} {
+       set fd [open "| git count-objects -v" r]
+       while {[gets $fd line] > 0} {
+               if {[regexp {^([^:]+): (\d+)$} $line _ name value]} {
+                       set stats($name) $value
+               }
+       }
+       close $fd
+
+       set packed_sz 0
+       foreach p [glob -directory [gitdir objects pack] \
+               -type f \
+               -nocomplain -- *] {
+               incr packed_sz [file size $p]
+       }
+       if {$packed_sz > 0} {
+               set stats(size-pack) [expr {$packed_sz / 1024}]
+       }
+
+       set w .stats_view
+       toplevel $w
+       wm geometry $w "+[winfo rootx .]+[winfo rooty .]"
+
+       label $w.header -text {Database Statistics} \
+               -font font_uibold
+       pack $w.header -side top -fill x
+
+       frame $w.buttons -border 1
+       button $w.buttons.close -text Close \
+               -font font_ui \
+               -command [list destroy $w]
+       button $w.buttons.gc -text {Compress Database} \
+               -font font_ui \
+               -command "destroy $w;do_gc"
+       pack $w.buttons.close -side right
+       pack $w.buttons.gc -side left
+       pack $w.buttons -side bottom -fill x -pady 10 -padx 10
+
+       frame $w.stat -borderwidth 1 -relief solid
+       foreach s {
+               {count           {Number of loose objects}}
+               {size            {Disk space used by loose objects} { KiB}}
+               {in-pack         {Number of packed objects}}
+               {packs           {Number of packs}}
+               {size-pack       {Disk space used by packed objects} { KiB}}
+               {prune-packable  {Packed objects waiting for pruning}}
+               {garbage         {Garbage files}}
+               } {
+               set name [lindex $s 0]
+               set label [lindex $s 1]
+               if {[catch {set value $stats($name)}]} continue
+               if {[llength $s] > 2} {
+                       set value "$value[lindex $s 2]"
+               }
+
+               label $w.stat.l_$name -text "$label:" -anchor w -font font_ui
+               label $w.stat.v_$name -text $value -anchor w -font font_ui
+               grid $w.stat.l_$name $w.stat.v_$name -sticky we -padx {0 5}
+       }
+       pack $w.stat -pady 10 -padx 10
+
+       bind $w <Visibility> "grab $w; focus $w"
+       bind $w <Key-Escape> [list destroy $w]
+       bind $w <Key-Return> [list destroy $w]
+       wm title $w "[appname] ([reponame]): Database Statistics"
+       tkwait window $w
+}
+
 proc do_gc {} {
        set w [new_console {gc} {Compressing the object database}]
        console_exec $w {git gc}
@@ -2720,12 +2986,13 @@ proc do_quit {} {
        #
        set save [gitdir GITGUI_MSG]
        set msg [string trim [$ui_comm get 0.0 end]]
-       if {![string match amend* $commit_type]
-               && [$ui_comm edit modified]
+       regsub -all -line {[ \r\t]+$} $msg {} msg
+       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]]
+                       puts -nonewline $fd $msg
                        close $fd
                }
        } else {
@@ -3074,11 +3341,9 @@ proc do_options {} {
        pack $w.buttons -side bottom -fill x -pady 10 -padx 10
 
        labelframe $w.repo -text "[reponame] Repository" \
-               -font font_ui \
-               -relief raised -borderwidth 2
+               -font font_ui
        labelframe $w.global -text {Global (All Repositories)} \
-               -font font_ui \
-               -relief raised -borderwidth 2
+               -font font_ui
        pack $w.repo -side left -fill both -expand 1 -pady 5 -padx 5
        pack $w.global -side right -fill both -expand 1 -pady 5 -padx 5
 
@@ -3507,15 +3772,17 @@ menu .mbar.repository
        -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 command \
+       -label {Visualize All Branches} \
+       -command {do_gitk {--all}} \
+       -font font_ui
 .mbar.repository add separator
 
 if {!$single_commit} {
+       .mbar.repository add command -label {Database Statistics} \
+               -command do_stats \
+               -font font_ui
+
        .mbar.repository add command -label {Compress Database} \
                -command do_gc \
                -font font_ui
@@ -3821,7 +4088,7 @@ frame .vpane.lower.commarea
 frame .vpane.lower.diff -relief sunken -borderwidth 1
 pack .vpane.lower.commarea -side top -fill x
 pack .vpane.lower.diff -side bottom -fill both -expand 1
-.vpane add .vpane.lower -stick nsew
+.vpane add .vpane.lower -sticky nsew
 
 # -- Commit Area Buttons
 #
@@ -3960,6 +4227,7 @@ bind_button3 $ui_comm "tk_popup $ctxm %X %Y"
 # -- Diff Header
 #
 set current_diff_path {}
+set current_diff_side {}
 set diff_actions [list]
 proc trace_current_diff_path {varname args} {
        global current_diff_path diff_actions file_states
@@ -4040,6 +4308,7 @@ 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 d_cr -elide true
 $ui_diff tag conf d_@ -foreground blue -font font_diffbold
 $ui_diff tag conf d_+ -foreground {#00a000}
 $ui_diff tag conf d_- -foreground red
@@ -4079,6 +4348,7 @@ $ctxm add command \
        -label {Refresh} \
        -font font_ui \
        -command reshow_diff
+lappend diff_actions [list $ctxm entryconf [$ctxm index last] -state]
 $ctxm add command \
        -label {Copy} \
        -font font_ui \
@@ -4099,6 +4369,13 @@ $ctxm add command \
        }
 lappend diff_actions [list $ctxm entryconf [$ctxm index last] -state]
 $ctxm add separator
+$ctxm add command \
+       -label {Apply/Reverse Hunk} \
+       -font font_ui \
+       -command {apply_hunk $cursorX $cursorY}
+set ui_diff_applyhunk [$ctxm index last]
+lappend diff_actions [list $ctxm entryconf $ui_diff_applyhunk -state]
+$ctxm add separator
 $ctxm add command \
        -label {Decrease Font Size} \
        -font font_ui \
@@ -4130,7 +4407,17 @@ $ctxm add separator
 $ctxm add command -label {Options...} \
        -font font_ui \
        -command do_options
-bind_button3 $ui_diff "tk_popup $ctxm %X %Y"
+bind_button3 $ui_diff "
+       set cursorX %x
+       set cursorY %y
+       if {\$ui_index eq \$current_diff_side} {
+               $ctxm entryconf $ui_diff_applyhunk -label {Unstage Hunk From Commit}
+       } else {
+               $ctxm entryconf $ui_diff_applyhunk -label {Stage Hunk For Commit}
+       }
+       tk_popup $ctxm %X %Y
+"
+unset ui_diff_applyhunk
 
 # -- Status Bar
 #
@@ -4183,6 +4470,7 @@ bind $ui_diff <Key-Up>     {catch {%W yview scroll -1 units};break}
 bind $ui_diff <Key-Down>   {catch {%W yview scroll  1 units};break}
 bind $ui_diff <Key-Left>   {catch {%W xview scroll -1 units};break}
 bind $ui_diff <Key-Right>  {catch {%W xview scroll  1 units};break}
+bind $ui_diff <Button-1>   {focus %W}
 
 if {!$single_commit} {
        bind . <$M1B-Key-n> do_create_branch