From 6eb70b7d7ce76a7c004d8f915d5489433fa4ba73 Mon Sep 17 00:00:00 2001
From: cwilvx <geoffreymungai45@gmail.com>
Date: Tue, 22 Apr 2025 09:37:11 +0300
Subject: [PATCH] add item counts

---
 app/api/folder.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/app/api/folder.py b/app/api/folder.py
index 1756d1d0..601e710f 100644
--- a/app/api/folder.py
+++ b/app/api/folder.py
@@ -123,15 +123,13 @@ def get_folder_tree(body: FolderTree):
             reverse=True,
         )
 
-        for playlist in playlists:
-            playlist.clear_lists()
-
         return {
             "path": req_dir,
             "folders": [
                 {
                     "name": p.name,
                     "path": f"$playlist/{p.id}",
+                    "trackcount": p.count,
                 }
                 for p in playlists
             ],
@@ -172,11 +170,13 @@ def get_folder_tree(body: FolderTree):
         playlists_item = {
             "name": "Playlists",
             "path": "$playlists",
+            "trackcount": sum(p.count for p in PlaylistTable.get_all()),
         }
 
         favorites_item = {
             "name": "Favorites",
             "path": "$favorites",
+            "trackcount": FavoritesTable.get_fav_tracks(0, -1)[1],
         }
 
         results["folders"].insert(0, playlists_item)