Add scratchpad mode for sway with icons
parent
f4da203915
commit
2df0e7cf9e
GPG Key ID:
43ACB900522EA740
|
|
@ -21,11 +21,19 @@
|
|||
#include "giomm/desktopappinfo.h"
|
||||
#include "util/json.hpp"
|
||||
#include "wlr-foreign-toplevel-management-unstable-v1-client-protocol.h"
|
||||
#include "modules/sway/ipc/client.hpp"
|
||||
|
||||
namespace waybar::modules::wlr {
|
||||
|
||||
class Taskbar;
|
||||
|
||||
class SwayScratchpad {
|
||||
public:
|
||||
int id;
|
||||
std::string app_id;
|
||||
std::string title;
|
||||
};
|
||||
|
||||
class Task {
|
||||
public:
|
||||
Task(const waybar::Bar &, const Json::Value &, Taskbar *,
|
||||
|
|
@ -54,6 +62,7 @@ class Task {
|
|||
struct wl_seat *seat_;
|
||||
|
||||
uint32_t id_;
|
||||
uint32_t swayId_;
|
||||
|
||||
Gtk::Box content_;
|
||||
Gtk::Image icon_;
|
||||
|
|
@ -62,6 +71,8 @@ class Task {
|
|||
Glib::RefPtr<Gio::DesktopAppInfo> app_info_;
|
||||
bool button_visible_ = false;
|
||||
bool ignored_ = false;
|
||||
bool only_scratchpad_ = false;
|
||||
bool isScratchpadMode_ = false;
|
||||
|
||||
bool with_icon_ = false;
|
||||
bool with_name_ = false;
|
||||
|
|
@ -86,12 +97,15 @@ class Task {
|
|||
bool image_load_icon(Gtk::Image &image, const Glib::RefPtr<Gtk::IconTheme> &icon_theme,
|
||||
Glib::RefPtr<Gio::DesktopAppInfo> app_info, int size);
|
||||
void hide_if_ignored();
|
||||
bool hideNoScratchpad();
|
||||
|
||||
public:
|
||||
/* Getter functions */
|
||||
uint32_t id() const { return id_; }
|
||||
uint32_t swayId() const { return swayId_; }
|
||||
std::string title() const { return title_; }
|
||||
std::string app_id() const { return app_id_; }
|
||||
bool isScratchpadMode() const { return isScratchpadMode_; }
|
||||
uint32_t state() const { return state_; }
|
||||
bool maximized() const { return state_ & MAXIMIZED; }
|
||||
bool minimized() const { return state_ & MINIMIZED; }
|
||||
|
|
@ -117,6 +131,8 @@ class Task {
|
|||
void handle_drag_data_received(const Glib::RefPtr<Gdk::DragContext> &context, int x, int y,
|
||||
Gtk::SelectionData selection_data, guint info, guint time);
|
||||
|
||||
void setManualForSway(const int id, const std::string app_id, const std::string title);
|
||||
|
||||
public:
|
||||
bool operator==(const Task &) const;
|
||||
bool operator!=(const Task &) const;
|
||||
|
|
@ -153,6 +169,15 @@ class Taskbar : public waybar::AModule {
|
|||
struct zwlr_foreign_toplevel_manager_v1 *manager_;
|
||||
struct wl_seat *seat_;
|
||||
|
||||
/* Sway scratchpad indicator */
|
||||
std::mutex mutex_;
|
||||
waybar::modules::sway::Ipc ipc_;
|
||||
auto getSwayTree() -> void;
|
||||
auto onSwayCmd(const struct waybar::modules::sway::Ipc::ipc_response&) -> void;
|
||||
auto onSwayEvent(const struct waybar::modules::sway::Ipc::ipc_response&) -> void;
|
||||
util::JsonParser parser_;
|
||||
std::map<std::string, Task> swayWorkspaceMap;
|
||||
|
||||
public:
|
||||
/* Callbacks for global registration */
|
||||
void register_manager(struct wl_registry *, uint32_t name, uint32_t version);
|
||||
|
|
|
|||
|
|
@ -23,6 +23,11 @@
|
|||
#include "util/gtk_icon.hpp"
|
||||
#include "util/rewrite_string.hpp"
|
||||
#include "util/string.hpp"
|
||||
#include <stdio.h>
|
||||
#include "modules/sway/ipc/ipc.hpp"
|
||||
#include "modules/sway/ipc/client.hpp"
|
||||
#include <cstdlib>
|
||||
#include <string>
|
||||
|
||||
namespace waybar::modules::wlr {
|
||||
|
||||
|
|
@ -278,7 +283,10 @@ Task::Task(const waybar::Bar &bar, const Json::Value &config, Taskbar *tbar,
|
|||
seat_{seat},
|
||||
id_{global_id++},
|
||||
content_{bar.orientation, 0} {
|
||||
zwlr_foreign_toplevel_handle_v1_add_listener(handle_, &toplevel_handle_impl, this);
|
||||
|
||||
if (handle_) {
|
||||
zwlr_foreign_toplevel_handle_v1_add_listener(handle_, &toplevel_handle_impl, this);
|
||||
}
|
||||
|
||||
button.set_relief(Gtk::RELIEF_NONE);
|
||||
|
||||
|
|
@ -333,6 +341,9 @@ Task::Task(const waybar::Bar &bar, const Json::Value &config, Taskbar *tbar,
|
|||
config_["on-click-right"].isString()) {
|
||||
}
|
||||
|
||||
/* Only show apps within scratchpad */
|
||||
only_scratchpad_ = config_["only-scratchpad"].asBool();
|
||||
|
||||
button.add_events(Gdk::BUTTON_PRESS_MASK);
|
||||
button.signal_button_release_event().connect(sigc::mem_fun(*this, &Task::handle_clicked), false);
|
||||
|
||||
|
|
@ -360,6 +371,7 @@ Task::~Task() {
|
|||
|
||||
std::string Task::repr() const {
|
||||
std::stringstream ss;
|
||||
//std::cout << "##### ID: " << id_ << " Title: " << title_ << " App_ID: " <<app_id_ << "\n";
|
||||
ss << "Task (" << id_ << ") " << title_ << " [" << app_id_ << "] <" << (active() ? "A" : "a")
|
||||
<< (maximized() ? "M" : "m") << (minimized() ? "I" : "i") << (fullscreen() ? "F" : "f") << ">";
|
||||
|
||||
|
|
@ -387,8 +399,21 @@ void Task::handle_title(const char *title) {
|
|||
hide_if_ignored();
|
||||
}
|
||||
|
||||
void Task::setManualForSway(const int id, const std::string app_id, const std::string title) {
|
||||
swayId_ = id;
|
||||
handle_app_id(app_id.c_str());
|
||||
title_ = title;
|
||||
name_ = "";
|
||||
isScratchpadMode_ = true;
|
||||
|
||||
//std::cout << "[[[[]]]] Having id " << global_id << "with app_id " << app_id_ << " and title: " << name_;
|
||||
hide_if_ignored();
|
||||
handle_output_enter(nullptr);
|
||||
handle_done();
|
||||
}
|
||||
|
||||
void Task::hide_if_ignored() {
|
||||
if (tbar_->ignore_list().count(app_id_) || tbar_->ignore_list().count(title_)) {
|
||||
if (tbar_->ignore_list().count(app_id_) || tbar_->ignore_list().count(title_) || hideNoScratchpad()) {
|
||||
ignored_ = true;
|
||||
if (button_visible_) {
|
||||
auto output = gdk_wayland_monitor_get_wl_output(bar_.output->monitor->gobj());
|
||||
|
|
@ -404,6 +429,13 @@ void Task::hide_if_ignored() {
|
|||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Checks if the current app_id_ should be hidden if it is not a scratchpad element
|
||||
*/
|
||||
bool Task::hideNoScratchpad() {
|
||||
return only_scratchpad_ && !isScratchpadMode_;
|
||||
}
|
||||
|
||||
void Task::handle_app_id(const char *app_id) {
|
||||
if (app_id_.empty()) {
|
||||
spdlog::debug(fmt::format("Task ({}) setting app_id to {}", id_, app_id));
|
||||
|
|
@ -455,7 +487,7 @@ void Task::handle_output_enter(struct wl_output *output) {
|
|||
|
||||
spdlog::debug("{} entered output {}", repr(), (void *)output);
|
||||
|
||||
if (!button_visible_ && (tbar_->all_outputs() || tbar_->show_output(output))) {
|
||||
if (!button_visible_ && (isScratchpadMode_ || tbar_->all_outputs() || tbar_->show_output(output))) {
|
||||
/* The task entered the output of the current bar make the button visible */
|
||||
tbar_->add_button(button);
|
||||
button.show();
|
||||
|
|
@ -467,7 +499,7 @@ void Task::handle_output_enter(struct wl_output *output) {
|
|||
void Task::handle_output_leave(struct wl_output *output) {
|
||||
spdlog::debug("{} left output {}", repr(), (void *)output);
|
||||
|
||||
if (button_visible_ && !tbar_->all_outputs() && tbar_->show_output(output)) {
|
||||
if (button_visible_ && (isScratchpadMode_ || (!tbar_->all_outputs() && tbar_->show_output(output)) )) {
|
||||
/* The task left the output of the current bar, make the button invisible */
|
||||
tbar_->remove_button(button);
|
||||
button.hide();
|
||||
|
|
@ -523,7 +555,9 @@ void Task::handle_done() {
|
|||
|
||||
void Task::handle_closed() {
|
||||
spdlog::debug("{} closed", repr());
|
||||
zwlr_foreign_toplevel_handle_v1_destroy(handle_);
|
||||
if (handle_) {
|
||||
zwlr_foreign_toplevel_handle_v1_destroy(handle_);
|
||||
}
|
||||
handle_ = nullptr;
|
||||
if (button_visible_) {
|
||||
tbar_->remove_button(button);
|
||||
|
|
@ -549,6 +583,27 @@ bool Task::handle_clicked(GdkEventButton *bt) {
|
|||
else if (config_["on-click-right"].isString() && bt->button == 3)
|
||||
action = config_["on-click-right"].asString();
|
||||
|
||||
// We can't handle click events because we have no wayland handler.
|
||||
// So we implement it on our own :)
|
||||
if (action == "close" || action == "activate") {
|
||||
// Change focus to workspace
|
||||
std::string command = "swaymsg [con_id=" + std::to_string(swayId_) + "] scratchpad show";
|
||||
if (system(command.c_str()) != 0) {
|
||||
spdlog::warn("Error while execution sway command {}", command);
|
||||
}
|
||||
}
|
||||
// Kill the window
|
||||
if (action == "close") {
|
||||
auto command = "swaymsg kill";
|
||||
if (system(command) != 0) {
|
||||
spdlog::warn("Error while execution sway command {}", command);
|
||||
}
|
||||
}
|
||||
|
||||
if (isScratchpadMode_) {
|
||||
return true;
|
||||
}
|
||||
|
||||
if (action.empty())
|
||||
return true;
|
||||
else if (action == "activate")
|
||||
|
|
@ -788,8 +843,112 @@ Taskbar::Taskbar(const std::string &id, const waybar::Bar &bar, const Json::Valu
|
|||
}
|
||||
|
||||
icon_themes_.push_back(Gtk::IconTheme::get_default());
|
||||
|
||||
// Sway IPC register
|
||||
ipc_.subscribe(R"(["window"])");
|
||||
ipc_.signal_event.connect(sigc::mem_fun(*this, &Taskbar::onSwayEvent));
|
||||
ipc_.signal_cmd.connect(sigc::mem_fun(*this, &Taskbar::onSwayCmd));
|
||||
|
||||
getSwayTree();
|
||||
|
||||
ipc_.setWorker([this] {
|
||||
try {
|
||||
ipc_.handleEvent();
|
||||
} catch (const std::exception& e) {
|
||||
spdlog::error("Scratchpad: {}", e.what());
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
auto Taskbar::getSwayTree() -> void {
|
||||
try {
|
||||
ipc_.sendCmd(IPC_GET_TREE);
|
||||
} catch (const std::exception& e) {
|
||||
spdlog::error("Scratchpad: {}", e.what());
|
||||
}
|
||||
}
|
||||
|
||||
auto Taskbar::onSwayCmd(const struct waybar::modules::sway::Ipc::ipc_response& res) -> void {
|
||||
try {
|
||||
std::lock_guard<std::mutex> lock(mutex_);
|
||||
auto tree = parser_.parse(res.payload);
|
||||
|
||||
// Sort the sway IDs aufsteigend
|
||||
std::vector<SwayScratchpad> scratchpad;
|
||||
for (const auto& window: tree["nodes"][0]["nodes"][0]["floating_nodes"]) {
|
||||
SwayScratchpad scratch;
|
||||
scratch.id = window["id"].asInt();
|
||||
scratch.app_id = window["app_id"].asString();
|
||||
scratch.title = window["name"].asString();
|
||||
|
||||
scratchpad.push_back(scratch);
|
||||
}
|
||||
std::sort(scratchpad.begin(), scratchpad.end(), [](SwayScratchpad a, SwayScratchpad b) -> bool {
|
||||
return a.id < b.id;
|
||||
});
|
||||
|
||||
// We can't remove elements while iterating.
|
||||
// So we cache them to remove them after iterating
|
||||
std::vector<uint32_t> removals;
|
||||
|
||||
uint32_t lastId = 0;
|
||||
for (const auto& window: scratchpad) {
|
||||
|
||||
// The list of apps within the scratchpad is ordered ascending.
|
||||
// If we have an ID higher than the last one and the current one, we remove it.
|
||||
bool taskFound = false;
|
||||
uint32_t currentId = window.id;
|
||||
|
||||
for (auto &t : tasks_) {
|
||||
// Only use scratchpad tasks
|
||||
if (!t->isScratchpadMode()) {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (t->swayId() == currentId) {
|
||||
taskFound = true;
|
||||
}
|
||||
|
||||
if (t->swayId() > lastId && t->swayId() < currentId) {
|
||||
removals.push_back(t->id());
|
||||
}
|
||||
}
|
||||
|
||||
lastId = currentId;
|
||||
|
||||
// No entry found -> add a new one
|
||||
if (!taskFound) {
|
||||
tasks_.push_back(std::make_unique<Task>(bar_, config_, this, nullptr, seat_));
|
||||
const int i = tasks_.size() - 1;
|
||||
tasks_[i]->setManualForSway(window.id, window.app_id, window.title);
|
||||
}
|
||||
}
|
||||
|
||||
// Cleanup
|
||||
for (auto &t : tasks_) {
|
||||
// Only use scratchpad tasks
|
||||
if (!t->isScratchpadMode()) {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (t->swayId() > lastId) {
|
||||
removals.push_back(t->id());
|
||||
}
|
||||
}
|
||||
|
||||
// Remove them
|
||||
for (const auto& id: removals) {
|
||||
remove_task(id);
|
||||
}
|
||||
|
||||
dp.emit();
|
||||
} catch (const std::exception& e) {
|
||||
spdlog::error("Scratchpad: {}", e.what());
|
||||
}
|
||||
}
|
||||
|
||||
auto Taskbar::onSwayEvent(const struct waybar::modules::sway::Ipc::ipc_response& res) -> void { getSwayTree(); }
|
||||
|
||||
Taskbar::~Taskbar() {
|
||||
if (manager_) {
|
||||
struct wl_display *display = Client::inst()->wl_display;
|
||||
|
|
@ -810,6 +969,8 @@ Taskbar::~Taskbar() {
|
|||
}
|
||||
|
||||
void Taskbar::update() {
|
||||
std::lock_guard<std::mutex> lock(mutex_);
|
||||
|
||||
for (auto &t : tasks_) {
|
||||
t->update();
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue