chiark - git - mdw - stgit/blob - contrib/stgit.el

   1 ;; stgit.el: An emacs mode for StGit
   2 ;;
   3 ;; Copyright (C) 2007 David Kågedal <davidk@lysator.liu.se>
   4 ;;
   5 ;; To install: put this file on the load-path and place the following
   6 ;; in your .emacs file:
   7 ;;
   8 ;;    (require 'stgit)
   9 ;;
  10 ;; To start: `M-x stgit'
  11
  12 (require 'git nil t)
  13 (require 'cl)
  14 (require 'ewoc)
  15
  16 (defun stgit (dir)
  17   "Manage StGit patches for the tree in DIR."
  18   (interactive "DDirectory: \n")
  19   (switch-to-stgit-buffer (git-get-top-dir dir))
  20   (stgit-reload))
  21
  22 (unless (fboundp 'git-get-top-dir)
  23   (defun git-get-top-dir (dir)
  24     "Retrieve the top-level directory of a git tree."
  25     (let ((cdup (with-output-to-string
  26                   (with-current-buffer standard-output
  27                     (cd dir)
  28                     (unless (eq 0 (call-process "git" nil t nil
  29                                                 "rev-parse" "--show-cdup"))
  30                       (error "Cannot find top-level git tree for %s" dir))))))
  31       (expand-file-name (concat (file-name-as-directory dir)
  32                                 (car (split-string cdup "\n")))))))
  33
  34 (defun stgit-refresh-git-status (&optional dir)
  35   "If it exists, refresh the `git-status' buffer belonging to
  36 directory DIR or `default-directory'"
  37   (when (and (fboundp 'git-find-status-buffer)
  38              (fboundp 'git-refresh-status))
  39     (let* ((top-dir (git-get-top-dir (or dir default-directory)))
  40            (git-status-buffer (and top-dir (git-find-status-buffer top-dir))))
  41       (when git-status-buffer
  42         (with-current-buffer git-status-buffer
  43           (git-refresh-status))))))
  44
  45 (defun stgit-find-buffer (dir)
  46   "Return the buffer displaying StGit patches for DIR, or nil if none."
  47   (setq dir (file-name-as-directory dir))
  48   (let ((buffers (buffer-list)))
  49     (while (and buffers
  50                 (not (with-current-buffer (car buffers)
  51                        (and (eq major-mode 'stgit-mode)
  52                             (string= default-directory dir)))))
  53       (setq buffers (cdr buffers)))
  54     (and buffers (car buffers))))
  55
  56 (defun switch-to-stgit-buffer (dir)
  57   "Switch to a (possibly new) buffer displaying StGit patches for DIR."
  58   (setq dir (file-name-as-directory dir))
  59   (let ((buffer (stgit-find-buffer dir)))
  60     (switch-to-buffer (or buffer
  61                           (create-stgit-buffer dir)))))
  62
  63 (defstruct (stgit-patch)
  64   status name desc empty files-ewoc)
  65
  66 (defun stgit-patch-pp (patch)
  67   (let ((status (stgit-patch-status patch))
  68         (start (point))
  69         (name (stgit-patch-name patch)))
  70     (case name
  71        (:index (insert (propertize "  Index" 'face 'italic)))
  72        (:work (insert (propertize "  Work tree" 'face 'italic)))
  73        (t (insert (case status
  74                     ('applied "+")
  75                     ('top ">")
  76                     ('unapplied "-"))
  77                   (if (memq name stgit-marked-patches)
  78                       "*" " ")
  79                   (propertize (format "%-30s"
  80                                       (symbol-name name))
  81                               'face (case status
  82                                       ('applied 'stgit-applied-patch-face)
  83                                       ('top 'stgit-top-patch-face)
  84                                       ('unapplied 'stgit-unapplied-patch-face)
  85                                       ('index nil)
  86                                       ('work nil)))
  87                   "  "
  88                   (if (stgit-patch-empty patch) "(empty) " "")
  89                   (propertize (or (stgit-patch-desc patch) "")
  90                               'face 'stgit-description-face))))
  91     (put-text-property start (point) 'entry-type 'patch)
  92     (when (memq name stgit-expanded-patches)
  93       (stgit-insert-patch-files patch))
  94     (put-text-property start (point) 'patch-data patch)))
  95
  96 (defun create-stgit-buffer (dir)
  97   "Create a buffer for showing StGit patches.
  98 Argument DIR is the repository path."
  99   (let ((buf (create-file-buffer (concat dir "*stgit*")))
 100         (inhibit-read-only t))
 101     (with-current-buffer buf
 102       (setq default-directory dir)
 103       (stgit-mode)
 104       (set (make-local-variable 'stgit-ewoc)
 105            (ewoc-create #'stgit-patch-pp "Branch:\n" "--"))
 106       (setq buffer-read-only t))
 107     buf))
 108
 109 (defmacro stgit-capture-output (name &rest body)
 110   "Capture StGit output and, if there was any output, show it in a window
 111 at the end.
 112 Returns nil if there was no output."
 113   (declare (debug ([&or stringp null] body))
 114            (indent 1))
 115   `(let ((output-buf (get-buffer-create ,(or name "*StGit output*")))
 116          (stgit-dir default-directory)
 117          (inhibit-read-only t))
 118      (with-current-buffer output-buf
 119        (erase-buffer)
 120        (setq default-directory stgit-dir)
 121        (setq buffer-read-only t))
 122      (let ((standard-output output-buf))
 123        ,@body)
 124      (with-current-buffer output-buf
 125        (set-buffer-modified-p nil)
 126        (setq buffer-read-only t)
 127        (if (< (point-min) (point-max))
 128            (display-buffer output-buf t)))))
 129
 130 (defun stgit-make-run-args (args)
 131   "Return a copy of ARGS with its elements converted to strings."
 132   (mapcar (lambda (x)
 133             ;; don't use (format "%s" ...) to limit type errors
 134             (cond ((stringp x) x)
 135                   ((integerp x) (number-to-string x))
 136                   ((symbolp x) (symbol-name x))
 137                   (t
 138                    (error "Bad element in stgit-make-run-args args: %S" x))))
 139           args))
 140
 141 (defun stgit-run-silent (&rest args)
 142   (setq args (stgit-make-run-args args))
 143   (apply 'call-process "stg" nil standard-output nil args))
 144
 145 (defun stgit-run (&rest args)
 146   (setq args (stgit-make-run-args args))
 147   (let ((msgcmd (mapconcat #'identity args " ")))
 148     (message "Running stg %s..." msgcmd)
 149     (apply 'call-process "stg" nil standard-output nil args)
 150     (message "Running stg %s...done" msgcmd)))
 151
 152 (defun stgit-run-git (&rest args)
 153   (setq args (stgit-make-run-args args))
 154   (let ((msgcmd (mapconcat #'identity args " ")))
 155     (message "Running git %s..." msgcmd)
 156     (apply 'call-process "git" nil standard-output nil args)
 157     (message "Running git %s...done" msgcmd)))
 158
 159 (defun stgit-run-git-silent (&rest args)
 160   (setq args (stgit-make-run-args args))
 161   (apply 'call-process "git" nil standard-output nil args))
 162
 163 (defun stgit-index-empty-p ()
 164   "Returns non-nil if the index contains no changes from HEAD."
 165   (zerop (stgit-run-git-silent "diff-index" "--cached" "--quiet" "HEAD")))
 166
 167 (defvar stgit-index-node nil)
 168 (defvar stgit-worktree-node nil)
 169
 170 (defun stgit-refresh-index ()
 171   (when stgit-index-node
 172     (ewoc-invalidate (car stgit-index-node) (cdr stgit-index-node))))
 173
 174 (defun stgit-refresh-worktree ()
 175   (when stgit-worktree-node
 176     (ewoc-invalidate (car stgit-worktree-node) (cdr stgit-worktree-node))))
 177
 178 (defun stgit-run-series (ewoc)
 179   (let ((first-line t))
 180     (with-temp-buffer
 181       (let ((exit-status (stgit-run-silent "series" "--description" "--empty")))
 182         (goto-char (point-min))
 183         (if (not (zerop exit-status))
 184             (cond ((looking-at "stg series: \\(.*\\)")
 185                    (ewoc-set-hf ewoc (car (ewoc-get-hf ewoc))
 186                                 "-- not initialized (run M-x stgit-init)"))
 187                   ((looking-at ".*")
 188                    (error "Error running stg: %s"
 189                           (match-string 0))))
 190           (while (not (eobp))
 191             (unless (looking-at
 192                      "\\([0 ]\\)\\([>+-]\\)\\( \\)\\([^ ]+\\) *[|#] \\(.*\\)")
 193               (error "Syntax error in output from stg series"))
 194             (let* ((state-str (match-string 2))
 195                    (state (cond ((string= state-str ">") 'top)
 196                                 ((string= state-str "+") 'applied)
 197                                 ((string= state-str "-") 'unapplied))))
 198               (ewoc-enter-last ewoc
 199                                (make-stgit-patch
 200                                 :status state
 201                                 :name (intern (match-string 4))
 202                                 :desc (match-string 5)
 203                                 :empty (string= (match-string 1) "0"))))
 204             (setq first-line nil)
 205             (forward-line 1)))))
 206     (if stgit-show-worktree
 207         (setq stgit-index-node (cons ewoc (ewoc-enter-last ewoc
 208                                                            (make-stgit-patch
 209                                                             :status 'index
 210                                                             :name :index
 211                                                             :desc nil
 212                                                             :empty nil)))
 213               stgit-worktree-node (cons ewoc (ewoc-enter-last ewoc
 214                                                               (make-stgit-patch
 215                                                                :status 'work
 216                                                                :name :work
 217                                                                :desc nil
 218                                                                :empty nil))))
 219       (setq stgit-worktree-node nil))))
 220
 221
 222 (defun stgit-reload ()
 223   "Update the contents of the StGit buffer."
 224   (interactive)
 225   (let ((inhibit-read-only t)
 226         (curline (line-number-at-pos))
 227         (curpatch (stgit-patch-name-at-point)))
 228     (ewoc-filter stgit-ewoc #'(lambda (x) nil))
 229     (ewoc-set-hf stgit-ewoc
 230                  (concat "Branch: "
 231                          (propertize
 232                           (with-temp-buffer
 233                             (stgit-run-silent "branch")
 234                             (buffer-substring (point-min) (1- (point-max))))
 235                           'face 'bold)
 236                          "\n")
 237                  (if stgit-show-worktree
 238                      "--"
 239                    (propertize
 240                     (substitute-command-keys "--\n\"\\[stgit-toggle-worktree]\"\
 241  shows the working tree\n")
 242                    'face 'stgit-description-face)))
 243     (stgit-run-series stgit-ewoc)
 244     (if curpatch
 245         (stgit-goto-patch curpatch)
 246       (goto-line curline)))
 247   (stgit-refresh-git-status))
 248
 249 (defgroup stgit nil
 250   "A user interface for the StGit patch maintenance tool."
 251   :group 'tools)
 252
 253 (defface stgit-description-face
 254   '((((background dark)) (:foreground "tan"))
 255     (((background light)) (:foreground "dark red")))
 256   "The face used for StGit descriptions"
 257   :group 'stgit)
 258
 259 (defface stgit-top-patch-face
 260   '((((background dark)) (:weight bold :foreground "yellow"))
 261     (((background light)) (:weight bold :foreground "purple"))
 262     (t (:weight bold)))
 263   "The face used for the top patch names"
 264   :group 'stgit)
 265
 266 (defface stgit-applied-patch-face
 267   '((((background dark)) (:foreground "light yellow"))
 268     (((background light)) (:foreground "purple"))
 269     (t ()))
 270   "The face used for applied patch names"
 271   :group 'stgit)
 272
 273 (defface stgit-unapplied-patch-face
 274   '((((background dark)) (:foreground "gray80"))
 275     (((background light)) (:foreground "orchid"))
 276     (t ()))
 277   "The face used for unapplied patch names"
 278   :group 'stgit)
 279
 280 (defface stgit-modified-file-face
 281   '((((class color) (background light)) (:foreground "purple"))
 282     (((class color) (background dark)) (:foreground "salmon")))
 283   "StGit mode face used for modified file status"
 284   :group 'stgit)
 285
 286 (defface stgit-unmerged-file-face
 287   '((((class color) (background light)) (:foreground "red" :bold t))
 288     (((class color) (background dark)) (:foreground "red" :bold t)))
 289   "StGit mode face used for unmerged file status"
 290   :group 'stgit)
 291
 292 (defface stgit-unknown-file-face
 293   '((((class color) (background light)) (:foreground "goldenrod" :bold t))
 294     (((class color) (background dark)) (:foreground "goldenrod" :bold t)))
 295   "StGit mode face used for unknown file status"
 296   :group 'stgit)
 297
 298 (defface stgit-file-permission-face
 299   '((((class color) (background light)) (:foreground "green" :bold t))
 300     (((class color) (background dark)) (:foreground "green" :bold t)))
 301   "StGit mode face used for permission changes."
 302   :group 'stgit)
 303
 304 (defcustom stgit-expand-find-copies-harder
 305   nil
 306   "Try harder to find copied files when listing patches.
 307
 308 When not nil, runs git diff-tree with the --find-copies-harder
 309 flag, which reduces performance."
 310   :type 'boolean
 311   :group 'stgit)
 312
 313 (defconst stgit-file-status-code-strings
 314   (mapcar (lambda (arg)
 315             (cons (car arg)
 316                   (propertize (cadr arg) 'face (car (cddr arg)))))
 317           '((add         "Added"       stgit-modified-file-face)
 318             (copy        "Copied"      stgit-modified-file-face)
 319             (delete      "Deleted"     stgit-modified-file-face)
 320             (modify      "Modified"    stgit-modified-file-face)
 321             (rename      "Renamed"     stgit-modified-file-face)
 322             (mode-change "Mode change" stgit-modified-file-face)
 323             (unmerged    "Unmerged"    stgit-unmerged-file-face)
 324             (unknown     "Unknown"     stgit-unknown-file-face)))
 325   "Alist of code symbols to description strings")
 326
 327 (defun stgit-file-status-code-as-string (file)
 328   "Return stgit status code for FILE as a string"
 329   (let* ((code (assq (stgit-file-status file)
 330                      stgit-file-status-code-strings))
 331          (score (stgit-file-cr-score file)))
 332     (when code
 333       (format "%-11s  "
 334               (if (and score (/= score 100))
 335                   (format "%s %s" (cdr code)
 336                           (propertize (format "%d%%" score)
 337                                       'face 'stgit-description-face))
 338                 (cdr code))))))
 339
 340 (defun stgit-file-status-code (str &optional score)
 341   "Return stgit status code from git status string"
 342   (let ((code (assoc str '(("A" . add)
 343                            ("C" . copy)
 344                            ("D" . delete)
 345                            ("M" . modify)
 346                            ("R" . rename)
 347                            ("T" . mode-change)
 348                            ("U" . unmerged)
 349                            ("X" . unknown)))))
 350     (setq code (if code (cdr code) 'unknown))
 351     (when (stringp score)
 352       (if (> (length score) 0)
 353           (setq score (string-to-number score))
 354         (setq score nil)))
 355     (if score (cons code score) code)))
 356
 357 (defconst stgit-file-type-strings
 358   '((#o100 . "file")
 359     (#o120 . "symlink")
 360     (#o160 . "subproject"))
 361   "Alist of names of file types")
 362
 363 (defun stgit-file-type-string (type)
 364   "Return string describing file type TYPE (the high bits of file permission).
 365 Cf. `stgit-file-type-strings' and `stgit-file-type-change-string'."
 366   (let ((type-str (assoc type stgit-file-type-strings)))
 367     (or (and type-str (cdr type-str))
 368         (format "unknown type %o" type))))
 369
 370 (defun stgit-file-type-change-string (old-perm new-perm)
 371   "Return string describing file type change from OLD-PERM to NEW-PERM.
 372 Cf. `stgit-file-type-string'."
 373   (let ((old-type (lsh old-perm -9))
 374         (new-type (lsh new-perm -9)))
 375     (cond ((= old-type new-type) "")
 376           ((zerop new-type) "")
 377           ((zerop old-type)
 378            (if (= new-type #o100)
 379                ""
 380              (format "   (%s)" (stgit-file-type-string new-type))))
 381           (t (format "   (%s -> %s)"
 382                      (stgit-file-type-string old-type)
 383                      (stgit-file-type-string new-type))))))
 384
 385 (defun stgit-file-mode-change-string (old-perm new-perm)
 386   "Return string describing file mode change from OLD-PERM to NEW-PERM.
 387 Cf. `stgit-file-type-change-string'."
 388   (setq old-perm (logand old-perm #o777)
 389         new-perm (logand new-perm #o777))
 390   (if (or (= old-perm new-perm)
 391           (zerop old-perm)
 392           (zerop new-perm))
 393       ""
 394     (let* ((modified       (logxor old-perm new-perm))
 395            (not-x-modified (logand (logxor old-perm new-perm) #o666)))
 396       (cond ((zerop modified) "")
 397             ((and (zerop not-x-modified)
 398                   (or (and (eq #o111 (logand old-perm #o111))
 399                            (propertize "-x" 'face 'stgit-file-permission-face))
 400                       (and (eq #o111 (logand new-perm #o111))
 401                            (propertize "+x" 'face
 402                                        'stgit-file-permission-face)))))
 403             (t (concat (propertize (format "%o" old-perm)
 404                                    'face 'stgit-file-permission-face)
 405                        (propertize " -> "
 406                                    'face 'stgit-description-face)
 407                        (propertize (format "%o" new-perm)
 408                                    'face 'stgit-file-permission-face)))))))
 409
 410 (defstruct (stgit-file)
 411   old-perm new-perm copy-or-rename cr-score cr-from cr-to status file)
 412
 413 (defun stgit-file-pp (file)
 414   (let ((status (stgit-file-status file))
 415         (name (if (stgit-file-copy-or-rename file)
 416                   (concat (stgit-file-cr-from file)
 417                           (propertize " -> "
 418                                       'face 'stgit-description-face)
 419                           (stgit-file-cr-to file))
 420                 (stgit-file-file file)))
 421         (mode-change (stgit-file-mode-change-string
 422                       (stgit-file-old-perm file)
 423                       (stgit-file-new-perm file)))
 424         (start (point)))
 425     (insert (format "    %-12s%1s%s%s\n"
 426                     (stgit-file-status-code-as-string file)
 427                     mode-change
 428                     name
 429                     (propertize (stgit-file-type-change-string
 430                                  (stgit-file-old-perm file)
 431                                  (stgit-file-new-perm file))
 432                                 'face 'stgit-description-face)))
 433     (add-text-properties start (point)
 434                          (list 'entry-type 'file
 435                                'file-data file))))
 436
 437 (defun stgit-insert-patch-files (patch)
 438   "Expand (show modification of) the patch PATCH after the line
 439 at point."
 440   (let* ((patchsym (stgit-patch-name patch))
 441          (end (progn (insert "#") (prog1 (point-marker) (forward-char -1))))
 442          (args (list "-z" (if stgit-expand-find-copies-harder
 443                               "--find-copies-harder"
 444                             "-C")))
 445          (ewoc (ewoc-create #'stgit-file-pp nil nil t)))
 446     (setf (stgit-patch-files-ewoc patch) ewoc)
 447     (when (eq patchsym :work)
 448       (setq stgit-work-ewoc ewoc))
 449     (with-temp-buffer
 450       (apply 'stgit-run-git
 451              (cond ((eq patchsym :work)
 452                     `("diff-files" ,@args))
 453                    ((eq patchsym :index)
 454                     `("diff-index" ,@args "--cached" "HEAD"))
 455                    (t
 456                     `("diff-tree" ,@args "-r" ,(stgit-id patchsym)))))
 457       (goto-char (point-min))
 458       (unless (or (eobp) (memq patchsym '(:work :index)))
 459         (forward-char 41))
 460       (while (looking-at ":\\([0-7]+\\) \\([0-7]+\\) [0-9A-Fa-f]\\{40\\} [0-9A-Fa-f]\\{40\\} ")
 461         (let ((old-perm (string-to-number (match-string 1) 8))
 462               (new-perm (string-to-number (match-string 2) 8)))
 463           (goto-char (match-end 0))
 464           (let ((file
 465                  (cond ((looking-at
 466                          "\\([CR]\\)\\([0-9]*\\)\0\\([^\0]*\\)\0\\([^\0]*\\)\0")
 467                         (make-stgit-file
 468                          :old-perm       old-perm
 469                          :new-perm       new-perm
 470                          :copy-or-rename t
 471                          :cr-score       (string-to-number (match-string 2))
 472                          :cr-from        (match-string 3)
 473                          :cr-to          (match-string 4)
 474                          :status         (stgit-file-status-code (match-string 1))
 475                          :file           (match-string 3)))
 476                        ((looking-at "\\([ABD-QS-Z]\\)\0\\([^\0]*\\)\0")
 477                         (make-stgit-file
 478                          :old-perm       old-perm
 479                          :new-perm       new-perm
 480                          :copy-or-rename nil
 481                          :cr-score       nil
 482                          :cr-from        nil
 483                          :cr-to          nil
 484                          :status         (stgit-file-status-code (match-string 1))
 485                          :file           (match-string 2))))))
 486             (ewoc-enter-last ewoc file))
 487           (goto-char (match-end 0))))
 488       (unless (ewoc-nth ewoc 0)
 489         (ewoc-set-hf ewoc "" (propertize "    <no files>\n"
 490                                          'face 'stgit-description-face))))
 491     (goto-char end)
 492     (delete-char -2)))
 493
 494 (defun stgit-select-file ()
 495   (let ((filename (expand-file-name
 496                    (stgit-file-file (stgit-patched-file-at-point)))))
 497     (unless (file-exists-p filename)
 498       (error "File does not exist"))
 499     (find-file filename)))
 500
 501 (defun stgit-select-patch ()
 502   (let ((patchname (stgit-patch-name-at-point)))
 503     (if (memq patchname stgit-expanded-patches)
 504         (setq stgit-expanded-patches (delq patchname stgit-expanded-patches))
 505       (setq stgit-expanded-patches (cons patchname stgit-expanded-patches)))
 506     (ewoc-invalidate stgit-ewoc (ewoc-locate stgit-ewoc)))
 507   (move-to-column (stgit-goal-column)))
 508
 509 (defun stgit-select ()
 510   "Expand or collapse the current entry"
 511   (interactive)
 512   (case (get-text-property (point) 'entry-type)
 513     ('patch
 514      (stgit-select-patch))
 515     ('file
 516      (stgit-select-file))
 517     (t
 518      (error "No patch or file on line"))))
 519
 520 (defun stgit-find-file-other-window ()
 521   "Open file at point in other window"
 522   (interactive)
 523   (let ((patched-file (stgit-patched-file-at-point)))
 524     (unless patched-file
 525       (error "No file on the current line"))
 526     (let ((filename (expand-file-name (stgit-file-file patched-file))))
 527       (unless (file-exists-p filename)
 528         (error "File does not exist"))
 529       (find-file-other-window filename))))
 530
 531 (defun stgit-quit ()
 532   "Hide the stgit buffer."
 533   (interactive)
 534   (bury-buffer))
 535
 536 (defun stgit-git-status ()
 537   "Show status using `git-status'."
 538   (interactive)
 539   (unless (fboundp 'git-status)
 540     (error "The stgit-git-status command requires git-status"))
 541   (let ((dir default-directory))
 542     (save-selected-window
 543       (pop-to-buffer nil)
 544       (git-status dir))))
 545
 546 (defun stgit-goal-column ()
 547   "Return goal column for the current line"
 548   (case (get-text-property (point) 'entry-type)
 549     ('patch 2)
 550     ('file 4)
 551     (t 0)))
 552
 553 (defun stgit-next-line (&optional arg)
 554   "Move cursor vertically down ARG lines"
 555   (interactive "p")
 556   (next-line arg)
 557   (move-to-column (stgit-goal-column)))
 558
 559 (defun stgit-previous-line (&optional arg)
 560   "Move cursor vertically up ARG lines"
 561   (interactive "p")
 562   (previous-line arg)
 563   (move-to-column (stgit-goal-column)))
 564
 565 (defun stgit-next-patch (&optional arg)
 566   "Move cursor down ARG patches."
 567   (interactive "p")
 568   (ewoc-goto-next stgit-ewoc (or arg 1))
 569   (move-to-column goal-column))
 570
 571 (defun stgit-previous-patch (&optional arg)
 572   "Move cursor up ARG patches."
 573   (interactive "p")
 574   (ewoc-goto-prev stgit-ewoc (or arg 1))
 575   (move-to-column goal-column))
 576
 577 (defvar stgit-mode-hook nil
 578   "Run after `stgit-mode' is setup.")
 579
 580 (defvar stgit-mode-map nil
 581   "Keymap for StGit major mode.")
 582
 583 (unless stgit-mode-map
 584   (let ((toggle-map (make-keymap)))
 585     (suppress-keymap toggle-map)
 586     (mapc (lambda (arg) (define-key toggle-map (car arg) (cdr arg)))
 587           '(("t" .        stgit-toggle-worktree)))
 588     (setq stgit-mode-map (make-keymap))
 589     (suppress-keymap stgit-mode-map)
 590     (mapc (lambda (arg) (define-key stgit-mode-map (car arg) (cdr arg)))
 591           `((" " .        stgit-mark)
 592             ("m" .        stgit-mark)
 593             ("\d" .       stgit-unmark-up)
 594             ("u" .        stgit-unmark-down)
 595             ("?" .        stgit-help)
 596             ("h" .        stgit-help)
 597             ("\C-p" .     stgit-previous-line)
 598             ("\C-n" .     stgit-next-line)
 599             ([up] .       stgit-previous-line)
 600             ([down] .     stgit-next-line)
 601             ("p" .        stgit-previous-patch)
 602             ("n" .        stgit-next-patch)
 603             ("\M-{" .     stgit-previous-patch)
 604             ("\M-}" .     stgit-next-patch)
 605             ("s" .        stgit-git-status)
 606             ("g" .        stgit-reload)
 607             ("r" .        stgit-refresh)
 608             ("\C-c\C-r" . stgit-rename)
 609             ("e" .        stgit-edit)
 610             ("M" .        stgit-move-patches)
 611             ("S" .        stgit-squash)
 612             ("N" .        stgit-new)
 613             ("R" .        stgit-repair)
 614             ("C" .        stgit-commit)
 615             ("U" .        stgit-uncommit)
 616             ("\r" .       stgit-select)
 617             ("o" .        stgit-find-file-other-window)
 618             ("i" .        stgit-file-toggle-index)
 619             (">" .        stgit-push-next)
 620             ("<" .        stgit-pop-next)
 621             ("P" .        stgit-push-or-pop)
 622             ("G" .        stgit-goto)
 623             ("=" .        stgit-show)
 624             ("D" .        stgit-delete)
 625             ([(control ?/)] . stgit-undo)
 626             ("\C-_" .     stgit-undo)
 627             ("B" .        stgit-branch)
 628             ("t" .        ,toggle-map)
 629             ("q" .        stgit-quit)))))
 630
 631 (defun stgit-mode ()
 632   "Major mode for interacting with StGit.
 633 Commands:
 634 \\{stgit-mode-map}"
 635   (kill-all-local-variables)
 636   (buffer-disable-undo)
 637   (setq mode-name "StGit"
 638         major-mode 'stgit-mode
 639         goal-column 2)
 640   (use-local-map stgit-mode-map)
 641   (set (make-local-variable 'list-buffers-directory) default-directory)
 642   (set (make-local-variable 'stgit-marked-patches) nil)
 643   (set (make-local-variable 'stgit-expanded-patches) nil)
 644   (set (make-local-variable 'stgit-show-worktree) stgit-default-show-worktree)
 645   (set-variable 'truncate-lines 't)
 646   (add-hook 'after-save-hook 'stgit-update-saved-file)
 647   (run-hooks 'stgit-mode-hook))
 648
 649 (defun stgit-update-saved-file ()
 650   (let* ((file (expand-file-name buffer-file-name))
 651          (dir (file-name-directory file))
 652          (gitdir (condition-case nil (git-get-top-dir dir)
 653                    (error nil)))
 654          (buffer (and gitdir (stgit-find-buffer gitdir))))
 655     (when buffer
 656       (with-current-buffer buffer
 657         (stgit-refresh-worktree)))))
 658
 659 (defun stgit-add-mark (patchsym)
 660   "Mark the patch PATCHSYM."
 661   (setq stgit-marked-patches (cons patchsym stgit-marked-patches)))
 662
 663 (defun stgit-remove-mark (patchsym)
 664   "Unmark the patch PATCHSYM."
 665   (setq stgit-marked-patches (delq patchsym stgit-marked-patches)))
 666
 667 (defun stgit-clear-marks ()
 668   "Unmark all patches."
 669   (setq stgit-marked-patches '()))
 670
 671 (defun stgit-patch-at-point (&optional cause-error)
 672   (get-text-property (point) 'patch-data))
 673
 674 (defun stgit-patch-name-at-point (&optional cause-error)
 675   "Return the patch name on the current line as a symbol.
 676 If CAUSE-ERROR is not nil, signal an error if none found."
 677   (let ((patch (stgit-patch-at-point)))
 678     (cond (patch
 679            (stgit-patch-name patch))
 680           (cause-error
 681            (error "No patch on this line")))))
 682
 683 (defun stgit-patched-file-at-point ()
 684   (get-text-property (point) 'file-data))
 685
 686 (defun stgit-patches-marked-or-at-point ()
 687   "Return the symbols of the marked patches, or the patch on the current line."
 688   (if stgit-marked-patches
 689       stgit-marked-patches
 690     (let ((patch (stgit-patch-name-at-point)))
 691       (if patch
 692           (list patch)
 693         '()))))
 694
 695 (defun stgit-goto-patch (patchsym)
 696   "Move point to the line containing patch PATCHSYM.
 697 If that patch cannot be found, do nothing."
 698   (let ((node (ewoc-nth stgit-ewoc 0)))
 699     (while (and node (not (eq (stgit-patch-name (ewoc-data node))
 700                               patchsym)))
 701       (setq node (ewoc-next stgit-ewoc node)))
 702     (when node
 703       (ewoc-goto-node stgit-ewoc node)
 704       (move-to-column goal-column))))
 705
 706 (defun stgit-init ()
 707   "Run stg init."
 708   (interactive)
 709   (stgit-capture-output nil
 710     (stgit-run "init"))
 711   (stgit-reload))
 712
 713 (defun stgit-mark ()
 714   "Mark the patch under point."
 715   (interactive)
 716   (let* ((node (ewoc-locate stgit-ewoc))
 717          (patch (ewoc-data node)))
 718     (stgit-add-mark (stgit-patch-name patch))
 719     (ewoc-invalidate stgit-ewoc node))
 720   (stgit-next-patch))
 721
 722 (defun stgit-unmark-up ()
 723   "Remove mark from the patch on the previous line."
 724   (interactive)
 725   (stgit-previous-patch)
 726   (let* ((node (ewoc-locate stgit-ewoc))
 727          (patch (ewoc-data node)))
 728     (stgit-remove-mark (stgit-patch-name patch))
 729     (ewoc-invalidate stgit-ewoc node))
 730   (move-to-column (stgit-goal-column)))
 731
 732 (defun stgit-unmark-down ()
 733   "Remove mark from the patch on the current line."
 734   (interactive)
 735   (let* ((node (ewoc-locate stgit-ewoc))
 736          (patch (ewoc-data node)))
 737     (stgit-remove-mark (stgit-patch-name patch))
 738     (ewoc-invalidate stgit-ewoc node))
 739   (stgit-next-patch))
 740
 741 (defun stgit-rename (name)
 742   "Rename the patch under point to NAME."
 743   (interactive (list (read-string "Patch name: "
 744                                   (symbol-name (stgit-patch-name-at-point t)))))
 745   (let ((old-patchsym (stgit-patch-name-at-point t)))
 746     (stgit-capture-output nil
 747       (stgit-run "rename" old-patchsym name))
 748     (let ((name-sym (intern name)))
 749       (when (memq old-patchsym stgit-expanded-patches)
 750         (setq stgit-expanded-patches
 751             (cons name-sym (delq old-patchsym stgit-expanded-patches))))
 752       (when (memq old-patchsym stgit-marked-patches)
 753         (setq stgit-marked-patches
 754             (cons name-sym (delq old-patchsym stgit-marked-patches))))
 755       (stgit-reload)
 756       (stgit-goto-patch name-sym))))
 757
 758 (defun stgit-repair ()
 759   "Run stg repair."
 760   (interactive)
 761   (stgit-capture-output nil
 762     (stgit-run "repair"))
 763   (stgit-reload))
 764
 765 (defun stgit-available-branches ()
 766   "Returns a list of the available stg branches"
 767   (let ((output (with-output-to-string
 768                   (stgit-run "branch" "--list")))
 769         (start 0)
 770         result)
 771     (while (string-match "^>?\\s-+s\\s-+\\(\\S-+\\)" output start)
 772       (setq result (cons (match-string 1 output) result))
 773       (setq start (match-end 0)))
 774     result))
 775
 776 (defun stgit-branch (branch)
 777   "Switch to branch BRANCH."
 778   (interactive (list (completing-read "Switch to branch: "
 779                                       (stgit-available-branches))))
 780   (stgit-capture-output nil (stgit-run "branch" "--" branch))
 781   (stgit-reload))
 782
 783 (defun stgit-commit (count)
 784   "Run stg commit on COUNT commits.
 785 Interactively, the prefix argument is used as COUNT."
 786   (interactive "p")
 787   (stgit-capture-output nil (stgit-run "commit" "-n" count))
 788   (stgit-reload))
 789
 790 (defun stgit-uncommit (count)
 791   "Run stg uncommit on COUNT commits.
 792 Interactively, the prefix argument is used as COUNT."
 793   (interactive "p")
 794   (stgit-capture-output nil (stgit-run "uncommit" "-n" count))
 795   (stgit-reload))
 796
 797 (defun stgit-push-next (npatches)
 798   "Push the first unapplied patch.
 799 With numeric prefix argument, push that many patches."
 800   (interactive "p")
 801   (stgit-capture-output nil (stgit-run "push" "-n" npatches))
 802   (stgit-reload)
 803   (stgit-refresh-git-status))
 804
 805 (defun stgit-pop-next (npatches)
 806   "Pop the topmost applied patch.
 807 With numeric prefix argument, pop that many patches."
 808   (interactive "p")
 809   (stgit-capture-output nil (stgit-run "pop" "-n" npatches))
 810   (stgit-reload)
 811   (stgit-refresh-git-status))
 812
 813 (defun stgit-applied-at-point ()
 814   "Is the patch on the current line applied?"
 815   (save-excursion
 816     (beginning-of-line)
 817     (looking-at "[>+]")))
 818
 819 (defun stgit-push-or-pop ()
 820   "Push or pop the patch on the current line."
 821   (interactive)
 822   (let ((patchsym (stgit-patch-name-at-point t))
 823         (applied (stgit-applied-at-point)))
 824     (stgit-capture-output nil
 825       (stgit-run (if applied "pop" "push") patchsym))
 826     (stgit-reload)))
 827
 828 (defun stgit-goto ()
 829   "Go to the patch on the current line."
 830   (interactive)
 831   (let ((patchsym (stgit-patch-name-at-point t)))
 832     (stgit-capture-output nil
 833       (stgit-run "goto" patchsym))
 834     (stgit-reload)))
 835
 836 (defun stgit-id (patchsym)
 837   "Return the git commit id for PATCHSYM.
 838 If PATCHSYM is a keyword, returns PATCHSYM unmodified."
 839   (if (keywordp patchsym)
 840       patchsym
 841     (let ((result (with-output-to-string
 842                     (stgit-run-silent "id" patchsym))))
 843       (unless (string-match "^\\([0-9A-Fa-f]\\{40\\}\\)$" result)
 844         (error "Cannot find commit id for %s" patchsym))
 845       (match-string 1 result))))
 846
 847 (defun stgit-show ()
 848   "Show the patch on the current line."
 849   (interactive)
 850   (stgit-capture-output "*StGit patch*"
 851     (case (get-text-property (point) 'entry-type)
 852       ('file
 853        (let* ((patched-file (stgit-patched-file-at-point))
 854               (patch-name (stgit-patch-name-at-point))
 855               (patch-id (stgit-id patch-name))
 856               (args (append (and (stgit-file-cr-from patched-file)
 857                                  (if stgit-expand-find-copies-harder
 858                                      '("--find-copies-harder")
 859                                    '("-C")))
 860                             (cond ((eq patch-id :index)
 861                                    '("--cached"))
 862                                   ((eq patch-id :work)
 863                                    nil)
 864                                   (t
 865                                    (list (concat patch-id "^") patch-id)))
 866                             '("--")
 867                               (if (stgit-file-copy-or-rename patched-file)
 868                                   (list (stgit-file-cr-from patched-file)
 869                                         (stgit-file-cr-to patched-file))
 870                                 (list (stgit-file-file patched-file))))))
 871          (apply 'stgit-run-git "diff" args)))
 872       ('patch
 873        (stgit-run "show" "-O" "--patch-with-stat" "-O" "-M"
 874                   (stgit-patch-name-at-point)))
 875       (t
 876        (error "No patch or file at point")))
 877     (with-current-buffer standard-output
 878       (goto-char (point-min))
 879       (diff-mode))))
 880
 881 (defun stgit-move-change-to-index (file status)
 882   "Copies the workspace state of FILE to index, using git add or git rm"
 883   (let ((op (if (file-exists-p file) "add" "rm")))
 884     (stgit-capture-output "*git output*"
 885       (stgit-run-git op "--" file))))
 886
 887 (defun stgit-remove-change-from-index (file status)
 888   "Unstages the change in FILE from the index"
 889   (stgit-capture-output "*git output*"
 890     (stgit-run-git "reset" "-q" "--" file)))
 891
 892 (defun stgit-file-toggle-index ()
 893   "Move modified file in or out of the index."
 894   (interactive)
 895   (let ((patched-file (stgit-patched-file-at-point)))
 896     (unless patched-file
 897       (error "No file on the current line"))
 898     (let ((patch-name (stgit-patch-name-at-point)))
 899       (cond ((eq patch-name :work)
 900              (stgit-move-change-to-index (stgit-file-file patched-file)
 901                                          (stgit-file-status patched-file)))
 902             ((eq patch-name :index)
 903              (stgit-remove-change-from-index (stgit-file-file patched-file)
 904                                              (stgit-file-status patched-file)))
 905             (t
 906              (error "Can only move files in the working tree to index")))))
 907   (stgit-refresh-worktree)
 908   (stgit-refresh-index))
 909
 910 (defun stgit-edit ()
 911   "Edit the patch on the current line."
 912   (interactive)
 913   (let ((patchsym (stgit-patch-name-at-point t))
 914         (edit-buf (get-buffer-create "*StGit edit*"))
 915         (dir default-directory))
 916     (log-edit 'stgit-confirm-edit t nil edit-buf)
 917     (set (make-local-variable 'stgit-edit-patchsym) patchsym)
 918     (setq default-directory dir)
 919     (let ((standard-output edit-buf))
 920       (stgit-run-silent "edit" "--save-template=-" patchsym))))
 921
 922 (defun stgit-confirm-edit ()
 923   (interactive)
 924   (let ((file (make-temp-file "stgit-edit-")))
 925     (write-region (point-min) (point-max) file)
 926     (stgit-capture-output nil
 927       (stgit-run "edit" "-f" file stgit-edit-patchsym))
 928     (with-current-buffer log-edit-parent-buffer
 929       (stgit-reload))))
 930
 931 (defun stgit-new (add-sign)
 932   "Create a new patch.
 933 With a prefix argument, include a \"Signed-off-by:\" line at the
 934 end of the patch."
 935   (interactive "P")
 936   (let ((edit-buf (get-buffer-create "*StGit edit*"))
 937         (dir default-directory))
 938     (log-edit 'stgit-confirm-new t nil edit-buf)
 939     (setq default-directory dir)
 940     (when add-sign
 941       (save-excursion
 942         (let ((standard-output (current-buffer)))
 943           (stgit-run-silent "new" "--sign" "--save-template=-"))))))
 944
 945 (defun stgit-confirm-new ()
 946   (interactive)
 947   (let ((file (make-temp-file "stgit-edit-")))
 948     (write-region (point-min) (point-max) file)
 949     (stgit-capture-output nil
 950       (stgit-run "new" "-f" file))
 951     (with-current-buffer log-edit-parent-buffer
 952       (stgit-reload))))
 953
 954 (defun stgit-create-patch-name (description)
 955   "Create a patch name from a long description"
 956   (let ((patch ""))
 957     (while (> (length description) 0)
 958       (cond ((string-match "\\`[a-zA-Z_-]+" description)
 959              (setq patch (downcase (concat patch
 960                                            (match-string 0 description))))
 961              (setq description (substring description (match-end 0))))
 962             ((string-match "\\` +" description)
 963              (setq patch (concat patch "-"))
 964              (setq description (substring description (match-end 0))))
 965             ((string-match "\\`[^a-zA-Z_-]+" description)
 966              (setq description (substring description (match-end 0))))))
 967     (cond ((= (length patch) 0)
 968            "patch")
 969           ((> (length patch) 20)
 970            (substring patch 0 20))
 971           (t patch))))
 972
 973 (defun stgit-delete (patchsyms &optional spill-p)
 974   "Delete the patches in PATCHSYMS.
 975 Interactively, delete the marked patches, or the patch at point.
 976
 977 With a prefix argument, or SPILL-P, spill the patch contents to
 978 the work tree and index."
 979   (interactive (list (stgit-patches-marked-or-at-point)
 980                      current-prefix-arg))
 981   (unless patchsyms
 982     (error "No patches to delete"))
 983   (let ((npatches (length patchsyms)))
 984     (when (yes-or-no-p (format "Really delete %d patch%s%s? "
 985                                npatches
 986                                (if (= 1 npatches) "" "es")
 987                                (if spill-p
 988                                    " (spilling contents to index)"
 989                                  "")))
 990       (let ((args (if spill-p
 991                       (cons "--spill" patchsyms)
 992                     patchsyms)))
 993         (stgit-capture-output nil
 994           (apply 'stgit-run "delete" args))
 995         (stgit-reload)))))
 996
 997 (defun stgit-move-patches-target ()
 998   "Return the patchsym indicating a target patch for
 999 `stgit-move-patches'.
1000
1001 This is either the patch at point, or one of :top and :bottom, if
1002 the point is after or before the applied patches."
1003
1004   (let ((patchsym (stgit-patch-name-at-point)))
1005     (cond (patchsym patchsym)
1006           ((save-excursion (re-search-backward "^>" nil t)) :top)
1007           (t :bottom))))
1008
1009 (defun stgit-sort-patches (patchsyms)
1010   "Returns the list of patches in PATCHSYMS sorted according to
1011 their position in the patch series, bottommost first.
1012
1013 PATCHSYMS may not contain duplicate entries."
1014   (let (sorted-patchsyms
1015         (series (with-output-to-string
1016                   (with-current-buffer standard-output
1017                     (stgit-run-silent "series" "--noprefix"))))
1018         start)
1019     (while (string-match "^\\(.+\\)" series start)
1020       (let ((patchsym (intern (match-string 1 series))))
1021         (when (memq patchsym patchsyms)
1022           (setq sorted-patchsyms (cons patchsym sorted-patchsyms))))
1023       (setq start (match-end 0)))
1024     (setq sorted-patchsyms (nreverse sorted-patchsyms))
1025
1026     (unless (= (length patchsyms) (length sorted-patchsyms))
1027       (error "Internal error"))
1028
1029     sorted-patchsyms))
1030
1031 (defun stgit-move-patches (patchsyms target-patch)
1032   "Move the patches in PATCHSYMS to below TARGET-PATCH.
1033 If TARGET-PATCH is :bottom or :top, move the patches to the
1034 bottom or top of the stack, respectively.
1035
1036 Interactively, move the marked patches to where the point is."
1037   (interactive (list stgit-marked-patches
1038                      (stgit-move-patches-target)))
1039   (unless patchsyms
1040     (error "Need at least one patch to move"))
1041
1042   (unless target-patch
1043     (error "Point not at a patch"))
1044
1045   (if (eq target-patch :top)
1046       (stgit-capture-output nil
1047         (apply 'stgit-run "float" patchsyms))
1048
1049     ;; need to have patchsyms sorted by position in the stack
1050     (let ((sorted-patchsyms (stgit-sort-patches patchsyms)))
1051       (while sorted-patchsyms
1052         (setq sorted-patchsyms
1053               (and (stgit-capture-output nil
1054                      (if (eq target-patch :bottom)
1055                          (stgit-run "sink" "--" (car sorted-patchsyms))
1056                        (stgit-run "sink" "--to" target-patch "--"
1057                                   (car sorted-patchsyms))))
1058                    (cdr sorted-patchsyms))))))
1059   (stgit-reload))
1060
1061 (defun stgit-squash (patchsyms)
1062   "Squash the patches in PATCHSYMS.
1063 Interactively, squash the marked patches.
1064
1065 Unless there are any conflicts, the patches will be merged into
1066 one patch, which will occupy the same spot in the series as the
1067 deepest patch had before the squash."
1068   (interactive (list stgit-marked-patches))
1069   (when (< (length patchsyms) 2)
1070     (error "Need at least two patches to squash"))
1071   (let ((stgit-buffer (current-buffer))
1072         (edit-buf (get-buffer-create "*StGit edit*"))
1073         (dir default-directory)
1074         (sorted-patchsyms (stgit-sort-patches patchsyms)))
1075     (log-edit 'stgit-confirm-squash t nil edit-buf)
1076     (set (make-local-variable 'stgit-patchsyms) sorted-patchsyms)
1077     (setq default-directory dir)
1078     (let ((result (let ((standard-output edit-buf))
1079                     (apply 'stgit-run-silent "squash"
1080                            "--save-template=-" sorted-patchsyms))))
1081
1082       ;; stg squash may have reordered the patches or caused conflicts
1083       (with-current-buffer stgit-buffer
1084         (stgit-reload))
1085
1086       (unless (eq 0 result)
1087         (fundamental-mode)
1088         (rename-buffer "*StGit error*")
1089         (resize-temp-buffer-window)
1090         (switch-to-buffer-other-window stgit-buffer)
1091         (error "stg squash failed")))))
1092
1093 (defun stgit-confirm-squash ()
1094   (interactive)
1095   (let ((file (make-temp-file "stgit-edit-")))
1096     (write-region (point-min) (point-max) file)
1097     (stgit-capture-output nil
1098       (apply 'stgit-run "squash" "-f" file stgit-patchsyms))
1099     (with-current-buffer log-edit-parent-buffer
1100       (stgit-clear-marks)
1101       ;; Go to first marked patch and stay there
1102       (goto-char (point-min))
1103       (re-search-forward (concat "^[>+-]\\*") nil t)
1104       (move-to-column goal-column)
1105       (let ((pos (point)))
1106         (stgit-reload)
1107         (goto-char pos)))))
1108
1109 (defun stgit-help ()
1110   "Display help for the StGit mode."
1111   (interactive)
1112   (describe-function 'stgit-mode))
1113
1114 (defun stgit-undo (&optional arg)
1115   "Run stg undo.
1116 With prefix argument, run it with the --hard flag."
1117   (interactive "P")
1118   (stgit-capture-output nil
1119     (if arg
1120         (stgit-run "undo" "--hard")
1121       (stgit-run "undo")))
1122   (stgit-reload))
1123
1124 (defun stgit-refresh (&optional arg)
1125   "Run stg refresh.
1126 With prefix argument, refresh the marked patch or the patch under point."
1127   (interactive "P")
1128   (let ((patchargs (if arg
1129                        (let ((patches (stgit-patches-marked-or-at-point)))
1130                          (cond ((null patches)
1131                                 (error "No patch to update"))
1132                                ((> (length patches) 1)
1133                                 (error "Too many patches selected"))
1134                                (t
1135                                 (cons "-p" patches))))
1136                      nil)))
1137     (stgit-capture-output nil
1138       (apply 'stgit-run "refresh" patchargs))
1139     (stgit-refresh-git-status))
1140   (stgit-reload))
1141
1142 (defcustom stgit-default-show-worktree
1143   nil
1144   "Set to non-nil to by default show the working tree in a new stgit buffer.
1145
1146 This value is used as the default value for `stgit-show-worktree'."
1147   :type 'boolean
1148   :group 'stgit)
1149
1150 (defvar stgit-show-worktree nil
1151   "Show work tree and index in the stgit buffer.
1152
1153 See `stgit-default-show-worktree' for its default value.")
1154
1155 (defun stgit-toggle-worktree (&optional arg)
1156   "Toggle the visibility of the work tree.
1157 With arg, show the work tree if arg is positive.
1158
1159 Its initial setting is controlled by `stgit-default-show-worktree'."
1160   (interactive)
1161   (setq stgit-show-worktree
1162         (if (numberp arg)
1163             (> arg 0)
1164           (not stgit-show-worktree)))
1165   (stgit-reload))
1166
1167 (provide 'stgit)