#include <Arduino.h>
#include "ELECHOUSE_CC1101_SRC_DRV.h"
#include "cc1101.h"
#include "config.h"

// ===== CONFIG =====
#define RAW_BUF_MAX 512

// ===== STATE =====
static bool cc1101Inited = false;

// ===== CAPTURE STATE =====
volatile unsigned long captureBuffer[RAW_BUF_MAX];
volatile int pulseIndex = 0;
volatile unsigned long lastEdgeTime = 0;
volatile bool capturing = false;

// ===== RF SETTINGS =====
float currentFreq = 433.92;
float dataRate = 3.79372;
float deviation = 0.0;
float rxBW = 325.0;
int powerLevel = 10;

// ===== ISR =====
void IRAM_ATTR pulseISR()
{
    unsigned long now = micros();

    if (!capturing) return;
    if (pulseIndex >= RAW_BUF_MAX) return;

    unsigned long duration = now - lastEdgeTime;

    if (duration < 50) return;

    captureBuffer[pulseIndex++] = duration;
    lastEdgeTime = now;
}

// ===== OOK SETUP =====
void setupOOKMode()
{
    ELECHOUSE_cc1101.SetRx();
    ELECHOUSE_cc1101.setMHZ(currentFreq);

    ELECHOUSE_cc1101.setModulation(0); // ASK/OOK
    ELECHOUSE_cc1101.setDRate(dataRate);
    ELECHOUSE_cc1101.setDeviation(0);
    ELECHOUSE_cc1101.setRxBW(rxBW);
    ELECHOUSE_cc1101.setSyncMode(0);
    ELECHOUSE_cc1101.setPA(powerLevel);
}

// ===== INIT (LAZY, SAFE) =====
bool initCC1101()
{
    if (cc1101Inited)
        return true;

    Serial.println("Initializing CC1101...");

    // IMPORTANT: no detachInterrupt here (causes crash if not installed)

    ELECHOUSE_cc1101.setSpiPin(
        cc1101_SCK,
        cc1101_MISO,
        cc1101_MOSI,
        CC1101_CS
    );

    ELECHOUSE_cc1101.setGDO(CC1101_GDO0, CC1101_GDO2);

    if (!ELECHOUSE_cc1101.getCC1101())
    {
        Serial.println("❌ CC1101 NOT FOUND");
        return false;
    }

    delay(10); // let SPI settle

    ELECHOUSE_cc1101.Init();   // THIS WAS YOUR FREEZE POINT
    setupOOKMode();

    pinMode(CC1101_GDO0, INPUT);

    Serial.println("✅ CC1101 READY");

    cc1101Inited = true;
    return true;
}

// ===== CAPTURE CONTROL =====
void startCapture()
{
    pulseIndex = 0;
    capturing = true;
    lastEdgeTime = micros();

    attachInterrupt(
        digitalPinToInterrupt(CC1101_GDO0),
        pulseISR,
        CHANGE
    );

    Serial.println("Capture started");
}

bool isCC1101Ready() {
    return cc1101Inited;
}

void stopCapture()
{
    capturing = false;

    detachInterrupt(digitalPinToInterrupt(CC1101_GDO0));

    Serial.println("Capture stopped");
}

// ===== DEBUG PRINT =====
void printCapture()
{
    Serial.println("Captured pulses:");

    for (int i = 0; i < pulseIndex; i++)
    {
        Serial.println(captureBuffer[i]);
    }
}


// ===== REPLAY =====
void replaySignal()
{
    Serial.println("Replaying...");

    ELECHOUSE_cc1101.SetTx();
    pinMode(CC1101_GDO0, OUTPUT);

    for (int i = 0; i < pulseIndex; i++)
    {
        digitalWrite(CC1101_GDO0, (i % 2 == 0) ? HIGH : LOW);
        delayMicroseconds(captureBuffer[i]);
    }

    digitalWrite(CC1101_GDO0, LOW);
    ELECHOUSE_cc1101.SetRx();
}