1
0
Fork 0
mirror of https://github.com/polybar/polybar.git synced 2024-11-11 13:50:56 -05:00
polybar/include/modules/fs.hpp
Jérôme BOULMIER 529843b6ba
Remove undefined behaviors and clean code (#1769)
* fix: avoid memory leaks in case of exceptions.

* fix(fs): remove undefined behavior when removing mounts in m_mounts

* cleanup: remove double checks

* fix: remove memory leaks

* cleanup(xresources): capture exception by reference

Co-authored-by: Patrick Ziegler <p.ziegler96@gmail.com>
2021-02-15 23:32:56 +01:00

79 lines
2 KiB
C++

#pragma once
#include <utility>
#include "components/config.hpp"
#include "modules/meta/timer_module.hpp"
#include "settings.hpp"
POLYBAR_NS
namespace modules {
/**
* Filesystem structure
*/
struct fs_mount {
string mountpoint;
bool mounted = false;
string type;
string fsname;
uint64_t bytes_free{0ULL};
uint64_t bytes_used{0ULL};
uint64_t bytes_avail{0ULL};
uint64_t bytes_total{0ULL};
int percentage_free{0};
int percentage_used{0};
explicit fs_mount(string mountpoint, bool mounted = false) : mountpoint(move(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:
explicit fs_module(const bar_settings&, string);
bool update();
string get_format() const;
string get_output();
bool build(builder* builder, const string& tag) const;
static constexpr auto TYPE = "internal/fs";
private:
static constexpr auto FORMAT_MOUNTED = "format-mounted";
static constexpr auto FORMAT_WARN = "format-warn";
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_LABEL_WARN = "<label-warn>";
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;
label_t m_labelwarn;
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};
bool m_remove_unmounted{false};
int m_spacing{2};
int m_perc_used_warn{90};
// used while formatting output
size_t m_index{0_z};
};
} // namespace modules
POLYBAR_NS_END