From ec2f8ec7967109eb9b83a8250b6d0a34990104f1 Mon Sep 17 00:00:00 2001
From: zadam <adam.zivner@gmail.com>
Date: Sun, 14 Mar 2021 20:47:33 +0100
Subject: [PATCH] added subtree note count into info widget

---
 .../app/widgets/collapsible_widgets/note_info.js     | 12 +++++++++---
 src/routes/api/stats.js                              |  3 ++-
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/src/public/app/widgets/collapsible_widgets/note_info.js b/src/public/app/widgets/collapsible_widgets/note_info.js
index c204590fe..8c677c4a1 100644
--- a/src/public/app/widgets/collapsible_widgets/note_info.js
+++ b/src/public/app/widgets/collapsible_widgets/note_info.js
@@ -50,7 +50,7 @@ const TPL = `
             <span class="note-sizes-wrapper">
                 <span class="note-size"></span>
                 
-                (subtree size: <span class="subtree-size"></span>)
+                <span class="subtree-size"></span>
             </span>
         </td>
     </tr>
@@ -88,8 +88,14 @@ export default class NoteInfoWidget extends CollapsibleWidget {
             const noteSizeResp = await server.get(`stats/note-size/${this.noteId}`);
             this.$noteSize.text(this.formatSize(noteSizeResp.noteSize));
 
-            const subTreeSizeResp = await server.get(`stats/subtree-size/${this.noteId}`);
-            this.$subTreeSize.text(this.formatSize(subTreeSizeResp.subTreeSize));
+            const subTreeResp = await server.get(`stats/subtree-size/${this.noteId}`);
+
+            if (subTreeResp.subTreeNoteCount > 1) {
+                this.$subTreeSize.text("(subtree size: " + this.formatSize(subTreeResp.subTreeSize) + ` in ${subTreeResp.subTreeNoteCount} notes)`);
+            }
+            else {
+                this.$subTreeSize.text("");
+            }
         });
     }
 
diff --git a/src/routes/api/stats.js b/src/routes/api/stats.js
index 610acd4cc..1bd030367 100644
--- a/src/routes/api/stats.js
+++ b/src/routes/api/stats.js
@@ -50,7 +50,8 @@ function getSubtreeSize(req) {
             )`);
 
     return {
-        subTreeSize
+        subTreeSize,
+        subTreeNoteCount: subTreeNoteIds.length
     };
 }