Compare commits

...

37 Commits
0.0.3 ... 0.0.6

Author SHA1 Message Date
53956d9d18 feat(ALabel): Toggleable labels 2018-08-27 01:36:25 +02:00
e9478f548e chore: add mediaplayer script 2018-08-26 21:47:35 +02:00
c8ca8b3725 fix(Custom): hide label when exec-if failed 2018-08-26 21:41:34 +02:00
0ad2bc7516 refactor(Network): clean nl socket 2018-08-24 15:32:06 +02:00
0dba3abc1d fix(custom): do not take the custom module ref 2018-08-21 10:50:09 +02:00
8be67d5008 chore: optional deps 2018-08-20 17:20:02 +02:00
49232eed8d Clean (#31) 2018-08-20 14:50:45 +02:00
b7e3d10fb7 revert(workspaces): ipc command out of update func 2018-08-20 00:19:27 +02:00
8ce33e0c64 fix(window): pick only con title 2018-08-19 20:37:33 +02:00
969c1ceedd chore: v0.0.5 2018-08-19 13:43:41 +02:00
52a4e761a8 fix(workspaces): avoid useless mutex lock 2018-08-19 13:43:00 +02:00
16b856c8bc fix: remove debug flag 2018-08-19 13:41:22 +02:00
6705134034 Handle screens disconnection (#29) 2018-08-19 13:39:57 +02:00
ce50a627be refactor: move command execution into their own file 2018-08-18 17:54:20 +02:00
b794ca63d1 feat(custom): exec-if 2018-08-18 17:27:40 +02:00
38ede5b3d5 refactor(ipc): clean 2018-08-18 16:01:56 +02:00
27dfffa4e3 refactor: style issue 2018-08-18 15:05:18 +02:00
b1fd4d7b82 feat(modules): generic label module to allow max-length on all labels 2018-08-18 11:43:48 +02:00
c128562284 feat(bar): clean exit 2018-08-17 20:28:26 +02:00
d280f5e8bd Network detect (#26) 2018-08-17 14:24:00 +02:00
0603b99714 fix(bar): proper center modules 2018-08-16 18:11:16 +02:00
0371271465 fix(custom): hide first 2018-08-16 17:59:45 +02:00
93f87f322f chore: v0.0.4 2018-08-16 17:19:02 +02:00
8768183f3d fea(workspaces): add disable-scroll config 2018-08-16 17:12:45 +02:00
e4f35d7ca0 fea(custom): add max-length config 2018-08-16 17:09:51 +02:00
57f3a01a5b refactor: remove assert 2018-08-16 15:41:09 +02:00
6635548d3e Style code (#25) 2018-08-16 14:29:41 +02:00
3fdc50163d feat(window): update when window title change 2018-08-16 00:02:57 +02:00
a9246a09eb feat(workspaces): add a option to show all workspaces from all outputs 2018-08-15 22:19:17 +02:00
3ed3416d75 fix(config): update sway workspaces key 2018-08-15 21:03:49 +02:00
008856cbb8 feat(clock): allow choose interval 2018-08-15 21:00:04 +02:00
608b791ac1 refactor(clock): use fmt::localtime 2018-08-15 20:53:27 +02:00
d427512d7d chore: update README 2018-08-15 20:18:00 +02:00
f94598c138 feat(sway): add focused window name 2018-08-15 20:17:17 +02:00
9b75302d22 refactor(client): cleanup 2018-08-15 17:31:45 +02:00
be66cc2dd1 feat(workspaces): add urgent, visible class 2018-08-15 15:03:51 +02:00
52e7b6148b feat(workspaces): add class to button when label is a icon 2018-08-15 14:58:55 +02:00
43 changed files with 1766 additions and 1133 deletions

View File

@ -6,6 +6,7 @@
**Current features** **Current features**
- Sway Workspaces - Sway Workspaces
- Sway focused window name
- Local time - Local time
- Battery - Battery
- Network - Network
@ -13,6 +14,7 @@
- Memory - Memory
- Cpu load average - Cpu load average
- Custom scripts - Custom scripts
- And much more customizations
**Configuration and Customization** **Configuration and Customization**
@ -27,7 +29,8 @@ $ ninja -C build
$ ./build/waybar $ ./build/waybar
``` ```
Contributions welcome! - have fun :) Contributions welcome! - have fun :)<br>
The style guidelines is [Google's](https://google.github.io/styleguide/cppguide.html)
## License ## License

26
include/ALabel.hpp Normal file
View File

@ -0,0 +1,26 @@
#pragma once
#include <json/json.h>
#include "IModule.hpp"
namespace waybar {
class ALabel : public IModule {
public:
ALabel(const Json::Value&, const std::string format);
virtual ~ALabel() = default;
virtual auto update() -> void;
virtual std::string getIcon(uint16_t percentage);
virtual operator Gtk::Widget &();
protected:
Gtk::EventBox event_box_;
Gtk::Label label_;
const Json::Value& config_;
std::string format_;
private:
bool handleToggle(GdkEventButton* const& ev);
bool alt = false;
const std::string default_format_;
};
}

View File

@ -3,10 +3,13 @@
#include <gtkmm.h> #include <gtkmm.h>
namespace waybar { namespace waybar {
class IModule {
public: class IModule {
virtual ~IModule() {} public:
virtual auto update() -> void = 0; virtual ~IModule() = default;
virtual operator Gtk::Widget &() = 0; virtual auto update() -> void = 0;
}; virtual operator Gtk::Widget &() = 0;
Glib::Dispatcher dp; // Hmmm Maybe I should create an abstract class ?
};
} }

View File

@ -4,47 +4,56 @@
#include <gtkmm.h> #include <gtkmm.h>
#include "wlr-layer-shell-unstable-v1-client-protocol.h" #include "wlr-layer-shell-unstable-v1-client-protocol.h"
#include "xdg-output-unstable-v1-client-protocol.h" #include "xdg-output-unstable-v1-client-protocol.h"
#include "IModule.hpp"
namespace waybar { namespace waybar {
struct Client; class Client;
class Factory;
struct Bar { class Bar {
Bar(Client& client, std::unique_ptr<struct wl_output *>&& output); public:
Bar(const Client&, std::unique_ptr<struct wl_output *>&&, uint32_t);
Bar(const Bar&) = delete; Bar(const Bar&) = delete;
Client& client;
auto toggle() -> void;
const Client& client;
Gtk::Window window; Gtk::Window window;
struct wl_surface *surface; struct wl_surface *surface;
struct zwlr_layer_surface_v1 *layerSurface; struct zwlr_layer_surface_v1 *layer_surface;
std::unique_ptr<struct wl_output *> output; std::unique_ptr<struct wl_output *> output;
std::string output_name;
uint32_t wl_name;
bool visible = true; bool visible = true;
std::string outputName;
auto setWidth(uint32_t) -> void;
auto toggle() -> void;
private: private:
static void _handleLogicalPosition(void *data, static void handleLogicalPosition(void *, struct zxdg_output_v1 *, int32_t,
struct zxdg_output_v1 *zxdg_output_v1, int32_t x, int32_t y); int32_t);
static void _handleLogicalSize(void *data, static void handleLogicalSize(void *, struct zxdg_output_v1 *, int32_t,
struct zxdg_output_v1 *zxdg_output_v1, int32_t width, int32_t height); int32_t);
static void _handleDone(void *data, struct zxdg_output_v1 *zxdg_output_v1); static void handleDone(void *, struct zxdg_output_v1 *);
static void _handleName(void *data, struct zxdg_output_v1 *xdg_output, static void handleName(void *, struct zxdg_output_v1 *, const char *);
const char *name); static void handleDescription(void *, struct zxdg_output_v1 *,
static void _handleDescription(void *data, const char *);
struct zxdg_output_v1 *zxdg_output_v1, const char *description); static void layerSurfaceHandleConfigure(void *,
static void _layerSurfaceHandleConfigure(void *data, struct zwlr_layer_surface_v1 *, uint32_t, uint32_t, uint32_t);
struct zwlr_layer_surface_v1 *surface, uint32_t serial, uint32_t width, static void layerSurfaceHandleClosed(void *,
uint32_t height); struct zwlr_layer_surface_v1 *);
static void _layerSurfaceHandleClosed(void *data,
struct zwlr_layer_surface_v1 *surface); auto setupConfig() -> void;
auto _setupConfig() -> void; auto setupWidgets() -> void;
auto _setupWidgets() -> void; auto setupCss() -> void;
auto _setupCss() -> void; void getModules(const Factory&, const std::string&);
uint32_t _width = 0;
uint32_t _height = 30; uint32_t width_ = 0;
Json::Value _config; uint32_t height_ = 30;
Glib::RefPtr<Gtk::StyleContext> _styleContext; Json::Value config_;
Glib::RefPtr<Gtk::CssProvider> _cssProvider; Glib::RefPtr<Gtk::StyleContext> style_context_;
struct zxdg_output_v1 *_xdgOutput; Glib::RefPtr<Gtk::CssProvider> css_provider_;
}; struct zxdg_output_v1 *xdg_output_;
std::vector<std::unique_ptr<waybar::IModule>> modules_left_;
std::vector<std::unique_ptr<waybar::IModule>> modules_center_;
std::vector<std::unique_ptr<waybar::IModule>> modules_right_;
};
} }

View File

@ -14,29 +14,30 @@
namespace waybar { namespace waybar {
struct Client { class Client {
std::string cssFile; public:
std::string configFile; Client(int argc, char *argv[]);
int main(int argc, char *argv[]);
Gtk::Main gtk_main;
Glib::RefPtr<Gtk::Application> gtk_app;
std::string css_file;
std::string config_file;
Glib::RefPtr<Gdk::Display> gdk_display; Glib::RefPtr<Gdk::Display> gdk_display;
struct wl_display *wlDisplay = nullptr; struct wl_display *wl_display = nullptr;
struct wl_registry *registry = nullptr; struct wl_registry *registry = nullptr;
struct zwlr_layer_shell_v1 *layer_shell = nullptr; struct zwlr_layer_shell_v1 *layer_shell = nullptr;
struct zxdg_output_manager_v1 *xdg_output_manager = nullptr; struct zxdg_output_manager_v1 *xdg_output_manager = nullptr;
struct wl_seat *seat = nullptr; struct wl_seat *seat = nullptr;
struct wl_output *wlOutput = nullptr;
std::vector<std::unique_ptr<Bar>> bars; std::vector<std::unique_ptr<Bar>> bars;
Client(int argc, char* argv[]); private:
void bind_interfaces(); void bindInterfaces();
auto setup_css(); auto setupCss();
int main(int argc, char* argv[]);
private: static void handleGlobal(void *data, struct wl_registry *registry,
static void _handle_global(void *data, struct wl_registry *registry, uint32_t name, const char *interface, uint32_t version);
uint32_t name, const char *interface, uint32_t version); static void handleGlobalRemove(void *data,
static void _handle_global_remove(void *data, struct wl_registry *registry, uint32_t name);
struct wl_registry *registry, uint32_t name); };
};
} }

View File

@ -2,23 +2,30 @@
#include <json/json.h> #include <json/json.h>
#include "modules/clock.hpp" #include "modules/clock.hpp"
#include "modules/workspaces.hpp" #ifdef HAVE_SWAY
#include "modules/sway/workspaces.hpp"
#include "modules/sway/window.hpp"
#endif
#include "modules/battery.hpp" #include "modules/battery.hpp"
#include "modules/memory.hpp" #include "modules/memory.hpp"
#include "modules/cpu.hpp" #include "modules/cpu.hpp"
#ifdef HAVE_LIBNL
#include "modules/network.hpp" #include "modules/network.hpp"
#endif
#ifdef HAVE_LIBPULSE
#include "modules/pulseaudio.hpp" #include "modules/pulseaudio.hpp"
#endif
#include "modules/custom.hpp" #include "modules/custom.hpp"
namespace waybar { namespace waybar {
class Factory { class Factory {
public: public:
Factory(Bar &bar, Json::Value config); Factory(Bar& bar, const Json::Value& config);
IModule *makeModule(std::string name); IModule* makeModule(const std::string &name) const;
private: private:
Bar &_bar; Bar& bar_;
Json::Value _config; const Json::Value& config_;
}; };
} }

View File

@ -1,32 +0,0 @@
#pragma once
#include <iostream>
#include "ipc.hpp"
/**
* IPC response including type of IPC response, size of payload and the json
* encoded payload string.
*/
struct ipc_response {
uint32_t size;
uint32_t type;
std::string payload;
};
/**
* Gets the path to the IPC socket from sway.
*/
std::string get_socketpath(void);
/**
* Opens the sway socket.
*/
int ipc_open_socket(std::string socket_path);
/**
* Issues a single IPC command and returns the buffer. len will be updated with
* the length of the buffer returned from sway.
*/
std::string ipc_single_command(int socketfd, uint32_t type, const char *payload, uint32_t *len);
/**
* Receives a single IPC response and returns an ipc_response.
*/
struct ipc_response ipc_recv_response(int socketfd);

View File

@ -1,6 +1,5 @@
#pragma once #pragma once
#include <json/json.h>
#include <filesystem> #include <filesystem>
#include <fstream> #include <fstream>
#include <iostream> #include <iostream>
@ -8,24 +7,25 @@
#include <sys/inotify.h> #include <sys/inotify.h>
#include <algorithm> #include <algorithm>
#include "util/chrono.hpp" #include "util/chrono.hpp"
#include "IModule.hpp" #include "ALabel.hpp"
namespace waybar::modules { namespace waybar::modules {
namespace fs = std::filesystem; namespace fs = std::filesystem;
class Battery : public IModule { class Battery : public ALabel {
public: public:
Battery(Json::Value config); Battery(const Json::Value&);
auto update() -> void; ~Battery();
operator Gtk::Widget&(); auto update() -> void;
private: private:
std::string _getIcon(uint16_t percentage); static inline const fs::path data_dir_ = "/sys/class/power_supply/";
static inline const fs::path _data_dir = "/sys/class/power_supply/";
std::vector<fs::path> _batteries; void worker();
util::SleeperThread _thread;
Gtk::Label _label; util::SleeperThread thread_;
Json::Value _config; std::vector<fs::path> batteries_;
}; int fd_;
};
} }

View File

@ -1,21 +1,18 @@
#pragma once #pragma once
#include <json/json.h>
#include <fmt/format.h> #include <fmt/format.h>
#include "fmt/time.h"
#include "util/chrono.hpp" #include "util/chrono.hpp"
#include "IModule.hpp" #include "ALabel.hpp"
namespace waybar::modules { namespace waybar::modules {
class Clock : public IModule { class Clock : public ALabel {
public: public:
Clock(Json::Value config); Clock(const Json::Value&);
auto update() -> void; auto update() -> void;
operator Gtk::Widget &(); private:
private: waybar::util::SleeperThread thread_;
Gtk::Label _label; };
waybar::util::SleeperThread _thread;
Json::Value _config;
};
} }

View File

@ -1,22 +1,18 @@
#pragma once #pragma once
#include <json/json.h>
#include <fmt/format.h> #include <fmt/format.h>
#include <sys/sysinfo.h> #include <sys/sysinfo.h>
#include "util/chrono.hpp" #include "util/chrono.hpp"
#include "IModule.hpp" #include "ALabel.hpp"
namespace waybar::modules { namespace waybar::modules {
class Cpu : public IModule { class Cpu : public ALabel {
public: public:
Cpu(Json::Value config); Cpu(const Json::Value&);
auto update() -> void; auto update() -> void;
operator Gtk::Widget &(); private:
private: waybar::util::SleeperThread thread_;
Gtk::Label _label; };
waybar::util::SleeperThread _thread;
Json::Value _config;
};
} }

View File

@ -1,22 +1,22 @@
#pragma once #pragma once
#include <json/json.h>
#include <fmt/format.h> #include <fmt/format.h>
#include <iostream>
#include "util/chrono.hpp" #include "util/chrono.hpp"
#include "IModule.hpp" #include "util/command.hpp"
#include "ALabel.hpp"
namespace waybar::modules { namespace waybar::modules {
class Custom : public IModule { class Custom : public ALabel {
public: public:
Custom(std::string name, Json::Value config); Custom(const std::string, const Json::Value&);
auto update() -> void; auto update() -> void;
operator Gtk::Widget &(); private:
private: void worker();
Gtk::Label _label;
waybar::util::SleeperThread _thread; const std::string name_;
const std::string _name; waybar::util::SleeperThread thread_;
Json::Value _config; };
};
} }

View File

@ -1,22 +1,18 @@
#pragma once #pragma once
#include <json/json.h>
#include <fmt/format.h> #include <fmt/format.h>
#include <sys/sysinfo.h> #include <sys/sysinfo.h>
#include "util/chrono.hpp" #include "util/chrono.hpp"
#include "IModule.hpp" #include "ALabel.hpp"
namespace waybar::modules { namespace waybar::modules {
class Memory : public IModule { class Memory : public ALabel {
public: public:
Memory(Json::Value config); Memory(const Json::Value&);
auto update() -> void; auto update() -> void;
operator Gtk::Widget &(); private:
private: waybar::util::SleeperThread thread_;
Gtk::Label _label; };
waybar::util::SleeperThread _thread;
Json::Value _config;
};
} }

View File

@ -5,31 +5,42 @@
#include <netlink/genl/genl.h> #include <netlink/genl/genl.h>
#include <netlink/genl/ctrl.h> #include <netlink/genl/ctrl.h>
#include <linux/nl80211.h> #include <linux/nl80211.h>
#include <json/json.h>
#include <fmt/format.h> #include <fmt/format.h>
#include "util/chrono.hpp" #include "util/chrono.hpp"
#include "IModule.hpp" #include "ALabel.hpp"
namespace waybar::modules { namespace waybar::modules {
class Network : public IModule { class Network : public ALabel {
public: public:
Network(Json::Value config); Network(const Json::Value&);
auto update() -> void; ~Network();
operator Gtk::Widget &(); auto update() -> void;
private: private:
void _parseEssid(struct nlattr **bss); static uint64_t netlinkRequest(int, void*, uint32_t, uint32_t groups = 0);
void _parseSignal(struct nlattr **bss); static uint64_t netlinkResponse(int, void*, uint32_t, uint32_t groups = 0);
bool _associatedOrJoined(struct nlattr **bss); static int scanCb(struct nl_msg*, void*);
static int _scanCb(struct nl_msg *msg, void *data);
auto _getInfo() -> void; void disconnected();
Gtk::Label _label; void initNL80211();
waybar::util::SleeperThread _thread; int getExternalInterface();
Json::Value _config; void parseEssid(struct nlattr**);
std::size_t _ifid; void parseSignal(struct nlattr**);
std::string _essid; bool associatedOrJoined(struct nlattr**);
int _signalStrengthdBm; auto getInfo() -> void;
int _signalStrength;
}; waybar::util::SleeperThread thread_;
int ifid_;
sa_family_t family_;
int sock_fd_;
struct sockaddr_nl nladdr_ = {0};
struct nl_sock* sk_ = nullptr;
int nl80211_id_;
std::string essid_;
std::string ifname_;
int signal_strength_dbm_;
uint16_t signal_strength_;
};
} }

View File

@ -1,36 +1,31 @@
#pragma once #pragma once
#include <pulse/pulseaudio.h> #include <pulse/pulseaudio.h>
#include <json/json.h>
#include <fmt/format.h> #include <fmt/format.h>
#include <algorithm> #include <algorithm>
#include "IModule.hpp" #include "ALabel.hpp"
namespace waybar::modules { namespace waybar::modules {
class Pulseaudio : public IModule { class Pulseaudio : public ALabel {
public: public:
Pulseaudio(Json::Value config); Pulseaudio(const Json::Value&);
auto update() -> void; ~Pulseaudio();
operator Gtk::Widget &(); auto update() -> void;
private: private:
std::string _getIcon(uint16_t percentage); static void subscribeCb(pa_context*, pa_subscription_event_type_t,
static void _subscribeCb(pa_context *context, uint32_t, void*);
pa_subscription_event_type_t type, uint32_t idx, void *data); static void contextStateCb(pa_context*, void*);
static void _contextStateCb(pa_context *c, void *data); static void sinkInfoCb(pa_context*, const pa_sink_info*, int, void*);
static void _sinkInfoCb(pa_context *context, const pa_sink_info *i, static void serverInfoCb(pa_context*, const pa_server_info*, void*);
int eol, void *data);
static void _serverInfoCb(pa_context *context, const pa_server_info *i, pa_threaded_mainloop* mainloop_;
void *data); pa_mainloop_api* mainloop_api_;
Gtk::Label _label; pa_context* context_;
Json::Value _config; uint32_t sink_idx_{0};
pa_threaded_mainloop *_mainloop; uint16_t volume_;
pa_mainloop_api *_mainloop_api; bool muted_;
pa_context *_context; std::string desc_;
uint32_t _sinkIdx{0}; };
int _volume;
bool _muted;
std::string _desc;
};
} }

View File

@ -0,0 +1,41 @@
#pragma once
#include <iostream>
#include <unistd.h>
#include <sys/socket.h>
#include <sys/un.h>
#include "ipc.hpp"
namespace waybar::modules::sway {
class Ipc {
public:
Ipc();
~Ipc();
struct ipc_response {
uint32_t size;
uint32_t type;
std::string payload;
};
void connect();
struct ipc_response sendCmd(uint32_t type,
const std::string& payload = "") const;
void subscribe(const std::string& payload) const;
struct ipc_response handleEvent() const;
protected:
static inline const std::string ipc_magic_ = "i3-ipc";
static inline const size_t ipc_header_size_ = ipc_magic_.size() + 8;
const std::string getSocketPath() const;
int open(const std::string&) const;
struct ipc_response send(int fd, uint32_t type,
const std::string& payload = "") const;
struct ipc_response recv(int fd) const;
int fd_;
int fd_event_;
};
}

View File

@ -1,6 +1,6 @@
#pragma once #pragma once
#define event_mask(ev) (1 << (ev & 0x7F)) #define event_mask(ev) (1u << (ev & 0x7F))
enum ipc_command_type { enum ipc_command_type {
// i3 command types - see i3's I3_REPLY_TYPE constants // i3 command types - see i3's I3_REPLY_TYPE constants

View File

@ -0,0 +1,29 @@
#pragma once
#include <fmt/format.h>
#include "bar.hpp"
#include "client.hpp"
#include "util/chrono.hpp"
#include "util/json.hpp"
#include "ALabel.hpp"
#include "modules/sway/ipc/client.hpp"
namespace waybar::modules::sway {
class Window : public ALabel {
public:
Window(waybar::Bar&, const Json::Value&);
auto update() -> void;
private:
void worker();
std::string getFocusedNode(Json::Value nodes);
void getFocusedWindow();
Bar& bar_;
waybar::util::SleeperThread thread_;
util::JsonParser parser_;
Ipc ipc_;
std::string window_;
};
}

View File

@ -0,0 +1,38 @@
#pragma once
#include <fmt/format.h>
#include "bar.hpp"
#include "client.hpp"
#include "util/chrono.hpp"
#include "util/json.hpp"
#include "IModule.hpp"
#include "modules/sway/ipc/client.hpp"
namespace waybar::modules::sway {
class Workspaces : public IModule {
public:
Workspaces(waybar::Bar&, const Json::Value&);
auto update() -> void;
operator Gtk::Widget &();
private:
void worker();
void addWorkspace(Json::Value);
std::string getIcon(std::string);
bool handleScroll(GdkEventScroll*);
int getPrevWorkspace();
int getNextWorkspace();
Bar& bar_;
const Json::Value& config_;
waybar::util::SleeperThread thread_;
Gtk::Box box_;
util::JsonParser parser_;
std::mutex mutex_;
bool scrolling_;
std::unordered_map<int, Gtk::Button> buttons_;
Json::Value workspaces_;
Ipc ipc_;
};
}

View File

@ -1,37 +0,0 @@
#pragma once
#include <fmt/format.h>
#include "bar.hpp"
#include "client.hpp"
#include "util/chrono.hpp"
#include "util/json.hpp"
#include "IModule.hpp"
namespace waybar::modules {
class Workspaces : public IModule {
public:
Workspaces(waybar::Bar &bar, Json::Value config);
auto update() -> void;
operator Gtk::Widget &();
private:
void _addWorkspace(Json::Value node);
std::string _getIcon(std::string name);
Json::Value _getWorkspaces(const std::string data);
bool _handleScroll(GdkEventScroll *e);
int _getPrevWorkspace();
int _getNextWorkspace();
Bar &_bar;
Json::Value _config;
waybar::util::SleeperThread _thread;
Gtk::Box _box;
util::JsonParser _parser;
std::mutex _mutex;
bool _scrolling;
std::unordered_map<int, Gtk::Button> _buttons;
Json::Value _workspaces;
int _ipcfd;
int _ipcEventfd;
};
}

View File

@ -5,88 +5,83 @@
#include <functional> #include <functional>
#include <condition_variable> #include <condition_variable>
#include <thread> #include <thread>
#include <gtkmm.h>
namespace waybar::chrono { namespace waybar::chrono {
using namespace std::chrono; using namespace std::chrono;
using clock = std::chrono::system_clock; using clock = std::chrono::system_clock;
using duration = clock::duration; using duration = clock::duration;
using time_point = std::chrono::time_point<clock, duration>; using time_point = std::chrono::time_point<clock, duration>;
inline struct timespec to_timespec(time_point t) noexcept
{
long secs = duration_cast<seconds>(t.time_since_epoch()).count();
long nsc = duration_cast<nanoseconds>(t.time_since_epoch() % seconds(1)).count();
return {secs, nsc};
}
inline time_point to_time_point(struct timespec t) noexcept
{
return time_point(duration_cast<duration>(seconds(t.tv_sec) + nanoseconds(t.tv_nsec)));
}
} }
namespace waybar::util { namespace waybar::util {
struct SleeperThread { struct SleeperThread {
SleeperThread() = default; SleeperThread() = default;
SleeperThread(std::function<void()> func) SleeperThread(std::function<void()> func)
: thread{[this, func] { : thread_{[this, func] {
do { while(true) {
func(); {
} while (do_run); std::lock_guard<std::mutex> lock(mutex_);
}} if (!do_run_) {
{ break;
defined = true; }
} }
SleeperThread& operator=(std::function<void()> func)
{
thread = std::thread([this, func] {
do {
func(); func();
} while (do_run); }
}); }}
defined = true; {}
return *this;
}
SleeperThread& operator=(std::function<void()> func)
auto sleep_for(chrono::duration dur) {
{ thread_ = std::thread([this, func] {
auto lock = std::unique_lock(mutex); while(true) {
return condvar.wait_for(lock, dur); {
} std::lock_guard<std::mutex> lock(mutex_);
if (!do_run_) {
auto sleep_until(chrono::time_point time) break;
{ }
auto lock = std::unique_lock(mutex); }
return condvar.wait_until(lock, time); func();
}
auto wake_up()
{
condvar.notify_all();
}
~SleeperThread()
{
do_run = false;
if (defined) {
condvar.notify_all();
thread.join();
} }
} });
return *this;
}
private:
std::thread thread; auto sleep_for(chrono::duration dur)
std::condition_variable condvar; {
std::mutex mutex; auto lock = std::unique_lock(mutex_);
bool defined = false; return condvar_.wait_for(lock, dur);
bool do_run = true; }
};
auto sleep_until(chrono::time_point time)
{
auto lock = std::unique_lock(mutex_);
return condvar_.wait_until(lock, time);
}
auto wake_up()
{
condvar_.notify_all();
}
~SleeperThread()
{
do_run_ = false;
condvar_.notify_all();
thread_.detach();
}
private:
std::thread thread_;
std::condition_variable condvar_;
std::mutex mutex_;
bool do_run_ = true;
};
} }

35
include/util/command.hpp Normal file
View File

@ -0,0 +1,35 @@
#pragma once
#include <sys/wait.h>
namespace waybar::util::command {
struct cmd_res {
int exit_code;
std::string out;
};
inline struct cmd_res exec(const std::string cmd)
{
FILE* fp(popen(cmd.c_str(), "r"));
if (!fp) {
return { -1, "" };
}
std::array<char, 128> buffer = {0};
std::string output;
while (feof(fp) == 0) {
if (fgets(buffer.data(), 128, fp) != nullptr) {
output += buffer.data();
}
}
// Remove last newline
if (!output.empty() && output[output.length()-1] == '\n') {
output.erase(output.length()-1);
}
int exit_code = WEXITSTATUS(pclose(fp));
return { exit_code, output };
}
}

View File

@ -4,31 +4,28 @@
namespace waybar::util { namespace waybar::util {
struct JsonParser { struct JsonParser {
JsonParser() JsonParser()
: _reader(_builder.newCharReader()) : reader_(builder_.newCharReader())
{} {}
Json::Value parse(const std::string data) const Json::Value parse(const std::string data) const
{ {
Json::Value root; Json::Value root;
std::string err; std::string err;
bool res = bool res =
_reader->parse(data.c_str(), data.c_str() + data.size(), &root, &err); reader_->parse(data.c_str(), data.c_str() + data.size(), &root, &err);
if (!res) if (!res)
throw std::runtime_error(err); throw std::runtime_error(err);
return root; return root;
} }
~JsonParser() ~JsonParser() = default;
{
delete _reader;
}
private: private:
Json::CharReaderBuilder _builder; Json::CharReaderBuilder builder_;
Json::CharReader *_reader; std::unique_ptr<Json::CharReader> const reader_;
}; };
} }

View File

@ -1,6 +1,6 @@
project( project(
'waybar', 'cpp', 'c', 'waybar', 'cpp', 'c',
version: '0.0.3', version: '0.0.5',
license: 'MIT', license: 'MIT',
default_options : ['cpp_std=c++17'], default_options : ['cpp_std=c++17'],
) )
@ -31,15 +31,47 @@ wlroots = dependency('wlroots', fallback: ['wlroots', 'wlroots'])
gtkmm = dependency('gtkmm-3.0') gtkmm = dependency('gtkmm-3.0')
jsoncpp = dependency('jsoncpp') jsoncpp = dependency('jsoncpp')
sigcpp = dependency('sigc++-2.0') sigcpp = dependency('sigc++-2.0')
libnl = dependency('libnl-3.0') libnl = dependency('libnl-3.0', required: false)
libnlgen = dependency('libnl-genl-3.0') libnlgen = dependency('libnl-genl-3.0', required: false)
libpulse = dependency('libpulse') libpulse = dependency('libpulse', required: false)
src_files = files(
'src/factory.cpp',
'src/ALabel.cpp',
'src/modules/memory.cpp',
'src/modules/battery.cpp',
'src/modules/clock.cpp',
'src/modules/custom.cpp',
'src/modules/cpu.cpp',
'src/main.cpp',
'src/bar.cpp',
'src/client.cpp'
)
if find_program('sway').found()
add_project_arguments('-DHAVE_SWAY', language: 'cpp')
src_files += [
'src/modules/sway/ipc/client.cpp',
'src/modules/sway/window.cpp',
'src/modules/sway/workspaces.cpp'
]
endif
if libnl.found() and libnlgen.found()
add_project_arguments('-DHAVE_LIBNL', language: 'cpp')
src_files += 'src/modules/network.cpp'
endif
if libpulse.found()
add_project_arguments('-DHAVE_LIBPULSE', language: 'cpp')
src_files += 'src/modules/pulseaudio.cpp'
endif
subdir('protocol') subdir('protocol')
executable( executable(
'waybar', 'waybar',
run_command('find', './src', '-name', '*.cpp').stdout().strip().split('\n'), src_files,
dependencies: [ dependencies: [
thread_dep, thread_dep,
wlroots, wlroots,
@ -64,3 +96,15 @@ install_data(
'./resources/style.css', './resources/style.css',
install_dir: '/etc/xdg/waybar', install_dir: '/etc/xdg/waybar',
) )
clangtidy = find_program('clang-tidy', required: false)
if clangtidy.found()
run_target(
'tidy',
command: [
clangtidy,
'-checks=*,-fuchsia-default-arguments',
'-p', meson.build_root()
] + src_files)
endif

View File

@ -1,20 +1,29 @@
{ {
// "layer": "top", // Waybar at top layer "layer": "top", // Waybar at top layer
// "position": "bottom", // Waybar at the bottom of your screen // "position": "bottom", // Waybar at the bottom of your screen
// "height": 30, // Waybar height // "height": 30, // Waybar height
// "width": 1280, // Waybar width // "width": 1280, // Waybar width
// Choose the order of the modules // Choose the order of the modules
"modules-left": ["workspaces", "custom/spotify"], "modules-left": ["sway/workspaces", "custom/spotify"],
"modules-center": ["sway/window"],
"modules-right": ["pulseaudio", "network", "cpu", "memory", "battery", "clock"], "modules-right": ["pulseaudio", "network", "cpu", "memory", "battery", "clock"],
// Modules configuration // Modules configuration
"workspaces": { "sway/workspaces": {
"format-icons": { // "disable-scroll": true,
"1": "", // "all-outputs": true,
"2": "", // "format-icons": {
"3": "", // "1": "",
"4": "", // "2": "",
"5": "" // "3": "",
} // "4": "",
// "5": ""
// }
},
"sway/window": {
"max-length": 50
},
"clock": {
"format-alt": "{:%Y-%m-%d}"
}, },
"cpu": { "cpu": {
"format": "{}% " "format": "{}% "
@ -27,8 +36,10 @@
"format-icons": ["", "", "", "", ""] "format-icons": ["", "", "", "", ""]
}, },
"network": { "network": {
"interface": "wlp2s0", // "interface": "wlp2s0", // (Optional) To force the use of this interface
"format": "{essid} ({signalStrength}%) " "format-wifi": "{essid} ({signalStrength}%) ",
"format-ethernet": "{ifname} ",
"format-disconnected": "Disconnected ⚠"
}, },
"pulseaudio": { "pulseaudio": {
"format": "{volume}% {icon}", "format": "{volume}% {icon}",
@ -37,6 +48,8 @@
}, },
"custom/spotify": { "custom/spotify": {
"format": " {}", "format": " {}",
"exec": "$HOME/.bin/mediaplayer.sh" "max-length": 40,
"exec": "$HOME/.config/waybar/mediaplayer.sh", // Script in resources folder
"exec-if": "pgrep spotify"
} }
} }

7
resources/mediaplayer.sh Executable file
View File

@ -0,0 +1,7 @@
#!/bin/sh
player_status=$(playerctl status 2> /dev/null)
if [ "$player_status" = "Playing" ]; then
echo "$(playerctl metadata artist) - $(playerctl metadata title)"
elif [ "$player_status" = "Paused" ]; then
echo "$(playerctl metadata artist) - $(playerctl metadata title)"
fi

View File

@ -12,17 +12,17 @@ window {
} }
#workspaces button { #workspaces button {
padding: 0 8px; padding: 0 5px;
background: transparent; background: transparent;
color: white; color: white;
border-bottom: 3px solid transparent; border-bottom: 3px solid transparent;
} }
#workspaces button label { #workspaces button.icon label {
font-size: 12px; font-size: 10px;
} }
#workspaces button.current { #workspaces button.focused {
background: #64727D; background: #64727D;
border-bottom: 3px solid white; border-bottom: 3px solid white;
} }
@ -76,6 +76,10 @@ window {
background: #2980b9; background: #2980b9;
} }
#network.disconnected {
background: #f53c3c;
}
#pulseaudio { #pulseaudio {
background: #f1c40f; background: #f1c40f;
color: black; color: black;

