src/mpdpp.h

   1 /***************************************************************************
   2  *   Copyright (C) 2008-2016 by Andrzej Rybczak                            *
   3  *   electricityispower@gmail.com                                          *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) any later version.                                   *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA.              *
  19  ***************************************************************************/
  20
  21 #ifndef NCMPCPP_MPDPP_H
  22 #define NCMPCPP_MPDPP_H
  23
  24 #include <cassert>
  25 #include <exception>
  26 #include <random>
  27 #include <set>
  28 #include <vector>
  29
  30 #include <mpd/client.h>
  31 #include "song.h"
  32
  33 namespace MPD {
  34
  35 void checkConnectionErrors(mpd_connection *conn);
  36
  37 enum PlayerState { psUnknown, psStop, psPlay, psPause };
  38 enum ReplayGainMode { rgmOff, rgmTrack, rgmAlbum };
  39
  40 struct ClientError: public std::exception
  41 {
  42         ClientError(mpd_error code_, std::string msg, bool clearable_)
  43         : m_code(code_), m_msg(msg), m_clearable(clearable_) { }
  44         virtual ~ClientError() noexcept { }
  45
  46         virtual const char *what() const noexcept { return m_msg.c_str(); }
  47         mpd_error code() const { return m_code; }
  48         bool clearable() const { return m_clearable; }
  49
  50 private:
  51         mpd_error m_code;
  52         std::string m_msg;
  53         bool m_clearable;
  54 };
  55
  56 struct ServerError: public std::exception
  57 {
  58         ServerError(mpd_server_error code_, std::string msg, bool clearable_)
  59         : m_code(code_), m_msg(msg), m_clearable(clearable_) { }
  60         virtual ~ServerError() noexcept { }
  61
  62         virtual const char *what() const noexcept { return m_msg.c_str(); }
  63         mpd_server_error code() const { return m_code; }
  64         bool clearable() const { return m_clearable; }
  65
  66 private:
  67         mpd_server_error m_code;
  68         std::string m_msg;
  69         bool m_clearable;
  70 };
  71
  72 struct Statistics
  73 {
  74         friend struct Connection;
  75
  76         bool empty() const { return m_stats.get() == nullptr; }
  77
  78         unsigned artists() const { return mpd_stats_get_number_of_artists(m_stats.get()); }
  79         unsigned albums() const { return mpd_stats_get_number_of_albums(m_stats.get()); }
  80         unsigned songs() const { return mpd_stats_get_number_of_songs(m_stats.get()); }
  81         unsigned long playTime() const { return mpd_stats_get_play_time(m_stats.get()); }
  82         unsigned long uptime() const { return mpd_stats_get_uptime(m_stats.get()); }
  83         unsigned long dbUpdateTime() const { return mpd_stats_get_db_update_time(m_stats.get()); }
  84         unsigned long dbPlayTime() const { return mpd_stats_get_db_play_time(m_stats.get()); }
  85
  86 private:
  87         Statistics(mpd_stats *stats) : m_stats(stats, mpd_stats_free) { }
  88
  89         std::shared_ptr<mpd_stats> m_stats;
  90 };
  91
  92 struct Status
  93 {
  94         friend struct Connection;
  95
  96         Status() { }
  97
  98         void clear() { m_status.reset(); }
  99         bool empty() const { return m_status.get() == nullptr; }
 100
 101         int volume() const { return mpd_status_get_volume(m_status.get()); }
 102         bool repeat() const { return mpd_status_get_repeat(m_status.get()); }
 103         bool random() const { return mpd_status_get_random(m_status.get()); }
 104         bool single() const { return mpd_status_get_single(m_status.get()); }
 105         bool consume() const { return mpd_status_get_consume(m_status.get()); }
 106         unsigned playlistLength() const { return mpd_status_get_queue_length(m_status.get()); }
 107         unsigned playlistVersion() const { return mpd_status_get_queue_version(m_status.get()); }
 108         PlayerState playerState() const { return PlayerState(mpd_status_get_state(m_status.get())); }
 109         unsigned crossfade() const { return mpd_status_get_crossfade(m_status.get()); }
 110         int currentSongPosition() const { return mpd_status_get_song_pos(m_status.get()); }
 111         int currentSongID() const { return mpd_status_get_song_id(m_status.get()); }
 112         int nextSongPosition() const { return mpd_status_get_next_song_pos(m_status.get()); }
 113         int nextSongID() const { return mpd_status_get_next_song_id(m_status.get()); }
 114         unsigned elapsedTime() const { return mpd_status_get_elapsed_time(m_status.get()); }
 115         unsigned totalTime() const { return mpd_status_get_total_time(m_status.get()); }
 116         unsigned kbps() const { return mpd_status_get_kbit_rate(m_status.get()); }
 117         unsigned updateID() const { return mpd_status_get_update_id(m_status.get()); }
 118         const char *error() const { return mpd_status_get_error(m_status.get()); }
 119
 120 private:
 121         Status(mpd_status *status) : m_status(status, mpd_status_free) { }
 122
 123         std::shared_ptr<mpd_status> m_status;
 124 };
 125
 126 struct Directory
 127 {
 128         Directory()
 129         : m_last_modified(0)
 130         { }
 131         Directory(const mpd_directory *directory)
 132         {
 133                 assert(directory != nullptr);
 134                 m_path = mpd_directory_get_path(directory);
 135                 m_last_modified = mpd_directory_get_last_modified(directory);
 136         }
 137         Directory(std::string path_, time_t last_modified = 0)
 138         : m_path(std::move(path_))
 139         , m_last_modified(last_modified)
 140         { }
 141
 142         bool operator==(const Directory &rhs) const
 143         {
 144                 return m_path == rhs.m_path
 145                     && m_last_modified == rhs.m_last_modified;
 146         }
 147         bool operator!=(const Directory &rhs) const
 148         {
 149                 return !(*this == rhs);
 150         }
 151
 152         const std::string &path() const
 153         {
 154                 return m_path;
 155         }
 156         time_t lastModified() const
 157         {
 158                 return m_last_modified;
 159         }
 160
 161 private:
 162         std::string m_path;
 163         time_t m_last_modified;
 164 };
 165
 166 struct Playlist
 167 {
 168         Playlist()
 169         : m_last_modified(0)
 170         { }
 171         Playlist(const mpd_playlist *playlist)
 172         {
 173                 assert(playlist != nullptr);
 174                 m_path = mpd_playlist_get_path(playlist);
 175                 m_last_modified = mpd_playlist_get_last_modified(playlist);
 176         }
 177         Playlist(std::string path_, time_t last_modified = 0)
 178         : m_path(std::move(path_))
 179         , m_last_modified(last_modified)
 180         {
 181                 if (m_path.empty())
 182                         throw std::runtime_error("empty path");
 183         }
 184
 185         bool operator==(const Playlist &rhs) const
 186         {
 187                 return m_path == rhs.m_path
 188                     && m_last_modified == rhs.m_last_modified;
 189         }
 190         bool operator!=(const Playlist &rhs) const
 191         {
 192                 return !(*this == rhs);
 193         }
 194
 195         const std::string &path() const
 196         {
 197                 return m_path;
 198         }
 199         time_t lastModified() const
 200         {
 201                 return m_last_modified;
 202         }
 203
 204 private:
 205         std::string m_path;
 206         time_t m_last_modified;
 207 };
 208
 209 struct Item
 210 {
 211         enum class Type { Directory, Song, Playlist };
 212
 213         Item(mpd_entity *entity)
 214         {
 215                 assert(entity != nullptr);
 216                 switch (mpd_entity_get_type(entity))
 217                 {
 218                         case MPD_ENTITY_TYPE_DIRECTORY:
 219                                 m_type = Type::Directory;
 220                                 m_directory = Directory(mpd_entity_get_directory(entity));
 221                                 break;
 222                         case MPD_ENTITY_TYPE_SONG:
 223                                 m_type = Type::Song;
 224                                 m_song = Song(mpd_song_dup(mpd_entity_get_song(entity)));
 225                                 break;
 226                         case MPD_ENTITY_TYPE_PLAYLIST:
 227                                 m_type = Type::Playlist;
 228                                 m_playlist = Playlist(mpd_entity_get_playlist(entity));
 229                                 break;
 230                         default:
 231                                 throw std::runtime_error("unknown mpd_entity type");
 232                 }
 233                 mpd_entity_free(entity);
 234         }
 235         Item(Directory directory_)
 236         : m_type(Type::Directory)
 237         , m_directory(std::move(directory_))
 238         { }
 239         Item(Song song_)
 240         : m_type(Type::Song)
 241         , m_song(std::move(song_))
 242         { }
 243         Item(Playlist playlist_)
 244         : m_type(Type::Playlist)
 245         , m_playlist(std::move(playlist_))
 246         { }
 247
 248         bool operator==(const Item &rhs) const
 249         {
 250                 return m_directory == rhs.m_directory
 251                     && m_song == rhs.m_song
 252                     && m_playlist == rhs.m_playlist;
 253         }
 254         bool operator!=(const Item &rhs) const
 255         {
 256                 return !(*this == rhs);
 257         }
 258
 259         Type type() const
 260         {
 261                 return m_type;
 262         }
 263
 264         Directory &directory()
 265         {
 266                 return const_cast<Directory &>(
 267                         static_cast<const Item &>(*this).directory());
 268         }
 269         Song &song()
 270         {
 271                 return const_cast<Song &>(
 272                         static_cast<const Item &>(*this).song());
 273         }
 274         Playlist &playlist()
 275         {
 276                 return const_cast<Playlist &>(
 277                         static_cast<const Item &>(*this).playlist());
 278         }
 279
 280         const Directory &directory() const
 281         {
 282                 assert(m_type == Type::Directory);
 283                 return m_directory;
 284         }
 285         const Song &song() const
 286         {
 287                 assert(m_type == Type::Song);
 288                 return m_song;
 289         }
 290         const Playlist &playlist() const
 291         {
 292                 assert(m_type == Type::Playlist);
 293                 return m_playlist;
 294         }
 295
 296 private:
 297         Type m_type;
 298         Directory m_directory;
 299         Song m_song;
 300         Playlist m_playlist;
 301 };
 302
 303 struct Output
 304 {
 305         Output() { }
 306         Output(mpd_output *output)
 307         : m_output(output, mpd_output_free)
 308         { }
 309
 310         bool operator==(const Output &rhs) const
 311         {
 312                 if (empty() && rhs.empty())
 313                         return true;
 314                 else if (!empty() && !rhs.empty())
 315                         return id() == rhs.id()
 316                             && strcmp(name(), rhs.name()) == 0
 317                             && enabled() == rhs.enabled();
 318                 else
 319                         return false;
 320         }
 321         bool operator!=(const Output &rhs) const
 322         {
 323                 return !(*this == rhs);
 324         }
 325
 326         unsigned id() const
 327         {
 328                 assert(m_output.get() != nullptr);
 329                 return mpd_output_get_id(m_output.get());
 330         }
 331         const char *name() const
 332         {
 333                 assert(m_output.get() != nullptr);
 334                 return mpd_output_get_name(m_output.get());
 335         }
 336         bool enabled() const
 337         {
 338                 assert(m_output.get() != nullptr);
 339                 return mpd_output_get_enabled(m_output.get());
 340         }
 341
 342         bool empty() const { return m_output.get() == nullptr; }
 343
 344 private:
 345         std::shared_ptr<mpd_output> m_output;
 346 };
 347
 348 template <typename ObjectT>
 349 struct Iterator: std::iterator<std::input_iterator_tag, ObjectT>
 350 {
 351         // shared state of the iterator
 352         struct State
 353         {
 354                 friend Iterator;
 355
 356                 typedef std::function<bool(State &)> Fetcher;
 357
 358                 State(mpd_connection *connection_, Fetcher fetcher)
 359                 : m_connection(connection_)
 360                 , m_fetcher(fetcher)
 361                 {
 362                         assert(m_connection != nullptr);
 363                         assert(m_fetcher != nullptr);
 364                 }
 365                 ~State()
 366                 {
 367                         mpd_response_finish(m_connection);
 368                 }
 369
 370                 mpd_connection *connection() const
 371                 {
 372                         return m_connection;
 373                 }
 374
 375                 void setObject(ObjectT object)
 376                 {
 377                         if (hasObject())
 378                                 *m_object = std::move(object);
 379                         else
 380                                 m_object.reset(new ObjectT(std::move(object)));
 381                 }
 382
 383         private:
 384                 bool operator==(const State &rhs) const
 385                 {
 386                         return m_connection == rhs.m_connection
 387                             && m_object == m_object;
 388                 }
 389                 bool operator!=(const State &rhs) const
 390                 {
 391                         return !(*this == rhs);
 392                 }
 393
 394                 bool fetch()
 395                 {
 396                         return m_fetcher(*this);
 397                 }
 398                 ObjectT &getObject() const
 399                 {
 400                         return *m_object;
 401                 }
 402                 bool hasObject() const
 403                 {
 404                         return m_object.get() != nullptr;
 405                 }
 406
 407                 mpd_connection *m_connection;
 408                 Fetcher m_fetcher;
 409                 std::unique_ptr<ObjectT> m_object;
 410         };
 411
 412         Iterator()
 413         : m_state(nullptr)
 414         { }
 415         Iterator(mpd_connection *connection, typename State::Fetcher fetcher)
 416         : m_state(std::make_shared<State>(connection, std::move(fetcher)))
 417         {
 418                 // get the first element
 419                 ++*this;
 420         }
 421         ~Iterator()
 422         {
 423                 if (m_state)
 424                         checkConnectionErrors(m_state->connection());
 425         }
 426
 427         void finish()
 428         {
 429                 assert(m_state);
 430                 // check errors and change the iterator into end iterator
 431                 checkConnectionErrors(m_state->connection());
 432                 m_state = nullptr;
 433         }
 434
 435         ObjectT &operator*() const
 436         {
 437                 if (!m_state)
 438                         throw std::runtime_error("no object associated with the iterator");
 439                 assert(m_state->hasObject());
 440                 return m_state->getObject();
 441         }
 442         ObjectT *operator->() const
 443         {
 444                 return &**this;
 445         }
 446
 447         Iterator &operator++()
 448         {
 449                 assert(m_state);
 450                 if (!m_state->fetch())
 451                         finish();
 452                 return *this;
 453         }
 454         Iterator operator++(int)
 455         {
 456                 Iterator it(*this);
 457                 ++*this;
 458                 return it;
 459         }
 460
 461         bool operator==(const Iterator &rhs) const
 462         {
 463                 return m_state == rhs.m_state;
 464         }
 465         bool operator!=(const Iterator &rhs) const
 466         {
 467                 return !(*this == rhs);
 468         }
 469
 470 private:
 471         std::shared_ptr<State> m_state;
 472 };
 473
 474 typedef Iterator<Directory> DirectoryIterator;
 475 typedef Iterator<Item> ItemIterator;
 476 typedef Iterator<Output> OutputIterator;
 477 typedef Iterator<Playlist> PlaylistIterator;
 478 typedef Iterator<Song> SongIterator;
 479 typedef Iterator<std::string> StringIterator;
 480
 481 struct Connection
 482 {
 483         Connection();
 484
 485         void Connect();
 486         bool Connected() const;
 487         void Disconnect();
 488
 489         const std::string &GetHostname() { return m_host; }
 490         int GetPort() { return m_port; }
 491
 492         unsigned Version() const;
 493
 494         int GetFD() const { return m_fd; }
 495
 496         void SetHostname(const std::string &);
 497         void SetPort(int port) { m_port = port; }
 498         void SetTimeout(int timeout) { m_timeout = timeout; }
 499         void SetPassword(const std::string &password) { m_password = password; }
 500         void SendPassword();
 501
 502         Statistics getStatistics();
 503         Status getStatus();
 504
 505         void UpdateDirectory(const std::string &);
 506
 507         void Play();
 508         void Play(int);
 509         void PlayID(int);
 510         void Pause(bool);
 511         void Toggle();
 512         void Stop();
 513         void Next();
 514         void Prev();
 515         void Move(unsigned int from, unsigned int to);
 516         void Swap(unsigned, unsigned);
 517         void Seek(unsigned int pos, unsigned int where);
 518         void Shuffle();
 519         void ShuffleRange(unsigned start, unsigned end);
 520         void ClearMainPlaylist();
 521
 522         SongIterator GetPlaylistChanges(unsigned);
 523
 524         Song GetCurrentSong();
 525         Song GetSong(const std::string &);
 526         SongIterator GetPlaylistContent(const std::string &name);
 527         SongIterator GetPlaylistContentNoInfo(const std::string &name);
 528
 529         StringIterator GetSupportedExtensions();
 530
 531         void SetRepeat(bool);
 532         void SetRandom(bool);
 533         void SetSingle(bool);
 534         void SetConsume(bool);
 535         void SetCrossfade(unsigned);
 536         void SetVolume(unsigned int vol);
 537
 538         std::string GetReplayGainMode();
 539         void SetReplayGainMode(ReplayGainMode);
 540
 541         void SetPriority(const MPD::Song &s, int prio);
 542
 543         int AddSong(const std::string &, int = -1); // returns id of added song
 544         int AddSong(const Song &, int = -1); // returns id of added song
 545         bool AddRandomTag(mpd_tag_type, size_t, std::mt19937 &rng);
 546         bool AddRandomSongs(size_t number, std::mt19937 &rng);
 547         void Add(const std::string &path);
 548         void Delete(unsigned int pos);
 549         void PlaylistDelete(const std::string &playlist, unsigned int pos);
 550         void StartCommandsList();
 551         void CommitCommandsList();
 552
 553         void DeletePlaylist(const std::string &name);
 554         void LoadPlaylist(const std::string &name);
 555         void SavePlaylist(const std::string &);
 556         void ClearPlaylist(const std::string &playlist);
 557         void AddToPlaylist(const std::string &, const Song &);
 558         void AddToPlaylist(const std::string &, const std::string &);
 559         void PlaylistMove(const std::string &path, int from, int to);
 560         void Rename(const std::string &from, const std::string &to);
 561
 562         void StartSearch(bool);
 563         void StartFieldSearch(mpd_tag_type);
 564         void AddSearch(mpd_tag_type item, const std::string &str) const;
 565         void AddSearchAny(const std::string &str) const;
 566         void AddSearchURI(const std::string &str) const;
 567         SongIterator CommitSearchSongs();
 568
 569         PlaylistIterator GetPlaylists();
 570         StringIterator GetList(mpd_tag_type type);
 571         ItemIterator GetDirectory(const std::string &directory);
 572         SongIterator GetDirectoryRecursive(const std::string &directory);
 573         SongIterator GetSongs(const std::string &directory);
 574         DirectoryIterator GetDirectories(const std::string &directory);
 575
 576         OutputIterator GetOutputs();
 577         void EnableOutput(int id);
 578         void DisableOutput(int id);
 579
 580         StringIterator GetURLHandlers();
 581         StringIterator GetTagTypes();
 582
 583         void idle();
 584         int noidle();
 585
 586 private:
 587         struct ConnectionDeleter {
 588                 void operator()(mpd_connection *connection) {
 589                         mpd_connection_free(connection);
 590                 }
 591         };
 592
 593         void checkConnection() const;
 594         void prechecks();
 595         void prechecksNoCommandsList();
 596         void checkErrors() const;
 597
 598         std::unique_ptr<mpd_connection, ConnectionDeleter> m_connection;
 599         bool m_command_list_active;
 600
 601         int m_fd;
 602         bool m_idle;
 603
 604         std::string m_host;
 605         int m_port;
 606         int m_timeout;
 607         std::string m_password;
 608 };
 609
 610 }
 611
 612 extern MPD::Connection Mpd;
 613
 614 #endif // NCMPCPP_MPDPP_H