Use magit-revert-buffer-hook

Closes #19
References magit/magit#603

diff --git a/diff-hl.el b/diff-hl.el
index 4b25fc7f26030d48e4f8d5612d0c43497f222422..450ee9b40cd433bb57262dd9d8a358e822afd3b5 100644 (file)
--- a/diff-hl.el
+++ b/diff-hl.el
@@ -429,12 +429,14 @@ in the source file, or the last line of the hunk above it."
           (add-hook 'find-file-hook 'diff-hl-update t t))\r
         (add-hook 'vc-checkin-hook 'diff-hl-update nil t)\r
         (add-hook 'after-revert-hook 'diff-hl-update nil t)\r
+        (add-hook 'magit-revert-buffer-hook 'diff-hl-update nil t)\r
         (add-hook 'text-scale-mode-hook 'diff-hl-define-bitmaps nil t))\r
     (remove-hook 'after-save-hook 'diff-hl-update t)\r
     (remove-hook 'after-change-functions 'diff-hl-edit t)\r
     (remove-hook 'find-file-hook 'diff-hl-update t)\r
     (remove-hook 'vc-checkin-hook 'diff-hl-update t)\r
     (remove-hook 'after-revert-hook 'diff-hl-update t)\r
+    (remove-hook 'magit-revert-buffer-hook 'diff-hl-update t)\r
     (remove-hook 'text-scale-mode-hook 'diff-hl-define-bitmaps t)\r
     (diff-hl-remove-overlays)))\r
 \r