ce71de167a27f0f1a049513072e720f2edbec9e6
[ncmpc-debian.git] / src / AlbumListPage.cxx
1 /* ncmpc (Ncurses MPD Client)
2  * (c) 2004-2018 The Music Player Daemon Project
3  * Project homepage: http://musicpd.org
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License as published by
7  * the Free Software Foundation; either version 2 of the License, or
8  * (at your option) any later version.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License along
16  * with this program; if not, write to the Free Software Foundation, Inc.,
17  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18  */
19
20 #include "AlbumListPage.hxx"
21 #include "screen_interface.hxx"
22 #include "screen_status.hxx"
23 #include "screen_find.hxx"
24 #include "screen_browser.hxx"
25 #include "screen.hxx"
26 #include "i18n.h"
27 #include "charset.hxx"
28 #include "mpdclient.hxx"
29
30 #include <glib.h>
31
32 #include <algorithm>
33
34 #include <assert.h>
35 #include <string.h>
36
37 #define BUFSIZE 1024
38
39 gcc_pure
40 static bool
41 CompareUTF8(const std::string &a, const std::string &b)
42 {
43         char *key1 = g_utf8_collate_key(a.c_str(), -1);
44         char *key2 = g_utf8_collate_key(b.c_str(), -1);
45         int n = strcmp(key1,key2);
46         g_free(key1);
47         g_free(key2);
48         return n < 0;
49 }
50
51 /* list_window callback */
52 static const char *
53 album_lw_callback(unsigned idx, void *data)
54 {
55         const auto &list = *(const std::vector<std::string> *)data;
56
57         assert(idx < list.size());
58
59         const char *str_utf8 = list[idx].c_str();
60
61         static char buf[BUFSIZE];
62         g_strlcpy(buf, Utf8ToLocale(str_utf8).c_str(), sizeof(buf));
63         return buf;
64 }
65
66 /* list_window callback */
67 static const char *
68 AlbumListCallback(unsigned idx, void *data)
69 {
70         const auto &list = *(const std::vector<std::string> *)data;
71
72         if (idx == 0)
73                 return "..";
74         else if (idx == list.size() + 1)
75                 return _("All tracks");
76
77         --idx;
78
79         return album_lw_callback(idx, data);
80 }
81
82 static void
83 recv_tag_values(struct mpd_connection *connection, enum mpd_tag_type tag,
84                 std::vector<std::string> &list)
85 {
86         struct mpd_pair *pair;
87
88         while ((pair = mpd_recv_pair_tag(connection, tag)) != nullptr) {
89                 list.emplace_back(pair->value);
90                 mpd_return_pair(connection, pair);
91         }
92 }
93
94 void
95 AlbumListPage::LoadAlbumList(struct mpdclient &c)
96 {
97         struct mpd_connection *connection = mpdclient_get_connection(&c);
98
99         album_list.clear();
100
101         if (connection != nullptr) {
102                 mpd_search_db_tags(connection, MPD_TAG_ALBUM);
103                 mpd_search_add_tag_constraint(connection,
104                                               MPD_OPERATOR_DEFAULT,
105                                               MPD_TAG_ARTIST,
106                                               artist.c_str());
107                 mpd_search_commit(connection);
108
109                 recv_tag_values(connection, MPD_TAG_ALBUM, album_list);
110
111                 mpdclient_finish_command(&c);
112         }
113
114         /* sort list */
115         std::sort(album_list.begin(), album_list.end(), CompareUTF8);
116         lw.SetLength(album_list.size() + 2);
117 }
118
119 void
120 AlbumListPage::Reload(struct mpdclient &c)
121 {
122         LoadAlbumList(c);
123 }
124
125 /**
126  * Paint one item in the album list.  There are two virtual items
127  * inserted: at the beginning, there's the special item ".." to go to
128  * the parent directory, and at the end, there's the item "All tracks"
129  * to view the tracks of all albums.
130  */
131 static void
132 paint_album_callback(WINDOW *w, unsigned i,
133                      gcc_unused unsigned y, unsigned width,
134                      bool selected, const void *data)
135 {
136         const auto &list = *(const std::vector<std::string> *)data;
137         const char *p;
138         char *q = nullptr;
139
140         if (i == 0)
141                 p = "..";
142         else if (i == list.size() + 1)
143                 p = _("All tracks");
144         else
145                 p = q = utf8_to_locale(list[i - 1].c_str());
146
147         screen_browser_paint_directory(w, width, selected, p);
148         g_free(q);
149 }
150
151 void
152 AlbumListPage::Paint() const
153 {
154         lw.Paint(paint_album_callback, &album_list);
155 }
156
157 const char *
158 AlbumListPage::GetTitle(char *str, size_t size) const
159 {
160         if (artist.empty())
161                 return _("Albums");
162
163         g_snprintf(str, size, _("Albums of artist: %s"),
164                    Utf8ToLocale(artist.c_str()).c_str());
165         return str;
166 }
167
168 void
169 AlbumListPage::Update(struct mpdclient &c, unsigned events)
170 {
171         if (events & MPD_IDLE_DATABASE) {
172                 /* the db has changed -> update the list */
173                 Reload(c);
174                 SetDirty();
175         }
176 }
177
178 /* add_query - Add all songs satisfying specified criteria.
179    _artist is actually only used in the ALBUM case to distinguish albums with
180    the same name from different artists. */
181 static void
182 add_query(struct mpdclient *c, enum mpd_tag_type table, const char *_filter,
183           const char *_artist)
184 {
185         struct mpd_connection *connection = mpdclient_get_connection(c);
186
187         assert(_filter != nullptr);
188
189         if (connection == nullptr)
190                 return;
191
192         screen_status_printf(_("Adding \'%s\' to queue"),
193                              Utf8ToLocale(_filter).c_str());
194
195         mpd_search_add_db_songs(connection, true);
196         mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
197                                       table, _filter);
198         if (table == MPD_TAG_ALBUM)
199                 mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
200                                               MPD_TAG_ARTIST, _artist);
201         mpd_search_commit(connection);
202         mpdclient_finish_command(c);
203 }
204
205 bool
206 AlbumListPage::OnCommand(struct mpdclient &c, command_t cmd)
207 {
208         switch(cmd) {
209                 const char *selected;
210
211         case CMD_PLAY:
212                 if (lw.selected == 0) {
213                         /* handle ".." */
214                         screen.OnCommand(c, CMD_GO_PARENT_DIRECTORY);
215                         return true;
216                 }
217
218                 break;
219
220         case CMD_SELECT:
221         case CMD_ADD:
222                 for (const unsigned i : lw.GetRange()) {
223                         if(i == album_list.size() + 1)
224                                 add_query(&c, MPD_TAG_ARTIST, artist.c_str(),
225                                           nullptr);
226                         else if (i > 0) {
227                                 selected = album_list[lw.selected - 1].c_str();
228                                 add_query(&c, MPD_TAG_ALBUM, selected,
229                                           artist.c_str());
230                                 cmd = CMD_LIST_NEXT; /* continue and select next item... */
231                         }
232                 }
233                 break;
234
235                 /* continue and update... */
236         case CMD_SCREEN_UPDATE:
237                 Reload(c);
238                 return false;
239
240         case CMD_LIST_FIND:
241         case CMD_LIST_RFIND:
242         case CMD_LIST_FIND_NEXT:
243         case CMD_LIST_RFIND_NEXT:
244                 screen_find(screen, &lw, cmd,
245                             AlbumListCallback, &album_list);
246                 SetDirty();
247                 return true;
248
249         case CMD_LIST_JUMP:
250                 screen_jump(screen, &lw,
251                             AlbumListCallback, &album_list,
252                             paint_album_callback, &album_list);
253                 SetDirty();
254                 return true;
255
256         default:
257                 break;
258         }
259
260         if (lw.HandleCommand(cmd)) {
261                 SetDirty();
262                 return true;
263         }
264
265         return false;
266 }