// https://forum.arduino.cc/t/jetpack-lights-and-sounds/1327550/

#include <Adafruit_NeoPixel.h>

// Pin definitions
#define LED_PIN 6       // NeoPixel ring data pin
#define BUTTON_PIN 2    // Button pin
#define NUM_LEDS 61     // Number of LEDs in the ring
#define MP3_TX 10       // These are the exact names and values
#define MP3_RX 11       // from your working code

// Initialize NeoPixel ring
Adafruit_NeoPixel ring = Adafruit_NeoPixel(NUM_LEDS, LED_PIN, NEO_GRB + NEO_KHZ800);

// Timing and transition variables
unsigned long lastUpdate = 0;

// Update interval for the effect CHANGES HOW LONG THE FADE FROM YELLOW TO RED LASTS*********************************************
const int updateInterval = 21; 

bool buttonPressed = false;
bool hasStarted = false;
bool transitionToRed = true;
bool fadingOut = false;
bool jetFiring = false;
bool isPlaying = false;

// Color and brightness variables
int currentRed = 255;
int currentGreen = 0; // Initial yellow color
int currentBlue = 0;
int brightness = 0; // Start with low brightness

// Flame and swirl parameters
int flameRed = 255;
int flameOrange = 0;
int flameBlue = 0;
int swirlPosition = 0;
int swirlLength = 5;
int maxFlames = 6;
int sparkleCount = 10;

void setup() {
  // Initialize LED ring
  ring.begin();
  ring.show();

  // Initialize pins exactly as in your working code
  pinMode(BUTTON_PIN, INPUT_PULLUP);
  pinMode(MP3_TX, OUTPUT);
  pinMode(MP3_RX, OUTPUT);

  // Set initial states
  digitalWrite(MP3_TX, HIGH);
  digitalWrite(MP3_RX, HIGH);
  setAllPixelsWithBrightness(0, 0, 0, brightness);

  Serial.begin(9600);
  Serial.println(F("Controller initialized"));
}

// Your exact working audio control functions
void playClip() {
  digitalWrite(MP3_TX, LOW);
  digitalWrite(MP3_RX, HIGH);
  Serial.println(F("Playing clip"));
}

void stopAndReset() {
  digitalWrite(MP3_TX, HIGH);
  digitalWrite(MP3_RX, HIGH);
  delay(30);
  digitalWrite(MP3_RX, LOW);
  delay(30);
  digitalWrite(MP3_RX, HIGH);
  Serial.println(F("Stopped and reset clip"));
}

void loop() {
  bool currentButtonState = !digitalRead(BUTTON_PIN);

  // Handle button state changes
  if (currentButtonState != buttonPressed) {
    buttonPressed = currentButtonState;

    if (buttonPressed) {
      // Button was pressed - Play sound immediately
      playClip();

      // Initialize LED effect with delay
      if (!hasStarted) {
        delay(1250);  // DELAY THE STARTING OF THE LEDS delay here - LEDs will start after sound******************************************************************
        currentRed = 255;
        currentGreen = 120; // Initial yellow color
        currentBlue = 0;
        brightness = 0;
        fadingOut = false;
        hasStarted = true;
      }
      transitionToRed = true;
    } else {
      // Button was released
      stopAndReset();
      transitionToRed = false;
      jetFiring = false;
    }
  }

  // Update effects
  if (hasStarted && millis() - lastUpdate >= updateInterval) {
    lastUpdate = millis();

    if (buttonPressed && transitionToRed) {
      // Fade from yellow to red *********************************************************************************************************************************
      if (currentGreen > 0) {
        currentGreen -= 1;
        brightness += 1;
        if (brightness > 255) brightness = 255;
        setAllPixelsWithBrightness(currentRed, currentGreen, currentBlue, brightness);
      } else {
        jetFiring = true;
        transitionToRed = false;
      }
    } else if (buttonPressed && jetFiring) {
      enhancedFireJetEffect();
    } else if (!buttonPressed && !transitionToRed) {
      fadeToYellowAndOut();
    }
  }
}

void enhancedFireJetEffect() {
  randomizedSwirl();
  enhancedFlickerFlame();
  addSparkles();
}

void randomizedSwirl() {
  for (int i = 0; i < NUM_LEDS; i++) {
    ring.setPixelColor(i, ring.Color(0, 0, 0));
  }

  swirlLength = random(3, 8);
  swirlPosition = (swirlPosition + random(1, 5)) % NUM_LEDS;

  for (int i = 0; i < swirlLength; i++) {
    int ledIndex = (swirlPosition + i) % NUM_LEDS;
    int redIntensity = random(200, 255);
    ring.setPixelColor(ledIndex, ring.Color(redIntensity, 0, 0));
  }

  ring.show();
}

void enhancedFlickerFlame() {
  for (int i = 0; i < maxFlames; i++) {
    int currentLED = random(NUM_LEDS);
    int twinkleBrightness = random(150, 255);
    int twinkleRed = (flameRed * twinkleBrightness) / 255;
    int twinkleOrange = (flameOrange * twinkleBrightness) / 0;

    ring.setPixelColor(currentLED, ring.Color(twinkleRed, twinkleOrange, flameBlue));
  }
  ring.show();
}

void addSparkles() {
  for (int i = 0; i < sparkleCount; i++) {
    int sparkleIndex = random(NUM_LEDS);
    int sparkleRed = random(240, 80);
    int sparkleOrange = random(60);

    ring.setPixelColor(sparkleIndex, ring.Color(sparkleRed, sparkleOrange, 1));
  }
  ring.show();
}

void fadeToYellowAndOut() {   //CHANGING THE LENGTH OF THE FADE OUT*******************************************************************************
  if (brightness > 0) {
    brightness -= .25;
    if (currentGreen < 255) {
      currentGreen += 1;
    }
    setAllPixelsWithBrightness(currentRed, currentGreen, currentBlue, brightness);

    if (brightness == 0) {
      hasStarted = false;
      fadingOut = false;
    }
  }
}

void setAllPixelsWithBrightness(int red, int green, int blue, int bright) {
  red = (red * bright) / 255;
  green = (green * bright) / 255;
  blue = (blue * bright) / 255;

  for (int i = 0; i < NUM_LEDS; i++) {
    ring.setPixelColor(i, ring.Color(red, green, blue));
  }
  ring.show();
}