conky/src/mpd.cc

/*
 *
 * Conky, a system monitor, based on torsmo
 *
 * Any original torsmo code is licensed under the BSD license
 *
 * All code written since the fork of torsmo is licensed under the GPL
 *
 * Please see COPYING for details
 *
 * Copyright (c) 2005-2018 Brenden Matthews, Philip Kovacs, et. al.
 *	(see AUTHORS)
 * All rights reserved.
 *
 * 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 3 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, see <http://www.gnu.org/licenses/>.
 *
 */

#include "mpd.h"
#include "conky.h"
#include "libmpdclient.h"
#include "logging.h"
#include "timeinfo.h"
#include "update-cb.hh"
#include <cmath>
#include <mutex>

namespace {

/* this is true if the current host was set from MPD_HOST */
bool mpd_environment_host = false;

class mpd_host_setting : public conky::simple_config_setting<std::string> {
  using Base = conky::simple_config_setting<std::string>;

 protected:
  void lua_setter(lua::state &l, bool init) override;

 public:
  mpd_host_setting() : Base("mpd_host", "localhost", false) {}
};

void mpd_host_setting::lua_setter(lua::state &l, bool init) {
  lua::stack_sentry s(l, -2);

  if (l.isnil(-2)) {
    // get the value from environment
    mpd_environment_host = true;
    const char *t = getenv("MPD_HOST");
    if (t != nullptr) {
      l.checkstack(1);
      const char *h = strchr(t, '@');
      if (h != nullptr) {
        if (h[1] != 0) {
          l.pushstring(h + 1);
        }
      } else {
        l.pushstring(t);
      }
      l.replace(-3);
    }
  }

  Base::lua_setter(l, init);

  ++s;
}

class mpd_password_setting : public conky::simple_config_setting<std::string> {
  using Base = conky::simple_config_setting<std::string>;

 protected:
  void lua_setter(lua::state &l, bool init) override;

 public:
  mpd_password_setting() : Base("mpd_password", std::string(), false) {}
};

void mpd_password_setting::lua_setter(lua::state &l, bool init) {
  lua::stack_sentry s(l, -2);

  /* for security, dont use environment password when user specifies host in
   * config */
  if (l.isnil(-2) && mpd_environment_host) {
    // get the value from environment
    const char *t = getenv("MPD_HOST");
    if (t != nullptr) {
      const char *p = strchr(t, '@');
      if (p != nullptr) {
        l.checkstack(1);
        l.pushstring(t, p - t);
        l.replace(-3);
      }
    }
  }

  Base::lua_setter(l, init);

  ++s;
}

conky::range_config_setting<in_port_t> mpd_port("mpd_port", 1, 65535, 6600,
                                                false);
mpd_host_setting mpd_host;
mpd_password_setting mpd_password;

struct mpd_result {
  std::string title;
  std::string artist;
  std::string albumartist;
  std::string album;
  std::string date;
  std::string status;
  std::string random;
  std::string repeat;
  std::string track;
  std::string name;
  std::string file;
  int is_playing{};
  int vol{};
  float progress{};
  int bitrate{};
  int length{};
  int elapsed{};
};

class mpd_cb : public conky::callback<mpd_result> {
  using Base = conky::callback<mpd_result>;

  mpd_Connection *conn;

 protected:
  void work() override;

 public:
  explicit mpd_cb(uint32_t period)
      : Base(period, false, Tuple()), conn(nullptr) {}

