#pragma once

#define MAIN_DEVICE_ID "rc-gateway"

#include "ha.h"

using namespace Ha;

DeviceConfig* gatewayConfig = (new DeviceConfig{MAIN_DEVICE_ID, "RC Gateway"})->withManufacturer("Adafruit")->withModel("Huzzah Esp8266");

struct PollinSwitch : Switch {
    const char* group;
    unsigned char channel;

    PollinSwitch(const char* name, const char* group, const unsigned char channel)
        : Switch(name, Protocol_1::buildId(group, channel)), group(group), channel(channel) {
        asDevice((new DeviceConfig{id, name})->withManufacturer("Pollin")->withParent(gatewayConfig));
    }

    void onCommand(const char* msg) override {
        (String{ "ON" }.equals(msg)) ? mySwitch.switchOn(group, channel) : mySwitch.switchOff(group, channel);
        publisher(stateTopic, msg);
    }

    void addToMap() override {
        p1Switches.insert({ string(id), this });
    }

};

struct EasyHomeSwitch : Switch {
    unsigned long on[8] = { 4326554, 4537114, 4767530, 4972714 };
    unsigned long off[8] = { 4483146, 4626810, 4661562, 4819642 };

    EasyHomeSwitch(const char* name, const char* id, unsigned long on[4], unsigned long off[4])
        : Switch(name, id) {
        memcpy(&this->on[4], on, 4 * sizeof(unsigned long));
        memcpy(&this->off[4], off, 4 * sizeof(unsigned long));
        asDevice((new DeviceConfig{id, name})->withManufacturer("Intertek")->withModel("Easy Home")->withParent(gatewayConfig));
    }

    void onCommand(const char* msg) override {
        mySwitch.setProtocol(4);
        String{ "ON" }.equals(msg) ? mySwitch.send(on[4], 24) : mySwitch.send(off[4], 24);
        publisher(stateTopic, msg);
    }

    void addToMap() override {
        for (int i = 0; i < 8; i++) {
            onSwitches.insert({ on[i], this });
            offSwitches.insert({ off[i], this });
        }
    }
};