From 36c124f9e344335ff867bc723b02eacbff3225fb Mon Sep 17 00:00:00 2001
From: Stephan Dilly <dilly.stephan@gmail.com>
Date: Wed, 3 Mar 2021 18:40:00 +0100
Subject: [PATCH] refresh compare state at the right time (#564)

---
 src/tabs/status.rs | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/tabs/status.rs b/src/tabs/status.rs
index b9cf8afb..3009d39c 100644
--- a/src/tabs/status.rs
+++ b/src/tabs/status.rs
@@ -27,7 +27,7 @@ use tui::{
     widgets::Paragraph,
 };
 
-///
+/// what part of the screen is focused
 #[derive(PartialEq)]
 enum Focus {
     WorkDir,
@@ -35,7 +35,7 @@ enum Focus {
     Stage,
 }
 
-///
+/// which target are we showing a diff against
 #[derive(PartialEq, Copy, Clone)]
 enum DiffTarget {
     Stage,
@@ -317,6 +317,9 @@ impl Status {
         match ev {
             AsyncNotification::Diff => self.update_diff()?,
             AsyncNotification::Status => self.update_status()?,
+            AsyncNotification::Push
+            | AsyncNotification::Fetch
+            | AsyncNotification::CommitFiles => self.branch_compare(),
             _ => (),
         }