  ~mpd_cb() override {
    if (conn != nullptr) {
      mpd_closeConnection(conn);
    }
  }
};

void mpd_cb::work() {
  mpd_Status *status;
  mpd_InfoEntity *entity;
  mpd_result mpd_info;

  do {
    if (conn == nullptr) {
      conn = mpd_newConnection(mpd_host.get(*state).c_str(),
                               mpd_port.get(*state), 10);
    }

    if (static_cast<unsigned int>(!mpd_password.get(*state).empty()) != 0u) {
      mpd_sendPasswordCommand(conn, mpd_password.get(*state).c_str());
      mpd_finishCommand(conn);
    }

    if (conn->error != 0) {
      NORM_ERR("MPD error: %s\n", conn->errorStr);
      mpd_closeConnection(conn);
      conn = nullptr;

      mpd_info.status = "MPD not responding";
      break;
    }

    mpd_sendStatusCommand(conn);
    if ((status = mpd_getStatus(conn)) == nullptr) {
      NORM_ERR("MPD error: %s\n", conn->errorStr);
      mpd_closeConnection(conn);
      conn = nullptr;

      mpd_info.status = "MPD not responding";
      break;
    }
    mpd_finishCommand(conn);
    if ((conn == nullptr) || (conn->error != 0)) {
      // fprintf(stderr, "%s\n", conn->errorStr);
      mpd_closeConnection(conn);
      conn = nullptr;
      break;
    }

    mpd_info.vol = status->volume;
    if (status->random == 0) {
      mpd_info.random = "Off";
    } else if (status->random == 1) {
      mpd_info.random = "On";
    } else {
      mpd_info.random = "";
    }
    if (status->repeat == 0) {
      mpd_info.repeat = "Off";
    } else if (status->repeat == 1) {
      mpd_info.repeat = "On";
    } else {
      mpd_info.repeat = "";
    }
    /* if (status->error) {
       printf("error: %s\n", status->error);
       } */

    switch (status->state) {
      case MPD_STATUS_STATE_PLAY:
        mpd_info.status = "Playing";
        break;
      case MPD_STATUS_STATE_STOP:
        mpd_info.status = "Stopped";
        break;
      case MPD_STATUS_STATE_PAUSE:
        mpd_info.status = "Paused";
        break;
      default:
        mpd_info.status = "";
        break;
    }

    if (status->state == MPD_STATUS_STATE_PLAY ||
        status->state == MPD_STATUS_STATE_PAUSE) {
      mpd_info.is_playing = 1;
      mpd_info.bitrate = status->bitRate;
      mpd_info.progress =
          ((0 != status->totalTime) ?
           static_cast<float>(status->elapsedTime) / status->totalTime
           : 0.0);
      mpd_info.elapsed = status->elapsedTime;
      mpd_info.length = status->totalTime;
    } else {
      mpd_info.progress = 0;
      mpd_info.is_playing = 0;
      mpd_info.elapsed = 0;
    }

    if (conn->error != 0) {
      // fprintf(stderr, "%s\n", conn->errorStr);
      mpd_closeConnection(conn);
      conn = nullptr;
      break;
    }

    mpd_sendCurrentSongCommand(conn);
    while ((entity = mpd_getNextInfoEntity(conn)) != nullptr) {
      mpd_Song *song = entity->info.song;

      if (entity->type != MPD_INFO_ENTITY_TYPE_SONG) {
        mpd_freeInfoEntity(entity);
        continue;
      }
#define SETSTRING(a, b) \
  if (b)                \
    (a) = b;            \
  else                  \
    (a) = "";
      SETSTRING(mpd_info.artist, song->artist);
      SETSTRING(mpd_info.albumartist, song->albumartist);
      SETSTRING(mpd_info.album, song->album);
      SETSTRING(mpd_info.title, song->title);
      SETSTRING(mpd_info.date, song->date);
      SETSTRING(mpd_info.track, song->track);
      SETSTRING(mpd_info.name, song->name);
      SETSTRING(mpd_info.file, song->file);
      if (entity != nullptr) {
        mpd_freeInfoEntity(entity);
        entity = nullptr;
      }
    }
    mpd_finishCommand(conn);
    if ((conn != nullptr) && (conn->error != 0)) {
      // fprintf(stderr, "%s\n", conn->errorStr);
      mpd_closeConnection(conn);
      conn = nullptr;
      break;
    }

    if (conn->error != 0) {
      // fprintf(stderr, "%s\n", conn->errorStr);
      mpd_closeConnection(conn);
      conn = nullptr;
      break;
    }

    mpd_freeStatus(status);
    /* if (conn) {
       mpd_closeConnection(conn);
       conn = 0;
       } */
  } while (0);
  std::lock_guard<std::mutex> lock(Base::result_mutex);
  result = mpd_info;  // don't forget to save results!
}

mpd_result get_mpd() {
  uint32_t period = std::max(
      lround(music_player_interval.get(*state) / active_update_interval()), 1l);
  return conky::register_cb<mpd_cb>(period)->get_result_copy();
}
}  // namespace

static inline void format_media_player_time(char *buf, const int size,
                                            int seconds) {
  int days, hours, minutes;

  if (times_in_seconds.get(*state)) {
    snprintf(buf, size, "%d", seconds);
    return;
  }

  days = seconds / (24 * 60 * 60);
  seconds %= (24 * 60 * 60);
  hours = seconds / (60 * 60);
  seconds %= (60 * 60);
  minutes = seconds / 60;
  seconds %= 60;

  if (days > 0) {
    snprintf(buf, size, "%i days %i:%02i:%02i", days, hours, minutes, seconds);
  } else if (hours > 0) {
    snprintf(buf, size, "%i:%02i:%02i", hours, minutes, seconds);
  } else {
    snprintf(buf, size, "%i:%02i", minutes, seconds);
  }
}

void print_mpd_elapsed(struct text_object *obj, char *p, int p_max_size) {
  (void)obj;
  format_media_player_time(p, p_max_size, get_mpd().elapsed);
}

void print_mpd_length(struct text_object *obj, char *p, int p_max_size) {
  (void)obj;
  format_media_player_time(p, p_max_size, get_mpd().length);
}

uint8_t mpd_percentage(struct text_object *obj) {
  (void)obj;
  return round_to_int(get_mpd().progress * 100.0f);
}

double mpd_barval(struct text_object *obj) {
  (void)obj;
  return get_mpd().progress;
}

void print_mpd_smart(struct text_object *obj, char *p, int p_max_size) {
  const mpd_result &mpd_info = get_mpd();
  int len = obj->data.i;
  if (len == 0 || len > p_max_size) {
    len = p_max_size;
  }

  memset(p, 0, p_max_size);
  if ((static_cast<unsigned int>(!mpd_info.artist.empty()) != 0u) &&
      (static_cast<unsigned int>(!mpd_info.title.empty()) != 0u)) {
    snprintf(p, len, "%s - %s", mpd_info.artist.c_str(),
             mpd_info.title.c_str());
  } else if (static_cast<unsigned int>(!get_mpd().title.empty()) != 0u) {
    snprintf(p, len, "%s", mpd_info.title.c_str());
  } else if (static_cast<unsigned int>(!mpd_info.artist.empty()) != 0u) {
    snprintf(p, len, "%s", mpd_info.artist.c_str());
  } else if (static_cast<unsigned int>(!mpd_info.file.empty()) != 0u) {
    snprintf(p, len, "%s", mpd_info.file.c_str());
  } else {
    *p = 0;
  }
}

int check_mpd_playing(struct text_object *obj) {
  (void)obj;
  return get_mpd().is_playing;
}

#define MPD_PRINT_GENERATOR(name, fmt, acc)                                 \
  void print_mpd_##name(struct text_object *obj, char *p, int p_max_size) { \
    if (obj->data.i && obj->data.i < p_max_size) p_max_size = obj->data.i;  \
    snprintf(p, p_max_size, fmt, get_mpd().name acc);                       \
  }

MPD_PRINT_GENERATOR(title, "%s", .c_str())
MPD_PRINT_GENERATOR(artist, "%s", .c_str())
MPD_PRINT_GENERATOR(albumartist, "%s", .c_str())
MPD_PRINT_GENERATOR(album, "%s", .c_str())
MPD_PRINT_GENERATOR(date, "%s", .c_str())
MPD_PRINT_GENERATOR(random, "%s", .c_str())
MPD_PRINT_GENERATOR(repeat, "%s", .c_str())
MPD_PRINT_GENERATOR(track, "%s", .c_str())
MPD_PRINT_GENERATOR(name, "%s", .c_str())
MPD_PRINT_GENERATOR(file, "%s", .c_str())
MPD_PRINT_GENERATOR(vol, "%d", )
MPD_PRINT_GENERATOR(bitrate, "%d", )
MPD_PRINT_GENERATOR(status, "%s", .c_str())

#undef MPD_PRINT_GENERATOR