From 965fa79bbfaacdf24e2bf91b08540a82c90a4772 Mon Sep 17 00:00:00 2001
From: Zlatin Balevsky <zlatinb@gmail.com>
Date: Mon, 14 Oct 2019 20:57:50 +0100
Subject: [PATCH] fix count of shared files in tree view mode

---
 gui/griffon-app/models/com/muwire/gui/MainFrameModel.groovy | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/gui/griffon-app/models/com/muwire/gui/MainFrameModel.groovy b/gui/griffon-app/models/com/muwire/gui/MainFrameModel.groovy
index 7587f979..e5fd6ded 100644
--- a/gui/griffon-app/models/com/muwire/gui/MainFrameModel.groovy
+++ b/gui/griffon-app/models/com/muwire/gui/MainFrameModel.groovy
@@ -338,6 +338,7 @@ class MainFrameModel {
                 table.model.fireTableDataChanged()
             } else {
                 insertIntoTree(e.sharedFile)
+                loadedFiles = fileToNode.size()
             }
         }
     }
@@ -351,6 +352,7 @@ class MainFrameModel {
                 table.model.fireTableDataChanged()
             } else {
                 insertIntoTree(e.loadedFile)
+                loadedFiles = fileToNode.size()
             }
         }
     }
@@ -523,6 +525,7 @@ class MainFrameModel {
                 table.model.fireTableDataChanged()
             } else {
                 insertIntoTree(e.downloadedFile)
+                loadedFiles = fileToNode.size()
             }
         }
     }
@@ -550,9 +553,8 @@ class MainFrameModel {
         }
         
         def dmtn = new DefaultMutableTreeNode(file)
-        fileToNode[file] = dmtn
+        fileToNode.put(file, dmtn)
         node.add(dmtn)
-        loadedFiles = fileToNode.size()
         view.refreshSharedFiles()
     }