limit_number.h

#pragma once
#include "esphome/components/number/number.h"
#include "esphome/core/preferences.h"
#include "esphome/core/component.h"
#include "LD2450.h"

namespace esphome::ld2450
{

    /**
     * Enum which determines which property this number component affects.
     */
    enum LimitType
    {
        MAX_DISTANCE,
        MAX_TILT_ANGLE,
        MIN_TILT_ANGLE,
    };

    /**
     * User controlled number component which updates a limitation property on it's parent.
     */
    class LimitNumber : public number::Number, public Component, public Parented<LD2450>
    {
    public:
        void setup() override;

        /**
         * @brief Sets the initial value of this number.
         * @param value new value
         */
        void set_initial_state(float value)
        {
            initial_value_ = value;
        }

        /**
         * @brief Sets which property is affected by this number component
         * @param type new type
         */
        void set_type(LimitType type)
        {
            type_ = type;
        }

        /**
         * @brief Sets the restore flag. If set to true, this component will attempt to restore the value instead of using the initial value.
         * @param restore_value new value
         */
        void set_restore(bool restore_value)
        {
            restore_value_ = restore_value;
        }

    protected:
        /**
         * @brief Action performed when a new value is available. Forwards the value to the parent component. Updates preferences if required.
         * @param value new limitation value
         */
        void control(float value) override;

        /// @brief Determines which property is affected by this component
        LimitType type_ = MAX_DISTANCE;

        /// @brief Initial value of this number component
        float initial_value_ = 6.0f;

        /// @brief Value restore flag. If set to true, the initial value will be restored from memory.
        bool restore_value_ = true;

        /// @brief Preference management reference
        ESPPreferenceObject pref_;
    };

} // namespace esphome::ld2450