git add, git reset and git stash implementation added.

1 files changed, 17 insertions, 1 deletions

diff --git a/pkg/gui/statusview.go b/pkg/gui/statusview.go
index b1873e8..fbc90ee 100644
--- a/pkg/gui/statusview.go
+++ b/pkg/gui/statusview.go
@@ -119,6 +119,13 @@ func (gui *Gui) closeStatusView(g *gocui.Gui, v *gocui.View) error {
 	if _, err := g.SetCurrentView(mainViewFeature.Name); err != nil {
 		return err
 	}
+	entity := gui.getSelectedRepository()
+	if err := gui.refreshMain(g); err != nil {
+		return err
+	}
+	if err := gui.refreshViews(g, entity); err != nil {
+		return err
+	}
 	gui.updateKeyBindingsView(g, mainViewFeature.Name)
 	return nil
 }
@@ -129,7 +136,7 @@ func generateFileLists(entity *git.RepoEntity) (staged, unstaged []*git.File, er
 		return nil, nil, err
 	}
 	for _, file := range files {
-		if file.X != git.StatusNotupdated && file.X != git.StatusUntracked && file.X != git.StatusIgnored {
+		if file.X != git.StatusNotupdated && file.X != git.StatusUntracked && file.X != git.StatusIgnored && file.X != git.StatusUpdated {
 			staged = append(staged, file)
 		}
 		if file.Y != git.StatusNotupdated {
@@ -156,3 +163,12 @@ func refreshStatusView(viewName string, g *gocui.Gui, entity *git.RepoEntity) er
 	}
 	return nil
 }
+
+func refreshAllStatusView(g *gocui.Gui, entity *git.RepoEntity) error {
+	for _, v := range statusViews {
+		if err := refreshStatusView(v.Name, g, entity); err != nil {
+			return err
+		}
+	}
+	return nil
+}