#include "modules/jack.hpp"

waybar::modules::JACK::JACK(const std::string &id, const Json::Value &config)
    : ALabel(config, "jack", id, "{load}%", 1) {
  xruns_ = 0;
  state_ = "disconnected";
  client_ = NULL;

  state_ = JACKState();
  thread_ = [this] {
    dp.emit();
    thread_.sleep_for(interval_);
  };
}

std::string waybar::modules::JACK::JACKState() {
  if (state_.compare("xrun") == 0) return "xrun";
  if (state_.compare("connected") == 0) return "connected";

  client_ = jack_client_open("waybar", JackNoStartServer, NULL);
  if (client_) {
    pthread_t jack_thread = jack_client_thread_id(client_);
    if (config_["realtime"].isBool() && !config_["realtime"].asBool())
      jack_drop_real_time_scheduling(jack_thread);

    bufsize_ = jack_get_buffer_size(client_);
    samplerate_ = jack_get_sample_rate(client_);
    jack_set_buffer_size_callback(client_, bufSizeCallback, this);
    jack_set_xrun_callback(client_, xrunCallback, this);
    jack_on_shutdown(client_, shutdownCallback, this);

    if (!jack_activate(client_)) return "connected";
  }
  return "disconnected";
}

auto waybar::modules::JACK::update() -> void {
  std::string format;
  float latency = 1000 * (float)bufsize_ / (float)samplerate_;
  auto state = JACKState();
  float load;

  if (label_.get_style_context()->has_class("xrun")) {
    label_.get_style_context()->remove_class("xrun");
    state = "connected";
  }

  if (state.compare("disconnected") != 0)
    load = jack_cpu_load(client_);
  else {
    load = 0;
    bufsize_ = 0;
    samplerate_ = 0;
    latency = 0;
  }

  if (label_.get_style_context()->has_class(state_))
    label_.get_style_context()->remove_class(state_);

  if (config_["format-" + state].isString()) {
    format = config_["format-" + state].asString();
  } else if (config_["format"].isString()) {
    format = config_["format"].asString();
  } else
    format = "DSP {load}%";

  if (!label_.get_style_context()->has_class(state))
    label_.get_style_context()->add_class(state);
  state_ = state;

  label_.set_markup(fmt::format(format, fmt::arg("load", std::round(load)),
                                fmt::arg("bufsize", bufsize_), fmt::arg("samplerate", samplerate_),
                                fmt::arg("latency", fmt::format("{:.2f}", latency)),
                                fmt::arg("xruns", xruns_)));

  if (tooltipEnabled()) {
    std::string tooltip_format = "{bufsize}/{samplerate} {latency}ms";
    if (config_["tooltip-format"].isString()) tooltip_format = config_["tooltip-format"].asString();
    label_.set_tooltip_text(fmt::format(tooltip_format, fmt::arg("load", std::round(load)),
                                       	fmt::arg("bufsize", bufsize_),
                                       	fmt::arg("samplerate", samplerate_),
                                        fmt::arg("latency", fmt::format("{:.2f}", latency)),
                                        fmt::arg("xruns", xruns_)));

  }

  // Call parent update
  ALabel::update();
}

int waybar::modules::JACK::bufSize(unsigned int size) {
  bufsize_ = size;
  return size;
}

int waybar::modules::JACK::xrun() {
  xruns_ += 1;
  state_ = "xrun";
  return 0;
}

void waybar::modules::JACK::shutdown() {
  client_ = NULL;
  state_ = "disconnected";
  xruns_ = 0;
}

int bufSizeCallback(unsigned int size, void *obj) {
  return static_cast<waybar::modules::JACK *>(obj)->bufSize(size);
}

int xrunCallback(void *obj) { return static_cast<waybar::modules::JACK *>(obj)->xrun(); }

void shutdownCallback(void *obj) { return static_cast<waybar::modules::JACK *>(obj)->shutdown(); }