Add track count to Tracks item in sidebar

Author: NeuroXS

src/library/librarytablemodel.cpp

@@ -101,3 +101,8 @@ TrackModel::Capabilities LibraryTableModel::getCapabilities() const {
             Capability::Properties |
             Capability::Sorting;
 }
+
+void LibraryTableModel::select() {
+    BaseSqlTableModel::select();
+    emit updateTrackCount();
+}

src/library/librarytablemodel.h

@@ -16,4 +16,9 @@ class LibraryTableModel : public BaseSqlTableModel {
     // number of successful additions.
     int addTracks(const QModelIndex& index, const QList<QString>& locations) final;
     TrackModel::Capabilities getCapabilities() const final;
+
+    void select() override;
+
+  signals:
+    void updateTrackCount();
 };

src/library/mixxxlibraryfeature.cpp

@@ -32,7 +32,9 @@ MixxxLibraryFeature::MixxxLibraryFeature(Library* pLibrary,
           m_pLibraryTableModel(nullptr),
           m_pSidebarModel(make_parented<TreeItemModel>(this)),
           m_pMissingView(nullptr),
-          m_pHiddenView(nullptr) {
+          m_pHiddenView(nullptr),
+          m_pTracksTreeItem(nullptr),
+          m_trackCount{0} {
     QString idColumn = LIBRARYTABLE_ID;
     QStringList columns = {
             LIBRARYTABLE_ID,
@@ -114,7 +116,13 @@ MixxxLibraryFeature::MixxxLibraryFeature(Library* pLibrary,
             pLibrary->trackCollectionManager(),
             "mixxx.db.model.library");
 
+    connect(m_pLibraryTableModel,
+            &LibraryTableModel::updateTrackCount,
+            this,
+            &MixxxLibraryFeature::slotUpdateTrackCount);
+
     std::unique_ptr<TreeItem> pRootItem = TreeItem::newRoot(this);
+    m_pTracksTreeItem = pRootItem.get();
     pRootItem->appendChild(kMissingTitle);
     pRootItem->appendChild(kHiddenTitle);
 
@@ -149,7 +157,8 @@ void MixxxLibraryFeature::bindLibraryWidget(WLibrary* pLibraryWidget,
 }
 
 QVariant MixxxLibraryFeature::title() {
-    return tr("Tracks");
+    const QString title = tr("Tracks") + QStringLiteral(" (%1)").arg(m_trackCount);
+    return title;
 }
 
 TreeItemModel* MixxxLibraryFeature::sidebarModel() const {
@@ -183,6 +192,17 @@ void MixxxLibraryFeature::bindSidebarWidget(WLibrarySidebar* pSidebarWidget) {
 }
 #endif
 
+void MixxxLibraryFeature::slotUpdateTrackCount() {
+    m_trackCount = m_pLibraryTableModel->rowCount();
+
+    if (m_pTracksTreeItem && m_pSidebarModel) {
+        QModelIndex rootIndex = m_pSidebarModel->getRootIndex();
+        QModelIndex tracksIndex = m_pSidebarModel->index(
+            m_pTracksTreeItem->parentRow(), 0, rootIndex);
+        m_pSidebarModel->triggerRepaint(tracksIndex);
+    }
+}
+
 void MixxxLibraryFeature::activate() {
     //qDebug() << "MixxxLibraryFeature::activate()";
     emit saveModelState();

src/library/mixxxlibraryfeature.h

@@ -50,6 +50,7 @@ class MixxxLibraryFeature final : public LibraryFeature {
   public slots:
     void activate() override;
     void activateChild(const QModelIndex& index) override;
+    void slotUpdateTrackCount();
 #ifdef __ENGINEPRIME__
     void onRightClick(const QPoint& globalPos) override;
 #endif
@@ -74,6 +75,10 @@ class MixxxLibraryFeature final : public LibraryFeature {
     DlgMissing* m_pMissingView;
     DlgHidden* m_pHiddenView;
 
+    TreeItem* m_pTracksTreeItem;
+
+    int m_trackCount;
+
 #ifdef __ENGINEPRIME__
     parented_ptr<QAction> m_pExportLibraryAction;