From 047742a729f8f4f8f4e9c85bf475f75b16f4f7b7 Mon Sep 17 00:00:00 2001 From: in1tiate <32779090+in1tiate@users.noreply.github.com> Date: Mon, 13 Jan 2025 17:28:56 -0600 Subject: [PATCH] Add the ability to "favorite" songs and pin them to the the top of the songlist (#1066) * add song favoriting * remove incorrectly placed sort() * store as qstringlist instead of using keys --- src/aoapplication.h | 3 ++ src/courtroom.cpp | 70 ++++++++++++++++++++++++++++++++++++- src/courtroom.h | 2 ++ src/packet_distribution.cpp | 6 ++-- 4 files changed, 77 insertions(+), 4 deletions(-) diff --git a/src/aoapplication.h b/src/aoapplication.h index 6022ca1..ebcdab9 100644 --- a/src/aoapplication.h +++ b/src/aoapplication.h @@ -321,6 +321,9 @@ public: const QString default_theme = "default"; // don't change this!!! don't do it!!! + // The name of the currently connected server. + QString server_name; + // The file name of the log file in base/logs. QString log_filename; diff --git a/src/courtroom.cpp b/src/courtroom.cpp index 349e5a5..b085f67 100644 --- a/src/courtroom.cpp +++ b/src/courtroom.cpp @@ -150,8 +150,9 @@ Courtroom::Courtroom(AOApplication *p_ao_app) ui_area_list->setObjectName("ui_area_list"); ui_music_list = new QTreeWidget(this); - ui_music_list->setColumnCount(2); + ui_music_list->setColumnCount(3); ui_music_list->hideColumn(1); + ui_music_list->hideColumn(2); ui_music_list->setHeaderHidden(true); ui_music_list->header()->setStretchLastSection(false); ui_music_list->header()->setSectionResizeMode(QHeaderView::ResizeToContents); @@ -1703,6 +1704,40 @@ void Courtroom::list_music() QBrush missing_brush(ao_app->get_color("missing_song_color", f_file)); QTreeWidgetItem *parent = nullptr; + + // Handle favorites first so they're at the top of the list + QSettings favorite_songs_ini(get_base_path() + "favorite_songs.ini", QSettings::IniFormat); + const QStringList &favorite_songs = favorite_songs_ini.value(ao_app->server_name).toStringList(); + if (!favorite_songs.isEmpty()) + { + QTreeWidgetItem *favCategory; + favCategory = new QTreeWidgetItem(ui_music_list); + favCategory->setText(0, tr("== FAVORITES ==")); + favCategory->setText(1, tr("== FAVORITES ==")); + favCategory->setText(2, "1"); + favCategory->setBackground(0, missing_brush); + for (const QString &song : favorite_songs) + { + QTreeWidgetItem *favSong = new QTreeWidgetItem(favCategory); + QString f_song_listname = song.left(song.lastIndexOf(".")); + + favSong->setText(0, f_song_listname); + favSong->setText(1, song); + favSong->setText(2, "1"); + + QString song_path = ao_app->get_real_path(ao_app->get_music_path(song)); + + if (file_exists(song_path)) + { + favSong->setBackground(0, found_brush); + } + else + { + favSong->setBackground(0, missing_brush); + } + } + } + for (int n_song = 0; n_song < music_list.size(); ++n_song) { QString i_song = music_list.at(n_song); @@ -1730,6 +1765,7 @@ void Courtroom::list_music() } treeItem->setText(0, i_song_listname); treeItem->setText(1, i_song); + treeItem->setText(2, "0"); QString song_path = ao_app->get_real_path(ao_app->get_music_path(i_song)); @@ -5749,6 +5785,18 @@ void Courtroom::on_music_list_context_menu_requested(const QPoint &pos) menu->addAction(QString(tr("Collapse All Categories")), this, &Courtroom::music_list_collapse_all); menu->addSeparator(); + QTreeWidgetItem *current_song = ui_music_list->currentItem(); + if (ui_music_list->currentItem()->text(2) == "1") + { + menu->addAction(QString(tr("Remove Favorite")), this, [this, current_song] { Courtroom::remove_favorite_song(current_song); }); + menu->addSeparator(); + } + else + { + menu->addAction(QString(tr("Add Favorite")), this, [this, current_song] { Courtroom::add_favorite_song(current_song); }); + menu->addSeparator(); + } + menu->addAction(new QAction(tr("Fade Out Previous"), this)); menu->actions().back()->setCheckable(true); menu->actions().back()->setChecked(music_flags & FADE_OUT); @@ -5777,6 +5825,26 @@ void Courtroom::on_music_list_context_menu_requested(const QPoint &pos) menu->popup(ui_music_list->mapToGlobal(pos)); } +void Courtroom::add_favorite_song(QTreeWidgetItem *p_item) +{ + QSettings favorite_songs_ini(get_base_path() + "favorite_songs.ini", QSettings::IniFormat); + QStringList favorite_songs = favorite_songs_ini.value(ao_app->server_name).toStringList(); + favorite_songs.append(p_item->text(1)); + + favorite_songs_ini.setValue(ao_app->server_name, favorite_songs); + list_music(); +} + +void Courtroom::remove_favorite_song(QTreeWidgetItem *p_item) +{ + QSettings favorite_songs_ini(get_base_path() + "favorite_songs.ini", QSettings::IniFormat); + QStringList favorite_songs = favorite_songs_ini.value(ao_app->server_name).toStringList(); + favorite_songs.removeAll(p_item->text(1)); + + favorite_songs_ini.setValue(ao_app->server_name, favorite_songs); + list_music(); +} + void Courtroom::music_fade_out(bool toggle) { if (toggle) diff --git a/src/courtroom.h b/src/courtroom.h index 6cae0ff..7c075b9 100644 --- a/src/courtroom.h +++ b/src/courtroom.h @@ -834,6 +834,8 @@ private Q_SLOTS: void on_music_search_edited(QString p_text); void on_music_list_double_clicked(QTreeWidgetItem *p_item, int column); void on_music_list_context_menu_requested(const QPoint &pos); + void add_favorite_song(QTreeWidgetItem *p_item); + void remove_favorite_song(QTreeWidgetItem *p_item); void music_fade_out(bool toggle); void music_fade_in(bool toggle); void music_synchronize(bool toggle); diff --git a/src/packet_distribution.cpp b/src/packet_distribution.cpp index 6cf5c74..babdae2 100644 --- a/src/packet_distribution.cpp +++ b/src/packet_distribution.cpp @@ -120,7 +120,6 @@ void AOApplication::server_packet_received(AOPacket packet) int selected_server = w_lobby->get_selected_server(); QString server_address; - QString server_name; switch (w_lobby->pageSelected()) { case 0: @@ -161,10 +160,11 @@ void AOApplication::server_packet_received(AOPacket packet) // Remove any characters not accepted in folder names for the server_name // here + QString server_name_stripped = server_name; if (Options::getInstance().logToDemoFileEnabled() && server_name != "Demo playback") { - this->log_filename = QDateTime::currentDateTime().toUTC().toString("'logs/" + server_name.remove(QRegularExpression("[\\\\/:*?\"<>|\']")) + "/'yyyy-MM-dd hh-mm-ss t'.log'"); - this->write_to_file("Joined server " + server_name + " hosted on address " + server_address + " on " + QDateTime::currentDateTime().toUTC().toString(), log_filename, true); + this->log_filename = QDateTime::currentDateTime().toUTC().toString("'logs/" + server_name_stripped.remove(QRegularExpression("[\\\\/:*?\"<>|\']")) + "/'yyyy-MM-dd hh-mm-ss t'.log'"); + this->write_to_file("Joined server " + server_name_stripped + " hosted on address " + server_address + " on " + QDateTime::currentDateTime().toUTC().toString(), log_filename, true); } else {