Add a timeout to regularly update disk usage information

https://bugzilla.gnome.org/show_bug.cgi?id=776272

2 files changed, 19 insertions, 1 deletions

diff --git a/src/baobab-location-list.vala b/src/baobab-location-list.vala
index d31eb10..63043c8 100644
--- a/src/baobab-location-list.vala
+++ b/src/baobab-location-list.vala
@@ -39,7 +39,11 @@ namespace Baobab {
 
         public LocationRow (Location l) {
             location = l;
+            update ();
+        }
 
+        public void update () {
+            location.get_fs_usage ();
             image.gicon = location.icon;
 
             var escaped = GLib.Markup.escape_text (location.name, -1);
@@ -111,9 +115,19 @@ namespace Baobab {
             remote_list_box.set_header_func (update_header);
             remote_list_box.row_activated.connect (row_activated);
 
+            Timeout.add_seconds(3, (() => {
+                update_existing ();
+                return Source.CONTINUE;
+            }));
+
             populate ();
         }
 
+        void update_existing () {
+            local_list_box.foreach ((widget) => { ((LocationRow)widget).update (); });
+            remote_list_box.foreach ((widget) => { ((LocationRow)widget).update (); });
+        }
+
         bool already_present (File file) {
             foreach (var l in locations) {
                 if (l.file != null && l.file.equal (file)) {
diff --git a/src/baobab-location.vala b/src/baobab-location.vala
index c14c9d8..085225d 100644
--- a/src/baobab-location.vala
+++ b/src/baobab-location.vala
@@ -200,7 +200,11 @@ namespace Baobab {
             }
         }
 
-        void get_fs_usage () {
+        public void get_fs_usage () {
+            if (file == null) {
+                return;
+            }
+
             size = null;
             used = null;
             reserved = null;