Waybar/src/modules/sway/workspaces.cpp

#include "modules/sway/workspaces.hpp"

waybar::modules::sway::Workspaces::Workspaces(const std::string &id, const Bar &bar,
                                              const Json::Value &config)
    : bar_(bar),
      config_(config),
      box_(bar.vertical ? Gtk::ORIENTATION_VERTICAL : Gtk::ORIENTATION_HORIZONTAL, 0),
      scrolling_(false) {
  box_.set_name("workspaces");
  if (!id.empty()) {
    box_.get_style_context()->add_class(id);
  }
  ipc_.subscribe("[ \"workspace\" ]");
  // Launch worker
  worker();
}

void waybar::modules::sway::Workspaces::worker() {
  thread_ = [this] {
    try {
      if (!workspaces_.empty()) {
        ipc_.handleEvent();
      }
      {
        std::lock_guard<std::mutex> lock(mutex_);
        auto                        res = ipc_.sendCmd(IPC_GET_WORKSPACES);
        if (thread_.isRunning()) {
          workspaces_ = parser_.parse(res.payload);
        }
      }
      dp.emit();
    } catch (const std::exception &e) {
      std::cerr << "Workspaces: " << e.what() << std::endl;
    }
  };
}

auto waybar::modules::sway::Workspaces::update() -> void {
  bool                        needReorder = false;
  std::lock_guard<std::mutex> lock(mutex_);
  for (auto it = buttons_.begin(); it != buttons_.end();) {
    auto ws = std::find_if(workspaces_.begin(), workspaces_.end(), [it](auto node) -> bool {
      return node["name"].asString() == it->first;
    });
    if (ws == workspaces_.end() ||
        (!config_["all-outputs"].asBool() && (*ws)["output"].asString() != bar_.output->name)) {
      it          = buttons_.erase(it);
      needReorder = true;
    } else {
      ++it;
    }
  }
  for (auto const &node : workspaces_) {
    if (!config_["all-outputs"].asBool() && bar_.output->name != node["output"].asString()) {
      continue;
    }
    auto it = buttons_.find(node["name"].asString());
    if (it == buttons_.end()) {
      addWorkspace(node);
      needReorder = true;
    } else {
      auto &button = it->second;
      if (node["focused"].asBool()) {
        button.get_style_context()->add_class("focused");
      } else {
        button.get_style_context()->remove_class("focused");
      }
      if (node["visible"].asBool()) {
        button.get_style_context()->add_class("visible");
      } else {
        button.get_style_context()->remove_class("visible");
      }
      if (node["urgent"].asBool()) {
        button.get_style_context()->add_class("urgent");
      } else {
        button.get_style_context()->remove_class("urgent");
      }
      if (needReorder) {
        box_.reorder_child(button, getWorkspaceIndex(node["name"].asString()));
      }
      auto        icon   = getIcon(node["name"].asString(), node);
      std::string output = icon;
      if (config_["format"].isString()) {
        auto format = config_["format"].asString();
        output      = fmt::format(format,
                             fmt::arg("icon", icon),
                             fmt::arg("name", trimWorkspaceName(node["name"].asString())),
                             fmt::arg("index", node["num"].asString()));
      }
      if (!config_["disable-markup"].asBool()) {
        static_cast<Gtk::Label *>(button.get_children()[0])->set_markup(output);
      } else {
        button.set_label(output);
      }
      onButtonReady(node, button);
    }
  }
  if (scrolling_) {
    scrolling_ = false;
  }
}

void waybar::modules::sway::Workspaces::addWorkspace(const Json::Value &node) {
  auto icon   = getIcon(node["name"].asString(), node);
  auto format = config_["format"].isString()
                    ? fmt::format(config_["format"].asString(),
                                  fmt::arg("icon", icon),
                                  fmt::arg("name", trimWorkspaceName(node["name"].asString())),
                                  fmt::arg("index", node["num"].asString()))
                    : icon;
  auto  pair   = buttons_.emplace(node["name"].asString(), format);
  auto &button = pair.first->second;
  if (!config_["disable-markup"].asBool()) {
    static_cast<Gtk::Label *>(button.get_children()[0])->set_markup(format);
  }
  box_.pack_start(button, false, false, 0);
  button.set_relief(Gtk::RELIEF_NONE);
  button.signal_clicked().connect([this, pair] {
    try {
      std::lock_guard<std::mutex> lock(mutex_);
      auto                        cmd = fmt::format("workspace \"{}\"", pair.first->first);
      ipc_.sendCmd(IPC_COMMAND, cmd);
    } catch (const std::exception &e) {
      std::cerr << e.what() << std::endl;
    }
  });
  if (!config_["disable-scroll"].asBool()) {
    button.add_events(Gdk::SCROLL_MASK | Gdk::SMOOTH_SCROLL_MASK);
    button.signal_scroll_event().connect(sigc::mem_fun(*this, &Workspaces::handleScroll));
  }
  box_.reorder_child(button, getWorkspaceIndex(node["name"].asString()));
  if (node["focused"].asBool()) {
    button.get_style_context()->add_class("focused");
  }
  if (node["visible"].asBool()) {
    button.get_style_context()->add_class("visible");
  }
  if (node["urgent"].asBool()) {
    button.get_style_context()->add_class("urgent");
  }

  onButtonReady(node, button);
}

