Andrzej Rybczak
16 years ago
parent
d4a5574146
commit
a1cd5ae1aa
8 changed files with 308 additions and 328 deletions
@ -1,291 +0,0 @@ |
||||
/***************************************************************************
|
||||
* Copyright (C) 2008-2010 by Andrzej Rybczak * |
||||
* electricityispower@gmail.com * |
||||
* * |
||||
* 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 St, Fifth Floor, Boston, MA 02110-1301, USA. * |
||||
***************************************************************************/ |
||||
|
||||
#include "info.h" |
||||
#include "curl_handle.h" |
||||
|
||||
#ifdef HAVE_CURL_CURL_H |
||||
# include <fstream> |
||||
# ifdef WIN32 |
||||
# include <io.h> |
||||
# else |
||||
# include <sys/stat.h> |
||||
# endif // WIN32
|
||||
#endif |
||||
|
||||
#include "browser.h" |
||||
#include "charset.h" |
||||
#include "global.h" |
||||
#include "media_library.h" |
||||
#include "playlist.h" |
||||
#include "playlist_editor.h" |
||||
#include "search_engine.h" |
||||
#include "status.h" |
||||
#include "tag_editor.h" |
||||
|
||||
using Global::MainHeight; |
||||
using Global::MainStartY; |
||||
using Global::myScreen; |
||||
using Global::myOldScreen; |
||||
|
||||
#ifdef HAVE_CURL_CURL_H |
||||
const std::string Info::Folder = home_path + HOME_FOLDER"artists"; |
||||
bool Info::ArtistReady = 0; |
||||
|
||||
pthread_t *Info::Downloader = 0; |
||||
|
||||
#endif // HAVE_CURL_CURL_H
|
||||
|
||||
Info *myInfo = new Info; |
||||
|
||||
void Info::Init() |
||||
{ |
||||
w = new Scrollpad(0, MainStartY, COLS, MainHeight, "", Config.main_color, brNone); |
||||
isInitialized = 1; |
||||
} |
||||
|
||||
void Info::Resize() |
||||
{ |
||||
w->Resize(COLS, MainHeight); |
||||
w->MoveTo(0, MainStartY); |
||||
hasToBeResized = 0; |
||||
} |
||||
|
||||
std::basic_string<my_char_t> Info::Title() |
||||
{ |
||||
return TO_WSTRING(itsTitle); |
||||
} |
||||
|
||||
#ifdef HAVE_CURL_CURL_H |
||||
void Info::Update() |
||||
{ |
||||
if (!ArtistReady) |
||||
return; |
||||
pthread_join(*Downloader, 0); |
||||
w->Flush(); |
||||
w->Refresh(); |
||||
delete Downloader; |
||||
Downloader = 0; |
||||
ArtistReady = 0; |
||||
} |
||||
|
||||
void Info::GetArtist() |
||||
{ |
||||
if (myScreen == this) |
||||
{ |
||||
myOldScreen->SwitchTo(); |
||||
} |
||||
else |
||||
{ |
||||
if (!isInitialized) |
||||
Init(); |
||||
|
||||
if (Downloader && !ArtistReady) |
||||
{ |
||||
ShowMessage("Artist info is being downloaded..."); |
||||
return; |
||||
} |
||||
else if (ArtistReady) |
||||
Update(); |
||||
|
||||
MPD::Song *s = myScreen->CurrentSong(); |
||||
|
||||
if (!s && myScreen->ActiveWindow() != myLibrary->Artists) |
||||
return; |
||||
|
||||
itsArtist = !s ? myLibrary->Artists->Current() : s->GetArtist(); |
||||
|
||||
if (itsArtist.empty()) |
||||
return; |
||||
|
||||
if (hasToBeResized) |
||||
Resize(); |
||||
myOldScreen = myScreen; |
||||
myScreen = this; |
||||
Global::RedrawHeader = 1; |
||||
itsTitle = "Artist info - " + itsArtist; |
||||
w->Clear(); |
||||
w->Reset(); |
||||
static_cast<Window &>(*w) << "Fetching artist info..."; |
||||
w->Window::Refresh(); |
||||
if (!Downloader) |
||||
{ |
||||
locale_to_utf(itsArtist); |
||||
|
||||
std::string file = itsArtist + ".txt"; |
||||
ToLower(file); |
||||
EscapeUnallowedChars(file); |
||||
|
||||
itsFilenamePath = Folder + "/" + file; |
||||
|
||||
mkdir(Folder.c_str() |
||||
# ifndef WIN32 |
||||
, 0755 |
||||
# endif // !WIN32
|
||||
); |
||||
|
||||
std::ifstream input(itsFilenamePath.c_str()); |
||||
if (input.is_open()) |
||||
{ |
||||
bool first = 1; |
||||
std::string line; |
||||
while (getline(input, line)) |
||||
{ |
||||
if (!first) |
||||
*w << "\n"; |
||||
utf_to_locale(line); |
||||
*w << line; |
||||
first = 0; |
||||
} |
||||
input.close(); |
||||
w->SetFormatting(fmtBold, U("\n\nSimilar artists:\n"), fmtBoldEnd, 0); |
||||
w->SetFormatting(Config.color2, U("\n * "), clEnd, 1); |
||||
// below is used so format won't be removed using RemoveFormatting() by accident.
|
||||
w->ForgetFormatting(); |
||||
w->Flush(); |
||||
} |
||||
else |
||||
{ |
||||
Downloader = new pthread_t; |
||||
pthread_create(Downloader, 0, PrepareArtist, this); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
void *Info::PrepareArtist(void *screen_void_ptr) |
||||
{ |
||||
Info *screen = static_cast<Info *>(screen_void_ptr); |
||||
|
||||
std::string url = "http://ws.audioscrobbler.com/2.0/?method=artist.getinfo&artist="; |
||||
url += Curl::escape(screen->itsArtist); |
||||
url += "&api_key=d94e5b6e26469a2d1ffae8ef20131b79"; |
||||
|
||||
std::string result; |
||||
CURLcode code = Curl::perform(result, url); |
||||
|
||||
if (code != CURLE_OK) |
||||
{ |
||||
*screen->w << "Error while fetching artist's info: " << curl_easy_strerror(code); |
||||
ArtistReady = 1; |
||||
pthread_exit(0); |
||||
} |
||||
|
||||
size_t a, b; |
||||
bool save = 1; |
||||
|
||||
a = result.find("status=\"failed\""); |
||||
|
||||
if (a != std::string::npos) |
||||
{ |
||||
StripHtmlTags(result); |
||||
*screen->w << "Last.fm returned an error message: " << result; |
||||
ArtistReady = 1; |
||||
pthread_exit(0); |
||||
} |
||||
|
||||
std::vector<std::string> similar; |
||||
for (size_t i = result.find("<name>"); i != std::string::npos; i = result.find("<name>")) |
||||
{ |
||||
result[i] = '.'; |
||||
size_t j = result.find("</name>"); |
||||
result[j] = '.'; |
||||
i += static_strlen("<name>"); |
||||
similar.push_back(result.substr(i, j-i)); |
||||
StripHtmlTags(similar.back()); |
||||
} |
||||
std::vector<std::string> urls; |
||||
for (size_t i = result.find("<url>"); i != std::string::npos; i = result.find("<url>")) |
||||
{ |
||||
result[i] = '.'; |
||||
size_t j = result.find("</url>"); |
||||
result[j] = '.'; |
||||
i += static_strlen("<url>"); |
||||
urls.push_back(result.substr(i, j-i)); |
||||
} |
||||
|
||||
bool parse_failed = 0; |
||||
if ((a = result.find("<content>")) != std::string::npos) |
||||
{ |
||||
a += static_strlen("<content>"); |
||||
if ((b = result.find("</content>")) == std::string::npos) |
||||
parse_failed = 1; |
||||
} |
||||
else |
||||
parse_failed = 1; |
||||
if (parse_failed) |
||||
{ |
||||
*screen->w << "Error: Fetched data could not be parsed"; |
||||
ArtistReady = 1; |
||||
pthread_exit(0); |
||||
} |
||||
|
||||
if (a == b) |
||||
{ |
||||
result = "No description available for this artist."; |
||||
save = 0; |
||||
} |
||||
else |
||||
{ |
||||
a += static_strlen("<![CDATA["); |
||||
b -= static_strlen("]]>"); |
||||
result = result.substr(a, b-a); |
||||
} |
||||
|
||||
StripHtmlTags(result); |
||||
Trim(result); |
||||
|
||||
std::ostringstream filebuffer; |
||||
if (save) |
||||
filebuffer << result; |
||||
utf_to_locale(result); |
||||
*screen->w << result; |
||||
|
||||
if (save) |
||||
filebuffer << "\n\nSimilar artists:\n"; |
||||
*screen->w << fmtBold << "\n\nSimilar artists:\n" << fmtBoldEnd; |
||||
for (size_t i = 1; i < similar.size(); ++i) |
||||
{ |
||||
if (save) |
||||
filebuffer << "\n * " << similar[i] << " (" << urls[i] << ")"; |
||||
utf_to_locale(similar[i]); |
||||
utf_to_locale(urls[i]); |
||||
*screen->w << "\n" << Config.color2 << " * " << clEnd << similar[i] << " (" << urls[i] << ")"; |
||||
} |
||||
|
||||
if (save) |
||||
filebuffer << "\n\n" << urls.front(); |
||||
utf_to_locale(urls.front()); |
||||
*screen->w << "\n\n" << urls.front(); |
||||
|
||||
if (save) |
||||
{ |
||||
std::ofstream output(screen->itsFilenamePath.c_str()); |
||||
if (output.is_open()) |
||||
{ |
||||
output << filebuffer.str(); |
||||
output.close(); |
||||
} |
||||
} |
||||
ArtistReady = 1; |
||||
pthread_exit(0); |
||||
} |
||||
#endif // HVAE_CURL_CURL_H
|
||||
|
||||
@ -0,0 +1,225 @@ |
||||
/***************************************************************************
|
||||
* Copyright (C) 2008-2010 by Andrzej Rybczak * |
||||
* electricityispower@gmail.com * |
||||
* * |
||||
* 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 St, Fifth Floor, Boston, MA 02110-1301, USA. * |
||||
***************************************************************************/ |
||||
|
||||
#include "lastfm.h" |
||||
|
||||
#ifdef HAVE_CURL_CURL_H |
||||
|
||||
#ifdef WIN32 |
||||
# include <io.h> |
||||
#else |
||||
# include <sys/stat.h> |
||||
#endif // WIN32
|
||||
|
||||
#include <cassert> |
||||
#include <cerrno> |
||||
#include <fstream> |
||||
#include <iostream> |
||||
|
||||
#include "charset.h" |
||||
#include "global.h" |
||||
|
||||
using Global::MainHeight; |
||||
using Global::MainStartY; |
||||
using Global::myScreen; |
||||
using Global::myOldScreen; |
||||
|
||||
Lastfm *myLastfm = new Lastfm; |
||||
|
||||
void Lastfm::Init() |
||||
{ |
||||
w = new Scrollpad(0, MainStartY, COLS, MainHeight, "", Config.main_color, brNone); |
||||
isInitialized = 1; |
||||
} |
||||
|
||||
void Lastfm::Resize() |
||||
{ |
||||
w->Resize(COLS, MainHeight); |
||||
w->MoveTo(0, MainStartY); |
||||
hasToBeResized = 0; |
||||
} |
||||
|
||||
std::basic_string<my_char_t> Lastfm::Title() |
||||
{ |
||||
return itsTitle; |
||||
} |
||||
|
||||
void Lastfm::Update() |
||||
{ |
||||
if (isReadyToTake) |
||||
{ |
||||
pthread_join(itsDownloader, 0); |
||||
w->Flush(); |
||||
w->Refresh(); |
||||
isDownloadInProgress = 0; |
||||
isReadyToTake = 0; |
||||
} |
||||
} |
||||
|
||||
void Lastfm::SwitchTo() |
||||
{ |
||||
if (myScreen == this) |
||||
return myOldScreen->SwitchTo(); |
||||
|
||||
if (!isInitialized) |
||||
Init(); |
||||
|
||||
if (hasToBeResized) |
||||
Resize(); |
||||
|
||||
// if something is ready to take, take it
|
||||
Update(); |
||||
|
||||
Load(); |
||||
|
||||
myOldScreen = myScreen; |
||||
myScreen = this; |
||||
|
||||
Global::RedrawHeader = 1; |
||||
} |
||||
|
||||
void Lastfm::Load() |
||||
{ |
||||
if (isDownloadInProgress) |
||||
return; |
||||
|
||||
assert(itsService.get()); |
||||
assert(itsService->checkArgs(itsArgs)); |
||||
|
||||
SetTitleAndFolder(); |
||||
|
||||
w->Clear(); |
||||
w->Reset(); |
||||
|
||||
std::string artist = itsArgs.find("artist")->second; |
||||
locale_to_utf(artist); |
||||
|
||||
std::string file = artist + ".txt"; |
||||
ToLower(file); |
||||
EscapeUnallowedChars(file); |
||||
|
||||
itsFilename = itsFolder + "/" + file; |
||||
|
||||
mkdir(itsFolder.c_str() |
||||
# ifndef WIN32 |
||||
, 0755 |
||||
# endif // !WIN32
|
||||
); |
||||
|
||||
std::ifstream input(itsFilename.c_str()); |
||||
if (input.is_open()) |
||||
{ |
||||
bool first = 1; |
||||
std::string line; |
||||
while (getline(input, line)) |
||||
{ |
||||
if (!first) |
||||
*w << "\n"; |
||||
utf_to_locale(line); |
||||
*w << line; |
||||
first = 0; |
||||
} |
||||
input.close(); |
||||
itsService->colorizeOutput(*w); |
||||
} |
||||
else |
||||
{ |
||||
*w << "Fetching informations... "; |
||||
pthread_create(&itsDownloader, 0, DownloadWrapper, this); |
||||
isDownloadInProgress = 1; |
||||
} |
||||
w->Flush(); |
||||
} |
||||
|
||||
void Lastfm::SetTitleAndFolder() |
||||
{ |
||||
if (dynamic_cast<ArtistInfo *>(itsService.get())) |
||||
{ |
||||
itsTitle = U("Artist info - "); |
||||
itsTitle += TO_WSTRING(itsArgs.find("artist")->second); |
||||
itsFolder = home_path + HOME_FOLDER + "artists"; |
||||
} |
||||
} |
||||
|
||||
void *Lastfm::DownloadWrapper(void *this_ptr) |
||||
{ |
||||
static_cast<Lastfm *>(this_ptr)->Download(); |
||||
pthread_exit(0); |
||||
} |
||||
|
||||
void Lastfm::Download() |
||||
{ |
||||
LastfmService::Result result = itsService->fetch(itsArgs); |
||||
|
||||
if (result.first) |
||||
{ |
||||
Save(result.second); |
||||
w->Clear(); |
||||
utf_to_locale(result.second); |
||||
*w << result.second; |
||||
itsService->colorizeOutput(*w); |
||||
} |
||||
else |
||||
*w << clRed << result.second << clEnd; |
||||
|
||||
isReadyToTake = 1; |
||||
} |
||||
|
||||
void Lastfm::Save(const std::string &data) |
||||
{ |
||||
std::ofstream output(itsFilename.c_str()); |
||||
if (output.is_open()) |
||||
{ |
||||
output << data; |
||||
output.close(); |
||||
} |
||||
else |
||||
Error("couldn't save file \"" << itsFilename << "\""); |
||||
} |
||||
|
||||
void Lastfm::Refetch() |
||||
{ |
||||
if (!remove(itsFilename.c_str())) |
||||
{ |
||||
Load(); |
||||
} |
||||
else |
||||
{ |
||||
static const char msg[] = "Couldn't remove \"%s\": %s"; |
||||
ShowMessage(msg, Shorten(TO_WSTRING(itsFilename), COLS-static_strlen(msg)-25).c_str(), strerror(errno)); |
||||
} |
||||
} |
||||
|
||||
bool Lastfm::SetArtistInfoArgs(const std::string &artist, const std::string &lang) |
||||
{ |
||||
if (isDownloadInProgress) |
||||
return false; |
||||
|
||||
itsService.reset(new ArtistInfo); |
||||
itsArgs.clear(); |
||||
itsArgs["artist"] = artist; |
||||
if (!lang.empty()) |
||||
itsArgs["lang"] = lang; |
||||
|
||||
return true; |
||||
} |
||||
|
||||
#endif // HVAE_CURL_CURL_H
|
||||
|
||||
Loading…
Reference in new issue