51
src/ALabel.cpp Normal file
View File

@ -0,0 +1,51 @@
#include "ALabel.hpp"
#include <iostream>
waybar::ALabel::ALabel(const Json::Value& config, const std::string format)
: config_(config),
format_(config_["format"] ? config_["format"].asString() : format),
default_format_(format_)
{
event_box_.add(label_);
if (config_["max-length"]) {
label_.set_max_width_chars(config_["max-length"].asUInt());
label_.set_ellipsize(Pango::EllipsizeMode::ELLIPSIZE_END);
}
if (config_["format-alt"]) {
event_box_.add_events(Gdk::BUTTON_PRESS_MASK);
event_box_.signal_button_press_event()
.connect(sigc::mem_fun(*this, &ALabel::handleToggle));
}
}
auto waybar::ALabel::update() -> void
{
// Nothing here
}
bool waybar::ALabel::handleToggle(GdkEventButton* const& ev)
{
alt = !alt;
if (alt) {
format_ = config_["format-alt"].asString();
} else {
format_ = default_format_;
}
dp.emit();
return true;
}
std::string waybar::ALabel::getIcon(uint16_t percentage)
{
if (!config_["format-icons"] || !config_["format-icons"].isArray()) {
return "";
}
auto size = config_["format-icons"].size();
auto idx = std::clamp(percentage / (100 / size), 0U, size - 1);
return config_["format-icons"][idx].asString();
}
waybar::ALabel::operator Gtk::Widget &() {
return event_box_;
}

