#pragma once

#include "../ratgdo.h"
#include "../ratgdo_child.h"
#include "../ratgdo_state.h"
#include "esphome/components/cover/cover.h"
#include "esphome/core/component.h"

namespace esphome {
namespace ratgdo {

    class RATGDOCover : public cover::Cover, public RATGDOClient, public Component {
    public:
        void dump_config() override;
        cover::CoverTraits get_traits() override;

        void on_motion_state(esphome::ratgdo::MotionState state) override;
        void on_obstruction_state(esphome::ratgdo::ObstructionState state) override;
        void on_door_state(esphome::ratgdo::DoorState state) override;
        void on_light_state(esphome::ratgdo::LightState state) override;
        void on_lock_state(esphome::ratgdo::LockState state) override;

    protected:
        void control(const cover::CoverCall& call) override;
    };

} // namespace ratgdo
} // namespace esphome