polybar/include/modules/network.hpp

#ifndef _MODULES_NETWORK_HPP_
#define _MODULES_NETWORK_HPP_

#include <atomic>
#include <chrono>
#include <memory>
#include <string>

#include "modules/base.hpp"
#include "interfaces/net.hpp"
#include "services/logger.hpp"
#include "drawtypes/icon.hpp"
#include "drawtypes/label.hpp"
#include "drawtypes/animation.hpp"
#include "drawtypes/ramp.hpp"

namespace modules
{
  DefineModule(NetworkModule, TimerModule)
  {
    const char *FORMAT_CONNECTED = "format:connected";
    const char *FORMAT_PACKETLOSS = "format:packetloss";
    const char *FORMAT_DISCONNECTED = "format:disconnected";

    const char *TAG_RAMP_SIGNAL = "<ramp:signal>";
    const char *TAG_LABEL_CONNECTED = "<label:connected>";
    const char *TAG_LABEL_DISCONNECTED = "<label:disconnected>";
    const char *TAG_LABEL_PACKETLOSS = "<label:packetloss>";
    const char *TAG_ANIMATION_PACKETLOSS = "<animation:packetloss>";

    std::unique_ptr<net::WiredNetwork> wired_network;
    std::unique_ptr<net::WirelessNetwork> wireless_network;

    std::unique_ptr<drawtypes::Ramp> ramp_signal;
    std::unique_ptr<drawtypes::Animation> animation_packetloss;
    std::unique_ptr<drawtypes::Label> label_connected;
    std::unique_ptr<drawtypes::Label> label_connected_tokenized;
    std::unique_ptr<drawtypes::Label> label_disconnected;
    std::unique_ptr<drawtypes::Label> label_packetloss;
    std::unique_ptr<drawtypes::Label> label_packetloss_tokenized;

    std::string interface;

    concurrency::Atomic<bool> connected;
    concurrency::Atomic<bool> conseq_packetloss;
    concurrency::Atomic<int> signal_quality;

    int ping_nth_update;
    int counter = -1; // Set to -1 to ignore the first run

    void subthread_routine();

    public:
      NetworkModule(const std::string& name);

      void start();
      bool update();

      std::string get_format();

      bool build(Builder *builder, const std::string& tag);

  };
}

#endif