polybar/include/modules/fs.hpp

75 lines
1.7 KiB
C++

#pragma once
#include "components/config.hpp"
#include "config.hpp"
#include "modules/meta.hpp"
POLYBAR_NS
namespace modules {
/**
* Filesystem structure
*/
struct fs_mount {
string mountpoint;
bool mounted = false;
string type;
string fsname;
unsigned long long bytes_free = 0;
unsigned long long bytes_used = 0;
unsigned long long bytes_total = 0;
float percentage_free = 0;
float percentage_used = 0;
string percentage_free_s;
string percentage_used_s;
explicit fs_mount(const string& mountpoint, bool mounted = false)
: mountpoint(mountpoint), mounted(mounted) {}
};
using fs_mount_t = unique_ptr<fs_mount>;
/**
* Module used to display filesystem stats.
*/
class fs_module : public timer_module<fs_module> {
public:
using timer_module::timer_module;
void setup();
bool update();
string get_format() const;
string get_output();
bool build(builder* builder, string tag) const;
private:
static constexpr auto FORMAT_MOUNTED = "format-mounted";
static constexpr auto FORMAT_UNMOUNTED = "format-unmounted";
static constexpr auto TAG_LABEL_MOUNTED = "<label-mounted>";
static constexpr auto TAG_LABEL_UNMOUNTED = "<label-unmounted>";
static constexpr auto TAG_BAR_USED = "<bar-used>";
static constexpr auto TAG_BAR_FREE = "<bar-free>";
static constexpr auto TAG_RAMP_CAPACITY = "<ramp-capacity>";
label_t m_labelmounted;
label_t m_labelunmounted;
progressbar_t m_barused;
progressbar_t m_barfree;
ramp_t m_rampcapacity;
vector<string> m_mountpoints;
vector<fs_mount_t> m_mounts;
bool m_fixed = false;
int m_spacing = 2;
// used while formatting output
size_t m_index = 0;
};
}
POLYBAR_NS_END