![[tokkee]](http://tokkee.org/images/avatar.png){kind=avatar}
diff --git a/src/screen_search.c b/src/screen_search.c
index 6b6719e301710692133c0d6e293373bfc8803cdf..81ca8e4840c3fe7e9a71e753575146b152c78fad 100644 (file)
--- a/src/screen_search.c
+++ b/src/screen_search.c
-#include <stdlib.h>
+/* ncmpc (Ncurses MPD Client)
+ * (c) 2004-2010 The Music Player Daemon Project
+ * Project homepage: http://musicpd.org
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+#include "screen_search.h"
+#include "screen_interface.h"
+#include "screen_status.h"
+#include "screen.h"
+#include "i18n.h"
+#include "options.h"
+#include "charset.h"
+#include "mpdclient.h"
+#include "strfsong.h"
+#include "utils.h"
+#include "screen_utils.h"
+#include "screen_browser.h"
+#include "filelist.h"
+
#include <string.h>
#include <glib.h>
-#include <ncurses.h>
-#include "libmpdclient.h"
-#include "config.h"
-#include "mpc.h"
-#include "command.h"
-#include "screen.h"
-#include "screen_search.h"
+enum {
+ SEARCH_URI = MPD_TAG_COUNT + 100,
+ SEARCH_ARTIST_TITLE
+};
+
+static const struct {
+ const char *name;
+ const char *localname;
+} search_tag[MPD_TAG_COUNT] = {
+ [MPD_TAG_ARTIST] = { "artist", N_("artist") },
+ [MPD_TAG_ALBUM] = { "album", N_("album") },
+ [MPD_TAG_TITLE] = { "title", N_("title") },
+ [MPD_TAG_TRACK] = { "track", N_("track") },
+ [MPD_TAG_NAME] = { "name", N_("name") },
+ [MPD_TAG_GENRE] = { "genre", N_("genre") },
+ [MPD_TAG_DATE] = { "date", N_("date") },
+ [MPD_TAG_COMPOSER] = { "composer", N_("composer") },
+ [MPD_TAG_PERFORMER] = { "performer", N_("performer") },
+ [MPD_TAG_COMMENT] = { "comment", N_("comment") },
+};
+
+static int
+search_get_tag_id(const char *name)
+{
+ unsigned i;
+
+ if (g_ascii_strcasecmp(name, "file") == 0 ||
+ strcasecmp(name, _("file")) == 0)
+ return SEARCH_URI;
+
+ for (i = 0; i < MPD_TAG_COUNT; ++i)
+ if (search_tag[i].name != NULL &&
+ (strcasecmp(search_tag[i].name, name) == 0 ||
+ strcasecmp(search_tag[i].localname, name) == 0))
+ return i;
+
+ return -1;
+}
+
+typedef struct {
+ enum mpd_tag_type table;
+ const char *label;
+} search_type_t;
+
+static search_type_t mode[] = {
+ { MPD_TAG_TITLE, N_("Title") },
+ { MPD_TAG_ARTIST, N_("Artist") },
+ { MPD_TAG_ALBUM, N_("Album") },
+ { SEARCH_URI, N_("Filename") },
+ { SEARCH_ARTIST_TITLE, N_("Artist + Title") },
+ { 0, NULL }
+};
+
+static GList *search_history = NULL;
+static gchar *pattern = NULL;
+static gboolean advanced_search_mode = FALSE;
+
+static struct screen_browser browser;
+
+static const char *const help_text[] = {
+ "Quick - Enter a string and ncmpc will search according",
+ " to the current search mode (displayed above).",
+ "",
+ "Advanced - <tag>:<search term> [<tag>:<search term>...]",
+ " Example: artist:radiohead album:pablo honey",
+ "",
+ " Available tags: artist, album, title, track,",
+ " name, genre, date composer, performer, comment, file",
+ "",
+};
+
+/* search info */
+static const char *
+lw_search_help_callback(unsigned idx, G_GNUC_UNUSED void *data)
+{
+ assert(idx < G_N_ELEMENTS(help_text));
+
+ return help_text[idx];
+}
+
+static void
+screen_search_paint(void);
+
+static void
+search_repaint(void)
+{
+ screen_search_paint();
+ wrefresh(browser.lw->w);
+}
+
+/* sanity check search mode value */
+static void
+search_check_mode(void)
+{
+ int max = 0;
+
+ while (mode[max].label != NULL)
+ max++;
+ if (options.search_mode < 0)
+ options.search_mode = 0;
+ else if (options.search_mode >= max)
+ options.search_mode = max-1;
+}
+
+static void
+search_clear(bool clear_pattern)
+{
+ if (browser.filelist) {
+ filelist_free(browser.filelist);
+ browser.filelist = filelist_new();
+ list_window_set_length(browser.lw, 0);
+ }
+ if (clear_pattern && pattern) {
+ g_free(pattern);
+ pattern = NULL;
+ }
+}
+
+static struct filelist *
+search_simple_query(struct mpd_connection *connection, bool exact_match,
+ int table, gchar *local_pattern)
+{
+ struct filelist *list;
+ gchar *filter_utf8 = locale_to_utf8(local_pattern);
+
+ if (table == SEARCH_ARTIST_TITLE) {
+ mpd_command_list_begin(connection, false);
+
+ mpd_search_db_songs(connection, exact_match);
+ mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
+ MPD_TAG_ARTIST, filter_utf8);
+ mpd_search_commit(connection);
+
+ mpd_search_db_songs(connection, exact_match);
+ mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
+ MPD_TAG_TITLE, filter_utf8);
+ mpd_search_commit(connection);
+
+ mpd_command_list_end(connection);
+
+ list = filelist_new_recv(connection);
+ filelist_no_duplicates(list);
+ } else if (table == SEARCH_URI) {
+ mpd_search_db_songs(connection, exact_match);
+ mpd_search_add_uri_constraint(connection, MPD_OPERATOR_DEFAULT,
+ filter_utf8);
+ mpd_search_commit(connection);
+
+ list = filelist_new_recv(connection);
+ } else {
+ mpd_search_db_songs(connection, exact_match);
+ mpd_search_add_tag_constraint(connection, MPD_OPERATOR_DEFAULT,
+ table, filter_utf8);
+ mpd_search_commit(connection);
+
+ list = filelist_new_recv(connection);
+ }
+
+ g_free(filter_utf8);
+ return list;
+}
+
+/*-----------------------------------------------------------------------
+ * NOTE: This code exists to test a new search ui,
+ * Its ugly and MUST be redesigned before the next release!
+ *-----------------------------------------------------------------------
+ */
+static struct filelist *
+search_advanced_query(struct mpd_connection *connection, char *query)
+{
+ int i,j;
+ char **strv;
+ int table[10];
+ char *arg[10];
+ struct filelist *fl = NULL;
+
+ advanced_search_mode = FALSE;
+ if (strchr(query, ':') == NULL)
+ return NULL;
+
+ strv = g_strsplit_set(query, ": ", 0);
+
+ memset(table, 0, 10*sizeof(int));
+ memset(arg, 0, 10*sizeof(char *));
+
+ i=0;
+ j=0;
+ while (strv[i] && strlen(strv[i]) > 0 && i < 9) {
+ int id = search_get_tag_id(strv[i]);
+ if (id == -1) {
+ if (table[j]) {
+ char *tmp = arg[j];
+ arg[j] = g_strdup_printf("%s %s", arg[j], strv[i]);
+ g_free(tmp);
+ } else {
+ screen_status_printf(_("Bad search tag %s"), strv[i]);
+ }
+ i++;
+ } else if (strv[i+1] == NULL || strlen(strv[i+1]) == 0) {
+ screen_status_printf(_("No argument for search tag %s"), strv[i]);
+ i++;
+ // j--;
+ //table[j] = -1;
+ } else {
+ table[j] = id;
+ arg[j] = locale_to_utf8(strv[i+1]); // FREE ME
+ j++;
+ table[j] = -1;
+ arg[j] = NULL;
+ i = i + 2;
+ advanced_search_mode = TRUE;
+ }
+ }
+
+ g_strfreev(strv);
+
+ if (!advanced_search_mode || j == 0) {
+ for (i = 0; arg[i] != NULL; ++i)
+ g_free(arg[i]);
+ return NULL;
+ }
+
+ /*-----------------------------------------------------------------------
+ * NOTE (again): This code exists to test a new search ui,
+ * Its ugly and MUST be redesigned before the next release!
+ * + the code below should live in mpdclient.c
+ *-----------------------------------------------------------------------
+ */
+ /** stupid - but this is just a test...... (fulhack) */
+ mpd_search_db_songs(connection, false);
+
+ for (i = 0; i < 10 && arg[i] != NULL; i++) {
+ if (table[i] == SEARCH_URI)
+ mpd_search_add_uri_constraint(connection,
+ MPD_OPERATOR_DEFAULT,
+ arg[i]);
+ else
+ mpd_search_add_tag_constraint(connection,
+ MPD_OPERATOR_DEFAULT,
+ table[i], arg[i]);
+ }
+
+ mpd_search_commit(connection);
+ fl = filelist_new_recv(connection);
+ if (!mpd_response_finish(connection)) {
+ filelist_free(fl);
+ fl = NULL;
+ }
+
+ for (i = 0; arg[i] != NULL; ++i)
+ g_free(arg[i]);
+
+ return fl;
+}
+
+static struct filelist *
+do_search(struct mpdclient *c, char *query)
+{
+ struct mpd_connection *connection = mpdclient_get_connection(c);
+ struct filelist *fl;
+
+ fl = search_advanced_query(connection, query);
+ if (fl != NULL)
+ return fl;
+
+ if (mpd_connection_get_error(connection) != MPD_ERROR_SUCCESS) {
+ mpdclient_handle_error(c);
+ return NULL;
+ }
+
+ fl = search_simple_query(connection, FALSE,
+ mode[options.search_mode].table,
+ query);
+ if (fl == NULL)
+ mpdclient_handle_error(c);
+ return fl;
+}
+
+static void
+screen_search_reload(struct mpdclient *c)
+{
+ if (pattern == NULL)
+ return;
+
+ if (browser.filelist != NULL) {
+ filelist_free(browser.filelist);
+ browser.filelist = NULL;
+ }
+
+ browser.filelist = do_search(c, pattern);
+ if (browser.filelist == NULL)
+ browser.filelist = filelist_new();
+ list_window_set_length(browser.lw, filelist_length(browser.filelist));
+
+ screen_browser_sync_highlights(browser.filelist, &c->playlist);
+}
+
+static void
+search_new(struct mpdclient *c)
+{
+ if (!mpdclient_is_connected(c))
+ return;
+
+ search_clear(true);
+
+ g_free(pattern);
+ pattern = screen_readln(_("Search"),
+ NULL,
+ &search_history,
+ NULL);
+
+ if (pattern == NULL) {
+ list_window_reset(browser.lw);
+ return;
+ }
+
+ screen_search_reload(c);
+}
+
+static void
+screen_search_init(WINDOW *w, int cols, int rows)
+{
+ browser.lw = list_window_init(w, cols, rows);
+ list_window_set_length(browser.lw, G_N_ELEMENTS(help_text));
+}
+
+static void
+screen_search_quit(void)
+{
+ if (search_history)
+ string_list_free(search_history);
+ if (browser.filelist)
+ filelist_free(browser.filelist);
+ list_window_free(browser.lw);
+
+ if (pattern) {
+ g_free(pattern);
+ pattern = NULL;
+ }
+}
+
+static void
+screen_search_open(G_GNUC_UNUSED struct mpdclient *c)
+{
+ // if( pattern==NULL )
+ // search_new(screen, c);
+ // else
+ screen_status_printf(_("Press %s for a new search"),
+ get_key_names(CMD_SCREEN_SEARCH, false));
+ search_check_mode();
+}
+
+static void
+screen_search_resize(int cols, int rows)
+{
+ list_window_resize(browser.lw, cols, rows);
+}
+
+static void
+screen_search_paint(void)
+{
+ if (browser.filelist) {
+ browser.lw->hide_cursor = false;
+ screen_browser_paint(&browser);
+ } else {
+ browser.lw->hide_cursor = true;
+ list_window_paint(browser.lw, lw_search_help_callback, NULL);
+ }
+}
+
+static const char *
+screen_search_get_title(char *str, size_t size)
+{
+ if (advanced_search_mode && pattern)
+ g_snprintf(str, size, _("Search: %s"), pattern);
+ else if (pattern)
+ g_snprintf(str, size,
+ _("Search: Results for %s [%s]"),
+ pattern,
+ _(mode[options.search_mode].label));
+ else
+ g_snprintf(str, size, _("Search: Press %s for a new search [%s]"),
+ get_key_names(CMD_SCREEN_SEARCH, false),
+ _(mode[options.search_mode].label));
+
+ return str;
+}
+
+static void
+screen_search_update(struct mpdclient *c)
+{
+ if (browser.filelist != NULL && c->events & MPD_IDLE_QUEUE) {
+ screen_browser_sync_highlights(browser.filelist, &c->playlist);
+ search_repaint();
+ }
+}
+
+static bool
+screen_search_cmd(struct mpdclient *c, command_t cmd)
+{
+ switch (cmd) {
+ case CMD_SEARCH_MODE:
+ options.search_mode++;
+ if (mode[options.search_mode].label == NULL)
+ options.search_mode = 0;
+ screen_status_printf(_("Search mode: %s"),
+ _(mode[options.search_mode].label));
+ /* continue and update... */
+ case CMD_SCREEN_UPDATE:
+ screen_search_reload(c);
+ search_repaint();
+ return true;
+
+ case CMD_SCREEN_SEARCH:
+ search_new(c);
+ search_repaint();
+ return true;
+
+ case CMD_CLEAR:
+ search_clear(true);
+ list_window_reset(browser.lw);
+ search_repaint();
+ return true;
+
+ default:
+ break;
+ }
+
+ if (browser.filelist != NULL &&
+ browser_cmd(&browser, c, cmd)) {
+ if (screen_is_visible(&screen_search))
+ search_repaint();
+ return true;
+ }
+
+ return false;
+}
+const struct screen_functions screen_search = {
+ .init = screen_search_init,
+ .exit = screen_search_quit,
+ .open = screen_search_open,
+ .resize = screen_search_resize,
+ .paint = screen_search_paint,
+ .update = screen_search_update,
+ .cmd = screen_search_cmd,
+ .get_title = screen_search_get_title,
+};