#ifndef VFD_H
#define VFD_H

#include <Streaming.h>
#include "./Addon.h"
#include "./enums.h"
#include "./config.h"

#ifdef HAS_STATES
#include <ArduinoJson.h>
#endif

class VFD : public Addon
{
public:
    enum DIRECTION
    {
        FORWARD = 1,
        STOP = 0,
        REVERSE = 2
    };

    VFD() : Addon(VFD_STR, VFD_CONTROL),
            direction(STOP){};

    void rev(short nop)
    {
        update(DIRECTION::REVERSE);
    }

    void fwd(short nop)
    {
        update(DIRECTION::FORWARD);
    }

    short setup()
    {
        pinMode(FWD_PIN, OUTPUT);
        pinMode(REV_PIN, OUTPUT);

        stop();
    }

    short stop(short nop = 0)
    {
        update(DIRECTION::STOP);
    }

    void speed(int aValue)
    {
    }

    virtual void debug(Stream *stream)
    {
        //  *stream << this->name << ":" << SPACE(direction);
    }

    virtual void info(Stream *stream)
    {
        // *stream << this->name << "\n\t" << SPACE(": FWD PIN " << FWD_PIN << " | REV PIN " << REV_PIN);
    }

    uchar direction;
    uchar lastDirection;
    millis_t dt;

#ifdef HAS_STATES
    String state()
    {
        const int capacity = JSON_OBJECT_SIZE(2);
        StaticJsonDocument<capacity> doc;
        doc["0"] = id;
        doc["1"] = direction;
        return doc.as<String>();
    }
#endif
private:
    void update(uchar newDirection)
    {
        if (direction != newDirection)
        {
            dt = now;
            lastDirection = direction;
            direction = newDirection;
            switch (direction)
            {
            case DIRECTION::FORWARD:
            {
                digitalWrite(REV_PIN, LOW);
                digitalWrite(FWD_PIN, HIGH);
                break;
            }
            case DIRECTION::REVERSE:
            {
                digitalWrite(FWD_PIN, LOW);
                digitalWrite(REV_PIN, HIGH);
                break;
            }
            case DIRECTION::STOP:
            {
                digitalWrite(FWD_PIN, LOW);
                digitalWrite(REV_PIN, LOW);
            }
            }
        }
    }
};

#endif