View File

@ -3,186 +3,209 @@
#include "factory.hpp" #include "factory.hpp"
#include "util/json.hpp" #include "util/json.hpp"
waybar::Bar::Bar(Client &client, std::unique_ptr<struct wl_output *> &&p_output) waybar::Bar::Bar(const Client& client,
std::unique_ptr<struct wl_output *> &&p_output, uint32_t p_wl_name)
: client(client), window{Gtk::WindowType::WINDOW_TOPLEVEL}, : client(client), window{Gtk::WindowType::WINDOW_TOPLEVEL},
output(std::move(p_output)) surface(nullptr), layer_surface(nullptr),
output(std::move(p_output)), wl_name(p_wl_name)
{ {
static const struct zxdg_output_v1_listener xdgOutputListener = { static const struct zxdg_output_v1_listener xdgOutputListener = {
.logical_position = _handleLogicalPosition, .logical_position = handleLogicalPosition,
.logical_size = _handleLogicalSize, .logical_size = handleLogicalSize,
.done = _handleDone, .done = handleDone,
.name = _handleName, .name = handleName,
.description = _handleDescription, .description = handleDescription,
}; };
_xdgOutput = xdg_output_ =
zxdg_output_manager_v1_get_xdg_output(client.xdg_output_manager, *output); zxdg_output_manager_v1_get_xdg_output(client.xdg_output_manager, *output);
zxdg_output_v1_add_listener(_xdgOutput, &xdgOutputListener, this); zxdg_output_v1_add_listener(xdg_output_, &xdgOutputListener, this);
window.set_title("waybar"); window.set_title("waybar");
window.set_decorated(false); window.set_decorated(false);
_setupConfig(); setupConfig();
_setupCss(); setupCss();
_setupWidgets(); setupWidgets();
if (_config["height"])
_height = _config["height"].asUInt(); Gtk::Widget& wrap(window);
bool positionBottom = _config["position"] == "bottom"; gtk_widget_realize(wrap.gobj());
bool layerTop = _config["layer"] == "top"; GdkWindow *gdk_window = gtk_widget_get_window(wrap.gobj());
gtk_widget_realize(GTK_WIDGET(window.gobj())); gdk_wayland_window_set_use_custom_surface(gdk_window);
GdkWindow *gdkWindow = gtk_widget_get_window(GTK_WIDGET(window.gobj())); surface = gdk_wayland_window_get_wl_surface(gdk_window);
gdk_wayland_window_set_use_custom_surface(gdkWindow);
surface = gdk_wayland_window_get_wl_surface(gdkWindow); std::size_t layer_top = config_["layer"] == "top"
layerSurface = zwlr_layer_shell_v1_get_layer_surface( ? ZWLR_LAYER_SHELL_V1_LAYER_TOP : ZWLR_LAYER_SHELL_V1_LAYER_BOTTOM;
client.layer_shell, surface, *output, layer_surface = zwlr_layer_shell_v1_get_layer_surface(
(layerTop ? ZWLR_LAYER_SHELL_V1_LAYER_TOP : ZWLR_LAYER_SHELL_V1_LAYER_BOTTOM), client.layer_shell, surface, *output, layer_top, "waybar");
"waybar");
zwlr_layer_surface_v1_set_anchor(layerSurface, static const struct zwlr_layer_surface_v1_listener layer_surface_listener = {
ZWLR_LAYER_SURFACE_V1_ANCHOR_LEFT | ZWLR_LAYER_SURFACE_V1_ANCHOR_RIGHT | .configure = layerSurfaceHandleConfigure,
(positionBottom ? ZWLR_LAYER_SURFACE_V1_ANCHOR_BOTTOM : ZWLR_LAYER_SURFACE_V1_ANCHOR_TOP)); .closed = layerSurfaceHandleClosed,
zwlr_layer_surface_v1_set_size(layerSurface, _width, _height);
static const struct zwlr_layer_surface_v1_listener layerSurfaceListener = {
.configure = _layerSurfaceHandleConfigure,
.closed = _layerSurfaceHandleClosed,
}; };
zwlr_layer_surface_v1_add_listener(layerSurface, &layerSurfaceListener, zwlr_layer_surface_v1_add_listener(layer_surface,
this); &layer_surface_listener, this);
std::size_t anchor = ZWLR_LAYER_SURFACE_V1_ANCHOR_LEFT
| ZWLR_LAYER_SURFACE_V1_ANCHOR_RIGHT;
if (config_["position"] == "bottom") {
anchor |= ZWLR_LAYER_SURFACE_V1_ANCHOR_BOTTOM;
} else {
anchor |= ZWLR_LAYER_SURFACE_V1_ANCHOR_TOP;
}
auto height = config_["height"] ? config_["height"].asUInt() : height_;
auto width = config_["width"] ? config_["width"].asUInt() : width_;
zwlr_layer_surface_v1_set_anchor(layer_surface, anchor);
zwlr_layer_surface_v1_set_exclusive_zone(layer_surface, height);
zwlr_layer_surface_v1_set_size(layer_surface, width, height);
wl_surface_commit(surface); wl_surface_commit(surface);
} }
void waybar::Bar::_handleLogicalPosition(void *data, void waybar::Bar::handleLogicalPosition(void* /*data*/,
struct zxdg_output_v1 *zxdg_output_v1, int32_t x, int32_t y) struct zxdg_output_v1* /*zxdg_output_v1*/, int32_t /*x*/, int32_t /*y*/)
{ {
// Nothing here // Nothing here
} }
void waybar::Bar::_handleLogicalSize(void *data, void waybar::Bar::handleLogicalSize(void* /*data*/,
struct zxdg_output_v1 *zxdg_output_v1, int32_t width, int32_t height) struct zxdg_output_v1* /*zxdg_output_v1*/, int32_t /*width*/,
int32_t /*height*/)
{ {
// Nothing here // Nothing here
} }
void waybar::Bar::_handleDone(void *data, struct zxdg_output_v1 *zxdg_output_v1) void waybar::Bar::handleDone(void* /*data*/,
struct zxdg_output_v1* /*zxdg_output_v1*/)
{ {
// Nothing here // Nothing here
} }
void waybar::Bar::_handleName(void *data, struct zxdg_output_v1 *xdg_output, void waybar::Bar::handleName(void* data, struct zxdg_output_v1* /*xdg_output*/,
const char *name) const char* name)
{ {
auto o = reinterpret_cast<waybar::Bar *>(data); auto o = static_cast<waybar::Bar *>(data);
o->outputName = name; o->output_name = name;
} }
void waybar::Bar::_handleDescription(void *data, void waybar::Bar::handleDescription(void* /*data*/,
struct zxdg_output_v1 *zxdg_output_v1, const char *description) struct zxdg_output_v1* /*zxdg_output_v1*/, const char* /*description*/)
{ {
// Nothing here // Nothing here
} }
void waybar::Bar::_layerSurfaceHandleConfigure( void waybar::Bar::layerSurfaceHandleConfigure(void* data,
void *data, struct zwlr_layer_surface_v1 *surface, uint32_t serial, struct zwlr_layer_surface_v1* surface, uint32_t serial, uint32_t width,
uint32_t width, uint32_t height) uint32_t height)
{ {
auto o = reinterpret_cast<waybar::Bar *>(data); auto o = static_cast<waybar::Bar *>(data);
o->window.show_all(); o->window.show_all();
o->setWidth(o->_config["width"] ? o->_config["width"].asUInt() : width);
zwlr_layer_surface_v1_ack_configure(surface, serial); zwlr_layer_surface_v1_ack_configure(surface, serial);
if (o->_height != height) { if (width != o->width_ || height != o->height_) {
height = o->_height; o->width_ = width;
std::cout << fmt::format("New Height: {}", height) << std::endl; o->height_ = height;
zwlr_layer_surface_v1_set_size(surface, o->_width, height); std::cout << fmt::format(
zwlr_layer_surface_v1_set_exclusive_zone(surface, o->visible ? height : 0); "Bar configured (width: {}, height: {}) for output: {}",
o->width_, o->height_, o->output_name) << std::endl;
o->window.set_size_request(o->width_, o->height_);
o->window.resize(o->width_, o->height_);
zwlr_layer_surface_v1_set_exclusive_zone(surface, o->height_);
zwlr_layer_surface_v1_set_size(surface, o->width_, o->height_);
wl_surface_commit(o->surface); wl_surface_commit(o->surface);
} }
} }
void waybar::Bar::_layerSurfaceHandleClosed(void *data, void waybar::Bar::layerSurfaceHandleClosed(void* data,
struct zwlr_layer_surface_v1 *surface) struct zwlr_layer_surface_v1* /*surface*/)
{ {
auto o = reinterpret_cast<waybar::Bar *>(data); auto o = static_cast<waybar::Bar *>(data);
zwlr_layer_surface_v1_destroy(o->layerSurface); std::cout << "Bar removed from output: " + o->output_name << std::endl;
o->layerSurface = nullptr; zwlr_layer_surface_v1_destroy(o->layer_surface);
wl_surface_destroy(o->surface); wl_output_destroy(*o->output);
o->surface = nullptr; zxdg_output_v1_destroy(o->xdg_output_);
o->window.close(); o->modules_left_.clear();
} o->modules_center_.clear();
o->modules_right_.clear();
auto waybar::Bar::setWidth(uint32_t width) -> void
{
if (width == this->_width) return;
std::cout << fmt::format("Bar width configured: {}", width) << std::endl;
this->_width = width;
window.set_size_request(width);
window.resize(width, _height);
zwlr_layer_surface_v1_set_size(layerSurface, width, _height + 1);
wl_surface_commit(surface);
} }
auto waybar::Bar::toggle() -> void auto waybar::Bar::toggle() -> void
{ {
visible = !visible; visible = !visible;
auto zone = visible ? _height : 0; auto zone = visible ? height_ : 0;
zwlr_layer_surface_v1_set_exclusive_zone(layerSurface, zone); zwlr_layer_surface_v1_set_exclusive_zone(layer_surface, zone);
wl_surface_commit(surface); wl_surface_commit(surface);
} }
auto waybar::Bar::_setupConfig() -> void auto waybar::Bar::setupConfig() -> void
{ {
util::JsonParser parser; std::ifstream file(client.config_file);
std::ifstream file(client.configFile); if (!file.is_open()) {
if (!file.is_open())
throw std::runtime_error("Can't open config file"); throw std::runtime_error("Can't open config file");
}
std::string str((std::istreambuf_iterator<char>(file)), std::string str((std::istreambuf_iterator<char>(file)),
std::istreambuf_iterator<char>()); std::istreambuf_iterator<char>());
_config = parser.parse(str); util::JsonParser parser;
config_ = parser.parse(str);
} }
auto waybar::Bar::_setupCss() -> void auto waybar::Bar::setupCss() -> void
{ {
_cssProvider = Gtk::CssProvider::create(); css_provider_ = Gtk::CssProvider::create();
_styleContext = Gtk::StyleContext::create(); style_context_ = Gtk::StyleContext::create();
// load our css file, wherever that may be hiding // Load our css file, wherever that may be hiding
if (_cssProvider->load_from_path(client.cssFile)) { if (css_provider_->load_from_path(client.css_file)) {
Glib::RefPtr<Gdk::Screen> screen = window.get_screen(); Glib::RefPtr<Gdk::Screen> screen = window.get_screen();
_styleContext->add_provider_for_screen(screen, _cssProvider, style_context_->add_provider_for_screen(screen, css_provider_,
GTK_STYLE_PROVIDER_PRIORITY_USER); GTK_STYLE_PROVIDER_PRIORITY_USER);
} }
} }
auto waybar::Bar::_setupWidgets() -> void void waybar::Bar::getModules(const Factory& factory, const std::string& pos)
{
if (config_[pos]) {
for (const auto &name : config_[pos]) {
try {
auto module = factory.makeModule(name.asString());
if (pos == "modules-left") {
modules_left_.emplace_back(module);
}
if (pos == "modules-center") {
modules_center_.emplace_back(module);
}
if (pos == "modules-right") {
modules_right_.emplace_back(module);
}
module->dp.connect([module] { module->update(); });
} catch (const std::exception& e) {
std::cerr << e.what() << std::endl;
}
}
}
}
auto waybar::Bar::setupWidgets() -> void
{ {
auto &left = *Gtk::manage(new Gtk::Box(Gtk::ORIENTATION_HORIZONTAL, 0)); auto &left = *Gtk::manage(new Gtk::Box(Gtk::ORIENTATION_HORIZONTAL, 0));
auto &center = *Gtk::manage(new Gtk::Box(Gtk::ORIENTATION_HORIZONTAL, 0)); auto &center = *Gtk::manage(new Gtk::Box(Gtk::ORIENTATION_HORIZONTAL, 0));
auto &right = *Gtk::manage(new Gtk::Box(Gtk::ORIENTATION_HORIZONTAL, 0)); auto &right = *Gtk::manage(new Gtk::Box(Gtk::ORIENTATION_HORIZONTAL, 0));
auto &box1 = *Gtk::manage(new Gtk::Box(Gtk::ORIENTATION_HORIZONTAL, 0)); auto &box = *Gtk::manage(new Gtk::Box(Gtk::ORIENTATION_HORIZONTAL, 0));
window.add(box1); window.add(box);
box1.set_homogeneous(true); box.pack_start(left, true, true);
box1.pack_start(left, true, true); box.set_center_widget(center);
box1.pack_start(center, false, false); box.pack_end(right, true, true);
box1.pack_end(right, true, true);
Factory factory(*this, _config); Factory factory(*this, config_);
getModules(factory, "modules-left");
if (_config["modules-left"]) { getModules(factory, "modules-center");
for (auto name : _config["modules-left"]) { getModules(factory, "modules-right");
auto module = factory.makeModule(name.asString()); for (auto& module : modules_left_) {
if (module) left.pack_start(*module, false, true, 0);
left.pack_start(*module, false, true, 0);
}
} }
if (_config["modules-center"]) { for (auto& module : modules_center_) {
for (auto name : _config["modules-center"]) { center.pack_start(*module, true, true, 0);
auto module = factory.makeModule(name.asString());
if (module)
center.pack_start(*module, true, false, 10);
}
} }
if (_config["modules-right"]) { std::reverse(modules_right_.begin(), modules_right_.end());
std::reverse(_config["modules-right"].begin(), _config["modules-right"].end()); for (auto& module : modules_right_) {
for (auto name : _config["modules-right"]) { right.pack_end(*module, false, false, 0);
auto module = factory.makeModule(name.asString());
if (module)
right.pack_end(*module, false, false, 0);
}
} }
} }

