need to clear the free space and speed indicators if there's a disconnect due to error.

1 files changed, 8 insertions, 3 deletions

diff --git a/src/trg-status-bar.c b/src/trg-status-bar.c
index 7df8069..594ef16 100644
--- a/src/trg-status-bar.c
+++ b/src/trg-status-bar.c
@@ -42,15 +42,20 @@ static void trg_status_bar_class_init(TrgStatusBarClass * klass)
     g_type_class_add_private(klass, sizeof(TrgStatusBarPrivate));
 }
 
-void trg_status_bar_reset(TrgStatusBar *sb)
+void trg_status_bar_clear_indicators(TrgStatusBar *sb)
 {
     TrgStatusBarPrivate *priv = TRG_STATUS_BAR_GET_PRIVATE(sb);
-
-    gtk_label_set_text(GTK_LABEL(priv->info_lbl), _("Disconnected"));
     gtk_label_set_text(GTK_LABEL(priv->free_lbl), "");
     gtk_label_set_text(GTK_LABEL(priv->speed_lbl), "");
 }
 
+void trg_status_bar_reset(TrgStatusBar *sb)
+{
+    TrgStatusBarPrivate *priv = TRG_STATUS_BAR_GET_PRIVATE(sb);
+    trg_status_bar_clear_indicators(sb);
+    gtk_label_set_text(GTK_LABEL(priv->info_lbl), _("Disconnected"));
+}
+
 static void trg_status_bar_init(TrgStatusBar * self)
 {
     TrgStatusBarPrivate *priv = TRG_STATUS_BAR_GET_PRIVATE(self);