// sink starter
// vije miller
// updated with button restore
// updated with touchless flush + sink

#include "FastLED.h"
#include <VarSpeedServo.h>

#define NUM_LEDS 90
#define LED_TYPE WS2812B
#define COLOR_ORDER GRB
CRGB leds[NUM_LEDS];
int data = 255;
int r = 0;
int g = 0;
int b = 255;
int j = 0;
int k = 0;
int red_ran;
int blue_ran;
int green_ran;

VarSpeedServo left;
VarSpeedServo l_lift;
VarSpeedServo right;
VarSpeedServo r_lift;
VarSpeedServo toilet_flush;

//pins
const int toilet_pin = 2;
const int left_pin = 3;
const int l_lift_pin = 5;
const int right_pin = 7;
const int r_lift_pin = 9;
const int motion = 11;
const int toilet_motion = 14;
#define PIN 16
const int button = 18;

const int rldown = 100;
const int lldown = 80;
const int rlup = 6;
const int llup = 180;
const int lup = 10;
const int rup = 170;
const int ldown_second = 92;
const int rdown_second = 100;
const int toilet_up = 55;
const int toilet_down = 10;

int sink_moved;
int toilet_moved;

void setup() {

  pinMode(motion, INPUT);
  pinMode(toilet_motion, INPUT);
  pinMode(button, INPUT);

  FastLED.addLeds<LED_TYPE, PIN, COLOR_ORDER>(leds, NUM_LEDS).setCorrection( TypicalLEDStrip );

  // wait for user activation
  // power outage safety
  while (digitalRead(button) == LOW) {
    neoshow(255, 0, 0, 150);
    delay(500);
    neoshow(0, 0, 0, 150);
    delay(500);
  }

  l_lift.attach(l_lift_pin);
  r_lift.attach(r_lift_pin);
  delay(1000); // default error fix
  right.attach(right_pin);
  left.attach(left_pin);
  delay(500);

  l_lift.write(lldown, 255, false);
  r_lift.write(rldown, 255, true);
  delay(1000);
  left.write(lup, 100, false);
  right.write(rup, 100, true);
  delay(1000);
  l_lift.write(llup, 180, false);
  r_lift.write(rlup, 180, true);
  delay(500);

  toilet_flush.attach(toilet_pin);
  delay(500);
  toilet_flush.write(toilet_down, 180, true);
  delay(500);

  l_lift.detach();
  r_lift.detach();
  left.detach();
  right.detach();
  toilet_flush.detach();

  colorWipe(0x00, 0x00, 0xff, 50);
  neoshow(0, 0, 255, 220);
}

void loop() {
  // if motion
  sink_moved = digitalRead(motion);
  if (sink_moved == LOW) {
    doit();
  }
  toilet_moved = digitalRead(toilet_motion);
  if (toilet_moved == LOW) {
    poopie();
    doit();
  }
}

void doit() {
  // reconnect pins
  l_lift.attach(l_lift_pin);
  r_lift.attach(r_lift_pin);
  right.attach(right_pin);
  left.attach(left_pin);
  // give it a chance
  delay(100);
  // lift down first
  l_lift.write(lldown, 255, false);
  r_lift.write(rldown, 255, true);
  delay(500);
  // push
  left.write(ldown_second, 200, false);
  right.write(rdown_second, 200, true);
  delay(500);
  // return
  left.write(lup, 200, false);
  right.write(rup, 200, true);
  delay(500);
  // lift
  l_lift.write(llup, 100, false);
  r_lift.write(rlup, 100, true);
  delay(500);
  // disconnect pins
  l_lift.detach();
  r_lift.detach();
  left.detach();
  right.detach();
  // pavlov
  lightshow();
}

void poopie() {
  toilet_flush.attach(toilet_pin);
  delay(100);
  toilet_flush.write(toilet_up, 180, true);
  delay(3000);
  toilet_flush.write(toilet_down, 180, true);
  delay(100);
  toilet_flush.detach();
}

void lightshow() {
  j = random(1, 3);
  if (j == 1) {
    for (k = 0; k < 4; k++) {
      TwinkleRandom(20, 100, false);
    }
  } else if (j == 2) {
    for (k = 0; k < 1; k++) {
      rainbowCycle(20);
    }
  } else if (j == 3) {
    for (k = 0; k < 2; k++) {
      red_ran = random(255);
      green_ran = random(255);
      blue_ran = random(255);
      if (red_ran > 200 && green_ran > 200 && blue_ran > 200) {
        red_ran = random(0, 80);
      }
      colorWipe(red_ran, green_ran, blue_ran, 50);
      colorWipe(0, 0, 0, 50);
    }
  }
  red_ran = random(255);
  green_ran = random(255);
  blue_ran = random(255);
  if (red_ran > 200 && green_ran > 200 && blue_ran > 200) {
    red_ran = random(0, 80);
  }
  colorWipe(0, 0, 0, 50);
  colorWipe(red_ran, green_ran, blue_ran, 50);
  neoshow(red_ran, green_ran, blue_ran, 200);
}

void neoshow(int r, int g, int b, int brightness) {
  FastLED.setBrightness(brightness);
  for (int i = 0; i < NUM_LEDS; i++ )
  {
    leds[i] = CRGB(r, g, b);
  }
  FastLED.show();
}

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
}

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

void rainbowCycle(int SpeedDelay) {
  byte *c;
  uint16_t i, j;
  for (j = 0; j < 256 * 5; j++) { // 5 cycles of all colors on wheel
    for (i = 0; i < NUM_LEDS; i++) {
      c = Wheel(((i * 256 / NUM_LEDS) + j) & 255);
      setPixel(i, *c, *(c + 1), *(c + 2));
    }
    showStrip();
    delay(SpeedDelay);
  }
}

byte * Wheel(byte WheelPos) {
  static byte c[3];
  if (WheelPos < 85) {
    c[0] = WheelPos * 3;
    c[1] = 255 - WheelPos * 3;
    c[2] = 0;
  } else if (WheelPos < 170) {
    WheelPos -= 85;
    c[0] = 255 - WheelPos * 3;
    c[1] = 0;
    c[2] = WheelPos * 3;
  } else {
    WheelPos -= 170;
    c[0] = 0;
    c[1] = WheelPos * 3;
    c[2] = 255 - WheelPos * 3;
  }
  return c;
}

void TwinkleRandom(int Count, int SpeedDelay, boolean OnlyOne) {
  setAll(0, 0, 0);
  for (int i = 0; i < Count; i++) {
    setPixel(random(NUM_LEDS), random(0, 255), random(0, 255), random(0, 255));
    showStrip();
    delay(SpeedDelay);
    if (OnlyOne) {
      setAll(0, 0, 0);
    }
  }
  delay(SpeedDelay);
}

void colorWipe(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);
  }
}