/*
 * Instant Macropad - Complete Code
 * สำหรับ Arduino Pro Micro (ATmega32U4)
 * Global Byte Shop - Thailand
 */

#include <Keyboard.h>
#include <FastLED.h>

// Pin Definitions
const int switchPins[8] = {2, 3, 4, 5, 6, 7, 8, 9};
const int encoderCLK = 10;
const int encoderDT = 16;
const int encoderSW = 14;
const int ledPin = 15;

// LED Configuration
#define NUM_LEDS 8
CRGB leds[NUM_LEDS];

// Encoder Variables
volatile int encoderPos = 0;
int lastCLK = HIGH;
bool encoderPressed = false;

// Switch States
bool switchStates[8] = {false};
bool lastSwitchStates[8] = {false};
unsigned long lastDebounceTime[8] = {0};
const unsigned long debounceDelay = 50;

// Macro Definitions
struct MacroKey {
  String name;
  CRGB color;
  void (*function)();
};

// Function Prototypes
void macro1(); void macro2(); void macro3(); void macro4();
void macro5(); void macro6(); void macro7(); void macro8();

// Macro Configuration
MacroKey macros[8] = {
  {"REC", CRGB::Red, macro1},
  {"MUTE", CRGB::Blue, macro2},
  {"SCENE", CRGB::Green, macro3},
  {"CHAT", CRGB::Purple, macro4},
  {"SAVE", CRGB::Orange, macro5},
  {"BUILD", CRGB::Cyan, macro6},
  {"DEBUG", CRGB::Pink, macro7},
  {"RGB", CRGB::White, macro8}
};

void setup() {
  Serial.begin(115200);
  Serial.println("Instant Macropad Starting...");
  
  // Initialize Keyboard
  Keyboard.begin();
  
  // Initialize Switch Pins
  for (int i = 0; i < 8; i++) {
    pinMode(switchPins[i], INPUT_PULLUP);
  }
  
  // Initialize Encoder Pins
  pinMode(encoderCLK, INPUT_PULLUP);
  pinMode(encoderDT, INPUT_PULLUP);
  pinMode(encoderSW, INPUT_PULLUP);
  
  // Initialize LEDs
  FastLED.addLeds<WS2812B, ledPin, GRB>(leds, NUM_LEDS);
  FastLED.setBrightness(50);
  
  // Set initial LED colors
  updateLEDs();
  
  Serial.println("Macropad Ready!");
  
  // Startup Animation
  startupAnimation();
}

void loop() {
  // Read Switches
  readSwitches();
  
  // Read Encoder
  readEncoder();
  
  // Update LEDs based on activity
  updateLEDs();
  
  delay(1);
}

void readSwitches() {
  for (int i = 0; i < 8; i++) {
    int reading = !digitalRead(switchPins[i]); // Inverted because of pull-up
    
    // Debouncing
    if (reading != lastSwitchStates[i]) {
      lastDebounceTime[i] = millis();
    }
    
    if ((millis() - lastDebounceTime[i]) > debounceDelay) {
      if (reading != switchStates[i]) {
        switchStates[i] = reading;
        
        // Execute macro on key press
        if (switchStates[i]) {
          Serial.print("Key pressed: ");
          Serial.println(macros[i].name);
          
          // Visual feedback
          keyPressAnimation(i);
          
          // Execute macro function
          macros[i].function();
        }
      }
    }
    
    lastSwitchStates[i] = reading;
  }
}

void readEncoder() {
  int currentCLK = digitalRead(encoderCLK);
  
  if (currentCLK != lastCLK && currentCLK == LOW) {
    if (digitalRead(encoderDT) == LOW) {
      encoderPos--;
      volumeDown();
    } else {
      encoderPos++;
      volumeUp();
    }
    Serial.print("Encoder: ");
    Serial.println(encoderPos);
  }
  lastCLK = currentCLK;
  
  // Check encoder button
  bool currentEncoderSW = !digitalRead(encoderSW);
  if (currentEncoderSW && !encoderPressed) {
    encoderPressed = true;
    volumeMute();
    Serial.println("Encoder button pressed");
  } else if (!currentEncoderSW && encoderPressed) {
    encoderPressed = false;
  }
}

void updateLEDs() {
  for (int i = 0; i < 8; i++) {
    if (switchStates[i]) {
      // Brighter when pressed
      leds[i] = macros[i].color;
      leds[i].fadeToBlackBy(100);
    } else {
      // Normal brightness
      leds[i] = macros[i].color;
      leds[i].fadeToBlackBy(200);
    }
  }
  FastLED.show();
}

void keyPressAnimation(int keyIndex) {
  // Flash the pressed key
  leds[keyIndex] = CRGB::White;
  FastLED.show();
  delay(50);
  leds[keyIndex] = macros[keyIndex].color;
  FastLED.show();
}

void startupAnimation() {
  // Rainbow wave animation
  for (int j = 0; j < 3; j++) {
    for (int i = 0; i < NUM_LEDS; i++) {
      leds[i] = CHSV(i * 32, 255, 255);
      FastLED.show();
      delay(100);
    }
    FastLED.clear();
    FastLED.show();
    delay(200);
  }
}

// Macro Functions
void macro1() { // REC - OBS Recording
  Keyboard.press(KEY_LEFT_CTRL);
  Keyboard.press(KEY_LEFT_SHIFT);
  Keyboard.press('r');
  delay(100);
  Keyboard.releaseAll();
}

void macro2() { // MUTE - Microphone Mute
  Keyboard.press(KEY_LEFT_CTRL);
  Keyboard.press('m');
  delay(100);
  Keyboard.releaseAll();
}

void macro3() { // SCENE - Switch OBS Scene
  Keyboard.press(KEY_LEFT_CTRL);
  Keyboard.press('1');
  delay(100);
  Keyboard.releaseAll();
}

void macro4() { // CHAT - Open Chat
  Keyboard.press(KEY_LEFT_ALT);
  Keyboard.press(KEY_TAB);
  delay(100);
  Keyboard.releaseAll();
}

void macro5() { // SAVE - Ctrl+S
  Keyboard.press(KEY_LEFT_CTRL);
  Keyboard.press('s');
  delay(100);
  Keyboard.releaseAll();
}

void macro6() { // BUILD - F5 (Visual Studio)
  Keyboard.press(KEY_F5);
  delay(100);
  Keyboard.releaseAll();
}

void macro7() { // DEBUG - F9 (Breakpoint)
  Keyboard.press(KEY_F9);
  delay(100);
  Keyboard.releaseAll();
}

void macro8() { // RGB - Cycle RGB Mode
  static int rgbMode = 0;
  rgbMode = (rgbMode + 1) % 3;
  
  switch (rgbMode) {
    case 0: // Normal colors
      for (int i = 0; i < 8; i++) {
        leds[i] = macros[i].color;
      }
      break;
    case 1: // Rainbow
      for (int i = 0; i < 8; i++) {
        leds[i] = CHSV(i * 32, 255, 255);
      }
      break;
    case 2: // All white
      fill_solid(leds, NUM_LEDS, CRGB::White);
      break;
  }
  FastLED.show();
}

// Volume Control Functions
void volumeUp() {
  Keyboard.press(KEY_MEDIA_VOLUME_UP);
  delay(50);
  Keyboard.releaseAll();
}

void volumeDown() {
  Keyboard.press(KEY_MEDIA_VOLUME_DOWN);
  delay(50);
  Keyboard.releaseAll();
}

void volumeMute() {
  Keyboard.press(KEY_MEDIA_MUTE);
  delay(50);
  Keyboard.releaseAll();
}