Files
ncmpcpp/src/mpdpp.cpp
Andrzej Rybczak b0212a3779 fix MPD::Connection::GetDirectories()
they should return only dirs from main dir, not from all subdirs
2009-10-04 16:14:45 +02:00

795 lines
20 KiB
C++

/***************************************************************************
* Copyright (C) 2008-2009 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 <cstdlib>
#include <algorithm>
#include "charset.h"
#include "mpdpp.h"
using namespace MPD;
MPD::Connection Mpd;
const char *MPD::Message::PartOfSongsAdded = "Only part of requested songs' list added to playlist!";
const char *MPD::Message::FullPlaylist = "Playlist is full!";
const char *MPD::Message::FunctionDisabledFilteringEnabled = "Function disabled due to enabled filtering in playlist";
Connection::Connection() : itsConnection(0),
isConnected(0),
isCommandsListEnabled(0),
itsErrorCode(0),
itsMaxPlaylistLength(-1),
itsHost("localhost"),
itsPort(6600),
itsTimeout(15),
itsCurrentStatus(0),
itsOldStatus(0),
itsUpdater(0),
itsErrorHandler(0)
{
}
Connection::~Connection()
{
if (itsConnection)
mpd_connection_free(itsConnection);
if (itsOldStatus)
mpd_status_free(itsOldStatus);
if (itsCurrentStatus)
mpd_status_free(itsCurrentStatus);
}
bool Connection::Connect()
{
if (!isConnected && !itsConnection)
{
itsConnection = mpd_connection_new(itsHost.c_str(), itsPort, itsTimeout*1000 /* timeout is in ms now */);
isConnected = 1;
if (CheckForErrors())
return false;
if (!itsPassword.empty())
SendPassword();
return !CheckForErrors();
}
else
return true;
}
bool Connection::Connected() const
{
return isConnected;
}
void Connection::Disconnect()
{
if (itsConnection)
mpd_connection_free(itsConnection);
if (itsOldStatus)
mpd_status_free(itsOldStatus);
if (itsCurrentStatus)
mpd_status_free(itsCurrentStatus);
itsConnection = 0;
itsCurrentStatus = 0;
itsOldStatus = 0;
isConnected = 0;
isCommandsListEnabled = 0;
itsMaxPlaylistLength = -1;
}
float Connection::Version() const
{
if (!isConnected)
return 0;
const unsigned *version = mpd_connection_get_server_version(itsConnection);
return version[1] + version[2]*0.1;
}
void Connection::SetHostname(const std::string &host)
{
size_t at = host.find("@");
if (at != std::string::npos)
{
itsPassword = host.substr(0, at);
itsHost = host.substr(at+1);
}
else
itsHost = host;
}
bool Connection::SendPassword() const
{
return mpd_run_password(itsConnection, itsPassword.c_str());
}
void Connection::SetStatusUpdater(StatusUpdater updater, void *data)
{
itsUpdater = updater;
itsStatusUpdaterUserdata = data;
}
void Connection::SetErrorHandler(ErrorHandler handler, void *data)
{
itsErrorHandler = handler;
itsErrorHandlerUserdata = data;
}
void Connection::UpdateStatus()
{
if (!itsConnection)
return;
CheckForErrors();
if (!itsConnection)
return;
if (itsOldStatus)
mpd_status_free(itsOldStatus);
itsOldStatus = itsCurrentStatus;
itsCurrentStatus = 0;
itsCurrentStatus = mpd_run_status(itsConnection);
if (!itsMaxPlaylistLength)
itsMaxPlaylistLength = GetPlaylistLength();
if (CheckForErrors())
return;
if (itsCurrentStatus && itsUpdater)
{
if (!itsOldStatus)
{
itsChanges.Playlist = 1;
itsChanges.SongID = 1;
itsChanges.Database = 1;
itsChanges.DBUpdating = 1;
itsChanges.Volume = 1;
itsChanges.ElapsedTime = 1;
itsChanges.Crossfade = 1;
itsChanges.Random = 1;
itsChanges.Repeat = 1;
itsChanges.Single = 1;
itsChanges.Consume = 1;
itsChanges.PlayerState = 1;
itsChanges.StatusFlags = 1;
}
else
{
itsChanges.Playlist = mpd_status_get_queue_version(itsOldStatus) != mpd_status_get_queue_version(itsCurrentStatus);
itsChanges.SongID = mpd_status_get_song_id(itsOldStatus) != mpd_status_get_song_id(itsCurrentStatus);
itsChanges.Database = mpd_status_get_update_id(itsOldStatus) != mpd_status_get_update_id(itsCurrentStatus);
itsChanges.DBUpdating = itsChanges.Database;
itsChanges.Volume = mpd_status_get_volume(itsOldStatus) != mpd_status_get_volume(itsCurrentStatus);
itsChanges.ElapsedTime = mpd_status_get_elapsed_time(itsOldStatus) != mpd_status_get_elapsed_time(itsCurrentStatus);
itsChanges.Crossfade = mpd_status_get_crossfade(itsOldStatus) != mpd_status_get_crossfade(itsCurrentStatus);
itsChanges.Random = mpd_status_get_random(itsOldStatus) != mpd_status_get_random(itsCurrentStatus);
itsChanges.Repeat = mpd_status_get_repeat(itsOldStatus) != mpd_status_get_repeat(itsCurrentStatus);
itsChanges.Single = mpd_status_get_single(itsOldStatus) != mpd_status_get_single(itsCurrentStatus);
itsChanges.Consume = mpd_status_get_consume(itsOldStatus) != mpd_status_get_consume(itsCurrentStatus);
itsChanges.PlayerState = mpd_status_get_state(itsOldStatus) != mpd_status_get_state(itsCurrentStatus);
itsChanges.StatusFlags = itsChanges.Repeat || itsChanges.Random || itsChanges.Single || itsChanges.Consume || itsChanges.Crossfade || itsChanges.DBUpdating;
}
itsUpdater(this, itsChanges, itsErrorHandlerUserdata);
}
}
bool Connection::UpdateDirectory(const std::string &path)
{
if (!isConnected)
return false;
if (!mpd_run_update(itsConnection, path.c_str()))
return false;
UpdateStatus();
return true;
}
void Connection::Play() const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_play : mpd_run_play)(itsConnection);
}
void Connection::Play(int pos) const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_play_pos : mpd_run_play_pos)(itsConnection, pos);
}
void Connection::PlayID(int id) const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_play_id : mpd_run_play_id)(itsConnection, id);
}
void Connection::Pause(bool state) const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_pause : mpd_run_pause)(itsConnection, state);
}
void Connection::Toggle() const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_toggle_pause : mpd_run_toggle_pause)(itsConnection);
}
void Connection::Stop() const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_stop : mpd_run_stop)(itsConnection);
}
void Connection::Next() const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_next : mpd_run_next)(itsConnection);
}
void Connection::Prev() const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_previous : mpd_run_previous)(itsConnection);
}
void Connection::Move(unsigned from, unsigned to) const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_move : mpd_run_move)(itsConnection, from, to);
}
void Connection::Swap(unsigned from, unsigned to) const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_swap : mpd_run_swap)(itsConnection, from, to);
}
void Connection::Seek(unsigned where) const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_seek_pos : mpd_run_seek_pos)(itsConnection, Mpd.GetCurrentSongPos(), where);
}
void Connection::Shuffle() const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_shuffle : mpd_run_shuffle)(itsConnection);
}
void Connection::ClearPlaylist() const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_clear : mpd_run_clear)(itsConnection);
}
void Connection::ClearPlaylist(const std::string &playlist) const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_playlist_clear : mpd_run_playlist_clear)(itsConnection, playlist.c_str());
}
void Connection::AddToPlaylist(const std::string &path, const Song &s) const
{
if (!s.Empty())
AddToPlaylist(path, s.GetFile());
}
void Connection::AddToPlaylist(const std::string &path, const std::string &file) const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_playlist_add : mpd_run_playlist_add)(itsConnection, path.c_str(), file.c_str());
}
void Connection::Move(const std::string &path, int from, int to) const
{
if (!isConnected)
return;
mpd_send_playlist_move(itsConnection, path.c_str(), from, to);
if (!isCommandsListEnabled)
mpd_response_finish(itsConnection);
}
void Connection::Rename(const std::string &from, const std::string &to) const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_rename : mpd_run_rename)(itsConnection, from.c_str(), to.c_str());
}
void Connection::GetPlaylistChanges(unsigned version, SongList &v) const
{
if (isConnected)
{
if (!version)
v.reserve(GetPlaylistLength());
mpd_send_queue_changes_meta(itsConnection, version);
while (mpd_song *s = mpd_recv_song(itsConnection))
v.push_back(new Song(s, 1));
mpd_response_finish(itsConnection);
}
}
Song Connection::GetSong(const std::string &path) const
{
if (isConnected)
{
mpd_send_list_all_meta(itsConnection, path.c_str());
mpd_song *s = mpd_recv_song(itsConnection);
Song result(s);
mpd_response_finish(itsConnection);
return result;
}
else
return Song();
}
int Connection::GetCurrentSongPos() const
{
return itsCurrentStatus && isPlaying() ? mpd_status_get_song_pos(itsCurrentStatus) : -1;
}
Song Connection::GetCurrentSong() const
{
return Song(isConnected && isPlaying() ? mpd_run_current_song(itsConnection) : 0);
}
void Connection::GetPlaylistContent(const std::string &path, SongList &v) const
{
if (isConnected)
{
mpd_send_list_playlist_meta(itsConnection, path.c_str());
while (mpd_song *s = mpd_recv_song(itsConnection))
v.push_back(new Song(s));
mpd_response_finish(itsConnection);
}
}
void Connection::SetRepeat(bool mode) const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_repeat : mpd_run_repeat)(itsConnection, mode);
}
void Connection::SetRandom(bool mode) const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_random : mpd_run_random)(itsConnection, mode);
}
void Connection::SetSingle(bool mode) const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_single : mpd_run_single)(itsConnection, mode);
}
void Connection::SetConsume(bool mode) const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_consume : mpd_run_consume)(itsConnection, mode);
}
void Connection::SetVolume(unsigned vol)
{
if (isConnected && vol <= 100)
{
if (mpd_run_set_volume(itsConnection, vol) && itsUpdater)
UpdateStatus();
}
}
void Connection::SetCrossfade(unsigned crossfade) const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_crossfade : mpd_run_crossfade)(itsConnection, crossfade);
}
int Connection::AddSong(const std::string &path)
{
int id = -1;
if (isConnected)
{
if (GetPlaylistLength() < itsMaxPlaylistLength)
{
mpd_send_add_id(itsConnection, path.c_str());
if (!isCommandsListEnabled)
{
id = mpd_recv_song_id(itsConnection);
mpd_response_finish(itsConnection);
UpdateStatus();
}
else
id = 0;
}
else if (itsErrorHandler)
itsErrorHandler(this, MPD_SERVER_ERROR_PLAYLIST_MAX, Message::FullPlaylist, itsErrorHandlerUserdata);
}
return id;
}
int Connection::AddSong(const Song &s)
{
return !s.Empty() ? (AddSong((!s.isFromDB() ? "file://" : "") + (s.Localized() ? locale_to_utf_cpy(s.GetFile()) : s.GetFile()))) : -1;
}
void Connection::Add(const std::string &path) const
{
if (!isConnected)
return;
mpd_send_add(itsConnection, path.c_str());
mpd_response_finish(itsConnection);
}
bool Connection::AddRandomSongs(size_t number)
{
if (!isConnected && !number)
return false;
TagList files;
mpd_send_list_all(itsConnection, "/");
while (mpd_pair *item = mpd_recv_pair_named(itsConnection, "file"))
{
files.push_back(item->value);
mpd_return_pair(itsConnection, item);
}
mpd_response_finish(itsConnection);
if (number > files.size())
{
if (itsErrorHandler)
itsErrorHandler(this, 0, "Requested number of random songs is bigger than size of your library!", itsErrorHandlerUserdata);
return false;
}
else
{
srand(time(0));
std::random_shuffle(files.begin(), files.end());
StartCommandsList();
TagList::const_iterator it = files.begin()+rand()%(files.size()-number);
for (size_t i = 0; i < number && it != files.end(); ++i)
AddSong(*it++);
CommitCommandsList();
}
return true;
}
void Connection::Delete(unsigned pos) const
{
if (!isConnected)
return;
mpd_send_delete(itsConnection, pos);
if (!isCommandsListEnabled)
mpd_response_finish(itsConnection);
}
void Connection::DeleteID(unsigned id) const
{
if (!isConnected)
return;
mpd_send_delete_id(itsConnection, id);
if (!isCommandsListEnabled)
mpd_response_finish(itsConnection);
}
void Connection::Delete(const std::string &playlist, unsigned pos) const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_playlist_delete : mpd_run_playlist_delete)(itsConnection, playlist.c_str(), pos);
}
void Connection::StartCommandsList()
{
if (isConnected)
{
mpd_command_list_begin(itsConnection, 1);
isCommandsListEnabled = 1;
}
}
bool Connection::CommitCommandsList()
{
if (isConnected)
{
mpd_command_list_end(itsConnection);
mpd_response_finish(itsConnection);
UpdateStatus();
if (GetPlaylistLength() == itsMaxPlaylistLength && itsErrorHandler)
itsErrorHandler(this, MPD_SERVER_ERROR_PLAYLIST_MAX, Message::FullPlaylist, itsErrorHandlerUserdata);
isCommandsListEnabled = 0;
}
return !CheckForErrors();
}
void Connection::DeletePlaylist(const std::string &name) const
{
if (!isConnected)
return;
(isCommandsListEnabled ? mpd_send_rm : mpd_run_rm)(itsConnection, name.c_str());
}
bool Connection::SavePlaylist(const std::string &name) const
{
if (isConnected)
{
mpd_send_save(itsConnection, name.c_str());
mpd_response_finish(itsConnection);
return !(mpd_connection_get_error(itsConnection) == MPD_ERROR_SERVER
&& mpd_connection_get_server_error(itsConnection) == MPD_SERVER_ERROR_EXIST);
}
else
return false;
}
void Connection::GetPlaylists(TagList &v) const
{
if (isConnected)
{
ItemList list;
GetDirectory("/", list);
for (ItemList::const_iterator it = list.begin(); it != list.end(); ++it)
if (it->type == itPlaylist)
v.push_back(it->name);
FreeItemList(list);
}
}
void Connection::GetList(TagList &v, mpd_tag_type type) const
{
if (isConnected)
{
mpd_search_db_tags(itsConnection, type);
mpd_search_commit(itsConnection);
while (mpd_pair *item = mpd_recv_pair_tag(itsConnection, type))
{
if (item->value[0] != 0) // do not push empty item
v.push_back(item->value);
mpd_return_pair(itsConnection, item);
}
mpd_response_finish(itsConnection);
}
}
void Connection::GetAlbums(const std::string &artist, TagList &v) const
{
if (isConnected)
{
mpd_search_db_tags(itsConnection, MPD_TAG_ALBUM);
if (!artist.empty())
mpd_search_add_tag_constraint(itsConnection, MPD_OPERATOR_DEFAULT, MPD_TAG_ARTIST, artist.c_str());
mpd_search_commit(itsConnection);
while (mpd_pair *item = mpd_recv_pair_tag(itsConnection, MPD_TAG_ALBUM))
{
if (item->value[0] != 0) // do not push empty item
v.push_back(item->value);
mpd_return_pair(itsConnection, item);
}
mpd_response_finish(itsConnection);
}
}
void Connection::StartSearch(bool exact_match) const
{
if (isConnected)
mpd_search_db_songs(itsConnection, exact_match);
}
void Connection::StartFieldSearch(mpd_tag_type item)
{
if (isConnected)
{
itsSearchedField = item;
mpd_search_db_tags(itsConnection, item);
}
}
void Connection::AddSearch(mpd_tag_type item, const std::string &str) const
{
// mpd version < 0.14.* doesn't support empty search constraints
if (Version() < 14 && str.empty())
return;
if (isConnected)
mpd_search_add_tag_constraint(itsConnection, MPD_OPERATOR_DEFAULT, item, str.c_str());
}
void Connection::CommitSearch(SongList &v) const
{
if (isConnected)
{
mpd_search_commit(itsConnection);
while (mpd_song *s = mpd_recv_song(itsConnection))
v.push_back(new Song(s));
mpd_response_finish(itsConnection);
}
}
void Connection::CommitSearch(TagList &v) const
{
if (isConnected)
{
mpd_search_commit(itsConnection);
while (mpd_pair *tag = mpd_recv_pair_tag(itsConnection, itsSearchedField))
{
if (tag->value[0] != 0) // do not push empty item
v.push_back(tag->value);
mpd_return_pair(itsConnection, tag);
}
mpd_response_finish(itsConnection);
}
}
void Connection::GetDirectory(const std::string &path, ItemList &v) const
{
if (isConnected)
{
mpd_send_list_meta(itsConnection, path.c_str());
while (mpd_entity *item = mpd_recv_entity(itsConnection))
{
Item i;
switch (mpd_entity_get_type(item))
{
case MPD_ENTITY_TYPE_DIRECTORY:
i.name = mpd_directory_get_path(mpd_entity_get_directory(item));
i.type = itDirectory;
goto WRITE;
case MPD_ENTITY_TYPE_SONG:
i.song = new Song(mpd_song_dup(mpd_entity_get_song(item)));
i.type = itSong;
goto WRITE;
case MPD_ENTITY_TYPE_PLAYLIST:
i.name = mpd_playlist_get_path(mpd_entity_get_playlist(item));
i.type = itPlaylist;
goto WRITE;
WRITE:
v.push_back(i);
break;
default:
break;
}
mpd_entity_free(item);
}
mpd_response_finish(itsConnection);
}
}
void Connection::GetDirectoryRecursive(const std::string &path, SongList &v) const
{
if (isConnected)
{
mpd_send_list_all_meta(itsConnection, path.c_str());
while (mpd_song *s = mpd_recv_song(itsConnection))
v.push_back(new Song(s));
mpd_response_finish(itsConnection);
}
}
void Connection::GetSongs(const std::string &path, SongList &v) const
{
if (isConnected)
{
mpd_send_list_meta(itsConnection, path.c_str());
while (mpd_song *s = mpd_recv_song(itsConnection))
v.push_back(new Song(s));
mpd_response_finish(itsConnection);
}
}
void Connection::GetDirectories(const std::string &path, TagList &v) const
{
if (isConnected)
{
mpd_send_list_meta(itsConnection, path.c_str());
while (mpd_directory *dir = mpd_recv_directory(itsConnection))
{
v.push_back(mpd_directory_get_path(dir));
mpd_directory_free(dir);
}
mpd_response_finish(itsConnection);
}
}
void Connection::GetOutputs(OutputList &v) const
{
if (!isConnected)
return;
mpd_send_outputs(itsConnection);
while (mpd_output *output = mpd_recv_output(itsConnection))
{
v.push_back(std::make_pair(mpd_output_get_name(output), mpd_output_get_enabled(output)));
mpd_output_free(output);
}
mpd_response_finish(itsConnection);
}
bool Connection::EnableOutput(int id)
{
if (!isConnected)
return false;
return mpd_run_enable_output(itsConnection, id);
}
bool Connection::DisableOutput(int id)
{
if (!isConnected)
return false;
return mpd_run_disable_output(itsConnection, id);
}
int Connection::CheckForErrors()
{
if ((itsErrorCode = mpd_connection_get_error(itsConnection)) != MPD_ERROR_SUCCESS)
{
itsErrorMessage = mpd_connection_get_error_message(itsConnection);
if (itsErrorCode == MPD_ERROR_SERVER)
{
// this is to avoid setting too small max size as we check it before fetching current status
// setting real max playlist length is in UpdateStatus()
itsErrorCode = mpd_connection_get_server_error(itsConnection);
if (itsErrorCode == MPD_SERVER_ERROR_PLAYLIST_MAX && itsMaxPlaylistLength == size_t(-1))
itsMaxPlaylistLength = 0;
}
if (!mpd_connection_clear_error(itsConnection))
Disconnect();
if (itsErrorHandler)
itsErrorHandler(this, itsErrorCode, itsErrorMessage.c_str(), itsErrorHandlerUserdata);
}
return itsErrorCode;
}
void MPD::FreeSongList(SongList &l)
{
for (SongList::iterator i = l.begin(); i != l.end(); ++i)
delete *i;
l.clear();
}
void MPD::FreeItemList(ItemList &l)
{
for (ItemList::iterator i = l.begin(); i != l.end(); ++i)
delete i->song;
l.clear();
}