Replacing WR2E Module with TYWE2S in AliExpress 3-Way Switches for Home Assistant Integration

I got 2 of these 3-way switches from AliExpress. These switches came with a WR2E Module.
At first, I thought this module had a BK chip, so I spent some time trying to get a firmware dump but was not successful.

So, instead of wasting more time, I put ESP-02S aka "TYWE2S" instead of the Tuya module.



I needed this switch to work with my normal mechanical 3-way switch.



After some playing with the device, I figured that there is a sensor that tells if the load is ON or OFF. This sensor is linked with a blue LED which turns on when the load is on.

So, the way to configure this switch is to have a dummy relay linked with this sensor as it will tell me the state of the load when using both the normal and the WiFi switch.



GPIO5 is the button, GPIO14 is the relay, and GPIO13 is the blue LED (the sensor).
GPIO12 is a red LED, but I couldn't figure out how to make it work with the blue LED.
In the original firmware, when the load is on, the blue LED turns on. When the load is off, the red LED is on.

The rest of the configuration with Home Assistant can be found here (Tasmota rules and HA YAML configuration):
https://templates.blakadder.com/moes-SS01-1.html

I know most of the teardowns here are for openbeken, but unfortunately, this chip is not supported (I think). So, I did the teardowns anyway, even though I'm using Tasmota here.