View File

@ -1,35 +1,34 @@
#include "client.hpp" #include "client.hpp"
waybar::Client::Client(int argc, char* argv[]) waybar::Client::Client(int argc, char* argv[])
: gtk_main(argc, argv), : gtk_app(Gtk::Application::create(argc, argv, "org.alexays.waybar")),
gdk_display(Gdk::Display::get_default()), gdk_display(Gdk::Display::get_default()),
wlDisplay(gdk_wayland_display_get_wl_display(gdk_display->gobj())) wl_display(gdk_wayland_display_get_wl_display(gdk_display->gobj()))
{ {
auto getFirstValidPath = [] (std::vector<std::string> possiblePaths) { auto getFirstValidPath = [] (std::vector<std::string> possiblePaths) {
wordexp_t p; wordexp_t p;
for (std::string path: possiblePaths) { for (const std::string &path: possiblePaths) {
if (wordexp(path.c_str(), &p, 0) == 0) { if (wordexp(path.c_str(), &p, 0) == 0) {
if (access(p.we_wordv[0], F_OK) == 0) { if (access(*p.we_wordv, F_OK) == 0) {
std::string result = p.we_wordv[0]; std::string result = *p.we_wordv;
wordfree(&p); wordfree(&p);
return result; return result;
} else {
wordfree(&p);
} }
wordfree(&p);
} }
} }
return std::string(); return std::string();
}; };
configFile = getFirstValidPath({ config_file = getFirstValidPath({
"$XDG_CONFIG_HOME/waybar/config", "$XDG_CONFIG_HOME/waybar/config",
"$HOME/waybar/config", "$HOME/waybar/config",
"/etc/xdg/waybar/config", "/etc/xdg/waybar/config",
"./resources/config", "./resources/config",
}); });
cssFile = getFirstValidPath({ css_file = getFirstValidPath({
"$XDG_CONFIG_HOME/waybar/style.css", "$XDG_CONFIG_HOME/waybar/style.css",
"$HOME/waybar/style.css", "$HOME/waybar/style.css",
"/etc/xdg/waybar/style.css", "/etc/xdg/waybar/style.css",
@ -38,56 +37,64 @@ waybar::Client::Client(int argc, char* argv[])
} }
void waybar::Client::_handle_global(void *data, struct wl_registry *registry, void waybar::Client::handleGlobal(void *data, struct wl_registry *registry,
uint32_t name, const char *interface, uint32_t version) uint32_t name, const char *interface, uint32_t version)
{ {
auto o = reinterpret_cast<waybar::Client *>(data); auto o = static_cast<waybar::Client *>(data);
if (!strcmp(interface, zwlr_layer_shell_v1_interface.name)) { if (strcmp(interface, zwlr_layer_shell_v1_interface.name) == 0) {
o->layer_shell = (zwlr_layer_shell_v1 *)wl_registry_bind(registry, name, o->layer_shell = static_cast<struct zwlr_layer_shell_v1 *>(
&zwlr_layer_shell_v1_interface, version); wl_registry_bind(registry, name, &zwlr_layer_shell_v1_interface, version));
} else if (!strcmp(interface, wl_output_interface.name)) { } else if (strcmp(interface, wl_output_interface.name) == 0) {
o->wlOutput = (struct wl_output *)wl_registry_bind(registry, name,
&wl_output_interface, version);
auto output = std::make_unique<struct wl_output *>(); auto output = std::make_unique<struct wl_output *>();
*output = o->wlOutput; *output = static_cast<struct wl_output *>(wl_registry_bind(registry, name,
if (o->xdg_output_manager) &wl_output_interface, version));
o->bars.emplace_back(std::make_unique<Bar>(*o, std::move(output))); if (o->xdg_output_manager != nullptr) {
} else if (!strcmp(interface, wl_seat_interface.name)) { o->bars.emplace_back(std::make_unique<Bar>(*o, std::move(output), name));
o->seat = (struct wl_seat *)wl_registry_bind(registry, name,
&wl_seat_interface, version);
} else if (!strcmp(interface, zxdg_output_manager_v1_interface.name)
&& version >= ZXDG_OUTPUT_V1_NAME_SINCE_VERSION) {
o->xdg_output_manager =
(struct zxdg_output_manager_v1 *)wl_registry_bind(registry, name,
&zxdg_output_manager_v1_interface, ZXDG_OUTPUT_V1_NAME_SINCE_VERSION);
if (o->wlOutput) {
auto output = std::make_unique<struct wl_output *>();
*output = o->wlOutput;
o->bars.emplace_back(std::make_unique<Bar>(*o, std::move(output)));
}
} }
} else if (strcmp(interface, wl_seat_interface.name) == 0) {
o->seat = static_cast<struct wl_seat *>(wl_registry_bind(registry, name,
&wl_seat_interface, version));
} else if (strcmp(interface, zxdg_output_manager_v1_interface.name) == 0
&& version >= ZXDG_OUTPUT_V1_NAME_SINCE_VERSION) {
o->xdg_output_manager = static_cast<struct zxdg_output_manager_v1 *>(
wl_registry_bind(registry, name,
&zxdg_output_manager_v1_interface, ZXDG_OUTPUT_V1_NAME_SINCE_VERSION));
}
} }
void waybar::Client::_handle_global_remove(void *data, void waybar::Client::handleGlobalRemove(void* data,
struct wl_registry *registry, uint32_t name) struct wl_registry* /*registry*/, uint32_t name)
{ {
// TODO auto o = static_cast<waybar::Client *>(data);
for (auto it = o->bars.begin(); it != o->bars.end(); ++it) {
if ((**it).wl_name == name) {
o->bars.erase(it);
break;
}
}
} }
void waybar::Client::bind_interfaces() void waybar::Client::bindInterfaces()
{ {
registry = wl_display_get_registry(wlDisplay); registry = wl_display_get_registry(wl_display);
static const struct wl_registry_listener registry_listener = { static const struct wl_registry_listener registry_listener = {
.global = _handle_global, .global = handleGlobal,
.global_remove = _handle_global_remove, .global_remove = handleGlobalRemove,
}; };
wl_registry_add_listener(registry, &registry_listener, this); wl_registry_add_listener(registry, &registry_listener, this);
wl_display_roundtrip(wlDisplay); wl_display_roundtrip(wl_display);
} }
int waybar::Client::main(int argc, char* argv[]) int waybar::Client::main(int /*argc*/, char* /*argv*/[])
{ {
bind_interfaces(); bindInterfaces();
gtk_main.run(); gtk_app->hold();
gtk_app->run();
bars.clear();
zxdg_output_manager_v1_destroy(xdg_output_manager);
zwlr_layer_shell_v1_destroy(layer_shell);
wl_registry_destroy(registry);
wl_seat_destroy(seat);
wl_display_disconnect(wl_display);
return 0; return 0;
} }

View File

@ -1,35 +1,51 @@
#include "factory.hpp" #include "factory.hpp"
waybar::Factory::Factory(Bar &bar, Json::Value config) waybar::Factory::Factory(Bar& bar, const Json::Value& config)
: _bar(bar), _config(config) : bar_(bar), config_(config)
{} {}
waybar::IModule *waybar::Factory::makeModule(std::string name) waybar::IModule* waybar::Factory::makeModule(const std::string &name) const
{ {
try { try {
if (name == "battery") if (name == "battery") {
return new waybar::modules::Battery(_config[name]); return new waybar::modules::Battery(config_[name]);
if (name == "workspaces") }
return new waybar::modules::Workspaces(_bar, _config[name]); #ifdef HAVE_SWAY
if (name == "memory") if (name == "sway/workspaces") {
return new waybar::modules::Memory(_config[name]); return new waybar::modules::sway::Workspaces(bar_, config_[name]);
if (name == "cpu") }
return new waybar::modules::Cpu(_config[name]); #endif
if (name == "clock") if (name == "sway/window") {
return new waybar::modules::Clock(_config[name]); return new waybar::modules::sway::Window(bar_, config_[name]);
if (name == "network") }
return new waybar::modules::Network(_config[name]); if (name == "memory") {
if (name == "pulseaudio") return new waybar::modules::Memory(config_[name]);
return new waybar::modules::Pulseaudio(_config[name]); }
if (!name.compare(0, 7, "custom/") && name.size() > 7) if (name == "cpu") {
return new waybar::modules::Custom(name.substr(7), _config[name]); return new waybar::modules::Cpu(config_[name]);
std::cerr << "Unknown module: " + name << std::endl; }
if (name == "clock") {
return new waybar::modules::Clock(config_[name]);
}
#ifdef HAVE_LIBNL
if (name == "network") {
return new waybar::modules::Network(config_[name]);
}
#endif
#ifdef HAVE_LIBPULSE
if (name == "pulseaudio") {
return new waybar::modules::Pulseaudio(config_[name]);
}
#endif
if (name.compare(0, 7, "custom/") == 0 && name.size() > 7) {
return new waybar::modules::Custom(name.substr(7), config_[name]);
}
} catch (const std::exception& e) { } catch (const std::exception& e) {
auto err = fmt::format("Disabling module \"{}\", {}", name, e.what()); auto err = fmt::format("Disabling module \"{}\", {}", name, e.what());
std::cerr << err << std::endl; throw std::runtime_error(err);
} catch (...) { } catch (...) {
auto err = fmt::format("Disabling module \"{}\", Unknown reason", name); auto err = fmt::format("Disabling module \"{}\", Unknown reason", name);
std::cerr << err << std::endl; throw std::runtime_error(err);
} }
return nullptr; throw std::runtime_error("Unknown module: " + name);
} }

View File

@ -1,95 +0,0 @@
#define _POSIX_C_SOURCE 200809L
#include <string>
#include <stdio.h>
#include <sys/socket.h>
#include <sys/un.h>
#include "ipc/client.hpp"
static const char ipc_magic[] = {'i', '3', '-', 'i', 'p', 'c'};
static const size_t ipc_header_size = sizeof(ipc_magic)+8;
std::string get_socketpath(void) {
const char *env = getenv("SWAYSOCK");
if (env) return std::string(env);
std::string str;
{
std::string str_buf;
FILE* in;
char buf[512] = { 0 };
if (!(in = popen("sway --get-socketpath 2>/dev/null", "r"))) {
throw std::runtime_error("Failed to get socket path");
}
while (fgets(buf, sizeof(buf), in) != nullptr) {
str_buf.append(buf, sizeof(buf));
}
pclose(in);
str = str_buf;
}
if (str.back() == '\n') {
str.pop_back();
}
return str;
}
int ipc_open_socket(std::string socket_path) {
struct sockaddr_un addr;
int socketfd;
if ((socketfd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) {
throw std::runtime_error("Unable to open Unix socket");
}
addr.sun_family = AF_UNIX;
strncpy(addr.sun_path, socket_path.c_str(), sizeof(addr.sun_path) - 1);
addr.sun_path[sizeof(addr.sun_path) - 1] = 0;
int l = sizeof(struct sockaddr_un);
if (connect(socketfd, (struct sockaddr *)&addr, l) == -1) {
throw std::runtime_error("Unable to connect to " + socket_path);
}
return socketfd;
}
struct ipc_response ipc_recv_response(int socketfd) {
struct ipc_response response;
char data[ipc_header_size];
uint32_t *data32 = (uint32_t *)(data + sizeof(ipc_magic));
size_t total = 0;
while (total < ipc_header_size) {
ssize_t received = recv(socketfd, data + total, ipc_header_size - total, 0);
if (received <= 0) {
throw std::runtime_error("Unable to receive IPC response");
}
total += received;
}
total = 0;
response.size = data32[0];
response.type = data32[1];
char payload[response.size + 1];
while (total < response.size) {
ssize_t received = recv(socketfd, payload + total, response.size - total, 0);
if (received < 0) {
throw std::runtime_error("Unable to receive IPC response");
}
total += received;
}
payload[response.size] = '\0';
response.payload = std::string(payload);
return response;
}
std::string ipc_single_command(int socketfd, uint32_t type, const char *payload, uint32_t *len) {
char data[ipc_header_size];
uint32_t *data32 = (uint32_t *)(data + sizeof(ipc_magic));
memcpy(data, ipc_magic, sizeof(ipc_magic));
data32[0] = *len;
data32[1] = type;
if (send(socketfd, data, ipc_header_size, 0) == -1)
throw std::runtime_error("Unable to send IPC header");
if (send(socketfd, payload, *len, 0) == -1)
throw std::runtime_error("Unable to send IPC payload");
struct ipc_response resp = ipc_recv_response(socketfd);
*len = resp.size;
return resp.payload;
}

View File

@ -1,10 +1,12 @@
#include "client.hpp"
#include <csignal> #include <csignal>
#include <iostream> #include <iostream>
#include "client.hpp"
namespace waybar { namespace waybar {
static Client* client;
} static Client* client;
} // namespace waybar
int main(int argc, char* argv[]) int main(int argc, char* argv[])
{ {
@ -13,7 +15,7 @@ int main(int argc, char* argv[])
waybar::client = &c; waybar::client = &c;
std::signal(SIGUSR1, [] (int signal) { std::signal(SIGUSR1, [] (int signal) {
for (auto& bar : waybar::client->bars) { for (auto& bar : waybar::client->bars) {
bar.get()->toggle(); (*bar).toggle();
} }
}); });

View File

@ -1,35 +1,48 @@
#include "modules/battery.hpp" #include "modules/battery.hpp"
waybar::modules::Battery::Battery(Json::Value config) waybar::modules::Battery::Battery(const Json::Value& config)
: _config(config) : ALabel(config, "{capacity}%")
{ {
try { try {
for (auto &node : fs::directory_iterator(_data_dir)) { for (auto &node : fs::directory_iterator(data_dir_)) {
if (fs::is_directory(node) && fs::exists(node / "capacity") if (fs::is_directory(node) && fs::exists(node / "capacity")
&& fs::exists(node / "status") && fs::exists(node / "uevent")) && fs::exists(node / "status") && fs::exists(node / "uevent")) {
_batteries.push_back(node); batteries_.push_back(node);
}
} }
} catch (fs::filesystem_error &e) { } catch (fs::filesystem_error &e) {
throw std::runtime_error(e.what()); throw std::runtime_error(e.what());
} }
if (batteries_.empty()) {
if (!_batteries.size())
throw std::runtime_error("No batteries."); throw std::runtime_error("No batteries.");
}
auto fd = inotify_init(); fd_ = inotify_init1(IN_CLOEXEC);
if (fd == -1) if (fd_ == -1) {
throw std::runtime_error("Unable to listen batteries."); throw std::runtime_error("Unable to listen batteries.");
for (auto &bat : _batteries) }
inotify_add_watch(fd, (bat / "uevent").c_str(), IN_ACCESS); for (auto &bat : batteries_) {
// Trigger first value inotify_add_watch(fd_, (bat / "uevent").c_str(), IN_ACCESS);
}
label_.set_name("battery");
worker();
}
waybar::modules::Battery::~Battery()
{
close(fd_);
}
void waybar::modules::Battery::worker()
{
// Trigger first values
update(); update();
_label.set_name("battery"); thread_ = [this] {
_thread = [this, fd] { struct inotify_event event = {0};
struct inotify_event event; int nbytes = read(fd_, &event, sizeof(event));
int nbytes = read(fd, &event, sizeof(event)); if (nbytes != sizeof(event)) {
if (nbytes != sizeof(event))
return; return;
Glib::signal_idle().connect_once(sigc::mem_fun(*this, &Battery::update)); }
dp.emit();
}; };
} }
@ -37,45 +50,34 @@ auto waybar::modules::Battery::update() -> void
{ {
try { try {
uint16_t total = 0; uint16_t total = 0;
bool charging = false;
std::string status; std::string status;
for (auto &bat : _batteries) { for (auto &bat : batteries_) {
uint16_t capacity; uint16_t capacity;
std::string _status;
std::ifstream(bat / "capacity") >> capacity; std::ifstream(bat / "capacity") >> capacity;
std::ifstream(bat / "status") >> status; std::ifstream(bat / "status") >> _status;
if (status == "Charging") if (_status != "Unknown") {
charging = true; status = _status;
}
total += capacity; total += capacity;
} }
uint16_t capacity = total / _batteries.size(); uint16_t capacity = total / batteries_.size();
auto format = _config["format"] label_.set_text(fmt::format(format_, fmt::arg("capacity", capacity),
? _config["format"].asString() : "{capacity}%"; fmt::arg("icon", getIcon(capacity))));
_label.set_text(fmt::format(format, fmt::arg("capacity", capacity), label_.set_tooltip_text(status);
fmt::arg("icon", _getIcon(capacity)))); bool charging = status == "Charging";
_label.set_tooltip_text(status); if (charging) {
if (charging) label_.get_style_context()->add_class("charging");
_label.get_style_context()->add_class("charging"); } else {
else label_.get_style_context()->remove_class("charging");
_label.get_style_context()->remove_class("charging"); }
auto critical = _config["critical"] ? _config["critical"].asUInt() : 15; auto critical = config_["critical"] ? config_["critical"].asUInt() : 15;
if (capacity <= critical && !charging) if (capacity <= critical && !charging) {
_label.get_style_context()->add_class("warning"); label_.get_style_context()->add_class("warning");
else } else {
_label.get_style_context()->remove_class("warning"); label_.get_style_context()->remove_class("warning");
}
} catch (const std::exception& e) { } catch (const std::exception& e) {
std::cerr << e.what() << std::endl; std::cerr << e.what() << std::endl;
} }
} }
std::string waybar::modules::Battery::_getIcon(uint16_t percentage)
{
if (!_config["format-icons"] || !_config["format-icons"].isArray()) return "";
auto size = _config["format-icons"].size();
auto idx = std::clamp(percentage / (100 / size), 0U, size - 1);
return _config["format-icons"][idx].asString();
}
waybar::modules::Battery::operator Gtk::Widget &()
{
return _label;
}

View File

@ -1,27 +1,21 @@
#include "modules/clock.hpp" #include "modules/clock.hpp"
waybar::modules::Clock::Clock(Json::Value config) waybar::modules::Clock::Clock(const Json::Value& config)
: _config(config) : ALabel(config, "{:%H:%M}")
{ {
_label.set_name("clock"); label_.set_name("clock");
_thread = [this] { uint32_t interval = config_["interval"] ? config_["inveral"].asUInt() : 60;
Glib::signal_idle().connect_once(sigc::mem_fun(*this, &Clock::update)); thread_ = [this, interval] {
auto now = waybar::chrono::clock::now(); auto now = waybar::chrono::clock::now();
auto timeout = dp.emit();
std::chrono::floor<std::chrono::minutes>(now + std::chrono::minutes(1)); auto timeout = std::chrono::floor<std::chrono::seconds>(now
_thread.sleep_until(timeout); + std::chrono::seconds(interval));
thread_.sleep_until(timeout);
}; };
}; }
auto waybar::modules::Clock::update() -> void auto waybar::modules::Clock::update() -> void
{ {
auto t = std::time(nullptr); auto localtime = fmt::localtime(std::time(nullptr));
auto localtime = std::localtime(&t); label_.set_text(fmt::format(format_, localtime));
auto format =
_config["format"] ? _config["format"].asString() : "{:02}:{:02}";
_label.set_text(fmt::format(format, localtime->tm_hour, localtime->tm_min));
}
waybar::modules::Clock::operator Gtk::Widget &() {
return _label;
} }

View File

@ -1,27 +1,22 @@
#include "modules/cpu.hpp" #include "modules/cpu.hpp"
waybar::modules::Cpu::Cpu(Json::Value config) waybar::modules::Cpu::Cpu(const Json::Value& config)
: _config(config) : ALabel(config, "{}%")
{ {
_label.set_name("cpu"); label_.set_name("cpu");
int interval = _config["interval"] ? _config["inveral"].asInt() : 10; uint32_t interval = config_["interval"] ? config_["inveral"].asUInt() : 10;
_thread = [this, interval] { thread_ = [this, interval] {
Glib::signal_idle().connect_once(sigc::mem_fun(*this, &Cpu::update)); dp.emit();
_thread.sleep_for(chrono::seconds(interval)); thread_.sleep_for(chrono::seconds(interval));
}; };
}; }
auto waybar::modules::Cpu::update() -> void auto waybar::modules::Cpu::update() -> void
{ {
struct sysinfo info; struct sysinfo info = {0};
if (!sysinfo(&info)) { if (sysinfo(&info) == 0) {
float f_load = 1.f / (1 << SI_LOAD_SHIFT); float f_load = 1.f / (1u << SI_LOAD_SHIFT);
int load = info.loads[0] * f_load * 100 / get_nprocs(); uint16_t load = info.loads[0] * f_load * 100 / get_nprocs();
auto format = _config["format"] ? _config["format"].asString() : "{}%"; label_.set_text(fmt::format(format_, load));
_label.set_text(fmt::format(format, load));
} }
} }
waybar::modules::Cpu::operator Gtk::Widget &() {
return _label;
}

View File

@ -1,49 +1,46 @@
#include "modules/custom.hpp" #include "modules/custom.hpp"
#include <iostream>
waybar::modules::Custom::Custom(std::string name, Json::Value config) waybar::modules::Custom::Custom(const std::string name,
: _name(name), _config(config) const Json::Value& config)
: ALabel(config, "{}"), name_(name)
{ {
if (!_config["exec"]) if (!config_["exec"]) {
throw std::runtime_error(name + " has no exec path."); throw std::runtime_error(name_ + " has no exec path.");
int interval = _config["interval"] ? _config["inveral"].asInt() : 30; }
_thread = [this, interval] { label_.set_name("custom-" + name_);
Glib::signal_idle().connect_once(sigc::mem_fun(*this, &Custom::update)); worker();
_thread.sleep_for(chrono::seconds(interval)); }
void waybar::modules::Custom::worker()
{
uint32_t interval = config_["interval"] ? config_["inveral"].asUInt() : 30;
thread_ = [this, interval] {
bool can_update = true;
if (config_["exec-if"]) {
auto res = waybar::util::command::exec(config_["exec-if"].asString());
if (res.exit_code != 0) {
can_update = false;
label_.hide();
}
}
if (can_update) {
dp.emit();
}
thread_.sleep_for(chrono::seconds(interval));
}; };
}; }
auto waybar::modules::Custom::update() -> void auto waybar::modules::Custom::update() -> void
{ {
std::array<char, 128> buffer; auto res = waybar::util::command::exec(config_["exec"].asString());
std::string output;
std::shared_ptr<FILE> fp(popen(_config["exec"].asCString(), "r"), pclose);
if (!fp) {
std::cerr << _name + " can't exec " + _config["exec"].asString() << std::endl;
return;
}
while (!feof(fp.get())) {
if (fgets(buffer.data(), 128, fp.get()) != nullptr)
output += buffer.data();
}
// Remove last newline
if (!output.empty() && output[output.length()-1] == '\n') {
output.erase(output.length()-1);
}
// Hide label if output is empty // Hide label if output is empty
if (output.empty()) { if (res.out.empty() || res.exit_code != 0) {
_label.set_name(""); label_.hide();
_label.hide();
} else { } else {
_label.set_name("custom-" + _name); auto str = fmt::format(format_, res.out);
auto format = _config["format"] ? _config["format"].asString() : "{}"; label_.set_text(str);
_label.set_text(fmt::format(format, output)); label_.set_tooltip_text(str);
_label.show(); label_.show();
} }
} }
waybar::modules::Custom::operator Gtk::Widget &() {
return _label;
}

View File

@ -1,30 +1,25 @@
#include "modules/memory.hpp" #include "modules/memory.hpp"
waybar::modules::Memory::Memory(Json::Value config) waybar::modules::Memory::Memory(const Json::Value& config)
: _config(config) : ALabel(config, "{}%")
{ {
_label.set_name("memory"); label_.set_name("memory");
int interval = _config["interval"] ? _config["inveral"].asInt() : 30; uint32_t interval = config_["interval"] ? config_["inveral"].asUInt() : 30;
_thread = [this, interval] { thread_ = [this, interval] {
Glib::signal_idle().connect_once(sigc::mem_fun(*this, &Memory::update)); dp.emit();
_thread.sleep_for(chrono::seconds(interval)); thread_.sleep_for(chrono::seconds(interval));
}; };
}; }
auto waybar::modules::Memory::update() -> void auto waybar::modules::Memory::update() -> void
{ {
struct sysinfo info; struct sysinfo info = {0};
if (!sysinfo(&info)) { if (sysinfo(&info) == 0) {
auto total = info.totalram * info.mem_unit; auto total = info.totalram * info.mem_unit;
auto freeram = info.freeram * info.mem_unit; auto freeram = info.freeram * info.mem_unit;
int used_ram_percentage = 100 * (total - freeram) / total; int used_ram_percentage = 100 * (total - freeram) / total;
auto format = _config["format"] ? _config["format"].asString() : "{}%"; label_.set_text(fmt::format(format_, used_ram_percentage));
_label.set_text(fmt::format(format, used_ram_percentage));
auto used_ram_gigabytes = (total - freeram) / std::pow(1024, 3); auto used_ram_gigabytes = (total - freeram) / std::pow(1024, 3);
_label.set_tooltip_text(fmt::format("{:.{}f}Gb used", used_ram_gigabytes, 1)); label_.set_tooltip_text(fmt::format("{:.{}f}Gb used", used_ram_gigabytes, 1));
} }
} }
waybar::modules::Memory::operator Gtk::Widget &() {
return _label;
}

View File

@ -1,62 +1,324 @@
#include "modules/network.hpp" #include "modules/network.hpp"
waybar::modules::Network::Network(Json::Value config) waybar::modules::Network::Network(const Json::Value& config)
: _config(config), _ifid(if_nametoindex(config["interface"].asCString())) : ALabel(config, "{ifname}"), family_(AF_INET),
signal_strength_dbm_(0), signal_strength_(0)
{ {
if (_ifid == 0) sock_fd_ = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
throw std::runtime_error("Can't found network interface"); if (sock_fd_ < 0) {
_label.set_name("network"); throw std::runtime_error("Can't open network socket");
int interval = _config["interval"] ? _config["inveral"].asInt() : 30; }
_thread = [this, interval] { nladdr_.nl_family = AF_NETLINK;
Glib::signal_idle().connect_once(sigc::mem_fun(*this, &Network::update)); nladdr_.nl_groups = RTMGRP_LINK | RTMGRP_IPV4_IFADDR;
_thread.sleep_for(chrono::minutes(1)); if (bind(sock_fd_, reinterpret_cast<struct sockaddr *>(&nladdr_),
sizeof(nladdr_)) != 0) {
throw std::runtime_error("Can't bind network socket");
}
if (config_["interface"]) {
ifid_ = if_nametoindex(config_["interface"].asCString());
ifname_ = config_["interface"].asString();
if (ifid_ <= 0) {
throw std::runtime_error("Can't found network interface");
}
} else {
ifid_ = getExternalInterface();
if (ifid_ > 0) {
char ifname[IF_NAMESIZE];
if_indextoname(ifid_, ifname);
ifname_ = ifname;
}
}
initNL80211();
label_.set_name("network");
// Trigger first values
getInfo();
update();
thread_ = [this] {
char buf[4096];
uint64_t len = netlinkResponse(sock_fd_, buf, sizeof(buf),
RTMGRP_LINK | RTMGRP_IPV4_IFADDR);
bool need_update = false;
for (auto nh = reinterpret_cast<struct nlmsghdr *>(buf); NLMSG_OK(nh, len);
nh = NLMSG_NEXT(nh, len)) {
if (nh->nlmsg_type == NLMSG_DONE) {
break;
}
if (nh->nlmsg_type == NLMSG_ERROR) {
continue;
}
if (nh->nlmsg_type < RTM_NEWADDR) {
auto rtif = static_cast<struct ifinfomsg *>(NLMSG_DATA(nh));
if (rtif->ifi_index == static_cast<int>(ifid_)) {
need_update = true;
if (!(rtif->ifi_flags & IFF_RUNNING)) {
disconnected();
}
}
}
}
if (ifid_ <= 0 && !config_["interface"]) {
// Need to wait before get external interface
thread_.sleep_for(std::chrono::seconds(1));
ifid_ = getExternalInterface();
if (ifid_ > 0) {
char ifname[IF_NAMESIZE];
if_indextoname(ifid_, ifname);
ifname_ = ifname;
need_update = true;
}
}
if (need_update) {
getInfo();
dp.emit();
}
}; };
}; }
waybar::modules::Network::~Network()
{
close(sock_fd_);
nl_socket_free(sk_);
}
auto waybar::modules::Network::update() -> void auto waybar::modules::Network::update() -> void
{ {
_getInfo(); auto format = format_;
auto format = _config["format"] ? _config["format"].asString() : "{essid}"; if (ifid_ <= 0) {
_label.set_text(fmt::format(format, format = config_["format-disconnected"]
fmt::arg("essid", _essid), ? config_["format-disconnected"].asString() : format;
fmt::arg("signaldBm", _signalStrengthdBm), label_.get_style_context()->add_class("disconnected");
fmt::arg("signalStrength", _signalStrength) } else {
if (essid_.empty()) {
format = config_["format-ethernet"]
? config_["format-ethernet"].asString() : format;
} else {
format = config_["format-wifi"]
? config_["format-wifi"].asString() : format;
}
label_.get_style_context()->remove_class("disconnected");
}
label_.set_text(fmt::format(format,
fmt::arg("essid", essid_),
fmt::arg("signaldBm", signal_strength_dbm_),
fmt::arg("signalStrength", signal_strength_),
fmt::arg("ifname", ifname_)
)); ));
} }
int waybar::modules::Network::_scanCb(struct nl_msg *msg, void *data) { void waybar::modules::Network::disconnected()
auto net = static_cast<waybar::modules::Network *>(data); {
auto gnlh = static_cast<genlmsghdr *>(nlmsg_data(nlmsg_hdr(msg))); essid_.clear();
struct nlattr* tb[NL80211_ATTR_MAX + 1]; signal_strength_dbm_ = 0;
struct nlattr* bss[NL80211_BSS_MAX + 1]; signal_strength_ = 0;
struct nla_policy bss_policy[NL80211_BSS_MAX + 1]{}; ifname_.clear();
bss_policy[NL80211_BSS_TSF].type = NLA_U64; ifid_ = -1;
bss_policy[NL80211_BSS_FREQUENCY].type = NLA_U32;
bss_policy[NL80211_BSS_BSSID].type = NLA_UNSPEC;
bss_policy[NL80211_BSS_BEACON_INTERVAL].type = NLA_U16;
bss_policy[NL80211_BSS_CAPABILITY].type = NLA_U16;
bss_policy[NL80211_BSS_INFORMATION_ELEMENTS].type = NLA_UNSPEC;
bss_policy[NL80211_BSS_SIGNAL_MBM].type = NLA_U32;
bss_policy[NL80211_BSS_SIGNAL_UNSPEC].type = NLA_U8;
bss_policy[NL80211_BSS_STATUS].type = NLA_U32;
if (nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0), genlmsg_attrlen(gnlh, 0), nullptr) < 0)
return NL_SKIP;
if (!tb[NL80211_ATTR_BSS])
return NL_SKIP;
if (nla_parse_nested(bss, NL80211_BSS_MAX, tb[NL80211_ATTR_BSS], bss_policy))
return NL_SKIP;
if (!net->_associatedOrJoined(bss))
return NL_SKIP;
net->_parseEssid(bss);
net->_parseSignal(bss);
// TODO: parse quality
return NL_SKIP;
} }
void waybar::modules::Network::_parseEssid(struct nlattr **bss) void waybar::modules::Network::initNL80211()
{ {
_essid.clear(); sk_ = nl_socket_alloc();
if (genl_connect(sk_) != 0) {
nl_socket_free(sk_);
throw std::runtime_error("Can't connect to netlink socket");
}
if (nl_socket_modify_cb(sk_, NL_CB_VALID, NL_CB_CUSTOM, scanCb, this) < 0) {
nl_socket_free(sk_);
throw std::runtime_error("Can't connect to netlink socket");
}
nl80211_id_ = genl_ctrl_resolve(sk_, "nl80211");
if (nl80211_id_ < 0) {
nl_socket_free(sk_);
throw std::runtime_error("Can't resolve nl80211 interface");
}
}
// Based on https://gist.github.com/Yawning/c70d804d4b8ae78cc698
int waybar::modules::Network::getExternalInterface()
{
static const uint32_t route_buffer_size = 8192;
struct nlmsghdr *hdr = nullptr;
struct rtmsg *rt = nullptr;
char resp[route_buffer_size] = {0};
int ifidx = -1;
/* Prepare request. */
uint32_t reqlen = NLMSG_SPACE(sizeof(*rt));
char req[reqlen] = {0};
/* Build the RTM_GETROUTE request. */
hdr = reinterpret_cast<struct nlmsghdr *>(req);
hdr->nlmsg_len = NLMSG_LENGTH(sizeof(*rt));
hdr->nlmsg_flags = NLM_F_REQUEST | NLM_F_DUMP;
hdr->nlmsg_type = RTM_GETROUTE;
rt = static_cast<struct rtmsg *>(NLMSG_DATA(hdr));
rt->rtm_family = family_;
rt->rtm_table = RT_TABLE_MAIN;
/* Issue the query. */
if (netlinkRequest(sock_fd_, req, reqlen) < 0) {
goto out;
}
/* Read the response(s).
*
* WARNING: All the packets generated by the request must be consumed (as in,
* consume responses till NLMSG_DONE/NLMSG_ERROR is encountered).
*/
do {
uint64_t len = netlinkResponse(sock_fd_, resp, route_buffer_size);
if (len < 0) {
goto out;
}
/* Parse the response payload into netlink messages. */
for (hdr = reinterpret_cast<struct nlmsghdr *>(resp); NLMSG_OK(hdr, len);
hdr = NLMSG_NEXT(hdr, len)) {
if (hdr->nlmsg_type == NLMSG_DONE) {
goto out;
}
if (hdr->nlmsg_type == NLMSG_ERROR) {
/* Even if we found the interface index, something is broken with the
* netlink socket, so return an error.
*/
ifidx = -1;
goto out;
}
/* If we found the correct answer, skip parsing the attributes. */
if (ifidx != -1) {
continue;
}
/* Find the message(s) concerting the main routing table, each message
* corresponds to a single routing table entry.
*/
rt = static_cast<struct rtmsg *>(NLMSG_DATA(hdr));
if (rt->rtm_table != RT_TABLE_MAIN) {
continue;
}
/* Parse all the attributes for a single routing table entry. */
struct rtattr *attr = RTM_RTA(rt);
uint64_t attrlen = RTM_PAYLOAD(hdr);
bool has_gateway = false;
bool has_destination = false;
int temp_idx = -1;
for (; RTA_OK(attr, attrlen); attr = RTA_NEXT(attr, attrlen)) {
/* Determine if this routing table entry corresponds to the default
* route by seeing if it has a gateway, and if a destination addr is
* set, that it is all 0s.
*/
switch (attr->rta_type) {
case RTA_GATEWAY:
/* The gateway of the route.
*
* If someone every needs to figure out the gateway address as well,
* it's here as the attribute payload.
*/
has_gateway = true;
break;
case RTA_DST: {
/* The destination address.
* Should be either missing, or maybe all 0s. Accept both.
*/
const uint32_t nr_zeroes = (family_ == AF_INET) ? 4 : 16;
unsigned char c = 0;
size_t dstlen = RTA_PAYLOAD(attr);
if (dstlen != nr_zeroes) {
break;
}
for (uint32_t i = 0; i < dstlen; i += 1) {
c |= *(unsigned char *)(RTA_DATA(attr) + i);
}
has_destination = (c == 0);
break;
}
case RTA_OIF:
/* The output interface index. */
temp_idx = *static_cast<int*>(RTA_DATA(attr));
break;
default:
break;
}
}
/* If this is the default route, and we know the interface index,
* we can stop parsing this message.
*/
if (has_gateway && !has_destination && temp_idx != -1) {
ifidx = temp_idx;
break;
}
}
} while (true);
out:
return ifidx;
}
uint64_t waybar::modules::Network::netlinkRequest(int fd, void *req,
uint32_t reqlen, uint32_t groups)
{
struct sockaddr_nl sa = {0};
sa.nl_family = AF_NETLINK;
sa.nl_groups = groups;
struct iovec iov = { req, reqlen };
struct msghdr msg = { &sa, sizeof(sa), &iov, 1, nullptr, 0, 0 };
return sendmsg(fd, &msg, 0);
}
uint64_t waybar::modules::Network::netlinkResponse(int fd, void *resp,
uint32_t resplen, uint32_t groups)
{
uint64_t ret;
struct sockaddr_nl sa = {0};
sa.nl_family = AF_NETLINK;
sa.nl_groups = groups;
struct iovec iov = { resp, resplen };
struct msghdr msg = { &sa, sizeof(sa), &iov, 1, nullptr, 0, 0 };
ret = recvmsg(fd, &msg, 0);
if (msg.msg_flags & MSG_TRUNC) {
return -1;
}
return ret;
}
int waybar::modules::Network::scanCb(struct nl_msg *msg, void *data) {
auto net = static_cast<waybar::modules::Network *>(data);
auto gnlh = static_cast<genlmsghdr *>(nlmsg_data(nlmsg_hdr(msg)));
struct nlattr* tb[NL80211_ATTR_MAX + 1];
struct nlattr* bss[NL80211_BSS_MAX + 1];
struct nla_policy bss_policy[NL80211_BSS_MAX + 1]{};
bss_policy[NL80211_BSS_TSF].type = NLA_U64;
bss_policy[NL80211_BSS_FREQUENCY].type = NLA_U32;
bss_policy[NL80211_BSS_BSSID].type = NLA_UNSPEC;
bss_policy[NL80211_BSS_BEACON_INTERVAL].type = NLA_U16;
bss_policy[NL80211_BSS_CAPABILITY].type = NLA_U16;
bss_policy[NL80211_BSS_INFORMATION_ELEMENTS].type = NLA_UNSPEC;
bss_policy[NL80211_BSS_SIGNAL_MBM].type = NLA_U32;
bss_policy[NL80211_BSS_SIGNAL_UNSPEC].type = NLA_U8;
bss_policy[NL80211_BSS_STATUS].type = NLA_U32;
if (nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0), genlmsg_attrlen(gnlh, 0), nullptr) < 0) {
return NL_SKIP;
}
if (tb[NL80211_ATTR_BSS] == nullptr) {
return NL_SKIP;
}
if (nla_parse_nested(bss, NL80211_BSS_MAX, tb[NL80211_ATTR_BSS], bss_policy) != 0) {
return NL_SKIP;
}
if (!net->associatedOrJoined(bss)) {
return NL_SKIP;
}
net->parseEssid(bss);
net->parseSignal(bss);
// TODO(someone): parse quality
return NL_SKIP;
}
void waybar::modules::Network::parseEssid(struct nlattr **bss)
{
essid_.clear();
if (bss[NL80211_BSS_INFORMATION_ELEMENTS] != nullptr) { if (bss[NL80211_BSS_INFORMATION_ELEMENTS] != nullptr) {
auto ies = auto ies =
static_cast<char*>(nla_data(bss[NL80211_BSS_INFORMATION_ELEMENTS])); static_cast<char*>(nla_data(bss[NL80211_BSS_INFORMATION_ELEMENTS]));
@ -69,71 +331,53 @@ void waybar::modules::Network::_parseEssid(struct nlattr **bss)
if (ies_len > hdr_len && ies_len > ies[1] + hdr_len) { if (ies_len > hdr_len && ies_len > ies[1] + hdr_len) {
auto essid_begin = ies + hdr_len; auto essid_begin = ies + hdr_len;
auto essid_end = essid_begin + ies[1]; auto essid_end = essid_begin + ies[1];
std::copy(essid_begin, essid_end, std::back_inserter(_essid)); std::copy(essid_begin, essid_end, std::back_inserter(essid_));
} }
} }
} }
void waybar::modules::Network::_parseSignal(struct nlattr **bss) { void waybar::modules::Network::parseSignal(struct nlattr **bss) {
if (bss[NL80211_BSS_SIGNAL_MBM] != nullptr) { if (bss[NL80211_BSS_SIGNAL_MBM] != nullptr) {
// signalstrength in dBm // signalstrength in dBm
_signalStrengthdBm = signal_strength_dbm_ =
static_cast<int>(nla_get_u32(bss[NL80211_BSS_SIGNAL_MBM])) / 100; static_cast<int>(nla_get_u32(bss[NL80211_BSS_SIGNAL_MBM])) / 100;
// WiFi-hardware usually operates in the range -90 to -20dBm. // WiFi-hardware usually operates in the range -90 to -20dBm.
const int hardwareMax = -20; const int hardwareMax = -20;
const int hardwareMin = -90; const int hardwareMin = -90;
_signalStrength = ((double)(_signalStrengthdBm - hardwareMin) signal_strength_ = ((signal_strength_dbm_ - hardwareMin)
/ (double)(hardwareMax - hardwareMin)) * 100; / double{hardwareMax - hardwareMin}) * 100;
}
} }
}
bool waybar::modules::Network::_associatedOrJoined(struct nlattr** bss) bool waybar::modules::Network::associatedOrJoined(struct nlattr** bss)
{ {
if (!bss[NL80211_BSS_STATUS]) if (bss[NL80211_BSS_STATUS] == nullptr) {
return false;
}
auto status = nla_get_u32(bss[NL80211_BSS_STATUS]);
switch (status) {
case NL80211_BSS_STATUS_ASSOCIATED:
case NL80211_BSS_STATUS_IBSS_JOINED:
case NL80211_BSS_STATUS_AUTHENTICATED:
return true;
default:
return false; return false;
auto status = nla_get_u32(bss[NL80211_BSS_STATUS]);
switch (status) {
case NL80211_BSS_STATUS_ASSOCIATED:
case NL80211_BSS_STATUS_IBSS_JOINED:
case NL80211_BSS_STATUS_AUTHENTICATED:
return true;
default:
return false;
}
} }
}
auto waybar::modules::Network::_getInfo() -> void auto waybar::modules::Network::getInfo() -> void
{ {
struct nl_sock *sk = nl_socket_alloc(); struct nl_msg* nl_msg = nlmsg_alloc();
if (genl_connect(sk) != 0) { if (nl_msg == nullptr) {
nl_socket_free(sk); nl_socket_free(sk_);
return; return;
} }
if (nl_socket_modify_cb(sk, NL_CB_VALID, NL_CB_CUSTOM, _scanCb, this) < 0) { if (genlmsg_put(nl_msg, NL_AUTO_PORT, NL_AUTO_SEQ, nl80211_id_, 0, NLM_F_DUMP,
nl_socket_free(sk); NL80211_CMD_GET_SCAN, 0) == nullptr
|| nla_put_u32(nl_msg, NL80211_ATTR_IFINDEX, ifid_) < 0) {
nlmsg_free(nl_msg);
return; return;
} }
const int nl80211_id = genl_ctrl_resolve(sk, "nl80211"); nl_send_sync(sk_, nl_msg);
if (nl80211_id < 0) {
nl_socket_free(sk);
return;
}
struct nl_msg *msg = nlmsg_alloc();
if (!msg) {
nl_socket_free(sk);
return;
}
if (!genlmsg_put(msg, NL_AUTO_PORT, NL_AUTO_SEQ, nl80211_id, 0,
NLM_F_DUMP, NL80211_CMD_GET_SCAN, 0) ||
nla_put_u32(msg, NL80211_ATTR_IFINDEX, _ifid) < 0) {
nlmsg_free(msg);
return;
}
nl_send_sync(sk, msg);
nl_socket_free(sk);
}
waybar::modules::Network::operator Gtk::Widget &() {
return _label;
} }

View File

@ -1,39 +1,51 @@
#include "modules/pulseaudio.hpp" #include "modules/pulseaudio.hpp"
waybar::modules::Pulseaudio::Pulseaudio(Json::Value config) waybar::modules::Pulseaudio::Pulseaudio(const Json::Value& config)
: _config(config), _mainloop(nullptr), _mainloop_api(nullptr), : ALabel(config, "{volume}%"), mainloop_(nullptr), mainloop_api_(nullptr),
_context(nullptr), _sinkIdx(0), _volume(0), _muted(false) context_(nullptr), sink_idx_(0), volume_(0), muted_(false)
{ {
_label.set_name("pulseaudio"); label_.set_name("pulseaudio");
_mainloop = pa_threaded_mainloop_new(); mainloop_ = pa_threaded_mainloop_new();
if (!_mainloop) if (mainloop_ == nullptr) {
throw std::runtime_error("pa_mainloop_new() failed."); throw std::runtime_error("pa_mainloop_new() failed.");
pa_threaded_mainloop_lock(_mainloop); }
_mainloop_api = pa_threaded_mainloop_get_api(_mainloop); pa_threaded_mainloop_lock(mainloop_);
_context = pa_context_new(_mainloop_api, "waybar"); mainloop_api_ = pa_threaded_mainloop_get_api(mainloop_);
if (!_context) context_ = pa_context_new(mainloop_api_, "waybar");
if (context_ == nullptr) {
throw std::runtime_error("pa_context_new() failed."); throw std::runtime_error("pa_context_new() failed.");
if (pa_context_connect(_context, NULL, PA_CONTEXT_NOAUTOSPAWN, NULL) < 0) }
throw std::runtime_error(fmt::format("pa_context_connect() failed: {}", if (pa_context_connect(context_, nullptr, PA_CONTEXT_NOAUTOSPAWN,
pa_strerror(pa_context_errno(_context)))); nullptr) < 0) {
pa_context_set_state_callback(_context, _contextStateCb, this); auto err = fmt::format("pa_context_connect() failed: {}",
if (pa_threaded_mainloop_start(_mainloop) < 0) pa_strerror(pa_context_errno(context_)));
throw std::runtime_error(err);
}
pa_context_set_state_callback(context_, contextStateCb, this);
if (pa_threaded_mainloop_start(mainloop_) < 0) {
throw std::runtime_error("pa_mainloop_run() failed."); throw std::runtime_error("pa_mainloop_run() failed.");
pa_threaded_mainloop_unlock(_mainloop); }
}; pa_threaded_mainloop_unlock(mainloop_);
}
void waybar::modules::Pulseaudio::_contextStateCb(pa_context *c, void *data) waybar::modules::Pulseaudio::~Pulseaudio()
{
mainloop_api_->quit(mainloop_api_, 0);
pa_threaded_mainloop_stop(mainloop_);
pa_threaded_mainloop_free(mainloop_);
}
void waybar::modules::Pulseaudio::contextStateCb(pa_context *c, void *data)
{ {
auto pa = static_cast<waybar::modules::Pulseaudio *>(data); auto pa = static_cast<waybar::modules::Pulseaudio *>(data);
switch (pa_context_get_state(c)) { switch (pa_context_get_state(c)) {
case PA_CONTEXT_TERMINATED: case PA_CONTEXT_TERMINATED:
pa->_mainloop_api->quit(pa->_mainloop_api, 0); pa->mainloop_api_->quit(pa->mainloop_api_, 0);
break; break;
case PA_CONTEXT_READY: case PA_CONTEXT_READY:
pa_context_get_server_info(c, _serverInfoCb, data); pa_context_get_server_info(c, serverInfoCb, data);
pa_context_set_subscribe_callback(c, _subscribeCb, data); pa_context_set_subscribe_callback(c, subscribeCb, data);
pa_context_subscribe(c, PA_SUBSCRIPTION_MASK_SINK, nullptr, pa_context_subscribe(c, PA_SUBSCRIPTION_MASK_SINK, nullptr, nullptr);
nullptr);
break; break;
case PA_CONTEXT_CONNECTING: case PA_CONTEXT_CONNECTING:
case PA_CONTEXT_AUTHORIZING: case PA_CONTEXT_AUTHORIZING:
@ -41,7 +53,7 @@ void waybar::modules::Pulseaudio::_contextStateCb(pa_context *c, void *data)
break; break;
case PA_CONTEXT_FAILED: case PA_CONTEXT_FAILED:
default: default:
pa->_mainloop_api->quit(pa->_mainloop_api, 1); pa->mainloop_api_->quit(pa->mainloop_api_, 1);
break; break;
} }
} }
@ -49,38 +61,35 @@ void waybar::modules::Pulseaudio::_contextStateCb(pa_context *c, void *data)
/* /*
* Called when an event we subscribed to occurs. * Called when an event we subscribed to occurs.
*/ */
void waybar::modules::Pulseaudio::_subscribeCb(pa_context *context, void waybar::modules::Pulseaudio::subscribeCb(pa_context* context,
pa_subscription_event_type_t type, uint32_t idx, void *data) pa_subscription_event_type_t type, uint32_t idx, void* data)
{ {
unsigned facility = type & PA_SUBSCRIPTION_EVENT_FACILITY_MASK; unsigned facility = type & PA_SUBSCRIPTION_EVENT_FACILITY_MASK;
pa_operation *op = nullptr;
switch (facility) { switch (facility) {
case PA_SUBSCRIPTION_EVENT_SINK: case PA_SUBSCRIPTION_EVENT_SINK:
pa_context_get_sink_info_by_index(context, idx, _sinkInfoCb, data); pa_context_get_sink_info_by_index(context, idx, sinkInfoCb, data);
break; break;
default: default:
assert(0); break;
break;
} }
if (op)
pa_operation_unref(op);
} }
/* /*
* Called when the requested sink information is ready. * Called when the requested sink information is ready.
*/ */
void waybar::modules::Pulseaudio::_sinkInfoCb(pa_context *context, void waybar::modules::Pulseaudio::sinkInfoCb(pa_context* /*context*/,
const pa_sink_info *i, int eol, void *data) const pa_sink_info* i, int /*eol*/, void* data)
{ {
if (i) { if (i != nullptr) {
auto pa = static_cast<waybar::modules::Pulseaudio *>(data); auto pa = static_cast<waybar::modules::Pulseaudio *>(data);
float volume = (float)pa_cvolume_avg(&(i->volume)) / (float)PA_VOLUME_NORM; float volume = static_cast<float>(pa_cvolume_avg(&(i->volume)))
pa->_sinkIdx = i->index; / float{PA_VOLUME_NORM};
pa->_volume = volume * 100.0f; pa->sink_idx_ = i->index;
pa->_muted = i->mute; pa->volume_ = volume * 100.0f;
pa->_desc = i->description; pa->muted_ = i->mute != 0;
Glib::signal_idle().connect_once(sigc::mem_fun(*pa, &Pulseaudio::update)); pa->desc_ = i->description;
pa->dp.emit();
} }
} }
@ -88,36 +97,25 @@ void waybar::modules::Pulseaudio::_sinkInfoCb(pa_context *context,
* Called when the requested information on the server is ready. This is * Called when the requested information on the server is ready. This is
* used to find the default PulseAudio sink. * used to find the default PulseAudio sink.
*/ */
void waybar::modules::Pulseaudio::_serverInfoCb(pa_context *context, void waybar::modules::Pulseaudio::serverInfoCb(pa_context *context,
const pa_server_info *i, void *data) const pa_server_info *i, void *data)
{ {
pa_context_get_sink_info_by_name(context, i->default_sink_name, _sinkInfoCb, pa_context_get_sink_info_by_name(context, i->default_sink_name,
data); sinkInfoCb, data);
} }
auto waybar::modules::Pulseaudio::update() -> void auto waybar::modules::Pulseaudio::update() -> void
{ {
auto format = _config["format"] ? _config["format"].asString() : "{volume}%"; auto format = format_;
if (_muted) { if (muted_) {
format = format =
_config["format-muted"] ? _config["format-muted"].asString() : format; config_["format-muted"] ? config_["format-muted"].asString() : format;
_label.get_style_context()->add_class("muted"); label_.get_style_context()->add_class("muted");
} else } else {
_label.get_style_context()->remove_class("muted"); label_.get_style_context()->remove_class("muted");
_label.set_label(fmt::format(format, }
fmt::arg("volume", _volume), label_.set_label(fmt::format(format,
fmt::arg("icon", _getIcon(_volume)))); fmt::arg("volume", volume_),
_label.set_tooltip_text(_desc); fmt::arg("icon", getIcon(volume_))));
} label_.set_tooltip_text(desc_);
std::string waybar::modules::Pulseaudio::_getIcon(uint16_t percentage)
{
if (!_config["format-icons"] || !_config["format-icons"].isArray()) return "";
auto size = _config["format-icons"].size();
auto idx = std::clamp(percentage / (100 / size), 0U, size - 1);
return _config["format-icons"][idx].asString();
}
waybar::modules::Pulseaudio::operator Gtk::Widget &() {
return _label;
} }

View File

@ -0,0 +1,134 @@
#include "modules/sway/ipc/client.hpp"
waybar::modules::sway::Ipc::Ipc()
: fd_(-1), fd_event_(-1)
{}
waybar::modules::sway::Ipc::~Ipc()
{
close(fd_);
close(fd_event_);
}
const std::string waybar::modules::sway::Ipc::getSocketPath() const
{
const char *env = getenv("SWAYSOCK");
if (env != nullptr) {
return std::string(env);
}
std::string str;
{
std::string str_buf;
FILE* in;
char buf[512] = { 0 };
if ((in = popen("sway --get-socketpath 2>/dev/null", "r")) == nullptr) {
throw std::runtime_error("Failed to get socket path");
}
while (fgets(buf, sizeof(buf), in) != nullptr) {
str_buf.append(buf, sizeof(buf));
}
pclose(in);
str = str_buf;
}
if (str.back() == '\n') {
str.pop_back();
}
return str;
}
int waybar::modules::sway::Ipc::open(const std::string& socketPath) const
{
struct sockaddr_un addr = {0};
int fd = -1;
if ((fd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) {
throw std::runtime_error("Unable to open Unix socket");
}
addr.sun_family = AF_UNIX;
strncpy(addr.sun_path, socketPath.c_str(), sizeof(addr.sun_path) - 1);
addr.sun_path[sizeof(addr.sun_path) - 1] = 0;
int l = sizeof(struct sockaddr_un);
if (::connect(fd, reinterpret_cast<struct sockaddr *>(&addr), l) == -1) {
throw std::runtime_error("Unable to connect to " + socketPath);
}
return fd;
}
void waybar::modules::sway::Ipc::connect()
{
const std::string& socketPath = getSocketPath();
fd_ = open(socketPath);
fd_event_ = open(socketPath);
}
struct waybar::modules::sway::Ipc::ipc_response
waybar::modules::sway::Ipc::recv(int fd) const
{
std::string header;
header.reserve(ipc_header_size_);
auto data32 = reinterpret_cast<uint32_t *>(header.data() + ipc_magic_.size());
size_t total = 0;
while (total < ipc_header_size_) {
ssize_t res =
::recv(fd, header.data() + total, ipc_header_size_ - total, 0);
if (res <= 0) {
throw std::runtime_error("Unable to receive IPC response");
}
total += res;
}
total = 0;
std::string payload;
payload.reserve(data32[0] + 1);
while (total < data32[0]) {
ssize_t res =
::recv(fd, payload.data() + total, data32[0] - total, 0);
if (res < 0) {
throw std::runtime_error("Unable to receive IPC response");
}
total += res;
}
payload[data32[0]] = 0;
return { data32[0], data32[1], &payload.front() };
}
struct waybar::modules::sway::Ipc::ipc_response
waybar::modules::sway::Ipc::send(int fd, uint32_t type,
const std::string& payload) const
{
std::string header;
header.reserve(ipc_header_size_);
auto data32 = reinterpret_cast<uint32_t *>(header.data() + ipc_magic_.size());
memcpy(header.data(), ipc_magic_.c_str(), ipc_magic_.size());
data32[0] = payload.size();
data32[1] = type;
if (::send(fd, header.data(), ipc_header_size_, 0) == -1) {
throw std::runtime_error("Unable to send IPC header");
}
if (::send(fd, payload.c_str(), payload.size(), 0) == -1) {
throw std::runtime_error("Unable to send IPC payload");
}
return recv(fd);
}
struct waybar::modules::sway::Ipc::ipc_response
waybar::modules::sway::Ipc::sendCmd(uint32_t type,
const std::string& payload) const
{
return send(fd_, type, payload);
}
void waybar::modules::sway::Ipc::subscribe(const std::string& payload) const
{
auto res = send(fd_event_, IPC_SUBSCRIBE, payload);
if (res.payload != "{\"success\": true}") {
throw std::runtime_error("Unable to subscribe ipc event");
}
}
struct waybar::modules::sway::Ipc::ipc_response
waybar::modules::sway::Ipc::handleEvent() const
{
return recv(fd_event_);
}

View File

@ -0,0 +1,61 @@
#include "modules/sway/window.hpp"
waybar::modules::sway::Window::Window(Bar &bar, const Json::Value& config)
: ALabel(config, "{}"), bar_(bar)
{
label_.set_name("window");
ipc_.connect();
ipc_.subscribe("[ \"window\" ]");
getFocusedWindow();
// Launch worker
worker();
}
void waybar::modules::sway::Window::worker()
{
thread_ = [this] {
try {
auto res = ipc_.handleEvent();
auto parsed = parser_.parse(res.payload);
if ((parsed["change"] == "focus" || parsed["change"] == "title")
&& parsed["container"]["focused"].asBool()) {
window_ = parsed["container"]["name"].asString();
dp.emit();
}
} catch (const std::exception& e) {
std::cerr << e.what() << std::endl;
}
};
}
auto waybar::modules::sway::Window::update() -> void
{
label_.set_text(fmt::format(format_, window_));
label_.set_tooltip_text(window_);
}
std::string waybar::modules::sway::Window::getFocusedNode(Json::Value nodes)
{
for (auto &node : nodes) {
if (node["focused"].asBool() && node["type"] == "con") {
return node["name"].asString();
}
auto res = getFocusedNode(node["nodes"]);
if (!res.empty()) {
return res;
}
}
return std::string();
}
void waybar::modules::sway::Window::getFocusedWindow()
{
try {
auto res = ipc_.sendCmd(IPC_GET_TREE);
auto parsed = parser_.parse(res.payload);
window_ = getFocusedNode(parsed["nodes"]);
Glib::signal_idle().connect_once(sigc::mem_fun(*this, &Window::update));
} catch (const std::exception &e) {
std::cerr << e.what() << std::endl;
}
}

View File

@ -0,0 +1,215 @@
#include "modules/sway/workspaces.hpp"
waybar::modules::sway::Workspaces::Workspaces(Bar& bar,
const Json::Value& config)
: bar_(bar), config_(config), scrolling_(false)
{
box_.set_name("workspaces");
ipc_.connect();
ipc_.subscribe("[ \"workspace\" ]");
// Launch worker
worker();
}
void waybar::modules::sway::Workspaces::worker()
{
thread_ = [this] {
try {
// Wait for the name of the output
if (!config_["all-outputs"].asBool() && bar_.output_name.empty()) {
while (bar_.output_name.empty()) {
thread_.sleep_for(chrono::milliseconds(150));
}
} else if (!workspaces_.empty()) {
ipc_.handleEvent();
}
{
std::lock_guard<std::mutex> lock(mutex_);
auto res = ipc_.sendCmd(IPC_GET_WORKSPACES);
workspaces_ = parser_.parse(res.payload);
}
dp.emit();
} catch (const std::exception& e) {
std::cerr << 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["num"].asInt() == it->first; });
if (ws == workspaces_.end()) {
it = buttons_.erase(it);
needReorder = true;
} else {
++it;
}
}
for (auto node : workspaces_) {
if (!config_["all-outputs"].asBool()
&& bar_.output_name != node["output"].asString()) {
continue;
}
auto it = buttons_.find(node["num"].asInt());
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, node["num"].asInt());
}
button.show();
}
}
if (scrolling_) {
scrolling_ = false;
}
}
void waybar::modules::sway::Workspaces::addWorkspace(Json::Value node)
{
auto icon = getIcon(node["name"].asString());
auto pair = buttons_.emplace(node["num"].asInt(), icon);
auto &button = pair.first->second;
if (icon != node["name"].asString()) {
button.get_style_context()->add_class("icon");
}
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;
}
});
button.add_events(Gdk::SCROLL_MASK | Gdk::SMOOTH_SCROLL_MASK);
if (!config_["disable-scroll"].asBool()) {
button.signal_scroll_event()
.connect(sigc::mem_fun(*this, &Workspaces::handleScroll));
}
box_.reorder_child(button, node["num"].asInt());
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");
}
button.show();
}
std::string waybar::modules::sway::Workspaces::getIcon(std::string name)
{
if (config_["format-icons"][name]) {
return config_["format-icons"][name].asString();
}
if (config_["format-icons"]["default"]) {
return config_["format-icons"]["default"].asString();
}
return name;
}
bool waybar::modules::sway::Workspaces::handleScroll(GdkEventScroll *e)
{
// Avoid concurrent scroll event
if (scrolling_) {
return false;
}
scrolling_ = true;
int id = -1;
uint16_t idx = 0;
{
std::lock_guard<std::mutex> lock(mutex_);
for (; idx < workspaces_.size(); idx += 1) {
if (workspaces_[idx]["focused"].asBool()) {
id = workspaces_[idx]["num"].asInt();
break;
}
}
}
if (id == -1) {
scrolling_ = false;
return false;
}
if (e->direction == GDK_SCROLL_UP) {
id = getNextWorkspace();
}
if (e->direction == GDK_SCROLL_DOWN) {
id = getPrevWorkspace();
}
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) {
id = getNextWorkspace();
} else if (delta_y > 0) {
id = getPrevWorkspace();
}
}
{
std::lock_guard<std::mutex> lock(mutex_);
if (id == workspaces_[idx]["num"].asInt()) {
scrolling_ = false;
return false;
}
ipc_.sendCmd(IPC_COMMAND, fmt::format("workspace \"{}\"", id));
std::this_thread::sleep_for(std::chrono::milliseconds(150));
}
return true;
}
int waybar::modules::sway::Workspaces::getPrevWorkspace()
{
for (uint16_t i = 0; i != workspaces_.size(); i += 1) {
if (workspaces_[i]["focused"].asBool()) {
if (i > 0) {
return workspaces_[i - 1]["num"].asInt();
}
return workspaces_[workspaces_.size() - 1]["num"].asInt();
}
}
return -1;
}
int waybar::modules::sway::Workspaces::getNextWorkspace()
{
for (uint16_t i = 0; i != workspaces_.size(); i += 1) {
if (workspaces_[i]["focused"].asBool()) {
if (i + 1U < workspaces_.size()) {
return workspaces_[i + 1]["num"].asInt();
}
return workspaces_[0]["num"].asInt();
}
}
return -1;
}
waybar::modules::sway::Workspaces::operator Gtk::Widget &() {
return box_;
}

