// Arduino Plasma Toilet Air Freshener
// Version 1.4
// Vije Miller

#include <FastLED.h>
#include <Servo.h>

// Neopixel
#define DATA_PIN 2
#define LED_TYPE WS2812
#define COLOR_ORDER GRB
#define NUM_LEDS 3
#define BRIGHTNESS 100
#define FRAMES_PER_SECOND 120
CRGB leds[NUM_LEDS];

// Servos
Servo gyro;
Servo riser;
Servo level;

// Gyro
int gyro_pos = 95;
int gyro_stop = 95;
int rotate = 77;
int timer_rotate_initial = 3000;
int timer_rotate = 9150;
// hall sensor in next version

// Riser
int rise = 105;
int lower = 75;
int riser_stop = 91;
int timer_riser_up = 20;
int timer_riser_down = 20;

// Level
int level_pos = 125;
int up = 93;
int down = 125;

// Relay
int relay = 3;

// Trigger
// Mercury Switch
int trigger = 7;
int led = 8;
int state = 0;
int hold_state = 0;

void setup() {

  Serial.begin(9600);

  pinMode(trigger, INPUT);
  pinMode(led, OUTPUT);

  pinMode(relay, OUTPUT);
  digitalWrite(relay, LOW);

  gyro.attach(5);
  gyro.write(gyro_stop);

  riser.attach(6);
  riser.write(riser_stop);

  level.attach(4);
  level.write(down);

  FastLED.addLeds(leds, NUM_LEDS).setCorrection(TypicalLEDStrip);
  FastLED.setBrightness(BRIGHTNESS);

  // Alert Active
  colorFWD(0x00, 0x00, 0xff, 100);
  colorFWD(0x00, 0x00, 0x00, 100);
  colorFWD(0x00, 0x00, 0xff, 100);
  colorFWD(0x00, 0x00, 0x00, 100);

}

void loop() {

  // Clear Strip
  colorFWD(0x00, 0x00, 0x00, 100);

  // Read Button
  state = digitalRead(trigger);

  // Lid Down
  if (state == LOW) {
    digitalWrite(led, LOW);
    // Lid Down After Open Then..
    if (hold_state == 1) {
      delay(5000);
      freshen();
      hold_state = 0; // reset
    }
    // Lid Open
  } else if (state == HIGH) {
    hold_state = 1;
    digitalWrite(led, HIGH);
  }

  // Testing
  if (Serial.available() > 0)
  {
    int cmd = Serial.parseInt();
    Serial.println(cmd);
    if (cmd == 1) {
      rise_up();
    } else if (cmd == 2) {
      go_down();
    } else if (cmd == 3) {
      turn_gyro();
    } else if (cmd == 4) {
      freshen();
    }
  }

  delay(1);
}

// Whole Shabang
void freshen() {
  rise_up();
  delay(2000);
  turn_gyro();
  delay(2000);
  go_down();
}

void rise_up() {
  for (int x = 0; x < timer_riser_up; x += 1) {
    riser.write(rise);
    track_led();
    // Convert to Leveler Servo
    level_pos = map(x, 0, timer_riser_up, down, up);
    level_pos = round(level_pos);
    level.write(level_pos);
    delay(500);
  }
  riser.write(riser_stop);
}

void go_down() {
  for (int x = 0; x < timer_riser_down; x += 1) {
    riser.write(lower);
    track_led();
    level_pos = map(x, 0, timer_riser_down, up, down);
    level_pos = round(level_pos);
    level.write(level_pos);
    delay(500);
  }
  riser.write(riser_stop);
}

void turn_gyro() {
  // Spin Up Gyro
  for (gyro_pos = gyro_stop; gyro_pos >= rotate; gyro_pos -= 1) {
    gyro.write(gyro_pos);
    delay(50);
  }
  delay(timer_rotate_initial);

  // Activate Spark
  digitalWrite(relay, HIGH);
  delay(timer_rotate);
  digitalWrite(relay, LOW);

  gyro.write(gyro_stop);
}

void track_led() {
  colorFWD(0x00, 0x00, 0xff, 100);
  colorFWD(0x00, 0x00, 0x00, 100);
}

void colorFWD(byte red, byte green, byte blue, int SpeedDelay) {
  for (uint16_t i = 0; i < NUM_LEDS; i++) {
    setPixel(i, red, green, blue);
    showStrip();
    delay(SpeedDelay);
  }
}

void showStrip() {
#ifdef ADAFRUIT_NEOPIXEL_H
  // NeoPixel
  strip.show();
#endif
#ifndef ADAFRUIT_NEOPIXEL_H
  // FastLED
  FastLED.show();
#endif
}

void setPixel(int Pixel, byte red, byte green, byte blue) {
#ifdef ADAFRUIT_NEOPIXEL_H
  // NeoPixel
  strip.setPixelColor(Pixel, strip.Color(red, green, blue));
#endif
#ifndef ADAFRUIT_NEOPIXEL_H
  // FastLED
  leds[Pixel].r = red;
  leds[Pixel].g = green;
  leds[Pixel].b = blue;
#endif
}