From 26b7218f8edf321f3a1bba1b6ea1380ac34d5820 Mon Sep 17 00:00:00 2001
From: Andrzej Rybczak <electricityispower@gmail.com>
Date: Mon, 9 Aug 2010 13:14:07 +0200
Subject: [PATCH] lyrics fetcher: add lyricwiki.com database support

---
 src/lyrics_fetcher.cpp | 93 ++++++++++++++++++++++++++++++++++++------
 src/lyrics_fetcher.h   | 18 ++++++++
 2 files changed, 98 insertions(+), 13 deletions(-)

diff --git a/src/lyrics_fetcher.cpp b/src/lyrics_fetcher.cpp
index d5561be6..0a8c219d 100644
--- a/src/lyrics_fetcher.cpp
+++ b/src/lyrics_fetcher.cpp
@@ -22,11 +22,14 @@
 
 #ifdef HAVE_CURL_CURL_H
 
+#include <cstdlib>
+
 #include "conv.h"
 #include "lyrics_fetcher.h"
 
 LyricsFetcher *lyricsPlugins[] =
 {
+	new LyricwikiFetcher(),
 	new LyrcComArFetcher(),
 	new LyricsflyFetcher(),
 	0
@@ -51,20 +54,9 @@ LyricsFetcher::Result LyricsFetcher::fetch(const std::string &artist, const std:
 		return result;
 	}
 	
-	size_t a, b;
-	bool parse_failed = false;
-	if ((a = data.find(getOpenTag())) != std::string::npos)
-	{
-		a += strlen(getCloseTag())-1;
-		if ((b = data.find(getCloseTag(), a)) != std::string::npos)
-			data = data.substr(a, b-a);
-		else
-			parse_failed = true;
-	}
-	else
-		parse_failed = true;
+	bool parse_ok = getContent(getOpenTag(), getCloseTag(), data);
 	
-	if (parse_failed || notLyrics(data))
+	if (!parse_ok || notLyrics(data))
 	{
 		result.second = "Not found";
 		return result;
@@ -77,12 +69,87 @@ LyricsFetcher::Result LyricsFetcher::fetch(const std::string &artist, const std:
 	return result;
 }
 
+bool LyricsFetcher::getContent(const char *open_tag, const char *close_tag, std::string &data)
+{
+	size_t a, b;
+	if ((a = data.find(open_tag)) != std::string::npos)
+	{
+		a += strlen(open_tag);
+		if ((b = data.find(close_tag, a)) != std::string::npos)
+			data = data.substr(a, b-a);
+		else
+			return false;
+	}
+	else
+		return false;
+	return true;
+}
+
 void LyricsFetcher::postProcess(std::string &data)
 {
 	EscapeHtml(data);
 	Trim(data);
 }
 
+/***********************************************************************/
+
+LyricsFetcher::Result LyricwikiFetcher::fetch(const std::string &artist, const std::string &title)
+{
+	LyricsFetcher::Result result = LyricsFetcher::fetch(artist, title);
+	if (result.first == true)
+	{
+		result.first = false;
+		
+		std::string data;
+		CURLcode code = Curl::perform(result.second, data);
+		
+		if (code != CURLE_OK)
+		{
+			result.second = "Error while fetching lyrics: ";
+			result.second += curl_easy_strerror(code);
+			return result;
+		}
+		
+		bool parse_ok = getContent("'17'/></a></div>", "<!--", data);
+		
+		if (!parse_ok)
+		{
+			result.second = "Not found";
+			return result;
+		}
+		
+		Replace(data, "<br />", "\n");
+		
+		result.second = unescape(data);
+		result.first = true;
+	}
+	return result;
+}
+
+bool LyricwikiFetcher::notLyrics(const std::string &data)
+{
+	return data.find("action=edit") != std::string::npos;
+}
+
+std::string LyricwikiFetcher::unescape(const std::string &data)
+{
+	std::string result;
+	for (size_t i = 0, j; i < data.length(); ++i)
+	{
+		if (data[i] == '&' && data[i+1] == '#' && (j = data.find(';', i)) != std::string::npos)
+		{
+			int n = atoi(&data.c_str()[i+2]);
+			result += char(n);
+			i = j;
+		}
+		else
+			result += data[i];
+	}
+	return result;
+}
+
+/***********************************************************************/
+
 void LyricsflyFetcher::postProcess(std::string &data)
 {
 	Replace(data, "[br]", "");
diff --git a/src/lyrics_fetcher.h b/src/lyrics_fetcher.h
index 5aa2cfd7..e25080d8 100644
--- a/src/lyrics_fetcher.h
+++ b/src/lyrics_fetcher.h
@@ -43,6 +43,8 @@ struct LyricsFetcher
 		
 		virtual bool notLyrics(GNUC_UNUSED const std::string &data) { return false; }
 		virtual void postProcess(std::string &data);
+		
+		bool getContent(const char *open_tag, const char *close_tag, std::string &data);
 };
 
 struct LyrcComArFetcher : public LyricsFetcher
@@ -55,6 +57,22 @@ struct LyrcComArFetcher : public LyricsFetcher
 		virtual const char *getCloseTag() { return "<p>"; }
 };
 
+struct LyricwikiFetcher : public LyricsFetcher
+{
+	virtual const char *name() { return "lyricwiki.com"; }
+	virtual Result fetch(const std::string &artist, const std::string &title);
+	
+	protected:
+		virtual const char *getURL() { return "http://lyrics.wikia.com/api.php?action=lyrics&fmt=xml&func=getSong&artist=%artist%&song=%title%"; }
+		virtual const char *getOpenTag() { return "<url>"; }
+		virtual const char *getCloseTag() { return "</url>"; }
+		
+		virtual bool notLyrics(const std::string &data);
+		
+	private:
+		std::string unescape(const std::string &data);
+};
+
 struct LyricsflyFetcher : public LyricsFetcher
 {
 	virtual const char *name() { return "lyricsfly.com"; }