HardwareSetupRemote.h

/*
    HardwareSetupRemote.h
    Copyright (c) 2025 Phil Schatzmann. All right reserved.
 
    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Lesser General Public
    License as published by the Free Software Foundation; either
    version 2.1 of the License, or (at your option) any later version.
 
    This library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
    Lesser General Public License for more details.
 
    You should have received a copy of the GNU Lesser General Public
    License along with this library; if not, write to the Free Software
    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*/
 
#pragma once
#include <exception>
#include "ArduinoLogger.h"
#include "GPIOWrapper.h"
#include "I2CWrapper.h"
#include "RemoteGPIO.h"
#include "RemoteI2C.h"
#include "RemoteSPI.h"
#include "SPIWrapper.h"
#include "WiFiUdpStream.h"
 
namespace arduino {
 
class HardwareSetupRemote : public I2CSource,
                                 public SPISource,
                                 public GPIOSource {
 public:
  HardwareSetupRemote() = default;
 
  HardwareSetupRemote(Stream& stream) { begin(&stream, false); }
 
  HardwareSetupRemote(int port) { this->port = port; }
 
  bool begin(Stream* s, bool asDefault = true, bool doHandShake = true) {
    p_stream = s;
 
    i2c.setStream(s);
    spi.setStream(s);
    gpio.setStream(s);
 
    // setup global objects
    if (asDefault) {
      Logger.warning("GPIO, I2C, SPI set up for Remote");
      SPI.setSPI(&spi);
      Wire.setI2C(&i2c);
      GPIO.setGPIO(&gpio);
    }
 
    if (doHandShake) {
      handShake(s);
    }
    return i2c && spi && gpio;
  }
 
  void begin(int port, bool asDefault) {
    this->port = port;
    begin(asDefault);
  }
 
  void begin(bool asDefault = true) {
    is_default_objects_active = asDefault;
    if (p_stream == nullptr) {
      default_stream.begin(port);
      handShake(&default_stream);
      IPAddress ip = default_stream.remoteIP();
      int remote_port = default_stream.remotePort();
      default_stream.setTarget(ip, remote_port);
      default_stream.write((const uint8_t*)"OK", 2);
      default_stream.flush();
      begin(&default_stream, asDefault, false);
    } else {
      begin(p_stream, asDefault, true);
    }
  }
 
  void end() {
    if (is_default_objects_active) {
      GPIO.setGPIO(nullptr);
      SPI.setSPI(nullptr);
      Wire.setI2C(nullptr);
    }
    if (p_stream == &default_stream) {
      default_stream.stop();
    }
  }
 
  HardwareGPIO* getGPIO() { return &gpio; }
  HardwareI2C* getI2C() { return &i2c; }
  HardwareSPI* getSPI() { return &spi; }
 
 protected:
  WiFiUDPStream default_stream;
  Stream* p_stream = nullptr;
  RemoteI2C i2c;
  RemoteSPI spi;
  RemoteGPIO gpio;
  int port;
  bool is_default_objects_active = false;
 
  void handShake(Stream* s) {
    while (true) {
      Logger.warning("HardwareSetup", "waiting for device...");
      try {
        // we wait for the Arduino to send us the Arduino-Emulator string
        if (s->available() >= 16) {
          char buffer[30];
          int len = s->readBytes(buffer, 18);
          buffer[len] = 0;
          if (strncmp(buffer, "Arduino-Emulator", 16)) {
            Logger.info("WiFiUDPStream", "device found!");
            break;
          } else {
            Logger.info("WiFiUDPStream", "unknown command", buffer);
          }
        }
        delay(10000);
      } catch (const std::exception& ex) {
        Logger.error("WiFiUDPStream", ex.what());
      }
    }
  }
};
 
#if !defined(SKIP_HARDWARE_SETUP)
static HardwareSetupRemote Remote{7000};
#endif
 
}  // namespace arduino