filter.h

Go to the documentation of this file.

#pragma once
 
#include "esphome/core/component.h"
#include "esphome/core/helpers.h"
 
namespace esphome {
namespace text_sensor {
 
class TextSensor;
 
class Filter {
 public:
  virtual optional<std::string> new_value(std::string value) = 0;
 
  virtual void initialize(TextSensor *parent, Filter *next);
 
  void input(const std::string &value);
 
  void output(const std::string &value);
 
 protected:
  friend TextSensor;
 
  Filter *next_{nullptr};
  TextSensor *parent_{nullptr};
};
 
using lambda_filter_t = std::function<optional<std::string>(std::string)>;
 
class LambdaFilter : public Filter {
 public:
  explicit LambdaFilter(lambda_filter_t lambda_filter);
 
  optional<std::string> new_value(std::string value) override;
 
  const lambda_filter_t &get_lambda_filter() const;
  void set_lambda_filter(const lambda_filter_t &lambda_filter);
 
 protected:
  lambda_filter_t lambda_filter_;
};
 
class StatelessLambdaFilter : public Filter {
 public:
  explicit StatelessLambdaFilter(optional<std::string> (*lambda_filter)(std::string)) : lambda_filter_(lambda_filter) {}
 
  optional<std::string> new_value(std::string value) override { return this->lambda_filter_(value); }
 
 protected:
  optional<std::string> (*lambda_filter_)(std::string);
};
 
class ToUpperFilter : public Filter {
 public:
  optional<std::string> new_value(std::string value) override;
};
 
class ToLowerFilter : public Filter {
 public:
  optional<std::string> new_value(std::string value) override;
};
 
class AppendFilter : public Filter {
 public:
  AppendFilter(std::string suffix) : suffix_(std::move(suffix)) {}
  optional<std::string> new_value(std::string value) override;
 
 protected:
  std::string suffix_;
};
 
class PrependFilter : public Filter {
 public:
  PrependFilter(std::string prefix) : prefix_(std::move(prefix)) {}
  optional<std::string> new_value(std::string value) override;
 
 protected:
  std::string prefix_;
};
 
struct Substitution {
  std::string from;
  std::string to;
};
 
class SubstituteFilter : public Filter {
 public:
  explicit SubstituteFilter(const std::initializer_list<Substitution> &substitutions);
  optional<std::string> new_value(std::string value) override;
 
 protected:
  FixedVector<Substitution> substitutions_;
};
 
class MapFilter : public Filter {
 public:
  explicit MapFilter(const std::initializer_list<Substitution> &mappings);
  optional<std::string> new_value(std::string value) override;
 
 protected:
  FixedVector<Substitution> mappings_;
};
 
}  // namespace text_sensor
}  // namespace esphome