lastfm: redesigned plugins system

src/lastfm.cpp

@@ -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
+