View File

@ -1,184 +0,0 @@
#include "modules/workspaces.hpp"
#include "ipc/client.hpp"
waybar::modules::Workspaces::Workspaces(Bar &bar, Json::Value config)
: _bar(bar), _config(config), _scrolling(false)
{
_box.set_name("workspaces");
std::string socketPath = get_socketpath();
_ipcfd = ipc_open_socket(socketPath);
_ipcEventfd = ipc_open_socket(socketPath);
const char *subscribe = "[ \"workspace\" ]";
uint32_t len = strlen(subscribe);
ipc_single_command(_ipcEventfd, IPC_SUBSCRIBE, subscribe, &len);
_thread = [this] {
try {
if (_bar.outputName.empty()) {
// Wait for the name of the output
while (_bar.outputName.empty())
_thread.sleep_for(chrono::milliseconds(150));
} else
ipc_recv_response(_ipcEventfd);
uint32_t len = 0;
std::lock_guard<std::mutex> lock(_mutex);
auto str = ipc_single_command(_ipcfd, IPC_GET_WORKSPACES, nullptr, &len);
_workspaces = _getWorkspaces(str);
Glib::signal_idle().connect_once(sigc::mem_fun(*this, &Workspaces::update));
} catch (const std::exception& e) {
std::cerr << e.what() << std::endl;
}
};
}
auto waybar::modules::Workspaces::update() -> void
{
std::lock_guard<std::mutex> lock(_mutex);
bool needReorder = false;
for (auto it = _buttons.begin(); it != _buttons.end();) {
auto ws = std::find_if(_workspaces.begin(), _workspaces.end(),
[it](auto node) -> bool { return node["num"].asInt() == it->first; });
if (ws == _workspaces.end()) {
it = _buttons.erase(it);
needReorder = true;
} else
++it;
}
for (auto node : _workspaces) {
if (_bar.outputName != node["output"].asString())
continue;
auto it = _buttons.find(node["num"].asInt());
if (it == _buttons.end()) {
_addWorkspace(node);
needReorder = true;
} else {
auto &button = it->second;
if (node["focused"].asBool())
button.get_style_context()->add_class("current");
else
button.get_style_context()->remove_class("current");
if (needReorder)
_box.reorder_child(button, node["num"].asInt());
button.show();
}
}
if (_scrolling)
_scrolling = false;
}
void waybar::modules::Workspaces::_addWorkspace(Json::Value node)
{
auto pair = _buttons.emplace(node["num"].asInt(),
_getIcon(node["name"].asString()));
auto &button = pair.first->second;
_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 value = fmt::format("workspace \"{}\"", pair.first->first);
uint32_t size = value.size();
ipc_single_command(_ipcfd, IPC_COMMAND, value.c_str(), &size);
} catch (const std::exception& e) {
std::cerr << e.what() << std::endl;
}
});
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, node["num"].asInt());
if (node["focused"].asBool())
button.get_style_context()->add_class("current");
button.show();
}
std::string waybar::modules::Workspaces::_getIcon(std::string name)
{
if (_config["format-icons"][name])
return _config["format-icons"][name].asString();
if (_config["format-icons"]["default"])
return _config["format-icons"]["default"].asString();
return name;
}
bool waybar::modules::Workspaces::_handleScroll(GdkEventScroll *e)
{
std::lock_guard<std::mutex> lock(_mutex);
// Avoid concurrent scroll event
if (_scrolling)
return false;
_scrolling = true;
int id = -1;
uint16_t idx = 0;
for (; idx < _workspaces.size(); idx += 1)
if (_workspaces[idx]["focused"].asBool()) {
id = _workspaces[idx]["num"].asInt();
break;
}
if (id == -1) {
_scrolling = false;
return false;
}
if (e->direction == GDK_SCROLL_UP)
id = _getNextWorkspace();
if (e->direction == GDK_SCROLL_DOWN)
id = _getPrevWorkspace();
if (e->direction == GDK_SCROLL_SMOOTH) {
gdouble delta_x, delta_y;
gdk_event_get_scroll_deltas ((const GdkEvent *) e, &delta_x, &delta_y);
if (delta_y < 0)
id = _getNextWorkspace();
else if (delta_y > 0)
id = _getPrevWorkspace();
}
if (id == _workspaces[idx]["num"].asInt()) {
_scrolling = false;
return false;
}
auto value = fmt::format("workspace \"{}\"", id);
uint32_t size = value.size();
ipc_single_command(_ipcfd, IPC_COMMAND, value.c_str(), &size);
std::this_thread::sleep_for(std::chrono::milliseconds(150));
return true;
}
int waybar::modules::Workspaces::_getPrevWorkspace()
{
int current = -1;
for (uint16_t i = 0; i != _workspaces.size(); i += 1)
if (_workspaces[i]["focused"].asBool()) {
current = _workspaces[i]["num"].asInt();
if (i > 0)
return _workspaces[i - 1]["num"].asInt();
return _workspaces[_workspaces.size() - 1]["num"].asInt();
}
return current;
}
int waybar::modules::Workspaces::_getNextWorkspace()
{
int current = -1;
for (uint16_t i = 0; i != _workspaces.size(); i += 1)
if (_workspaces[i]["focused"].asBool()) {
current = _workspaces[i]["num"].asInt();
if (i + 1U < _workspaces.size())
return _workspaces[i + 1]["num"].asInt();
return _workspaces[0]["num"].asInt();
}
return current;
}
Json::Value waybar::modules::Workspaces::_getWorkspaces(const std::string data)
{
Json::Value res;
try {
std::string err;
res = _parser.parse(data);
} catch (const std::exception& e) {
std::cerr << e.what() << std::endl;
}
return res;
}
waybar::modules::Workspaces::operator Gtk::Widget &() {
return _box;
}