polybar/include/modules/memory.hpp

#pragma once

#include <istream>

#include "config.hpp"
#include "drawtypes/label.hpp"
#include "drawtypes/progressbar.hpp"
#include "modules/meta.hpp"

LEMONBUDDY_NS

namespace modules {
  enum class memtype { NONE = 0, TOTAL, USED, FREE, SHARED, BUFFERS, CACHE, AVAILABLE };

  class memory_module : public timer_module<memory_module> {
   public:
    using timer_module::timer_module;

    void setup() {
      m_interval = chrono::duration<double>(m_conf.get<float>(name(), "interval", 1));

      m_formatter->add(DEFAULT_FORMAT, TAG_LABEL, {TAG_LABEL, TAG_BAR_USED, TAG_BAR_FREE});

      if (m_formatter->has(TAG_BAR_USED))
        m_bars[memtype::USED] = get_config_bar(m_bar, m_conf, name(), TAG_BAR_USED);
      if (m_formatter->has(TAG_BAR_FREE))
        m_bars[memtype::FREE] = get_config_bar(m_bar, m_conf, name(), TAG_BAR_FREE);

      if (m_formatter->has(TAG_LABEL)) {
        m_label = get_optional_config_label(m_conf, name(), TAG_LABEL, "%percentage_used%");
        m_tokenized = m_label->clone();
      }
    }

    bool update() {
      float kb_total;
      float kb_avail;
      // long kb_free;

      try {
        std::ifstream in(PATH_MEMORY_INFO);
        std::stringstream buffer;
        string str, rdbuf;
        int i = 0;

        in.exceptions(in.failbit);

        buffer.imbue(std::locale::classic());

        while (std::getline(in, str) && i++ < 3) {
          size_t off = str.find_first_of("1234567890", str.find(':'));
          buffer << std::strtol(&str[off], 0, 10) << std::endl;
        }

        buffer >> rdbuf;
        kb_total = std::atol(rdbuf.c_str());
        buffer >> rdbuf;  // kb_free = std::atol(rdbuf.c_str());
        buffer >> rdbuf;
        kb_avail = std::atol(rdbuf.c_str());
      } catch (const std::ios_base::failure& e) {
        kb_total = 0;
        // kb_free = 0;
        kb_avail = 0;
        m_log.err("Failed to read memory values (what: %s)", e.what());
      }

      if (kb_total > 0)
        m_perc[memtype::FREE] = (kb_avail / kb_total) * 100.0f + 0.5f;
      else
        m_perc[memtype::FREE] = 0;

      m_perc[memtype::USED] = 100 - m_perc[memtype::FREE];

      // replace tokens
      if (m_tokenized) {
        m_tokenized->m_text = m_label->m_text;

        auto replace_unit = [](label_t& label, string token, float value, string unit) {
          if (label->m_text.find(token) == string::npos)
            return;
          auto formatted = string_util::from_stream(
              stringstream() << std::setprecision(2) << std::fixed << value << " " << unit);
          label->replace_token(token, formatted);
        };

        replace_unit(m_tokenized, "%gb_used%", (kb_total - kb_avail) / 1024 / 1024, "GB");
        replace_unit(m_tokenized, "%gb_free%", kb_avail / 1024 / 1024, "GB");
        replace_unit(m_tokenized, "%gb_total%", kb_total / 1024 / 1024, "GB");
        replace_unit(m_tokenized, "%mb_used%", (kb_total - kb_avail) / 1024, "MB");
        replace_unit(m_tokenized, "%mb_free%", kb_avail / 1024, "MB");
        replace_unit(m_tokenized, "%mb_total%", kb_total / 1024, "MB");

        m_tokenized->replace_token("%percentage_used%", to_string(m_perc[memtype::USED]) + "%");
        m_tokenized->replace_token("%percentage_free%", to_string(m_perc[memtype::FREE]) + "%");
      }

      return true;
    }

    bool build(builder* builder, string tag) {
      if (tag == TAG_BAR_USED) {
        builder->node(m_bars[memtype::USED]->output(m_perc[memtype::USED]));
      } else if (tag == TAG_BAR_FREE)
        builder->node(m_bars[memtype::FREE]->output(m_perc[memtype::FREE]));
      else if (tag == TAG_LABEL)
        builder->node(m_tokenized);
      else
        return false;
      return true;
    }

   private:
    static constexpr auto TAG_LABEL = "<label>";
    static constexpr auto TAG_BAR_USED = "<bar-used>";
    static constexpr auto TAG_BAR_FREE = "<bar-free>";

    label_t m_label;
    label_t m_tokenized;
    map<memtype, progressbar_t> m_bars;
    map<memtype, int> m_perc;
  };
}

LEMONBUDDY_NS_END