std::string waybar::modules::sway::Workspaces::getIcon(const std::string &name,
                                                       const Json::Value &node) {
  std::vector<std::string> keys = {name, "urgent", "focused", "visible", "default"};
  for (auto const &key : keys) {
    if (key == "focused" || key == "visible" || key == "urgent") {
      if (config_["format-icons"][key].isString() && node[key].asBool()) {
        return config_["format-icons"][key].asString();
      }
    } else if (config_["format-icons"][key].isString()) {
      return config_["format-icons"][key].asString();
    }
  }
  return name;
}

bool waybar::modules::sway::Workspaces::handleScroll(GdkEventScroll *e) {
  // Avoid concurrent scroll event
  if (scrolling_) {
    return false;
  }
  std::lock_guard<std::mutex> lock(mutex_);
  uint8_t                     idx;
  scrolling_ = true;
  for (idx = 0; idx < workspaces_.size(); idx += 1) {
    if (workspaces_[idx]["focused"].asBool()) {
      break;
    }
  }
  if (idx == workspaces_.size()) {
    scrolling_ = false;
    return false;
  }
  std::string name;
  if (e->direction == GDK_SCROLL_UP) {
    name = getCycleWorkspace(idx, true);
  }
  if (e->direction == GDK_SCROLL_DOWN) {
    name = getCycleWorkspace(idx, false);
  }
  if (e->direction == GDK_SCROLL_SMOOTH) {
    gdouble delta_x, delta_y;
    gdk_event_get_scroll_deltas(reinterpret_cast<const GdkEvent *>(e), &delta_x, &delta_y);
    if (delta_y < 0) {
      name = getCycleWorkspace(idx, true);
    } else if (delta_y > 0) {
      name = getCycleWorkspace(idx, false);
    }
  }
  if (name.empty() || name == workspaces_[idx]["name"].asString()) {
    scrolling_ = false;
    return false;
  }
  ipc_.sendCmd(IPC_COMMAND, fmt::format("workspace \"{}\"", name));
  std::this_thread::sleep_for(std::chrono::milliseconds(150));
  return true;
}

const std::string waybar::modules::sway::Workspaces::getCycleWorkspace(uint8_t focused_workspace,
                                                                       bool    prev) const {
  auto    inc  = prev ? -1 : 1;
  int     size = workspaces_.size();
  uint8_t idx  = 0;
  for (int i = focused_workspace; i < size && i >= 0; i += inc) {
    bool same_output = (workspaces_[i]["output"].asString() == bar_.output->name &&
                        !config_["all-outputs"].asBool()) ||
                       config_["all-outputs"].asBool();
    bool same_name =
        workspaces_[i]["name"].asString() == workspaces_[focused_workspace]["name"].asString();
    if (same_output && !same_name) {
      return workspaces_[i]["name"].asString();
    }
    if (prev && i - 1 < 0) {
      i = size;
    } else if (!prev && i + 1 >= size) {
      i = -1;
    } else if (idx >= workspaces_.size()) {
      return "";
    }
    idx += 1;
  }
  return "";
}

uint16_t waybar::modules::sway::Workspaces::getWorkspaceIndex(const std::string &name) const {
  uint16_t idx = 0;
  for (const auto &workspace : workspaces_) {
    if (workspace["name"].asString() == name) {
      return idx;
    }
    if (!(!config_["all-outputs"].asBool() &&
          workspace["output"].asString() != bar_.output->name)) {
      idx += 1;
    }
  }
  return workspaces_.size();
}

std::string waybar::modules::sway::Workspaces::trimWorkspaceName(std::string name) {
  std::size_t found = name.find(":");
  if (found != std::string::npos) {
    return name.substr(found + 1);
  }
  return name;
}

void waybar::modules::sway::Workspaces::onButtonReady(const Json::Value &node,
                                                      Gtk::Button &      button) {
  if (config_["current-only"].asBool()) {
    if (node["focused"].asBool()) {
      button.show();
    } else {
      button.hide();
    }
  } else {
    button.show();
  }
}

waybar::modules::sway::Workspaces::operator Gtk::Widget &() { return box_; }